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

feat: milvus support to use token as auth (#2018)

Browse Source
This commit is contained in:
wulorn
2025-09-08 09:29:47 +08:00
committed by GitHub
parent a7d0eee7c0
commit 3317fa3016
3 changed files with 11 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
backend/application/base/appinfra/app_infra.go
View File

@ -435,13 +435,17 @@ func getVectorStore(ctx context.Context) (searchstore.Manager, error) {
ctx, cancel := context.WithTimeout(ctx, time.Second*5)
defer cancel()
milvusAddr := os.Getenv("MILVUS_ADDR")
user := os.Getenv("MILVUS_USER")
password := os.Getenv("MILVUS_PASSWORD")
var (
milvusAddr = os.Getenv("MILVUS_ADDR")
user = os.Getenv("MILVUS_USER")
password = os.Getenv("MILVUS_PASSWORD")
milvusToken = os.Getenv("MILVUS_TOKEN")
)
mc, err := milvusclient.New(ctx, &milvusclient.ClientConfig{
Address: milvusAddr,
Username: user,
Password: password,
APIKey: milvusToken,
})
if err != nil {
return nil, fmt.Errorf("init milvus client failed, err=%w", err)

3
docker/.env.debug.example
View File

@ -89,6 +89,9 @@ export RMQ_SECRET_KEY=""
export VECTOR_STORE_TYPE="milvus"
# milvus vector store
export MILVUS_ADDR="127.0.0.1:19530"
export MILVUS_USER=""
export MILVUS_PASSWORD=""
export MILVUS_TOKEN=""
# vikingdb vector store for Volcengine
export VIKING_DB_HOST=""
export VIKING_DB_REGION=""

1
docker/.env.example
View File

@ -87,6 +87,7 @@ export VECTOR_STORE_TYPE="milvus"
export MILVUS_ADDR="milvus:19530"
export MILVUS_USER=""
export MILVUS_PASSWORD=""
export MILVUS_TOKEN=""
# vikingdb vector store for Volcengine
export VIKING_DB_HOST=""
export VIKING_DB_REGION=""