flows/domain.py

from pydantic import ConfigDict

from talentum_score.domain import (
    CompanyEvaluation,
    JobPosting,
    PostEvaluation,
    TalentumScore,
)
from talentum_score.flows.base import FlowState


class JobEvaluatorState(FlowState[JobPosting, TalentumScore]):
    model_config = ConfigDict(arbitrary_types_allowed=True)
    company_analysis: CompanyEvaluation | None = None
    description_eval: PostEvaluation | None = None