fix: correct token calculation error caused by ctxcache init (#559)

backend/domain/workflow/internal/compose/designate_option.go

@@ -32,7 +32,6 @@ import (
 	"github.com/coze-dev/coze-studio/backend/domain/workflow/internal/execute"
 	"github.com/coze-dev/coze-studio/backend/domain/workflow/internal/nodes"
 	"github.com/coze-dev/coze-studio/backend/domain/workflow/internal/nodes/llm"
-	"github.com/coze-dev/coze-studio/backend/pkg/ctxcache"
 	"github.com/coze-dev/coze-studio/backend/pkg/lang/ptr"
 )
 
@@ -47,8 +46,6 @@ func (r *WorkflowRunner) designateOptions(ctx context.Context) (context.Context,
 		sw           = r.streamWriter
 	)
 
-	const tokenCallbackKey = "token_callback_key"
-
 	if wb.AppID != nil && exeCfg.AppID == nil {
 		exeCfg.AppID = wb.AppID
 	}
@@ -124,12 +121,6 @@ func (r *WorkflowRunner) designateOptions(ctx context.Context) (context.Context,
 		opts = append(opts, einoCompose.WithCheckPointID(strconv.FormatInt(executeID, 10)))
 	}
 
-	if !ctxcache.HasKey(ctx, tokenCallbackKey) {
-		opts = append(opts, einoCompose.WithCallbacks(execute.GetTokenCallbackHandler()))
-		ctx = ctxcache.Init(ctx)
-		ctxcache.Store(ctx, tokenCallbackKey, true)
-	}
-
 	return ctx, opts, nil
 }
 

backend/domain/workflow/service/global_handler.go

@@ -0,0 +1,26 @@
+/*
+ * Copyright 2025 coze-dev Authors
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package service
+
+import (
+	"github.com/cloudwego/eino/callbacks"
+	"github.com/coze-dev/coze-studio/backend/domain/workflow/internal/execute"
+)
+
+func GetTokenCallbackHandler() callbacks.Handler {
+	return execute.GetTokenCallbackHandler()
+}