Update output_parser.py

output_parser.py

@@ -4,9 +4,9 @@ import json
 
 class AttachmentStyle(BaseModel):
     secured: float = 0.0
-    anxious_preoccupied: float = Field(0.0, alias="AnxiousPreoccupied")
-    dismissive_avoidant: float = Field(0.0, alias="DismissiveAvoidant")
-    fearful_avoidant: float = Field(0.0, alias="FearfulAvoidant")
+    anxious_preoccupied: float = Field(0.0, alias="Anxious-Preoccupied")
+    dismissive_avoidant: float = Field(0.0, alias="Dismissive-Avoidant")
+    fearful_avoidant: float = Field(0.0, alias="Fearful-Avoidant")
     self_model: int = Field(0, alias="Self")
     others_model: int = Field(0, alias="Others")
     anxiety: int = 0
@@ -24,12 +24,12 @@ class BigFiveTraits(BaseModel):
 class PersonalityDisorder(BaseModel):
     depressed: int = 0
     paranoid: int = 0
-    schizoid_schizotypal: int = Field(0, alias="SchizoidSchizotypal")
-    antisocial_psychopathic: int = Field(0, alias="AntisocialPsychopathic")
-    borderline_dysregulated: int = Field(0, alias="BorderlineDysregulated")
+    schizoid_schizotypal: int = Field(0, alias="Schizoid-Schizotypal")
+    antisocial_psychopathic: int = Field(0, alias="Antisocial-Psychopathic")
+    borderline_dysregulated: int = Field(0, alias="Borderline-Dysregulated")
     narcissistic: int = 0
-    anxious_avoidant: int = Field(0, alias="AnxiousAvoidant")
-    dependent_victimized: int = Field(0, alias="DependentVictimized")
+    anxious_avoidant: int = Field(0, alias="Anxious-Avoidant")
+    dependent_victimized: int = Field(0, alias="Dependent-Victimized")
     obsessional: int = 0
     explanation: str = "No explanation provided"
 
@@ -71,8 +71,8 @@ class OutputParser:
             general_impression=general_impression_str,
             attachment_style=AttachmentStyle(
                 secured=attachment_styles.get("Secured", 0.0),
-                anxious_preoccupied=attachment_styles.get("AnxiousPreoccupied", 0.0),
-                dismissive_avoidant=attachment_styles.get("DismissiveAvoidant", 0.0),
+                anxious_preoccupied=attachment_styles.get("Anxious-Preoccupied", 0.0),
+                dismissive_avoidant=attachment_styles.get("Dismissive-Avoidant", 0.0),
                 fearful_avoidant=attachment_styles.get("FearfulAvoidant", 0.0),
                 self_model=attachment_styles.get("Self", 0),
                 others_model=attachment_styles.get("Others", 0),
@@ -91,12 +91,12 @@ class OutputParser:
             personality_disorder=PersonalityDisorder(
                 depressed=personality_disorders.get("Depressed", 0),
                 paranoid=personality_disorders.get("Paranoid", 0),
-                schizoid_schizotypal=personality_disorders.get("SchizoidSchizotypal", 0),
-                antisocial_psychopathic=personality_disorders.get("AntisocialPsychopathic", 0),
-                borderline_dysregulated=personality_disorders.get("BorderlineDysregulated", 0),
+                schizoid_schizotypal=personality_disorders.get("Schizoid-Schizotypal", 0),
+                antisocial_psychopathic=personality_disorders.get("Antisocial-Psychopathic", 0),
+                borderline_dysregulated=personality_disorders.get("Borderline-Dysregulated", 0),
                 narcissistic=personality_disorders.get("Narcissistic", 0),
-                anxious_avoidant=personality_disorders.get("AnxiousAvoidant", 0),
-                dependent_victimized=personality_disorders.get("DependentVictimized", 0),
+                anxious_avoidant=personality_disorders.get("Anxious-Avoidant", 0),
+                dependent_victimized=personality_disorders.get("Dependent-Victimized", 0),
                 obsessional=personality_disorders.get("Obsessional", 0),
                 explanation=personality_disorders.get("Explanation", "No explanation provided")
             )