a !f#@sddlmZddlZddlmZmZmZmZddlm Z ddl Z ddl m Z m Z ddlmZmZmZmZmZddlmZmZdd lmZdd lmZdd lmZmZdd lmZm Z dd l!m"Z"erddl#m$Z$m%Z%ddgZ&GdddeZ'GdddeZ(GdddZ)GdddZ*dS)) annotationsN) TYPE_CHECKINGListUnioncast)Literal)CreateEmbeddingResponseembedding_create_params) NOT_GIVENBodyQueryHeadersNotGiven)is_givenmaybe_transform)numpy) has_numpy)SyncAPIResourceAsyncAPIResource)to_raw_response_wrapperasync_to_raw_response_wrapper)make_request_options)OpenAI AsyncOpenAI EmbeddingsAsyncEmbeddingsc s\eZdZUded<dddfdd Zeeddded d d d d dddddd ddZZS)rEmbeddingsWithRawResponsewith_raw_responserNoneclientreturncst|t||_dSN)super__init__rrselfr! __class__\/var/www/html/python-backend/venv/lib/python3.9/site-packages/openai/resources/embeddings.pyr%s zEmbeddings.__init__Nencoding_formatuser extra_headers extra_query extra_bodytimeout1Union[str, List[str], List[int], List[List[int]]]-Union[str, Literal['text-embedding-ada-002']]%Literal['float', 'base64'] | NotGivenstr | NotGivenHeaders | None Query | None Body | None'float | httpx.Timeout | None | NotGivenr inputmodelr-r.r/r0r1r2r"c s`|||d} ts$tr$d| d<dddfdd } |jdt| 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`) and cannot be an empty string. [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. 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-base64r-r objr"csNtr |S|jD]6}tt|j}t|ts.qtjt |dd |_q|SNZfloat32)Zdtype rdatarobject embedding isinstancestrnpZ frombufferr@ b64decodetolistrBrGrEr-r*r+parserXs    z!Embeddings.create..parser /embeddingsr/r0r1r2Z post_parserbodyoptionsZcast_torr_postrr ZEmbeddingCreateParamsrr r'r<r=r-r.r/r0r1r2paramsrOr*rNr+create!s(/ zEmbeddings.create__name__ __module__ __qualname____annotations__r%r rY __classcell__r*r*r(r+rs  c s\eZdZUded<dddfdd Zeeddded d d d d dddddd ddZZS)rAsyncEmbeddingsWithRawResponserrrr cst|t||_dSr#)r$r%r`rr&r(r*r+r%zs zAsyncEmbeddings.__init__Nr,r3r4r5r6r7r8r9r:r r;c sf|||d} ts$tr$d| d<dddfdd } |jdt| tjt||||| d td Id HS) r>r?r@r-r rAcsNtr |S|jD]6}tt|j}t|ts.qtjt |dd |_q|SrCrDrMrNr*r+rOs    z&AsyncEmbeddings.create..parserrPrQrRNrUrWr*rNr+rY~s(/ zAsyncEmbeddings.createrZr*r*r(r+rws  c@seZdZdddddZdS)rrr embeddingsr"cCst|j|_dSr#)rrYr'rbr*r*r+r%sz"EmbeddingsWithRawResponse.__init__Nr[r\r]r%r*r*r*r+rsrc@seZdZdddddZdS)r`rrracCst|j|_dSr#)rrYrcr*r*r+r%sz'AsyncEmbeddingsWithRawResponse.__init__Nrdr*r*r*r+r`sr`)+ __future__rr@typingrrrrZtyping_extensionsrZhttpxtypesr r _typesr r r rr_utilsrr_extrasrrJrZ _resourcerr _responserrZ _base_clientrZ_clientrr__all__rrrr`r*r*r*r+s&     ]]