fix web errors

web/agent/pyproject.toml

@@ -16,4 +16,5 @@ dependencies = [
     "docstring-parser>=0.17.0",
     "jsonschema>=4.25.1",
     "copilotkit>=0.1.0",
+    "ag-ui-langgraph>=0.0.4",
 ]

web/agent/server.py

@@ -1,5 +1,5 @@
 """
-Self-hosted LangGraph agent server using CopilotKit.
+Self-hosted LangGraph agent server using CopilotKit AG-UI protocol.
 """
 
 import os
@@ -7,38 +7,23 @@ from fastapi import FastAPI
 import uvicorn
 from dotenv import load_dotenv
 
-from copilotkit import CopilotKitRemoteEndpoint, LangGraphAgent
-from copilotkit.integrations.fastapi import add_fastapi_endpoint
+from copilotkit import LangGraphAGUIAgent
+from ag_ui_langgraph import add_langgraph_fastapi_endpoint
 from main import graph
 
 load_dotenv()
 
 app = FastAPI(title="Cavepedia Agent")
 
-
-def build_agents(context):
-    """Build agents with auth context from properties."""
-    # Get user roles from frontend properties
-    user_roles = context.get("properties", {}).get("auth0_user_roles", [])
-
-    return [
-        LangGraphAgent(
-            name="vpi_1000",
-            description="AI assistant with access to cave-related information through the Cavepedia MCP server",
-            graph=graph,
-            langgraph_config={
-                "configurable": {
-                    "context": {
-                        "auth0_user_roles": user_roles,
-                    }
-                }
-            },
-        )
-    ]
-
-
-sdk = CopilotKitRemoteEndpoint(agents=build_agents)
-add_fastapi_endpoint(app, sdk, "/copilotkit")
+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")

web/src/app/api/copilotkit/route.ts

@@ -2,7 +2,6 @@ import {
   CopilotRuntime,
   ExperimentalEmptyAdapter,
   copilotRuntimeNextJSAppRouterEndpoint,
-  LangGraphHttpAgent,
 } from "@copilotkit/runtime";
 
 import { NextRequest } from "next/server";
@@ -10,11 +9,11 @@ import { NextRequest } from "next/server";
 const serviceAdapter = new ExperimentalEmptyAdapter();
 
 const runtime = new CopilotRuntime({
-  agents: {
-    vpi_1000: new LangGraphHttpAgent({
-      url: `${process.env.LANGGRAPH_DEPLOYMENT_URL || "http://localhost:8000"}/copilotkit`,
-    }),
-  },
+  remoteEndpoints: [
+    {
+      url: process.env.LANGGRAPH_DEPLOYMENT_URL || "http://localhost:8000",
+    },
+  ],
 });
 
 export const POST = async (req: NextRequest) => {