prediction-market-agent-tooling 0.13.0__py3-none-any.whl → 0.13.2__py3-none-any.whl

prediction_market_agent_tooling/benchmark/utils.py

@@ -55,7 +55,7 @@ class PredictionsCache(BaseModel):
 
     def save(self, path: str) -> None:
         with open(path, "w") as f:
-            json.dump(self.dict(), f, indent=2)
+            json.dump(self.model_dump(), f, indent=2)
 
     @staticmethod
     def load(path: str) -> "PredictionsCache":

prediction_market_agent_tooling/markets/agent_market.py

@@ -45,7 +45,7 @@ class AgentMarket(BaseModel):
     outcomes: list[str]
     resolution: Resolution | None
     created_time: datetime | None
-    close_time: datetime
+    close_time: datetime | None
     p_yes: Probability
     url: str
     volume: Decimal | None  # Should be in currency of `currency` above.

prediction_market_agent_tooling/markets/categorize.py

@@ -3,7 +3,6 @@ from langchain.schema.output_parser import StrOutputParser
 from langchain_openai import ChatOpenAI
 
 from prediction_market_agent_tooling.config import APIKeys
-from prediction_market_agent_tooling.gtypes import secretstr_to_v1_secretstr
 
 
 def infer_category(
@@ -22,7 +21,7 @@ Write only the category itself, nothing else.
         prompt
         | ChatOpenAI(
             model=model,
-            api_key=secretstr_to_v1_secretstr(APIKeys().openai_api_key),
+            api_key=APIKeys().openai_api_key.get_secret_value(),
         )
         | StrOutputParser()
     )

prediction_market_agent_tooling/markets/omen/omen.py

@@ -75,6 +75,8 @@ class OmenAgentMarket(AgentMarket):
     market_maker_contract_address_checksummed: ChecksumAddress
     condition: Condition
     finalized_time: datetime | None
+    created_time: datetime
+    close_time: datetime
 
     INVALID_MARKET_ANSWER: HexStr = HexStr(
         "0xFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF"
@@ -310,14 +312,18 @@ def binary_omen_buy_outcome_tx(
 
 
 def omen_sell_outcome_tx(
-    amount: xDai,
+    amount: xDai,  # The xDai value of shares to sell.
     from_private_key: PrivateKey,
     market: OmenAgentMarket,
     outcome: str,
     auto_withdraw: bool,
 ) -> None:
     """
-    Sells the given amount of shares for the given outcome in the given market.
+    Sells the given xDai value of shares corresponding to the given outcome in
+    the given market.
+
+    The number of shares sold will depend on the share price at the time of the
+    transaction.
     """
     amount_wei = xdai_to_wei(amount)
 

prediction_market_agent_tooling/tools/is_predictable.py

@@ -42,7 +42,7 @@ def is_predictable_binary(
 
     prompt = ChatPromptTemplate.from_template(template=prompt_template)
     messages = prompt.format_messages(question=question)
-    completion = llm(messages, max_tokens=512).content
+    completion = str(llm(messages, max_tokens=512).content)
 
     try:
         decision = completion.lower().rsplit("decision", 1)[1]

{prediction_market_agent_tooling-0.13.0.dist-info → prediction_market_agent_tooling-0.13.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: prediction-market-agent-tooling
-Version: 0.13.0
+Version: 0.13.2
 Summary: Tools to benchmark, deploy and monitor prediction market agents.
 Author: Gnosis
 Requires-Python: >=3.10,<3.12
@@ -11,7 +11,7 @@ Provides-Extra: google
 Provides-Extra: langchain
 Requires-Dist: autoflake (>=2.2.1,<3.0.0)
 Requires-Dist: cron-validator (>=1.0.8,<2.0.0)
-Requires-Dist: eth-typing (>=4.0.0,<5.0.0)
+Requires-Dist: eth-typing (>=3.0.0,<4.0.0)
 Requires-Dist: functions-framework (>=3.5.0,<4.0.0)
 Requires-Dist: google-api-python-client (==2.95.0) ; extra == "google"
 Requires-Dist: google-cloud-functions (>=1.16.0,<2.0.0)
@@ -20,8 +20,9 @@ Requires-Dist: google-cloud-secret-manager (>=2.18.2,<3.0.0)
 Requires-Dist: isort (>=5.13.2,<6.0.0)
 Requires-Dist: langchain (>=0.1.9,<0.2.0) ; extra == "langchain"
 Requires-Dist: langchain-community (>=0.0.19)
-Requires-Dist: langchain-openai (>=0.0.8,<0.0.9) ; extra == "langchain"
+Requires-Dist: langchain-openai (>=0.0.5,<0.0.6) ; extra == "langchain"
 Requires-Dist: loguru (>=0.7.2,<0.8.0)
+Requires-Dist: mech-client (>=0.2.13,<0.3.0)
 Requires-Dist: numpy (>=1.26.4,<2.0.0)
 Requires-Dist: pydantic (>=2.6.1,<3.0.0)
 Requires-Dist: pydantic-settings (>=2.1.0,<3.0.0)
@@ -32,7 +33,7 @@ Requires-Dist: tabulate (>=0.9.0,<0.10.0)
 Requires-Dist: tqdm (>=4.66.2,<5.0.0)
 Requires-Dist: typer (>=0.9.0,<0.10.0)
 Requires-Dist: types-pytz (>=2024.1.0.20240203,<2025.0.0.0)
-Requires-Dist: types-requests (>=2.31.0.20240106,<3.0.0.0)
+Requires-Dist: types-requests (>=2.31.0.0,<3.0.0.0)
 Requires-Dist: web3 (>=6.15.1,<7.0.0)
 Description-Content-Type: text/markdown
 

{prediction_market_agent_tooling-0.13.0.dist-info → prediction_market_agent_tooling-0.13.2.dist-info}/RECORD

@@ -9,7 +9,7 @@ prediction_market_agent_tooling/abis/wxdai.abi.json,sha256=m3qC06Yug-pToI0lSFe1f
 prediction_market_agent_tooling/benchmark/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 prediction_market_agent_tooling/benchmark/agents.py,sha256=YZidRshIxhbj7NLLKcny5ezMhv-JBvIU0uwmfaD93AU,3737
 prediction_market_agent_tooling/benchmark/benchmark.py,sha256=0lcVX65IGyrGo0WHtOwBh5NFdSEY0M2wS8CB55CcYos,21237
-prediction_market_agent_tooling/benchmark/utils.py,sha256=sKXSRZspMZCBoVV4kRa6Wocwmlu6OViksDG-VjFZW6Q,2748
+prediction_market_agent_tooling/benchmark/utils.py,sha256=tV9eN71aFQnNU7jXsBDSbuz6VoYyvHqJL1ezyXViyHM,2754
 prediction_market_agent_tooling/config.py,sha256=eYKsOVebMYVb1JGqyIv70YEf68Xr9qIy--QzhHlXMEM,2596
 prediction_market_agent_tooling/deploy/agent.py,sha256=_67jmi2Sx0ksA-PDoOn0EGmnYRlMkrggvfo8az-hV94,7789
 prediction_market_agent_tooling/deploy/agent_example.py,sha256=vwswPCOutCG9vIqJT5lBcI1FZy3QRkNyVbNIJdn6wvk,652
@@ -17,8 +17,8 @@ prediction_market_agent_tooling/deploy/constants.py,sha256=M5ty8URipYMGe_G-RzxRy
 prediction_market_agent_tooling/deploy/gcp/deploy.py,sha256=CYUgnfy-9XVk04kkxA_5yp0GE9Mw5caYqlFUZQ2j3ks,3739
 prediction_market_agent_tooling/deploy/gcp/utils.py,sha256=x_SMhxYgNuRt8jmhlJkJAlpfX31zM6x_mvQ3Arr_3-s,5255
 prediction_market_agent_tooling/gtypes.py,sha256=ZDjXaszuTnCjnCThfJNeNJaLoxmGDSBQEcayyUfazwQ,2526
-prediction_market_agent_tooling/markets/agent_market.py,sha256=3Ms3N7OO2eHuSKdHZ2WZXINBB0WeolxOliZ7b-adR0A,5083
-prediction_market_agent_tooling/markets/categorize.py,sha256=LzmBkpLQech19yCxmsyKSJ-6trzAo6lXUmOuqgVZbco,1026
+prediction_market_agent_tooling/markets/agent_market.py,sha256=sL8ASQ1bDvvIjy1ZBUMcqA6Qm44VF0GMKcUCX_wxIyI,5090
+prediction_market_agent_tooling/markets/categorize.py,sha256=yTd-lDMPW4ESDSzmxeLLBuzLX0FggOF7Vbswh7295o0,941
 prediction_market_agent_tooling/markets/data_models.py,sha256=gijy3i1xSvDtM-BK0XbqZCSNqEjpysD1KJis7BngX5k,793
 prediction_market_agent_tooling/markets/manifold/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 prediction_market_agent_tooling/markets/manifold/api.py,sha256=dv-uUqt_yRLrQAymGEaTXX6C37NePo8ZPwK1l4CwLWk,6882
@@ -28,7 +28,7 @@ prediction_market_agent_tooling/markets/manifold/utils.py,sha256=cPPFWXm3vCYH1jy
 prediction_market_agent_tooling/markets/markets.py,sha256=pIYIlTAvXGrGh1XxNlHJQPEGi36_DHQ-WN2m0TCjEn0,1575
 prediction_market_agent_tooling/markets/omen/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 prediction_market_agent_tooling/markets/omen/data_models.py,sha256=LrpuO1e8D6EzftPLd15bmEQ9U8Q2IK2PLVfACFq6S_I,12214
-prediction_market_agent_tooling/markets/omen/omen.py,sha256=4hPNfUTks6CkTfKgpg9JblQolBxhhaabWPGq_czWeQw,26041
+prediction_market_agent_tooling/markets/omen/omen.py,sha256=ME7aWvaSEK-sUcKJ_8wQZNs2QV6fx3871FX5ow-R1Yw,26249
 prediction_market_agent_tooling/markets/omen/omen_contracts.py,sha256=OFKRpt59iVGvBCUScj0iGJJuqQwkjRefxhM58tOy57E,19354
 prediction_market_agent_tooling/markets/omen/omen_replicate.py,sha256=VOvatdK_9OMrYhF1f_7DL8c-kcUDuzUWa5VbKpR6LfQ,7427
 prediction_market_agent_tooling/markets/omen/omen_resolve_replicated.py,sha256=5Aqpl99TIXPbV7oL7qyrxPZd2hdl7rz1yhqSGKioUB4,11187
@@ -54,13 +54,13 @@ prediction_market_agent_tooling/tools/contract.py,sha256=Onm4vy3Tt8uH7w8CweDu6CD
 prediction_market_agent_tooling/tools/gnosis_rpc.py,sha256=9L3X5jBQcTZDQK4cHm5vzU7SOnjb0ERA6IWQ9C_bKUw,660
 prediction_market_agent_tooling/tools/google.py,sha256=P-ifiMk7bxQdU1ceON9apcZlw4rLHeOcYwxdeE4Of6Q,1761
 prediction_market_agent_tooling/tools/hexbytes_custom.py,sha256=2s_eMa0KwA_a9gLmcI4oXKHL3NwoDIACTSpceomeV7g,2037
-prediction_market_agent_tooling/tools/is_predictable.py,sha256=QEH2VEG-RsypXtxo4mPT37qtDVW5N4a8fwAFZGA65EI,2613
+prediction_market_agent_tooling/tools/is_predictable.py,sha256=MkElDkQRd2K3ZoAbqwk9WZluhqGDgx4vdt5cq_DoAz8,2618
 prediction_market_agent_tooling/tools/parallelism.py,sha256=_PCiD5apyO_jyXcmBDOOVwE7bQFZnhuA_MIjUwCYIjM,733
 prediction_market_agent_tooling/tools/singleton.py,sha256=CiIELUiI-OeS7U7eeHEt0rnVhtQGzwoUdAgn_7u_GBM,729
 prediction_market_agent_tooling/tools/utils.py,sha256=6MRfLr5xGPEQGobLz_eCsE_p-XaAeOCcmygwgNCIvqE,5044
 prediction_market_agent_tooling/tools/web3_utils.py,sha256=-ti5hPuyTmtgAHTpCJsP0zaxZEGa7H7zSMH_RtiSB8A,5314
-prediction_market_agent_tooling-0.13.0.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
-prediction_market_agent_tooling-0.13.0.dist-info/METADATA,sha256=r1vhW7v0N27cFuPOKsmGoHu_MRIBSblvFLp95LCwwQ4,4661
-prediction_market_agent_tooling-0.13.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-prediction_market_agent_tooling-0.13.0.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
-prediction_market_agent_tooling-0.13.0.dist-info/RECORD,,
+prediction_market_agent_tooling-0.13.2.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
+prediction_market_agent_tooling-0.13.2.dist-info/METADATA,sha256=Bmz1CcYHahP5ZNqzKP3aHjLK5bLk9gWY_nSsaHXfq2k,4699
+prediction_market_agent_tooling-0.13.2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+prediction_market_agent_tooling-0.13.2.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
+prediction_market_agent_tooling-0.13.2.dist-info/RECORD,,