pre-commit

jupyterlab · Aug 7, 2023 · 0a65fcb · 0a65fcb
1 parent 04a1814
commit 0a65fcb
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 9 deletions.
diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py b/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py
@@ -9,7 +9,7 @@
 
 from jsonpath_ng import parse
 from langchain import PromptTemplate
-from langchain.chat_models import ChatOpenAI, AzureChatOpenAI
+from langchain.chat_models import AzureChatOpenAI, ChatOpenAI
 from langchain.llms import (
     AI21,
     Anthropic,
@@ -419,11 +419,16 @@ class ChatOpenAINewProvider(BaseProvider, ChatOpenAI):
     auth_strategy = EnvAuthStrategy(name="OPENAI_API_KEY")
 
     fields = [
-        TextField(key="openai_api_base", label="Base API URL (optional)", format="text"),
-        TextField(key="openai_organization", label="Organization (optional)", format="text"),
+        TextField(
+            key="openai_api_base", label="Base API URL (optional)", format="text"
+        ),
+        TextField(
+            key="openai_organization", label="Organization (optional)", format="text"
+        ),
         TextField(key="openai_proxy", label="Proxy (optional)", format="text"),
     ]
 
+
 class AzureChatOpenAIProvider(BaseProvider, AzureChatOpenAI):
     id = "azure-chat-openai"
     name = "Azure OpenAI"
@@ -435,12 +440,19 @@ class AzureChatOpenAIProvider(BaseProvider, AzureChatOpenAI):
     registry = True
 
     fields = [
-        TextField(key="openai_api_base", label="Base API URL (required)", format="text"),
-        TextField(key="openai_api_version", label="API version (required)", format="text"),
-        TextField(key="openai_organization", label="Organization (optional)", format="text"),
+        TextField(
+            key="openai_api_base", label="Base API URL (required)", format="text"
+        ),
+        TextField(
+            key="openai_api_version", label="API version (required)", format="text"
+        ),
+        TextField(
+            key="openai_organization", label="Organization (optional)", format="text"
+        ),
         TextField(key="openai_proxy", label="Proxy (optional)", format="text"),
     ]
 
+
 class JsonContentHandler(LLMContentHandler):
     content_type = "application/json"
     accepts = "application/json"
@@ -491,8 +503,12 @@ class SmEndpointProvider(BaseProvider, SagemakerEndpoint):
     registry = True
     fields = [
         TextField(key="region_name", label="Region name (required)", format="text"),
-        MultilineTextField(key="request_schema", label="Request schema (required)", format="json"),
-        TextField(key="response_path", label="Response path (required)", format="jsonpath"),
+        MultilineTextField(
+            key="request_schema", label="Request schema (required)", format="json"
+        ),
+        TextField(
+            key="response_path", label="Response path (required)", format="jsonpath"
+        ),
     ]
 
     def __init__(self, *args, **kwargs):

diff --git a/packages/jupyter-ai/jupyter_ai/handlers.py b/packages/jupyter-ai/jupyter_ai/handlers.py
@@ -30,8 +30,9 @@
 )
 
 if TYPE_CHECKING:
-    from jupyter_ai_magics.providers import BaseProvider
     from jupyter_ai_magics.embedding_providers import BaseEmbeddingsProvider
+    from jupyter_ai_magics.providers import BaseProvider
+
 
 class ChatHistoryHandler(BaseAPIHandler):
     """Handler to return message history"""