nodes/description_eval.py

from talentum_score.configs import talentum_score_config
from talentum_score.const import PROMPTS_PATH
from talentum_score.domain import PostEvaluation
from talentum_score.nodes.generic_llm import GenericLLMNode


class DescriptionEvalNode(GenericLLMNode):
    def __init__(self, output_key: str = "description_eval"):
        query = "Job Posting:\n\t- Description: {state.flow_input.description}\n\t- Position: {state.flow_input.position}"
        super().__init__(
            system_prompt_path=PROMPTS_PATH / "description_eval_system.txt",
            query=query,
            output_key=output_key,
            model="openai:" + talentum_score_config.openai_model,
            structured_output=PostEvaluation,
            api_key=talentum_score_config.openai_api_key,
        )