lint

backend/app/api/runs.py

@@ -4,11 +4,10 @@
 import langsmith.client
 from fastapi import APIRouter, BackgroundTasks, HTTPException
 from fastapi.exceptions import RequestValidationError
-from pydantic import ValidationError
 from langchain_core.messages import AnyMessage
 from langchain_core.runnables import RunnableConfig
 from langsmith.utils import tracing_is_enabled
-from pydantic import BaseModel, Field
+from pydantic import BaseModel, Field, ValidationError
 from sse_starlette import EventSourceResponse
 
 from app.agent import agent, chat_retrieval, chatbot

backend/app/checkpoint.py

@@ -62,7 +62,7 @@ async def setup(self) -> None:
             pool = AsyncConnectionPool(
                 conninfo=conninfo,
                 kwargs={"autocommit": True, "prepare_threshold": 0},
-                open=False  # Don't open in constructor
+                open=False,  # Don't open in constructor
             )
             await pool.open()
 

backend/app/lifespan.py

@@ -5,6 +5,7 @@
 import orjson
 import structlog
 from fastapi import FastAPI
+
 from app.checkpoint import AsyncPostgresCheckpoint
 
 _pg_pool = None

backend/app/message_types.py

@@ -1,24 +1,25 @@
 from typing import Any
+
 from langchain_core.messages import (
     FunctionMessage,
     MessageLikeRepresentation,
     ToolMessage,
+    _message_from_dict,
 )
 from langgraph.graph.message import Messages, add_messages
-from langchain_core.messages import _message_from_dict
 from pydantic import Field
 
 
 class LiberalFunctionMessage(FunctionMessage):
-     content: Any = Field(default="")
+    content: Any = Field(default="")
 
 
 class LiberalToolMessage(ToolMessage):
     content: Any = Field(default="")
 
 
 def _convert_pydantic_dict_to_message(
-    data: MessageLikeRepresentation
+    data: MessageLikeRepresentation,
 ) -> MessageLikeRepresentation:
     """Convert a dictionary to a message object if it matches message format."""
     if (

backend/app/storage.py

@@ -126,7 +126,7 @@ async def get_thread_state(*, user_id: str, thread_id: str, assistant: Assistant
     )
     # Keep original format - return values as is
     values = state.values if state.values else None
-    
+
     return {
         "values": values,
         "next": state.next,
@@ -151,7 +151,7 @@ async def update_thread_state(
             }
         }
     )
-    
+
     # If current state is a dict (retrieval agent), maintain dict structure
     if current_state.values and isinstance(current_state.values, dict):
         if isinstance(values, dict):
@@ -161,8 +161,10 @@ async def update_thread_state(
             state_values = {**current_state.values, "messages": values}
     else:
         # For message-only states (tools_agent, chatbot), just use the messages
-        state_values = values if isinstance(values, dict) and "messages" in values else values
-
+        state_values = (
+            values if isinstance(values, dict) and "messages" in values else values
+        )
+
     await agent.aupdate_state(
         {
             "configurable": {