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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: prediction-market-agent-tooling
-Version: 0.13.0
+Version: 0.13.1
 Summary: Tools to benchmark, deploy and monitor prediction market agents.
 Author: Gnosis
 Requires-Python: >=3.10,<3.12
@@ -20,7 +20,7 @@ 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: numpy (>=1.26.4,<2.0.0)
 Requires-Dist: pydantic (>=2.6.1,<3.0.0)

{prediction_market_agent_tooling-0.13.0 → prediction_market_agent_tooling-0.13.1}/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.1}/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.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "prediction-market-agent-tooling"
-version = "0.13.0"
+version = "0.13.1"
 description = "Tools to benchmark, deploy and monitor prediction market agents."
 authors = ["Gnosis"]
 readme = "README.md"
@@ -32,7 +32,7 @@ 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"