Merge pull request #933 from parea-ai/fix-experiment-deserialization

Fix experiment deserialization

Author: jalexanderII

parea/cookbook/evals_and_experiments/list_experiments.py

@@ -9,7 +9,7 @@
 
 p = Parea(api_key=os.getenv("PAREA_API_KEY"))
 
-experiments = p.list_experiments(ListExperimentUUIDsFilters(experiment_name_filter="Greeting"))
+experiments = p.list_experiments(ListExperimentUUIDsFilters(experiment_name_filter="greeting"))
 print(f"Num. experiments: {len(experiments)}")
 trace_logs = p.get_experiment_trace_logs(experiments[0].uuid)
 print(f"Num. trace logs: {len(trace_logs)}")

parea/schemas/log.py

@@ -33,10 +33,10 @@ def to_dict(self) -> Dict[str, str]:
 
 @define
 class ModelParams:
-    temp: Union[float, None] = 1.0
-    top_p: Union[float, None] = 1.0
-    frequency_penalty: Union[float, None] = 0.0
-    presence_penalty: Union[float, None] = 0.0
+    temp: Optional[float] = 1.0
+    top_p: Optional[float] = 1.0
+    frequency_penalty: Optional[float] = 0.0
+    presence_penalty: Optional[float] = 0.0
     max_length: Optional[int] = None
     response_format: Optional[dict] = None
     safe_prompt: Optional[bool] = None

parea/schemas/models.py

@@ -147,7 +147,7 @@ class TraceLog(EvaluatedLog):
 
 @define
 class TraceLogTree(TraceLog):
-    children_logs: Optional[List[TraceLog]] = field(factory=list)
+    children_logs: Optional[List["TraceLogTree"]] = field(factory=list)
 
 
 @define

pyproject.toml

@@ -6,7 +6,7 @@ build-backend = "poetry.core.masonry.api"
 [tool.poetry]
 name = "parea-ai"
 packages = [{ include = "parea" }]
-version = "0.2.174"
+version = "0.2.175"
 description = "Parea python sdk"
 readme = "README.md"
 authors = ["joel-parea-ai <[email protected]>"]