relaxai 0.1.0__py3-none-any.whl → 0.17.0__py3-none-any.whl

relaxai/_base_client.py

@@ -532,7 +532,10 @@ class BaseClient(Generic[_HttpxClientT, _DefaultStreamT]):
         is_body_allowed = options.method.lower() != "get"
 
         if is_body_allowed:
-            kwargs["json"] = json_data if is_given(json_data) else None
+            if isinstance(json_data, bytes):
+                kwargs["content"] = json_data
+            else:
+                kwargs["json"] = json_data if is_given(json_data) else None
             kwargs["files"] = files
         else:
             headers.pop("Content-Type", None)

relaxai/_client.py

@@ -12,7 +12,9 @@ from . import _exceptions
 from ._qs import Querystring
 from ._types import (
     NOT_GIVEN,
+    Body,
     Omit,
+    Query,
     Headers,
     Timeout,
     NotGiven,
@@ -22,13 +24,20 @@ from ._types import (
 )
 from ._utils import is_given, get_async_library
 from ._version import __version__
-from .resources import chat, health, models, embeddings
+from ._response import (
+    to_raw_response_wrapper,
+    to_streamed_response_wrapper,
+    async_to_raw_response_wrapper,
+    async_to_streamed_response_wrapper,
+)
+from .resources import chat, models, embeddings
 from ._streaming import Stream as Stream, AsyncStream as AsyncStream
-from ._exceptions import APIStatusError
+from ._exceptions import RelaxaiError, APIStatusError
 from ._base_client import (
     DEFAULT_MAX_RETRIES,
     SyncAPIClient,
     AsyncAPIClient,
+    make_request_options,
 )
 
 __all__ = ["Timeout", "Transport", "ProxiesTypes", "RequestOptions", "Relaxai", "AsyncRelaxai", "Client", "AsyncClient"]
@@ -37,13 +46,12 @@ __all__ = ["Timeout", "Transport", "ProxiesTypes", "RequestOptions", "Relaxai",
 class Relaxai(SyncAPIClient):
     chat: chat.ChatResource
     embeddings: embeddings.EmbeddingsResource
-    health: health.HealthResource
     models: models.ModelsResource
     with_raw_response: RelaxaiWithRawResponse
     with_streaming_response: RelaxaiWithStreamedResponse
 
     # client options
-    api_key: str | None
+    api_key: str
 
     def __init__(
         self,
@@ -74,12 +82,16 @@ class Relaxai(SyncAPIClient):
         """
         if api_key is None:
             api_key = os.environ.get("RELAXAI_API_KEY")
+        if api_key is None:
+            raise RelaxaiError(
+                "The api_key client option must be set either by passing api_key to the client or by setting the RELAXAI_API_KEY environment variable"
+            )
         self.api_key = api_key
 
         if base_url is None:
             base_url = os.environ.get("RELAXAI_BASE_URL")
         if base_url is None:
-            base_url = f"http://127.0.0.1"
+            base_url = f"https://api.relax.ai"
 
         super().__init__(
             version=__version__,
@@ -94,7 +106,6 @@ class Relaxai(SyncAPIClient):
 
         self.chat = chat.ChatResource(self)
         self.embeddings = embeddings.EmbeddingsResource(self)
-        self.health = health.HealthResource(self)
         self.models = models.ModelsResource(self)
         self.with_raw_response = RelaxaiWithRawResponse(self)
         self.with_streaming_response = RelaxaiWithStreamedResponse(self)
@@ -108,8 +119,6 @@ class Relaxai(SyncAPIClient):
     @override
     def auth_headers(self) -> dict[str, str]:
         api_key = self.api_key
-        if api_key is None:
-            return {}
         return {"Authorization": f"Bearer {api_key}"}
 
     @property
@@ -121,17 +130,6 @@ class Relaxai(SyncAPIClient):
             **self._custom_headers,
         }
 
-    @override
-    def _validate_headers(self, headers: Headers, custom_headers: Headers) -> None:
-        if self.api_key and headers.get("Authorization"):
-            return
-        if isinstance(custom_headers.get("Authorization"), Omit):
-            return
-
-        raise TypeError(
-            '"Could not resolve authentication method. Expected the api_key to be set. Or for the `Authorization` headers to be explicitly omitted"'
-        )
-
     def copy(
         self,
         *,
@@ -183,6 +181,25 @@ class Relaxai(SyncAPIClient):
     # client.with_options(timeout=10).foo.create(...)
     with_options = copy
 
+    def health(
+        self,
+        *,
+        # 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,
+    ) -> str:
+        """Check the health of the service."""
+        return self.get(
+            "/v1/health",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=str,
+        )
+
     @override
     def _make_status_error(
         self,
@@ -220,13 +237,12 @@ class Relaxai(SyncAPIClient):
 class AsyncRelaxai(AsyncAPIClient):
     chat: chat.AsyncChatResource
     embeddings: embeddings.AsyncEmbeddingsResource
-    health: health.AsyncHealthResource
     models: models.AsyncModelsResource
     with_raw_response: AsyncRelaxaiWithRawResponse
     with_streaming_response: AsyncRelaxaiWithStreamedResponse
 
     # client options
-    api_key: str | None
+    api_key: str
 
     def __init__(
         self,
@@ -257,12 +273,16 @@ class AsyncRelaxai(AsyncAPIClient):
         """
         if api_key is None:
             api_key = os.environ.get("RELAXAI_API_KEY")
+        if api_key is None:
+            raise RelaxaiError(
+                "The api_key client option must be set either by passing api_key to the client or by setting the RELAXAI_API_KEY environment variable"
+            )
         self.api_key = api_key
 
         if base_url is None:
             base_url = os.environ.get("RELAXAI_BASE_URL")
         if base_url is None:
-            base_url = f"http://127.0.0.1"
+            base_url = f"https://api.relax.ai"
 
         super().__init__(
             version=__version__,
@@ -277,7 +297,6 @@ class AsyncRelaxai(AsyncAPIClient):
 
         self.chat = chat.AsyncChatResource(self)
         self.embeddings = embeddings.AsyncEmbeddingsResource(self)
-        self.health = health.AsyncHealthResource(self)
         self.models = models.AsyncModelsResource(self)
         self.with_raw_response = AsyncRelaxaiWithRawResponse(self)
         self.with_streaming_response = AsyncRelaxaiWithStreamedResponse(self)
@@ -291,8 +310,6 @@ class AsyncRelaxai(AsyncAPIClient):
     @override
     def auth_headers(self) -> dict[str, str]:
         api_key = self.api_key
-        if api_key is None:
-            return {}
         return {"Authorization": f"Bearer {api_key}"}
 
     @property
@@ -304,17 +321,6 @@ class AsyncRelaxai(AsyncAPIClient):
             **self._custom_headers,
         }
 
-    @override
-    def _validate_headers(self, headers: Headers, custom_headers: Headers) -> None:
-        if self.api_key and headers.get("Authorization"):
-            return
-        if isinstance(custom_headers.get("Authorization"), Omit):
-            return
-
-        raise TypeError(
-            '"Could not resolve authentication method. Expected the api_key to be set. Or for the `Authorization` headers to be explicitly omitted"'
-        )
-
     def copy(
         self,
         *,
@@ -366,6 +372,25 @@ class AsyncRelaxai(AsyncAPIClient):
     # client.with_options(timeout=10).foo.create(...)
     with_options = copy
 
+    async def health(
+        self,
+        *,
+        # 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,
+    ) -> str:
+        """Check the health of the service."""
+        return await self.get(
+            "/v1/health",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=str,
+        )
+
     @override
     def _make_status_error(
         self,
@@ -404,33 +429,45 @@ class RelaxaiWithRawResponse:
     def __init__(self, client: Relaxai) -> None:
         self.chat = chat.ChatResourceWithRawResponse(client.chat)
         self.embeddings = embeddings.EmbeddingsResourceWithRawResponse(client.embeddings)
-        self.health = health.HealthResourceWithRawResponse(client.health)
         self.models = models.ModelsResourceWithRawResponse(client.models)
 
+        self.health = to_raw_response_wrapper(
+            client.health,
+        )
+
 
 class AsyncRelaxaiWithRawResponse:
     def __init__(self, client: AsyncRelaxai) -> None:
         self.chat = chat.AsyncChatResourceWithRawResponse(client.chat)
         self.embeddings = embeddings.AsyncEmbeddingsResourceWithRawResponse(client.embeddings)
-        self.health = health.AsyncHealthResourceWithRawResponse(client.health)
         self.models = models.AsyncModelsResourceWithRawResponse(client.models)
 
+        self.health = async_to_raw_response_wrapper(
+            client.health,
+        )
+
 
 class RelaxaiWithStreamedResponse:
     def __init__(self, client: Relaxai) -> None:
         self.chat = chat.ChatResourceWithStreamingResponse(client.chat)
         self.embeddings = embeddings.EmbeddingsResourceWithStreamingResponse(client.embeddings)
-        self.health = health.HealthResourceWithStreamingResponse(client.health)
         self.models = models.ModelsResourceWithStreamingResponse(client.models)
 
+        self.health = to_streamed_response_wrapper(
+            client.health,
+        )
+
 
 class AsyncRelaxaiWithStreamedResponse:
     def __init__(self, client: AsyncRelaxai) -> None:
         self.chat = chat.AsyncChatResourceWithStreamingResponse(client.chat)
         self.embeddings = embeddings.AsyncEmbeddingsResourceWithStreamingResponse(client.embeddings)
-        self.health = health.AsyncHealthResourceWithStreamingResponse(client.health)
         self.models = models.AsyncModelsResourceWithStreamingResponse(client.models)
 
+        self.health = async_to_streamed_response_wrapper(
+            client.health,
+        )
+
 
 Client = Relaxai
 

relaxai/_files.py

@@ -69,12 +69,12 @@ def _transform_file(file: FileTypes) -> HttpxFileTypes:
         return file
 
     if is_tuple_t(file):
-        return (file[0], _read_file_content(file[1]), *file[2:])
+        return (file[0], read_file_content(file[1]), *file[2:])
 
     raise TypeError(f"Expected file types input to be a FileContent type or to be a tuple")
 
 
-def _read_file_content(file: FileContent) -> HttpxFileContent:
+def read_file_content(file: FileContent) -> HttpxFileContent:
     if isinstance(file, os.PathLike):
         return pathlib.Path(file).read_bytes()
     return file
@@ -111,12 +111,12 @@ async def _async_transform_file(file: FileTypes) -> HttpxFileTypes:
         return file
 
     if is_tuple_t(file):
-        return (file[0], await _async_read_file_content(file[1]), *file[2:])
+        return (file[0], await async_read_file_content(file[1]), *file[2:])
 
     raise TypeError(f"Expected file types input to be a FileContent type or to be a tuple")
 
 
-async def _async_read_file_content(file: FileContent) -> HttpxFileContent:
+async def async_read_file_content(file: FileContent) -> HttpxFileContent:
     if isinstance(file, os.PathLike):
         return await anyio.Path(file).read_bytes()
 

relaxai/_version.py

@@ -1,4 +1,4 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 __title__ = "relaxai"
-__version__ = "0.1.0"  # x-release-please-version
+__version__ = "0.17.0"  # x-release-please-version

relaxai/resources/__init__.py

@@ -8,14 +8,6 @@ from .chat import (
     ChatResourceWithStreamingResponse,
     AsyncChatResourceWithStreamingResponse,
 )
-from .health import (
-    HealthResource,
-    AsyncHealthResource,
-    HealthResourceWithRawResponse,
-    AsyncHealthResourceWithRawResponse,
-    HealthResourceWithStreamingResponse,
-    AsyncHealthResourceWithStreamingResponse,
-)
 from .models import (
     ModelsResource,
     AsyncModelsResource,
@@ -46,12 +38,6 @@ __all__ = [
     "AsyncEmbeddingsResourceWithRawResponse",
     "EmbeddingsResourceWithStreamingResponse",
     "AsyncEmbeddingsResourceWithStreamingResponse",
-    "HealthResource",
-    "AsyncHealthResource",
-    "HealthResourceWithRawResponse",
-    "AsyncHealthResourceWithRawResponse",
-    "HealthResourceWithStreamingResponse",
-    "AsyncHealthResourceWithStreamingResponse",
     "ModelsResource",
     "AsyncModelsResource",
     "ModelsResourceWithRawResponse",

relaxai/resources/chat.py

@@ -18,9 +18,10 @@ from .._response import (
     async_to_streamed_response_wrapper,
 )
 from .._base_client import make_request_options
+from ..types.stream_options_param import StreamOptionsParam
+from ..types.chat_completion_response import ChatCompletionResponse
 from ..types.function_definition_param import FunctionDefinitionParam
 from ..types.chat_completion_message_param import ChatCompletionMessageParam
-from ..types.chat_create_completion_response import ChatCreateCompletionResponse
 
 __all__ = ["ChatResource", "AsyncChatResource"]
 
@@ -50,6 +51,7 @@ class ChatResource(SyncAPIResource):
         *,
         messages: Iterable[ChatCompletionMessageParam],
         model: str,
+        query_stream: bool | NotGiven = NOT_GIVEN,
         chat_template_kwargs: object | NotGiven = NOT_GIVEN,
         frequency_penalty: float | NotGiven = NOT_GIVEN,
         function_call: object | NotGiven = NOT_GIVEN,
@@ -68,25 +70,29 @@ class ChatResource(SyncAPIResource):
         seed: int | NotGiven = NOT_GIVEN,
         stop: List[str] | NotGiven = NOT_GIVEN,
         store: bool | NotGiven = NOT_GIVEN,
-        stream: bool | NotGiven = NOT_GIVEN,
-        stream_options: chat_create_completion_params.StreamOptions | NotGiven = NOT_GIVEN,
+        body_stream: bool | NotGiven = NOT_GIVEN,
+        stream_options: StreamOptionsParam | NotGiven = NOT_GIVEN,
         temperature: float | NotGiven = NOT_GIVEN,
         tool_choice: object | NotGiven = NOT_GIVEN,
         tools: Iterable[chat_create_completion_params.Tool] | NotGiven = NOT_GIVEN,
         top_logprobs: int | NotGiven = NOT_GIVEN,
         top_p: float | NotGiven = NOT_GIVEN,
         user: str | NotGiven = NOT_GIVEN,
+        web_search_options: chat_create_completion_params.WebSearchOptions | 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,
-    ) -> ChatCreateCompletionResponse:
+    ) -> ChatCompletionResponse:
         """
         Creates a chat completion for the given model
 
         Args:
+          query_stream: If true, server responds as an SSE stream. Generators may produce an ergonomic
+              streaming method when this is set.
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -119,7 +125,7 @@ class ChatResource(SyncAPIResource):
                     "seed": seed,
                     "stop": stop,
                     "store": store,
-                    "stream": stream,
+                    "body_stream": body_stream,
                     "stream_options": stream_options,
                     "temperature": temperature,
                     "tool_choice": tool_choice,
@@ -127,13 +133,20 @@ class ChatResource(SyncAPIResource):
                     "top_logprobs": top_logprobs,
                     "top_p": top_p,
                     "user": user,
+                    "web_search_options": web_search_options,
                 },
                 chat_create_completion_params.ChatCreateCompletionParams,
             ),
             options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=maybe_transform(
+                    {"query_stream": query_stream}, chat_create_completion_params.ChatCreateCompletionParams
+                ),
             ),
-            cast_to=ChatCreateCompletionResponse,
+            cast_to=ChatCompletionResponse,
         )
 
 
@@ -162,6 +175,7 @@ class AsyncChatResource(AsyncAPIResource):
         *,
         messages: Iterable[ChatCompletionMessageParam],
         model: str,
+        query_stream: bool | NotGiven = NOT_GIVEN,
         chat_template_kwargs: object | NotGiven = NOT_GIVEN,
         frequency_penalty: float | NotGiven = NOT_GIVEN,
         function_call: object | NotGiven = NOT_GIVEN,
@@ -180,25 +194,29 @@ class AsyncChatResource(AsyncAPIResource):
         seed: int | NotGiven = NOT_GIVEN,
         stop: List[str] | NotGiven = NOT_GIVEN,
         store: bool | NotGiven = NOT_GIVEN,
-        stream: bool | NotGiven = NOT_GIVEN,
-        stream_options: chat_create_completion_params.StreamOptions | NotGiven = NOT_GIVEN,
+        body_stream: bool | NotGiven = NOT_GIVEN,
+        stream_options: StreamOptionsParam | NotGiven = NOT_GIVEN,
         temperature: float | NotGiven = NOT_GIVEN,
         tool_choice: object | NotGiven = NOT_GIVEN,
         tools: Iterable[chat_create_completion_params.Tool] | NotGiven = NOT_GIVEN,
         top_logprobs: int | NotGiven = NOT_GIVEN,
         top_p: float | NotGiven = NOT_GIVEN,
         user: str | NotGiven = NOT_GIVEN,
+        web_search_options: chat_create_completion_params.WebSearchOptions | 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,
-    ) -> ChatCreateCompletionResponse:
+    ) -> ChatCompletionResponse:
         """
         Creates a chat completion for the given model
 
         Args:
+          query_stream: If true, server responds as an SSE stream. Generators may produce an ergonomic
+              streaming method when this is set.
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -231,7 +249,7 @@ class AsyncChatResource(AsyncAPIResource):
                     "seed": seed,
                     "stop": stop,
                     "store": store,
-                    "stream": stream,
+                    "body_stream": body_stream,
                     "stream_options": stream_options,
                     "temperature": temperature,
                     "tool_choice": tool_choice,
@@ -239,13 +257,20 @@ class AsyncChatResource(AsyncAPIResource):
                     "top_logprobs": top_logprobs,
                     "top_p": top_p,
                     "user": user,
+                    "web_search_options": web_search_options,
                 },
                 chat_create_completion_params.ChatCreateCompletionParams,
             ),
             options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=await async_maybe_transform(
+                    {"query_stream": query_stream}, chat_create_completion_params.ChatCreateCompletionParams
+                ),
             ),
-            cast_to=ChatCreateCompletionResponse,
+            cast_to=ChatCompletionResponse,
         )
 
 

relaxai/resources/embeddings.py

@@ -4,7 +4,7 @@ from __future__ import annotations
 
 import httpx
 
-from ..types import embedding_create_params
+from ..types import embedding_create_embedding_params
 from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven
 from .._utils import maybe_transform, async_maybe_transform
 from .._compat import cached_property
@@ -16,7 +16,7 @@ from .._response import (
     async_to_streamed_response_wrapper,
 )
 from .._base_client import make_request_options
-from ..types.embedding_create_response import EmbeddingCreateResponse
+from ..types.embedding_response import EmbeddingResponse
 
 __all__ = ["EmbeddingsResource", "AsyncEmbeddingsResource"]
 
@@ -41,7 +41,7 @@ class EmbeddingsResource(SyncAPIResource):
         """
         return EmbeddingsResourceWithStreamingResponse(self)
 
-    def create(
+    def create_embedding(
         self,
         *,
         input: object,
@@ -55,7 +55,7 @@ class EmbeddingsResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> EmbeddingCreateResponse:
+    ) -> EmbeddingResponse:
         """
         Creates an embedding vector representing the input text.
 
@@ -78,12 +78,12 @@ class EmbeddingsResource(SyncAPIResource):
                     "encoding_format": encoding_format,
                     "user": user,
                 },
-                embedding_create_params.EmbeddingCreateParams,
+                embedding_create_embedding_params.EmbeddingCreateEmbeddingParams,
             ),
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=EmbeddingCreateResponse,
+            cast_to=EmbeddingResponse,
         )
 
 
@@ -107,7 +107,7 @@ class AsyncEmbeddingsResource(AsyncAPIResource):
         """
         return AsyncEmbeddingsResourceWithStreamingResponse(self)
 
-    async def create(
+    async def create_embedding(
         self,
         *,
         input: object,
@@ -121,7 +121,7 @@ class AsyncEmbeddingsResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> EmbeddingCreateResponse:
+    ) -> EmbeddingResponse:
         """
         Creates an embedding vector representing the input text.
 
@@ -144,12 +144,12 @@ class AsyncEmbeddingsResource(AsyncAPIResource):
                     "encoding_format": encoding_format,
                     "user": user,
                 },
-                embedding_create_params.EmbeddingCreateParams,
+                embedding_create_embedding_params.EmbeddingCreateEmbeddingParams,
             ),
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=EmbeddingCreateResponse,
+            cast_to=EmbeddingResponse,
         )
 
 
@@ -157,8 +157,8 @@ class EmbeddingsResourceWithRawResponse:
     def __init__(self, embeddings: EmbeddingsResource) -> None:
         self._embeddings = embeddings
 
-        self.create = to_raw_response_wrapper(
-            embeddings.create,
+        self.create_embedding = to_raw_response_wrapper(
+            embeddings.create_embedding,
         )
 
 
@@ -166,8 +166,8 @@ class AsyncEmbeddingsResourceWithRawResponse:
     def __init__(self, embeddings: AsyncEmbeddingsResource) -> None:
         self._embeddings = embeddings
 
-        self.create = async_to_raw_response_wrapper(
-            embeddings.create,
+        self.create_embedding = async_to_raw_response_wrapper(
+            embeddings.create_embedding,
         )
 
 
@@ -175,8 +175,8 @@ class EmbeddingsResourceWithStreamingResponse:
     def __init__(self, embeddings: EmbeddingsResource) -> None:
         self._embeddings = embeddings
 
-        self.create = to_streamed_response_wrapper(
-            embeddings.create,
+        self.create_embedding = to_streamed_response_wrapper(
+            embeddings.create_embedding,
         )
 
 
@@ -184,6 +184,6 @@ class AsyncEmbeddingsResourceWithStreamingResponse:
     def __init__(self, embeddings: AsyncEmbeddingsResource) -> None:
         self._embeddings = embeddings
 
-        self.create = async_to_streamed_response_wrapper(
-            embeddings.create,
+        self.create_embedding = async_to_streamed_response_wrapper(
+            embeddings.create_embedding,
         )