output_parser.py

from pydantic import BaseModel
from langchain.output_parsers import PydanticOutputParser

class AttachmentStyle(BaseModel):
    speaker: str
    secured: float
    anxious_preoccupied: float
    dismissive_avoidant: float
    fearful_avoidant: float
    self_rating: int
    others_rating: int
    anxiety: int
    avoidance: int
    explanation: str

class BigFiveTraits(BaseModel):
    speaker: str
    extraversion: int
    agreeableness: int
    conscientiousness: int
    neuroticism: int
    openness: int
    explanation: str

class PersonalityDisorder(BaseModel):
    speaker: str
    depressed: int
    paranoid: int
    schizoid_schizotypal: int
    antisocial_psychopathic: int
    borderline_dysregulated: int
    narcissistic: int
    anxious_avoidant: int
    dependent_victimized: int
    obsessional: int
    explanation: str

attachment_parser = PydanticOutputParser(pydantic_object=AttachmentStyle)
bigfive_parser = PydanticOutputParser(pydantic_object=BigFiveTraits)
personality_parser = PydanticOutputParser(pydantic_object=PersonalityDisorder)