examples, sensible defaults

Author: evalstate

…ol-use-agent-hooks/fastagent.config.yaml …/tool-runner-hooks/fastagent.config.yamlexamples/tool-use-agent-hooks/fastagent.config.yaml renamed to examples/tool-runner-hooks/fastagent.config.yaml examples/tool-use-agent-hooks/fastagent.config.yaml renamed to examples/tool-runner-hooks/fastagent.config.yaml

@@ -0,0 +1,47 @@
+import asyncio
+
+from mcp.types import TextContent
+
+from fast_agent.agents.agent_types import AgentConfig
+from fast_agent.agents.tool_agent import ToolAgent
+from fast_agent.agents.tool_runner import ToolRunner
+from fast_agent.core import Core
+from fast_agent.llm.model_factory import ModelFactory
+from fast_agent.types import PromptMessageExtended
+
+
+def lookup_order_status(order_id: str) -> str:
+    return f"Order {order_id} is packed and ready to ship."
+
+
+async def main() -> None:
+    core: Core = Core()
+    await core.initialize()
+
+    config = AgentConfig(name="order_bot")
+    agent = ToolAgent(config, tools=[lookup_order_status], context=core.context)
+    await agent.attach_llm(ModelFactory.create_factory("haiku"))
+
+    messages = [
+        PromptMessageExtended(
+            role="user",
+            content=[
+                TextContent(type="text", text="Check order 12345, then summarize in one line.")
+            ],
+        )
+    ]
+
+    runner = ToolRunner(
+        agent=agent,
+        messages=messages,
+    )
+
+    async for assistant_message in runner:
+        text = assistant_message.last_text() or "<no text>"
+        print(f"[assistant] {text}")
+
+    await core.cleanup()
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

examples/tool-use-agent-hooks/agent.py …s/tool-runner-hooks/tool_runner_hooks.pyexamples/tool-use-agent-hooks/agent.py renamed to examples/tool-runner-hooks/tool_runner_hooks.py examples/tool-use-agent-hooks/agent.py renamed to examples/tool-runner-hooks/tool_runner_hooks.py

@@ -9,7 +9,7 @@
     Union,
 )
 
-from mcp.types import TextContent
+from mcp.types import ListToolsResult, TextContent
 
 from fast_agent.constants import DEFAULT_MAX_ITERATIONS, FAST_AGENT_ERROR_CHANNEL
 from fast_agent.types import PromptMessageExtended, RequestParams
@@ -35,6 +35,8 @@ async def _tool_runner_llm_step(
 
     async def run_tools(self, request: PromptMessageExtended) -> PromptMessageExtended: ...
 
+    async def list_tools(self) -> ListToolsResult: ...
+
 
 @dataclass(frozen=True)
 class ToolRunnerHooks:
@@ -67,8 +69,8 @@ def __init__(
         *,
         agent: _ToolLoopAgent,
         messages: list[PromptMessageExtended],
-        request_params: RequestParams | None,
-        tools: list[Tool] | None,
+        request_params: RequestParams | None = None,
+        tools: list[Tool] | None = None,
         hooks: ToolRunnerHooks | None = None,
     ) -> None:
         self._agent = agent
@@ -95,6 +97,8 @@ async def __anext__(self) -> PromptMessageExtended:
         if self._done:
             raise StopAsyncIteration
 
+        await self._ensure_tools_ready()
+
         if self._hooks.before_llm_call is not None:
             await self._hooks.before_llm_call(self, self._delta_messages)
 
@@ -188,6 +192,10 @@ def _stage_tool_response(self, tool_message: PromptMessageExtended) -> None:
                 self._delta_messages.append(self._last_message)
             self._delta_messages.append(tool_message)
 
+    async def _ensure_tools_ready(self) -> None:
+        if self._tools is None:
+            self._tools = (await self._agent.list_tools()).tools
+
     async def _ensure_tool_response_staged(self) -> None:
         if self._pending_tool_request is None:
             return