131 lines
5.1 KiB
Python
131 lines
5.1 KiB
Python
"""
|
|
Парсер задания: сырой текст → плоская карточка TaskCard.
|
|
|
|
Пример:
|
|
python main.py
|
|
python main.py "Сдайте до понедельника презентацию по RAG: 10 слайдов. Оценка: за полноту и визуализацию."
|
|
"""
|
|
|
|
import json
|
|
import os
|
|
import sys
|
|
from typing import Optional
|
|
|
|
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()
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Pydantic-модель карточки задания
|
|
# ---------------------------------------------------------------------------
|
|
|
|
|
|
class TaskCard(BaseModel):
|
|
"""Плоская карточка задания — все поля одного уровня вложенности."""
|
|
|
|
title: str = Field(
|
|
...,
|
|
description="Краткое название задачи (2–6 слов)",
|
|
)
|
|
subject: str = Field(
|
|
...,
|
|
description="Область знаний или технология (LangChain, Python, ML и т.п.)",
|
|
)
|
|
deadline_hint: str = Field(
|
|
...,
|
|
description="Срок сдачи в свободной форме, пустая строка если не указан",
|
|
)
|
|
deliverable_type: str = Field(
|
|
...,
|
|
description="Что нужно сдать: отчёт, код, презентация, видео, ссылка и т.п.",
|
|
)
|
|
grading_hints: list[str] = Field(
|
|
default_factory=list,
|
|
description="Список критериев оценки, упомянутых в тексте задания",
|
|
)
|
|
notes: Optional[str] = Field(
|
|
default=None,
|
|
description="Любые дополнительные заметки или уточнения из текста",
|
|
)
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# LLM — BroJS
|
|
# ---------------------------------------------------------------------------
|
|
|
|
llm = ChatOpenAI(
|
|
model="openai/gpt-oss-20b:free",
|
|
base_url="https://platform.brojs.ru/jrnl-bh/api/inference/v1",
|
|
api_key=os.getenv("JOURNAL_MCP_PAT"),
|
|
temperature=0.3,
|
|
)
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Цепочка: PromptTemplate → LLM → PydanticOutputParser
|
|
# ---------------------------------------------------------------------------
|
|
|
|
parser = PydanticOutputParser(pydantic_object=TaskCard)
|
|
|
|
_TEMPLATE = """Ты — ассистент, который извлекает структурированную информацию из описания учебного задания.
|
|
Верни ответ строго в формате JSON согласно инструкции ниже — без лишних пояснений.
|
|
|
|
{format_instructions}
|
|
|
|
Описание задания:
|
|
{task_text}
|
|
"""
|
|
|
|
prompt = PromptTemplate(
|
|
template=_TEMPLATE,
|
|
input_variables=["task_text"],
|
|
partial_variables={"format_instructions": parser.get_format_instructions()},
|
|
)
|
|
|
|
chain = prompt | llm | parser
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Публичный интерфейс
|
|
# ---------------------------------------------------------------------------
|
|
|
|
|
|
def parse_task(task_text: str) -> TaskCard:
|
|
"""Парсит текст задания и возвращает валидированный объект TaskCard."""
|
|
return chain.invoke({"task_text": task_text})
|
|
|
|
|
|
def print_card(card: TaskCard) -> None:
|
|
"""Выводит карточку в двух форматах: JSON-объект и читаемая сводка."""
|
|
print("\n─── Валидированный объект (model_dump) ─────────────────────")
|
|
print(json.dumps(card.model_dump(), ensure_ascii=False, indent=2))
|
|
|
|
print("\n─── Человекочитаемая сводка ─────────────────────────────────")
|
|
print(f" Задача: {card.title}")
|
|
print(f" Тема: {card.subject}")
|
|
print(f" Срок: {card.deadline_hint or '—'}")
|
|
print(f" Тип сдачи: {card.deliverable_type}")
|
|
if card.grading_hints:
|
|
print(f" Критерии: {', '.join(card.grading_hints)}")
|
|
if card.notes:
|
|
print(f" Заметки: {card.notes}")
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# CLI
|
|
# ---------------------------------------------------------------------------
|
|
|
|
_EXAMPLE = (
|
|
"Сдайте к пятнице мини-отчёт по LangChain: 2 страницы, упор на агентов. "
|
|
"Оценка: за полноту и за пример кода."
|
|
)
|
|
|
|
if __name__ == "__main__":
|
|
text = " ".join(sys.argv[1:]) if len(sys.argv) > 1 else _EXAMPLE
|
|
print(f"Вход:\n {text}")
|
|
card = parse_task(text)
|
|
print_card(card)
|