prediction-market-agent-tooling 0.57.6__tar.gz → 0.57.7__tar.gz

{prediction_market_agent_tooling-0.57.6 → prediction_market_agent_tooling-0.57.7}/PKG-INFO

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

{prediction_market_agent_tooling-0.57.6 → prediction_market_agent_tooling-0.57.7}/prediction_market_agent_tooling/tools/caches/db_cache.py

@@ -98,7 +98,9 @@ def db_cache(
         if not api_keys.ENABLE_CACHE:
             return func(*args, **kwargs)
 
-        DBManager(api_keys).create_tables([FunctionCache])
+        DBManager(api_keys.sqlalchemy_db_url.get_secret_value()).create_tables(
+            [FunctionCache]
+        )
 
         # Convert *args and **kwargs to a single dictionary, where we have names for arguments passed as args as well.
         signature = inspect.signature(func)
@@ -143,7 +145,9 @@ def db_cache(
             return_type
         )
 
-        with DBManager(api_keys).get_session() as session:
+        with DBManager(
+            api_keys.sqlalchemy_db_url.get_secret_value()
+        ).get_session() as session:
             # Try to get cached result
             statement = (
                 select(FunctionCache)
@@ -196,7 +200,9 @@ def db_cache(
                 result=computed_result,
                 created_at=utcnow(),
             )
-            with DBManager(api_keys).get_session() as session:
+            with DBManager(
+                api_keys.sqlalchemy_db_url.get_secret_value()
+            ).get_session() as session:
                 logger.info(f"Saving {cache_entry} into database.")
                 session.add(cache_entry)
                 session.commit()

{prediction_market_agent_tooling-0.57.6 → prediction_market_agent_tooling-0.57.7}/prediction_market_agent_tooling/tools/db/db_manager.py

@@ -15,24 +15,26 @@ from prediction_market_agent_tooling.tools.caches.serializers import (
 class DBManager:
     _instances: dict[str, "DBManager"] = {}
 
-    def __new__(cls, api_keys: APIKeys | None = None) -> "DBManager":
-        sqlalchemy_db_url = (api_keys or APIKeys()).sqlalchemy_db_url
+    def __new__(cls, sqlalchemy_db_url: str | None = None) -> "DBManager":
+        if sqlalchemy_db_url is None:
+            sqlalchemy_db_url = APIKeys().sqlalchemy_db_url.get_secret_value()
+
         # Hash the secret value to not store secrets in plain text.
-        url_hash = hashlib.md5(
-            sqlalchemy_db_url.get_secret_value().encode()
-        ).hexdigest()
+        url_hash = hashlib.md5(sqlalchemy_db_url.encode()).hexdigest()
         # Return singleton per database connection.
         if url_hash not in cls._instances:
             instance = super(DBManager, cls).__new__(cls)
             cls._instances[url_hash] = instance
         return cls._instances[url_hash]
 
-    def __init__(self, api_keys: APIKeys | None = None) -> None:
+    def __init__(self, sqlalchemy_db_url: str | None = None) -> None:
         if hasattr(self, "_initialized"):
             return
-        sqlalchemy_db_url = (api_keys or APIKeys()).sqlalchemy_db_url
+        sqlalchemy_db_url = (
+            sqlalchemy_db_url or APIKeys().sqlalchemy_db_url.get_secret_value()
+        )
         self._engine = create_engine(
-            sqlalchemy_db_url.get_secret_value(),
+            sqlalchemy_db_url,
             json_serializer=json_serializer,
             json_deserializer=json_deserializer,
             pool_size=2,
@@ -53,7 +55,6 @@ class DBManager:
     def create_tables(
         self, sqlmodel_tables: Sequence[type[SQLModel]] | None = None
     ) -> None:
-        # Determine tables to create
         if sqlmodel_tables is not None:
             tables_to_create = []
             for sqlmodel_table in sqlmodel_tables:

{prediction_market_agent_tooling-0.57.6 → prediction_market_agent_tooling-0.57.7}/prediction_market_agent_tooling/tools/relevant_news_analysis/relevant_news_cache.py

@@ -25,7 +25,9 @@ class RelevantNewsCacheModel(SQLModel, table=True):
 
 class RelevantNewsResponseCache:
     def __init__(self, api_keys: APIKeys | None = None):
-        self.db_manager = DBManager(api_keys)
+        self.db_manager = DBManager(
+            (api_keys or APIKeys()).sqlalchemy_db_url.get_secret_value()
+        )
         self._initialize_db()
 
     def _initialize_db(self) -> None:

{prediction_market_agent_tooling-0.57.6 → prediction_market_agent_tooling-0.57.7}/pyproject.toml

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