exa-py 1.8.7__py3-none-any.whl → 1.8.9__py3-none-any.whl

exa_py/api.py

@@ -1,37 +1,39 @@
 from __future__ import annotations
-from dataclasses import dataclass
+
 import dataclasses
-from functools import wraps
+import json
+import os
 import re
-import requests
+from dataclasses import dataclass
+from functools import wraps
 from typing import (
     Callable,
+    Dict,
+    Generic,
     Iterable,
+    Iterator,
     List,
+    Literal,
     Optional,
-    Dict,
-    Generic,
     TypeVar,
-    overload,
     Union,
-    Literal,
-    get_origin,
     get_args,
-    Iterator,
+    get_origin,
+    overload,
 )
-from typing_extensions import TypedDict
-import json
 
+import requests
 from openai import OpenAI
 from openai.types.chat.chat_completion_message_param import ChatCompletionMessageParam
 from openai.types.chat_model import ChatModel
+from typing_extensions import TypedDict
+
 from exa_py.utils import (
     ExaOpenAICompletion,
     add_message_to_messages,
     format_exa_result,
     maybe_get_query,
 )
-import os
 
 is_beta = os.getenv("IS_BETA") == "True"
 
@@ -102,17 +104,29 @@ def to_snake_case(data: dict) -> dict:
 SEARCH_OPTIONS_TYPES = {
     "query": [str],  # The query string.
     "num_results": [int],  # Number of results (Default: 10, Max for basic: 10).
-    "include_domains": [list],  # Domains to search from; exclusive with 'exclude_domains'.
+    "include_domains": [
+        list
+    ],  # Domains to search from; exclusive with 'exclude_domains'.
     "exclude_domains": [list],  # Domains to omit; exclusive with 'include_domains'.
     "start_crawl_date": [str],  # Results after this crawl date. ISO 8601 format.
     "end_crawl_date": [str],  # Results before this crawl date. ISO 8601 format.
-    "start_published_date": [str],  # Results after this publish date; excludes links with no date. ISO 8601 format.
-    "end_published_date": [str],  # Results before this publish date; excludes links with no date. ISO 8601 format.
-    "include_text": [list],  # Must be present in webpage text. (One string, up to 5 words)
-    "exclude_text": [list],  # Must not be present in webpage text. (One string, up to 5 words)
+    "start_published_date": [
+        str
+    ],  # Results after this publish date; excludes links with no date. ISO 8601 format.
+    "end_published_date": [
+        str
+    ],  # Results before this publish date; excludes links with no date. ISO 8601 format.
+    "include_text": [
+        list
+    ],  # Must be present in webpage text. (One string, up to 5 words)
+    "exclude_text": [
+        list
+    ],  # Must not be present in webpage text. (One string, up to 5 words)
     "use_autoprompt": [bool],  # Convert query to Exa. (Default: false)
     "type": [str],  # 'keyword', 'neural', or 'auto' (Default: auto)
-    "category": [str],  # A data category to focus on: 'company', 'research paper', 'news', 'pdf', 'github', 'tweet', 'personal site', 'linkedin profile', 'financial report'
+    "category": [
+        str
+    ],  # A data category to focus on: 'company', 'research paper', 'news', 'pdf', 'github', 'tweet', 'personal site', 'linkedin profile', 'financial report'
     "flags": [list],  # Experimental flags array for Exa usage.
     "moderation": [bool],  # If true, moderate search results for safety.
 }
@@ -188,6 +202,25 @@ def is_valid_type(value, expected_type):
     return False  # For any other case
 
 
+def parse_cost_dollars(raw: dict) -> Optional[CostDollars]:
+    """
+    Parse the costDollars JSON into a CostDollars object, or return None if missing/invalid.
+    """
+    if not raw:
+        return None
+
+    total = raw.get("total")
+    if total is None:
+        # If there's no total, treat as absent
+        return None
+
+    # search and contents can be dictionaries or None
+    search_part = raw.get("search")
+    contents_part = raw.get("contents")
+
+    return CostDollars(total=total, search=search_part, contents=contents_part)
+
+
 class TextContentsOptions(TypedDict, total=False):
     """A class representing the options that you can specify when requesting text
 
@@ -231,6 +264,30 @@ class ExtrasOptions(TypedDict, total=False):
     image_links: int
 
 
+class CostDollarsSearch(TypedDict, total=False):
+    """Represents the cost breakdown for search."""
+
+    neural: float
+    keyword: float
+
+
+class CostDollarsContents(TypedDict, total=False):
+    """Represents the cost breakdown for contents."""
+
+    text: float
+    highlights: float
+    summary: float
+
+
+@dataclass
+class CostDollars:
+    """Represents costDollars field in the API response."""
+
+    total: float
+    search: CostDollarsSearch = None
+    contents: CostDollarsContents = None
+
+
 @dataclass
 class _Result:
     """A class representing the base fields of a search result.
@@ -515,7 +572,8 @@ class AnswerResult:
         author (str, optional): If available, the author of the content.
         text (str, optional): The full page text from each search result.
     """
-    id: str 
+
+    id: str
     url: str
     title: Optional[str] = None
     published_date: Optional[str] = None
@@ -523,12 +581,12 @@ class AnswerResult:
     text: Optional[str] = None
 
     def __init__(self, **kwargs):
-        self.id = kwargs['id']
-        self.url = kwargs['url']
-        self.title = kwargs.get('title')    
-        self.published_date = kwargs.get('published_date')
-        self.author = kwargs.get('author')  
-        self.text = kwargs.get('text')
+        self.id = kwargs["id"]
+        self.url = kwargs["url"]
+        self.title = kwargs.get("title")
+        self.published_date = kwargs.get("published_date")
+        self.author = kwargs.get("author")
+        self.text = kwargs.get("text")
 
     def __str__(self):
         return (
@@ -539,18 +597,20 @@ class AnswerResult:
             f"Author: {self.author}\n"
             f"Text: {self.text}\n\n"
         )
-    
+
+
 @dataclass
 class StreamChunk:
     """A class representing a single chunk of streaming data.
-    
+
     Attributes:
         content (Optional[str]): The partial text content of the answer
         citations (Optional[List[AnswerResult]]): List of citations if provided in this chunk
     """
+
     content: Optional[str] = None
     citations: Optional[List[AnswerResult]] = None
-    
+
     def has_data(self) -> bool:
         """Check if this chunk contains any data."""
         return self.content is not None or self.citations is not None
@@ -594,6 +654,7 @@ class AnswerResponse:
 
 class StreamAnswerResponse:
     """A class representing a streaming answer response."""
+
     def __init__(self, raw_response: requests.Response):
         self._raw_response = raw_response
         self._ensure_ok_status()
@@ -621,8 +682,14 @@ class StreamAnswerResponse:
                 if "delta" in chunk["choices"][0]:
                     content = chunk["choices"][0]["delta"].get("content")
 
-            if "citations" in chunk and chunk["citations"] and chunk["citations"] != "null":
-                citations = [AnswerResult(**to_snake_case(s)) for s in chunk["citations"]]
+            if (
+                "citations" in chunk
+                and chunk["citations"]
+                and chunk["citations"] != "null"
+            ):
+                citations = [
+                    AnswerResult(**to_snake_case(s)) for s in chunk["citations"]
+                ]
 
             stream_chunk = StreamChunk(content=content, citations=citations)
             if stream_chunk.has_data():
@@ -651,6 +718,7 @@ class SearchResponse(Generic[T]):
     autoprompt_string: Optional[str]
     resolved_search_type: Optional[str]
     auto_date: Optional[str]
+    cost_dollars: Optional[CostDollars] = None
 
     def __str__(self):
         output = "\n\n".join(str(result) for result in self.results)
@@ -658,7 +726,12 @@ class SearchResponse(Generic[T]):
             output += f"\n\nAutoprompt String: {self.autoprompt_string}"
         if self.resolved_search_type:
             output += f"\nResolved Search Type: {self.resolved_search_type}"
-
+        if self.cost_dollars:
+            output += f"\nCostDollars: total={self.cost_dollars.total}"
+            if self.cost_dollars.search:
+                output += f"\n  - search: {self.cost_dollars.search}"
+            if self.cost_dollars.contents:
+                output += f"\n  - contents: {self.cost_dollars.contents}"
         return output
 
 
@@ -686,7 +759,7 @@ class Exa:
         self,
         api_key: Optional[str],
         base_url: str = "https://api.exa.ai",
-        user_agent: str = "exa-py 1.8.7",
+        user_agent: str = "exa-py 1.8.9",
     ):
         """Initialize the Exa client with the provided API key and optional base URL and user agent.
 
@@ -720,12 +793,16 @@ class Exa:
             ValueError: If the request fails (non-200 status code).
         """
         if data.get("stream"):
-            res = requests.post(self.base_url + endpoint, json=data, headers=self.headers, stream=True)
+            res = requests.post(
+                self.base_url + endpoint, json=data, headers=self.headers, stream=True
+            )
             return res
 
         res = requests.post(self.base_url + endpoint, json=data, headers=self.headers)
         if res.status_code != 200:
-            raise ValueError(f"Request failed with status code {res.status_code}: {res.text}")
+            raise ValueError(
+                f"Request failed with status code {res.status_code}: {res.text}"
+            )
         return res.json()
 
     def search(
@@ -773,11 +850,13 @@ class Exa:
         validate_search_options(options, SEARCH_OPTIONS_TYPES)
         options = to_camel_case(options)
         data = self.request("/search", options)
+        cost_dollars = parse_cost_dollars(data.get("costDollars"))
         return SearchResponse(
             [Result(**to_snake_case(result)) for result in data["results"]],
             data["autopromptString"] if "autopromptString" in data else None,
             data["resolvedSearchType"] if "resolvedSearchType" in data else None,
             data["autoDate"] if "autoDate" in data else None,
+            cost_dollars=cost_dollars,
         )
 
     @overload
@@ -805,8 +884,7 @@ class Exa:
         subpages: Optional[int] = None,
         subpage_target: Optional[Union[str, List[str]]] = None,
         extras: Optional[ExtrasOptions] = None,
-    ) -> SearchResponse[ResultWithText]:
-        ...
+    ) -> SearchResponse[ResultWithText]: ...
 
     @overload
     def search_and_contents(
@@ -826,14 +904,15 @@ class Exa:
         use_autoprompt: Optional[bool] = None,
         type: Optional[str] = None,
         category: Optional[str] = None,
+        flags: Optional[List[str]] = None,
+        moderation: Optional[bool] = None,
         subpages: Optional[int] = None,
         livecrawl_timeout: Optional[int] = None,
         livecrawl: Optional[LIVECRAWL_OPTIONS] = None,
         filter_empty_results: Optional[bool] = None,
         subpage_target: Optional[Union[str, List[str]]] = None,
         extras: Optional[ExtrasOptions] = None,
-    ) -> SearchResponse[ResultWithText]:
-        ...
+    ) -> SearchResponse[ResultWithText]: ...
 
     @overload
     def search_and_contents(
@@ -861,8 +940,7 @@ class Exa:
         livecrawl: Optional[LIVECRAWL_OPTIONS] = None,
         filter_empty_results: Optional[bool] = None,
         extras: Optional[ExtrasOptions] = None,
-    ) -> SearchResponse[ResultWithHighlights]:
-        ...
+    ) -> SearchResponse[ResultWithHighlights]: ...
 
     @overload
     def search_and_contents(
@@ -891,8 +969,7 @@ class Exa:
         livecrawl: Optional[LIVECRAWL_OPTIONS] = None,
         filter_empty_results: Optional[bool] = None,
         extras: Optional[ExtrasOptions] = None,
-    ) -> SearchResponse[ResultWithTextAndHighlights]:
-        ...
+    ) -> SearchResponse[ResultWithTextAndHighlights]: ...
 
     @overload
     def search_and_contents(
@@ -920,8 +997,7 @@ class Exa:
         livecrawl: Optional[LIVECRAWL_OPTIONS] = None,
         filter_empty_results: Optional[bool] = None,
         extras: Optional[ExtrasOptions] = None,
-    ) -> SearchResponse[ResultWithSummary]:
-        ...
+    ) -> SearchResponse[ResultWithSummary]: ...
 
     @overload
     def search_and_contents(
@@ -950,8 +1026,7 @@ class Exa:
         livecrawl: Optional[LIVECRAWL_OPTIONS] = None,
         filter_empty_results: Optional[bool] = None,
         extras: Optional[ExtrasOptions] = None,
-    ) -> SearchResponse[ResultWithTextAndSummary]:
-        ...
+    ) -> SearchResponse[ResultWithTextAndSummary]: ...
 
     @overload
     def search_and_contents(
@@ -974,12 +1049,13 @@ class Exa:
         category: Optional[str] = None,
         subpages: Optional[int] = None,
         subpage_target: Optional[Union[str, List[str]]] = None,
+        flags: Optional[List[str]] = None,
+        moderation: Optional[bool] = None,
         livecrawl_timeout: Optional[int] = None,
         livecrawl: Optional[LIVECRAWL_OPTIONS] = None,
         filter_empty_results: Optional[bool] = None,
         extras: Optional[ExtrasOptions] = None,
-    ) -> SearchResponse[ResultWithHighlightsAndSummary]:
-        ...
+    ) -> SearchResponse[ResultWithHighlightsAndSummary]: ...
 
     @overload
     def search_and_contents(
@@ -1002,14 +1078,14 @@ class Exa:
         type: Optional[str] = None,
         category: Optional[str] = None,
         flags: Optional[List[str]] = None,
+        moderation: Optional[bool] = None,
         livecrawl_timeout: Optional[int] = None,
         livecrawl: Optional[LIVECRAWL_OPTIONS] = None,
-        filter_empty_results: Optional[bool] = None,
         subpages: Optional[int] = None,
         subpage_target: Optional[Union[str, List[str]]] = None,
+        filter_empty_results: Optional[bool] = None,
         extras: Optional[ExtrasOptions] = None,
-    ) -> SearchResponse[ResultWithTextAndHighlightsAndSummary]:
-        ...
+    ) -> SearchResponse[ResultWithTextAndHighlightsAndSummary]: ...
 
     def search_and_contents(self, query: str, **kwargs):
         options = {k: v for k, v in {"query": query, **kwargs}.items() if v is not None}
@@ -1048,11 +1124,13 @@ class Exa:
         )
         options = to_camel_case(options)
         data = self.request("/search", options)
+        cost_dollars = parse_cost_dollars(data.get("costDollars"))
         return SearchResponse(
             [Result(**to_snake_case(result)) for result in data["results"]],
             data["autopromptString"] if "autopromptString" in data else None,
             data["resolvedSearchType"] if "resolvedSearchType" in data else None,
             data["autoDate"] if "autoDate" in data else None,
+            cost_dollars=cost_dollars,
         )
 
     @overload
@@ -1066,8 +1144,7 @@ class Exa:
         subpage_target: Optional[Union[str, List[str]]] = None,
         extras: Optional[ExtrasOptions] = None,
         flags: Optional[List[str]] = None,
-    ) -> SearchResponse[ResultWithText]:
-        ...
+    ) -> SearchResponse[ResultWithText]: ...
 
     @overload
     def get_contents(
@@ -1082,8 +1159,7 @@ class Exa:
         subpage_target: Optional[Union[str, List[str]]] = None,
         extras: Optional[ExtrasOptions] = None,
         flags: Optional[List[str]] = None,
-    ) -> SearchResponse[ResultWithText]:
-        ...
+    ) -> SearchResponse[ResultWithText]: ...
 
     @overload
     def get_contents(
@@ -1098,8 +1174,7 @@ class Exa:
         subpage_target: Optional[Union[str, List[str]]] = None,
         extras: Optional[ExtrasOptions] = None,
         flags: Optional[List[str]] = None,
-    ) -> SearchResponse[ResultWithHighlights]:
-        ...
+    ) -> SearchResponse[ResultWithHighlights]: ...
 
     @overload
     def get_contents(
@@ -1115,8 +1190,7 @@ class Exa:
         subpage_target: Optional[Union[str, List[str]]] = None,
         extras: Optional[ExtrasOptions] = None,
         flags: Optional[List[str]] = None,
-    ) -> SearchResponse[ResultWithTextAndHighlights]:
-        ...
+    ) -> SearchResponse[ResultWithTextAndHighlights]: ...
 
     @overload
     def get_contents(
@@ -1131,8 +1205,7 @@ class Exa:
         subpage_target: Optional[Union[str, List[str]]] = None,
         extras: Optional[ExtrasOptions] = None,
         flags: Optional[List[str]] = None,
-    ) -> SearchResponse[ResultWithSummary]:
-        ...
+    ) -> SearchResponse[ResultWithSummary]: ...
 
     @overload
     def get_contents(
@@ -1148,8 +1221,7 @@ class Exa:
         subpage_target: Optional[Union[str, List[str]]] = None,
         extras: Optional[ExtrasOptions] = None,
         flags: Optional[List[str]] = None,
-    ) -> SearchResponse[ResultWithTextAndSummary]:
-        ...
+    ) -> SearchResponse[ResultWithTextAndSummary]: ...
 
     @overload
     def get_contents(
@@ -1165,8 +1237,7 @@ class Exa:
         subpage_target: Optional[Union[str, List[str]]] = None,
         extras: Optional[ExtrasOptions] = None,
         flags: Optional[List[str]] = None,
-    ) -> SearchResponse[ResultWithHighlightsAndSummary]:
-        ...
+    ) -> SearchResponse[ResultWithHighlightsAndSummary]: ...
 
     @overload
     def get_contents(
@@ -1183,8 +1254,7 @@ class Exa:
         subpage_target: Optional[Union[str, List[str]]] = None,
         extras: Optional[ExtrasOptions] = None,
         flags: Optional[List[str]] = None,
-    ) -> SearchResponse[ResultWithTextAndHighlightsAndSummary]:
-        ...
+    ) -> SearchResponse[ResultWithTextAndHighlightsAndSummary]: ...
 
     def get_contents(self, urls: Union[str, List[str], List[_Result]], **kwargs):
         options = {
@@ -1206,11 +1276,13 @@ class Exa:
         )
         options = to_camel_case(options)
         data = self.request("/contents", options)
+        cost_dollars = parse_cost_dollars(data.get("costDollars"))
         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,
         )
 
     def find_similar(
@@ -1254,11 +1326,13 @@ class Exa:
         validate_search_options(options, FIND_SIMILAR_OPTIONS_TYPES)
         options = to_camel_case(options)
         data = self.request("/findSimilar", options)
+        cost_dollars = parse_cost_dollars(data.get("costDollars"))
         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,
         )
 
     @overload
@@ -1284,8 +1358,7 @@ class Exa:
         subpages: Optional[int] = None,
         subpage_target: Optional[Union[str, List[str]]] = None,
         extras: Optional[ExtrasOptions] = None,
-    ) -> SearchResponse[ResultWithText]:
-        ...
+    ) -> SearchResponse[ResultWithText]: ...
 
     @overload
     def find_similar_and_contents(
@@ -1311,8 +1384,7 @@ class Exa:
         subpages: Optional[int] = None,
         subpage_target: Optional[Union[str, List[str]]] = None,
         extras: Optional[ExtrasOptions] = None,
-    ) -> SearchResponse[ResultWithText]:
-        ...
+    ) -> SearchResponse[ResultWithText]: ...
 
     @overload
     def find_similar_and_contents(
@@ -1338,8 +1410,7 @@ class Exa:
         livecrawl: Optional[LIVECRAWL_OPTIONS] = None,
         filter_empty_results: Optional[bool] = None,
         extras: Optional[ExtrasOptions] = None,
-    ) -> SearchResponse[ResultWithHighlights]:
-        ...
+    ) -> SearchResponse[ResultWithHighlights]: ...
 
     @overload
     def find_similar_and_contents(
@@ -1366,8 +1437,7 @@ class Exa:
         subpages: Optional[int] = None,
         subpage_target: Optional[Union[str, List[str]]] = None,
         extras: Optional[ExtrasOptions] = None,
-    ) -> SearchResponse[ResultWithTextAndHighlights]:
-        ...
+    ) -> SearchResponse[ResultWithTextAndHighlights]: ...
 
     @overload
     def find_similar_and_contents(
@@ -1393,8 +1463,7 @@ class Exa:
         subpages: Optional[int] = None,
         subpage_target: Optional[Union[str, List[str]]] = None,
         extras: Optional[ExtrasOptions] = None,
-    ) -> SearchResponse[ResultWithSummary]:
-        ...
+    ) -> SearchResponse[ResultWithSummary]: ...
 
     @overload
     def find_similar_and_contents(
@@ -1421,8 +1490,7 @@ class Exa:
         subpages: Optional[int] = None,
         subpage_target: Optional[Union[str, List[str]]] = None,
         extras: Optional[ExtrasOptions] = None,
-    ) -> SearchResponse[ResultWithTextAndSummary]:
-        ...
+    ) -> SearchResponse[ResultWithTextAndSummary]: ...
 
     @overload
     def find_similar_and_contents(
@@ -1449,8 +1517,7 @@ class Exa:
         livecrawl: Optional[LIVECRAWL_OPTIONS] = None,
         filter_empty_results: Optional[bool] = None,
         extras: Optional[ExtrasOptions] = None,
-    ) -> SearchResponse[ResultWithHighlightsAndSummary]:
-        ...
+    ) -> SearchResponse[ResultWithHighlightsAndSummary]: ...
 
     @overload
     def find_similar_and_contents(
@@ -1478,8 +1545,7 @@ class Exa:
         subpages: Optional[int] = None,
         subpage_target: Optional[Union[str, List[str]]] = None,
         extras: Optional[ExtrasOptions] = None,
-    ) -> SearchResponse[ResultWithTextAndHighlightsAndSummary]:
-        ...
+    ) -> SearchResponse[ResultWithTextAndHighlightsAndSummary]: ...
 
     def find_similar_and_contents(self, url: str, **kwargs):
         options = {k: v for k, v in {"url": url, **kwargs}.items() if v is not None}
@@ -1516,18 +1582,20 @@ class Exa:
         )
         options = to_camel_case(options)
         data = self.request("/findSimilar", options)
+        cost_dollars = parse_cost_dollars(data.get("costDollars"))
         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,
         )
 
     def wrap(self, client: OpenAI):
         """Wrap an OpenAI client with Exa functionality.
 
-        After wrapping, any call to `client.chat.completions.create` will be intercepted 
-        and enhanced with Exa RAG functionality. To disable Exa for a specific call, 
+        After wrapping, any call to `client.chat.completions.create` will be intercepted
+        and enhanced with Exa RAG functionality. To disable Exa for a specific call,
         set `use_exa="none"` in the `create` method.
 
         Args:
@@ -1656,8 +1724,8 @@ class Exa:
         *,
         stream: Optional[bool] = False,
         text: Optional[bool] = False,
-    ) -> Union[AnswerResponse, StreamAnswerResponse]:
-        ...
+        model: Optional[Literal["exa", "exa-pro"]] = None,
+    ) -> Union[AnswerResponse, StreamAnswerResponse]: ...
 
     def answer(
         self,
@@ -1665,12 +1733,14 @@ class Exa:
         *,
         stream: Optional[bool] = False,
         text: Optional[bool] = False,
+        model: Optional[Literal["exa", "exa-pro"]] = None,
     ) -> Union[AnswerResponse, StreamAnswerResponse]:
         """Generate an answer to a query using Exa's search and LLM capabilities.
 
         Args:
             query (str): The query to answer.
             text (bool, optional): Whether to include full text in the results. Defaults to False.
+            model (str, optional): The model to use for answering. Either "exa" or "exa-pro". Defaults to None.
 
         Returns:
             AnswerResponse: An object containing the answer and citations.
@@ -1684,17 +1754,13 @@ class Exa:
                 "Please use `stream_answer(...)` for streaming."
             )
 
-        options = {
-            k: v
-            for k, v in locals().items()
-            if k != "self" and v is not None
-        }
+        options = {k: v for k, v in locals().items() if k != "self" and v is not None}
         options = to_camel_case(options)
         response = self.request("/answer", options)
 
         return AnswerResponse(
             response["answer"],
-            [AnswerResult(**to_snake_case(result)) for result in response["citations"]]
+            [AnswerResult(**to_snake_case(result)) for result in response["citations"]],
         )
 
     def stream_answer(
@@ -1702,24 +1768,21 @@ class Exa:
         query: str,
         *,
         text: bool = False,
+        model: Optional[Literal["exa", "exa-pro"]] = None,
     ) -> StreamAnswerResponse:
         """Generate a streaming answer response.
 
         Args:
             query (str): The query to answer.
             text (bool): Whether to include full text in the results. Defaults to False.
+            model (str, optional): The model to use for answering. Either "exa" or "exa-pro". Defaults to None.
 
         Returns:
             StreamAnswerResponse: An object that can be iterated over to retrieve (partial text, partial citations).
                 Each iteration yields a tuple of (Optional[str], Optional[List[AnswerResult]]).
         """
-        options = {
-            k: v
-            for k, v in locals().items()
-            if k != "self" and v is not None
-        }
+        options = {k: v for k, v in locals().items() if k != "self" and v is not None}
         options = to_camel_case(options)
         options["stream"] = True
         raw_response = self.request("/answer", options)
         return StreamAnswerResponse(raw_response)
-

{exa_py-1.8.7.dist-info → exa_py-1.8.9.dist-info}/METADATA

@@ -1,20 +1,23 @@
-Metadata-Version: 2.3
-Name: exa-py
-Version: 1.8.7
+Metadata-Version: 2.1
+Name: exa_py
+Version: 1.8.9
 Summary: Python SDK for Exa API.
-Author: Exa AI
+Home-page: https://github.com/exa-labs/exa-py
+Author: Exa
 Author-email: hello@exa.ai
-Requires-Python: >=3.9,<4.0
-Classifier: Programming Language :: Python :: 3
+Classifier: Development Status :: 5 - Production/Stable
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Typing :: Typed
+Classifier: Programming Language :: Python :: 3.8
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
-Classifier: Programming Language :: Python :: 3.13
-Requires-Dist: openai (>=1.48,<2.0)
-Requires-Dist: requests (>=2.32.3,<3.0.0)
-Requires-Dist: typing-extensions (>=4.12.2,<5.0.0)
 Description-Content-Type: text/markdown
+Requires-Dist: requests
+Requires-Dist: typing-extensions
+Requires-Dist: openai >=1.10.0
 
 # Exa
 
@@ -91,8 +94,8 @@ exa = Exa(api_key="your-api-key")
   # basic answer
   response = exa.answer("This is a query to answer a question")
 
-  # answer with full text
-  response = exa.answer("This is a query to answer a question", text=True)
+  # answer with full text, using the exa-pro model (sends 2 expanded quries to exa search)
+  response = exa.answer("This is a query to answer a question", text=True, model="exa-pro")
 
   # answer with streaming
   response = exa.stream_answer("This is a query to answer:")
@@ -103,4 +106,3 @@ exa = Exa(api_key="your-api-key")
 
 ```
 
-

exa_py-1.8.9.dist-info/RECORD

@@ -0,0 +1,8 @@
+exa_py/__init__.py,sha256=1selemczpRm1y8V9cWNm90LARnU1jbtyp-Qpx3c7cTw,28
+exa_py/api.py,sha256=BSZ-uSRYfCyHWBMUpV4SV-2yaWustPS7xc63gArbJSE,65156
+exa_py/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+exa_py/utils.py,sha256=Rc1FJjoR9LQ7L_OJM91Sd1GNkbHjcLyEvJENhRix6gc,2405
+exa_py-1.8.9.dist-info/METADATA,sha256=O_ivBX4PUV2yWQPL69nSmXlqP6S0BTlTYiVMpz4nLAM,3522
+exa_py-1.8.9.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+exa_py-1.8.9.dist-info/top_level.txt,sha256=Mfkmscdw9HWR1PtVhU1gAiVo6DHu_tyiVdb89gfZBVI,7
+exa_py-1.8.9.dist-info/RECORD,,

{exa_py-1.8.7.dist-info → exa_py-1.8.9.dist-info}/WHEEL

@@ -1,4 +1,5 @@
 Wheel-Version: 1.0
-Generator: poetry-core 2.0.1
+Generator: bdist_wheel (0.43.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
+

exa_py-1.8.9.dist-info/top_level.txt

@@ -0,0 +1 @@
+exa_py

exa_py/example.py

@@ -1,11 +0,0 @@
-from exa_py import Exa
-
-exa = Exa(api_key = "995da727-7b91-42d0-b282-326daa290b39")
-
-response = exa.stream_answer(
-    "How close are we to meeting aliens?",
-    text=True
-)
-
-for chunk in response:
-    print(chunk, end='', flush=True)

exa_py-1.8.7.dist-info/RECORD

@@ -1,8 +0,0 @@
-exa_py/__init__.py,sha256=1selemczpRm1y8V9cWNm90LARnU1jbtyp-Qpx3c7cTw,28
-exa_py/api.py,sha256=5uuKVZpLtXQvxEi_X7W_x8JYqTQlsO8R2Ln0IRAQCrk,62663
-exa_py/example.py,sha256=V2uZvFTQFLVr61lVQ_HbZz8G8TFT6Ic44-TTE5ixzBk,235
-exa_py/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-exa_py/utils.py,sha256=Rc1FJjoR9LQ7L_OJM91Sd1GNkbHjcLyEvJENhRix6gc,2405
-exa_py-1.8.7.dist-info/METADATA,sha256=xbby0b9LVLEzzBMu1mGivh0ZO0gDm_xjvPoF219riLk,3337
-exa_py-1.8.7.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-exa_py-1.8.7.dist-info/RECORD,,