mirror of https://github.com/Mai-with-u/MaiBot.git
20 lines
615 B
Python
20 lines
615 B
Python
from dataclasses import dataclass
|
|
from typing import Optional, List, TYPE_CHECKING
|
|
|
|
from . import BaseDataModel
|
|
|
|
if TYPE_CHECKING:
|
|
from src.common.data_models.message_data_model import ReplySetModel
|
|
from src.llm_models.payload_content.tool_option import ToolCall
|
|
|
|
|
|
@dataclass
|
|
class LLMGenerationDataModel(BaseDataModel):
|
|
content: Optional[str] = None
|
|
reasoning: Optional[str] = None
|
|
model: Optional[str] = None
|
|
tool_calls: Optional[List["ToolCall"]] = None
|
|
prompt: Optional[str] = None
|
|
selected_expressions: Optional[List[int]] = None
|
|
reply_set: Optional["ReplySetModel"] = None
|