28 lines
517 B
Python
28 lines
517 B
Python
from dataclasses import dataclass
|
|
from typing import Any, List, Literal, Optional, TypedDict
|
|
|
|
|
|
class _FunctionToolCall(TypedDict):
|
|
name: str
|
|
arguments: Optional[dict[str, Any]]
|
|
|
|
|
|
class ToolCall(TypedDict):
|
|
id: str
|
|
function: _FunctionToolCall
|
|
type: Literal["function"]
|
|
|
|
|
|
@dataclass
|
|
class MessageEvent:
|
|
content: Any
|
|
role: str = "user"
|
|
tool_calls: Optional[List[ToolCall]] = None
|
|
|
|
|
|
@dataclass
|
|
class ChoiceEvent:
|
|
index: int
|
|
message: MessageEvent
|
|
finish_reason: str = "unknown"
|