change MILVUS_USER TO MILVUS_USERNAME (#9)

run-llama · Mar 19, 2024 · 9ac6f27 · 9ac6f27
1 parent 9f6a0ef
commit 9ac6f27
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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")),