mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-06-02 20:14:36 +00:00
34 lines
581 B
Python
34 lines
581 B
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
|
|
|
|
name: typing.Optional[str] = None
|
|
|
|
content: typing.Optional[str] = None
|
|
|
|
function_call: typing.Optional[FunctionCall] = None
|
|
|
|
tool_calls: typing.Optional[list[ToolCall]] = None
|
|
|
|
tool_call_id: typing.Optional[str] = None
|