prediction-market-agent-tooling 0.48.15__tar.gz → 0.48.17__tar.gz

{prediction_market_agent_tooling-0.48.15 → prediction_market_agent_tooling-0.48.17}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: prediction-market-agent-tooling
-Version: 0.48.15
+Version: 0.48.17
 Summary: Tools to benchmark, deploy and monitor prediction market agents.
 Author: Gnosis
 Requires-Python: >=3.10,<3.12
@@ -26,6 +26,7 @@ Requires-Dist: langchain-community (>=0.0.19)
 Requires-Dist: langchain-openai (>=0.1.0,<0.2.0) ; extra == "langchain"
 Requires-Dist: langfuse (>=2.42.0,<3.0.0)
 Requires-Dist: loguru (>=0.7.2,<0.8.0)
+Requires-Dist: loky (>=3.4.1,<4.0.0)
 Requires-Dist: numpy (>=1.26.4,<2.0.0)
 Requires-Dist: openai (>=1.0.0,<2.0.0) ; extra == "openai"
 Requires-Dist: prompt-toolkit (>=3.0.43,<4.0.0)

{prediction_market_agent_tooling-0.48.15 → prediction_market_agent_tooling-0.48.17}/prediction_market_agent_tooling/loggers.py

@@ -49,6 +49,11 @@ def patch_logger() -> None:
     Function to patch loggers according to the deployed environment.
     Patches Loguru's logger, Python's default logger, warnings library and also monkey-patch print function as many libraries just use it.
     """
+    if not getattr(logger, "_patched", False):
+        logger._patched = True  # type: ignore[attr-defined] # Hacky way to store a flag on the logger object, to not patch it multiple times.
+    else:
+        return
+
     config = LogConfig()
 
     if config.LOG_FORMAT == LogFormat.GCP:
@@ -116,6 +121,4 @@ def simple_warning_format(message, category, filename, lineno, line=None):  # ty
     )  # Escape new lines, because otherwise logs will be broken.
 
 
-if not getattr(logger, "_patched", False):
-    patch_logger()
-    logger._patched = True  # type: ignore[attr-defined] # Hacky way to store a flag on the logger object, to not patch it multiple times.
+patch_logger()

{prediction_market_agent_tooling-0.48.15 → prediction_market_agent_tooling-0.48.17}/prediction_market_agent_tooling/tools/langfuse_client_utils.py

@@ -150,10 +150,5 @@ def get_trace_for_bet(
             add_utc_timezone_validator(bet.created_time),
             [t.timestamp for t in traces_for_bet],
         )
-        # Sanity check - the trace should be after the bet
-        if traces_for_bet[closest_trace_index].timestamp < add_utc_timezone_validator(
-            bet.created_time
-        ):
-            return None
 
         return traces_for_bet[closest_trace_index]

prediction_market_agent_tooling-0.48.17/prediction_market_agent_tooling/tools/parallelism.py

@@ -0,0 +1,33 @@
+from typing import Callable, Generator, TypeVar
+
+from loky import get_reusable_executor
+
+from prediction_market_agent_tooling.loggers import patch_logger
+
+A = TypeVar("A")
+B = TypeVar("B")
+
+
+def par_map(
+    items: list[A],
+    func: Callable[[A], B],
+    max_workers: int = 5,
+) -> "list[B]":
+    """Applies the function to each element using the specified executor. Awaits for all results."""
+    executor = get_reusable_executor(max_workers=max_workers, initializer=patch_logger)
+    futures = [executor.submit(func, item) for item in items]
+    results = []
+    for fut in futures:
+        results.append(fut.result())
+    return results
+
+
+def par_generator(
+    items: list[A],
+    func: Callable[[A], B],
+    max_workers: int = 5,
+) -> Generator[B, None, None]:
+    """Applies the function to each element using the specified executor. Yields results as they come."""
+    executor = get_reusable_executor(max_workers=max_workers, initializer=patch_logger)
+    for res in executor.map(func, items):
+        yield res

{prediction_market_agent_tooling-0.48.15 → prediction_market_agent_tooling-0.48.17}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "prediction-market-agent-tooling"
-version = "0.48.15"
+version = "0.48.17"
 description = "Tools to benchmark, deploy and monitor prediction market agents."
 authors = ["Gnosis"]
 readme = "README.md"
@@ -47,6 +47,7 @@ tavily-python = "^0.3.9"
 sqlmodel = "^0.0.21"
 psycopg2-binary = "^2.9.9"
 base58 = ">=1.0.2,<2.0"
+loky = "^3.4.1"
 
 [tool.poetry.extras]
 openai = ["openai"]

prediction_market_agent_tooling-0.48.15/prediction_market_agent_tooling/tools/parallelism.py

@@ -1,44 +0,0 @@
-import concurrent
-from concurrent.futures import Executor
-from concurrent.futures.process import ProcessPoolExecutor
-from concurrent.futures.thread import ThreadPoolExecutor
-from typing import Callable, Generator, TypeVar
-
-# Max workers to 5 to avoid rate limiting on some APIs, create a custom executor if you need more workers.
-DEFAULT_THREADPOOL_EXECUTOR = ThreadPoolExecutor(max_workers=5)
-DEFAULT_PROCESSPOOL_EXECUTOR = ProcessPoolExecutor(max_workers=5)
-
-A = TypeVar("A")
-B = TypeVar("B")
-
-
-def par_map(
-    items: list[A],
-    func: Callable[[A], B],
-    executor: Executor = DEFAULT_THREADPOOL_EXECUTOR,
-) -> "list[B]":
-    """Applies the function to each element using the specified executor. Awaits for all results.
-    If executor is ProcessPoolExecutor, make sure the function passed is pickable, e.g. no lambda functions
-    """
-    futures: list[concurrent.futures._base.Future[B]] = [
-        executor.submit(func, item) for item in items
-    ]
-    results = []
-    for fut in futures:
-        results.append(fut.result())
-    return results
-
-
-def par_generator(
-    items: list[A],
-    func: Callable[[A], B],
-    executor: Executor = DEFAULT_THREADPOOL_EXECUTOR,
-) -> Generator[B, None, None]:
-    """Applies the function to each element using the specified executor. Yields results as they come.
-    If executor is ProcessPoolExecutor, make sure the function passed is pickable, e.g. no lambda functions.
-    """
-    futures: list[concurrent.futures._base.Future[B]] = [
-        executor.submit(func, item) for item in items
-    ]
-    for fut in concurrent.futures.as_completed(futures):
-        yield fut.result()

{prediction_market_agent_tooling-0.48.15 → prediction_market_agent_tooling-0.48.17}/prediction_market_agent_tooling/jobs/jobs_models.py

@@ -36,8 +36,8 @@ class JobAgentMarket(AgentMarket, ABC):
     def get_reward(self, max_bond: float) -> float:
         """Reward for completing this job."""
 
-    @abstractmethod
     @classmethod
+    @abstractmethod
     def get_jobs(
         cls, limit: int | None, filter_by: FilterBy, sort_by: SortBy
     ) -> t.Sequence["JobAgentMarket"]: