airflow.providers.openai.operators.openai
¶
Module Contents¶
Classes¶
Operator that accepts input text to generate OpenAI embeddings using the specified model. |
- class airflow.providers.openai.operators.openai.OpenAIEmbeddingOperator(conn_id, input_text, model='text-embedding-ada-002', embedding_kwargs=None, **kwargs)[source]¶
Bases:
airflow.models.BaseOperator
Operator that accepts input text to generate OpenAI embeddings using the specified model.
- Parameters
conn_id (str) – The OpenAI connection ID to use.
input_text (str | list[str] | list[int] | list[list[int]]) – The text to generate OpenAI embeddings for. This can be a string, a list of strings, a list of integers, or a list of lists of integers.
model (str) – The OpenAI model to be used for generating the embeddings.
embedding_kwargs (dict | None) – Additional keyword arguments to pass to the OpenAI create_embeddings method.
See also
For more information on how to use this operator, take a look at the guide: OpenAIEmbeddingOperator For possible options for embedding_kwargs, see: https://platform.openai.com/docs/api-reference/embeddings/create