Add an example of integrating smolagents

Author: RussellLuo

README.md

@@ -446,14 +446,14 @@ triage = AgentSpec(
 - [mcp](examples/mcp)
 - [structured-outputs](examples/structured-outputs)
 - [deepseek-r1](examples/deepseek-r1)
-- [ping-pong](examples/ping-pong)
-- [stream-ping-pong](examples/stream-ping-pong)
-- [discovery](examples/discovery)
 - [translator](examples/translator)
+- [discovery](examples/discovery)
 - [notification](examples/notification)
-- [opencsg](examples/opencsg)
 - [app-builder](examples/app-builder)
-- [autogen](examples/autogen)
+- [opencsg](examples/opencsg)
+- [framework-integration](examples/framework-integration)
+- [ping-pong](examples/ping-pong)
+- [stream-ping-pong](examples/stream-ping-pong)
 - [cos](examples/cos)
 
 

examples/framework-integration/README.md

@@ -0,0 +1,5 @@
+# Integration with other Agent Frameworks
+
+
+- [Integration with smolagents](smolagents/README.md)
+- [Integration with AutoGen](autogen/README.md)

examples/autogen/README.md renamed to examples/framework-integration/autogen/README.md

@@ -30,13 +30,13 @@ export MODEL_BASE_URL="your-base-url"
 export MODEL_API_VERSION="your-api-version"
 export MODEL_API_KEY="your-api-key"
 
-python examples/using-autogen/autogen.py
+python examples/framework-integration/autogen/agent.py
 ```
 
-Then start a client in another terminal.
+Then communicate with the agent using the `coagent` CLI:
 
 ```bash
-python examples/rich_client.py autogen
+coagent agent -H type:ChatMessage --chat -d '{"role":"user","content":"What is the weather like in Beijing?"}'
 ```
 
 

examples/autogen/autogen.py renamed to examples/framework-integration/autogen/agent.py

@@ -1,7 +1,7 @@
 import asyncio
 import os
 
-from coagent.agents.chat_agent import ChatHistory, ChatMessage
+from coagent.agents import ChatMessage
 from coagent.core import (
     AgentSpec,
     BaseAgent,
@@ -12,6 +12,7 @@
     set_stderr_logger,
 )
 from coagent.runtimes import NATSRuntime
+
 from autogen_agentchat.agents import AssistantAgent
 from autogen_agentchat.task import TextMentionTermination
 from autogen_agentchat.teams import RoundRobinGroupChat
@@ -53,24 +54,22 @@ class AutoGenWeatherAgent(BaseAgent):
     agent_team = RoundRobinGroupChat([weather_agent], termination_condition=termination)
 
     @handler
-    async def handle(self, msg: ChatHistory, ctx: Context) -> ChatHistory:
+    async def handle(self, msg: ChatMessage, ctx: Context) -> ChatMessage:
         # Run the team and return the result.
-        result = await self.agent_team.run(task=msg.messages[-1].content)
+        result = await self.agent_team.run(task=msg.content)
         content = result.messages[-2].content
-        msg.messages.append(ChatMessage(role="assistant", content=content))
-        return msg
+        return ChatMessage(role="assistant", content=content)
 
 
-autogen = AgentSpec("autogen", new(AutoGenWeatherAgent))
+agent = AgentSpec("agent", new(AutoGenWeatherAgent))
 
 
 async def main():
     async with NATSRuntime.from_servers() as runtime:
-        await runtime.register(autogen)
+        await runtime.register(agent)
         await idle_loop()
 
 
 if __name__ == "__main__":
     set_stderr_logger("TRACE")
-
     asyncio.run(main())

examples/framework-integration/smolagents/README.md

@@ -0,0 +1,41 @@
+# Using AutoGen
+
+This example demonstrates how to use [smolagents' ToolCallingAgent][1] in Coagent.
+
+References:
+
+- [Agent from any LLM][2]
+
+
+## Prerequisites
+
+- Install `coagent` (see [Installation](../../README.md#installation)).
+- Start a NATS server (see [Distributed](../../README.md#distributed)).
+- Install `smolagents`:
+
+    ```bash
+    pip install smolagents
+    ```
+
+
+## Quick Start
+
+First, start a server in one terminal:
+
+```bash
+export MODEL_ID="your-model-id"
+export MODEL_BASE_URL="your-base-url"
+export MODEL_API_KEY="your-api-key"
+
+python examples/framework-integration/smolagents/agent.py
+```
+
+Then communicate with the agent using the `coagent` CLI:
+
+```bash
+coagent agent -H type:ChatMessage --chat -d '{"role":"user","content":"What is the weather like in Beijing?"}'
+```
+
+
+[1]: https://huggingface.co/docs/smolagents/reference/agents#smolagents.ToolCallingAgent
+[2]: https://github.com/huggingface/smolagents/blob/main/examples/agent_from_any_llm.py

examples/framework-integration/smolagents/agent.py

@@ -0,0 +1,80 @@
+import asyncio
+import os
+from typing import Callable
+
+from coagent.agents import ChatMessage, ModelClient
+from coagent.agents.util import run_in_thread
+from coagent.core import (
+    AgentSpec,
+    BaseAgent,
+    Context,
+    handler,
+    idle_loop,
+    new,
+    set_stderr_logger,
+)
+from coagent.runtimes import NATSRuntime
+
+from smolagents import LiteLLMModel, tool as smolagents_tool
+from smolagents.agents import ToolCallingAgent
+
+
+class ReActAgent(BaseAgent):
+    def __init__(self, tools: list[Callable], client: ModelClient):
+        super().__init__()
+
+        model = LiteLLMModel(
+            model_id=client.model,
+            api_base=client.base_url,
+            api_key=client.api_key,
+        )
+        # Convert tools to smolagents tools.
+        tools = [smolagents_tool(t) for t in tools]
+
+        self.smol_agent = ToolCallingAgent(tools=tools, model=model)
+
+    @handler
+    async def handle(self, msg: ChatMessage, ctx: Context) -> ChatMessage:
+        response = await self.run_agent(msg.content)
+        return ChatMessage(role="assistant", content=response)
+
+    @run_in_thread
+    def run_agent(self, task: str):
+        return self.smol_agent.run(task)
+
+
+def get_weather(location: str, celsius: bool = False) -> str:
+    """
+    Get weather in the next days at given location.
+    Secretly this tool does not care about the location, it hates the weather everywhere.
+
+    Args:
+        location: the location
+        celsius: the temperature
+    """
+    return "The weather is UNGODLY with torrential rains and temperatures below -10°C"
+
+
+agent = AgentSpec(
+    "agent",
+    new(
+        ReActAgent,
+        tools=[get_weather],
+        client=ModelClient(
+            model=os.getenv("MODEL_ID"),
+            base_url=os.getenv("MODEL_BASE_URL"),
+            api_key=os.getenv("MODEL_API_KEY"),
+        ),
+    ),
+)
+
+
+async def main():
+    async with NATSRuntime.from_servers() as runtime:
+        await runtime.register(agent)
+        await idle_loop()
+
+
+if __name__ == "__main__":
+    set_stderr_logger()
+    asyncio.run(main())