Functional agent with typed I/O

from vectorshift.agent import Agent, AgentType, IoConfig, LlmInfo
from vectorshift.agent.tool import ToolInput, ToolInputType
from vectorshift.agent.tools import ExaAiTool, GoogleSearchTool

# Create a tool instance directly
search = GoogleSearchTool(
    tool_name="google_web_search",
    query=ToolInput(type=ToolInputType.DYNAMIC, description=None),
    num_results=10,
)

exa_search = ExaAiTool(
    tool_name="exa_ai_search",
    query=ToolInput(type=ToolInputType.DYNAMIC, description="Search query"),
)

# Create a functional agent with typed inputs and outputs
agent = Agent.new(
    name="Topic briefing",
    type=AgentType.FUNCTIONAL,
    llm_info=LlmInfo(provider="openai", model_id="gpt-4o"),
    tools=[search],
    instructions=(
        "For the given topic, search the web if you need "
        "fresh facts, then write a short brief."
    ),
    inputs={
        "topic": IoConfig(io_type="string", description="Subject to brief"),
    },
    outputs={
        "brief": IoConfig(io_type="string", description="Markdown brief"),
    },
)

print(f"Created functional agent: {agent.name} (id={agent.id})")
print(f"  Inputs:  {list(agent.inputs.keys())}")
print(f"  Outputs: {list(agent.outputs.keys())}")
print(f"  Tools:   {[t.name for t in agent.tools]}")
print(f"  Type:    {agent.agent_type}")

# Clean up
agent.delete()
print("Deleted agent.")