`airflow.providers.openai.hooks.openai`¶

Module Contents¶

Classes¶

Use OpenAI SDK to interact with OpenAI APIs.

class airflow.providers.openai.hooks.openai.OpenAIHook(conn_id=default_conn_name, *args, **kwargs)[source]¶

Bases: airflow.hooks.base.BaseHook

Use OpenAI SDK to interact with OpenAI APIs.

See also

https://platform.openai.com/docs/introduction/overview

Parameters: conn_id (str) – OpenAI connection id

conn_name_attr = 'conn_id'[source]¶

default_conn_name = 'openai_default'[source]¶

conn_type = 'openai'[source]¶

hook_name = 'OpenAI'[source]¶

classmethod get_ui_field_behaviour()[source]¶

Return custom field behaviour.

test_connection()[source]¶

conn()[source]¶

Return an OpenAI connection object.

get_conn()[source]¶

Return an OpenAI connection object.

create_embeddings(text, model='text-embedding-ada-002', **kwargs)[source]¶

Generate embeddings for the given text using the given model.

Parameters

text (str | list[str] | list[int] | list[list[int]]) – The text to generate embeddings for.
model (str) – The model to use for generating embeddings.

Was this entry helpful?