|
@@ -20,9 +20,9 @@ class MilvusClient:
|
|
def __init__(self):
|
|
def __init__(self):
|
|
self.collection_prefix = "open_webui"
|
|
self.collection_prefix = "open_webui"
|
|
if MILVUS_TOKEN is None:
|
|
if MILVUS_TOKEN is None:
|
|
- self.client = Client(uri=MILVUS_URI, database=MILVUS_DB)
|
|
|
|
|
|
+ self.client = Client(uri=MILVUS_URI, db_name=MILVUS_DB)
|
|
else:
|
|
else:
|
|
- self.client = Client(uri=MILVUS_URI, database=MILVUS_DB, token=MILVUS_TOKEN)
|
|
|
|
|
|
+ self.client = Client(uri=MILVUS_URI, db_name=MILVUS_DB, token=MILVUS_TOKEN)
|
|
|
|
|
|
def _result_to_get_result(self, result) -> GetResult:
|
|
def _result_to_get_result(self, result) -> GetResult:
|
|
ids = []
|
|
ids = []
|