From aaec540f87d77eee946d36f867b4ac3eed1a504b Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Fri, 12 Jan 2024 22:13:27 +0000
Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 .../jupyter_ai_magics/embedding_providers.py  |  3 +--
 .../jupyter_ai_magics/providers.py            | 26 ++++++++++---------
 2 files changed, 15 insertions(+), 14 deletions(-)

diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/embedding_providers.py b/packages/jupyter-ai-magics/jupyter_ai_magics/embedding_providers.py
index 7a7744d6e..3f109f72f 100644
--- a/packages/jupyter-ai-magics/jupyter_ai_magics/embedding_providers.py
+++ b/packages/jupyter-ai-magics/jupyter_ai_magics/embedding_providers.py
@@ -142,8 +142,7 @@ class QianfanEmbeddingsEndpointProvider(
     auth_strategy = MultiEnvAuthStrategy(names=["QIANFAN_AK", "QIANFAN_SK"])
 
 
-class NVIDIAEmbeddingsProvider(BaseEmbeddingsProvider, NVIDIAEmbeddings
-):
+class NVIDIAEmbeddingsProvider(BaseEmbeddingsProvider, NVIDIAEmbeddings):
     id = "nvidia"
     name = "NVIDIA"
     models = ["playground_nvolveqa_40k"]
diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py b/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py
index 60179a65e..062bb1863 100644
--- a/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py
+++ b/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py
@@ -360,6 +360,7 @@ class ChatAnthropicProvider(BaseProvider, ChatAnthropic):
     def allows_concurrency(self):
         return False
 
+
 class CohereProvider(BaseProvider, Cohere):
     id = "cohere"
     name = "Cohere"
@@ -775,17 +776,18 @@ class QianfanProvider(BaseProvider, QianfanChatEndpoint):
 class ChatNVIDIAProvider(BaseProvider, ChatNVIDIA):
     id = "nvidia-chat"
     name = "NVIDIA"
-    models = ['playground_llama2_70b',
-              'playground_nemotron_steerlm_8b',
-              'playground_mistral_7b',
-              'playground_nv_llama2_rlhf_70b',
-              'playground_llama2_13b',
-              'playground_steerlm_llama_70b',
-              'playground_llama2_code_13b',
-              'playground_yi_34b',
-              'playground_mixtral_8x7b',
-              'playground_neva_22b',
-              'playground_llama2_code_34b'
-             ]
+    models = [
+        "playground_llama2_70b",
+        "playground_nemotron_steerlm_8b",
+        "playground_mistral_7b",
+        "playground_nv_llama2_rlhf_70b",
+        "playground_llama2_13b",
+        "playground_steerlm_llama_70b",
+        "playground_llama2_code_13b",
+        "playground_yi_34b",
+        "playground_mixtral_8x7b",
+        "playground_neva_22b",
+        "playground_llama2_code_34b",
+    ]
     model_id_key = "model"
     auth_strategy = EnvAuthStrategy(name="NVIDIA_API_KEY")