Update output_parser.py

output_parser.py

@@ -3,7 +3,7 @@ from pydantic import BaseModel, Field
 import json
 
 class AttachmentStyle(BaseModel):
-    speaker: str = "Unknown Speaker"
+    Speaker: str = "Unknown Speaker"
     Secured: float = Field(0.0, alias="Secured")
     Anxious_Preoccupied: float = Field(0.0, alias="Anxious-Preoccupied")
     Dismissive_Avoidant: float = Field(0.0, alias="Dismissive-Avoidant")
@@ -15,7 +15,7 @@ class AttachmentStyle(BaseModel):
     Explanation: str = "No explanation provided"
 
 class BigFiveTraits(BaseModel):
-    speaker: str = "Unknown Speaker"
+    Speaker: str = "Unknown Speaker"
     Extraversion: int = 0
     Agreeableness: int = 0
     Conscientiousness: int = 0
@@ -24,7 +24,7 @@ class BigFiveTraits(BaseModel):
     Explanation: str = "No explanation provided"
 
 class PersonalityDisorder(BaseModel):
-    speaker: str = "Unknown Speaker"
+    Speaker: str = "Unknown Speaker"
     Depressed: int = 0
     Paranoid: int = 0
     Schizoid_Schizotypal: int = Field(0, alias="Schizoid-Schizotypal")
@@ -60,7 +60,7 @@ class AttachmentStyleOutputParser:
         {
           "speaker_analyses": [
             {
-              "speaker": "Speaker 1",
+              "Speaker": "1",
               "attachment_styles": {
                 "Secured": 0.5,
                 "Anxious-Preoccupied": 0.2,
@@ -94,7 +94,7 @@ class AttachmentStyleOutputParser:
               }
             },
             {
-              "speaker": "Speaker 2",
+              "Speaker": "2",
               ...
             }
           ]