crewplus 0.2.13__py3-none-any.whl → 0.2.14__py3-none-any.whl

crewplus/vectorstores/milvus/vdb_service.py

@@ -6,7 +6,7 @@
 
 import logging
 from typing import List, Dict, Union, Optional
-from langchain_milvus import Zilliz
+from langchain_milvus import Milvus
 from langchain_core.embeddings import Embeddings
 from langchain_openai import AzureOpenAIEmbeddings
 from pymilvus import MilvusClient
@@ -60,8 +60,8 @@ class VDBService(object):
         ...         }
         ...     },
         ...     "index_params": {
-        ...         "metric_type": "L2",
-        ...         "index_type": "AUTOINDEX",
+        ...         "metric_type": "IP",
+        ...         "index_type": "HNSW",
         ...         "params": {}
         ...     }
         ... }
@@ -83,7 +83,7 @@ class VDBService(object):
         >>> assert vector_store is same_vector_store
     """
     _client: MilvusClient
-    _instances: Dict[str, Zilliz] = {}
+    _instances: Dict[str, Milvus] = {}
 
     schema: str
     embedding_function: Embeddings
@@ -275,7 +275,7 @@ class VDBService(object):
             self.logger.error(f"An error occurred while ensuring collection '{collection_name}' : {e}")
             raise RuntimeError(f"Failed to ensure collection '{collection_name}' .") from e
 
-    def get_vector_store(self, collection_name: str, embeddings: Embeddings = None, metric_type: str = "L2") -> Zilliz:
+    def get_vector_store(self, collection_name: str, embeddings: Embeddings = None, metric_type: str = "IP") -> Milvus:
         """
         Gets a vector store instance, creating it if it doesn't exist for the collection.
         This method validates both the embedding function and the vector store connection
@@ -284,10 +284,10 @@ class VDBService(object):
         Args:
             collection_name (str): The name of the collection in the vector database.
             embeddings (Embeddings, optional): An embedding model instance. If None, one is created.
-            metric_type (str): The distance metric for the index. Defaults to "L2".
+            metric_type (str): The distance metric for the index. Defaults to "IP".
 
         Returns:
-            Zilliz: LangChain Zilliz instance, which is compatible with both Zilliz and Milvus.
+            Milvus: LangChain Milvus instance, which is compatible with both Zilliz and Milvus.
         """
         if not collection_name:
             self.logger.error("get_vector_store called with no collection_name.")
@@ -317,14 +317,14 @@ class VDBService(object):
             )
             raise RuntimeError(f"Invalid embedding function provided.") from e
 
-        # If embeddings are valid, proceed to create the Zilliz instance.
+        # If embeddings are valid, proceed to create the Milvus instance.
         index_params = self.index_params or {
             "metric_type": metric_type,
             "index_type": "AUTOINDEX",
             "params": {}
         }
         
-        vdb = Zilliz(
+        vdb = Milvus(
             embedding_function=embeddings,
             collection_name=collection_name,
             connection_args=self.connection_args,
@@ -336,12 +336,12 @@ class VDBService(object):
 
         return vdb
 
-    def delete_old_indexes(self, url: str = None, vdb: Zilliz = None) -> (bool | None):
+    def delete_old_indexes(self, url: str = None, vdb: Milvus = None) -> (bool | None):
         """ Delete old indexes of the same source_url
 
         Args:
             url (str): source url
-            vdb (Zilliz): Zilliz instance
+            vdb (Milvus): Milvus/Zilliz instance
         """
         self.logger.info(f"Delete old indexes of the same source_url:{url}")
 
@@ -358,7 +358,7 @@ class VDBService(object):
             self.logger.info("Deleted old indexes result: " + str(res))
             return res
 
-    def delete_old_indexes_by_id(self, source_id: str = None, vdb: Zilliz = None) -> (bool | None):
+    def delete_old_indexes_by_id(self, source_id: str = None, vdb: Milvus = None) -> (bool | None):
         """ Delete old indexes of the same source_id
 
         Args:

{crewplus-0.2.13.dist-info → crewplus-0.2.14.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: crewplus
-Version: 0.2.13
+Version: 0.2.14
 Summary: Base services for CrewPlus AI applications
 Author-Email: Tim Liu <tim@opsmateai.com>
 License: MIT

{crewplus-0.2.13.dist-info → crewplus-0.2.14.dist-info}/RECORD

@@ -1,7 +1,7 @@
-crewplus-0.2.13.dist-info/METADATA,sha256=qOJZb76_-Cgyg9-L3uP4nRaylOVrmSoPOH6tpyzvFtU,5087
-crewplus-0.2.13.dist-info/WHEEL,sha256=9P2ygRxDrTJz3gsagc0Z96ukrxjr-LFBGOgv3AuKlCA,90
-crewplus-0.2.13.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
-crewplus-0.2.13.dist-info/licenses/LICENSE,sha256=2_NHSHRTKB_cTcT_GXgcenOCtIZku8j343mOgAguTfc,1087
+crewplus-0.2.14.dist-info/METADATA,sha256=r1Czargd06hF_rDNnysOU4sH2mFoq6qczAb_fSQijoU,5087
+crewplus-0.2.14.dist-info/WHEEL,sha256=9P2ygRxDrTJz3gsagc0Z96ukrxjr-LFBGOgv3AuKlCA,90
+crewplus-0.2.14.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
+crewplus-0.2.14.dist-info/licenses/LICENSE,sha256=2_NHSHRTKB_cTcT_GXgcenOCtIZku8j343mOgAguTfc,1087
 crewplus/__init__.py,sha256=m46HkZL1Y4toD619NL47Sn2Qe084WFFSFD7e6VoYKZc,284
 crewplus/services/__init__.py,sha256=ra_ciHcJN_sbv7q8UCP2kY91SbD32-QBpQLRgIosEcE,267
 crewplus/services/gemini_chat_model.py,sha256=i9p5KvSJYaHSUBLPKM_bpyGVLWCDQoNeah_WjQVJRXs,26227
@@ -13,9 +13,9 @@ crewplus/utils/schema_document_updater.py,sha256=frvffxn2vbi71fHFPoGb9hq7gH2azmm
 crewplus/vectorstores/milvus/__init__.py,sha256=egGncAdjlXG6ekTQvKMKqhvKBifrUrPlsSB0-bpvq4A,229
 crewplus/vectorstores/milvus/milvus_schema_manager.py,sha256=2IZT61LVui21Pt5Z3y8YYS2dYcwzkgUKxMq2NA0-lQE,9222
 crewplus/vectorstores/milvus/schema_milvus.py,sha256=IvKdUCH451HJ-F3TUR5jDjqwQlQs4SEXAQ_th4JAnfc,12117
-crewplus/vectorstores/milvus/vdb_service.py,sha256=R3YakhJEVpPN1yCGgprFCcJflcBIR6BY8eOglUJmduM,19192
+crewplus/vectorstores/milvus/vdb_service.py,sha256=i-imqk4tnIXkf5O406lHaq5H7YSPy8UZhqN1o5NpanI,19194
 docs/GeminiChatModel.md,sha256=_IQyup3ofAa2HxfSurO1GYUEezTHYYt5Q1khYNVThGM,8040
 docs/ModelLoadBalancer.md,sha256=aGHES1dcXPz4c7Y8kB5-vsCNJjriH2SWmjBkSGoYKiI,4398
 docs/VDBService.md,sha256=Dw286Rrf_fsi13jyD3Bo4Sy7nZ_G7tYm7d8MZ2j9hxk,9375
 docs/index.md,sha256=3tlc15uR8lzFNM5WjdoZLw0Y9o1P1gwgbEnOdIBspqc,1643
-crewplus-0.2.13.dist-info/RECORD,,
+crewplus-0.2.14.dist-info/RECORD,,