prediction-market-agent-tooling 0.56.0.dev112__tar.gz → 0.56.0.dev113__tar.gz

{prediction_market_agent_tooling-0.56.0.dev112 → prediction_market_agent_tooling-0.56.0.dev113}/PKG-INFO

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

{prediction_market_agent_tooling-0.56.0.dev112 → prediction_market_agent_tooling-0.56.0.dev113}/prediction_market_agent_tooling/tools/caches/db_cache.py

@@ -47,7 +47,7 @@ def db_cache(
     cache_none: bool = True,
     api_keys: APIKeys | None = None,
     ignore_args: Sequence[str] | None = None,
-    ignore_arg_types: Sequence[type] | None = (APIKeys,),
+    ignore_arg_types: Sequence[type] | None = None,
 ) -> Callable[[FunctionT], FunctionT]:
     ...
 
@@ -60,7 +60,7 @@ def db_cache(
     cache_none: bool = True,
     api_keys: APIKeys | None = None,
     ignore_args: Sequence[str] | None = None,
-    ignore_arg_types: Sequence[type] | None = (APIKeys,),
+    ignore_arg_types: Sequence[type] | None = None,
 ) -> FunctionT:
     ...
 
@@ -72,10 +72,7 @@ def db_cache(
     cache_none: bool = True,
     api_keys: APIKeys | None = None,
     ignore_args: Sequence[str] | None = None,
-    ignore_arg_types: Sequence[type]
-    | None = (
-        APIKeys,
-    ),  # Ignore `APIKeys` by default from storing it in DB and using to compute look-up hash, because api keys shouldn't change the computed results.
+    ignore_arg_types: Sequence[type] | None = None,
 ) -> FunctionT | Callable[[FunctionT], FunctionT]:
     if func is None:
         # Ugly Pythonic way to support this decorator as `@postgres_cache` but also `@postgres_cache(max_age=timedelta(days=3))`

{prediction_market_agent_tooling-0.56.0.dev112 → prediction_market_agent_tooling-0.56.0.dev113}/prediction_market_agent_tooling/tools/google.py

@@ -1,4 +1,5 @@
 import typing as t
+from datetime import timedelta
 
 import tenacity
 from googleapiclient.discovery import build
@@ -13,7 +14,7 @@ from prediction_market_agent_tooling.tools.caches.db_cache import db_cache
     stop=tenacity.stop_after_attempt(3),
     after=lambda x: logger.debug(f"search_google failed, {x.attempt_number=}."),
 )
-@db_cache
+@db_cache(max_age=timedelta(days=1))
 def search_google(
     query: str | None = None,
     num: int = 3,

{prediction_market_agent_tooling-0.56.0.dev112 → prediction_market_agent_tooling-0.56.0.dev113}/prediction_market_agent_tooling/tools/tavily/tavily_search.py

@@ -1,5 +1,5 @@
 import typing as t
-from datetime import date
+from datetime import date, timedelta
 
 import tenacity
 from tavily import TavilyClient
@@ -14,7 +14,7 @@ from prediction_market_agent_tooling.tools.tavily.tavily_models import (
 DEFAULT_SCORE_THRESHOLD = 0.75  # Based on some empirical testing, anything lower wasn't very relevant to the question being asked
 
 
-@db_cache
+@db_cache(max_age=timedelta(days=1), ignore_args=["api_keys"])
 def tavily_search(
     query: str,
     search_depth: t.Literal["basic", "advanced"] = "advanced",

{prediction_market_agent_tooling-0.56.0.dev112 → prediction_market_agent_tooling-0.56.0.dev113}/pyproject.toml

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