prediction-market-agent-tooling 0.30.0__py3-none-any.whl → 0.32.0__py3-none-any.whl

prediction_market_agent_tooling/deploy/agent.py

@@ -273,11 +273,11 @@ class DeployableTraderAgent(DeployableAgent):
         for market in markets:
             result = self.answer_binary_market(market)
             if result is None:
-                logger.debug(f"Skipping market {market} as no answer was provided")
+                logger.info(f"Skipping market {market} as no answer was provided")
                 continue
             if self.place_bet:
                 amount = self.calculate_bet_amount(result, market)
-                logger.debug(
+                logger.info(
                     f"Placing bet on {market} with result {result} and amount {amount}"
                 )
                 market.place_bet(

prediction_market_agent_tooling/loggers.py

@@ -14,12 +14,21 @@ class LogFormat(str, Enum):
     GCP = "gcp"
 
 
+class LogLevel(str, Enum):
+    CRITICAL = "CRITICAL"
+    ERROR = "ERROR"
+    WARNING = "WARNING"
+    INFO = "INFO"
+    DEBUG = "DEBUG"
+
+
 class LogConfig(BaseSettings):
     model_config = SettingsConfigDict(
         env_file=".env", env_file_encoding="utf-8", extra="ignore"
     )
 
     LOG_FORMAT: LogFormat = LogFormat.DEFAULT
+    LOG_LEVEL: LogLevel = LogLevel.DEBUG
 
 
 GCP_LOG_LOGURU_FORMAT = (
@@ -49,7 +58,7 @@ def patch_logger() -> None:
     # Change built-in logging.
     if format_logging is not None:
         logging.basicConfig(
-            level=logging.DEBUG, format=format_logging, datefmt=datefmt_logging
+            level=config.LOG_LEVEL.value, format=format_logging, datefmt=datefmt_logging
         )
 
     # Change loguru.
@@ -58,7 +67,7 @@ def patch_logger() -> None:
         logger.add(
             sys.stdout,
             format=format_loguru,
-            level="DEBUG",  # Can be the lowest level, higher ones will use by default this one.
+            level=config.LOG_LEVEL.value,
             colorize=True,
         )
 

prediction_market_agent_tooling/markets/omen/omen.py

@@ -156,7 +156,7 @@ class OmenAgentMarket(AgentMarket):
         )
 
     def sell_tokens(
-        self, outcome: bool, amount: TokenAmount, auto_withdraw: bool = True
+        self, outcome: bool, amount: TokenAmount, auto_withdraw: bool = False
     ) -> None:
         if not self.can_be_traded():
             raise ValueError(

prediction_market_agent_tooling/markets/omen/omen_subgraph_handler.py

@@ -205,8 +205,9 @@ class OmenSubgraphHandler(metaclass=SingletonMeta):
                 finalized_before
             )
 
+        # `excluded_question_titles` can not be an empty list, otherwise the API bugs out and returns nothing.
         excluded_question_titles = [""]
-        if excluded_questions is not None:
+        if excluded_questions:
             excluded_question_titles = [i for i in excluded_questions]
 
         where_stms["question_"]["title_not_in"] = excluded_question_titles
@@ -439,6 +440,8 @@ class OmenSubgraphHandler(metaclass=SingletonMeta):
         market_id: t.Optional[ChecksumAddress] = None,
         filter_by_answer_finalized_not_null: bool = False,
         type_: t.Literal["Buy", "Sell"] | None = None,
+        market_opening_after: datetime | None = None,
+        collateral_amount_more_than: Wei | None = None,
     ) -> list[OmenBet]:
         if not end_time:
             end_time = utcnow()
@@ -457,6 +460,12 @@ class OmenSubgraphHandler(metaclass=SingletonMeta):
             where_stms.append(trade.fpmm == market_id.lower())
         if filter_by_answer_finalized_not_null:
             where_stms.append(trade.fpmm.answerFinalizedTimestamp != None)
+        if market_opening_after is not None:
+            where_stms.append(
+                trade.fpmm.openingTimestamp > to_int_timestamp(market_opening_after)
+            )
+        if collateral_amount_more_than is not None:
+            where_stms.append(trade.collateralAmount > collateral_amount_more_than)
 
         trades = self.trades_subgraph.Query.fpmmTrades(
             first=sys.maxsize, where=where_stms
@@ -473,6 +482,8 @@ class OmenSubgraphHandler(metaclass=SingletonMeta):
         end_time: t.Optional[datetime] = None,
         market_id: t.Optional[ChecksumAddress] = None,
         filter_by_answer_finalized_not_null: bool = False,
+        market_opening_after: datetime | None = None,
+        collateral_amount_more_than: Wei | None = None,
     ) -> list[OmenBet]:
         return self.get_trades(
             better_address=better_address,
@@ -481,6 +492,8 @@ class OmenSubgraphHandler(metaclass=SingletonMeta):
             market_id=market_id,
             filter_by_answer_finalized_not_null=filter_by_answer_finalized_not_null,
             type_="Buy",  # We consider `bet` to be only the `Buy` trade types.
+            market_opening_after=market_opening_after,
+            collateral_amount_more_than=collateral_amount_more_than,
         )
 
     def get_resolved_bets(

{prediction_market_agent_tooling-0.30.0.dist-info → prediction_market_agent_tooling-0.32.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: prediction-market-agent-tooling
-Version: 0.30.0
+Version: 0.32.0
 Summary: Tools to benchmark, deploy and monitor prediction market agents.
 Author: Gnosis
 Requires-Python: >=3.10,<3.12
@@ -61,7 +61,6 @@ Deploying and monitoring agents using GCP requires that you set up the gcloud CL
 
 ```bash
 MANIFOLD_API_KEY=...
-BET_FROM_ADDRESS=...
 BET_FROM_PRIVATE_KEY=...
 OPENAI_API_KEY=...
 ```
@@ -76,46 +75,54 @@ For example:
 
 ```python
 import prediction_market_agent_tooling.benchmark.benchmark as bm
+from prediction_market_agent_tooling.benchmark.agents import RandomAgent
 from prediction_market_agent_tooling.markets.markets import MarketType, get_binary_markets
 
 benchmarker = bm.Benchmarker(
     markets=get_binary_markets(limit=10, market_type=MarketType.MANIFOLD),
-    agents=[...],
+    agents=[RandomAgent(agent_name="a_random_agent")],
 )
 benchmarker.run_agents()
 md = benchmarker.generate_markdown_report()
 ```
 
-This produces a markdown report comparing agents:
+This produces a markdown report that you can use for comparing agents side-by-side, like:
 
 ![Benchmark results](assets/comparison-report.png)
 
 ## Deploying
 
-Create a deployable agent by subclassing the `DeployableAgent` base class, and implementing the 
+> **Deprecated**: We suggest using your own infrastructure to deploy, but you may still find this useful.
+
+Create a deployable agent by subclassing the `DeployableTraderAgent` base class, and implementing the `answer_binary_market` method.
 
 For example, deploy an agent that randomly picks an outcome:
 
 ```python
 import random
-from prediction_market_agent_tooling.deploy.agent import DeployableAgent
+from prediction_market_agent_tooling.deploy.agent import DeployableTraderAgent
 from prediction_market_agent_tooling.markets.agent_market import AgentMarket
 
-class DeployableCoinFlipAgent(DeployableAgent):
+class DeployableCoinFlipAgent(DeployableTraderAgent):
     def answer_binary_market(self, market: AgentMarket) -> bool | None:
         return random.choice([True, False])
 
 DeployableCoinFlipAgent().deploy_gcp(...)
 ```
 
-For deploying a Safe manually for a given agent, run the script below:
+### Safe
+
+Agents can control funds via a wallet primary key only, or optionally via a [Safe](https://safe.global/) as well. For deploying a Safe manually for a given agent, run the script below:
 
 ```commandline
 poetry run python scripts/create_safe_for_agent.py  --from-private-key <YOUR_AGENT_PRIVATE_KEY> --salt-nonce 42
 ```
+
 This will output the newly created Safe in the terminal, and it can then be copied over to the deployment part (e.g. Terraform).
 Note that `salt_nonce` can be passed so that the created safe is deterministically created for each agent, so that, if the same `salt_nonce` is used, the script will not create a new Safe for the agent, instead it will output the previously existent Safe.
 
+You can then specify this agent's Safe address with the `SAFE_ADDRESS` environment variable.
+
 ## Monitoring
 
 Monitor the performance of the agents deployed to GCP, as well as meta-metrics of the prediction market platforms they are deployed to.
@@ -132,15 +139,42 @@ Which launches in the browser:
 
 ## The Market Platforms
 
-The following markets platforms are supported:
+The following prediction market platforms are supported:
+
+| Platform                              | Benchmarking | Deployment | Monitoring |
+|---------------------------------------|--------------|------------|------------|
+| [Manifold](https://manifold.markets/) | ✅ | ✅ | ✅ |
+| [AIOmen](https://aiomen.eth.limo/)    | ✅ | ✅ | ✅ |
+| [Polymarket](https://polymarket.com/) | ✅ | ❌ | ❌ |
+
+## Prediction Markets Python API
+
+We have built clean abstractions for taking actions on the different prediction market platforms (retrieving markets, buying and selling tokens, etc.). This is currently undocumented, but for now, inspecting the [`AgentMarket`](https://github.com/gnosis/prediction-market-agent-tooling/blob/1e497fff9f2b53e4e3e1beb5dda08b4d49da881b/prediction_market_agent_tooling/markets/agent_market.py) class and its methods is your best bet.
+
+For example:
+
+```python
+from prediction_market_agent_tooling.config import APIKeys
+from prediction_market_agent_tooling.markets.agent_market import SortBy
+from prediction_market_agent_tooling.markets.omen.omen import OmenAgentMarket
+
+# Place a bet on the market closing soonest
+market = OmenAgentMarket.get_binary_markets(limit=1, sort_by=SortBy.CLOSING_SOONEST)[0]
+market.place_bet(outcome=True, amount=market.get_bet_amount(0.1))
+
+# View your positions
+my_positions = OmenAgentMarket.get_positions(user_id=APIKeys().bet_from_address)
+print(my_positions)
+
+# Sell position (accounting for fees)
+market.sell_tokens(outcome=True, amount=market.get_bet_amount(0.095))
+```
 
-- [Manifold](https://manifold.markets/)
-- [AIOmen](https://aiomen.eth.limo/)
-- [Polymarket](https://polymarket.com/) - Benchmarking only. Deploy and monitor TODO
+This API can be built on top of to create your application. See [here](https://github.com/gnosis/prediction-market-agent/tree/main/prediction_market_agent/agents/microchain_agent) for an example.
 
 ## Contributing
 
-See the [Issues](https://github.com/gnosis/prediction-market-agent-tooling/issues) for ideas of things that need fixing or implementing. Or come up with your own :D.
+See the [Issues](https://github.com/gnosis/prediction-market-agent-tooling/issues) for ideas of things that need fixing or implementing. The team is also receptive to new issues and PRs.
 
 We use `mypy` for static type checking, and `isort`, `black` and `autoflake` for linting. These all run as steps in CI.
 

{prediction_market_agent_tooling-0.30.0.dist-info → prediction_market_agent_tooling-0.32.0.dist-info}/RECORD

@@ -12,14 +12,14 @@ prediction_market_agent_tooling/benchmark/agents.py,sha256=HPIFJvackW110ch3Ukktb
 prediction_market_agent_tooling/benchmark/benchmark.py,sha256=xiHKzZx5GHSsDerFHMZ9j_LXAXnSaITSvv67iPe3MEU,21095
 prediction_market_agent_tooling/benchmark/utils.py,sha256=iS1BzyXcCMfMm1Rx--1QCH0pHvBTblTndcDQFbTUJ2s,2897
 prediction_market_agent_tooling/config.py,sha256=yELIlzAm2yBwZzGRvHtHBZZV3NZy5CllJfo3chMQMo0,4297
-prediction_market_agent_tooling/deploy/agent.py,sha256=P4MaueaXpNakAWXhB0EYsWLNRXSeRisaN55IVDb8AR8,10187
+prediction_market_agent_tooling/deploy/agent.py,sha256=h33uU97x3DjBy8INONtS1Hc5B-NnNDV6MENeeCp0qoA,10185
 prediction_market_agent_tooling/deploy/agent_example.py,sha256=tqXVA2HSFK3pdZ49tMmta8aKdJFT8JN8WeJ1akjrpBk,909
 prediction_market_agent_tooling/deploy/constants.py,sha256=M5ty8URipYMGe_G-RzxRydK3AFL6CyvmqCraJUrLBnE,82
 prediction_market_agent_tooling/deploy/gcp/deploy.py,sha256=CYUgnfy-9XVk04kkxA_5yp0GE9Mw5caYqlFUZQ2j3ks,3739
 prediction_market_agent_tooling/deploy/gcp/kubernetes_models.py,sha256=qYIHRxQLac3yxtZ8ChikiPG9O1aUQucHW0muTSm1nto,2627
 prediction_market_agent_tooling/deploy/gcp/utils.py,sha256=oyW0jgrUT2Tr49c7GlpcMsYNQjoCSOcWis3q-MmVAhU,6089
 prediction_market_agent_tooling/gtypes.py,sha256=xGSJXw12hzp8LwvQ956l01GiZMWd07MZTYqo8CXVeLY,2417
-prediction_market_agent_tooling/loggers.py,sha256=p_ibaEpUGGdcpPtzIJxh5UusJNCQyN1y_RDvoTaeUfA,3001
+prediction_market_agent_tooling/loggers.py,sha256=ua9rynYmsbOJZjxPIFxRBooomeN08zuLSJ7lxZMDS7w,3133
 prediction_market_agent_tooling/markets/agent_market.py,sha256=SMvkXct_RgHXqF-fVq3ooTIQ_99MG77kVlvS3rM8Ozo,7019
 prediction_market_agent_tooling/markets/categorize.py,sha256=yTd-lDMPW4ESDSzmxeLLBuzLX0FggOF7Vbswh7295o0,941
 prediction_market_agent_tooling/markets/data_models.py,sha256=uODY3aoFp8YYeLAUcrzMk1yU8pIKsTLobB9xIEGTmKs,1170
@@ -31,10 +31,10 @@ prediction_market_agent_tooling/markets/manifold/utils.py,sha256=cPPFWXm3vCYH1jy
 prediction_market_agent_tooling/markets/markets.py,sha256=w05Oo7yCA2axpCw69Q9y4i9Gcdpht0u5bZGbWqld3rU,2964
 prediction_market_agent_tooling/markets/omen/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 prediction_market_agent_tooling/markets/omen/data_models.py,sha256=EXtjmcujx68Xu50BVkYXvLuf_Asx5o65RvFS3ZS6HGs,14405
-prediction_market_agent_tooling/markets/omen/omen.py,sha256=wQIdECDREiRnUROlp1BF1W25so_Gs-9YXKjjj0kksAs,32041
+prediction_market_agent_tooling/markets/omen/omen.py,sha256=9le-tvs-zclHtzpvK7bMVqg2lZV-VZh1c5OB_uAdSiY,32042
 prediction_market_agent_tooling/markets/omen/omen_contracts.py,sha256=eDS8vN4Klv_-Y1wwfIeLDt3twhl9U_AJjPQov0JScb0,19888
 prediction_market_agent_tooling/markets/omen/omen_resolving.py,sha256=g77QsQ5WnSI2rzBlX87L_EhWMwobkyXyfRhHQmpAdzo,9012
-prediction_market_agent_tooling/markets/omen/omen_subgraph_handler.py,sha256=QZWwkqvOqQ-b15jidwTNsn1K64x3FY_Un-l6A6g3Twg,22299
+prediction_market_agent_tooling/markets/omen/omen_subgraph_handler.py,sha256=RlaQMDF75lo2fos8LUb_OAdrw_ILLOXQLdejzE7tOmA,23053
 prediction_market_agent_tooling/markets/polymarket/api.py,sha256=HXmA1akA0qDj0m3e-GEvWG8x75pm6BX4H7YJPQcST7I,4767
 prediction_market_agent_tooling/markets/polymarket/data_models.py,sha256=9CJzakyEcsn6DQBK2nOXjOMzTZBLAmK_KqevXvW17DI,4292
 prediction_market_agent_tooling/markets/polymarket/data_models_web.py,sha256=f8SRQy0Rn-gIHSEMrJJAI8H3J7l8lzOLj3aCMe0vJv8,11324
@@ -65,8 +65,8 @@ prediction_market_agent_tooling/tools/safe.py,sha256=h0xOO0eNtitClf0fPkn-0oTc6A_
 prediction_market_agent_tooling/tools/singleton.py,sha256=CiIELUiI-OeS7U7eeHEt0rnVhtQGzwoUdAgn_7u_GBM,729
 prediction_market_agent_tooling/tools/utils.py,sha256=zkmwPi3YisgZDPCeNwaRbL8sInOYOkvFgFY4Q8PbEo4,5077
 prediction_market_agent_tooling/tools/web3_utils.py,sha256=cboATXNmEdn5RmPbVblHOwOdUMKBYrUK3GiI6i6Vzxo,9855
-prediction_market_agent_tooling-0.30.0.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
-prediction_market_agent_tooling-0.30.0.dist-info/METADATA,sha256=1mx60T6NftBRU_qnCNnRU0emgGbTOcQIfex4S_UM1MI,5465
-prediction_market_agent_tooling-0.30.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-prediction_market_agent_tooling-0.30.0.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
-prediction_market_agent_tooling-0.30.0.dist-info/RECORD,,
+prediction_market_agent_tooling-0.32.0.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
+prediction_market_agent_tooling-0.32.0.dist-info/METADATA,sha256=Xs3kCp-N5XHtsM_LuoVRJTg0orQMWgWLJT0U8DEah74,7514
+prediction_market_agent_tooling-0.32.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+prediction_market_agent_tooling-0.32.0.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
+prediction_market_agent_tooling-0.32.0.dist-info/RECORD,,