Bump PMAT, prophet, and remove dead code (#550)

gnosis · Nov 21, 2024 · a785d76 · a785d76
1 parent c9ddb44
commit a785d76
Show file tree

Hide file tree

Showing 10 changed files with 10 additions and 216 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/prediction_market_agent/agents/social_media_agent/deploy.py b/prediction_market_agent/agents/social_media_agent/deploy.py
@@ -88,4 +88,4 @@ def post(self, tweet: str | None, reasoning_reply_tweet: str | None) -> None:
 
 if __name__ == "__main__":
     agent = DeployableSocialMediaAgent()
-    agent.deploy_local(market_type=MarketType.OMEN, sleep_time=540, timeout=180)
+    agent.deploy_local(market_type=MarketType.OMEN, sleep_time=540, run_time=180)
diff --git a/prediction_market_agent/agents/think_thoroughly_agent/deploy.py b/prediction_market_agent/agents/think_thoroughly_agent/deploy.py
@@ -59,10 +59,10 @@ def get_betting_strategy(self, market: AgentMarket) -> BettingStrategy:
 
 if __name__ == "__main__":
     agent = DeployableThinkThoroughlyAgent(
-        place_trades=False, store_prediction=False, store_trades=False
+        place_trades=False, store_predictions=False, store_trades=False
     )
     agent.deploy_local(
         market_type=MarketType.OMEN,
         sleep_time=540,
-        timeout=180,
+        run_time=180,
     )
diff --git a/prediction_market_agent/ai_models/abstract_ai_models.py b/prediction_market_agent/ai_models/abstract_ai_models.py
diff --git a/prediction_market_agent/ai_models/llama_ai_models.py b/prediction_market_agent/ai_models/llama_ai_models.py
diff --git a/prediction_market_agent/ai_models/openai_ai_models.py b/prediction_market_agent/ai_models/openai_ai_models.py
diff --git a/prediction_market_agent/tools/web_scrape/markdown.py b/prediction_market_agent/tools/web_scrape/markdown.py
@@ -22,7 +22,7 @@ def fetch_html(url: str, timeout: int) -> Response:
 
 
 @observe()
-@db_cache(max_age=timedelta(days=1))
+@db_cache(max_age=timedelta(days=1), ignore_args=["timeout"])
 def web_scrape(url: str, timeout: int = 10) -> str:
     """
     Taken from agentcoinorg/predictionprophet

diff --git a/pyproject.toml b/pyproject.toml
@@ -34,7 +34,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.56.3", extras = ["langchain", "google"] }
+prediction-market-agent-tooling = { version = "^0.57.0", extras = ["langchain", "google"] }
 pydantic-settings = "^2.1.0"
 autoflake = "^2.2.1"
 isort = "^5.13.2"

diff --git a/scripts/agent_app.py b/scripts/agent_app.py
@@ -79,7 +79,7 @@ def predict(
 ) -> None:
     agent = AgentClass(
         place_trades=False,
-        store_prediction=False,
+        store_predictions=False,
         store_trades=False,
         enable_langfuse=enable_langfuse,
     )

diff --git a/tests/ai_models/test_llama_ai_models.py b/tests/ai_models/test_llama_ai_models.py