revert some stuff

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 remote endpoint protocol.
+Self-hosted LangGraph agent server using AG-UI protocol.
 """
 
 import os
@@ -7,24 +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")
 
-sdk = CopilotKitRemoteEndpoint(
-    agents=[
-        LangGraphAgent(
-            name="vpi_1000",
-            description="AI assistant with access to cave-related information through the Cavepedia MCP server",
-            graph=graph,
-        )
-    ]
+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="/",
 )
-add_fastapi_endpoint(app, sdk, "/copilotkit")
 
 
 @app.get("/health")

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

@@ -4,16 +4,19 @@ import {
   copilotRuntimeNextJSAppRouterEndpoint,
 } from "@copilotkit/runtime";
 
+import { LangGraphAgent } from "@ag-ui/langgraph";
 import { NextRequest } from "next/server";
 
 const serviceAdapter = new ExperimentalEmptyAdapter();
 
 const runtime = new CopilotRuntime({
-  remoteEndpoints: [
-    {
-      url: `${process.env.LANGGRAPH_DEPLOYMENT_URL || "http://localhost:8000"}/copilotkit`,
-    },
-  ],
+  agents: {
+    vpi_1000: new LangGraphAgent({
+      deploymentUrl: process.env.LANGGRAPH_DEPLOYMENT_URL || "http://localhost:8000",
+      graphId: "vpi_1000",
+      langsmithApiKey: process.env.LANGSMITH_API_KEY || "",
+    }),
+  },
 });
 
 export const POST = async (req: NextRequest) => {