rootdriver 0.1.0__py3-none-any.whl

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/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/constants.py

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

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/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/exception.py

@@ -0,0 +1 @@
+

rootdriver-0.1.0.dist-info/METADATA

@@ -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.dist-info/RECORD

@@ -0,0 +1,10 @@
+rootdriver/__init__.py,sha256=E7fG0npdGMr935XTHf0ByQbphNxQTW-nfWx6L0bXXFg,622
+rootdriver/agent.py,sha256=WDj0uc8Xt_KXwhOz3oyQwUSpmaIOrmAqEwHcyhGDiOQ,1454
+rootdriver/constants.py,sha256=btldAAsVhgJwj9zmk0E0WP-q26ydyEYQNSUkrXVfgb0,55
+rootdriver/conversation.py,sha256=_9lpx8K3ay32YI-vEf1ILfOaXgBPMCxRLmspYGbCEZE,2581
+rootdriver/engine.py,sha256=D-6CjvykK8V0Ppo7b3EP2VLIeh3GrtKd9a5ipOABK90,3155
+rootdriver/exception.py,sha256=frcCV1k9oG9oKj3dpUqdJg1PxRT2RSN_XKdLCPjaYaY,2
+rootdriver-0.1.0.dist-info/METADATA,sha256=YryuokUY6_huJ7r5tdhfccd7z8qtswijmSYwli96dow,2333
+rootdriver-0.1.0.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+rootdriver-0.1.0.dist-info/top_level.txt,sha256=m8YK68IRM5Pzia9feIdOJkfBXPEjVzA_KUTawqeA1us,11
+rootdriver-0.1.0.dist-info/RECORD,,

rootdriver-0.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (82.0.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

rootdriver-0.1.0.dist-info/top_level.txt

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