refactor: Handle errors and refusals in chat responses

provos · Oct 24, 2024 · 5c34bd3 · 5c34bd3
1 parent f892a53
commit 5c34bd3
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/planai/llm_interface.py b/src/planai/llm_interface.py
@@ -89,6 +89,11 @@ def _cached_chat(
             # Cache the response with hashed prompt as key
             self.disk_cache.set(prompt_hash, response)
 
+        if "error" in response:
+            self.logger.error("Error in chat response: %s", response["error"])
+        elif "refusal" in response:
+            self.logger.error("Model refused the request: %s", response["refusal"])
+
         return response["message"]["content"]
 
     def chat(