[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
jupyterlab · Mar 4, 2024 · caad50c · caad50c
1 parent c9e6994
commit caad50c
Showing 1 changed file with 16 additions and 13 deletions.
diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py b/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py
@@ -31,13 +31,13 @@
 from langchain_community.llms import (
     AI21,
     Anthropic,
-    Together,
     Bedrock,
     Cohere,
     GPT4All,
     HuggingFaceHub,
     OpenAI,
     SagemakerEndpoint,
+    Together,
 )
 
 # this is necessary because `langchain.pydantic_v1.main` does not include
@@ -858,25 +858,28 @@ class TogetherAIProvider(BaseProvider, Together):
     id = "togetherai"
     name = "Together AI"
     model_id_key = "model"
-    models = ['Austism/chronos-hermes-13b',
-              'DiscoResearch/DiscoLM-mixtral-8x7b-v2',
-              'EleutherAI/llemma_7b',
-              'Gryphe/MythoMax-L2-13b',
-              'Meta-Llama/Llama-Guard-7b',
-              'Nexusflow/NexusRaven-V2-13B',
-              'NousResearch/Nous-Capybara-7B-V1p9',
-              'NousResearch/Nous-Hermes-2-Yi-34B',
-              'NousResearch/Nous-Hermes-Llama2-13b',
-              'NousResearch/Nous-Hermes-Llama2-70b'
-              ]
+    models = [
+        "Austism/chronos-hermes-13b",
+        "DiscoResearch/DiscoLM-mixtral-8x7b-v2",
+        "EleutherAI/llemma_7b",
+        "Gryphe/MythoMax-L2-13b",
+        "Meta-Llama/Llama-Guard-7b",
+        "Nexusflow/NexusRaven-V2-13B",
+        "NousResearch/Nous-Capybara-7B-V1p9",
+        "NousResearch/Nous-Hermes-2-Yi-34B",
+        "NousResearch/Nous-Hermes-Llama2-13b",
+        "NousResearch/Nous-Hermes-Llama2-70b",
+    ]
     pypi_package_deps = ["together"]
     auth_strategy = EnvAuthStrategy(name="TOGETHER_API_KEY")
 
     def __init__(self, **kwargs):
         model = kwargs.get("model_id")
 
         if model not in self.models:
-            kwargs["responses"] = ["Model not supported! Please check model list with %ai list"]
+            kwargs["responses"] = [
+                "Model not supported! Please check model list with %ai list"
+            ]
 
         super().__init__(**kwargs)