OpenManus代码详解(四)：app/schema.py

南城同学

已于 2025-03-13 17:52:32 修改

阅读量359

点赞数 8

分类专栏： OpenManus 文章标签： OpenManus 智能体

于 2025-03-13 17:15:24 首次发布

本文链接：https://blog.youkuaiyun.com/Nine_Yao/article/details/146236932

版权

OpenManus 专栏收录该内容

7 篇文章

订阅专栏

请添加图片描述

官方地址：https://github.com/mannaandpoem/OpenManus

这个文件定义了项目中使用的核心数据模型。

1. AgentState 枚举类

class AgentState(str, Enum):
    """代理执行状态枚举"""
    IDLE = "IDLE"           # 空闲状态
    RUNNING = "RUNNING"     # 运行状态
    FINISHED = "FINISHED"   # 完成状态
    ERROR = "ERROR"         # 错误状态

继承自 str 和 Enum，定义了代理可能的状态
用于跟踪和管理代理的执行状态

2. Function 和 ToolCall 类

class Function(BaseModel):
    name: str          # 函数名称
    arguments: str     # 函数参数（JSON字符串）

class ToolCall(BaseModel):
    """表示消息中的工具/函数调用"""
    id: str           # 调用ID
    type: str = "function"  # 调用类型
    function: Function      # 函数信息

定义了工具调用的数据结构
用于处理 LLM 的函数调用请求

3. Message 类

class Message(BaseModel):
    """表示对话中的消息"""
    # 基本属性
    role: Literal["system", "user", "assistant", "tool"]  # 消息角色
    content: Optional[str] = None                         # 消息内容
    tool_calls: Optional[List[ToolCall]] = None          # 工具调用列表
    name: Optional[str] = None                           # 工具名称
    tool_call_id: Optional[str] = None                   # 工具调用ID

    # 运算符重载
    def __add__(self, other) -> List["Message"]:
        """支持 Message + list 或 Message + Message"""
    
    def __radd__(self, other) -> List["Message"]:
        """支持 list + Message"""

    # 转换方法
    def to_dict(self) -> dict:
        """转换为字典格式"""

    # 工厂方法
    @classmethod
    def user_message(cls, content: str) -> "Message":
        """创建用户消息"""

    @classmethod
    def system_message(cls, content: str) -> "Message":
        """创建系统消息"""

    @classmethod
    def assistant_message(cls, content: Optional[str] = None) -> "Message":
        """创建助手消息"""

    @classmethod
    def tool_message(cls, content: str, name, tool_call_id: str) -> "Message":
        """创建工具消息"""

    @classmethod
    def from_tool_calls(cls, tool_calls: List[Any], content: Union[str, List[str]] = ""):
        """从原始工具调用创建消息"""

特点：

支持四种消息角色：系统、用户、助手、工具
提供便捷的消息创建方法
支持消息列表的加法运算
可以转换为字典格式

4. Memory 类

class Memory(BaseModel):
    """内存管理类"""
    messages: List[Message] = Field(default_factory=list)  # 消息列表
    max_messages: int = Field(default=100)                 # 最大消息数量

    def add_message(self, message: Message) -> None:
        """添加单条消息"""
        self.messages.append(message)
        # 超出限制时裁剪
        if len(self.messages) > self.max_messages:
            self.messages = self.messages[-self.max_messages:]

    def add_messages(self, messages: List[Message]) -> None:
        """添加多条消息"""
        self.messages.extend(messages)

    def clear(self) -> None:
        """清空所有消息"""
        self.messages.clear()

    def get_recent_messages(self, n: int) -> List[Message]:
        """获取最近的n条消息"""
        return self.messages[-n:]

    def to_dict_list(self) -> List[dict]:
        """转换为字典列表"""
        return [msg.to_dict() for msg in self.messages]