Internal refactor of create react-agent

2026-09-09 11:17:53 +02:00 · 2025-08-12 21:50:19 -04:00
parent d43eaf1f42
commit 2fed0e4852
3 changed files with 537 additions and 446 deletions
@@ -49,4 +49,14 @@ def test_react_agent_graph_structure(
        post_model_hook=post_model_hook,
        response_format=response_format,
    )
-    assert agent.get_graph().draw_mermaid(with_styles=False) == snapshot
+    try:
+        assert agent.get_graph().draw_mermaid(with_styles=False) == snapshot
+    except Exception as e:
+        raise ValueError(
+            "The graph structure has changed. Please update the snapshot."
+            "Configuration used:\n"
+            f"tools: {tools}, "
+            f"pre_model_hook: {pre_model_hook}, "
+            f"post_model_hook: {post_model_hook}, "
+            f"response_format: {response_format}"
+        ) from e