sources only

2025-12-25 02:51:49 +01:00
parent 29b111080f
commit b156094691
5 changed files with 98 additions and 55 deletions
--- a/web/agent/src/agent.py
+++ b/web/agent/src/agent.py
@@ -87,6 +87,8 @@ Rules:
 7. Use tools sparingly—one search usually suffices.
 8. If you hit the search limit, end your reply with an italicized note: *Your question may be too broad. Try asking something more specific.* Do NOT mention "tools" or "tool limits"—the user doesn't know what those are."""

+SOURCES_ONLY_INSTRUCTIONS = """SOURCES ONLY MODE: Give a 1-2 sentence summary maximum. Focus on listing sources in a bulleted list. No detailed explanations."""
+

 def create_tool_call_limiter(max_calls: int = 3):
    """Create a process_tool_call callback that limits tool calls."""
@@ -110,7 +112,7 @@ def create_tool_call_limiter(max_calls: int = 3):
    return process_tool_call


-def create_agent(user_roles: list[str] | None = None):
+def create_agent(user_roles: list[str] | None = None, sources_only: bool = False):
    """Create an agent with MCP tools configured for the given user roles."""
    toolsets = []

@@ -140,10 +142,15 @@ def create_agent(user_roles: list[str] | None = None):
    else:
        logger.info("MCP server unavailable - running without MCP tools")

+    # Build instructions based on mode
+    instructions = AGENT_INSTRUCTIONS
+    if sources_only:
+        instructions = f"{SOURCES_ONLY_INSTRUCTIONS}\n\n{AGENT_INSTRUCTIONS}"
+
    return Agent(
        model="anthropic:claude-sonnet-4-5",
        toolsets=toolsets if toolsets else None,
-        instructions=AGENT_INSTRUCTIONS,
+        instructions=instructions,
        history_processors=[limit_history],
        model_settings=ModelSettings(max_tokens=4096),
    )
--- a/web/agent/src/main.py
+++ b/web/agent/src/main.py
@@ -67,8 +67,13 @@ async def handle_agent_request(request: Request) -> Response:
        except json.JSONDecodeError as e:
            logger.warning(f"Failed to parse x-user-roles header: {e}")

-    # Create agent with the user's roles
-    agent = create_agent(user_roles)
+    # Extract sources-only mode from header
+    sources_only = request.headers.get("x-sources-only", "false") == "true"
+    if sources_only:
+        logger.info("Sources-only mode enabled")
+
+    # Create agent with the user's roles and mode
+    agent = create_agent(user_roles, sources_only=sources_only)

    # Dispatch the request - tool limits handled by ToolCallLimiter in agent.py
    return await AGUIAdapter.dispatch_request(
--- a/web/src/app/api/copilotkit/route.ts
+++ b/web/src/app/api/copilotkit/route.ts
@@ -15,13 +15,19 @@ export const POST = async (req: NextRequest) => {
  const session = await auth0.getSession();
  const userRoles = (session?.user?.roles as string[]) || [];

-  console.log("DEBUG: User roles from session:", userRoles);
+  // Get sources-only mode from query param
+  const url = new URL(req.url);
+  const sourcesOnly = url.searchParams.get("sourcesOnly") === "true";

-  // Create HttpAgent with user roles header
+  console.log("DEBUG: User roles from session:", userRoles);
+  console.log("DEBUG: Sources only mode:", sourcesOnly);
+
+  // Create HttpAgent with user roles and sources-only headers
  const agent = new HttpAgent({
    url: process.env.AGENT_URL || "http://localhost:8000/",
    headers: {
      "x-user-roles": JSON.stringify(userRoles),
+      "x-sources-only": sourcesOnly ? "true" : "false",
    },
  });

--- a/web/src/app/layout.tsx
+++ b/web/src/app/layout.tsx
@@ -1,6 +1,5 @@
 import type { Metadata } from "next";

-import { CopilotKit } from "@copilotkit/react-core";
 import { Auth0Provider } from "@auth0/nextjs-auth0/client";
 import "./globals.css";
 import "@copilotkit/react-ui/styles.css";
@@ -19,9 +18,7 @@ export default function RootLayout({
    <html lang="en">
      <body className={"antialiased"}>
        <Auth0Provider>
-          <CopilotKit runtimeUrl="/api/copilotkit" agent="vpi_1000">
-            {children}
-          </CopilotKit>
+          {children}
        </Auth0Provider>
      </body>
    </html>
--- a/web/src/app/page.tsx
+++ b/web/src/app/page.tsx
@@ -1,6 +1,6 @@
 "use client";

-import { useCopilotAction, useCopilotChat } from "@copilotkit/react-core";
+import { CopilotKit, useCopilotAction, useCopilotChat } from "@copilotkit/react-core";
 import { CopilotKitCSSProperties, CopilotChat } from "@copilotkit/react-ui";
 import { useState } from "react";
 import { useUser } from "@auth0/nextjs-auth0/client";
@@ -36,10 +36,8 @@ function LoadingOverlay() {
  }
 }

-export default function CopilotKitPage() {
-  const [themeColor, setThemeColor] = useState("#6366f1");
-  const { user, isLoading: authLoading } = useUser();
-
+// Chat content with CopilotKit hooks - must be inside CopilotKit provider
+function ChatContent({ themeColor, setThemeColor }: { themeColor: string; setThemeColor: (color: string) => void }) {
  useCopilotAction({
    name: "setThemeColor",
    parameters: [{
@@ -52,6 +50,35 @@ export default function CopilotKitPage() {
    },
  });

+  return (
+    <div
+      className="flex-1 flex justify-center py-8 px-2 overflow-hidden relative"
+      style={{ "--copilot-kit-primary-color": themeColor } as CopilotKitCSSProperties}
+    >
+      <div className="h-full w-full max-w-5xl flex flex-col">
+        <CopilotChat
+          labels={{
+            title: "AI Cartwright",
+            initial: "Hello! I'm here to help with anything related to caving. Ask me about caves, techniques, safety, equipment, or anything else caving-related!",
+          }}
+          className="h-full w-full"
+        />
+      </div>
+      <LoadingOverlay />
+    </div>
+  );
+}
+
+export default function CopilotKitPage() {
+  const [themeColor, setThemeColor] = useState("#6366f1");
+  const [sourcesOnlyMode, setSourcesOnlyMode] = useState(false);
+  const { user, isLoading: authLoading } = useUser();
+
+  // Dynamic runtime URL based on sources-only mode
+  const runtimeUrl = sourcesOnlyMode
+    ? "/api/copilotkit?sourcesOnly=true"
+    : "/api/copilotkit";
+
  // Show loading state while checking authentication
  if (authLoading) {
    return (
@@ -88,50 +115,51 @@ export default function CopilotKitPage() {

  // If authenticated, show the CopilotKit chat with user profile
  return (
-    <main
-      style={{ "--copilot-kit-primary-color": themeColor } as CopilotKitCSSProperties}
-      className="h-screen w-screen flex flex-col bg-gray-50"
+    <CopilotKit
+      runtimeUrl={runtimeUrl}
+      agent="vpi_1000"
+      key={sourcesOnlyMode ? "sources" : "normal"}
    >
-      {/* Header with user profile and logout */}
-      <div className="w-full bg-white shadow-sm border-b border-gray-200 px-4 py-3">
-        <div className="max-w-7xl mx-auto flex justify-between items-center">
-          <div className="flex items-center gap-4">
-            <h1 className="text-xl font-semibold text-gray-900">Cavepedia</h1>
-          </div>
-          <div className="flex items-center gap-4">
-            {user.picture && (
-              <img
-                src={user.picture}
-                alt={user.name || 'User'}
-                className="w-8 h-8 rounded-full"
-              />
-            )}
-            <div className="flex flex-col items-end">
-              <span className="text-sm text-gray-700">{user.name}</span>
-              {(user as any).roles && (user as any).roles.length > 0 && (
-                <span className="text-xs text-gray-500">
-                  {(user as any).roles.join(', ')}
-                </span>
-              )}
+      <main className="h-screen w-screen flex flex-col bg-gray-50">
+        {/* Header with user profile and logout */}
+        <div className="w-full bg-white shadow-sm border-b border-gray-200 px-4 py-3">
+          <div className="max-w-7xl mx-auto flex justify-between items-center">
+            <div className="flex items-center gap-4">
+              <h1 className="text-xl font-semibold text-gray-900">Cavepedia</h1>
+              <label className="flex items-center gap-2 text-sm text-gray-600 cursor-pointer">
+                <input
+                  type="checkbox"
+                  checked={sourcesOnlyMode}
+                  onChange={(e) => setSourcesOnlyMode(e.target.checked)}
+                  className="w-4 h-4 rounded border-gray-300 text-indigo-600 focus:ring-indigo-500"
+                />
+                Sources only
+              </label>
+            </div>
+            <div className="flex items-center gap-4">
+              {user.picture && (
+                <img
+                  src={user.picture}
+                  alt={user.name || 'User'}
+                  className="w-8 h-8 rounded-full"
+                />
+              )}
+              <div className="flex flex-col items-end">
+                <span className="text-sm text-gray-700">{user.name}</span>
+                {(user as any).roles && (user as any).roles.length > 0 && (
+                  <span className="text-xs text-gray-500">
+                    {(user as any).roles.join(', ')}
+                  </span>
+                )}
+              </div>
+              <LogoutButton />
            </div>
-            <LogoutButton />
          </div>
        </div>
-      </div>

-      {/* CopilotKit Chat */}
-      <div className="flex-1 flex justify-center py-8 px-2 overflow-hidden relative">
-        <div className="h-full w-full max-w-5xl flex flex-col">
-          <CopilotChat
-            labels={{
-              title: "AI Cartwright",
-              initial: "Hello! I'm here to help with anything related to caving. Ask me about caves, techniques, safety, equipment, or anything else caving-related!",
-            }}
-            className="h-full w-full"
-          />
-        </div>
-        <LoadingOverlay />
-      </div>
-    </main>
+        {/* CopilotKit Chat */}
+        <ChatContent themeColor={themeColor} setThemeColor={setThemeColor} />
+      </main>
+    </CopilotKit>
  );
 }