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/gemini/back/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) 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 all embedding models), 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. In addition to the per-input token limit, all embedding models enforce a maximum of 300,000 tokens summed across all inputs in a single request. 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&parsern    z!Embeddings.create..parser /embeddingsr1r2r3r4 post_parserbodyoptionscast_toNrCrrrr_postrr EmbeddingCreateParamsrr r$r5r7r.r/r0r1r2r3r4paramsrVr%rUr&create/s*6 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,6  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        tt