Добавлен вывод сгенерированных данных

llm/schemas.py

@@ -0,0 +1,39 @@
+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