check status codes

CopilotC-Nvim · Feb 1, 2024 · f3f14de · f3f14de
1 parent f8823af
commit f3f14de
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/rplugin/python3/copilot.py b/rplugin/python3/copilot.py
@@ -103,6 +103,8 @@ def ask(self, system_prompt: str, prompt: str, code: str, language: str = ""):
         response = self.session.post(
             url, headers=self._headers(), json=data, stream=True
         )
+        if response.status_code != 200:
+            raise Exception(f"Error fetching response: {response}")
         for line in response.iter_lines():
             line = line.decode("utf-8").replace("data: ", "").strip()
             if line.startswith("[DONE]"):