This commit is contained in:
ArnoChen
2025-02-13 17:32:05 +08:00
parent adb234d5cf
commit e4562c761c
5 changed files with 6 additions and 6 deletions

View File

@@ -14,7 +14,7 @@ from pymilvus import MilvusClient
@dataclass
class MilvusVectorDBStorge(BaseVectorStorage):
class MilvusVectorDBStorage(BaseVectorStorage):
@staticmethod
def create_collection_if_not_exist(
client: MilvusClient, collection_name: str, **kwargs
@@ -37,7 +37,7 @@ class MilvusVectorDBStorge(BaseVectorStorage):
db_name=os.environ.get("MILVUS_DB_NAME", ""),
)
self._max_batch_size = self.global_config["embedding_batch_num"]
MilvusVectorDBStorge.create_collection_if_not_exist(
MilvusVectorDBStorage.create_collection_if_not_exist(
self._client,
self.namespace,
dimension=self.embedding_func.embedding_dim,