ben
/
coze-studio
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat(model): change env name from ARK_EMBEDDING_AK to ARK_EMBEDDING_A… (#482)

This commit is contained in:
IPender 2025-08-01 19:49:19 +08:00 committed by GitHub
parent afaaadec16
commit 4b03f8de78
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 13 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
backend/application/knowledge/init.go
View File

@ -335,8 +335,11 @@ func getEmbedding(ctx context.Context) (embedding.Embedder, error) {
var ( var (
arkEmbeddingBaseURL = os.Getenv("ARK_EMBEDDING_BASE_URL") arkEmbeddingBaseURL = os.Getenv("ARK_EMBEDDING_BASE_URL")
arkEmbeddingModel = os.Getenv("ARK_EMBEDDING_MODEL") arkEmbeddingModel = os.Getenv("ARK_EMBEDDING_MODEL")
arkEmbeddingAK = os.Getenv("ARK_EMBEDDING_AK") arkEmbeddingApiKey = os.Getenv("ARK_EMBEDDING_API_KEY")
arkEmbeddingDims = os.Getenv("ARK_EMBEDDING_DIMS") // deprecated: use ARK_EMBEDDING_API_KEY instead
// ARK_EMBEDDING_AK will be removed in the future
arkEmbeddingAK = os.Getenv("ARK_EMBEDDING_AK")
arkEmbeddingDims = os.Getenv("ARK_EMBEDDING_DIMS")
) )
dims, err := strconv.ParseInt(arkEmbeddingDims, 10, 64) dims, err := strconv.ParseInt(arkEmbeddingDims, 10, 64)
@ -345,7 +348,12 @@ func getEmbedding(ctx context.Context) (embedding.Embedder, error) {
} }
emb, err = arkemb.NewArkEmbedder(ctx, &ark.EmbeddingConfig{ emb, err = arkemb.NewArkEmbedder(ctx, &ark.EmbeddingConfig{
APIKey: arkEmbeddingAK, APIKey: func() string {
if arkEmbeddingApiKey != "" {
return arkEmbeddingApiKey
}
return arkEmbeddingAK
}(),
Model: arkEmbeddingModel, Model: arkEmbeddingModel,
BaseURL: arkEmbeddingBaseURL, BaseURL: arkEmbeddingBaseURL,
}, dims, batchSize) }, dims, batchSize)

2
docker/.env.debug.example
View File

@ -111,7 +111,7 @@ export OPENAI_EMBEDDING_REQUEST_DIMS=1024
# ark embedding # ark embedding
export ARK_EMBEDDING_MODEL="" export ARK_EMBEDDING_MODEL=""
export ARK_EMBEDDING_AK="" export ARK_EMBEDDING_API_KEY=""
export ARK_EMBEDDING_DIMS="2048" export ARK_EMBEDDING_DIMS="2048"
export ARK_EMBEDDING_BASE_URL="" export ARK_EMBEDDING_BASE_URL=""

2
docker/.env.example
View File

@ -111,7 +111,7 @@ export OPENAI_EMBEDDING_REQUEST_DIMS=1024
# ark embedding # ark embedding
export ARK_EMBEDDING_MODEL="" export ARK_EMBEDDING_MODEL=""
export ARK_EMBEDDING_AK="" export ARK_EMBEDDING_API_KEY=""
export ARK_EMBEDDING_DIMS="2048" export ARK_EMBEDDING_DIMS="2048"
export ARK_EMBEDDING_BASE_URL="" export ARK_EMBEDDING_BASE_URL=""