prediction-market-agent-tooling 0.57.6__py3-none-any.whl → 0.57.7__py3-none-any.whl

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/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/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.dist-info → prediction_market_agent_tooling-0.57.7.dist-info}/METADATA

@@ -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.dist-info → prediction_market_agent_tooling-0.57.7.dist-info}/RECORD

@@ -73,14 +73,14 @@ prediction_market_agent_tooling/tools/betting_strategies/market_moving.py,sha256
 prediction_market_agent_tooling/tools/betting_strategies/minimum_bet_to_win.py,sha256=-FUSuQQgjcWSSnoFxnlAyTeilY6raJABJVM2QKkFqAY,438
 prediction_market_agent_tooling/tools/betting_strategies/stretch_bet_between.py,sha256=THMXwFlskvzbjnX_OiYtDSzI8XVFyULWfP2525_9UGc,429
 prediction_market_agent_tooling/tools/betting_strategies/utils.py,sha256=kpIb-ci67Vc1Yqqaa-_S4OUkbhWSIYog4_Iwp69HU_k,97
-prediction_market_agent_tooling/tools/caches/db_cache.py,sha256=rkA7somZZo5G6OZVvNrCuCVc4kn7TUoPxmOq4J6BZ54,10391
+prediction_market_agent_tooling/tools/caches/db_cache.py,sha256=aafau_n_AUbLIwkyIRiTPgKB0dmM0767mSqyPDLF2A4,10576
 prediction_market_agent_tooling/tools/caches/inmemory_cache.py,sha256=ZW5iI5rmjqeAebu5T7ftRnlkxiL02IC-MxCfDB80x7w,1506
 prediction_market_agent_tooling/tools/caches/serializers.py,sha256=upSXN5__rmRlzJ6tv1h7FodKzJu9eCkFrN_zeuwroJM,2151
 prediction_market_agent_tooling/tools/contract.py,sha256=tJHwbDXRRRS7g-vIzqfYTOPQBcU5lqb9o_FF9OECZQ8,22941
 prediction_market_agent_tooling/tools/costs.py,sha256=EaAJ7v9laD4VEV3d8B44M4u3_oEO_H16jRVCdoZ93Uw,954
 prediction_market_agent_tooling/tools/custom_exceptions.py,sha256=Fh8z1fbwONvP4-j7AmV_PuEcoqb6-QXa9PJ9m7guMcM,93
 prediction_market_agent_tooling/tools/datetime_utc.py,sha256=2JSWF7AXQnv04_D_cu9Vmdkq0TWmGJ1QcK9AeqrA-U8,2765
-prediction_market_agent_tooling/tools/db/db_manager.py,sha256=_d_oJe4GZ59Htq0iOosRz8yEps2c6MmfTNvNvMLpL8Y,3002
+prediction_market_agent_tooling/tools/db/db_manager.py,sha256=GtzHH1NLl8HwqC8Z7s6eTlIQXuV0blxfaV2PeQrBnfQ,3013
 prediction_market_agent_tooling/tools/google_utils.py,sha256=t3_UEEvKX3L0biSIQ560GdRbllQ6eprhK_upE243A-0,3185
 prediction_market_agent_tooling/tools/hexbytes_custom.py,sha256=Bp94qgPjvjWf1Vb4lNzGFDXRdThw1rJ91vL6r2PWq5E,2096
 prediction_market_agent_tooling/tools/httpx_cached_client.py,sha256=RxD-hwtZCMctnMwfzy8t51W9Z9gxFGtDYxBIMChazpc,406
@@ -95,7 +95,7 @@ prediction_market_agent_tooling/tools/omen/reality_accuracy.py,sha256=M1SF7iSW1g
 prediction_market_agent_tooling/tools/parallelism.py,sha256=6Gou0hbjtMZrYvxjTDFUDZuxmE2nqZVbb6hkg1hF82A,1022
 prediction_market_agent_tooling/tools/relevant_news_analysis/data_models.py,sha256=95l84aztFaxcRLLcRQ46yKJbIlOEuDAbIGLouyliDzA,1316
 prediction_market_agent_tooling/tools/relevant_news_analysis/relevant_news_analysis.py,sha256=CddJem7tk15NAudJDwmuL8znTycbR-YI8kTNtd3LzG8,5474
-prediction_market_agent_tooling/tools/relevant_news_analysis/relevant_news_cache.py,sha256=ErLp284YlD7k_M8Si9OT86HuzfmSo1bPlLJtegXx1TA,3011
+prediction_market_agent_tooling/tools/relevant_news_analysis/relevant_news_cache.py,sha256=kNWq92T11Knb9mYBZlMiZUzOpKgCd-5adanylQUMRJA,3085
 prediction_market_agent_tooling/tools/safe.py,sha256=9vxGGLvSPnfy-sxUFDpBTe8omqpGXP7MzvGPp6bRxrU,5197
 prediction_market_agent_tooling/tools/singleton.py,sha256=CiIELUiI-OeS7U7eeHEt0rnVhtQGzwoUdAgn_7u_GBM,729
 prediction_market_agent_tooling/tools/streamlit_user_login.py,sha256=NXEqfjT9Lc9QtliwSGRASIz1opjQ7Btme43H4qJbzgE,3010
@@ -104,8 +104,8 @@ prediction_market_agent_tooling/tools/tavily/tavily_search.py,sha256=Kw2mXNkMTYT
 prediction_market_agent_tooling/tools/transaction_cache.py,sha256=K5YKNL2_tR10Iw2TD9fuP-CTGpBbZtNdgbd0B_R7pjg,1814
 prediction_market_agent_tooling/tools/utils.py,sha256=WvuUCHgMCiMq8_wMm5PHNwvLhcdDk2zGKaAM8OUC-qY,6438
 prediction_market_agent_tooling/tools/web3_utils.py,sha256=wqUDCed3iNrn1Wao1iwGN6tzIrhpzrTRj319wlveJEo,12275
-prediction_market_agent_tooling-0.57.6.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
-prediction_market_agent_tooling-0.57.6.dist-info/METADATA,sha256=VCWshrH1hcQnfq30Exw4IinOZLRsS0gS4AqtpO0vYhY,8188
-prediction_market_agent_tooling-0.57.6.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-prediction_market_agent_tooling-0.57.6.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
-prediction_market_agent_tooling-0.57.6.dist-info/RECORD,,
+prediction_market_agent_tooling-0.57.7.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
+prediction_market_agent_tooling-0.57.7.dist-info/METADATA,sha256=dDVW3NpNlZdOWksAXIUHFFynWSvxbA_kZXffQ-Af1Ys,8188
+prediction_market_agent_tooling-0.57.7.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+prediction_market_agent_tooling-0.57.7.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
+prediction_market_agent_tooling-0.57.7.dist-info/RECORD,,