Skip to content
17 changes: 16 additions & 1 deletion backend/api/handler/coze/config_service.go

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

152 changes: 102 additions & 50 deletions backend/api/model/admin/config/config.go

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

5 changes: 2 additions & 3 deletions backend/bizpkg/config/knowledge/knowledge.go
Original file line number Diff line number Diff line change
Expand Up @@ -82,9 +82,8 @@ func getKnowledgeConfigurationFromOldConfig() *config.KnowledgeConfig {
APIType: envkey.GetStringD("ARK_EMBEDDING_API_TYPE", "text_api"),
},
Openai: &config.OpenAIConnInfo{
ByAzure: envkey.GetBoolD("OPENAI_EMBEDDING_BY_AZURE", false),
APIVersion: envkey.GetString("OPENAI_EMBEDDING_API_VERSION"),
RequestDims: envkey.GetI32D("OPENAI_EMBEDDING_REQUEST_DIMS", 1024),
ByAzure: envkey.GetBoolD("OPENAI_EMBEDDING_BY_AZURE", false),
APIVersion: envkey.GetString("OPENAI_EMBEDDING_API_VERSION"),
},

Gemini: &config.GeminiConnInfo{
Expand Down
17 changes: 16 additions & 1 deletion backend/bizpkg/llm/modelbuilder/ark.go
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ type arkModelBuilder struct {
cfg *config.Model
}

func newArkModelBuilder(cfg *config.Model) *arkModelBuilder {
func newArkModelBuilder(cfg *config.Model) Service {
return &arkModelBuilder{
cfg: cfg,
}
Expand Down Expand Up @@ -84,6 +84,21 @@ func (b *arkModelBuilder) Build(ctx context.Context, params *LLMParams) (ToolCal
chatModelConf.BaseURL = base.BaseURL
}

switch base.ThinkingType {
case config.ThinkingType_Enable:
chatModelConf.Thinking = &model.Thinking{
Type: model.ThinkingTypeEnabled,
}
case config.ThinkingType_Disable:
chatModelConf.Thinking = &model.Thinking{
Type: model.ThinkingTypeDisabled,
}
case config.ThinkingType_Auto:
chatModelConf.Thinking = &model.Thinking{
Type: model.ThinkingTypeAuto,
}
}

arkConn := b.cfg.Connection.Ark
if arkConn != nil {
chatModelConf.Region = arkConn.Region
Expand Down
13 changes: 12 additions & 1 deletion backend/bizpkg/llm/modelbuilder/claude.go
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@ type claudeModelBuilder struct {
cfg *config.Model
}

func newClaudeModelBuilder(cfg *config.Model) *claudeModelBuilder {
func newClaudeModelBuilder(cfg *config.Model) Service {
return &claudeModelBuilder{
cfg: cfg,
}
Expand Down Expand Up @@ -71,6 +71,17 @@ func (c *claudeModelBuilder) Build(ctx context.Context, params *LLMParams) (Tool
conf.BaseURL = &base.BaseURL
}

switch base.ThinkingType {
case config.ThinkingType_Enable:
conf.Thinking = &claude.Thinking{
Enable: true,
}
case config.ThinkingType_Disable:
conf.Thinking = &claude.Thinking{
Enable: false,
}
}

c.applyParamsToChatModelConfig(conf, params)

return claude.NewChatModel(ctx, conf)
Expand Down
2 changes: 1 addition & 1 deletion backend/bizpkg/llm/modelbuilder/deepseek.go
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,7 @@ type deepseekModelBuilder struct {
cfg *config.Model
}

func newDeepseekModelBuilder(cfg *config.Model) *deepseekModelBuilder {
func newDeepseekModelBuilder(cfg *config.Model) Service {
return &deepseekModelBuilder{
cfg: cfg,
}
Expand Down
Loading
Loading