Bump PMAT (#525)

gnosis · Oct 21, 2024 · 4d888e7 · 4d888e7
1 parent 9d772d6
commit 4d888e7
Show file tree

Hide file tree

Showing 6 changed files with 452 additions and 548 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/prediction_market_agent/agents/microchain_agent/market_functions.py b/prediction_market_agent/agents/microchain_agent/market_functions.py
@@ -16,9 +16,7 @@
 from prediction_market_agent_tooling.tools.betting_strategies.kelly_criterion import (
     get_kelly_bet_simplified,
 )
-from prediction_market_agent_tooling.tools.tavily_storage.tavily_models import (
-    TavilyStorage,
-)
+from prediction_market_agent_tooling.tools.tavily.tavily_storage import TavilyStorage
 from prediction_market_agent_tooling.tools.utils import utcnow
 
 from prediction_market_agent.agents.microchain_agent.utils import (

diff --git a/prediction_market_agent/agents/prophet_agent/deploy.py b/prediction_market_agent/agents/prophet_agent/deploy.py
@@ -6,9 +6,7 @@
 from prediction_market_agent_tooling.loggers import logger
 from prediction_market_agent_tooling.markets.agent_market import AgentMarket
 from prediction_market_agent_tooling.markets.data_models import ProbabilisticAnswer
-from prediction_market_agent_tooling.tools.tavily_storage.tavily_models import (
-    TavilyStorage,
-)
+from prediction_market_agent_tooling.tools.tavily.tavily_storage import TavilyStorage
 from prediction_prophet.benchmark.agents import (
     EmbeddingModel,
     OlasAgent,

diff --git a/prediction_market_agent/agents/think_thoroughly_agent/think_thoroughly_agent.py b/prediction_market_agent/agents/think_thoroughly_agent/think_thoroughly_agent.py
@@ -23,9 +23,7 @@
 )
 from prediction_market_agent_tooling.tools.langfuse_ import langfuse_context, observe
 from prediction_market_agent_tooling.tools.parallelism import par_generator, par_map
-from prediction_market_agent_tooling.tools.tavily_storage.tavily_models import (
-    TavilyStorage,
-)
+from prediction_market_agent_tooling.tools.tavily.tavily_storage import TavilyStorage
 from prediction_market_agent_tooling.tools.utils import (
     LLM_SUPER_LOW_TEMPERATURE,
     DatetimeUTC,

diff --git a/prediction_market_agent/tools/prediction_prophet/research.py b/prediction_market_agent/tools/prediction_prophet/research.py
@@ -1,7 +1,5 @@
 from prediction_market_agent_tooling.loggers import logger
-from prediction_market_agent_tooling.tools.tavily_storage.tavily_models import (
-    TavilyStorage,
-)
+from prediction_market_agent_tooling.tools.tavily.tavily_storage import TavilyStorage
 from prediction_prophet.benchmark.agents import (  # noqa: F401 # Just to make it available for the user of research.
     _make_prediction as prophet_make_prediction,
 )

diff --git a/pyproject.toml b/pyproject.toml
@@ -35,7 +35,7 @@ poetry = "^1.7.1"
 poetry-plugin-export = "^1.6.0"
 functions-framework = "^3.5.0"
 cron-validator = "^1.0.8"
-prediction-market-agent-tooling = { version = "^0.52.2", extras = ["langchain", "google"] }
+prediction-market-agent-tooling = { version = "^0.54.0", extras = ["langchain", "google"] }
 pydantic-settings = "^2.1.0"
 autoflake = "^2.2.1"
 isort = "^5.13.2"
@@ -52,7 +52,7 @@ tweepy = "^4.14.0"
 langchain-pinecone = "^0.1.1"
 pinatapy-vourhey = "^0.2.0"
 plotly = "^5.22.0"
-prediction-prophet = { git = "https://github.com/agentcoinorg/predictionprophet.git", rev = "63fe84e5c00a992d59f44fe240a3d99debfc703c" }
+prediction-prophet = { git = "https://github.com/agentcoinorg/predictionprophet.git", rev = "69b3ab619b25eab5890e58d5dc65be3aadf5bb65" }
 transformers = "^4.43.3"
 openfactverification-kongzii = "^0.2.0"