Merge pull request #6 from KintsugiMindfulWellness/update_parsing

updating prediction result parsing.

Author: peeyushkintsugi

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "kintsugi-python"
-version = "0.1.7"
+version = "0.1.8"
 
 description = "Python SDK to access Kintsugi Voice API V2."
 authors = [

src/kintsugi/model.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 from datetime import datetime
-
+from dataclasses import dataclass, field
+from typing import Optional
 
 class FeedbackScore:
     category: str
@@ -9,14 +10,15 @@ class FeedbackScore:
     phq_9: str
 
 
+@dataclass
 class Prediction:
     session_id: str
-    predicted_score: dict = {}
-    feedback_score: FeedbackScore
     created_at: datetime
     updated_at: datetime
     is_calibrated: bool
     status: str
+    feedback_score: Optional[FeedbackScore] = None
+    predicted_score: dict = field(default_factory=dict)
 
     def get_score(self, category: str) -> str | None:
         return self.predicted_score.get(category, None)

src/kintsugi/parsers.py

@@ -21,13 +21,14 @@ def parse(self, data: dict) -> FeedbackScore:
 
 class PredictionParser:
     def parse(self, data: dict) -> Prediction:
-        output = Prediction()
+        output = Prediction(
+            session_id=data['session_id'],
+            created_at=data['created_at'],
+            updated_at=data['updated_at'],
+            is_calibrated=data['is_calibrated'],
+            status=data['status'],
+        )
 
-        output.session_id = data['session_id']
-        output.created_at = data['created_at']
-        output.updated_at = data['updated_at']
-        output.is_calibrated = data['is_calibrated']
-        output.status = data['status']
 
         if output.status != 'processing':
             categories = data['model_category'].split(',')