exa-py 1.14.3__tar.gz → 1.14.4__tar.gz

{exa_py-1.14.3 → exa_py-1.14.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: exa-py
-Version: 1.14.3
+Version: 1.14.4
 Summary: Python SDK for Exa API.
 Home-page: https://github.com/exa-labs/exa-py
 Author: Exa

{exa_py-1.14.3 → exa_py-1.14.4}/exa_py/api.py

@@ -788,6 +788,15 @@ class AsyncStreamAnswerResponse:
 
 T = TypeVar("T")
 
+@dataclass
+class ContentStatus:
+    """A class representing the status of a content retrieval operation."""
+
+    id: str
+    status: str
+    source: str
+
+
 
 @dataclass
 class SearchResponse(Generic[T]):
@@ -804,6 +813,7 @@ class SearchResponse(Generic[T]):
     autoprompt_string: Optional[str]
     resolved_search_type: Optional[str]
     auto_date: Optional[str]
+    statuses: Optional[List[ContentStatus]] = None
     cost_dollars: Optional[CostDollars] = None
 
     def __str__(self):
@@ -818,6 +828,8 @@ class SearchResponse(Generic[T]):
                 output += f"\n  - search: {self.cost_dollars.search}"
             if self.cost_dollars.contents:
                 output += f"\n  - contents: {self.cost_dollars.contents}"
+        if self.statuses:
+            output += f"\nStatuses: {self.statuses}"
         return output
 
 
@@ -1402,15 +1414,18 @@ class Exa:
             options,
             {**CONTENTS_OPTIONS_TYPES, **CONTENTS_ENDPOINT_OPTIONS_TYPES},
         )
+        
         options = to_camel_case(options)
         data = self.request("/contents", options)
         cost_dollars = parse_cost_dollars(data.get("costDollars"))
+        statuses = [ContentStatus(**status) for status in data.get("statuses", [])]
         return SearchResponse(
             [Result(**to_snake_case(result)) for result in data["results"]],
             data.get("autopromptString"),
             data.get("resolvedSearchType"),
             data.get("autoDate"),
             cost_dollars=cost_dollars,
+            statuses=statuses,
         )
 
     def find_similar(
@@ -2092,12 +2107,14 @@ class AsyncExa(Exa):
         options = to_camel_case(options)
         data = await self.async_request("/contents", options)
         cost_dollars = parse_cost_dollars(data.get("costDollars"))
+        statuses = [ContentStatus(**status) for status in data.get("statuses", [])]
         return SearchResponse(
             [Result(**to_snake_case(result)) for result in data["results"]],
             data.get("autopromptString"),
             data.get("resolvedSearchType"),
             data.get("autoDate"),
             cost_dollars=cost_dollars,
+            statuses=statuses,
         )
 
     async def find_similar(

{exa_py-1.14.3 → exa_py-1.14.4}/exa_py.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: exa-py
-Version: 1.14.3
+Version: 1.14.4
 Summary: Python SDK for Exa API.
 Home-page: https://github.com/exa-labs/exa-py
 Author: Exa

{exa_py-1.14.3 → exa_py-1.14.4}/exa_py.egg-info/SOURCES.txt

@@ -31,4 +31,5 @@ exa_py/websets/searches/client.py
 exa_py/websets/webhooks/__init__.py
 exa_py/websets/webhooks/client.py
 tests/test_monitors.py
+tests/test_search_api.py
 tests/test_websets.py

{exa_py-1.14.3 → exa_py-1.14.4}/pyproject.toml

@@ -32,7 +32,7 @@ in-project = true
 
 [project]
 name = "exa-py"
-version = "1.14.3"
+version = "1.14.4"
 description = "Python SDK for Exa API."
 readme = "README.md"
 requires-python = ">=3.9"

{exa_py-1.14.3 → exa_py-1.14.4}/setup.py

@@ -2,7 +2,7 @@ from setuptools import find_packages, setup
 
 setup(
     name="exa_py",
-    version="1.12.4",
+    version="1.14.4",
     description="Python SDK for Exa API.",
     long_description_content_type="text/markdown",
     long_description=open("README.md").read(),

exa_py-1.14.4/tests/test_search_api.py

@@ -0,0 +1,107 @@
+import os
+from unittest.mock import AsyncMock, patch
+
+import httpx
+import pytest
+
+from exa_py import Exa, AsyncExa, api as exa_api
+
+API_KEY = os.getenv("EXA_API_KEY", "test-key")
+
+
+def _have_real_key() -> bool:
+    return API_KEY != "test-key" and len(API_KEY) > 10
+
+########################################
+# Offline unit tests (no network)
+########################################
+
+def test_contentstatus_parsing_offline():
+    payload_status = {"id": "u", "status": "success", "source": "cached"}
+    cs = exa_api.ContentStatus(**payload_status)
+    assert cs.id == "u" and cs.status == "success" and cs.source == "cached"
+
+
+def test_answerresponse_accepts_dict():
+    dummy = exa_api.AnswerResult(id="1", url="u", title="t")
+    resp = exa_api.AnswerResponse(answer={"foo": "bar"}, citations=[dummy])
+    assert isinstance(resp.answer, dict) and resp.answer["foo"] == "bar"
+
+
+@pytest.mark.asyncio
+async def test_async_request_accepts_201():
+    ax = AsyncExa(API_KEY)
+
+    async def _fake_post(url, json, headers):
+        return httpx.Response(201, json={"ok": True})
+
+    with patch.object(ax.client, "post", new=AsyncMock(side_effect=_fake_post)):
+        data = await ax.async_request("/dummy", {})
+    assert data == {"ok": True}
+
+
+########################################
+# Live integration tests (skipped without key)
+########################################
+
+@pytest.mark.skipif(not _have_real_key(), reason="EXA_API_KEY not provided")
+def test_user_agent_header():
+    exa = Exa(API_KEY)
+    assert exa.headers["User-Agent"] == "exa-py 1.12.4"
+
+
+@pytest.mark.skipif(not _have_real_key(), reason="EXA_API_KEY not provided")
+def test_research_client_attrs():
+    exa = Exa(API_KEY)
+    aexa = AsyncExa(API_KEY)
+    assert hasattr(exa, "research") and hasattr(aexa, "research")
+
+
+# ---- Core live endpoint smoke checks ----
+
+@pytest.mark.skipif(not _have_real_key(), reason="EXA_API_KEY not provided")
+def test_get_contents_live_preferred():
+    exa = Exa(API_KEY)
+    resp = exa.get_contents(urls=["https://techcrunch.com"], text=True, livecrawl="preferred")
+    assert isinstance(resp, exa_api.ContentResponse)
+    # statuses may be empty when cached – still fine
+    assert len(resp.results) >= 1
+
+
+@pytest.mark.skipif(not _have_real_key(), reason="EXA_API_KEY not provided")
+def test_search_and_contents_live():
+    exa = Exa(API_KEY)
+    resp = exa.search_and_contents("openai", num_results=1, text=True)
+    assert resp.results and resp.results[0].text
+
+
+@pytest.mark.skipif(not _have_real_key(), reason="EXA_API_KEY not provided")
+def test_find_similar_live():
+    exa = Exa(API_KEY)
+    resp = exa.find_similar("https://www.openai.com", num_results=1)
+    assert resp.results
+
+
+@pytest.mark.skipif(not _have_real_key(), reason="EXA_API_KEY not provided")
+def test_get_contents_sync_live():
+    exa = Exa(API_KEY)
+    resp = exa.get_contents(urls=["https://example.com"], text=True, livecrawl="never")
+    assert resp.results
+
+
+@pytest.mark.asyncio
+@pytest.mark.skipif(not _have_real_key(), reason="EXA_API_KEY not provided")
+async def test_get_contents_async_live():
+    ax = AsyncExa(API_KEY)
+    resp = await ax.get_contents(urls=["https://example.com"], text=True, livecrawl="never")
+    assert resp.results
+
+
+# researchTask endpoint is still beta; mark as xfail if 404 returned
+@pytest.mark.skipif(not _have_real_key(), reason="EXA_API_KEY not provided")
+@pytest.mark.xfail(strict=False)
+def test_research_task_live():
+    exa = Exa(API_KEY)
+    schema = {"type": "object", "properties": {"answer": {"type": "string"}}, "required": ["answer"]}
+    resp = exa.researchTask(input_instructions="Return the string 'pong'", output_schema=schema)
+    assert resp.id