fix: set tools executed if them have not reported error (#253)

Author: meguminnnnnnnnn

compose/tool_node.go

@@ -258,8 +258,10 @@ func runToolCallTaskByInvoke(ctx context.Context, task *toolCallTask, opts ...to
 	})
 
 	ctx = setToolCallInfo(ctx, &toolCallInfo{toolCallID: task.callID})
-	task.executed = true
 	task.output, task.err = task.r.Invoke(ctx, task.arg, opts...)
+	if task.err == nil {
+		task.executed = true
+	}
 }
 
 func runToolCallTaskByStream(ctx context.Context, task *toolCallTask, opts ...tool.Option) {
@@ -270,8 +272,10 @@ func runToolCallTaskByStream(ctx context.Context, task *toolCallTask, opts ...to
 	})
 
 	ctx = setToolCallInfo(ctx, &toolCallInfo{toolCallID: task.callID})
-	task.executed = true
 	task.sOutput, task.err = task.r.Stream(ctx, task.arg, opts...)
+	if task.err == nil {
+		task.executed = true
+	}
 }
 
 func sequentialRunToolCall(ctx context.Context, run func(ctx2 context.Context, callTask *toolCallTask, opts ...tool.Option), tasks []toolCallTask, opts ...tool.Option) {

compose/tool_node_test.go

@@ -720,7 +720,7 @@ func TestToolRerun(t *testing.T) {
 	r, err := g.Compile(ctx, WithCheckPointStore(&inMemoryStore{m: map[string][]byte{}}))
 	assert.NoError(t, err)
 
-	_, err = r.Invoke(ctx, &schema.Message{Role: schema.Assistant, ToolCalls: tc}, WithCheckPointID("1"))
+	_, err = r.Stream(ctx, &schema.Message{Role: schema.Assistant, ToolCalls: tc}, WithCheckPointID("1"))
 	info, ok := ExtractInterruptInfo(err)
 	assert.True(t, ok)
 	assert.Equal(t, []string{"tool node"}, info.RerunNodes)