From a518d5a553231fe61af48b95d4089981bf02f3b2 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Tue, 26 Mar 2024 22:18:34 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- .../jupyter_ai_magics/partner_providers/openai.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/partner_providers/openai.py b/packages/jupyter-ai-magics/jupyter_ai_magics/partner_providers/openai.py index f9c88dbd8..382a480e1 100644 --- a/packages/jupyter-ai-magics/jupyter_ai_magics/partner_providers/openai.py +++ b/packages/jupyter-ai-magics/jupyter_ai_magics/partner_providers/openai.py @@ -1,6 +1,8 @@ -from ..providers import BaseProvider, EnvAuthStrategy, TextField +from langchain_openai import AzureChatOpenAI, ChatOpenAI, OpenAI, OpenAIEmbeddings + from ..embedding_providers import BaseEmbeddingsProvider -from langchain_openai import OpenAI, ChatOpenAI, AzureChatOpenAI, OpenAIEmbeddings +from ..providers import BaseProvider, EnvAuthStrategy, TextField + class OpenAIProvider(BaseProvider, OpenAI): id = "openai" @@ -104,4 +106,3 @@ class OpenAIEmbeddingsProvider(BaseEmbeddingsProvider, OpenAIEmbeddings): model_id_key = "model" pypi_package_deps = ["langchain_openai"] auth_strategy = EnvAuthStrategy(name="OPENAI_API_KEY") -