add main.py

2026-05-25 22:27:51 +00:00
parent 5a12bd3128
commit 215a0087cc
1 changed files with 91 additions and 0 deletions
@@ -0,0 +1,91 @@
+"""Разбор сырого текста задания в плоскую карточку (Pydantic + LangChain + OpenRouter)."""
+from __future__ import annotations
+
+import os
+
+from dotenv import load_dotenv
+from langchain_core.output_parsers import PydanticOutputParser
+from langchain_core.prompts import PromptTemplate
+from langchain_openai import ChatOpenAI
+from pydantic import BaseModel, Field
+
+load_dotenv()
+
+OPENROUTER_BASE_URL = os.getenv("OPENROUTER_BASE_URL", "https://openrouter.ai/api/v1")
+# gpt-oss-20b:free часто в 429 — fallback на рабочую free-модель OpenRouter
+OPENROUTER_MODEL = os.getenv(
+    "OPENROUTER_MODEL",
+    "poolside/laguna-m.1:free",
+)
+
+SAMPLE_TASK_TEXT = (
+    "Сдайте к пятнице мини-отчёт по LangChain: 2 страницы, упор на агентов. "
+    "Оценка: за полноту и за пример кода."
+)
+
+
+class TaskCard(BaseModel):
+    """Плоская карточка задания."""
+
+    title: str = Field(description="Краткий заголовок задания")
+    subject: str = Field(description="Предмет или тема (LangChain, Python и т.д.)")
+    deadline_hint: str = Field(
+        description="Подсказка по сроку в свободной форме (к пятнице, до 20 марта)"
+    )
+    deliverable_type: str = Field(
+        description="Что нужно сдать: отчёт, код, презентация и т.п."
+    )
+    grading_hints: list[str] = Field(
+        description="Список критериев или подсказок по оценке"
+    )
+
+
+def build_chain():
+    parser = PydanticOutputParser(pydantic_object=TaskCard)
+    prompt = PromptTemplate(
+        template=(
+            "Ты помощник студента. Из текста ниже извлеки структурированную карточку задания.\n"
+            "Верни только данные в формате из инструкций, без пояснений.\n\n"
+            "Текст задания:\n{task_text}\n\n"
+            "{format_instructions}"
+        ),
+        input_variables=["task_text"],
+        partial_variables={"format_instructions": parser.get_format_instructions()},
+    )
+    llm = ChatOpenAI(
+        model=OPENROUTER_MODEL,
+        base_url=OPENROUTER_BASE_URL,
+        api_key=os.getenv("OPENAI_API_KEY"),
+        temperature=0,
+    )
+    return prompt | llm | parser
+
+
+def print_summary(card: TaskCard) -> None:
+    print("\n--- Краткая сводка ---")
+    print(f"title: {card.title}")
+    print(f"subject: {card.subject}")
+    print(f"deadline_hint: {card.deadline_hint}")
+    print(f"deliverable_type: {card.deliverable_type}")
+    print(f"grading_hints: {card.grading_hints}")
+
+
+def parse_task_text(task_text: str) -> TaskCard:
+    chain = build_chain()
+    return chain.invoke({"task_text": task_text})
+
+
+def main() -> None:
+    print(f"Модель OpenRouter: {OPENROUTER_MODEL}")
+    print(f"\nВход:\n{SAMPLE_TASK_TEXT}\n")
+
+    card = parse_task_text(SAMPLE_TASK_TEXT)
+
+    print("\n--- Валидированный объект (model_dump) ---")
+    print(card.model_dump())
+
+    print_summary(card)
+
+
+if __name__ == "__main__":
+    main()