Обновил структуру проекта

llm/schemas.py

@@ -1,4 +1,4 @@
-from typing import Literal
+from typing import Literal, Optional
 
 from pydantic import BaseModel, Field
 
@@ -10,7 +10,7 @@ class RobotPlan(BaseModel):
     )
 
 
-class Message(BaseModel):
+class ChatMessage(BaseModel):
     role: Literal["system",
                   "user",
                   "assistant"]
@@ -18,22 +18,32 @@ class Message(BaseModel):
     content: str
 
 
-class SFTFormat(BaseModel):
-    message: list[Message]
-
-
-class DatasetRecord(BaseModel):
-    task: str
+class ChatRecord(BaseModel):
+    messages: list[ChatMessage]
 
+class Prompts(BaseModel):
     system_prompt: str
     instructions: str
     user_prompt: str
 
-    model_response: str
-    assistant_content: str
 
+class ModelResponse(BaseModel):
+    model_response: str
+    assistant_content: Optional[str] = None
+
+class ModelInfo(BaseModel):
     provider: str
     model: str
 
+class GenerationSettings(BaseModel):
     temperature: float
-    max_tokens: int
+    max_tokens: int
+
+class BaseDatasetRecord(BaseModel):
+    prompts: Prompts
+    response: ModelResponse
+    model_info: ModelInfo
+    generation: GenerationSettings
+
+class RobotDatasetRecord(BaseDatasetRecord):
+    task: str