refactor: 重构openai包基础组件架构

This commit is contained in:
RockChinQ
2024-01-27 00:06:38 +08:00
parent 411034902a
commit 850a4eeb7c
35 changed files with 779 additions and 59 deletions
+31
View File
@@ -0,0 +1,31 @@
from __future__ import annotations
import typing
import enum
import pydantic
class MessageRole(enum.Enum):
SYSTEM = 'system'
USER = 'user'
ASSISTANT = 'assistant'
FUNCTION = 'function'
class FunctionCall(pydantic.BaseModel):
name: str
args: dict[str, typing.Any]
class Message(pydantic.BaseModel):
role: MessageRole
content: typing.Optional[str] = None
function_call: typing.Optional[FunctionCall] = None