U ܟj/@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 @sbeZdZeddddZeddddZeeeddded d d d d ddddddd ddZdS)rEmbeddingsWithRawResponsereturncCst|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 )rselfr$?/tmp/pip-unpacked-wheel-2qj3ofc_/openai/resources/embeddings.pywith_raw_responseszEmbeddings.with_raw_responseEmbeddingsWithStreamingResponsecCst|Sz 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&sz"Embeddings.with_streaming_responseN dimensionsencoding_formatuser extra_headers extra_query extra_bodytimeout=Union[str, List[str], Iterable[int], Iterable[Iterable[int]]]Union[str, EmbeddingModel]int | NotGiven'Literal[('float', 'base64')] | NotGivenstr | NotGivenHeaders | None Query | None Body | None'float | httpx.Timeout | None | NotGivenr inputmodelr+r,r-r.r/r0r1r c s\||||d} ts d| d<dddfdd } |jdt| 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 r<r=r-r+r,base64r,robjr csntr |S|jD]V}tt|j}t|ts.qtsNtdt | |_qt j t |dd |_q|SNfZfloat32)Zdtyperdatarobject embedding isinstancestrrarrayr@ b64decodetolistnpZ frombufferrBrHrFr,r$r%parserms   z!Embeddings.create..parser /embeddingsr.r/r0r1Z post_parserbodyoptionsZcast_tor_postrr ZEmbeddingCreateParamsrr r#r<r=r+r,r-r.r/r0r1paramsrQr$rPr%create/s*5 zEmbeddings.create__name__ __module__ __qualname__rr&r)r r[r$r$r$r%rs  c @sbeZdZeddddZeddddZeeeddded d d d d ddddddd ddZdS)rAsyncEmbeddingsWithRawResponsercCst|Sr!)r`r"r$r$r%r&sz!AsyncEmbeddings.with_raw_response$AsyncEmbeddingsWithStreamingResponsecCst|Sr()rar"r$r$r%r)sz'AsyncEmbeddings.with_streaming_responseNr*r2r3r4r5r6r7r8r9r:rr;c sb||||d} ts d| d<dddfdd } |jdt| tjt|||| | d td Id HS) r>r?r@r,rrAcsntr |S|jD]V}tt|j}t|ts.qtsNtdt | |_qt j t |dd |_q|SrCrErOrPr$r%rQs   z&AsyncEmbeddings.create..parserrRrSrTNrWrYr$rPr%r[s*5 zAsyncEmbeddings.creater\r$r$r$r%rs  c@seZdZdddddZdS)rrNone embeddingsr cCs||_t|j|_dSN) _embeddingsr Zto_raw_response_wrapperr[r#rdr$r$r%__init__sz"EmbeddingsWithRawResponse.__init__Nr]r^r_rhr$r$r$r%rsrc@seZdZdddddZdS)r`rrbrccCs||_t|j|_dSre)rfr Zasync_to_raw_response_wrapperr[rgr$r$r%rh sz'AsyncEmbeddingsWithRawResponse.__init__Nrir$r$r$r%r` sr`c@seZdZdddddZdS)r'rrbrccCs||_t|j|_dSre)rfrr[rgr$r$r%rhsz(EmbeddingsWithStreamingResponse.__init__Nrir$r$r$r%r'sr'c@seZdZdddddZdS)rarrbrccCs||_t|j|_dSre)rfrr[rgr$r$r%rhsz-AsyncEmbeddingsWithStreamingResponse.__init__Nrir$r$r$r%rasra)2 __future__rrKr@typingrrrrtyping_extensionsrZhttpxr typesr _typesr r r rr_utilsrr_compatrZ_extrasrrNrZ _resourcerr _responserrZ _base_clientrZtypes.embedding_modelrZtypes.create_embedding_responser__all__rrrr`r'rar$r$r$r%s.        ss