Bump PMAT (#117)

agentcoinorg · Oct 21, 2024 · af0e002 · af0e002
1 parent aaa621b
commit af0e002
Show file tree

Hide file tree

Showing 6 changed files with 1,655 additions and 1,433 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/prediction_prophet/benchmark/agents.py b/prediction_prophet/benchmark/agents.py
@@ -26,7 +26,7 @@
 from pydantic.types import SecretStr
 from prediction_prophet.autonolas.research import Prediction as LLMCompletionPredictionDict
 from prediction_market_agent_tooling.tools.langfuse_ import observe
-from prediction_market_agent_tooling.tools.tavily_storage.tavily_models import TavilyStorage
+from prediction_market_agent_tooling.tools.tavily.tavily_storage import TavilyStorage
 
 if t.TYPE_CHECKING:
     from loguru import Logger

diff --git a/prediction_prophet/functions/research.py b/prediction_prophet/functions/research.py
@@ -12,7 +12,7 @@
 from pydantic.types import SecretStr
 from pydantic import BaseModel
 from prediction_market_agent_tooling.tools.langfuse_ import observe
-from prediction_market_agent_tooling.tools.tavily_storage.tavily_models import TavilyStorage
+from prediction_market_agent_tooling.tools.tavily.tavily_storage import TavilyStorage
 
 if t.TYPE_CHECKING:
     from loguru import Logger

diff --git a/prediction_prophet/functions/search.py b/prediction_prophet/functions/search.py
@@ -3,7 +3,7 @@
 from prediction_prophet.functions.web_search import WebSearchResult, web_search
 from concurrent.futures import ThreadPoolExecutor, as_completed
 from pydantic.types import SecretStr
-from prediction_market_agent_tooling.tools.tavily_storage.tavily_models import TavilyStorage
+from prediction_market_agent_tooling.tools.tavily.tavily_storage import TavilyStorage
 
 if t.TYPE_CHECKING:
     from loguru import Logger

diff --git a/prediction_prophet/functions/web_search.py b/prediction_prophet/functions/web_search.py
@@ -2,8 +2,8 @@
 
 from prediction_prophet.models.WebSearchResult import WebSearchResult
 from prediction_market_agent_tooling.config import APIKeys
-from prediction_market_agent_tooling.tools.tavily_storage.tavily_models import TavilyStorage
-from prediction_market_agent_tooling.tools.tavily_storage.tavily_storage import tavily_search
+from prediction_market_agent_tooling.tools.tavily.tavily_storage import TavilyStorage
+from prediction_market_agent_tooling.tools.tavily.tavily_search import tavily_search
 
 
 

diff --git a/pyproject.toml b/pyproject.toml
@@ -31,7 +31,7 @@ scikit-learn = "^1.4.0"
 typer = ">=0.9.0,<1.0.0"
 types-requests = "^2.31.0.20240125"
 types-python-dateutil = "^2.9.0"
-prediction-market-agent-tooling = { version = ">=0.48.3,<1", extras = ["langchain", "google"] }
+prediction-market-agent-tooling = { version = ">=0.54.0,<1", extras = ["langchain", "google"] }
 langchain-community = "^0.2.6"
 memory-profiler = "^0.61.0"
 matplotlib = "^3.8.3"