jpmschweitzerandClaude Opus 4.5
f4e8552298
Implements the first Claude-like agent for codebase exploration: Core Features: - Explore agent with glob, grep, read, and bash tools - Native PydanticAI tool calling with Ollama/Mistral Nemo - Sanitized Ollama provider (fixes content:null issue) - REST API endpoints for agent execution Tool Infrastructure: - BaseTool abstract class with ToolResult dataclass - ReadFileTool, GlobFilesTool, GrepContentTool, BashReadOnlyTool - Path validation and sandboxing support CLI Client (separate package for future extraction): - webber-cli command with chat, explore, status commands - Communicates with Webber API backend - Rich console output with theming Configuration: - Dev server on port 8095 (production uses 8086) - Mistral Nemo optimizations (temp 0.3, tool_choice required) Tests: 24 tests covering tools and API endpoints Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
142 lines
4.0 KiB
Python
142 lines
4.0 KiB
Python
"""
|
|
Webber API client.
|
|
|
|
Communicates with the Webber API backend for agent execution.
|
|
"""
|
|
import httpx
|
|
from dataclasses import dataclass
|
|
from typing import Any
|
|
|
|
|
|
@dataclass
|
|
class AgentResponse:
|
|
"""Response from agent execution."""
|
|
response: str
|
|
agent_type: str
|
|
success: bool
|
|
error: str | None = None
|
|
|
|
|
|
@dataclass
|
|
class AgentInfo:
|
|
"""Information about an available agent."""
|
|
name: str
|
|
description: str
|
|
|
|
|
|
class WebberClient:
|
|
"""
|
|
Client for the Webber API.
|
|
|
|
Usage:
|
|
client = WebberClient("http://localhost:8086")
|
|
response = await client.run_agent("explore", "find python files", "/path/to/project")
|
|
"""
|
|
|
|
def __init__(
|
|
self,
|
|
base_url: str = "http://localhost:8086",
|
|
api_key: str | None = None,
|
|
timeout: float = 120.0,
|
|
):
|
|
"""
|
|
Initialize the Webber client.
|
|
|
|
Args:
|
|
base_url: Webber API URL
|
|
api_key: Optional API key for authentication
|
|
timeout: Request timeout in seconds
|
|
"""
|
|
self.base_url = base_url.rstrip("/")
|
|
self.api_key = api_key
|
|
self.timeout = timeout
|
|
self._client: httpx.AsyncClient | None = None
|
|
|
|
async def _get_client(self) -> httpx.AsyncClient:
|
|
"""Get or create the HTTP client."""
|
|
if self._client is None or self._client.is_closed:
|
|
headers = {}
|
|
if self.api_key:
|
|
headers["X-API-Key"] = self.api_key
|
|
self._client = httpx.AsyncClient(
|
|
base_url=self.base_url,
|
|
headers=headers,
|
|
timeout=self.timeout,
|
|
)
|
|
return self._client
|
|
|
|
async def close(self) -> None:
|
|
"""Close the HTTP client."""
|
|
if self._client and not self._client.is_closed:
|
|
await self._client.aclose()
|
|
self._client = None
|
|
|
|
async def health_check(self) -> bool:
|
|
"""Check if the API is healthy."""
|
|
try:
|
|
client = await self._get_client()
|
|
response = await client.get("/health")
|
|
return response.status_code == 200
|
|
except httpx.RequestError:
|
|
return False
|
|
|
|
async def list_agents(self) -> list[AgentInfo]:
|
|
"""List available agents."""
|
|
client = await self._get_client()
|
|
response = await client.get("/agents/")
|
|
response.raise_for_status()
|
|
data = response.json()
|
|
return [AgentInfo(**a) for a in data.get("agents", [])]
|
|
|
|
async def get_agent(self, agent_type: str) -> AgentInfo | None:
|
|
"""Get information about a specific agent."""
|
|
client = await self._get_client()
|
|
response = await client.get(f"/agents/{agent_type}")
|
|
if response.status_code == 404:
|
|
return None
|
|
response.raise_for_status()
|
|
return AgentInfo(**response.json())
|
|
|
|
async def run_agent(
|
|
self,
|
|
agent_type: str,
|
|
prompt: str,
|
|
working_dir: str = ".",
|
|
) -> AgentResponse:
|
|
"""
|
|
Run an agent with the given prompt.
|
|
|
|
Args:
|
|
agent_type: Type of agent (e.g., "explore")
|
|
prompt: User prompt/query
|
|
working_dir: Working directory for the agent
|
|
|
|
Returns:
|
|
AgentResponse with the result
|
|
"""
|
|
client = await self._get_client()
|
|
response = await client.post(
|
|
"/agents/run",
|
|
json={
|
|
"agent_type": agent_type,
|
|
"prompt": prompt,
|
|
"working_dir": working_dir,
|
|
},
|
|
)
|
|
response.raise_for_status()
|
|
data = response.json()
|
|
return AgentResponse(
|
|
response=data.get("response", ""),
|
|
agent_type=data.get("agent_type", agent_type),
|
|
success=data.get("success", True),
|
|
error=data.get("error"),
|
|
)
|
|
|
|
async def __aenter__(self) -> "WebberClient":
|
|
"""Async context manager entry."""
|
|
return self
|
|
|
|
async def __aexit__(self, *args: Any) -> None:
|
|
"""Async context manager exit."""
|
|
await self.close()
|