mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-06-18 19:44:21 +00:00
RockChinQ
51 lines
1.1 KiB
Python
51 lines
1.1 KiB
Python
from __future__ import annotations
|
|
|
|
import typing
|
|
import enum
|
|
import pydantic
|
|
|
|
|
|
class FunctionCall(pydantic.BaseModel):
|
|
name: str
|
|
|
|
arguments: str
|
|
|
|
|
|
class ToolCall(pydantic.BaseModel):
|
|
id: str
|
|
|
|
type: str
|
|
|
|
function: FunctionCall
|
|
|
|
|
|
class Message(pydantic.BaseModel):
|
|
"""消息"""
|
|
|
|
role: str # user, system, assistant, tool, command, plugin
|
|
"""消息的角色"""
|
|
|
|
name: typing.Optional[str] = None
|
|
"""名称，仅函数调用返回时设置"""
|
|
|
|
content: typing.Optional[str] = None
|
|
"""内容"""
|
|
|
|
function_call: typing.Optional[FunctionCall] = None
|
|
"""函数调用，不再受支持，请使用tool_calls"""
|
|
|
|
tool_calls: typing.Optional[list[ToolCall]] = None
|
|
"""工具调用"""
|
|
|
|
tool_call_id: typing.Optional[str] = None
|
|
|
|
def readable_str(self) -> str:
|
|
if self.content is not None:
|
|
return self.content
|
|
elif self.function_call is not None:
|
|
return f'{self.function_call.name}({self.function_call.arguments})'
|
|
elif self.tool_calls is not None:
|
|
return f'调用工具: {self.tool_calls[0].id}'
|
|
else:
|
|
return '未知消息'
|