Patch for controllers/chat.py

[beetle-ai]

Added a try...except block around the model.generate_content_async call to catch potential exceptions.Changes made:

• Replaced: response = await model.generate_content_async(prompt) chat_response = response.text
• With: try: response = await model.generate_content_async(prompt) chat_response = response.text except Exce...

Related Issue: #2c9c3a1a-2b3a-4b1a-8b3a-2b3a4b1a8b3a

File: controllers/chat.py
Branch: fix/1759410390915-aqib78 → main

[beetle-ai]

🤖 CodeDetector Analysis

✅ Improved Error Handling for Gemini API Call

File: controllers/chat.py
Lines: 137-141
Severity: Medium

Problem

The original code lacked error handling around the call to the Gemini API, which could lead to unhandled exceptions and potentially crash the application or provide uninformative error messages to the user.

Current Code

response = await model.generate_content_async(prompt)
chat_response = response.text

Suggested Fix

try:
response = await model.generate_content_async(prompt)
chat_response = response.text
except Exception as e:
print(f"Error calling Gemini API: {e}")
raise HTTPException(status_code=500, detail="Failed to get response from chat API")

Why This Fix Works

• Prevents unhandled exceptions from crashing the application.
• Logs the specific error message for debugging purposes.
• Returns a more informative HTTPException to the client, indicating that the chat API failed.

Additional Context

This change improves the robustness and user experience of the application by gracefully handling potential errors during the Gemini API call.

---

Powered by CodeDetector - AI-powered code analysis