diff --git a/templates/components/vectordbs/python/milvus/generate.py b/templates/components/vectordbs/python/milvus/generate.py index 862ddd84181392a10e74054151f3882f7aebbff6..b980d9f870132615ef800b776c023e41e125feeb 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 130840ccb46bdffe009e00d81713c0e818ddc7e0..b3cd93d55eb76a75579b6f63b39e0b7f39e2ce87 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")),