rootdriver 0.1.0__tar.gz

rootdriver-0.1.0/PKG-INFO

@@ -0,0 +1,102 @@
+Metadata-Version: 2.4
+Name: rootdriver
+Version: 0.1.0
+Summary: Rooted in Origin, Driving All Things
+Author-email: zimvir <zimvir@qq.com>
+License: MIT
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: openai>=1.0.0
+Requires-Dist: pydantic>=2.0.0
+Provides-Extra: dev
+Requires-Dist: pytest>=7.0.0; extra == "dev"
+
+# RootDriver
+
+根源出发， 驱动万物
+
+一个轻量级的 Python AI Agent 开发框架。
+
+## 特性
+
+- **简洁易用**：装饰器方式定义工具，快速构建 Agent
+- **模块化设计**：LLM 适配器、工具系统、会话管理解耦
+- **工具调用**：支持 function calling，自动执行工具并返回结果
+- **多适配器**：基于适配器模式，方便扩展其他 LLM Provider
+
+## 安装
+
+```bash
+pip install rootdriver
+```
+
+## 快速开始
+
+### 定义工具
+
+```python
+from rootdriver import tool
+
+@tool
+def get_weather(city: str) -> str:
+    """获取城市天气"""
+    return f"{city} 晴天"
+```
+
+### 创建 Agent
+
+```python
+from rootdriver import Agent, AgentLLM, OpenAIAdapter
+agent_llm=AgentLLM(
+    adapter=OpenAIAdapter(
+        api_key="YOUR_API_KEY",
+        base_url="BASE_URL"
+    ),
+    model="gpt-4",
+        
+    ),
+agent = Agent(
+    agent_llm=agent_llm,
+    tools=[get_weather],
+    system_prompt="你是一个有用的助手",
+)
+
+# 单次对话
+response = agent.talk("北京天气怎么样？")
+print(response)
+```
+
+### 使用工具
+
+```python
+# 完整对话循环（包含工具调用）
+response = Agent.react("帮我查下上海天气")
+print(response)
+```
+
+## 核心组件
+
+| 组件 | 说明 |
+|------|------|
+| `Agent` | 智能体入口，整合 LLM、工具、会话 |
+| `Engine` | 核心引擎，处理对话循环和工具调用 |
+| `Conversation` | 会话管理，维护消息历史 |
+| `LLM` | LLM 调用封装 |
+| `Tool` | 工具集合，管理所有可调用工具 |
+
+## 项目结构
+
+```
+rootdriver/
+├── agent.py          # Agent 智能体
+├── engine.py         # 引擎核心
+├── conversation.py   # 对话管理
+├── llm/              # LLM 适配器
+├── tool/             # 工具系统
+├── types/            # 类型定义
+└── utils/            # 工具函数
+```
+
+## License
+
+MIT

rootdriver-0.1.0/README.md

@@ -0,0 +1,89 @@
+# RootDriver
+
+根源出发， 驱动万物
+
+一个轻量级的 Python AI Agent 开发框架。
+
+## 特性
+
+- **简洁易用**：装饰器方式定义工具，快速构建 Agent
+- **模块化设计**：LLM 适配器、工具系统、会话管理解耦
+- **工具调用**：支持 function calling，自动执行工具并返回结果
+- **多适配器**：基于适配器模式，方便扩展其他 LLM Provider
+
+## 安装
+
+```bash
+pip install rootdriver
+```
+
+## 快速开始
+
+### 定义工具
+
+```python
+from rootdriver import tool
+
+@tool
+def get_weather(city: str) -> str:
+    """获取城市天气"""
+    return f"{city} 晴天"
+```
+
+### 创建 Agent
+
+```python
+from rootdriver import Agent, AgentLLM, OpenAIAdapter
+agent_llm=AgentLLM(
+    adapter=OpenAIAdapter(
+        api_key="YOUR_API_KEY",
+        base_url="BASE_URL"
+    ),
+    model="gpt-4",
+        
+    ),
+agent = Agent(
+    agent_llm=agent_llm,
+    tools=[get_weather],
+    system_prompt="你是一个有用的助手",
+)
+
+# 单次对话
+response = agent.talk("北京天气怎么样？")
+print(response)
+```
+
+### 使用工具
+
+```python
+# 完整对话循环（包含工具调用）
+response = Agent.react("帮我查下上海天气")
+print(response)
+```
+
+## 核心组件
+
+| 组件 | 说明 |
+|------|------|
+| `Agent` | 智能体入口，整合 LLM、工具、会话 |
+| `Engine` | 核心引擎，处理对话循环和工具调用 |
+| `Conversation` | 会话管理，维护消息历史 |
+| `LLM` | LLM 调用封装 |
+| `Tool` | 工具集合，管理所有可调用工具 |
+
+## 项目结构
+
+```
+rootdriver/
+├── agent.py          # Agent 智能体
+├── engine.py         # 引擎核心
+├── conversation.py   # 对话管理
+├── llm/              # LLM 适配器
+├── tool/             # 工具系统
+├── types/            # 类型定义
+└── utils/            # 工具函数
+```
+
+## License
+
+MIT

rootdriver-0.1.0/pyproject.toml

@@ -0,0 +1,26 @@
+[build-system]
+requires = ["setuptools>=61.0", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "rootdriver"
+version = "0.1.0"
+description = "Rooted in Origin, Driving All Things"
+readme = "README.md"
+license = {text = "MIT"}
+authors = [
+    {name = "zimvir", email = "zimvir@qq.com"}
+]
+requires-python = ">=3.10"
+dependencies = [
+    "openai>=1.0.0",
+    "pydantic>=2.0.0",
+]
+
+[project.optional-dependencies]
+dev = [
+    "pytest>=7.0.0",
+]
+
+[tool.setuptools]
+packages = ["rootdriver"]

rootdriver-0.1.0/rootdriver/__init__.py

@@ -0,0 +1,32 @@
+__version__ = "0.1.0"
+__author__ = "zimvir"
+__email__ = "zimvir@qq.com"
+
+
+
+
+
+from .agent import Agent
+from .engine import Engine
+from .conversation import Conversation
+from .llm import LLM
+from .llm.base_adapter import BaseAdapter
+from .llm.adapter import OpenAIAdapter
+from .tool import tool, Tool
+from .types.agent import AgentLLM
+from .types import Message, ToolDefinition, ToolCall
+
+__all__ = [
+    "Agent",
+    "Engine",
+    "Conversation",
+    "LLM",
+    "BaseAdapter",
+    "OpenAIAdapter",
+    "tool",
+    "Tool",
+    "AgentLLM",
+    "Message",
+    "ToolDefinition",
+    "ToolCall",
+]

rootdriver-0.1.0/rootdriver/agent.py

@@ -0,0 +1,48 @@
+from pathlib import Path
+from uuid import uuid4
+
+from .engine import Engine
+from .llm import LLM
+from .conversation import Conversation
+from .tool import Tool, BaseTool
+from .types.agent import AgentLLM
+from .types import ToolDefinition
+from .utils import build_system_message, build_message, build_tool_message, build_user_message, build_assistant_message
+
+class Agent:
+
+    def __init__(
+        self,
+        agent_llm: AgentLLM,
+        *,
+        tools: list[BaseTool] | None = None,
+        system_prompt: str | None = None,
+        id: str | None = None,
+        # save_path: str | Path | None = None,
+        # max_retries: int = 3,
+        timeout: float | None = None,
+    ):
+        self.conversation = Conversation(system_prompt)
+        self.tool = Tool(tools if tools else [])
+        self.id = id if id else uuid4().hex
+        # self.save_path = save_path
+
+        self.engine = Engine(
+            model=agent_llm.model,
+            llm=LLM(agent_llm.adapter),
+            conversation=self.conversation,
+            tool=self.tool,
+        )
+
+
+    def react(self, input_prompt:str) -> "str":
+        """一次 react 循环"""
+        response = self.engine.run(build_user_message(input_prompt))
+        return response.content
+
+    def talk(self, input_prompt:str) -> "str":
+        """一次 agent 调用"""
+        return self.engine.invoke(build_user_message(input_prompt)).content
+
+
+

rootdriver-0.1.0/rootdriver/constants.py

@@ -0,0 +1,2 @@
+ROLE_ENUM = ("system", "user", "assistant", "tool")
+

rootdriver-0.1.0/rootdriver/conversation.py

@@ -0,0 +1,72 @@
+from .types import Message
+from .utils import get_iso_timestamp, build_system_message
+
+
+class Conversation:
+    def __init__(self, system_prompt: str = None):
+        self.messages: list[Message] = []
+        self.system_prompt = system_prompt
+        if self.system_prompt:
+            self.append(build_system_message(system_prompt))
+
+    def append(self, message: Message) -> "Conversation":
+        self.messages.append(message)
+        return self
+
+    def append_many(self, messages: list[Message]) -> "Conversation":
+        self.messages.extend(messages)
+        return self
+
+    # def append_user(self, content: str) -> "Conversation":
+    #     self.messages.append(Message(role="user", content=content, created_at=get_iso_timestamp()))
+    #     return self
+    #
+    # def append_assistant(self, content: str | None = None, tool_calls: list | None = None) -> "Conversation":
+    #     msg = Message(role="assistant", content=content, created_at=get_iso_timestamp())
+    #     if tool_calls:
+    #         msg.tool_calls = tool_calls
+    #     self.messages.append(msg)
+    #     return self
+    #
+    # def append_tool(self, tool_call_id: str, content: str) -> "Conversation":
+    #     self.messages.append(Message(
+    #         role="tool",
+    #         tool_call_id=tool_call_id,
+    #         content=content,
+    #         created_at=get_iso_timestamp()
+    #     ))
+    #     return self
+
+    def append_system(self, content: str) -> "Conversation":
+        self.messages.append(Message(role="system", content=content, created_at=get_iso_timestamp()))
+        return self
+
+    def delete(self, index: int=-1) -> "Conversation":
+        self.messages.pop(index)
+        return self
+
+    def get_messages(self) -> list[Message]:
+        '''返回 list[message<obj>]'''
+        return self.messages
+
+    def get_messages_in_list(self) -> list[dict]:
+        '''返回 字典加列表组成的message（可转成json用于网络、跨语言传输） 组成的列表'''
+        return [m.model_dump(exclude_none=True) for m in self.messages]
+
+
+    @classmethod
+    def from_dict_list(cls, messages: list[dict]) -> "Conversation":
+        conv = cls()
+        conv.messages = [Message.model_validate(m) for m in messages]
+        return conv
+
+    @classmethod
+    def from_list(cls, messages: list[Message]) -> "Conversation":
+        conv = cls()
+        conv.messages = list(messages)
+        return conv
+
+    def clear(self) -> "Conversation":
+        self.messages = []
+        return self
+

rootdriver-0.1.0/rootdriver/engine.py

@@ -0,0 +1,108 @@
+
+
+from .llm import LLM
+from .conversation import Conversation
+from .tool import Tool
+from .types import LLMRequest, Message
+from .utils import  get_iso_timestamp,build_tool_message, build_message
+
+class Engine:
+    def __init__(self, model:str, llm:LLM, conversation:Conversation, tool:Tool, ):
+        self.llm = llm
+        self.conversation = conversation
+        self.tool = tool
+        self.model = model
+
+    def invoke(self, input_message:Message) -> Message:
+        self.conversation.append(input_message)
+        message = self.chat()
+        return message
+
+    def chat(self) -> Message:
+        """
+
+        1. 读会话消息
+        2. 大模型交流
+        3. 追加会话消息
+        4. 判断 是否调用工具
+        5. 是： tool 返回结果并追加，否： 追加并返回
+        """
+        # 1. 读会话消息
+        message = self.conversation.get_messages()
+
+        # 2. llm交互
+        response = self.llm.invoke(self.messages_to_llm_request(message))
+
+        # 3. 追加会话消息
+        response_message = self.llm_response_to_message(response)
+        self.conversation.append(response_message)
+
+        return response_message
+
+    def deal_tool_or_output(self, response_message:Message) -> Message|None:
+        """
+                1. 追加输入
+                2. 读会话消息
+                3. 大模型交流
+                4. 追加会话消息
+                5. 判断 是否调用工具
+                6. 是： tool 返回结果并追加，否： 追加并返回
+
+
+                Returns:
+
+                """
+
+        # 5. 判断 是否调用工具
+        if response_message.tool_calls:
+            # 6. 是： tool 返回结果追加
+            for tool_call in response_message.tool_calls:
+                tool_result = self.tool.invoke(tool_call)
+                self.conversation.append(build_tool_message(tool_result.tool_call_id, tool_result.content))
+            return None
+        else:
+            # 6. 否： 追加并返回
+            return response_message
+
+
+    def run(self, input_message:Message) ->Message:
+        """
+        1. 追加输入
+        2. 读会话消息
+        3. 大模型交流
+        4. 追加会话消息
+        5. 判断 是否调用工具
+        6. 是： tool 返回结果并追加，否： 追加并返回
+
+
+        Returns:
+
+        """
+        self.conversation.append(input_message)
+        while True:
+
+            response_message = self.chat()
+            result = self.deal_tool_or_output(response_message)
+            if result is None:
+                continue
+            elif isinstance(result, Message):
+                return result
+
+
+
+    
+
+        
+
+
+    def messages_to_llm_request(self,messages:list[Message]) -> LLMRequest:
+        return LLMRequest(
+            model=self.model,
+            messages=messages,
+            tools=self.tool.to_definitions(),
+        )
+
+    def llm_response_to_message(self, response) -> Message:
+        """把 LLM 响应转成 Message 。"""
+        return response.message
+

rootdriver-0.1.0/rootdriver/exception.py

@@ -0,0 +1 @@
+

rootdriver-0.1.0/rootdriver.egg-info/PKG-INFO

@@ -0,0 +1,102 @@
+Metadata-Version: 2.4
+Name: rootdriver
+Version: 0.1.0
+Summary: Rooted in Origin, Driving All Things
+Author-email: zimvir <zimvir@qq.com>
+License: MIT
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: openai>=1.0.0
+Requires-Dist: pydantic>=2.0.0
+Provides-Extra: dev
+Requires-Dist: pytest>=7.0.0; extra == "dev"
+
+# RootDriver
+
+根源出发， 驱动万物
+
+一个轻量级的 Python AI Agent 开发框架。
+
+## 特性
+
+- **简洁易用**：装饰器方式定义工具，快速构建 Agent
+- **模块化设计**：LLM 适配器、工具系统、会话管理解耦
+- **工具调用**：支持 function calling，自动执行工具并返回结果
+- **多适配器**：基于适配器模式，方便扩展其他 LLM Provider
+
+## 安装
+
+```bash
+pip install rootdriver
+```
+
+## 快速开始
+
+### 定义工具
+
+```python
+from rootdriver import tool
+
+@tool
+def get_weather(city: str) -> str:
+    """获取城市天气"""
+    return f"{city} 晴天"
+```
+
+### 创建 Agent
+
+```python
+from rootdriver import Agent, AgentLLM, OpenAIAdapter
+agent_llm=AgentLLM(
+    adapter=OpenAIAdapter(
+        api_key="YOUR_API_KEY",
+        base_url="BASE_URL"
+    ),
+    model="gpt-4",
+        
+    ),
+agent = Agent(
+    agent_llm=agent_llm,
+    tools=[get_weather],
+    system_prompt="你是一个有用的助手",
+)
+
+# 单次对话
+response = agent.talk("北京天气怎么样？")
+print(response)
+```
+
+### 使用工具
+
+```python
+# 完整对话循环（包含工具调用）
+response = Agent.react("帮我查下上海天气")
+print(response)
+```
+
+## 核心组件
+
+| 组件 | 说明 |
+|------|------|
+| `Agent` | 智能体入口，整合 LLM、工具、会话 |
+| `Engine` | 核心引擎，处理对话循环和工具调用 |
+| `Conversation` | 会话管理，维护消息历史 |
+| `LLM` | LLM 调用封装 |
+| `Tool` | 工具集合，管理所有可调用工具 |
+
+## 项目结构
+
+```
+rootdriver/
+├── agent.py          # Agent 智能体
+├── engine.py         # 引擎核心
+├── conversation.py   # 对话管理
+├── llm/              # LLM 适配器
+├── tool/             # 工具系统
+├── types/            # 类型定义
+└── utils/            # 工具函数
+```
+
+## License
+
+MIT

rootdriver-0.1.0/rootdriver.egg-info/SOURCES.txt

@@ -0,0 +1,16 @@
+README.md
+pyproject.toml
+rootdriver/__init__.py
+rootdriver/agent.py
+rootdriver/constants.py
+rootdriver/conversation.py
+rootdriver/engine.py
+rootdriver/exception.py
+rootdriver.egg-info/PKG-INFO
+rootdriver.egg-info/SOURCES.txt
+rootdriver.egg-info/dependency_links.txt
+rootdriver.egg-info/requires.txt
+rootdriver.egg-info/top_level.txt
+tests/test_base_tool.py
+tests/test_conversation.py
+tests/test_tool.py

rootdriver-0.1.0/rootdriver.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

rootdriver-0.1.0/rootdriver.egg-info/requires.txt

@@ -0,0 +1,5 @@
+openai>=1.0.0
+pydantic>=2.0.0
+
+[dev]
+pytest>=7.0.0

rootdriver-0.1.0/rootdriver.egg-info/top_level.txt

@@ -0,0 +1 @@
+rootdriver

rootdriver-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

rootdriver-0.1.0/tests/test_base_tool.py

@@ -0,0 +1,103 @@
+import pytest
+from rootdriver.tool.base_tool import BaseTool
+
+
+def get_weather(city: str) -> str:
+    """获取城市天气"""
+    return f"{city} 晴天"
+
+
+@pytest.fixture
+def weather_tool():
+    return BaseTool(get_weather)
+
+
+def test_base_tool_name(weather_tool):
+    assert weather_tool.name == "get_weather"
+
+
+def test_base_tool_description(weather_tool):
+    assert weather_tool.description == "获取城市天气"
+
+
+def test_invoke_success(weather_tool):
+    result = weather_tool.invoke({"city": "北京"})
+    assert result == "北京 晴天"
+
+
+def test_invoke_error_returns_string(weather_tool):
+    def bad_func(city: str) -> str:
+        raise ValueError("参数错误")
+    bad_tool = BaseTool(bad_func)
+    result = bad_tool.invoke({"city": "北京"})
+    assert isinstance(result, str)
+    assert "ValueError" in result or "参数错误" in result
+
+
+def test_to_definition(weather_tool):
+    definition = weather_tool.to_definition()
+    assert definition.name == "get_weather"
+    assert definition.description == "获取城市天气"
+    assert "properties" in definition.parameters
+    assert "city" in definition.parameters["properties"]
+
+
+def test_custom_name():
+    def foo():
+        pass
+    tool = BaseTool(foo, name="custom_name")
+    assert tool.name == "custom_name"
+
+
+def test_custom_description():
+    def foo():
+        """custom desc"""
+        pass
+    tool = BaseTool(foo)
+    assert tool.description == "custom desc"
+
+
+def test_callable(weather_tool):
+    result = weather_tool(city="上海")
+    assert result == "上海 晴天"
+
+
+def test_invoke_retry_success():
+    """retry=1 第一次就成功"""
+    call_count = 0
+    def counter(city: str) -> str:
+        nonlocal call_count
+        call_count += 1
+        return f"{city} 晴天"
+    t = BaseTool(counter)
+    result = t.invoke({"city": "北京"}, retry=3)
+    assert result == "北京 晴天"
+    assert call_count == 1
+
+
+def test_invoke_retry_then_success():
+    """retry=3 前两次失败，第三次成功"""
+    call_count = 0
+    def flaky(city: str) -> str:
+        nonlocal call_count
+        call_count += 1
+        if call_count < 3:
+            raise ValueError(f"第{call_count}次失败")
+        return f"{city} 晴天"
+    t = BaseTool(flaky)
+    result = t.invoke({"city": "北京"}, retry=3)
+    assert result == "北京 晴天"
+    assert call_count == 3
+
+
+def test_invoke_retry_all_fail():
+    """retry=3 三次都失败，返回最后一次错误"""
+    call_count = 0
+    def always_fail(city: str) -> str:
+        nonlocal call_count
+        call_count += 1
+        raise ValueError(f"错误{call_count}")
+    t = BaseTool(always_fail)
+    result = t.invoke({"city": "北京"}, retry=3)
+    assert "错误3" in result
+    assert call_count == 3

rootdriver-0.1.0/tests/test_conversation.py

@@ -0,0 +1,78 @@
+import pytest
+from rootdriver.conversation import Conversation
+from rootdriver.types import Message
+
+
+def test_conversation_empty():
+    conv = Conversation()
+    assert conv.get_messages() == []
+
+
+def test_conversation_with_system_prompt():
+    conv = Conversation(system_prompt="你是一个助手")
+    messages = conv.get_messages()
+    assert len(messages) == 1
+    assert messages[0].role == "system"
+    assert messages[0].content == "你是一个助手"
+
+
+def test_append_message():
+    conv = Conversation()
+    conv.append(Message(role="user", content="你好", created_at="2026-01-01T00:00:00"))
+    assert len(conv.get_messages()) == 1
+    assert conv.get_messages()[0].content == "你好"
+
+
+def test_append_many():
+    conv = Conversation()
+    messages = [
+        Message(role="user", content="你好", created_at="2026-01-01T00:00:00"),
+        Message(role="assistant", content="你好！", created_at="2026-01-01T00:00:01"),
+    ]
+    conv.append_many(messages)
+    assert len(conv.get_messages()) == 2
+
+
+def test_append_system():
+    conv = Conversation()
+    conv.append_system("新的系统提示")
+    assert conv.get_messages()[-1].role == "system"
+    assert conv.get_messages()[-1].content == "新的系统提示"
+
+
+def test_delete_message():
+    conv = Conversation()
+    conv.append(Message(role="user", content="你好", created_at="2026-01-01T00:00:00"))
+    conv.append(Message(role="assistant", content="你好！", created_at="2026-01-01T00:00:01"))
+    conv.delete()
+    assert len(conv.get_messages()) == 1
+
+
+def test_clear():
+    conv = Conversation(system_prompt="test")
+    conv.clear()
+    assert len(conv.get_messages()) == 0
+
+
+def test_get_messages_in_list():
+    conv = Conversation()
+    conv.append(Message(role="user", content="你好", created_at="2026-01-01T00:00:00"))
+    result = conv.get_messages_in_list()
+    assert isinstance(result, list)
+    assert result[0]["content"] == "你好"
+
+
+def test_from_dict_list():
+    data = [
+        {"role": "user", "content": "你好", "created_at": "2026-01-01T00:00:00"},
+    ]
+    conv = Conversation.from_dict_list(data)
+    assert len(conv.get_messages()) == 1
+
+
+def test_from_list():
+    messages = [
+        Message(role="user", content="你好", created_at="2026-01-01T00:00:00"),
+    ]
+    conv = Conversation.from_list(messages)
+    assert len(conv.get_messages()) == 1

rootdriver-0.1.0/tests/test_tool.py

@@ -0,0 +1,76 @@
+import pytest
+from rootdriver.tool.base_tool import BaseTool
+from rootdriver.tool.tools import Tool
+from rootdriver.types.tool import ToolCall
+
+
+def get_weather(city: str) -> str:
+    """获取城市天气"""
+    return f"{city} 晴天"
+
+
+def get_time(city: str) -> str:
+    """获取城市时间"""
+    return f"{city} 10:00"
+
+
+@pytest.fixture
+def weather_tool():
+    return BaseTool(get_weather)
+
+
+@pytest.fixture
+def time_tool():
+    return BaseTool(get_time)
+
+
+@pytest.fixture
+def tool_registry(weather_tool, time_tool):
+    return Tool([weather_tool, time_tool])
+
+
+def test_tool_add():
+    t = Tool()
+    t.add(BaseTool(get_weather))
+    assert t.get("get_weather") is not None
+
+
+def test_tool_get(weather_tool):
+    t = Tool([weather_tool])
+    assert t.get("get_weather") is weather_tool
+
+
+def test_tool_get_not_found():
+    t = Tool()
+    assert t.get("not_exist") is None
+
+
+def test_tool_invoke_success(tool_registry):
+    result = tool_registry.invoke(ToolCall(id="1", name="get_weather", arguments={"city": "北京"}))
+    assert result.tool_call_id == "1"
+    assert result.content == "北京 晴天"
+
+
+def test_tool_invoke_not_found(tool_registry):
+    result = tool_registry.invoke(ToolCall(id="2", name="not_exist", arguments={}))
+    assert result.tool_call_id == "2"
+    assert "not found" in result.content
+
+
+def test_tool_invoke_many(tool_registry):
+    calls = [
+        ToolCall(id="1", name="get_weather", arguments={"city": "北京"}),
+        ToolCall(id="2", name="get_time", arguments={"city": "北京"}),
+    ]
+    results = tool_registry.invoke_many(calls)
+    assert len(results) == 2
+    assert results[0].content == "北京 晴天"
+    assert results[1].content == "北京 10:00"
+
+
+def test_tool_to_definitions(tool_registry):
+    defs = tool_registry.to_definitions()
+    assert len(defs) == 2
+    names = [d.name for d in defs]
+    assert "get_weather" in names
+    assert "get_time" in names