diff --git a/agents-flex-store/agents-flex-store-milvus/src/main/java/com/agentsflex/store/milvus/MilvusVectorStore.java b/agents-flex-store/agents-flex-store-milvus/src/main/java/com/agentsflex/store/milvus/MilvusVectorStore.java index fc187db..18b3878 100644 --- a/agents-flex-store/agents-flex-store-milvus/src/main/java/com/agentsflex/store/milvus/MilvusVectorStore.java +++ b/agents-flex-store/agents-flex-store-milvus/src/main/java/com/agentsflex/store/milvus/MilvusVectorStore.java @@ -57,6 +57,8 @@ public class MilvusVectorStore extends DocumentStore { .uri(config.getUri()) .dbName(config.getDatabaseName()) .token(config.getToken()) + .username(config.getUsername()) + .password(config.getPassword()) .build(); this.client = new MilvusClientV2(connectConfig); diff --git a/agents-flex-store/agents-flex-store-milvus/src/main/java/com/agentsflex/store/milvus/MilvusVectorStoreConfig.java b/agents-flex-store/agents-flex-store-milvus/src/main/java/com/agentsflex/store/milvus/MilvusVectorStoreConfig.java index edf0fc2..a4f4789 100644 --- a/agents-flex-store/agents-flex-store-milvus/src/main/java/com/agentsflex/store/milvus/MilvusVectorStoreConfig.java +++ b/agents-flex-store/agents-flex-store-milvus/src/main/java/com/agentsflex/store/milvus/MilvusVectorStoreConfig.java @@ -24,8 +24,9 @@ public class MilvusVectorStoreConfig implements Serializable { private String uri; private String token; private String databaseName = "default"; + private String username; + private String password; private String defaultCollectionName; - private boolean autoCreateCollection = true; public MilvusVectorStoreConfig() { @@ -70,4 +71,20 @@ public class MilvusVectorStoreConfig implements Serializable { public void setAutoCreateCollection(boolean autoCreateCollection) { this.autoCreateCollection = autoCreateCollection; } + + public String getUsername() { + return username; + } + + public void setUsername(String username) { + this.username = username; + } + + public String getPassword() { + return password; + } + + public void setPassword(String password) { + this.password = password; + } }