From 31fb3c8921c5c1b257db7d8cb7ea3f0b1918bfa6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=D0=9A=D0=B8=D1=80=D0=B8=D0=BB=D0=BB=20=D0=9A=D1=83=D1=82?=
 =?UTF-8?q?=D0=BB=D0=B0=D1=85=D0=BC=D0=B5=D1=82=D0=BE=D0=B2?=
 <kirillkutlahmetov10@gmail.com>
Date: Wed, 27 May 2026 13:54:50 +0000
Subject: [PATCH] add models.py

---
 models.py | 95 +++++++++++++++++++++++--------------------------------
 1 file changed, 39 insertions(+), 56 deletions(-)

diff --git a/models.py b/models.py
index 7302698..bf850f9 100644
--- a/models.py
+++ b/models.py
@@ -1,61 +1,44 @@
-import os
+"""
+Pydantic model representing a structured assignment card.
+
+Fields:
+- title: short title of the task (string)
+- subject: main topic or subject area (string, optional)
+- deadline_hint: free‑form hint about due date (string, optional)
+- deliverable_type: what is expected to be submitted (string, optional)
+- grading_hints: list of strings describing evaluation criteria (list[str], optional)
+"""
+from __future__ import annotations
+
 from typing import List, Optional
-from pydantic import BaseModel, Field, validator
 
-class TaskCard(BaseModel):
-    """Pydantic model representing a structured task card.
+from pydantic import BaseModel, Field
 
-    Attributes
-    ----------
-    title: str
-        Short title of the task (max 10 words).
-    subject: str
-        Subject or topic of the task.
-    deadline: Optional[str]
-        Deadline string if mentioned, otherwise None.
-    deliverable: str
-        What needs to be submitted.
-    requirements: List[str]
-        List of concrete requirements extracted from the raw text.
-    difficulty: str
-        Difficulty level: "легко", "средне" or "сложно".
-    grading_criteria: Optional[str]
-        Grading criteria if present.
-    tags: List[str]
-        Tags describing the task.
-    """
+class AssignmentCard(BaseModel):
+    """Structured representation of an informal assignment description."""
 
-    title: str = Field(..., description="Short title of the task (max 10 words)")
-    subject: str = Field(..., description="Subject or topic of the task")
-    deadline: Optional[str] = Field(None, description="Deadline if mentioned")
-    deliverable: str = Field(..., description="What needs to be submitted")
-    requirements: List[str] = Field(default_factory=list, description="Concrete requirements")
-    difficulty: str = Field(..., description="Difficulty level: легко, средне, сложно")
-    grading_criteria: Optional[str] = Field(None, description="Grading criteria if present")
-    tags: List[str] = Field(default_factory=list, description="Tags describing the task")
+    title: str = Field(..., description="Short title or main action of the task")
+    subject: Optional[str] = Field(None, description="Primary subject or topic of the task")
+    deadline_hint: Optional[str] = Field(
+        None,
+        description="Free‑form hint about when the task is due (e.g., 'к пятнице')",
+    )
+    deliverable_type: Optional[str] = Field(
+        None, description="What should be submitted: report, code, presentation, etc."
+    )
+    grading_hints: List[str] = Field(
+        default_factory=list,
+        description="List of evaluation criteria mentioned in the text",
+    )
 
-    @validator("title")
-    def title_word_limit(cls, v: str) -> str:
-        if len(v.split()) > 10:
-            raise ValueError("Title must be at most 10 words")
-        return v
-
-    def to_markdown(self) -> str:
-        """Return a nicely formatted Markdown representation of the card."""
-        md = [f"## {self.title}"]
-        md.append(f"**Предмет:** {self.subject}")
-        if self.deadline:
-            md.append(f"**Дедлайн:** {self.deadline}")
-        md.append(f"**Сдать:** {self.deliverable}")
-        if self.requirements:
-            md.append("**Требования:**")
-            for req in self.requirements:
-                md.append(f"- {req}")
-        md.append(f"**Сложность:** {self.difficulty}")
-        if self.grading_criteria:
-            md.append(f"**Критерии оценки:** {self.grading_criteria}")
-        if self.tags:
-            md.append("**Теги:** " + ", ".join(self.tags))
-        return "\n".join(md)
-
-# End of models.py
+    class Config:
+        arbitrary_types_allowed = True
+        json_schema_extra = {
+            "example": {
+                "title": "Мини‑отчёт по LangChain",
+                "subject": "LangChain",
+                "deadline_hint": "к пятнице",
+                "deliverable_type": "отчёт",
+                "grading_hints": ["полнота", "пример кода"],
+            }
+        }