agent endpoint updated

app.py

@@ -79,24 +79,43 @@ async def upload_dataframe(file: UploadFile = File(...), styling_instructions: s
 @app.post("/chat/{agent_name}", response_model=dict)
 async def chat_with_agent(agent_name: str, request: QueryRequest):
     if agent_name not in AVAILABLE_AGENTS:
-        raise HTTPException(status_code=404, detail="Agent not found")
+        raise HTTPException(status_code=404, detail=f"Agent '{agent_name}' not found. Available agents: {list(AVAILABLE_AGENTS.keys())}")
     
-    agent = AVAILABLE_AGENTS[agent_name]
-    response_text = agent(request.query)
-    return {
-        "agent_name": agent_name,
-        "query": request.query,
-        "response": response_text,
-    }
+    try:
+        print("agent_name: ", agent_name)
+        agent = AVAILABLE_AGENTS[agent_name]
+        response = agent(request.query)
+        # Format the response using the same formatter as the main chat endpoint
+        formatted_response = format_response_to_markdown(response)
+        
+        return {
+            "agent_name": agent_name,
+            "query": request.query,
+            "response": formatted_response,
+        }
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
 
 @app.post("/chat", response_model=dict)
 async def chat_with_all(request: QueryRequest):
-    response_text = ai_system(request.query)
-    response_text = format_response_to_markdown(response_text)
+    try:
+        response = ai_system(request.query)
+        formatted_response = format_response_to_markdown(response)
+        
+        return {
+            "agent_name": "ai_system",
+            "query": request.query,
+            "response": formatted_response,
+        }
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+# Add an endpoint to list available agents
+@app.get("/agents", response_model=dict)
+async def list_agents():
     return {
-        "agent_name": "ai_system",
-        "query": request.query,
-        "response": response_text,
+        "available_agents": list(AVAILABLE_AGENTS.keys()),
+        "description": "List of available specialized agents that can be called using @agent_name"
     }
 
 @app.get("/health", response_model=dict)