Spaces:
Runtime error
Runtime error
File size: 1,064 Bytes
8df2e34 e3551a8 9e68e0a 1cca70d e3551a8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 |
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) |