Run pre-commit (#1423)

Author: dlqqq

packages/jupyter-ai-magics/jupyter_ai_magics/partner_providers/vertexai.py

@@ -20,6 +20,7 @@ class VertexAIProvider(BaseProvider, VertexAI):
         "For more information, see the [Vertex AI authentication documentation](https://python.langchain.com/docs/integrations/llms/google_vertex_ai_palm/)."
     )
 
+
 class VertexAIEmbeddingsProvider(BaseProvider, VertexAIEmbeddings):
     id = "vertexai"
     name = "Vertex AI"
@@ -35,4 +36,4 @@ class VertexAIEmbeddingsProvider(BaseProvider, VertexAIEmbeddings):
         "- Store the path to a service account JSON file as the GOOGLE_APPLICATION_CREDENTIALS environment variable\n\n"
         "This codebase uses the google.auth library which first looks for the application credentials variable mentioned above, and then looks for system-level auth. "
         "For more information, see the [Vertex AI authentication documentation](https://python.langchain.com/docs/integrations/llms/google_vertex_ai_palm/)."
-    )
+    )

packages/jupyter-ai/jupyter_ai/config/__init__.py

@@ -1 +1 @@
-from .config_models import JaiConfig, UpdateConfigRequest, DescribeConfigResponse
+from .config_models import DescribeConfigResponse, JaiConfig, UpdateConfigRequest

packages/jupyter-ai/jupyter_ai/config/config_models.py

@@ -1,6 +1,8 @@
-from pydantic import BaseModel, field_validator
 from typing import Any, Optional
 
+from pydantic import BaseModel, field_validator
+
+
 class JaiConfig(BaseModel):
     """
     Pydantic model that serializes and validates the Jupyter AI config.

packages/jupyter-ai/jupyter_ai/config_manager.py

@@ -2,9 +2,9 @@
 import logging
 import os
 import time
+from copy import deepcopy
 from typing import Optional, Union
 
-from copy import deepcopy
 from deepmerge import always_merger
 from jupyter_ai_magics.utils import (
     AnyProvider,
@@ -13,11 +13,12 @@
     get_em_provider,
     get_lm_provider,
 )
-from .config import JaiConfig, DescribeConfigResponse, UpdateConfigRequest
 from jupyter_core.paths import jupyter_data_dir
 from traitlets import Integer, Unicode
 from traitlets.config import Configurable
 
+from .config import DescribeConfigResponse, JaiConfig, UpdateConfigRequest
+
 Logger = Union[logging.Logger, logging.LoggerAdapter]
 
 # default path to config
@@ -26,6 +27,7 @@
 # default no. of spaces to use when formatting config
 DEFAULT_INDENTATION_DEPTH = 4
 
+
 class AuthError(Exception):
     pass
 
@@ -62,7 +64,7 @@ def remove_none_entries(d: dict):
     Returns a deep copy of the given dictionary that excludes all top-level
     entries whose value is `None`.
     """
-    d = { k: deepcopy(d[k]) for k in d if d[k] is not None }
+    d = {k: deepcopy(d[k]) for k in d if d[k] is not None}
     return d
 
 
@@ -105,7 +107,6 @@ class ConfigManager(Configurable):
     `self._config`.
     """
 
-
     def __init__(
         self,
         log: Logger,
@@ -161,10 +162,11 @@ def _init_config(self):
         # configuration on init.
         if self._defaults:
             existing_config_args = self._read_config().model_dump()
-            merged_config_args = always_merger.merge(existing_config_args, self._defaults)
+            merged_config_args = always_merger.merge(
+                existing_config_args, self._defaults
+            )
             merged_config = JaiConfig(**merged_config_args)
             self._write_config(merged_config)
-    
 
     def _process_existing_config(self):
         """

packages/jupyter-ai/jupyter_ai/handlers.py

@@ -6,13 +6,11 @@
 from tornado import web
 from tornado.web import HTTPError
 
+from .config import UpdateConfigRequest
 from .models import (
     ListProvidersEntry,
     ListProvidersResponse,
 )
-from .config import (
-    UpdateConfigRequest
-)
 
 if TYPE_CHECKING:
     from jupyter_ai_magics.embedding_providers import BaseEmbeddingsProvider

packages/jupyter-ai/jupyter_ai/models.py

@@ -36,4 +36,3 @@ class IndexedDir(BaseModel):
 
 class IndexMetadata(BaseModel):
     dirs: list[IndexedDir]
-

packages/jupyter-ai/jupyter_ai/personas/base_persona.py

@@ -352,7 +352,7 @@ def process_attachments(self, message: Message) -> Optional[str]:
         """
         Process file attachments in the message and return their content as a string.
         """
-        
+
         if not message.attachments:
             return None
 
@@ -363,22 +363,22 @@ def process_attachments(self, message: Message) -> Optional[str]:
             try:
                 # Try to resolve attachment using multiple strategies
                 file_path = self.resolve_attachment_to_path(attachment_id)
-                
+
                 if not file_path:
-                    self.log.warning(f"Could not resolve attachment ID: {attachment_id}")
+                    self.log.warning(
+                        f"Could not resolve attachment ID: {attachment_id}"
+                    )
                     continue
-                
+
                 # Read the file content
-                with open(file_path, "r", encoding="utf-8") as f:
+                with open(file_path, encoding="utf-8") as f:
                     file_content = f.read()
-                
+
                 # Get relative path for display
                 rel_path = os.path.relpath(file_path, self.get_workspace_dir())
-                
+
                 # Add file content with header
-                context_parts.append(
-                    f"File: {rel_path}\n```\n{file_content}\n```"
-                )
+                context_parts.append(f"File: {rel_path}\n```\n{file_content}\n```")
 
             except Exception as e:
                 self.log.warning(f"Failed to read attachment {attachment_id}: {e}")
@@ -393,34 +393,34 @@ def resolve_attachment_to_path(self, attachment_id: str) -> Optional[str]:
         """
         Resolve an attachment ID to its file path using multiple strategies.
         """
-        
+
         try:
             attachment_data = self.ychat.get_attachments().get(attachment_id)
-            
+
             if attachment_data and isinstance(attachment_data, dict):
                 # If attachment has a 'value' field with filename
-                if 'value' in attachment_data:
-                    filename = attachment_data['value']
-                    
+                if "value" in attachment_data:
+                    filename = attachment_data["value"]
+
                     # Try relative to workspace directory
                     workspace_path = os.path.join(self.get_workspace_dir(), filename)
                     if os.path.exists(workspace_path):
                         return workspace_path
-                    
+
                     # Try as absolute path
                     if os.path.exists(filename):
                         return filename
-                        
+
             return None
-            
+
         except Exception as e:
             self.log.error(f"Failed to resolve attachment {attachment_id}: {e}")
             return None
-        
+
     def shutdown(self) -> None:
         """
         Shuts the persona down. This method should:
-         
+
         - Halt all background tasks started by this instance.
         - Remove the persona from the chat awareness
 

packages/jupyter-ai/jupyter_ai/personas/jupyternaut/jupyternaut.py

@@ -64,4 +64,3 @@ def build_runnable(self) -> Any:
         )
 
         return runnable
-