prediction-market-agent-tooling 0.64.2.dev603__tar.gz → 0.64.2.dev604__tar.gz

{prediction_market_agent_tooling-0.64.2.dev603 → prediction_market_agent_tooling-0.64.2.dev604}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: prediction-market-agent-tooling
-Version: 0.64.2.dev603
+Version: 0.64.2.dev604
 Summary: Tools to benchmark, deploy and monitor prediction market agents.
 Author: Gnosis
 Requires-Python: >=3.10,<3.13

{prediction_market_agent_tooling-0.64.2.dev603 → prediction_market_agent_tooling-0.64.2.dev604}/prediction_market_agent_tooling/tools/caches/db_cache.py

@@ -3,6 +3,7 @@ import inspect
 import json
 from datetime import timedelta
 from functools import wraps
+from types import UnionType
 from typing import (
     Any,
     Callable,
@@ -276,6 +277,17 @@ def convert_cached_output_to_pydantic(return_type: Any, data: Any) -> Any:
                 ): convert_cached_output_to_pydantic(value_type, v)
                 for k, v in data.items()
             }
+        # If the origin is a union and one of the unions is basemodel, convert it to it.
+        elif (
+            origin is UnionType
+            and (
+                base_model_from_args := next(
+                    (x for x in args if issubclass(x, BaseModel)), None
+                )
+            )
+            is not None
+        ):
+            return base_model_from_args.model_validate(data)
         else:
             # If the origin is not a dictionary, return the data as is
             return data

{prediction_market_agent_tooling-0.64.2.dev603 → prediction_market_agent_tooling-0.64.2.dev604}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "prediction-market-agent-tooling"
-version = "0.64.2.dev603"
+version = "0.64.2.dev604"
 description = "Tools to benchmark, deploy and monitor prediction market agents."
 authors = ["Gnosis"]
 readme = "README.md"