exa-py 1.14.0__py3-none-any.whl → 1.14.1__py3-none-any.whl

exa_py/research/client.py

@@ -9,7 +9,7 @@ block, but at runtime we only pay the cost if/when a helper is actually used.
 
 from __future__ import annotations
 
-from typing import TYPE_CHECKING, Any, Dict, Optional
+from typing import TYPE_CHECKING, Any, Dict, Optional, Literal
 
 if TYPE_CHECKING:  # pragma: no cover – only for static analysers
     # Import with full type info when static type-checking.  `_Result` still
@@ -39,15 +39,20 @@ class ResearchClient:
         self,
         *,
         instructions: str,
-        model: str = "exa-research",
-        output_schema: Dict[str, Any],
+        model: Literal["exa-research", "exa-research-pro"] = "exa-research",
+        output_infer_schema: bool = None,
+        output_schema: Dict[str, Any] = None,
     ) -> "ResearchTaskId":
         """Submit a research request and return the *task identifier*."""
-        payload = {
-            "instructions": instructions,
-            "model": model,
-            "output": {"schema": output_schema},
-        }
+        payload = {"instructions": instructions}
+        if model is not None:
+            payload["model"] = model
+        if output_schema is not None or output_infer_schema is not None:
+            payload["output"] = {}
+            if output_schema is not None:
+                payload["output"]["schema"] = output_schema
+            if output_infer_schema is not None:
+                payload["output"]["inferSchema"] = output_infer_schema
 
         raw_response: Dict[str, Any] = self._client.request(
             "/research/v0/tasks", payload
@@ -64,9 +69,7 @@ class ResearchClient:
 
         return ResearchTaskId(id=raw_response["id"])
 
-    def get_task(
-        self, id: str
-    ) -> "ResearchTask":  # noqa: D401 – imperative mood is fine
+    def get_task(self, id: str) -> "ResearchTask":  # noqa: D401 – imperative mood is fine
         """Fetch the current status / result for a research task."""
         endpoint = f"/research/v0/tasks/{id}"
 
@@ -175,7 +178,7 @@ class AsyncResearchClient:
         self,
         *,
         instructions: str,
-        model: str = "exa-research",
+        model: Literal["exa-research", "exa-research-pro"] = "exa-research",
         output_schema: Dict[str, Any],
     ) -> "ResearchTaskId":
         """Submit a research request and return the *task identifier* (async)."""

{exa_py-1.14.0.dist-info → exa_py-1.14.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: exa-py
-Version: 1.14.0
+Version: 1.14.1
 Summary: Python SDK for Exa API.
 License: MIT
 Author: Exa AI

{exa_py-1.14.0.dist-info → exa_py-1.14.1.dist-info}/RECORD

@@ -2,7 +2,7 @@ exa_py/__init__.py,sha256=M2GC9oSdoV6m2msboW0vMWWl8wrth4o6gmEV4MYLGG8,66
 exa_py/api.py,sha256=zwTEiIkWa6Gynx5NW9VMMY4Nws63gVDRSmzKdT4v2aY,84903
 exa_py/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 exa_py/research/__init__.py,sha256=QeY-j6bP4QP5tF9ytX0IeQhJvd0Wn4cJCD69U8pP7kA,271
-exa_py/research/client.py,sha256=oFQpczMcfA20Sn_yq36As8GlFo2u8B77IqVMMRqt3qE,11481
+exa_py/research/client.py,sha256=C2ukFq_dE1xUfhMlHwpD9cY5rDClgn8N92pH4_FEVpE,11901
 exa_py/research/models.py,sha256=j7YgRoMRp2MLgnaij7775x_hJEeV5gksKpfLwmawqxY,3704
 exa_py/utils.py,sha256=Rc1FJjoR9LQ7L_OJM91Sd1GNkbHjcLyEvJENhRix6gc,2405
 exa_py/websets/__init__.py,sha256=uOBAb9VrIHrPKoddGOp2ai2KgWlyUVCLMZqfbGOlboA,70
@@ -23,6 +23,6 @@ exa_py/websets/streams/runs/client.py,sha256=ja63-G0Y74YDgsWq30aoa0llIx8I9mIwbCb
 exa_py/websets/types.py,sha256=-KmC6N9LMh5fPhhXZslmUyQcSmBqIdPF-qQ27qS4GEw,34756
 exa_py/websets/webhooks/__init__.py,sha256=iTPBCxFd73z4RifLQMX6iRECx_6pwlI5qscLNjMOUHE,77
 exa_py/websets/webhooks/client.py,sha256=zsIRMTeJU65yj-zo7Zz-gG02Prtzgcx6utGFSoY4HQQ,4222
-exa_py-1.14.0.dist-info/METADATA,sha256=jOIKH214iEOdk7HE46-tyhST5d9iGvgCaaff12tPxiY,4120
-exa_py-1.14.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-exa_py-1.14.0.dist-info/RECORD,,
+exa_py-1.14.1.dist-info/METADATA,sha256=Mf9xdxUdIxEXkxcXa_mTDKRt34M4QR0lR7IPF2H26k8,4120
+exa_py-1.14.1.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+exa_py-1.14.1.dist-info/RECORD,,