def call_gpt4o_mini_endpoint()

in Lab2_nlp_evaluation/target_nlp_api/target_nlp_api.py [0:0]

• 8 lines of code
• 1 McCabe index (conditional complexity)

    def call_gpt4o_mini_endpoint(self: Self, query: str) -> Response:
        endpoint = self.env["gpt-4o-mini"]["endpoint"]
        key = self.env["gpt-4o-mini"]["key"]

        headers = {"Content-Type": "application/json", "api-key": key}

        payload = {"messages": [{"role": "user", "content": query}], "max_tokens": 500}

        output = self.query(endpoint=endpoint, headers=headers, payload=payload)
        response = output["choices"][0]["message"]["content"]
        return {"query": query, "response": response}