cavepedia/cavepediav2
1
0
Fork 0
Code Issues 1 Pull Requests Actions Packages Projects 1 Releases Wiki Activity

sources only

Browse Source
This commit is contained in:
Paul Walko
2025-12-25 02:51:49 +01:00
parent 29b111080f
commit b156094691
5 changed files with 98 additions and 55 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
web/agent/src/agent.py
View File

@@ -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),
)

9
web/agent/src/main.py
View File

@@ -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(

10
web/src/app/api/copilotkit/route.ts
View File

@@ -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",
},
});

3
web/src/app/layout.tsx
View File

@@ -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>
</Auth0Provider>
</body>
</html>

68
web/src/app/page.tsx
View File

@@ -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,15 +115,26 @@ 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"}
>
<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 && (
@@ -120,18 +158,8 @@ export default function CopilotKitPage() {
</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>
<ChatContent themeColor={themeColor} setThemeColor={setThemeColor} />
</main>
</CopilotKit>
);
}