랭체인으로 코드 문제 테스트케이스 생성기 만들기_241007

def generate_question(quiz_domain: str, estimated_time: str, session_id: str):
    question_prompt = ChatPromptTemplate.from_messages([
        ("system", """다음 조건을 만족하는 코딩 문제를 하나 제안해 주세요.
도메인: "{quiz_domain}"
예상시간: "{estimated_time}"
요청이 있기 전까지는 예시 코드 답안을 제공하지 마세요.
문제: """),
        MessagesPlaceholder(variable_name="history"),
    ])

    question_stream = chatsession | question_prompt

    response = question_stream.invoke(
        input={"quiz_domain": quiz_domain, "estimated_time": estimated_time},
        config={"configurable": {"session_id": session_id}},
    )
    return response.content

def generate_answer(session_id: str):
    answer_prompt = ChatPromptTemplate.from_messages([
        ("system", """위 문제의 답을 제공해주세요."""),
        MessagesPlaceholder(variable_name="history"),
    ])
    
    answer_stream = chatsession | answer_prompt

    response = answer_stream.invoke(
        input={},
        config={"configurable": {"session_id": session_id}},
    )
    return response.content