prediction-market-agent-tooling 0.56.2.dev138__tar.gz → 0.56.2.dev139__tar.gz

{prediction_market_agent_tooling-0.56.2.dev138 → prediction_market_agent_tooling-0.56.2.dev139}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: prediction-market-agent-tooling
-Version: 0.56.2.dev138
+Version: 0.56.2.dev139
 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.dev138 → prediction_market_agent_tooling-0.56.2.dev139}/prediction_market_agent_tooling/tools/caches/db_cache.py

@@ -1,6 +1,7 @@
 import hashlib
 import inspect
 import json
+import logging
 from datetime import date, timedelta
 from functools import wraps
 from typing import (
@@ -15,12 +16,11 @@ from typing import (
 )
 
 from pydantic import BaseModel
-from sqlalchemy import Column, Engine
+from sqlalchemy import Column
 from sqlalchemy.dialects.postgresql import JSONB
 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
@@ -90,17 +90,16 @@ def db_cache(
 
         return decorator
 
+    logger = logging  # For some reason, `loguru` doesn't work in multiprocessing if it's used in in-function-defined functions, so force-patch for just standard logging here.
     api_keys = api_keys if api_keys is not None else APIKeys()
-
-    def engine_initialiser() -> Engine:
-        return create_engine(
+    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,
         )
-
-    wrapped_engine = InitialiseNonPickable(engine_initialiser)
+    )
 
     if api_keys.ENABLE_CACHE:
         SQLModel.metadata.create_all(wrapped_engine.get_value())

{prediction_market_agent_tooling-0.56.2.dev138 → prediction_market_agent_tooling-0.56.2.dev139}/pyproject.toml

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