This is an automated email from the ASF dual-hosted git repository.
kaxil pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git
The following commit(s) were added to refs/heads/main by this push:
new 427117ee911 Update OpenAI provider to OpenAI Python SDK 2.x (#69068)
427117ee911 is described below
commit 427117ee9114d4e29138ac5f24e78a0778853956
Author: Kaxil Naik <[email protected]>
AuthorDate: Sat Jun 27 02:41:29 2026 +0100
Update OpenAI provider to OpenAI Python SDK 2.x (#69068)
---
providers/openai/README.rst | 2 +-
providers/openai/docs/index.rst | 2 +-
providers/openai/pyproject.toml | 2 +-
providers/openai/src/airflow/providers/openai/hooks/openai.py | 6 +++---
providers/openai/src/airflow/providers/openai/operators/openai.py | 2 +-
providers/openai/tests/system/openai/example_openai.py | 6 +++---
6 files changed, 10 insertions(+), 10 deletions(-)
diff --git a/providers/openai/README.rst b/providers/openai/README.rst
index a3c4ba53ac5..5d78147fb2d 100644
--- a/providers/openai/README.rst
+++ b/providers/openai/README.rst
@@ -57,7 +57,7 @@ PIP package Version required
========================================== ==================
``apache-airflow`` ``>=2.11.0``
``apache-airflow-providers-common-compat`` ``>=1.12.0``
-``openai[datalib]`` ``>=1.66.0``
+``openai[datalib]`` ``>=2.37.0``
========================================== ==================
Cross provider package dependencies
diff --git a/providers/openai/docs/index.rst b/providers/openai/docs/index.rst
index 70ba43a0a9b..7883d87efdd 100644
--- a/providers/openai/docs/index.rst
+++ b/providers/openai/docs/index.rst
@@ -98,7 +98,7 @@ PIP package Version required
========================================== ==================
``apache-airflow`` ``>=2.11.0``
``apache-airflow-providers-common-compat`` ``>=1.12.0``
-``openai[datalib]`` ``>=1.66.0``
+``openai[datalib]`` ``>=2.37.0``
========================================== ==================
Cross provider package dependencies
diff --git a/providers/openai/pyproject.toml b/providers/openai/pyproject.toml
index 489e9c69c49..5e2f97bd29e 100644
--- a/providers/openai/pyproject.toml
+++ b/providers/openai/pyproject.toml
@@ -61,7 +61,7 @@ requires-python = ">=3.10"
dependencies = [
"apache-airflow>=2.11.0",
"apache-airflow-providers-common-compat>=1.12.0",
- "openai[datalib]>=1.66.0",
+ "openai[datalib]>=2.37.0",
]
[dependency-groups]
diff --git a/providers/openai/src/airflow/providers/openai/hooks/openai.py
b/providers/openai/src/airflow/providers/openai/hooks/openai.py
index c245bd10ca0..dd98b239848 100644
--- a/providers/openai/src/airflow/providers/openai/hooks/openai.py
+++ b/providers/openai/src/airflow/providers/openai/hooks/openai.py
@@ -129,7 +129,7 @@ class OpenAIHook(BaseHook):
| ChatCompletionToolMessageParam
| ChatCompletionFunctionMessageParam
],
- model: str = "gpt-3.5-turbo",
+ model: str = "gpt-4o-mini",
**kwargs: Any,
) -> list[ChatCompletionMessage]:
"""
@@ -141,7 +141,7 @@ class OpenAIHook(BaseHook):
response = self.conn.chat.completions.create(model=model,
messages=messages, **kwargs)
return response.choices
- def create_assistant(self, model: str = "gpt-3.5-turbo", **kwargs: Any) ->
Assistant:
+ def create_assistant(self, model: str = "gpt-4o-mini", **kwargs: Any) ->
Assistant:
"""
Create an OpenAI assistant using the given model.
@@ -297,7 +297,7 @@ class OpenAIHook(BaseHook):
def create_embeddings(
self,
text: str | list[str] | list[int] | list[list[int]],
- model: str = "text-embedding-ada-002",
+ model: str = "text-embedding-3-small",
**kwargs: Any,
) -> list[float]:
"""
diff --git a/providers/openai/src/airflow/providers/openai/operators/openai.py
b/providers/openai/src/airflow/providers/openai/operators/openai.py
index 574d7fb85d8..192986ab456 100644
--- a/providers/openai/src/airflow/providers/openai/operators/openai.py
+++ b/providers/openai/src/airflow/providers/openai/operators/openai.py
@@ -54,7 +54,7 @@ class OpenAIEmbeddingOperator(BaseOperator):
self,
conn_id: str,
input_text: str | list[str] | list[int] | list[list[int]],
- model: str = "text-embedding-ada-002",
+ model: str = "text-embedding-3-small",
embedding_kwargs: dict | None = None,
**kwargs: Any,
):
diff --git a/providers/openai/tests/system/openai/example_openai.py
b/providers/openai/tests/system/openai/example_openai.py
index f8df2bd6024..6cbbe2b47f3 100644
--- a/providers/openai/tests/system/openai/example_openai.py
+++ b/providers/openai/tests/system/openai/example_openai.py
@@ -78,7 +78,7 @@ def example_openai_dag():
task_id="embedding_using_xcom_data",
conn_id="openai_default",
input_text=task_to_store_input_text_in_xcom(),
- model="text-embedding-ada-002",
+ model="text-embedding-3-small",
)
OpenAIEmbeddingOperator(
@@ -90,13 +90,13 @@ def example_openai_dag():
input_kwarg1="input_kwarg1_value",
input_kwarg2="input_kwarg2_value",
),
- model="text-embedding-ada-002",
+ model="text-embedding-3-small",
)
OpenAIEmbeddingOperator(
task_id="embedding_using_text",
conn_id="openai_default",
input_text=texts,
- model="text-embedding-ada-002",
+ model="text-embedding-3-small",
)
# [END howto_operator_openai_embedding]
