MCP Server

from pydantic import BaseModel
from afk.tools import tool

class SearchArgs(BaseModel):
    query: str
    limit: int = 10

@tool(args_model=SearchArgs, name="search_docs", description="Search the documentation.")
def search_docs(args: SearchArgs) -> dict:
    return {"results": ["doc1", "doc2"]}

class CalcArgs(BaseModel):
    expression: str

_OPS = {
    "+": lambda a, b: a + b,
    "-": lambda a, b: a - b,
    "*": lambda a, b: a * b,
    "/": lambda a, b: a / b,
}

@tool(args_model=CalcArgs, name="calculate", description="Evaluate a math expression.")
def calculate(args: CalcArgs) -> dict:
    left, op, right = args.expression.split()
    return {"result": _OPS[op](float(left), float(right))}

from afk.mcp import MCPServer

server = MCPServer(
    name="my-tools",
    tools=[search_docs, calculate],
    host="0.0.0.0",
    port=3001,
)

await server.start()
# Server is now accepting MCP connections on port 3001

from afk.mcp import MCPStore

store = MCPStore()
await store.connect("https://tools.example.com:3001")

tools = await store.list_tools()
for t in tools:
    print(f"{t.name}: {t.description}")

from afk.agents import Agent

agent = Agent(
    name="assistant",
    model="gpt-5.5",
    instructions="Use available tools to help the user.",
    tools=tools,  # ← MCP tools work like local tools
)