Remove extra main.py

This commit is contained in:
2026-05-28 08:41:44 +00:00
parent 38939e24a9
commit 482a0ae2c6
-63
View File
@@ -1,63 +0,0 @@
"""
Simple module that converts informal assignment description into a structured data card.
Usage:
from main import parse_assignment
card = parse_assignment("Write an essay on climate change by next Friday")
"""
from typing import Dict, Any
from langchain.output_parsers import PydanticOutputParser
from pydantic import BaseModel, Field
class AssignmentCard(BaseModel):
title: str = Field(..., description="Short title of the assignment")
subject: str | None = Field(None, description="Subject or topic of the assignment")
deadline_hint: str | None = Field(
None,
description="Humanreadable hint about when the assignment is due",
)
deliverable_type: str | None = Field(
None,
description="What kind of work should be submitted (essay, report, code, etc.)",
)
grading_hints: str | None = Field(
None,
description="Any hints about how the assignment will be graded",
)
parser = PydanticOutputParser(pydantic_object=AssignmentCard)
# Prompt template that asks the model to output JSON matching AssignmentCard
PROMPT_TEMPLATE = (
"You are an assistant that converts a short informal description of a study assignment into a structured data card."
" Return only valid JSON that matches the following schema:\n{schema}\n"
" Description: {description}"
)
def parse_assignment(description: str) -> AssignmentCard:
"""Return an AssignmentCard parsed from the given description.
The function uses LangChain's PydanticOutputParser to enforce type safety.
"""
from langchain import PromptTemplate, LLMChain
from langchain.chat_models import ChatOpenAI
# Use a small model for demonstration; replace with your own key if needed.
llm = ChatOpenAI(temperature=0.2)
prompt = PromptTemplate(
input_variables=["description", "schema"],
template=PROMPT_TEMPLATE,
)
chain = LLMChain(llm=llm, prompt=prompt, output_parser=parser)
result = chain.run(description=description, schema=parser.get_format_instructions())
return result
if __name__ == "__main__":
import sys
if len(sys.argv) < 2:
print("Usage: python main.py '<assignment description>'")
sys.exit(1)
desc = sys.argv[1]
card = parse_assignment(desc)
print(card.json(indent=4))