From 3ac5700bf01c76add9aaefd8a8ca646d1449b2a6 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Wed, 7 Aug 2024 13:59:42 +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 | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 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 e804968f8..5d8f49073 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,4 +1,10 @@ -from langchain_openai import AzureChatOpenAI, ChatOpenAI, OpenAI, OpenAIEmbeddings, AzureOpenAIEmbeddings +from langchain_openai import ( + AzureChatOpenAI, + AzureOpenAIEmbeddings, + ChatOpenAI, + OpenAI, + OpenAIEmbeddings, +) from ..embedding_providers import BaseEmbeddingsProvider from ..providers import BaseProvider, EnvAuthStrategy, TextField @@ -121,5 +127,5 @@ class AzureOpenAIEmbeddingsProvider(BaseEmbeddingsProvider, AzureOpenAIEmbedding auth_strategy = EnvAuthStrategy( name="AZURE_OPENAI_API_KEY", keyword_param="openai_api_key" ) - + registry = True