sources only mode

mcp/server.py

@@ -49,6 +49,11 @@ def get_user_roles() -> list[str]:
     print("DEBUG: No roles header found, returning empty list")
     return []
 
+def is_sources_only() -> bool:
+    """Check if sources-only mode is enabled via header."""
+    headers = get_http_headers()
+    return headers.get("x-sources-only", "false") == "true"
+
 def embed(text, input_type):
     resp = co.embed(
         texts=[text],
@@ -94,6 +99,7 @@ def search_caving_documents(query: str, priority_prefixes: list[str] | None = No
 
     # Build results with optional priority boost
     docs = []
+    sources_only = is_sources_only()
     for result in rerank_resp.results:
         row = rows[result.index]
         score = result.relevance_score
@@ -104,8 +110,11 @@ def search_caving_documents(query: str, priority_prefixes: list[str] | None = No
             if any(key.startswith(prefix) for prefix in priority_prefixes):
                 score = min(1.0, score * 1.3)
 
-        content = row['content'] or ''
-        docs.append({'key': row['key'], 'content': content, 'relevance': round(score, 3)})
+        if sources_only:
+            docs.append({'key': row['key'], 'relevance': round(score, 3)})
+        else:
+            content = row['content'] or ''
+            docs.append({'key': row['key'], 'content': content, 'relevance': round(score, 3)})
 
     # Re-sort by boosted score and return top_n
     docs.sort(key=lambda x: x['relevance'], reverse=True)

web/agent/src/agent.py

@@ -123,7 +123,10 @@ def create_agent(user_roles: list[str] | None = None, sources_only: bool = False
 
             mcp_server = MCPServerStreamableHTTP(
                 url=CAVE_MCP_URL,
-                headers={"x-user-roles": roles_header},
+                headers={
+                    "x-user-roles": roles_header,
+                    "x-sources-only": "true" if sources_only else "false",
+                },
                 timeout=30.0,
                 process_tool_call=create_search_limiter(),
             )