exa-py 1.0.10__py3-none-any.whl → 1.0.12__py3-none-any.whl

exa_py/api.py

@@ -20,17 +20,8 @@ from typing_extensions import TypedDict
 
 import httpx
 from openai import NOT_GIVEN, NotGiven, OpenAI
-from openai.types.chat.chat_completion_stream_options_param import (
-    ChatCompletionStreamOptionsParam,
-)
 from openai.types.chat.chat_completion_message_param import ChatCompletionMessageParam
 from openai.types.chat_model import ChatModel
-from openai.types.chat.chat_completion_tool_choice_option_param import (
-    ChatCompletionToolChoiceOptionParam,
-)
-from openai.types.chat.chat_completion_tool_param import ChatCompletionToolParam
-from openai._types import Headers, Query, Body
-from openai.types.chat import completion_create_params
 from exa_py.utils import (
     ExaOpenAICompletion,
     add_message_to_messages,
@@ -344,7 +335,7 @@ class Exa:
         self,
         api_key: Optional[str],
         base_url: str = "https://api.exa.ai",
-        user_agent: str = "exa-py 1.0.10",
+        user_agent: str = "exa-py 1.0.12",
     ):
         """Initialize the Exa client with the provided API key and optional base URL and user agent.
 
@@ -692,38 +683,6 @@ class Exa:
             # Mandatory OpenAI args
             messages: Iterable[ChatCompletionMessageParam],
             model: Union[str, ChatModel],
-            # Optional OpenAI args
-            frequency_penalty: Optional[float] | NotGiven = NOT_GIVEN,
-            function_call: completion_create_params.FunctionCall | NotGiven = NOT_GIVEN,
-            functions: (
-                Iterable[completion_create_params.Function] | NotGiven
-            ) = NOT_GIVEN,
-            logit_bias: Optional[Dict[str, int]] | NotGiven = NOT_GIVEN,
-            logprobs: Optional[bool] | NotGiven = NOT_GIVEN,
-            max_tokens: Optional[int] | NotGiven = NOT_GIVEN,
-            n: Optional[int] | NotGiven = NOT_GIVEN,
-            presence_penalty: Optional[float] | NotGiven = NOT_GIVEN,
-            response_format: (
-                completion_create_params.ResponseFormat | NotGiven
-            ) = NOT_GIVEN,
-            seed: Optional[int] | NotGiven = NOT_GIVEN,
-            stop: Union[Optional[str], List[str]] | NotGiven = NOT_GIVEN,
-            stream: Optional[Literal[False]] | NotGiven = NOT_GIVEN,
-            stream_options: (
-                Optional[ChatCompletionStreamOptionsParam] | NotGiven
-            ) = NOT_GIVEN,
-            temperature: Optional[float] | NotGiven = NOT_GIVEN,
-            tool_choice: ChatCompletionToolChoiceOptionParam | NotGiven = NOT_GIVEN,
-            tools: Iterable[ChatCompletionToolParam] | NotGiven = NOT_GIVEN,
-            top_logprobs: Optional[int] | NotGiven = NOT_GIVEN,
-            top_p: Optional[float] | NotGiven = NOT_GIVEN,
-            user: str | NotGiven = NOT_GIVEN,
-            # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
-            # The extra values given here take precedence over values defined on the client or passed to this method.
-            extra_headers: Headers | None = None,
-            extra_query: Query | None = None,
-            extra_body: Body | None = None,
-            timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
             # Exa args
             use_exa: Optional[Literal["required", "none", "auto"]] = "auto",
             highlights: Union[HighlightsContentsOptions, Literal[True], None] = None,
@@ -738,6 +697,8 @@ class Exa:
             type: Optional[str] = None,
             category: Optional[str] = None,
             result_max_len: int = 2048,
+            # OpenAI args
+            **openai_kwargs,
         ):
             exa_kwargs = {
                 "num_results": num_results,
@@ -755,33 +716,11 @@ class Exa:
 
             create_kwargs = {
                 "model": model,
-                "frequency_penalty": frequency_penalty,
-                "function_call": function_call,
-                "functions": functions,
-                "logit_bias": logit_bias,
-                "logprobs": logprobs,
-                "max_tokens": max_tokens,
-                "n": n,
-                "presence_penalty": presence_penalty,
-                "response_format": response_format,
-                "seed": seed,
-                "stop": stop,
-                "stream": stream,
-                "stream_options": stream_options,
-                "temperature": temperature,
-                "tool_choice": tool_choice,
-                "tools": tools,
-                "top_logprobs": top_logprobs,
-                "top_p": top_p,
-                "user": user,
-                "extra_headers": extra_headers,
-                "extra_query": extra_query,
-                "extra_body": extra_body,
-                "timeout": timeout,
+                **openai_kwargs,
             }
 
             if use_exa != "none":
-                assert tools is NOT_GIVEN, "Tool use is not supported with Exa"
+                assert "tools" not in create_kwargs, "Tool use is not supported with Exa"
                 create_kwargs["tool_choice"] = use_exa
 
             return self._create_with_tool(

{exa_py-1.0.10.dist-info → exa_py-1.0.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: exa-py
-Version: 1.0.10
+Version: 1.0.12
 Summary: Python SDK for Exa API.
 Home-page: https://github.com/exa-labs/exa-py
 Author: Exa
@@ -19,8 +19,6 @@ Description-Content-Type: text/markdown
 Requires-Dist: requests
 Requires-Dist: typing-extensions
 Requires-Dist: openai
-Provides-Extra: openai
-Requires-Dist: openai ; extra == 'openai'
 
 # Exa
 

exa_py-1.0.12.dist-info/RECORD

@@ -0,0 +1,8 @@
+exa_py/__init__.py,sha256=aVF1zB_UV3dagJ5Vn2WrdcInzibdIW61M89sjwRCU_g,29
+exa_py/api.py,sha256=fmPCOzC31meRaBYt2V3skZZDg2xiQaWt60G1Zs5DIHY,27839
+exa_py/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+exa_py/utils.py,sha256=pF91TKGGg4SbSVrdWroQze90-D9AlvqLoKWlHIeyLaE,2402
+exa_py-1.0.12.dist-info/METADATA,sha256=rUIwnAb5tx5U4e9ue8QguSsEZhriJNIJNWxTVEVkSM0,3072
+exa_py-1.0.12.dist-info/WHEEL,sha256=pkctZYzUS4AYVn6dJ-7367OJZivF2e8RA9b_ZBjif18,92
+exa_py-1.0.12.dist-info/top_level.txt,sha256=Mfkmscdw9HWR1PtVhU1gAiVo6DHu_tyiVdb89gfZBVI,7
+exa_py-1.0.12.dist-info/RECORD,,

exa_py-1.0.10.dist-info/RECORD

@@ -1,8 +0,0 @@
-exa_py/__init__.py,sha256=aVF1zB_UV3dagJ5Vn2WrdcInzibdIW61M89sjwRCU_g,29
-exa_py/api.py,sha256=FdGJG46Ix8TvStF-mmuJ-QMtXzvqPRQcLnOogHj24Uc,31032
-exa_py/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-exa_py/utils.py,sha256=pF91TKGGg4SbSVrdWroQze90-D9AlvqLoKWlHIeyLaE,2402
-exa_py-1.0.10.dist-info/METADATA,sha256=NOvPj8ZfTUiMFVXufUSXz5EySt2zmYhBfnXsaBqdvj4,3137
-exa_py-1.0.10.dist-info/WHEEL,sha256=pkctZYzUS4AYVn6dJ-7367OJZivF2e8RA9b_ZBjif18,92
-exa_py-1.0.10.dist-info/top_level.txt,sha256=Mfkmscdw9HWR1PtVhU1gAiVo6DHu_tyiVdb89gfZBVI,7
-exa_py-1.0.10.dist-info/RECORD,,