valid serper response status code

Libr-AI · Apr 18, 2024 · b7d4dfc · b7d4dfc
1 parent aedf0ce
commit b7d4dfc
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/factcheck/core/Retriever/SerperEvidenceRetrieve.py b/factcheck/core/Retriever/SerperEvidenceRetrieve.py
@@ -185,8 +185,12 @@ def _request_serper_api(self, questions):
         questions_data = [{"q": question, "autocorrect": False} for question in questions]
         payload = json.dumps(questions_data)
         response = None
-        try:
-            response = requests.request("POST", url, headers=headers, data=payload)
-        except Exception as e:
-            logger.info(f"Warning, Serper API error: {e}.")
-        return response
+        response = requests.request("POST", url, headers=headers, data=payload)
+
+        if response.status_code == 200:
+            return response
+        elif response.status_code == 403:
+            raise Exception("Failed to authenticate. Check your API key.")
+        else:
+            raise Exception(f"Error occurred: {response.text}")
+