o Th/@sPddlmZddlZddlZddlmZmZmZmZddl m Z ddl Z ddl m Z ddlmZddlmZmZmZmZmZdd lmZmZdd lmZdd lmZmZdd lm Z m!Z!dd l"m#Z#m$Z$ddl%m&Z&ddl'm(Z(ddl)m*Z*ddgZ+Gddde Z,Gddde!Z-GdddZ.GdddZ/GdddZ0GdddZ1dS)) annotationsN)ListUnionIterablecast)Literal)_legacy_response)embedding_create_params) NOT_GIVENBodyQueryHeadersNotGiven)is_givenmaybe_transform)cached_property)numpy has_numpy)SyncAPIResourceAsyncAPIResource)to_streamed_response_wrapper"async_to_streamed_response_wrapper)make_request_options)EmbeddingModel)CreateEmbeddingResponse EmbeddingsAsyncEmbeddingsc@DeZdZedddZed ddZeeeddded d!ddZdS)"rreturnEmbeddingsWithRawResponsecCt|Sa This property can be used as a prefix for any HTTP method call to return the raw response object instead of the parsed content. For more information, see https://www.github.com/openai/openai-python#accessing-raw-response-data-eg-headers )r selfr%[/home/air/segue/gpt/backup/venv/lib/python3.10/site-packages/openai/resources/embeddings.pywith_raw_responsezEmbeddings.with_raw_responseEmbeddingsWithStreamingResponsecCr!z An alternative to `.with_raw_response` that doesn't eagerly read the response body. For more information, see https://www.github.com/openai/openai-python#with_streaming_response )r)r#r%r%r&with_streaming_response&z"Embeddings.with_streaming_responseN dimensionsencoding_formatuser extra_headers extra_query extra_bodytimeoutinput=Union[str, List[str], Iterable[int], Iterable[Iterable[int]]]modelUnion[str, EmbeddingModel]r.int | NotGivenr/%Literal['float', 'base64'] | NotGivenr0str | NotGivenr1Headers | Noner2 Query | Noner3 Body | Noner4'float | httpx.Timeout | None | NotGivenrc sV||||d} tsd| d<d fdd } |jd t| tjt|||| | d td S)c Creates an embedding vector representing the input text. Args: input: Input text to embed, encoded as a string or array of tokens. To embed multiple inputs in a single request, pass an array of strings or array of token arrays. The input must not exceed the max input tokens for the model (8192 tokens for `text-embedding-ada-002`), cannot be an empty string, and any array must be 2048 dimensions or less. [Example Python code](https://cookbook.openai.com/examples/how_to_count_tokens_with_tiktoken) for counting tokens. Some models may also impose a limit on total number of tokens summed across inputs. model: ID of the model to use. You can use the [List models](https://platform.openai.com/docs/api-reference/models/list) API to see all of your available models, or see our [Model overview](https://platform.openai.com/docs/models) for descriptions of them. dimensions: The number of dimensions the resulting output embeddings should have. Only supported in `text-embedding-3` and later models. encoding_format: The format to return the embeddings in. Can be either `float` or [`base64`](https://pypi.org/project/pybase64/). user: A unique identifier representing your end-user, which can help OpenAI to monitor and detect abuse. [Learn more](https://platform.openai.com/docs/guides/safety-best-practices#end-user-ids). extra_headers: Send extra headers extra_query: Add additional query parameters to the request extra_body: Add additional JSON properties to the request timeout: Override the client-level default timeout for this request, in seconds r5r7r0r.r/base64r/objrrcntr|S|jD]+}tt|j}t|tsq ts'tdt | |_q t j t |dd |_q |SNffloat32)dtyperdatarobject embedding isinstancestrrarrayrB b64decodetolistnp frombufferrCrLrJr/r%r&parserm    z!Embeddings.create..parser /embeddingsr1r2r3r4 post_parserbodyoptionscast_toNrCrrrr_postrr EmbeddingCreateParamsrr r$r5r7r.r/r0r1r2r3r4paramsrVr%rUr&create/s*5 zEmbeddings.create)rr )rr)r5r6r7r8r.r9r/r:r0r;r1r<r2r=r3r>r4r?rr__name__ __module__ __qualname__rr'r+r rer%r%r%r&r   c@r)"rrAsyncEmbeddingsWithRawResponsecCr!r")rlr#r%r%r&r'r(z!AsyncEmbeddings.with_raw_response$AsyncEmbeddingsWithStreamingResponsecCr!r*)rmr#r%r%r&r+r,z'AsyncEmbeddings.with_streaming_responseNr-r5r6r7r8r.r9r/r:r0r;r1r<r2r=r3r>r4r?rc s^||||d} tsd| d<d fdd } |jd t| tjt|||| | d td Id HS)r@rArBr/rCrrcrDrErIrTrUr%r&rVrWz&AsyncEmbeddings.create..parserrXrYr[Nr_r`rcr%rUr&res,5  zAsyncEmbeddings.create)rrl)rrmrfrgr%r%r%r&rrkc@eZdZdddZdS) r embeddingsrrNonecC||_t|j|_dSN) _embeddingsr to_raw_response_wrapperrer$ror%r%r&__init__ z"EmbeddingsWithRawResponse.__init__Nrorrrprhrirjrvr%r%r%r&r r c@rn) rlrorrrpcCrqrr)rsr async_to_raw_response_wrapperrerur%r%r&rv rwz'AsyncEmbeddingsWithRawResponse.__init__Nrorrrpryr%r%r%r&rl rzrlc@rn) r)rorrrpcC||_t|j|_dSrr)rsrrerur%r%r&rv z(EmbeddingsWithStreamingResponse.__init__Nrxryr%r%r%r&r)rzr)c@rn) rmrorrrpcCr}rr)rsrrerur%r%r&rvr~z-AsyncEmbeddingsWithStreamingResponse.__init__Nr|ryr%r%r%r&rmrzrm)2 __future__rrOrBtypingrrrrtyping_extensionsrhttpxr typesr _typesr r r rr_utilsrr_compatr_extrasrrRr _resourcerr _responserr _base_clientrtypes.embedding_modelrtypes.create_embedding_responser__all__rrr rlr)rmr%r%r%r&s0        ss