Fix default condition name

This commit is contained in:
Nuno Campos
2024-03-31 18:56:38 -07:00
parent b668bf8eab
commit 0d2bbe85c6
2 changed files with 85 additions and 85 deletions
+1 -1
View File
@@ -136,7 +136,7 @@ class Graph:
if not isinstance(condition, Runnable):
condition = RunnableLambda(condition)
if condition.name is None:
condition.name = f"{start_key}_condition"
condition.name = "condition"
if condition.name in self.branches[start_key]:
raise ValueError(
f"Branch with name `{condition.name}` already exists for node "
+84 -84
View File
@@ -63,7 +63,7 @@
}
},
{
"id": "left_left_condition",
"id": "left_condition",
"type": "runnable",
"data": {
"id": [
@@ -72,7 +72,7 @@
"base",
"RunnableLambda"
],
"name": "left_condition"
"name": "condition"
}
}
],
@@ -97,20 +97,20 @@
},
{
"source": "left",
"target": "left_left_condition"
"target": "left_condition"
},
{
"source": "left_left_condition",
"source": "left_condition",
"target": "left",
"data": "left"
},
{
"source": "left_left_condition",
"source": "left_condition",
"target": "right",
"data": "right"
},
{
"source": "left_left_condition",
"source": "left_condition",
"target": "__end__",
"data": "__end__"
}
@@ -120,39 +120,39 @@
# ---
# name: test_conditional_entrypoint_graph.3
'''
+-----------+
| __start__ |
+-----------+
*
*
*
+------------------------+
| __start___should_start |
+------------------------+
*** **
* **
** **
+------+ **
| left | *
+------+ *
* *
* *
* *
+---------------------+ *
| left_left_condition | *
+---------------------+ *
* ***** *
* **** *
* *** *
** +-------+
* | right |
*** +-------+
* ***
*** *
* **
+---------+
| __end__ |
+---------+
+-----------+
| __start__ |
+-----------+
*
*
*
+------------------------+
| __start___should_start |
+------------------------+
*** **
* **
** **
+------+ *
| left | *
+------+ *
* *
* *
* *
+----------------+ *
| left_condition | *
+----------------+* *
* ***** *
* *** *
* *** *
** +-------+
* | right |
*** +-------+
* ***
*** *
* **
+---------+
| __end__ |
+---------+
'''
# ---
# name: test_conditional_entrypoint_graph_state
@@ -247,7 +247,7 @@
}
},
{
"id": "left_left_condition",
"id": "left_condition",
"type": "runnable",
"data": {
"id": [
@@ -256,7 +256,7 @@
"base",
"RunnableLambda"
],
"name": "left_condition"
"name": "condition"
}
}
],
@@ -281,20 +281,20 @@
},
{
"source": "left",
"target": "left_left_condition"
"target": "left_condition"
},
{
"source": "left_left_condition",
"source": "left_condition",
"target": "left",
"data": "left"
},
{
"source": "left_left_condition",
"source": "left_condition",
"target": "right",
"data": "right"
},
{
"source": "left_left_condition",
"source": "left_condition",
"target": "__end__",
"data": "__end__"
}
@@ -304,39 +304,39 @@
# ---
# name: test_conditional_entrypoint_graph_state.3
'''
+-----------+
| __start__ |
+-----------+
*
*
*
+------------------------+
| __start___should_start |
+------------------------+
*** **
* **
** **
+------+ **
| left | *
+------+ *
* *
* *
* *
+---------------------+ *
| left_left_condition | *
+---------------------+ *
* ***** *
* **** *
* *** *
** +-------+
* | right |
*** +-------+
* ***
*** *
* **
+---------+
| __end__ |
+---------+
+-----------+
| __start__ |
+-----------+
*
*
*
+------------------------+
| __start___should_start |
+------------------------+
*** **
* **
** **
+------+ *
| left | *
+------+ *
* *
* *
* *
+----------------+ *
| left_condition | *
+----------------+* *
* ***** *
* *** *
* *** *
** +-------+
* | right |
*** +-------+
* ***
*** *
* **
+---------+
| __end__ |
+---------+
'''
# ---
# name: test_conditional_graph
@@ -399,11 +399,11 @@
],
"edges": [
{
"source": "__start__",
"source": "tools",
"target": "agent"
},
{
"source": "tools",
"source": "__start__",
"target": "agent"
},
{
@@ -670,11 +670,11 @@
"target": 3
},
{
"source": "__start__",
"source": "tools",
"target": 2
},
{
"source": "tools",
"source": "__start__",
"target": 2
},
{
@@ -1020,11 +1020,11 @@
],
"edges": [
{
"source": "__start__",
"source": "tools",
"target": "agent"
},
{
"source": "tools",
"source": "__start__",
"target": "agent"
},
{