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

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

@ -335,6 +335,9 @@ 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")
arkEmbeddingApiKey = os.Getenv("ARK_EMBEDDING_API_KEY")
// deprecated: use ARK_EMBEDDING_API_KEY instead
// ARK_EMBEDDING_AK will be removed in the future
arkEmbeddingAK = os.Getenv("ARK_EMBEDDING_AK") arkEmbeddingAK = os.Getenv("ARK_EMBEDDING_AK")
arkEmbeddingDims = os.Getenv("ARK_EMBEDDING_DIMS") arkEmbeddingDims = os.Getenv("ARK_EMBEDDING_DIMS")
) )
@ -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=""