prediction-market-agent-tooling 0.65.4__py3-none-any.whl → 0.65.6__py3-none-any.whl

prediction_market_agent_tooling/config.py

@@ -65,6 +65,8 @@ class APIKeys(BaseSettings):
 
     SQLALCHEMY_DB_URL: t.Optional[SecretStr] = None
 
+    PERPLEXITY_API_KEY: t.Optional[SecretStr] = None
+
     ENABLE_CACHE: bool = False
     CACHE_DIR: str = "./.cache"
 
@@ -259,6 +261,12 @@ class APIKeys(BaseSettings):
             if self.model_fields[k].annotation not in SECRET_TYPES and v is not None
         }
 
+    @property
+    def perplexity_api_key(self) -> SecretStr:
+        return check_not_none(
+            self.PERPLEXITY_API_KEY, "PERPLEXITY_API_KEY missing in the environment."
+        )
+
     def model_dump_secrets(self) -> dict[str, t.Any]:
         return {
             k: v.get_secret_value() if isinstance(v, SecretStr) else v

prediction_market_agent_tooling/deploy/agent.py

@@ -222,6 +222,7 @@ class DeployablePredictionAgent(DeployableAgent):
         self.have_bet_on_market_since = observe()(self.have_bet_on_market_since)  # type: ignore[method-assign]
         self.verify_market = observe()(self.verify_market)  # type: ignore[method-assign]
         self.answer_binary_market = observe()(self.answer_binary_market)  # type: ignore[method-assign]
+        self.answer_categorical_market = observe()(self.answer_categorical_market)  # type: ignore[method-assign]
         self.process_market = observe()(self.process_market)  # type: ignore[method-assign]
 
     def update_langfuse_trace_by_market(
@@ -321,7 +322,7 @@ class DeployablePredictionAgent(DeployableAgent):
     def fetch_categorical_markets(self) -> bool:
         # Check if the subclass has implemented the answer_categorical_market method, if yes, fetch categorical markets as well.
         if (
-            self.answer_categorical_market.__func__  # type: ignore[attr-defined] # This works just fine, but mypy doesn't know about it for some reason.
+            self.answer_categorical_market.__wrapped__.__func__  # type: ignore[attr-defined] # This works just fine, but mypy doesn't know about it for some reason.
             is not DeployablePredictionAgent.answer_categorical_market
         ):
             return True

prediction_market_agent_tooling/logprobs_parser.py

@@ -61,7 +61,8 @@ class LogprobsParser:
             (
                 i
                 for i in range(result_start_index, len(logprobs))
-                if logprobs[i]["token"] in {",", '"', ",\n", "\",\n'", '",\n'}
+                if logprobs[i]["token"]
+                in {",", '"', ",\n", "\",\n'", '",\n', '"\n', "\n"}
             ),
             len(logprobs) - 1,
         )

prediction_market_agent_tooling/tools/perplexity/perplexity_client.py

@@ -0,0 +1,86 @@
+from typing import Any, Dict, List, Optional
+
+import httpx
+from pydantic import SecretStr
+
+from prediction_market_agent_tooling.tools.perplexity.perplexity_models import (
+    PerplexityModelSettings,
+    PerplexityRequestParameters,
+    PerplexityResponse,
+)
+
+
+class PerplexityModel:
+    def __init__(
+        self,
+        model_name: str,
+        *,
+        api_key: SecretStr,
+        completition_endpoint: str = "https://api.perplexity.ai/chat/completions",
+    ) -> None:
+        self.model_name = model_name
+        self.api_key = api_key
+        self.completition_endpoint = completition_endpoint
+
+    async def request(
+        self,
+        messages: List[dict[str, str]],
+        model_settings: Optional[PerplexityModelSettings],
+        model_request_parameters: PerplexityRequestParameters,
+    ) -> PerplexityResponse:
+        payload: Dict[str, Any] = {"model": self.model_name, "messages": messages}
+
+        if model_settings:
+            model_settings_dict = model_settings.model_dump()
+            model_settings_dict = {
+                k: v for k, v in model_settings_dict.items() if v is not None
+            }
+            payload.update(model_settings_dict)
+
+        params_dict = model_request_parameters.model_dump()
+        params_dict = {k: v for k, v in params_dict.items() if v is not None}
+
+        # Extract and handle search_context_size specially
+        if "search_context_size" in params_dict:
+            search_context_size = params_dict.pop("search_context_size")
+            payload["web_search_options"] = {"search_context_size": search_context_size}
+
+        # Add remaining Perplexity parameters to payload
+        payload.update(params_dict)
+
+        try:
+            async with httpx.AsyncClient(timeout=180) as client:
+                response = await client.post(
+                    self.completition_endpoint,
+                    headers={
+                        "Authorization": f"Bearer {self.api_key.get_secret_value()}",
+                        "Content-Type": "application/json",
+                    },
+                    json=payload,
+                )
+                response.raise_for_status()
+                result: dict[str, Any] = response.json()
+
+                choices = result.get("choices", [])
+                if not choices:
+                    raise ValueError("Invalid response: no choices")
+
+                content = choices[0].get("message", {}).get("content")
+                if not content:
+                    raise ValueError("Invalid response: no content")
+
+                return PerplexityResponse(
+                    content=content,
+                    citations=result.get("citations", []),
+                    usage=result.get("usage", {}),
+                )
+        except httpx.HTTPStatusError as e:
+            raise ValueError(
+                f"HTTP error from Perplexity API: {e.response.status_code} - {e.response.text}"
+            ) from e
+        except httpx.RequestError as e:
+            raise ValueError(f"Request error to Perplexity API: {str(e)}") from e
+        except Exception as e:
+            raise ValueError(
+                f"Unexpected error in Perplexity API request: {str(e)}"
+            ) from e

prediction_market_agent_tooling/tools/perplexity/perplexity_models.py

@@ -0,0 +1,26 @@
+from typing import Any, List, Literal, Optional
+
+from pydantic import BaseModel
+
+
+class PerplexityRequestParameters(BaseModel):
+    search_context_size: Optional[Literal["low", "medium", "high"]]
+    search_recency_filter: Optional[Literal["any", "day", "week", "month", "year"]]
+    search_return_related_questions: Optional[bool]
+    search_domain_filter: Optional[List[str]]
+    search_after_date_filter: Optional[str]
+    search_before_date_filter: Optional[str]
+
+
+class PerplexityResponse(BaseModel):
+    content: str
+    citations: list[str]
+    usage: dict[str, Any]
+
+
+class PerplexityModelSettings(BaseModel):
+    max_tokens: Optional[int] = None
+    temperature: Optional[float] = None
+    top_p: Optional[float] = None
+    frequency_penalty: Optional[float] = None
+    presence_penalty: Optional[float] = None

prediction_market_agent_tooling/tools/perplexity/perplexity_search.py

@@ -0,0 +1,73 @@
+import asyncio
+import typing as t
+from datetime import date, timedelta
+
+import tenacity
+
+from prediction_market_agent_tooling.config import APIKeys
+from prediction_market_agent_tooling.tools.caches.db_cache import db_cache
+from prediction_market_agent_tooling.tools.perplexity.perplexity_client import (
+    PerplexityModel,
+)
+from prediction_market_agent_tooling.tools.perplexity.perplexity_models import (
+    PerplexityModelSettings,
+    PerplexityRequestParameters,
+    PerplexityResponse,
+)
+
+SYSTEM_PROMPT = "You are a helpful search assistant. Your task is to provide accurate information based on web searches."
+
+
+@tenacity.retry(stop=tenacity.stop_after_attempt(3), wait=tenacity.wait_fixed(1))
+@db_cache(
+    max_age=timedelta(days=1),
+    ignore_args=["api_keys"],
+    log_error_on_unsavable_data=False,
+)
+def perplexity_search(
+    query: str,
+    api_keys: APIKeys,
+    search_context_size: t.Literal["low", "medium", "high"] = "medium",
+    search_recency_filter: t.Literal["any", "day", "week", "month", "year"]
+    | None = None,
+    search_filter_before_date: date | None = None,
+    search_filter_after_date: date | None = None,
+    search_return_related_questions: bool | None = None,
+    include_domains: list[str] | None = None,
+    temperature: float = 0,
+    model_name: str = "sonar-pro",
+    max_tokens: int = 2048,
+) -> PerplexityResponse:
+    # Create messages in ModelMessage format
+    messages = [
+        {"role": "system", "content": SYSTEM_PROMPT},
+        {"role": "user", "content": query},
+    ]
+
+    # Define special parameters for the request and create the settings
+    model_settings = PerplexityModelSettings(
+        max_tokens=max_tokens, temperature=temperature
+    )
+
+    # Create a basic request parameters object with required base parameters
+    request_params = PerplexityRequestParameters(
+        search_domain_filter=include_domains,
+        search_after_date_filter=search_filter_after_date.strftime("%Y-%m-%d")
+        if search_filter_after_date
+        else None,
+        search_before_date_filter=search_filter_before_date.strftime("%Y-%m-%d")
+        if search_filter_before_date
+        else None,
+        search_recency_filter=search_recency_filter,
+        search_context_size=search_context_size,
+        search_return_related_questions=search_return_related_questions,
+    )
+
+    model = PerplexityModel(model_name=model_name, api_key=api_keys.perplexity_api_key)
+    return asyncio.run(
+        model.request(
+            messages=messages,
+            model_settings=model_settings,
+            model_request_parameters=request_params,
+        )
+    )

{prediction_market_agent_tooling-0.65.4.dist-info → prediction_market_agent_tooling-0.65.6.dist-info}/METADATA

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

{prediction_market_agent_tooling-0.65.4.dist-info → prediction_market_agent_tooling-0.65.6.dist-info}/RECORD

@@ -23,8 +23,8 @@ prediction_market_agent_tooling/benchmark/agents.py,sha256=zC5tUM6pPTWtqSddOOSYV
 prediction_market_agent_tooling/benchmark/benchmark.py,sha256=KwMZzwise3sgmhdjw7xCgHMmjKHdHqQC-zc9untOLWg,17832
 prediction_market_agent_tooling/benchmark/utils.py,sha256=xQd7p9H08-OtN3iC4QT2i9bkUTmrXa6rxGXeg9yMhgU,2986
 prediction_market_agent_tooling/chains.py,sha256=1qQstoqXMwqwM7k-KH7MjMz8Ei-D83KZByvDbCZpAxs,116
-prediction_market_agent_tooling/config.py,sha256=UyhK6u261tBLVsRwvXJm9C0wy6UrUpfAKH6xk4CSfk8,11535
-prediction_market_agent_tooling/deploy/agent.py,sha256=kxfJNfuvGJ2LFReI-PCyjGuubZKIrPEYiyphaS6eKcM,25476
+prediction_market_agent_tooling/config.py,sha256=-kJfdDr-m0R-tGZ1KRI-hJJk0mXDt142CAlvwaJ2N2I,11778
+prediction_market_agent_tooling/deploy/agent.py,sha256=ddFvBUhB_WVWtuUGkiZNKBME4Hd3AdquVv588c1rpu0,25602
 prediction_market_agent_tooling/deploy/agent_example.py,sha256=yS1fWkHynr9MYGNOM2WsCnRWLPaffY4bOc6bIudrdd4,1377
 prediction_market_agent_tooling/deploy/betting_strategy.py,sha256=YYayGjTKW02d3BUavJ8M3NmFk41oldEM3FHbwppZGRM,17184
 prediction_market_agent_tooling/deploy/constants.py,sha256=Qe9cllgsGMkecfmbhXoFkPxuJyG6ATsrT87RF9SmPWM,249
@@ -37,7 +37,7 @@ prediction_market_agent_tooling/jobs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeu
 prediction_market_agent_tooling/jobs/jobs_models.py,sha256=DoZ9dlvVhpNrnINiR1uy6YUOsuzI_L-avBt362y5xXM,2467
 prediction_market_agent_tooling/jobs/omen/omen_jobs.py,sha256=qbTZ9HVvu_iP4dDxuvOZxAp6JsRKejvEW2YDYCnRmd4,5039
 prediction_market_agent_tooling/loggers.py,sha256=kFZ1BrI8hvWgZO1vzptFnYiOEDx9Ozs86DA9yF3bSgY,5212
-prediction_market_agent_tooling/logprobs_parser.py,sha256=Du1Yc-fAVSixQX_Zx6KWpgSzI_ZYhv5tS1b8IcOPPr8,4979
+prediction_market_agent_tooling/logprobs_parser.py,sha256=DBlBQtWX8_URXhzTU3YWIPa76Zx3QDHlx1ARqbgJsVI,5008
 prediction_market_agent_tooling/markets/agent_market.py,sha256=smKuMaP1zyTtw31Robbj3ospF03xptWX0EDKPJbHH9I,19070
 prediction_market_agent_tooling/markets/base_subgraph_handler.py,sha256=7RaYO_4qAmQ6ZGM8oPK2-CkiJfKmV9MxM-rJlduaecU,1971
 prediction_market_agent_tooling/markets/blockchain_utils.py,sha256=gZMwCTO-1d2ALXeY7-LP6U4kCpotJ6GMPZwN11kFOfE,2604
@@ -100,6 +100,9 @@ prediction_market_agent_tooling/tools/langfuse_client_utils.py,sha256=6IvsqqNNfB
 prediction_market_agent_tooling/tools/omen/reality_accuracy.py,sha256=M1SF7iSW1gVlQSTskdVFTn09uPLST23YeipVIWj54io,2236
 prediction_market_agent_tooling/tools/omen/sell_positions.py,sha256=Q4oI7_QI3AkyxlH10VvxDahYVrphQa1Wnox2Ce_cf_k,2452
 prediction_market_agent_tooling/tools/parallelism.py,sha256=6Gou0hbjtMZrYvxjTDFUDZuxmE2nqZVbb6hkg1hF82A,1022
+prediction_market_agent_tooling/tools/perplexity/perplexity_client.py,sha256=Oi5a_uaCAPtpTAEEVwD0SXWTsKkhCXOQjCBfa1eAuC0,3215
+prediction_market_agent_tooling/tools/perplexity/perplexity_models.py,sha256=SfeR5FGQ2PqaxG4f4ij50LEnS9f2aeLplVwD7IZQh1s,820
+prediction_market_agent_tooling/tools/perplexity/perplexity_search.py,sha256=Ii-5bqk6DygOCK1G78PDdz2sRTNvFYHVYVl2sajUbY4,2616
 prediction_market_agent_tooling/tools/relevant_news_analysis/data_models.py,sha256=95l84aztFaxcRLLcRQ46yKJbIlOEuDAbIGLouyliDzA,1316
 prediction_market_agent_tooling/tools/relevant_news_analysis/relevant_news_analysis.py,sha256=r4MdP5uEMlaCwoa2XQZzGq3oZEqnoo9S4dg8uzXfSOY,5473
 prediction_market_agent_tooling/tools/relevant_news_analysis/relevant_news_cache.py,sha256=kNWq92T11Knb9mYBZlMiZUzOpKgCd-5adanylQUMRJA,3085
@@ -116,8 +119,8 @@ prediction_market_agent_tooling/tools/tokens/usd.py,sha256=yuW8iPPtcpP4eLH2nORMD
 prediction_market_agent_tooling/tools/transaction_cache.py,sha256=K5YKNL2_tR10Iw2TD9fuP-CTGpBbZtNdgbd0B_R7pjg,1814
 prediction_market_agent_tooling/tools/utils.py,sha256=Jzpck3_QwShhejhgbAhmNxPSOPQJssBQep0eVemVjZ4,7064
 prediction_market_agent_tooling/tools/web3_utils.py,sha256=zRq-eeBGWt8uUGN9G_WfjmJ0eVvO8aWE9S0Pz_Y6AOA,12342
-prediction_market_agent_tooling-0.65.4.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
-prediction_market_agent_tooling-0.65.4.dist-info/METADATA,sha256=bG8oQxcVytt5TcAg1I-7qsmGqL2aCOZJ-FvKB-7tgQE,8734
-prediction_market_agent_tooling-0.65.4.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-prediction_market_agent_tooling-0.65.4.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
-prediction_market_agent_tooling-0.65.4.dist-info/RECORD,,
+prediction_market_agent_tooling-0.65.6.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
+prediction_market_agent_tooling-0.65.6.dist-info/METADATA,sha256=XT7Kp9vv8HBSZjedQV7VgRp7H4Awv4GQdjrzz1F7Z-E,8734
+prediction_market_agent_tooling-0.65.6.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+prediction_market_agent_tooling-0.65.6.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
+prediction_market_agent_tooling-0.65.6.dist-info/RECORD,,