def generate_completion()

in distilvit/curate_api.py [0:0]

• 9 lines of code
• 2 McCabe index (conditional complexity)

    def generate_completion(self, prompt):
        url = f"{self.base_url}/api/generate"
        headers = {"Content-Type": "application/json"}
        data = {"model": self.model, "prompt": prompt, "stream": False}
        response = requests.post(url, headers=headers, json=data)
        if response.status_code == 200:
            return response.json()["response"]
        else:
            return f"Error: {response.status_code}, {response.text}"