exa-py 1.14.3__py3-none-any.whl → 1.14.5__py3-none-any.whl

exa_py/api.py

@@ -166,6 +166,7 @@ CONTENTS_OPTIONS_TYPES = {
     "text": [dict, bool],
     "highlights": [dict, bool],
     "summary": [dict, bool],
+    "context": [dict, bool],
     "metadata": [dict, bool],
     "livecrawl_timeout": [int],
     "livecrawl": [LIVECRAWL_OPTIONS],
@@ -292,6 +293,16 @@ class SummaryContentsOptions(TypedDict, total=False):
     schema: JSONSchema
 
 
+class ContextContentsOptions(TypedDict, total=False):
+    """Options for retrieving aggregated context from a set of search results.
+
+    Attributes:
+        max_characters (int): The maximum number of characters to include in the context string.
+    """
+
+    max_characters: int
+
+
 class ExtrasOptions(TypedDict, total=False):
     """A class representing additional extraction fields (e.g. links, images)"""
 
@@ -788,6 +799,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]):
@@ -798,16 +818,23 @@ class SearchResponse(Generic[T]):
         autoprompt_string (str, optional): The Exa query created by autoprompt.
         resolved_search_type (str, optional): 'neural' or 'keyword' if auto.
         auto_date (str, optional): A date for filtering if autoprompt found one.
+        context (str, optional): Combined context string when requested via contents.context.
+        statuses (List[ContentStatus], optional): Status list from get_contents.
+        cost_dollars (CostDollars, optional): Cost breakdown.
     """
 
     results: List[T]
     autoprompt_string: Optional[str]
     resolved_search_type: Optional[str]
     auto_date: Optional[str]
+    context: Optional[str] = None
+    statuses: Optional[List[ContentStatus]] = None
     cost_dollars: Optional[CostDollars] = None
 
     def __str__(self):
         output = "\n\n".join(str(result) for result in self.results)
+        if self.context:
+            output += f"\nContext: {self.context}"
         if self.autoprompt_string:
             output += f"\n\nAutoprompt String: {self.autoprompt_string}"
         if self.resolved_search_type:
@@ -818,6 +845,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
 
 
@@ -1242,6 +1271,7 @@ class Exa:
                 "text",
                 "highlights",
                 "summary",
+                "context",
                 "subpages",
                 "subpage_target",
                 "livecrawl",
@@ -1258,6 +1288,7 @@ class Exa:
             data["autopromptString"] if "autopromptString" in data else None,
             data["resolvedSearchType"] if "resolvedSearchType" in data else None,
             data["autoDate"] if "autoDate" in data else None,
+            context=data.get("context"),
             cost_dollars=cost_dollars,
         )
 
@@ -1405,12 +1436,15 @@ class Exa:
         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"),
+            context=data.get("context"),
             cost_dollars=cost_dollars,
+            statuses=statuses,
         )
 
     def find_similar(
@@ -1700,6 +1734,7 @@ class Exa:
                 "text",
                 "highlights",
                 "summary",
+                "context",
                 "subpages",
                 "subpage_target",
                 "livecrawl",
@@ -1716,6 +1751,7 @@ class Exa:
             data.get("autopromptString"),
             data.get("resolvedSearchType"),
             data.get("autoDate"),
+            context=data.get("context"),
             cost_dollars=cost_dollars,
         )
 
@@ -2052,6 +2088,7 @@ class AsyncExa(Exa):
                 "text",
                 "highlights",
                 "summary",
+                "context",
                 "subpages",
                 "subpage_target",
                 "livecrawl",
@@ -2068,6 +2105,7 @@ class AsyncExa(Exa):
             data["autopromptString"] if "autopromptString" in data else None,
             data["resolvedSearchType"] if "resolvedSearchType" in data else None,
             data["autoDate"] if "autoDate" in data else None,
+            context=data.get("context"),
             cost_dollars=cost_dollars,
         )
 
@@ -2092,12 +2130,15 @@ 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"),
+            context=data.get("context"),
             cost_dollars=cost_dollars,
+            statuses=statuses,
         )
 
     async def find_similar(
@@ -2175,6 +2216,7 @@ class AsyncExa(Exa):
                 "text",
                 "highlights",
                 "summary",
+                "context",
                 "subpages",
                 "subpage_target",
                 "livecrawl",
@@ -2191,6 +2233,7 @@ class AsyncExa(Exa):
             data.get("autopromptString"),
             data.get("resolvedSearchType"),
             data.get("autoDate"),
+            context=data.get("context"),
             cost_dollars=cost_dollars,
         )
 

{exa_py-1.14.3.dist-info → exa_py-1.14.5.dist-info}/METADATA

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

{exa_py-1.14.3.dist-info → exa_py-1.14.5.dist-info}/RECORD

@@ -1,5 +1,5 @@
 exa_py/__init__.py,sha256=M2GC9oSdoV6m2msboW0vMWWl8wrth4o6gmEV4MYLGG8,66
-exa_py/api.py,sha256=HEv52SYYtG6f4FkJb9wGhNDaknZkcgxp0Fr7Kw2BrPc,84916
+exa_py/api.py,sha256=ez32FoYovsujjMKmnY0eArgilVgG3EpAVZycOdN8Z8w,86443
 exa_py/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 exa_py/utils.py,sha256=Rc1FJjoR9LQ7L_OJM91Sd1GNkbHjcLyEvJENhRix6gc,2405
 exa_py/research/__init__.py,sha256=QeY-j6bP4QP5tF9ytX0IeQhJvd0Wn4cJCD69U8pP7kA,271
@@ -22,7 +22,7 @@ exa_py/websets/searches/__init__.py,sha256=_0Zx8ES5fFTEL3T8mhLxq_xK2t0JONx6ad6At
 exa_py/websets/searches/client.py,sha256=X3f7axWGfecmxf-2tBTX0Yf_--xToz1X8ZHbbudEzy0,1790
 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.3.dist-info/METADATA,sha256=-UJmHBKQXJq-uc_KalQQ4EinqRuRGvh2eu2gI35nukE,3835
-exa_py-1.14.3.dist-info/WHEEL,sha256=0CuiUZ_p9E4cD6NyLD6UG80LBXYyiSYZOKDm5lp32xk,91
-exa_py-1.14.3.dist-info/top_level.txt,sha256=Mfkmscdw9HWR1PtVhU1gAiVo6DHu_tyiVdb89gfZBVI,7
-exa_py-1.14.3.dist-info/RECORD,,
+exa_py-1.14.5.dist-info/METADATA,sha256=PFRpXGH3t0yYznx9VIX2FHATR9AJ6MbCr251jMxFddo,3835
+exa_py-1.14.5.dist-info/WHEEL,sha256=0CuiUZ_p9E4cD6NyLD6UG80LBXYyiSYZOKDm5lp32xk,91
+exa_py-1.14.5.dist-info/top_level.txt,sha256=Mfkmscdw9HWR1PtVhU1gAiVo6DHu_tyiVdb89gfZBVI,7
+exa_py-1.14.5.dist-info/RECORD,,