models.py

from typing import Optional, TypedDict, List
from pydantic import BaseModel, Field

class GraphState(TypedDict):
    name: str
    questions: List[str]
    answers: List[str]
    curr_question: int
    done: bool
    score_evaluation: List[int]
    last_message: Optional[str]

class InterviewInput(BaseModel):
    candidate_message: str = Field(..., description="The candidate's response to the current question, instruction or message.")
    
class InterviewOutput(BaseModel):
    question: Optional[str] = Field(..., description="Current question asked by the interviewer.")
    message: Optional[str] = Field(..., description="A message by the interviewer. Can be anything other that intevriew question.")
    
class EvalInput(BaseModel):
    question: str = Field(..., description="A questions asked in the interview.")
    answer: str = Field(..., description="The candidate's answer to the interview question.")