o +if*@s<ddlmZddlZddlmZmZmZmZddlm Z ddl Z ddl m Z ddl mZddlmZmZmZmZmZdd lmZmZdd lmZdd lmZmZdd lmZm Z dd l!m"Z"m#Z#ddl$m%Z%ddl&m'Z'ddgZ(GdddeZ)Gddde Z*GdddZ+GdddZ,GdddZ-GdddZ.dS)) 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)CreateEmbeddingResponse EmbeddingsAsyncEmbeddingsc@DeZdZedddZed ddZeeeddded d!ddZdS)"rreturnEmbeddingsWithRawResponsecCt|SN)rselfr$T/var/www/html/corbot_env/lib/python3.10/site-packages/openai/resources/embeddings.pywith_raw_responsezEmbeddings.with_raw_responseEmbeddingsWithStreamingResponsecCr r!)r(r"r$r$r%with_streaming_response r'z"Embeddings.with_streaming_responseN dimensionsencoding_formatuser extra_headers extra_query extra_bodytimeoutinput=Union[str, List[str], Iterable[int], Iterable[Iterable[int]]]modelaUnion[str, Literal['text-embedding-ada-002', 'text-embedding-3-small', 'text-embedding-3-large']]r+int | NotGivenr,%Literal['float', 'base64'] | NotGivenr-str | NotGivenr.Headers | Noner/ Query | Noner0 Body | Noner1'float | httpx.Timeout | None | NotGivenrc s\||||d} tstrd| d<d fdd } |jd t| tjt|||| | d td S)  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. 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/overview) 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 r2r4r-r+r,base64r,objrrcNtr|S|jD]}tt|j}t|tsq tjt |dd |_q |SNfloat32)dtype rdatarobject embedding isinstancestrnp frombufferr? b64decodetolistr@rHrFr,r$r%parsera    z!Embeddings.create..parser /embeddingsr.r/r0r1 post_parserbodyoptionscast_toNr@rrrrr_postrr EmbeddingCreateParamsrr r#r2r4r+r,r-r.r/r0r1paramsrQr$rPr%create$s*4 zEmbeddings.create)rr)rr(r2r3r4r5r+r6r,r7r-r8r.r9r/r:r0r;r1r<rr__name__ __module__ __qualname__rr&r)r r`r$r$r$r%r  c@r)"rrAsyncEmbeddingsWithRawResponsecCr r!)rgr"r$r$r%r&r'z!AsyncEmbeddings.with_raw_response$AsyncEmbeddingsWithStreamingResponsecCr r!)rhr"r$r$r%r)r'z'AsyncEmbeddings.with_streaming_responseNr*r2r3r4r5r+r6r,r7r-r8r.r9r/r:r0r;r1r<rc sd||||d} tstrd| d<d fdd } |jd t| tjt|||| | d td Id HS)r=r>r?r,r@rrcrArBrErOrPr$r%rQrRz&AsyncEmbeddings.create..parserrSrTrVNrZr[r^r$rPr%r`s,4  zAsyncEmbeddings.create)rrg)rrhrarbr$r$r$r%rrfc@eZdZdddZdS) r embeddingsrrNonecC||_t|j|_dSr!) _embeddingsr to_raw_response_wrapperr`r#rjr$r$r%__init__ z"EmbeddingsWithRawResponse.__init__Nrjrrrkrcrdrerpr$r$r$r%rrc@ri) rgrjrrrkcCrlr!)rmr async_to_raw_response_wrapperr`ror$r$r%rprqz'AsyncEmbeddingsWithRawResponse.__init__Nrjrrrkrsr$r$r$r%rgrtrgc@ri) r(rjrrrkcC||_t|j|_dSr!)rmrr`ror$r$r%rp z(EmbeddingsWithStreamingResponse.__init__Nrrrsr$r$r$r%r(rtr(c@ri) rhrjrrrkcCrwr!)rmrr`ror$r$r%rprxz-AsyncEmbeddingsWithStreamingResponse.__init__Nrvrsr$r$r$r%rhrtrh)/ __future__rr?typingrrrrtyping_extensionsrhttpxr typesr _typesr r r rr_utilsrr_compatr_extrasrrKr _resourcerr _responserr _base_clientrtypes.create_embedding_responser__all__rrrrgr(rhr$r$r$r%s,       ee