diff --git a/providers/openai/pyproject.toml b/providers/openai/pyproject.toml index 07922d599bad4..2cb41064bf2ea 100644 --- a/providers/openai/pyproject.toml +++ b/providers/openai/pyproject.toml @@ -59,7 +59,7 @@ requires-python = ">=3.10" # After you modify the dependencies, and rebuild your Breeze CI image with ``breeze ci-image build`` dependencies = [ "apache-airflow>=2.11.0", - "apache-airflow-providers-common-compat>=1.10.1", + "apache-airflow-providers-common-compat>=1.10.1", # use next version "openai[datalib]>=1.66.0", ] diff --git a/providers/openai/src/airflow/providers/openai/operators/openai.py b/providers/openai/src/airflow/providers/openai/operators/openai.py index 9479335b7d7c0..574d7fb85d857 100644 --- a/providers/openai/src/airflow/providers/openai/operators/openai.py +++ b/providers/openai/src/airflow/providers/openai/operators/openai.py @@ -22,8 +22,7 @@ from functools import cached_property from typing import TYPE_CHECKING, Any, Literal -from airflow.configuration import conf -from airflow.providers.common.compat.sdk import BaseOperator +from airflow.providers.common.compat.sdk import BaseOperator, conf from airflow.providers.openai.exceptions import OpenAIBatchJobException from airflow.providers.openai.hooks.openai import OpenAIHook from airflow.providers.openai.triggers.openai import OpenAIBatchTrigger