feat(model/ark): Refactor callback handling for Responses API (#649)

Moves the callback initialization logic from the generic ChatModel to the specific ResponsesAPIChatModel.

This change ensures that EnsureRunInfo is only called for the Responses API flow, where it is actually needed. The GetType method was also added to ResponsesAPIChatModel to support this.

Author: zhuangjie1125

components/model/ark/chatmodel.go

@@ -27,8 +27,6 @@ import (
 	"github.com/volcengine/volcengine-go-sdk/service/arkruntime"
 	"github.com/volcengine/volcengine-go-sdk/service/arkruntime/model"
 
-	"github.com/cloudwego/eino/callbacks"
-	"github.com/cloudwego/eino/components"
 	fmodel "github.com/cloudwego/eino/components/model"
 	"github.com/cloudwego/eino/schema"
 )
@@ -389,9 +387,6 @@ type CacheInfo struct {
 
 func (cm *ChatModel) Generate(ctx context.Context, in []*schema.Message, opts ...fmodel.Option) (
 	outMsg *schema.Message, err error) {
-
-	ctx = callbacks.EnsureRunInfo(ctx, cm.GetType(), components.ComponentOfChatModel)
-
 	ok, err := cm.callByResponsesAPI(opts...)
 	if err != nil {
 		return nil, err
@@ -406,8 +401,6 @@ func (cm *ChatModel) Generate(ctx context.Context, in []*schema.Message, opts ..
 func (cm *ChatModel) Stream(ctx context.Context, in []*schema.Message, opts ...fmodel.Option) (
 	outStream *schema.StreamReader[*schema.Message], err error) {
 
-	ctx = callbacks.EnsureRunInfo(ctx, cm.GetType(), components.ComponentOfChatModel)
-
 	ok, err := cm.callByResponsesAPI(opts...)
 	if err != nil {
 		return nil, err

components/model/ark/responses_api.go

@@ -28,6 +28,7 @@ import (
 
 	"github.com/bytedance/sonic"
 	"github.com/cloudwego/eino/callbacks"
+	"github.com/cloudwego/eino/components"
 	"github.com/cloudwego/eino/components/model"
 	"github.com/cloudwego/eino/schema"
 	"github.com/volcengine/volcengine-go-sdk/service/arkruntime"
@@ -219,8 +220,14 @@ type cacheConfig struct {
 	ExpireAt *int64
 }
 
+func (cm *ResponsesAPIChatModel) GetType() string {
+	return "ResponsesAPI"
+}
+
 func (cm *ResponsesAPIChatModel) Generate(ctx context.Context, input []*schema.Message,
 	opts ...model.Option) (outMsg *schema.Message, err error) {
+	ctx = callbacks.EnsureRunInfo(ctx, cm.GetType(), components.ComponentOfChatModel)
+
 	options, specOptions, err := cm.getOptions(opts)
 	if err != nil {
 		return nil, err
@@ -288,7 +295,7 @@ func (cm *ResponsesAPIChatModel) Generate(ctx context.Context, input []*schema.M
 
 func (cm *ResponsesAPIChatModel) Stream(ctx context.Context, input []*schema.Message,
 	opts ...model.Option) (outStream *schema.StreamReader[*schema.Message], err error) {
-
+	ctx = callbacks.EnsureRunInfo(ctx, cm.GetType(), components.ComponentOfChatModel)
 	options, specOptions, err := cm.getOptions(opts)
 	if err != nil {
 		return nil, err