Use PMAT package (#28)

* Use PMAT package * Update poetry * Don't use git in pyproject * fix openai key
gnosis · Feb 12, 2024 · ba112e6 · ba112e6
1 parent 424b1b4
commit ba112e6
Show file tree

Hide file tree

Showing 39 changed files with 778 additions and 3,270 deletions.
diff --git a/examples/cloud_deployment/agent.py b/examples/cloud_deployment/agent.py
diff --git a/examples/cloud_deployment/deploy.py b/examples/cloud_deployment/deploy.py
diff --git a/main.py b/main.py
@@ -1,14 +1,10 @@
 import typer
 import prediction_market_agent as pma
 from decimal import Decimal
-from prediction_market_agent.tools.utils import should_not_happen
 from prediction_market_agent.agents.all_agents import AgentType, get_agent
-from prediction_market_agent.tools.gtypes import xDai, Mana
-from prediction_market_agent.markets.all_markets import (
+from prediction_market_agent_tooling.markets.markets import (
     MarketType,
     get_bet_amount,
-    omen,
-    manifold,
     get_binary_markets,
     place_bet,
 )
@@ -22,7 +18,7 @@ def main(
     """
     Picks one market and answers it, optionally placing a bet.
     """
-    keys = pma.utils.get_keys()
+    keys = pma.utils.APIKeys()
 
     # Pick a market
     market = get_binary_markets(market_type)[0].to_agent_market()
@@ -54,7 +50,6 @@ def main(
             market=market.original_market,
             amount=get_bet_amount(amount, market_type),
             outcome=result,
-            keys=keys,
             omen_auto_deposit=True,
         )
 

diff --git a/mypy.ini b/mypy.ini
@@ -1,6 +1,6 @@
 [mypy]
 python_version = 3.9
-files = main.py, prediction_market_agent/, scripts/, tests/, examples/
+files = main.py, prediction_market_agent/, tests/ 
 plugins = pydantic.mypy
 warn_redundant_casts = True
 warn_unused_ignores = True

diff --git a/poetry.lock b/poetry.lock
diff --git a/prediction_market_agent/__init__.py b/prediction_market_agent/__init__.py
@@ -1,4 +1,2 @@
 from . import utils
 from . import agents
-from .markets import manifold
-from .markets import omen