[REL-11697] lint

Author: andrewklatzke

packages/sdk/server-ai/src/ldai/agent_graph/__init__.py

@@ -150,14 +150,14 @@ def terminal_nodes(self) -> List[AgentGraphNode]:
             if len(self.get_child_nodes(node.get_key())) == 0
         ]
 
-    def root(self) -> AgentGraphNode:
+    def root(self) -> Optional[AgentGraphNode]:
         """Get the root node of the graph."""
         return self._nodes.get(self._agent_graph.root_config_key)
 
     def traverse(
         self,
         fn: Callable[["AgentGraphNode", Dict[str, Any]], Any],
-        execution_context: Dict[str, Any] = None,
+        execution_context: Optional[Dict[str, Any]] = None,
     ) -> Any:
         """Traverse from the root down to terminal nodes, visiting nodes in order of depth.
         Nodes with the longest paths from the root (deepest nodes) will always be visited last."""
@@ -205,9 +205,8 @@ def traverse(
     def reverse_traverse(
         self,
         fn: Callable[["AgentGraphNode", Dict[str, Any]], Any],
-        execution_context: Dict[str, Any] = None,
+        execution_context: Optional[Dict[str, Any]] = None,
     ) -> Any:
-    
         """Traverse from terminal nodes up to the root, visiting nodes level by level.
         The root node will always be visited last, even if multiple paths converge at it."""
         if execution_context is None:
@@ -255,4 +254,3 @@ def reverse_traverse(
                 )
 
         return execution_context[self._agent_graph.root_config_key]
-