U DAf.@sddlZddlmZmZmZmZddlmZddlm Z m Z m Z ddl m Z mZeejddZeejdd Ze Gd d d ZdS) N)AnyDictListOptional)OpenAI) componentdefault_from_dictdefault_to_dict)Secretdeserialize_secrets_inplaceOPENAI_TIMEOUTOPENAI_MAX_RETRIESc @seZdZdZedddddddddf eeeeeeeeeeee eed ddZ e ee fd d d Z e ee fd d d Zee ee fddddZejee e ee fdedddZdS)OpenAITextEmbeddera A component for embedding strings using OpenAI models. Usage example: ```python from haystack.components.embedders import OpenAITextEmbedder text_to_embed = "I love pizza!" text_embedder = OpenAITextEmbedder() print(text_embedder.run(text_to_embed)) # {'embedding': [0.017020374536514282, -0.023255806416273117, ...], # 'meta': {'model': 'text-embedding-ada-002-v2', # 'usage': {'prompt_tokens': 4, 'total_tokens': 4}}} ``` ZOPENAI_API_KEYztext-embedding-ada-002N) api_keymodel dimensions api_base_url organizationprefixsuffixtimeout max_retriesc Csz||_||_||_||_||_||_||_|dkrDttj dd}| dkr^t tj dd} t | |||| d|_dS)an Create an OpenAITextEmbedder component. By setting the 'OPENAI_TIMEOUT' and 'OPENAI_MAX_RETRIES' you can change the timeout and max_retries parameters in the OpenAI client. :param api_key: The OpenAI API key. :param model: The name of the model to use. :param dimensions: The number of dimensions the resulting output embeddings should have. Only supported in `text-embedding-3` a nd later models. :param api_base_url: Overrides default base url for all HTTP requests. :param organization: The Organization ID. See OpenAI's [production best practices](https://platform.openai.com/docs/guides/production-best-practices/setting-up-your-organization) for more information. :param prefix: A string to add at the beginning of each text. :param suffix: A string to add at the end of each text. :param timeout: Timeout for OpenAI Client calls, if not set it is inferred from the `OPENAI_TIMEOUT` environment variable or set to 30. :param max_retries: Maximum retries to stablish contact with OpenAI if it returns an internal error, if not set it is inferred from the `OPENAI_MAX_RETRIES` environment variable or set to 5. Nr g>@rr)rrbase_urlrr)rrrrrrrfloatosenvirongetintrZ resolve_valueclient) selfrrrrrrrrrr#V/tmp/pip-unpacked-wheel-z752163x/haystack/components/embedders/openai_text_embedder.py__init__&s$*zOpenAITextEmbedder.__init__)returncCs d|jiS)zC Data that is sent to Posthog for usage analytics. r)rr"r#r#r$_get_telemetry_dataesz&OpenAITextEmbedder._get_telemetry_datac Cs*t||j|j|j|j|j|j|jdS)z{ Serializes the component to a dictionary. :returns: Dictionary with serialized data. )rrrrrrr) r rrrrrrrto_dictr'r#r#r$r)kszOpenAITextEmbedder.to_dict)datar&cCst|ddgdt||S)z Deserializes the component from a dictionary. :param data: Dictionary to deserialize from. :returns: Deserialized component. Zinit_parametersr)keys)r r)clsr*r#r#r$ from_dict}s zOpenAITextEmbedder.from_dict embeddingmeta)textcCst|tstd|j||j}|dd}|jdk rR|jjj |j |j|d}n|jjj |j |d}|j t |j d}|j dj|d S) a Embed a single string. :param text: Text to embed. :returns: A dictionary with the following keys: - `embedding`: The embedding of the input text. - `meta`: Information about the usage of the model. zOpenAITextEmbedder expects a string as an input.In case you want to embed a list of Documents, please use the OpenAIDocumentEmbedder.  N)rrinput)rr4)rusagerr.) isinstancestr TypeErrorrrreplacerr!Z embeddingscreaterdictr5r*r/)r"r1Z text_to_embedresponser0r#r#r$runs   zOpenAITextEmbedder.run)__name__ __module__ __qualname____doc__r Z from_env_varr7rr rr%rrr(r) classmethodr-rZ output_typesrr=r#r#r#r$rs6 ? r)rtypingrrrrZopenairZhaystackrrr Zhaystack.utilsr r rrrr r rrr#r#r#r$s