prediction-market-agent-tooling 0.13.0__tar.gz → 0.13.2__tar.gz

{prediction_market_agent_tooling-0.13.0 → prediction_market_agent_tooling-0.13.2}/PKG-INFO

@@ -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 → prediction_market_agent_tooling-0.13.2}/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-0.13.0 → prediction_market_agent_tooling-0.13.2}/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-0.13.0 → prediction_market_agent_tooling-0.13.2}/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-0.13.0 → prediction_market_agent_tooling-0.13.2}/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-0.13.0 → prediction_market_agent_tooling-0.13.2}/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 → prediction_market_agent_tooling-0.13.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "prediction-market-agent-tooling"
-version = "0.13.0"
+version = "0.13.2"
 description = "Tools to benchmark, deploy and monitor prediction market agents."
 authors = ["Gnosis"]
 readme = "README.md"
@@ -12,14 +12,14 @@ sell_omen = "scripts.bet_omen:sell"
 [tool.poetry.dependencies]
 python = ">=3.10,<3.12"
 typer = "^0.9.0"
-types-requests = "^2.31.0.20240106"
+types-requests = "^2.31.0.0"
 google-cloud-functions = "^1.16.0"
 google-cloud-resource-manager = "^1.12.0"
 functions-framework = "^3.5.0"
 cron-validator = "^1.0.8"
 pydantic = "^2.6.1"
 web3 = "^6.15.1"
-eth-typing = "^4.0.0"
+eth-typing = "^3.0.0"
 pydantic-settings = "^2.1.0"
 numpy = "^1.26.4"
 autoflake = "^2.2.1"
@@ -32,10 +32,11 @@ tabulate = "^0.9.0"
 types-pytz = "^2024.1.0.20240203"
 google-cloud-secret-manager = "^2.18.2"
 langchain = { version = "^0.1.9", optional = true}
-langchain-openai = { version = "^0.0.8", optional = true}
+langchain-openai = { version = "^0.0.5", optional = true}
 google-api-python-client = { version = "2.95.0", optional = true}
 subgrounds = "^1.8.1"
 loguru = "^0.7.2"
+mech-client = "^0.2.13"
 
 [tool.poetry.extras]
 langchain = ["langchain", "langchain-openai"]
@@ -43,7 +44,7 @@ google = ["google-api-python-client"]
 
 [tool.poetry.group.dev.dependencies]
 pytest = "*"
-mypy = "^1.8.0"
+mypy = "^1.9.0"
 black = "^23.12.1"
 
 [build-system]