diff --git a/server/src/milvus/milvus.service.ts b/server/src/milvus/milvus.service.ts
index cd3193a8..19b3e563 100644
--- a/server/src/milvus/milvus.service.ts
+++ b/server/src/milvus/milvus.service.ts
@@ -55,6 +55,10 @@ export class MilvusService {
         if (process.env.SERVER_NAME) {
           clientConfig.tls.serverName = process.env.SERVER_NAME;
         }
+
+        if (process.env.DB_NAME) {
+          clientConfig.database = process.env.DB_NAME;
+        }
       }
       // create the client
       const milvusClient: MilvusClient = new MilvusClient(clientConfig);