feat: refactor AI engine to support new models and APIs

- Added a new configuration file with default settings for model, API, and formatting options.
- Updated `go.mod` to include new dependencies and remove unused ones.
- Refactored AI engine to use a new `Model` interface and simplified API key handling.
- Added support for new API endpoints and models, including `siliconflow` and `ark`.
- Removed unnecessary imports and streamlined the configuration handling in the AI engine.
- Updated the `vim` command in the settings editor to remove redundant options.
- Modified the auto-coder UI to display model aliases instead of the model name.

Signed-off-by: codiing-hui <wecoding@yeah.net>

Author: coding-hui

go.mod

@@ -37,12 +37,14 @@ require (
 	github.com/spf13/cobra v1.8.1
 	github.com/spf13/pflag v1.0.5
 	github.com/stretchr/testify v1.10.0
+	github.com/volcengine/volcengine-go-sdk v1.0.181
 	gopkg.in/yaml.v3 v3.0.1
 	k8s.io/klog/v2 v2.130.1
 	modernc.org/sqlite v1.34.4
 )
 
 require (
+	dario.cat/mergo v1.0.1 // indirect
 	github.com/Azure/go-ansiterm v0.0.0-20250102033503-faa5f7b0171c // indirect
 	github.com/Masterminds/goutils v1.1.1 // indirect
 	github.com/Masterminds/semver/v3 v3.3.1 // indirect
@@ -67,8 +69,8 @@ require (
 	github.com/h2non/filetype v1.1.3 // indirect
 	github.com/hashicorp/golang-lru/v2 v2.0.7 // indirect
 	github.com/huandu/xstrings v1.5.0 // indirect
-	github.com/imdario/mergo v0.3.11 // indirect
 	github.com/inconshreveable/mousetrap v1.1.0 // indirect
+	github.com/jmespath/go-jmespath v0.4.0 // indirect
 	github.com/json-iterator/go v1.1.12 // indirect
 	github.com/mattn/go-colorable v0.1.13 // indirect
 	github.com/mattn/go-localereader v0.0.1 // indirect
@@ -99,6 +101,7 @@ require (
 	github.com/shopspring/decimal v1.4.0 // indirect
 	github.com/sirupsen/logrus v1.9.3 // indirect
 	github.com/spf13/cast v1.7.1 // indirect
+	github.com/volcengine/volc-sdk-golang v1.0.23 // indirect
 	github.com/yargevad/filepathx v1.0.0 // indirect
 	github.com/yuin/goldmark v1.7.4 // indirect
 	github.com/yuin/goldmark-emoji v1.0.3 // indirect
@@ -109,7 +112,6 @@ require (
 	golang.org/x/sys v0.30.0 // indirect
 	golang.org/x/term v0.29.0 // indirect
 	golang.org/x/text v0.22.0 // indirect
-	golang.org/x/tools v0.30.0 // indirect
 	gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c // indirect
 	gopkg.in/yaml.v2 v2.4.0 // indirect
 	modernc.org/gc/v3 v3.0.0-20240107210532-573471604cb6 // indirect

go.sum

@@ -4,14 +4,10 @@ import (
 	"context"
 	"encoding/json"
 	"fmt"
-	"os"
-	"os/exec"
 	"strings"
 
-	"github.com/caarlos0/go-shellwords"
 	"github.com/coding-hui/common/util/slices"
 	"github.com/coding-hui/wecoding-sdk-go/services/ai/llms"
-	"github.com/coding-hui/wecoding-sdk-go/services/ai/llms/openai"
 
 	"github.com/coding-hui/ai-terminal/internal/convo"
 	"github.com/coding-hui/ai-terminal/internal/errbook"
@@ -29,7 +25,7 @@ type Engine struct {
 	channel chan StreamCompletionOutput
 
 	convoStore convo.Store
-	model      *openai.Model
+	model      Model
 
 	Config *options.Config
 }
@@ -160,8 +156,8 @@ func (e *Engine) callOptions(streamingFunc ...func(ctx context.Context, chunk []
 	if len(streamingFunc) > 0 && streamingFunc[0] != nil {
 		opts = append(opts, llms.WithStreamingFunc(streamingFunc[0]))
 	}
-	opts = append(opts, llms.WithModel(e.Config.Model))
-	opts = append(opts, llms.WithMaxLength(e.Config.MaxInputChars))
+	opts = append(opts, llms.WithModel(e.Config.CurrentModel.Name))
+	opts = append(opts, llms.WithMaxLength(e.Config.CurrentModel.MaxChars))
 	opts = append(opts, llms.WithTemperature(e.Config.Temperature))
 	opts = append(opts, llms.WithTopP(e.Config.TopP))
 	opts = append(opts, llms.WithTopK(e.Config.TopK))
@@ -199,39 +195,6 @@ func (e *Engine) appendAssistantMessage(content string) {
 	}
 }
 
-func ensureApiKey(api options.API) (string, error) {
-	key := api.APIKey
-	if key == "" && api.APIKeyEnv != "" && api.APIKeyCmd == "" {
-		key = os.Getenv(api.APIKeyEnv)
-	}
-	if key == "" && api.APIKeyCmd != "" {
-		args, err := shellwords.Parse(api.APIKeyCmd)
-		if err != nil {
-			return "", errbook.Wrap("Failed to parse api-key-cmd", err)
-		}
-		out, err := exec.Command(args[0], args[1:]...).CombinedOutput() //nolint:gosec
-		if err != nil {
-			return "", errbook.Wrap("Cannot exec api-key-cmd", err)
-		}
-		key = strings.TrimSpace(string(out))
-	}
-	if key != "" {
-		return key, nil
-	}
-	return "", errbook.Wrap(
-		fmt.Sprintf(
-			"%[1]s required; set the environment variable %[1]s or update %[2]s through %[3]s.",
-			console.StderrStyles().InlineCode.Render(api.APIKeyEnv),
-			console.StderrStyles().InlineCode.Render("Config.yaml"),
-			console.StderrStyles().InlineCode.Render("ai Config"),
-		),
-		errbook.NewUserErrorf(
-			"You can grab one at %s.",
-			console.StderrStyles().Link.Render(api.BaseURL),
-		),
-	)
-}
-
 func convert(msg llms.ChatMessage) llms.MessageContent {
 	return llms.MessageContent{
 		Role:  msg.GetType(),

internal/ai/ai.go

@@ -1,14 +1,14 @@
 package ai
 
 import (
-	"fmt"
+	"github.com/volcengine/volcengine-go-sdk/service/arkruntime"
 
 	"github.com/coding-hui/ai-terminal/internal/convo"
 	"github.com/coding-hui/ai-terminal/internal/errbook"
 	"github.com/coding-hui/ai-terminal/internal/options"
-	"github.com/coding-hui/ai-terminal/internal/ui/console"
 
 	"github.com/coding-hui/wecoding-sdk-go/services/ai/llms/openai"
+	"github.com/coding-hui/wecoding-sdk-go/services/ai/llms/volcengine"
 )
 
 type Option func(*Engine)
@@ -53,67 +53,39 @@ func applyOptions(engineOpts ...Option) (engine *Engine, err error) {
 		}
 	}
 
-	var api options.API
-	mod, ok := cfg.Models[cfg.Model]
-	if !ok {
-		if cfg.API == "" {
-			return nil, errbook.Wrap(
-				fmt.Sprintf(
-					"model %s is not in the settings file.",
-					console.StderrStyles().InlineCode.Render(cfg.Model),
-				),
-				errbook.NewUserErrorf(
-					"Please specify an API endpoint with %s or configure the model in the settings: %s",
-					console.StderrStyles().InlineCode.Render("--api"),
-					console.StderrStyles().InlineCode.Render("ai -s"),
-				),
-			)
-		}
-		mod.Name = cfg.Model
-		mod.API = cfg.API
-		mod.MaxChars = cfg.MaxInputChars
-	}
-	if cfg.API != "" {
-		mod.API = cfg.API
-	}
-	for _, a := range cfg.APIs {
-		if mod.API == a.Name {
-			api = a
-			break
-		}
-	}
-	if api.Name == "" {
-		eps := make([]string, 0)
-		for _, a := range cfg.APIs {
-			eps = append(eps, console.StderrStyles().InlineCode.Render(a.Name))
-		}
-		return nil, errbook.Wrap(
-			fmt.Sprintf(
-				"The API endpoint %s is not configured.",
-				console.StderrStyles().InlineCode.Render(cfg.API),
-			),
-			errbook.NewUserErrorf(
-				"Your configured API endpoints are: %s",
-				eps,
-			),
-		)
-	}
-
-	key, err := ensureApiKey(api)
+	cfg.CurrentModel, err = cfg.GetModel(cfg.Model)
 	if err != nil {
 		return nil, err
 	}
 
-	var opts []openai.Option
-	opts = append(opts,
-		openai.WithModel(mod.Name),
-		openai.WithBaseURL(api.BaseURL),
-		openai.WithToken(key),
-	)
-	engine.model, err = openai.New(opts...)
+	cfg.CurrentAPI, err = cfg.GetAPI(cfg.API)
 	if err != nil {
 		return nil, err
 	}
 
+	mod, api := cfg.CurrentModel, cfg.CurrentAPI
+	switch api.Name {
+	case ModelTypeARK:
+		engine.model, err = volcengine.NewClientWithApiKey(
+			cfg.CurrentAPI.APIKey,
+			arkruntime.WithBaseUrl(api.BaseURL),
+			arkruntime.WithRegion(api.Region),
+			arkruntime.WithTimeout(api.Timeout),
+			arkruntime.WithRetryTimes(api.RetryTimes),
+		)
+		if err != nil {
+			return nil, err
+		}
+	default:
+		engine.model, err = openai.New(
+			openai.WithModel(mod.Name),
+			openai.WithBaseURL(api.BaseURL),
+			openai.WithToken(api.APIKey),
+		)
+		if err != nil {
+			return nil, err
+		}
+	}
+
 	return engine, nil
 }

internal/ai/ai_options.go

@@ -0,0 +1,16 @@
+package ai
+
+import (
+	"context"
+
+	"github.com/coding-hui/wecoding-sdk-go/services/ai/llms"
+)
+
+const (
+	ModelTypeOpenAI = "openai"
+	ModelTypeARK    = "ark"
+)
+
+type Model interface {
+	GenerateContent(context.Context, []llms.MessageContent, ...llms.CallOption) (*llms.ContentResponse, error)
+}