refactor(ark): use arkruntime.WithResponsesExtraFields

Remove local JSON merge; map WithExtraFields to SDK requestOption. Update replace pseudo-version for volcengine PR.

Made-with: Cursor

Author: jesse-engineer

components/model/ark/go.mod

@@ -50,4 +50,4 @@ require (
 	gopkg.in/yaml.v3 v3.0.1 // indirect
 )
 
-replace github.com/volcengine/volcengine-go-sdk => github.com/jesse-engineer/volcengine-go-sdk v1.2.26-0.20260428150644-cf9db789ee32
+replace github.com/volcengine/volcengine-go-sdk => github.com/jesse-engineer/volcengine-go-sdk v1.2.26-0.20260428151814-5ca1476bfea4

components/model/ark/go.sum

@@ -66,8 +66,8 @@ github.com/goph/emperror v0.17.2/go.mod h1:+ZbQ+fUNO/6FNiUo0ujtMjhgad9Xa6fQL9KhH
 github.com/gopherjs/gopherjs v1.17.2 h1:fQnZVsXk8uxXIStYb0N4bGk7jeyTalG/wsZjQ25dO0g=
 github.com/gopherjs/gopherjs v1.17.2/go.mod h1:pRRIvn/QzFLrKfvEz3qUuEhtE/zLCWfreZ6J5gM2i+k=
 github.com/hpcloud/tail v1.0.0/go.mod h1:ab1qPbhIpdTxEkNHXyeSf5vhxWSCs/tWer42PpOxQnU=
-github.com/jesse-engineer/volcengine-go-sdk v1.2.26-0.20260428150644-cf9db789ee32 h1:MmTysoe+4IMf8JpHkDZg2UF3sN5nt976sNLKXMoI6eg=
-github.com/jesse-engineer/volcengine-go-sdk v1.2.26-0.20260428150644-cf9db789ee32/go.mod h1:oxoVo+A17kvkwPkIeIHPVLjSw7EQAm+l/Vau1YGHN+A=
+github.com/jesse-engineer/volcengine-go-sdk v1.2.26-0.20260428151814-5ca1476bfea4 h1:dujwYehwF2FdlnMltvQqaAdLdXB56ZICvzAUPfKz0sc=
+github.com/jesse-engineer/volcengine-go-sdk v1.2.26-0.20260428151814-5ca1476bfea4/go.mod h1:oxoVo+A17kvkwPkIeIHPVLjSw7EQAm+l/Vau1YGHN+A=
 github.com/jmespath/go-jmespath v0.4.0 h1:BEgLn5cpjn8UN1mAw4NjwDrS35OdebyEtFe+9YPoQUg=
 github.com/jmespath/go-jmespath v0.4.0/go.mod h1:T8mJZnbsbmF+m6zOOFylbeCJqk5+pHWvzYPziyZiYoo=
 github.com/jmespath/go-jmespath/internal/testify v1.5.1 h1:shLQSRRSCCPj3f2gpwzGwWFoC7ycTf1rcQZHOlsJ6N8=

components/model/ark/option.go

@@ -158,8 +158,8 @@ func WithEnableReasoningContentPassback(enable bool) model.Option {
 }
 
 // WithExtraFields sets extra top-level JSON fields on the Responses API request body.
-// Merged after inputs are built and after multimodal preprocessing; extra keys override existing keys.
-// Requires volcengine-go-sdk with PreprocessResponsesRequest, CreateResponsesFromJSON, CreateResponsesStreamFromJSON.
+// Merged inside volcengine-go-sdk after the request is built and multimodal inputs are preprocessed;
+// keys in extraFields override existing keys. See arkruntime.WithResponsesExtraFields.
 func WithExtraFields(extraFields map[string]any) model.Option {
 	return model.WrapImplSpecificOptFn(func(o *arkOptions) {
 		o.extraFields = extraFields

components/model/ark/option_test.go

@@ -46,3 +46,9 @@ func TestOptions(t *testing.T) {
 	assert.Equal(t, cacheOpt, *opt.cache)
 	assert.Equal(t, arkModel.ThinkingTypeEnabled, opt.thinking.Type)
 }
+
+func TestWithExtraFieldsOption(t *testing.T) {
+	fields := map[string]any{"k": "v"}
+	opt := model.GetImplSpecificOptions(&arkOptions{}, WithExtraFields(fields))
+	assert.Equal(t, fields, opt.extraFields)
+}

components/model/ark/responses_api.go

@@ -35,7 +35,6 @@ import (
 	arkModel "github.com/volcengine/volcengine-go-sdk/service/arkruntime/model"
 	"github.com/volcengine/volcengine-go-sdk/service/arkruntime/model/responses"
 	"github.com/volcengine/volcengine-go-sdk/service/arkruntime/utils"
-	"github.com/volcengine/volcengine-go-sdk/volcengine"
 )
 
 type Thinking = arkModel.Thinking
@@ -949,35 +948,31 @@ func (cm *ResponsesAPIChatModel) getOptions(opts []model.Option) (*model.Options
 	return options, arkOpts, nil
 }
 
+func extraFieldsToIface(m map[string]any) map[string]interface{} {
+	if len(m) == 0 {
+		return nil
+	}
+	out := make(map[string]interface{}, len(m))
+	for k, v := range m {
+		out[k] = v
+	}
+	return out
+}
+
 func (cm *ResponsesAPIChatModel) invokeCreateResponses(ctx context.Context, responseReq *responses.ResponsesRequest, spec *arkOptions) (*responses.ResponseObject, error) {
 	h := arkruntime.WithCustomHeaders(spec.customHeaders)
 	if len(spec.extraFields) == 0 {
 		return cm.client.CreateResponses(ctx, responseReq, h)
 	}
-	if err := cm.client.PreprocessResponsesRequest(ctx, responseReq); err != nil {
-		return nil, err
-	}
-	raw, err := mergeResponsesRequestExtraJSON(responseReq, spec.extraFields)
-	if err != nil {
-		return nil, err
-	}
-	return cm.client.CreateResponsesFromJSON(ctx, responseReq.Model, raw, h)
+	return cm.client.CreateResponses(ctx, responseReq, h, arkruntime.WithResponsesExtraFields(extraFieldsToIface(spec.extraFields)))
 }
 
 func (cm *ResponsesAPIChatModel) invokeCreateResponsesStream(ctx context.Context, responseReq *responses.ResponsesRequest, spec *arkOptions) (*utils.ResponsesStreamReader, error) {
 	h := arkruntime.WithCustomHeaders(spec.customHeaders)
 	if len(spec.extraFields) == 0 {
 		return cm.client.CreateResponsesStream(ctx, responseReq, h)
 	}
-	responseReq.Stream = volcengine.Bool(true)
-	if err := cm.client.PreprocessResponsesRequest(ctx, responseReq); err != nil {
-		return nil, err
-	}
-	raw, err := mergeResponsesRequestExtraJSON(responseReq, spec.extraFields)
-	if err != nil {
-		return nil, err
-	}
-	return cm.client.CreateResponsesStreamFromJSON(ctx, responseReq.Model, raw, h)
+	return cm.client.CreateResponsesStream(ctx, responseReq, h, arkruntime.WithResponsesExtraFields(extraFieldsToIface(spec.extraFields)))
 }
 
 func (cm *ResponsesAPIChatModel) toTools(tis []*schema.ToolInfo) ([]*responses.ResponsesTool, error) {