diff --git a/libs/prebuilt/langgraph/prebuilt/chat_agent_executor.py b/libs/prebuilt/langgraph/prebuilt/chat_agent_executor.py index e05b418b5..2a26eb42f 100644 --- a/libs/prebuilt/langgraph/prebuilt/chat_agent_executor.py +++ b/libs/prebuilt/langgraph/prebuilt/chat_agent_executor.py @@ -585,31 +585,31 @@ class _AgentBuilder: """Check if agent has tools enabled.""" return len(self._tool_classes) > 0 - def _get_model_paths(self) -> list[str]: - """Get possible paths from model node.""" - paths = [] + def _get_model_edges(self) -> list[str]: + """Get possible edge destinations from model node.""" + edges = [] - # If post_model_hook exists, we don't add paths here - we use direct edge instead + # If post_model_hook exists, we don't add edges here - we use direct edge instead if not self.post_model_hook: if self._has_tools(): - paths.append("tools") + edges.append("tools") if self.response_format: - paths.append("generate_structured_response") + edges.append("generate_structured_response") if not self._has_tools() and not self.response_format: - paths.append(END) + edges.append(END) - return paths + return edges - def _get_post_model_hook_paths(self) -> list[str]: - """Get possible paths from post_model_hook node.""" - paths = [self._get_entry_point()] + def _get_post_model_hook_edges(self) -> list[str]: + """Get possible edge destinations from post_model_hook node.""" + edges = [self._get_entry_point()] if self._has_tools(): - paths.append("tools") + edges.append("tools") if self.response_format: - paths.append("generate_structured_response") + edges.append("generate_structured_response") else: - paths.append(END) - return paths + edges.append(END) + return edges def build(self) -> StateGraph: """Build the agent workflow graph (uncompiled).""" @@ -651,15 +651,15 @@ class _AgentBuilder: # Direct edge from model node to post_model_hook when post_model_hook exists workflow.add_edge("agent", "post_model_hook") # Post-model hook conditional edges - post_hook_paths = self._get_post_model_hook_paths() + post_hook_edges = self._get_post_model_hook_edges() workflow.add_conditional_edges( - "post_model_hook", self.post_model_hook_router, path_map=post_hook_paths + "post_model_hook", self.post_model_hook_router, path_map=post_hook_edges ) # type: ignore[arg-type] else: # Conditional edges from model node when no post_model_hook model_router = self.create_model_router() - model_paths = self._get_model_paths() - workflow.add_conditional_edges("agent", model_router, path_map=model_paths) # type: ignore[arg-type] + model_edges = self._get_model_edges() + workflow.add_conditional_edges("agent", model_router, path_map=model_edges) # type: ignore[arg-type] # Tools edges if self._has_tools():