use pydantic

web/agent/server.py

@@ -1,35 +1,18 @@
 """
-Self-hosted LangGraph agent server using AG-UI protocol.
+Self-hosted PydanticAI agent server using AG-UI protocol.
 """
 
 import os
-from fastapi import FastAPI
 import uvicorn
 from dotenv import load_dotenv
 
-from copilotkit import LangGraphAGUIAgent
-from ag_ui_langgraph import add_langgraph_fastapi_endpoint
-from main import graph
+from pydantic_ai.ui.ag_ui.app import AGUIApp
+from main import agent
 
 load_dotenv()
 
-app = FastAPI(title="Cavepedia Agent")
-
-add_langgraph_fastapi_endpoint(
-    app=app,
-    agent=LangGraphAGUIAgent(
-        name="vpi_1000",
-        description="AI assistant with access to cave-related information through the Cavepedia MCP server",
-        graph=graph,
-    ),
-    path="/",
-)
-
-
-@app.get("/health")
-def health():
-    """Health check."""
-    return {"status": "ok"}
+# Convert PydanticAI agent to ASGI app with AG-UI protocol
+app = AGUIApp(agent)
 
 
 if __name__ == "__main__":