Better error handling for empty responses

monarch-initiative · May 31, 2024 · dd22be0 · dd22be0
1 parent 0efa39a
commit dd22be0
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/ontogpt/clients/llm_client.py b/src/ontogpt/clients/llm_client.py
@@ -63,7 +63,11 @@ def complete(self, prompt, show_prompt: bool = False, **kwargs) -> str:
         except Exception as e:
             print(f"Encountered error: {type(e)}, Error: {e}")
 
-        payload = response.choices[0].message.content
+        if response is not None:
+            payload = response.choices[0].message.content
+        else:
+            logger.error(f"No response or response is empty.")
+            payload = ""
 
         return payload