mirror of
https://github.com/langbot-app/LangBot.git
synced 2025-11-25 19:37:36 +08:00
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 '未知消息'
|