chore: delint

GoogleCloudPlatform · Nov 15, 2024 · b64bdc7 · b64bdc7
1 parent 5c94393
commit b64bdc7
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/llm_demo/orchestrator/langgraph/langgraph_orchestrator.py b/llm_demo/orchestrator/langgraph/langgraph_orchestrator.py
@@ -160,7 +160,9 @@ def user_session_reset(self, session: dict[str, Any], uuid: str):
         # Reset graph checkpointer
         checkpoint = empty_checkpoint()
         config = self.get_config(uuid)
-        self._checkpointer.put(config=config, checkpoint=checkpoint, metadata={}, new_versions={})
+        self._checkpointer.put(
+            config=config, checkpoint=checkpoint, metadata={}, new_versions={}
+        )
 
         # Update state with message history
         self._langgraph_app.update_state(config, {"messages": history})

diff --git a/llm_demo/orchestrator/langgraph/react_graph.py b/llm_demo/orchestrator/langgraph/react_graph.py
@@ -110,7 +110,9 @@ async def acall_model(state: UserState, config: RunnableConfig):
                     res = AIMessage(
                         content="suggesting a tool call",
                         tool_calls=[
-                            ToolCall(id=str(uuid.uuid4()), name=action, args=action_input)
+                            ToolCall(
+                                id=str(uuid.uuid4()), name=action, args=action_input
+                            )
                         ],
                     )
             except Exception as e: