prediction-market-agent-tooling 0.48.15__py3-none-any.whl → 0.48.17__py3-none-any.whl

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"]:

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/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/tools/parallelism.py

@@ -1,12 +1,8 @@
-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)
+from loky import get_reusable_executor
+
+from prediction_market_agent_tooling.loggers import patch_logger
 
 A = TypeVar("A")
 B = TypeVar("B")
@@ -15,14 +11,11 @@ B = TypeVar("B")
 def par_map(
     items: list[A],
     func: Callable[[A], B],
-    executor: Executor = DEFAULT_THREADPOOL_EXECUTOR,
+    max_workers: int = 5,
 ) -> "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
-    ]
+    """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())
@@ -32,13 +25,9 @@ def par_map(
 def par_generator(
     items: list[A],
     func: Callable[[A], B],
-    executor: Executor = DEFAULT_THREADPOOL_EXECUTOR,
+    max_workers: int = 5,
 ) -> 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()
+    """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.dist-info → prediction_market_agent_tooling-0.48.17.dist-info}/METADATA

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

@@ -25,9 +25,9 @@ prediction_market_agent_tooling/deploy/gcp/utils.py,sha256=oyW0jgrUT2Tr49c7GlpcM
 prediction_market_agent_tooling/gtypes.py,sha256=O77co9-GWmHJo_NyBzRVkli5L1xqweI28JmsyaAHUHs,2474
 prediction_market_agent_tooling/jobs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 prediction_market_agent_tooling/jobs/jobs.py,sha256=I07yh0GJ-xhlvQaOUQB8xlSnihhcbU2c7DZ4ZND14c0,1246
-prediction_market_agent_tooling/jobs/jobs_models.py,sha256=wlBhh-D8pUegBCTpzTfla6crGlNMSqqNX7yDqMcRLYI,1326
+prediction_market_agent_tooling/jobs/jobs_models.py,sha256=8JS9n_EVgmNzqRg1YUjopPVZRjqFneVYCKnX4UEFy3I,1326
 prediction_market_agent_tooling/jobs/omen/omen_jobs.py,sha256=EAzo63vhSt1P7ddoDubdmvgzFf-ieQRTDCRLzYkH3js,3922
-prediction_market_agent_tooling/loggers.py,sha256=JiBTgvb34O9dKHYKZyQ0UzojPUy6KSFQSTfbBIXopSY,3721
+prediction_market_agent_tooling/loggers.py,sha256=Am6HHXRNO545BO3l7Ue9Wb2TkYE1OK8KKhGbI3XypVU,3751
 prediction_market_agent_tooling/markets/agent_market.py,sha256=nVJO1MY7_l-YP1Q7-mGLtRgIHK8M-yt3Ht1yyQPwImM,10176
 prediction_market_agent_tooling/markets/categorize.py,sha256=jsoHWvZk9pU6n17oWSCcCxNNYVwlb_NXsZxKRI7vmsk,1301
 prediction_market_agent_tooling/markets/data_models.py,sha256=vaJ049j2Anf5sZWjNflsCEag-Y0G1mTu6bCWT74HxqM,3266
@@ -75,8 +75,8 @@ prediction_market_agent_tooling/tools/image_gen/image_gen.py,sha256=HzRwBx62hOXB
 prediction_market_agent_tooling/tools/image_gen/market_thumbnail_gen.py,sha256=8A3U2uxsCsOfLjru-6R_PPIAuiKY4qFkWp_GSBPV6-s,1280
 prediction_market_agent_tooling/tools/is_predictable.py,sha256=QapzvJVgUZdhucgmxhzWAQ885BwSwvYUi0SG8mkLQMQ,6738
 prediction_market_agent_tooling/tools/langfuse_.py,sha256=jI_4ROxqo41CCnWGS1vN_AeDVhRzLMaQLxH3kxDu3L8,1153
-prediction_market_agent_tooling/tools/langfuse_client_utils.py,sha256=EljGXIal3xApSzi5oaNJl4pRm66nayBWEpXR68SuVy0,4951
-prediction_market_agent_tooling/tools/parallelism.py,sha256=Rz8QdVUWX8KCbr8UZfaC_b1GBWIb3bXwITUumuvBJ60,1633
+prediction_market_agent_tooling/tools/langfuse_client_utils.py,sha256=HZ8PN__CahMeNLaEx4v9esOqow6XEeezuYOOZD9-Ol8,4741
+prediction_market_agent_tooling/tools/parallelism.py,sha256=6Gou0hbjtMZrYvxjTDFUDZuxmE2nqZVbb6hkg1hF82A,1022
 prediction_market_agent_tooling/tools/safe.py,sha256=h0xOO0eNtitClf0fPkn-0oTc6A_bflDTee98V_aiV-A,5195
 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
@@ -84,8 +84,8 @@ prediction_market_agent_tooling/tools/tavily_storage/tavily_models.py,sha256=Uq2
 prediction_market_agent_tooling/tools/tavily_storage/tavily_storage.py,sha256=xrtQH9v5pXycBRyc5j45pWqkSffkoc9efNIU1_G633Q,3706
 prediction_market_agent_tooling/tools/utils.py,sha256=JE9YWtPPhnTgLiOyGAZDNG5K8nCwUY9IZEuAlm9UcxA,6611
 prediction_market_agent_tooling/tools/web3_utils.py,sha256=IZDxHhUJH5RsaRkK9DW6z1RYdk2cz5RqLMZG3T6Gv1U,11602
-prediction_market_agent_tooling-0.48.15.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
-prediction_market_agent_tooling-0.48.15.dist-info/METADATA,sha256=vp4AH8yoRuOMnzAizjbHTWMTVxKWnvRLNrJxLt-Utm0,7811
-prediction_market_agent_tooling-0.48.15.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-prediction_market_agent_tooling-0.48.15.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
-prediction_market_agent_tooling-0.48.15.dist-info/RECORD,,
+prediction_market_agent_tooling-0.48.17.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
+prediction_market_agent_tooling-0.48.17.dist-info/METADATA,sha256=sAyQzhlLTgPYGu4SF0z2dfSpSkZeJkZJYStbvagwh18,7848
+prediction_market_agent_tooling-0.48.17.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+prediction_market_agent_tooling-0.48.17.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
+prediction_market_agent_tooling-0.48.17.dist-info/RECORD,,