From 9ac6f27ece802c55b0342c26bf47c6cc77c8463b Mon Sep 17 00:00:00 2001
From: "Huu Le (Lee)" <39040748+leehuwuj@users.noreply.github.com>
Date: Tue, 19 Mar 2024 16:21:01 +0700
Subject: [PATCH] change MILVUS_USER TO MILVUS_USERNAME (#9)

---
 templates/components/vectordbs/python/milvus/generate.py | 2 +-
 templates/components/vectordbs/python/milvus/index.py    | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/templates/components/vectordbs/python/milvus/generate.py b/templates/components/vectordbs/python/milvus/generate.py
index 862ddd84..b980d9f8 100644
--- a/templates/components/vectordbs/python/milvus/generate.py
+++ b/templates/components/vectordbs/python/milvus/generate.py
@@ -20,7 +20,7 @@ def generate_datasource():
     documents = get_documents()
     store = MilvusVectorStore(
         uri=os.environ["MILVUS_ADDRESS"],
-        user=os.getenv("MILVUS_USER"),
+        user=os.getenv("MILVUS_USERNAME"),
         password=os.getenv("MILVUS_PASSWORD"),
         collection_name=os.getenv("MILVUS_COLLECTION"),
         dim=int(os.getenv("MILVUS_DIMENSION", "1536")),
diff --git a/templates/components/vectordbs/python/milvus/index.py b/templates/components/vectordbs/python/milvus/index.py
index 130840cc..b3cd93d5 100644
--- a/templates/components/vectordbs/python/milvus/index.py
+++ b/templates/components/vectordbs/python/milvus/index.py
@@ -12,7 +12,7 @@ def get_index():
     logger.info("Connecting to index from Milvus...")
     store = MilvusVectorStore(
         uri=os.getenv("MILVUS_ADDRESS"),
-        user=os.getenv("MILVUS_USER"),
+        user=os.getenv("MILVUS_USERNAME"),
         password=os.getenv("MILVUS_PASSWORD"),
         collection_name=os.getenv("MILVUS_COLLECTION"),
         dim=int(os.getenv("EMBEDDING_DIM", "1536")),
-- 
GitLab