fix: filter for valid fields

Author: joschkabraun

parea/helpers.py

@@ -105,26 +105,25 @@ def structure_float_or_none(obj: Any, cl: type) -> Optional[float]:
     converter.register_structure_hook(float, structure_float_or_none)
     converter.register_structure_hook(Optional[float], structure_float_or_none)
 
+    # Helper function to filter valid fields
+    def filter_valid_fields(obj, cls):
+        return {k: v for k, v in obj.items() if k in fields_dict(cls)}
+
     # Register structure hooks for nested types
     converter.register_structure_hook(Role, lambda obj, _: Role(obj))
-    converter.register_structure_hook(Message, lambda obj, _: Message(**obj))
-    converter.register_structure_hook(LLMInputs, lambda obj, _: LLMInputs(**obj))
-    converter.register_structure_hook(EvaluationResult, lambda obj, _: EvaluationResult(**obj))
-    converter.register_structure_hook(TraceLogImage, lambda obj, _: TraceLogImage(**obj))
-    converter.register_structure_hook(TraceLogCommentSchema, lambda obj, _: TraceLogCommentSchema(**obj))
-    converter.register_structure_hook(TraceLogAnnotationSchema, lambda obj, _: TraceLogAnnotationSchema(**obj))
-
-    def structure_model_params(obj, _):
-        valid_params = {k: v for k, v in obj.items() if k in fields_dict(ModelParams)}
-        return ModelParams(**valid_params)
-
-    converter.register_structure_hook(ModelParams, structure_model_params)
+    converter.register_structure_hook(Message, lambda obj, _: Message(**filter_valid_fields(obj, Message)))
+    converter.register_structure_hook(EvaluationResult, lambda obj, _: EvaluationResult(**filter_valid_fields(obj, EvaluationResult)))
+    converter.register_structure_hook(TraceLogImage, lambda obj, _: TraceLogImage(**filter_valid_fields(obj, TraceLogImage)))
+    converter.register_structure_hook(TraceLogCommentSchema, lambda obj, _: TraceLogCommentSchema(**filter_valid_fields(obj, TraceLogCommentSchema)))
+    converter.register_structure_hook(TraceLogAnnotationSchema, lambda obj, _: TraceLogAnnotationSchema(**filter_valid_fields(obj, TraceLogAnnotationSchema)))
+    converter.register_structure_hook(ModelParams, lambda obj, _: ModelParams(**filter_valid_fields(obj, ModelParams)))
 
     def structure_llm_inputs(obj, _):
         if obj is None:
             return None
+        valid_fields = filter_valid_fields(obj, LLMInputs)
         kwargs = {}
-        for key, value in obj.items():
+        for key, value in valid_fields.items():
             if key == "messages":
                 kwargs[key] = [converter.structure(msg, Message) for msg in value]
             elif key == "model_params":
@@ -136,8 +135,9 @@ def structure_llm_inputs(obj, _):
     converter.register_structure_hook(LLMInputs, structure_llm_inputs)
 
     def structure_trace_log_tree(data, _):
+        valid_fields = filter_valid_fields(data, TraceLogTree)
         kwargs = {}
-        for key, value in data.items():
+        for key, value in valid_fields.items():
             if key == "children_logs":
                 kwargs["children_logs"] = [structure_trace_log_tree(child, TraceLogTree) for child in value]
             elif key == "configuration":
@@ -150,7 +150,7 @@ def structure_trace_log_tree(data, _):
                 kwargs["comments"] = [converter.structure(comment, TraceLogCommentSchema) for comment in value]
             elif key == "annotations":
                 kwargs["annotations"] = {int(k): {sk: converter.structure(sv, TraceLogAnnotationSchema) for sk, sv in v.items()} for k, v in value.items()}
-            elif key in fields_dict(TraceLogTree):
+            else:
                 field_type = fields_dict(TraceLogTree)[key].type
                 kwargs[key] = converter.structure(value, field_type)
         return TraceLogTree(**kwargs)

pyproject.toml

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