Server-side tools (zero-infra)

from pathlib import Path
import sys

ROOT = (
    Path.cwd().resolve().parent
    if Path.cwd().name == "notebooks"
    else Path.cwd().resolve()
)
if str(ROOT) not in sys.path:
    sys.path.insert(0, str(ROOT))

import json
from shipit_agent.llms import (
    web_search,
    code_execution,
    computer_use,
    bash,
    text_editor,
)

declarations = {
    "web_search": web_search(max_uses=3, allowed_domains=["docs.anthropic.com"]),
    "code_execution": code_execution(),
    "computer_use": computer_use(display_width_px=1280, display_height_px=800),
    "bash": bash(),
    "text_editor": text_editor(),
}
print(json.dumps(declarations, indent=2))

from shipit_agent.llms.server_tools import required_betas, is_server_tool, BETA_HEADERS

tools = [web_search(), code_execution(), computer_use(display_width_px=1024, display_height_px=768)]
print("is_server_tool(web_search()):", is_server_tool(web_search()))
print("is_server_tool(client tool):", is_server_tool({"function": {"name": "x"}}))
print("\nBETA_HEADERS map:", json.dumps(BETA_HEADERS, indent=2))
print("required betas for [web_search, code_execution, computer_use]:")
print(required_betas(tools))

from shipit_agent.llms import AnthropicChatLLM
from shipit_agent.models import Message

llm = AnthropicChatLLM(model="claude-opus-4-1", api_key="offline-demo-key")

req = llm._build_request_kwargs(
    messages=[Message(role="user", content="Search the web and run a quick calc.")],
    tools=[web_search(max_uses=3), code_execution()],
    system_prompt="You are a research assistant.",
)

print("tool declarations forwarded as-is (note top-level type):")
print(json.dumps(req["tools"], indent=2))
print("\nbeta headers attached by the adapter:")
print(req.get("betas"))

client_tool = {
    "function": {
        "name": "lookup_order",
        "description": "Look up an order by id.",
        "parameters": {"type": "object", "properties": {"order_id": {"type": "string"}}},
    }
}

req_mixed = llm._build_request_kwargs(
    messages=[Message(role="user", content="Find order 42 and search for its tracking page.")],
    tools=[client_tool, web_search()],
    system_prompt=None,
)
for t in req_mixed["tools"]:
    kind = "server-tool (verbatim)" if "type" in t else "client-tool (translated)"
    print(f"{t['name']:<28} -> {kind}")
print("\nbetas:", req_mixed.get("betas"), "(web_search is GA, so none)")

import sys, types
from types import SimpleNamespace as NS

# Fake response content: a server_tool_use block, its web_search_tool_result, and text.
fake_content = [
    NS(type="server_tool_use", id="srv_1", name="web_search",
       input={"query": "anthropic server tools"}),
    NS(type="web_search_tool_result", tool_use_id="srv_1",
       content=[{"type": "web_search_result", "title": "Docs", "url": "https://docs.anthropic.com"}]),
    NS(type="text", text="Based on the search, server tools run in Anthropic's sandbox."),
]
fake_response = NS(
    content=fake_content,
    usage=NS(input_tokens=120, output_tokens=30,
             cache_read_input_tokens=0, cache_creation_input_tokens=0),
)

fake_anthropic = types.ModuleType("anthropic")

class _Messages:
    def create(self, **_kwargs):
        return fake_response

class _Beta:
    def __init__(self):
        self.messages = _Messages()

class _Client:
    def __init__(self, **_kwargs):
        self.messages = _Messages()
        self.beta = _Beta()

fake_anthropic.Anthropic = _Client
sys.modules["anthropic"] = fake_anthropic

out = llm.complete(
    messages=[Message(role="user", content="What are server tools?")],
    tools=[web_search()],
)

print("text:", out.content)
print("client-side tool_calls (should be empty):", out.tool_calls)
print("\nmetadata['server_tool_use']:")
print(json.dumps(out.metadata.get("server_tool_use"), indent=2))
print("\nmetadata['server_tool_results']:")
print(json.dumps(out.metadata.get("server_tool_results"), indent=2))

import os
sys.modules.pop("anthropic", None)  # drop the fake so a real import is used

if os.getenv("ANTHROPIC_API_KEY"):
    live = AnthropicChatLLM(model="claude-opus-4-1")
    r = live.complete(
        messages=[Message(role="user", content="Search the web: what is the Anthropic Batches API?")],
        tools=[web_search(max_uses=2)],
    )
    print("text:", r.content[:400])
    print("server_tool_use:", r.metadata.get("server_tool_use"))
else:
    print("Set ANTHROPIC_API_KEY to run the live web-search call. Skipping (offline).")