prediction-market-agent-tooling 0.56.2.dev146__tar.gz → 0.56.2.dev148__tar.gz

{prediction_market_agent_tooling-0.56.2.dev146 → prediction_market_agent_tooling-0.56.2.dev148}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: prediction-market-agent-tooling
-Version: 0.56.2.dev146
+Version: 0.56.2.dev148
 Summary: Tools to benchmark, deploy and monitor prediction market agents.
 Author: Gnosis
 Requires-Python: >=3.10,<3.12

{prediction_market_agent_tooling-0.56.2.dev146 → prediction_market_agent_tooling-0.56.2.dev148}/prediction_market_agent_tooling/tools/caches/db_cache.py

@@ -22,7 +22,6 @@ from sqlmodel import Field, Session, SQLModel, create_engine, desc, select
 from prediction_market_agent_tooling.config import APIKeys
 from prediction_market_agent_tooling.loggers import logger
 from prediction_market_agent_tooling.tools.datetime_utc import DatetimeUTC
-from prediction_market_agent_tooling.tools.pickle_utils import InitialiseNonPickable
 from prediction_market_agent_tooling.tools.utils import utcnow
 
 FunctionT = TypeVar("FunctionT", bound=Callable[..., Any])
@@ -91,17 +90,6 @@ def db_cache(
         return decorator
 
     api_keys = api_keys if api_keys is not None else APIKeys()
-    wrapped_engine = InitialiseNonPickable(
-        lambda: create_engine(
-            api_keys.sqlalchemy_db_url.get_secret_value(),
-            # Use custom json serializer and deserializer, because otherwise, for example `datetime` serialization would fail.
-            json_serializer=json_serializer,
-            json_deserializer=json_deserializer,
-        )
-    )
-
-    if api_keys.ENABLE_CACHE:
-        SQLModel.metadata.create_all(wrapped_engine.get_value())
 
     @wraps(func)
     def wrapper(*args: Any, **kwargs: Any) -> Any:
@@ -109,7 +97,14 @@ def db_cache(
         if not api_keys.ENABLE_CACHE:
             return func(*args, **kwargs)
 
-        engine = wrapped_engine.get_value()
+        engine = create_engine(
+            api_keys.sqlalchemy_db_url.get_secret_value(),
+            # Use custom json serializer and deserializer, because otherwise, for example `datetime` serialization would fail.
+            json_serializer=json_serializer,
+            json_deserializer=json_deserializer,
+            pool_size=1,
+        )
+        SQLModel.metadata.create_all(engine)
 
         # Convert *args and **kwargs to a single dictionary, where we have names for arguments passed as args as well.
         signature = inspect.signature(func)
@@ -213,6 +208,7 @@ def db_cache(
                 session.add(cache_entry)
                 session.commit()
 
+        engine.dispose()
         return computed_result
 
     return cast(FunctionT, wrapper)

{prediction_market_agent_tooling-0.56.2.dev146 → prediction_market_agent_tooling-0.56.2.dev148}/prediction_market_agent_tooling/tools/relevant_news_analysis/relevant_news_cache.py

@@ -25,9 +25,12 @@ class RelevantNewsCacheModel(SQLModel, table=True):
 class RelevantNewsResponseCache:
     def __init__(self, sqlalchemy_db_url: str | None = None):
         self.engine = create_engine(
-            sqlalchemy_db_url
-            if sqlalchemy_db_url
-            else APIKeys().sqlalchemy_db_url.get_secret_value()
+            (
+                sqlalchemy_db_url
+                if sqlalchemy_db_url
+                else APIKeys().sqlalchemy_db_url.get_secret_value()
+            ),
+            pool_size=1,
         )
         self._initialize_db()
 

{prediction_market_agent_tooling-0.56.2.dev146 → prediction_market_agent_tooling-0.56.2.dev148}/pyproject.toml

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

prediction_market_agent_tooling-0.56.2.dev146/prediction_market_agent_tooling/tools/pickle_utils.py

@@ -1,31 +0,0 @@
-import typing as t
-
-InitialisedValue = t.TypeVar("InitialisedValue")
-
-
-class InitialiseNonPickable(t.Generic[InitialisedValue]):
-    """
-    Use this class to wrap values that you want to be shared within a thread,
-    but they are re-initialised for a new processes.
-
-    Initialiser for the value still needs to be pickable.
-    """
-
-    def __init__(self, initialiser: t.Callable[[], InitialisedValue]) -> None:
-        self.value: InitialisedValue | None = None
-        self.initialiser = initialiser
-
-    def __getstate__(self) -> dict[str, t.Any]:
-        # During pickling, always return `value` as just None, which is pickable and this class will re-initialise it in `get_value` when called.
-        return {"value": None, "initialiser": self.initialiser}
-
-    def __setstate__(self, d: dict[str, t.Any]) -> None:
-        self.value = d["value"]
-        self.initialiser = d["initialiser"]
-
-    def get_value(self) -> InitialisedValue:
-        """Use this function to get the wrapped value, which will be initialised if necessary."""
-        if self.value is None:
-            self.value = self.initialiser()
-
-        return self.value