prediction-market-agent-tooling 0.65.1__tar.gz → 0.65.3__tar.gz

{prediction_market_agent_tooling-0.65.1 → prediction_market_agent_tooling-0.65.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: prediction-market-agent-tooling
-Version: 0.65.1
+Version: 0.65.3
 Summary: Tools to benchmark, deploy and monitor prediction market agents.
 Author: Gnosis
 Requires-Python: >=3.10,<3.13
@@ -27,9 +27,9 @@ Requires-Dist: google-cloud-secret-manager (>=2.18.2,<3.0.0)
 Requires-Dist: hishel (>=0.0.31,<0.0.32)
 Requires-Dist: httpx (>=0.25.2,<1.0.0)
 Requires-Dist: isort (>=5.13.2,<6.0.0)
-Requires-Dist: langchain (>=0.2.6,<0.3.0) ; extra == "langchain"
+Requires-Dist: langchain (>=0.3.0,<0.4.0) ; extra == "langchain"
 Requires-Dist: langchain-community (>=0.0.19)
-Requires-Dist: langchain-openai (>=0.1.0,<0.2.0) ; extra == "langchain"
+Requires-Dist: langchain-openai (>=0.3.0,<0.4.0) ; extra == "langchain"
 Requires-Dist: langfuse (>=2.42.0,<3.0.0)
 Requires-Dist: loguru (>=0.7.2,<0.8.0)
 Requires-Dist: loky (>=3.4.1,<4.0.0)

{prediction_market_agent_tooling-0.65.1 → prediction_market_agent_tooling-0.65.3}/prediction_market_agent_tooling/markets/categorize.py

@@ -31,8 +31,8 @@ Write only the category itself, nothing else.
     research_evaluation_chain = (
         prompt
         | ChatOpenAI(
-            model=model,
-            api_key=APIKeys().openai_api_key_secretstr_v1,
+            model_name=model,
+            openai_api_key=APIKeys().openai_api_key,
         )
         | StrOutputParser()
     )

{prediction_market_agent_tooling-0.65.1 → prediction_market_agent_tooling-0.65.3}/prediction_market_agent_tooling/markets/seer/price_manager.py

@@ -122,10 +122,15 @@ class PriceManager:
         # Inspired by https://github.com/seer-pm/demo/blob/ca682153a6b4d4dd3dcc4ad8bdcbe32202fc8fe7/web/src/hooks/useMarketOdds.ts#L15
         price_data: dict[HexAddress, CollateralToken] = {}
 
-        for wrapped_token in self.seer_market.wrapped_tokens:
+        for idx, wrapped_token in enumerate(self.seer_market.wrapped_tokens):
             price = self.get_price_for_token(
                 token=Web3.to_checksum_address(wrapped_token),
             )
+            # It's okay if invalid (last) outcome has price 0, but not the other outcomes.
+            if price is None and idx != len(self.seer_market.wrapped_tokens) - 1:
+                raise ValueError(
+                    f"Couldn't get price for {wrapped_token} for market {self.seer_market.url}."
+                )
             price_data[wrapped_token] = (
                 price if price is not None else CollateralToken.zero()
             )
@@ -137,10 +142,9 @@ class PriceManager:
             sum(price_data.values(), start=CollateralToken.zero())
             == CollateralToken.zero()
         ):
-            return {
-                OutcomeStr(outcome): Probability(0)
-                for outcome in self.seer_market.outcomes
-            }
+            raise ValueError(
+                f"All prices for market {self.seer_market.url} are zero. This shouldn't happen."
+            )
 
         for outcome_token, price in price_data.items():
             old_price = price

{prediction_market_agent_tooling-0.65.1 → prediction_market_agent_tooling-0.65.3}/prediction_market_agent_tooling/markets/seer/seer_subgraph_handler.py

@@ -9,7 +9,7 @@ from prediction_market_agent_tooling.deploy.constants import (
     NO_OUTCOME_LOWERCASE_IDENTIFIER,
     YES_OUTCOME_LOWERCASE_IDENTIFIER,
 )
-from prediction_market_agent_tooling.gtypes import ChecksumAddress
+from prediction_market_agent_tooling.gtypes import ChecksumAddress, Wei
 from prediction_market_agent_tooling.loggers import logger
 from prediction_market_agent_tooling.markets.agent_market import FilterBy, SortBy
 from prediction_market_agent_tooling.markets.base_subgraph_handler import (
@@ -77,6 +77,7 @@ class SeerSubgraphHandler(BaseSubgraphHandler):
     @staticmethod
     def _build_where_statements(
         filter_by: FilterBy,
+        outcome_supply_gt_if_open: Wei,
         include_conditional_markets: bool = False,
         include_categorical_markets: bool = True,
     ) -> dict[Any, Any]:
@@ -88,6 +89,7 @@ class SeerSubgraphHandler(BaseSubgraphHandler):
             case FilterBy.OPEN:
                 and_stms["openingTs_gt"] = now
                 and_stms["hasAnswers"] = False
+                and_stms["outcomesSupply_gt"] = outcome_supply_gt_if_open.value
             case FilterBy.RESOLVED:
                 # We consider RESOLVED == CLOSED (on Seer UI)
                 and_stms["payoutReported"] = True
@@ -154,6 +156,7 @@ class SeerSubgraphHandler(BaseSubgraphHandler):
         filter_by: FilterBy,
         sort_by: SortBy = SortBy.NONE,
         limit: int | None = None,
+        outcome_supply_gt_if_open: Wei = Wei(0),
         include_conditional_markets: bool = True,
         include_categorical_markets: bool = True,
     ) -> list[SeerMarket]:
@@ -163,6 +166,7 @@ class SeerSubgraphHandler(BaseSubgraphHandler):
         # Binary markets on Seer contain 3 outcomes: OutcomeA, outcomeB and an Invalid option.
         where_stms = self._build_where_statements(
             filter_by=filter_by,
+            outcome_supply_gt_if_open=outcome_supply_gt_if_open,
             include_conditional_markets=include_conditional_markets,
             include_categorical_markets=include_categorical_markets,
         )

{prediction_market_agent_tooling-0.65.1 → prediction_market_agent_tooling-0.65.3}/prediction_market_agent_tooling/tools/image_gen/market_thumbnail_gen.py

@@ -17,9 +17,9 @@ def rewrite_question_into_image_generation_prompt(question: str) -> str:
             "openai not installed, please install extras `langchain` to use this function."
         )
     llm = ChatOpenAI(
-        model="gpt-4o-2024-08-06",
+        model_name="gpt-4o-2024-08-06",
         temperature=0.0,
-        api_key=APIKeys().openai_api_key_secretstr_v1,
+        openai_api_key=APIKeys().openai_api_key,
     )
     rewritten = str(
         llm.invoke(

{prediction_market_agent_tooling-0.65.1 → prediction_market_agent_tooling-0.65.3}/prediction_market_agent_tooling/tools/is_invalid.py

@@ -77,10 +77,10 @@ def is_invalid(
         return True
 
     llm = ChatOpenAI(
-        model=engine,
+        model_name=engine,
         temperature=temperature,
         seed=seed,
-        api_key=APIKeys().openai_api_key_secretstr_v1,
+        openai_api_key=APIKeys().openai_api_key,
     )
 
     prompt = ChatPromptTemplate.from_template(template=prompt_template)

{prediction_market_agent_tooling-0.65.1 → prediction_market_agent_tooling-0.65.3}/prediction_market_agent_tooling/tools/is_predictable.py

@@ -98,10 +98,10 @@ def is_predictable_binary(
         return True
 
     llm = ChatOpenAI(
-        model=engine,
+        model_name=engine,
         temperature=LLM_SUPER_LOW_TEMPERATURE,
         seed=LLM_SEED,
-        api_key=APIKeys().openai_api_key_secretstr_v1,
+        openai_api_key=APIKeys().openai_api_key,
     )
 
     prompt = ChatPromptTemplate.from_template(template=prompt_template)
@@ -138,10 +138,10 @@ def is_predictable_without_description(
         return True
 
     llm = ChatOpenAI(
-        model=engine,
+        model_name=engine,
         temperature=LLM_SUPER_LOW_TEMPERATURE,
         seed=LLM_SEED,
-        api_key=APIKeys().openai_api_key_secretstr_v1,
+        openai_api_key=APIKeys().openai_api_key,
     )
 
     prompt = ChatPromptTemplate.from_template(template=prompt_template)

{prediction_market_agent_tooling-0.65.1 → prediction_market_agent_tooling-0.65.3}/prediction_market_agent_tooling/tools/relevant_news_analysis/relevant_news_analysis.py

@@ -70,8 +70,8 @@ def analyse_news_relevance(
     )
     llm = ChatOpenAI(
         temperature=temperature,
-        model=model,
-        api_key=APIKeys().openai_api_key_secretstr_v1,
+        model_name=model,
+        openai_api_key=APIKeys().openai_api_key,
     )
     chain = prompt | llm | parser
 

{prediction_market_agent_tooling-0.65.1 → prediction_market_agent_tooling-0.65.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "prediction-market-agent-tooling"
-version = "0.65.1"
+version = "0.65.3"
 description = "Tools to benchmark, deploy and monitor prediction market agents."
 authors = ["Gnosis"]
 readme = "README.md"
@@ -31,8 +31,8 @@ scikit-learn = "^1.3.1"
 tabulate = "^0.9.0"
 types-pytz = "^2024.1.0.20240203"
 google-cloud-secret-manager = "^2.18.2"
-langchain = { version = "^0.2.6", optional = true}
-langchain-openai = { version = "^0.1.0", optional = true}
+langchain = { version = "^0.3.0", optional = true}
+langchain-openai = { version = "^0.3.0", optional = true}
 google-api-python-client = { version = "2.95.0", optional = true}
 subgrounds-gnosis = "^1.9.2"
 loguru = "^0.7.2"