agenta 0.39.4__py3-none-any.whl → 0.40.0__py3-none-any.whl

agenta/sdk/decorators/tracing.py

@@ -297,7 +297,9 @@ class instrument:  # pylint: disable=invalid-name
             not in (
                 ignore
                 if isinstance(ignore, list)
-                else io.keys() if ignore is True else []
+                else io.keys()
+                if ignore is True
+                else []
             )
         }
 

agenta/sdk/tracing/attributes.py

@@ -103,19 +103,63 @@ def _encode_key(
     return f"ag.{namespace}.{key}"
 
 
-def _encode_value(
-    value: Any,
-) -> Optional[Attribute]:
+def _make_serializable(value: Any) -> Any:
+    """
+    Transform complex nested structures into JSON-serializable form.
+    Handles Pydantic models, nested dictionaries and lists recursively.
+    """
+    if value is None or isinstance(value, (str, int, float, bool, bytes)):
+        return value
+
+    # Handle Pydantic objects (prioritize v2 over v1 API)
+    if hasattr(value, "model_dump"):  # Pydantic v2
+        return value.model_dump()
+    elif hasattr(value, "dict"):  # Pydantic v1
+        return value.dict()
+
+    if isinstance(value, dict):
+        try:
+            # Test serialization without modifying - optimizes for already-serializable dicts
+            dumps(
+                value
+            )  # If serialization fails, we'll catch the exception and process deeply
+            return value  # Avoid unnecessary recursion for serializable dicts
+        except TypeError:
+            return {k: _make_serializable(v) for k, v in value.items()}
+    elif isinstance(value, list):
+        try:
+            # Test serialization without modifying - optimizes for already-serializable lists
+            dumps(
+                value
+            )  # If serialization fails, we'll catch the exception and process deeply
+            return value  # Avoid unnecessary recursion for serializable lists
+        except TypeError:
+            return [_make_serializable(item) for item in value]
+
+    return repr(value)
+
+
+def _encode_value(value: Any) -> Optional[Attribute]:
+    """
+    Encode values for tracing, ensuring proper JSON serialization.
+    Adds the @ag.type=json: prefix only to appropriate values.
+    """
     if value is None:
         return None
 
     if isinstance(value, (str, int, float, bool, bytes)):
         return value
 
-    if isinstance(value, dict) or isinstance(value, list):
-        encoded = dumps(value)
-        value = "@ag.type=json:" + encoded
-        return value
+    try:
+        if (
+            isinstance(value, (dict, list))
+            or hasattr(value, "model_dump")
+            or hasattr(value, "dict")
+        ):
+            serializable_value = _make_serializable(value)
+            return "@ag.type=json:" + dumps(serializable_value)
+    except TypeError:
+        pass
 
     return repr(value)
 

{agenta-0.39.4.dist-info → agenta-0.40.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: agenta
-Version: 0.39.4
+Version: 0.40.0
 Summary: The SDK for agenta is an open-source LLMOps platform.
 Keywords: LLMOps,LLM,evaluation,prompt engineering
 Author: Mahmoud Mabrouk

{agenta-0.39.4.dist-info → agenta-0.40.0.dist-info}/RECORD

@@ -207,7 +207,7 @@ agenta/sdk/context/routing.py,sha256=FEsjw8EttI1SMyUo96ptcUsvHJnhoKwdr1szlkxxJNU
 agenta/sdk/context/tracing.py,sha256=xjErrXP1Nq1AfL-Cif1l-lNEfs12eQ3v_VCRgoKe7nY,743
 agenta/sdk/decorators/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agenta/sdk/decorators/routing.py,sha256=RUq55Y3GHkQilJlobQmUHnHBmPZrhHgQyPxKF_D9RXo,24799
-agenta/sdk/decorators/tracing.py,sha256=Px4X9thTlBH7rmq-Wi7BAxEhy_xdfb7XsEEmuaq-SzQ,10077
+agenta/sdk/decorators/tracing.py,sha256=n3AZWw8HHKUoIHD79_ktPQcoZRIw3RthSxU5xkO-skg,10109
 agenta/sdk/litellm/__init__.py,sha256=Bpz1gfHQc0MN1yolWcjifLWznv6GjHggvRGQSpxpihM,37
 agenta/sdk/litellm/litellm.py,sha256=vZYuzcSss04zE258eC-3xjKjY-HToafB1ekjmKjMR4U,10202
 agenta/sdk/litellm/mockllm.py,sha256=8V6dqdv8eA4P-VoXIwHNYlIjHG189P14POSfSfluVw0,678
@@ -230,7 +230,7 @@ agenta/sdk/middleware/otel.py,sha256=lHzhGUv4fq2RPuTPH2keJ16v-_cBUrLrTqWzHUmEVdI
 agenta/sdk/middleware/vault.py,sha256=Hd_S8Lw1PLFdavnFwW_bb-HYlyYvaCxNJNwjvd2Bxyc,4043
 agenta/sdk/router.py,sha256=mOguvtOwl2wmyAgOuWTsf98pQwpNiUILKIo67W_hR3A,119
 agenta/sdk/tracing/__init__.py,sha256=rQNe5-zT5Kt7_CDhq-lnUIi1EYTBVzVf_MbfcIxVD98,41
-agenta/sdk/tracing/attributes.py,sha256=zh8JQZSeYCLBeIRSopKJx6QQ-WEgw08Cr64DS_WOcT8,3833
+agenta/sdk/tracing/attributes.py,sha256=DwjjOk3mGOvz0jYu8EYr3hhItvawK5GX80_MfciqPrc,5559
 agenta/sdk/tracing/conventions.py,sha256=JBtznBXZ3aRkGKkLl7cPwdMNh3w1G-H2Ta2YrAxbr38,950
 agenta/sdk/tracing/exporters.py,sha256=ib-VSMaqKn3XDrChjjHlsPJ9Fu3ePgIYR-TAWxSJv9Q,3251
 agenta/sdk/tracing/inline.py,sha256=tY9mH1zMlJMqf2WWKJdakLtMHJKNeYRaynqvsPeqpTY,31312
@@ -250,6 +250,6 @@ agenta/sdk/utils/logging.py,sha256=gBk2ecRvltN7f8qtnUHXBiZPWOdArN_b5civ_3w_Oek,8
 agenta/sdk/utils/preinit.py,sha256=YlJL7RLfel0R7DFp-jK7OV-z4ZIQJM0oupYlk7g8b5o,1278
 agenta/sdk/utils/singleton.py,sha256=17Ph7LGnnV8HkPjImruKita2ni03Ari5jr0jqm__4sc,312
 agenta/sdk/utils/timing.py,sha256=hVzkF7ObjrBlRcB2lMowE8nUkWgYqU2kIJoqRZNXKdM,1556
-agenta-0.39.4.dist-info/METADATA,sha256=kj11TYSg4iuFeZL2xqhoqlxuvh5XtnvaNhoClWxRXP8,31346
-agenta-0.39.4.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-agenta-0.39.4.dist-info/RECORD,,
+agenta-0.40.0.dist-info/METADATA,sha256=Oga16OtunVBNMQivKGtc_cT0KtmTdtU_kMCfn6EySok,31346
+agenta-0.40.0.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+agenta-0.40.0.dist-info/RECORD,,