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

for more information, see https://pre-commit.ci

packages/jupyter-ai/jupyter_ai/callback_handlers/metadata.py

@@ -1,13 +1,14 @@
+import json
+
 from langchain_core.callbacks import BaseCallbackHandler
 from langchain_core.outputs import LLMResult
-import json
 
 
 def convert_to_serializable(obj):
     """Convert an object to a JSON serializable format"""
-    if hasattr(obj, 'dict') and callable(obj.dict):
+    if hasattr(obj, "dict") and callable(obj.dict):
         return obj.dict()
-    if hasattr(obj, '__dict__'):
+    if hasattr(obj, "__dict__"):
         return obj.__dict__
     return str(obj)
 
@@ -34,7 +35,7 @@ def on_llm_end(self, response: LLMResult, **kwargs) -> None:
             return
 
         metadata = response.generations[0][0].generation_info or {}
-        
+
         # Convert any non-serializable objects in metadata
         serializable_metadata = {}
         for key, value in metadata.items():
@@ -43,5 +44,5 @@ def on_llm_end(self, response: LLMResult, **kwargs) -> None:
                 serializable_metadata[key] = value
             except (TypeError, ValueError):
                 serializable_metadata[key] = convert_to_serializable(value)
-        
+
         self.jai_metadata = serializable_metadata

packages/jupyter-ai/jupyter_ai/models.py

@@ -1,5 +1,5 @@
-from typing import Any, Dict, List, Literal, Optional, Union
 import json
+from typing import Any, Dict, List, Literal, Optional, Union
 
 from jupyter_ai_magics import Persona
 from jupyter_ai_magics.providers import AuthStrategy, Field
@@ -128,7 +128,7 @@ class AgentStreamChunkMessage(BaseModel):
     chunk should override any metadata from previous chunks. See the docstring
     on `BaseAgentMessage.metadata` for information.
     """
-    
+
     @validator("metadata")
     def validate_metadata(cls, v):
         """Ensure metadata values are JSON serializable"""