feat(EXCEPTION_HANDLING): :bug: Implement general exception handler and improve error responses for text generation and summarization

app.py

@@ -3,6 +3,7 @@ from pydantic import BaseModel
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 from functools import partial
+from fastapi.responses import JSONResponse
 
 from langchain_core.messages import HumanMessage, AIMessage
 from langgraph.checkpoint.memory import MemorySaver
@@ -121,6 +122,14 @@ class SummaryRequest(BaseModel):
 # Create the FastAPI application
 app = FastAPI(title="LangChain FastAPI", description="API to generate text using LangChain and LangGraph - Máximo Fernández Núñez IriusRisk test challenge")
 
+# Add general exception handler
+@app.exception_handler(Exception)
+async def general_exception_handler(request, exc):
+    return JSONResponse(
+        status_code=500,
+        content={"error": f"Error interno: {str(exc)}", "type": type(exc).__name__}
+    )
+
 # Welcome endpoint
 @app.get("/")
 async def api_home():
@@ -170,7 +179,14 @@ async def generate(request: QueryRequest):
             "thread_id": request.thread_id
         }
     except Exception as e:
-        raise HTTPException(status_code=500, detail=f"Error generating text: {str(e)}")
+        return JSONResponse(
+            status_code=500,
+            content={
+                "error": f"Error generando texto: {str(e)}", 
+                "type": type(e).__name__,
+                "thread_id": request.thread_id
+            }
+        )
 
 @app.post("/summarize")
 async def summarize(request: SummaryRequest):
@@ -216,7 +232,14 @@ async def summarize(request: SummaryRequest):
             "thread_id": request.thread_id
         }
     except Exception as e:
-        raise HTTPException(status_code=500, detail=f"Error generating summary: {str(e)}")
+        return JSONResponse(
+            status_code=500,
+            content={
+                "error": f"Error generando resumen: {str(e)}", 
+                "type": type(e).__name__,
+                "thread_id": request.thread_id
+            }
+        )
 
 if __name__ == "__main__":
     import uvicorn