39 lines
706 B
Python
39 lines
706 B
Python
from typing import Literal
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
class RobotPlan(BaseModel):
|
|
steps: list[str] = Field(
|
|
description="Последовательность высокоуровневых действий робота",
|
|
min_length=1
|
|
)
|
|
|
|
|
|
class Message(BaseModel):
|
|
role: Literal["system",
|
|
"user",
|
|
"assistant"]
|
|
|
|
content: str
|
|
|
|
|
|
class SFTFormat(BaseModel):
|
|
message: list[Message]
|
|
|
|
|
|
class DatasetRecord(BaseModel):
|
|
task: str
|
|
|
|
system_prompt: str
|
|
instructions: str
|
|
user_prompt: str
|
|
|
|
model_response: str
|
|
assistant_content: str
|
|
|
|
provider: str
|
|
model: str
|
|
|
|
temperature: float
|
|
max_tokens: int |