jpmschweitzerandClaude Opus 4.5
100ebeae52
Move web search functionality to The Librarian agent, integrating with the library-desk /rag/search endpoint for enhanced search capabilities. Changes: - Add search_web, read_url, read_urls_batch tools to Librarian - Add WebSearchResult, ContentExtractionResult models to client - Add search_web, extract_content, extract_content_batch client methods - Update Librarian capability with web/url/internet domains - Remove search_web from tatlock_core tools and toolset - Update Tatlock system prompt to delegate web search to Librarian - Add comprehensive unit tests for new Librarian tools - Clean up legacy src/agents/tools.py 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
91 lines
2.6 KiB
Python
91 lines
2.6 KiB
Python
"""
|
|
Librarian capability registration for the Household Registry.
|
|
|
|
Defines The Librarian's capabilities and registers it as a
|
|
household member for coordination by the Steward and Tatlock.
|
|
"""
|
|
from src.agents.librarian.agent import get_librarian_agent
|
|
from src.agents.librarian.tools import LIBRARIAN_TOOLS
|
|
from src.core.household_registry import (
|
|
HouseholdCapability,
|
|
get_household_registry,
|
|
)
|
|
from src.core.logging_config import get_logger
|
|
|
|
logger = get_logger(__name__)
|
|
|
|
|
|
# The Librarian's capability summary for Steward coordination
|
|
LIBRARIAN_CAPABILITY = HouseholdCapability(
|
|
name="librarian",
|
|
role="The Librarian",
|
|
category="research",
|
|
description=(
|
|
"Research, web search, and wiki management: can SEARCH the web for current "
|
|
"information, READ URLs/articles, CREATE wiki pages about topics "
|
|
"(with automatic HybridRAG research), UPDATE existing pages, "
|
|
"and synthesize information from multiple sources. "
|
|
"Use for: 'search for X', 'what is X', 'create a page about X', 'read this URL'"
|
|
),
|
|
domains=[
|
|
"research",
|
|
"knowledge",
|
|
"information",
|
|
"wiki",
|
|
"documents",
|
|
"search",
|
|
"web",
|
|
"url",
|
|
"internet",
|
|
"synthesis",
|
|
"create",
|
|
"write",
|
|
"update",
|
|
],
|
|
cost="medium", # Multiple API calls to library-desk
|
|
requires_network=True, # Needs library-desk API access
|
|
)
|
|
|
|
|
|
def get_librarian_capability() -> HouseholdCapability:
|
|
"""Get The Librarian's capability definition."""
|
|
return LIBRARIAN_CAPABILITY
|
|
|
|
|
|
def register_librarian() -> None:
|
|
"""
|
|
Register The Librarian with the Household Registry.
|
|
|
|
This makes The Librarian available for:
|
|
- Steward recommendations (via capability summary)
|
|
- Tatlock delegation (via agent reference)
|
|
- Tool scoping (via tool list)
|
|
"""
|
|
registry = get_household_registry()
|
|
|
|
# Check if already registered
|
|
if "librarian" in registry:
|
|
logger.debug("librarian_already_registered")
|
|
return
|
|
|
|
registry.register(
|
|
name="librarian",
|
|
capability=LIBRARIAN_CAPABILITY,
|
|
tools=LIBRARIAN_TOOLS,
|
|
agent=get_librarian_agent(),
|
|
)
|
|
|
|
logger.info(
|
|
"librarian_registered",
|
|
role=LIBRARIAN_CAPABILITY.role,
|
|
domains=LIBRARIAN_CAPABILITY.domains,
|
|
tool_count=len(LIBRARIAN_TOOLS),
|
|
)
|
|
|
|
|
|
def unregister_librarian() -> None:
|
|
"""Unregister The Librarian from the Household Registry."""
|
|
registry = get_household_registry()
|
|
registry.unregister("librarian")
|
|
logger.info("librarian_unregistered")
|