prediction-market-agent-tooling 0.66.0.dev788__tar.gz → 0.66.0.dev791__tar.gz

{prediction_market_agent_tooling-0.66.0.dev788 → prediction_market_agent_tooling-0.66.0.dev791}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: prediction-market-agent-tooling
-Version: 0.66.0.dev788
+Version: 0.66.0.dev791
 Summary: Tools to benchmark, deploy and monitor prediction market agents.
 Author: Gnosis
 Requires-Python: >=3.10,<3.13

{prediction_market_agent_tooling-0.66.0.dev788 → prediction_market_agent_tooling-0.66.0.dev791}/prediction_market_agent_tooling/deploy/agent.py

@@ -373,7 +373,8 @@ class DeployablePredictionAgent(DeployableAgent):
                 binary_answer = self.answer_binary_market(market)
                 return (
                     CategoricalProbabilisticAnswer.from_probabilistic_answer(
-                        binary_answer
+                        binary_answer,
+                        market.outcomes,
                     )
                     if binary_answer is not None
                     else None

{prediction_market_agent_tooling-0.66.0.dev788 → prediction_market_agent_tooling-0.66.0.dev791}/prediction_market_agent_tooling/markets/data_models.py

@@ -15,6 +15,10 @@ from prediction_market_agent_tooling.gtypes import (
     Probability,
 )
 from prediction_market_agent_tooling.logprobs_parser import FieldLogprobs
+from prediction_market_agent_tooling.markets.omen.omen import (
+    OMEN_FALSE_OUTCOME,
+    OMEN_TRUE_OUTCOME,
+)
 from prediction_market_agent_tooling.tools.utils import DatetimeUTC, check_not_none
 
 
@@ -142,13 +146,20 @@ class CategoricalProbabilisticAnswer(BaseModel):
     @staticmethod
     def from_probabilistic_answer(
         answer: ProbabilisticAnswer,
+        market_outcomes: Sequence[OutcomeStr] | None = None,
     ) -> "CategoricalProbabilisticAnswer":
         return CategoricalProbabilisticAnswer(
             probabilities={
-                OutcomeStr(YES_OUTCOME_LOWERCASE_IDENTIFIER): answer.p_yes,
-                OutcomeStr(NO_OUTCOME_LOWERCASE_IDENTIFIER): Probability(
-                    1 - answer.p_yes
-                ),
+                (
+                    OMEN_TRUE_OUTCOME
+                    if market_outcomes and OMEN_TRUE_OUTCOME in market_outcomes
+                    else OutcomeStr(YES_OUTCOME_LOWERCASE_IDENTIFIER)
+                ): answer.p_yes,
+                (
+                    OMEN_FALSE_OUTCOME
+                    if market_outcomes and OMEN_FALSE_OUTCOME in market_outcomes
+                    else OutcomeStr(NO_OUTCOME_LOWERCASE_IDENTIFIER)
+                ): Probability(1 - answer.p_yes),
             },
             confidence=answer.confidence,
             reasoning=answer.reasoning,

{prediction_market_agent_tooling-0.66.0.dev788 → prediction_market_agent_tooling-0.66.0.dev791}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "prediction-market-agent-tooling"
-version = "0.66.0.dev788"
+version = "0.66.0.dev791"
 description = "Tools to benchmark, deploy and monitor prediction market agents."
 authors = ["Gnosis"]
 readme = "README.md"