sharedkernel 1.6.8__tar.gz → 1.6.9__tar.gz

{sharedkernel-1.6.8 → sharedkernel-1.6.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sharedkernel
-Version: 1.6.8
+Version: 1.6.9
 Summary: sharekernel is a shared package between all python projects
 Author: Smilinno
 Description-Content-Type: text/markdown
@@ -9,7 +9,6 @@ Requires-Dist: requests
 Requires-Dist: pymongo
 Requires-Dist: fastapi==0.111.0
 Requires-Dist: PyJWT
-Requires-Dist: pymilvus
 Requires-Dist: chromadb
 Requires-Dist: persian_tools
 Requires-Dist: sentry-sdk
@@ -21,6 +20,8 @@ this a shared kernel package
 
 # Change Log
 ### Version 1.6.8
+- Delete milvus
+### Version 1.6.8
 - Fix Date Converter Bug
 ### Version 1.6.7
 - Fix bug and remove additional features

{sharedkernel-1.6.8 → sharedkernel-1.6.9}/README.md

@@ -3,6 +3,8 @@ this a shared kernel package
 
 # Change Log
 ### Version 1.6.8
+- Delete milvus
+### Version 1.6.8
 - Fix Date Converter Bug
 ### Version 1.6.7
 - Fix bug and remove additional features

{sharedkernel-1.6.8 → sharedkernel-1.6.9}/setup.py

@@ -26,7 +26,7 @@ setup(
         "pymongo",
         "fastapi==0.111.0",
         "PyJWT",
-        "pymilvus",
+        #"pymilvus",
         "chromadb",
         "persian_tools",
         "sentry-sdk",
@@ -34,7 +34,7 @@ setup(
         "persiantools"
     ],
     # *strongly* suggested for sharing
-    version="1.6.8",
+    version="1.6.9",
     description="sharekernel is a shared package between all python projects",
     long_description=long_description,
     long_description_content_type="text/markdown",

sharedkernel-1.6.9/sharedkernel/database/vector_database_repository/__init__.py

@@ -0,0 +1,2 @@
+from .chroma_startegy import ChromaStrategy
+from .vector_database_repository import VectorRepository

{sharedkernel-1.6.8 → sharedkernel-1.6.9}/sharedkernel/database/vector_database_repository/vector_database_repository.py

@@ -1,6 +1,6 @@
 import numpy as np
 from .vector_database_strategy import VectorDatabaseStrategy
-from .milvus_strategy import MilvusStrategy
+# from .milvus_strategy import MilvusStrategy
 from .chroma_startegy import ChromaStrategy
 from sharedkernel.enum.vector_database_type import VectorDatabaseType
 
@@ -10,10 +10,10 @@ class VectorRepository:
         self.strategy.connect(**connection_params)
     
     def _get_strategy(self, database_type: VectorDatabaseType, collection_name: str) -> VectorDatabaseStrategy:
-        if database_type == VectorDatabaseType.MILVUS:
-            return MilvusStrategy(collection_name)
-        else:
-            return ChromaStrategy(collection_name)
+        # if database_type == VectorDatabaseType.MILVUS:
+        #     return MilvusStrategy(collection_name)
+        # else:
+        return ChromaStrategy(collection_name)
     
     def add_vector(self, vector: np.ndarray, metadata: dict) -> str:
         return self.strategy.insert_vector(vector, metadata)

{sharedkernel-1.6.8 → sharedkernel-1.6.9}/sharedkernel.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sharedkernel
-Version: 1.6.8
+Version: 1.6.9
 Summary: sharekernel is a shared package between all python projects
 Author: Smilinno
 Description-Content-Type: text/markdown
@@ -9,7 +9,6 @@ Requires-Dist: requests
 Requires-Dist: pymongo
 Requires-Dist: fastapi==0.111.0
 Requires-Dist: PyJWT
-Requires-Dist: pymilvus
 Requires-Dist: chromadb
 Requires-Dist: persian_tools
 Requires-Dist: sentry-sdk
@@ -21,6 +20,8 @@ this a shared kernel package
 
 # Change Log
 ### Version 1.6.8
+- Delete milvus
+### Version 1.6.8
 - Fix Date Converter Bug
 ### Version 1.6.7
 - Fix bug and remove additional features

{sharedkernel-1.6.8 → sharedkernel-1.6.9}/sharedkernel.egg-info/SOURCES.txt

@@ -14,7 +14,6 @@ sharedkernel/database/__init__.py
 sharedkernel/database/mongo_generic_repository.py
 sharedkernel/database/vector_database_repository/__init__.py
 sharedkernel/database/vector_database_repository/chroma_startegy.py
-sharedkernel/database/vector_database_repository/milvus_strategy.py
 sharedkernel/database/vector_database_repository/vector_database_repository.py
 sharedkernel/database/vector_database_repository/vector_database_strategy.py
 sharedkernel/enum/__init__.py

{sharedkernel-1.6.8 → sharedkernel-1.6.9}/sharedkernel.egg-info/requires.txt

@@ -3,7 +3,6 @@ requests
 pymongo
 fastapi==0.111.0
 PyJWT
-pymilvus
 chromadb
 persian_tools
 sentry-sdk

sharedkernel-1.6.8/sharedkernel/database/vector_database_repository/__init__.py

@@ -1,3 +0,0 @@
-from .chroma_startegy import ChromaStrategy
-from .milvus_strategy import MilvusStrategy
-from .vector_database_repository import VectorRepository

sharedkernel-1.6.8/sharedkernel/database/vector_database_repository/milvus_strategy.py

@@ -1,50 +0,0 @@
-import numpy as np
-from pymilvus import connections, Collection, FieldSchema, CollectionSchema, DataType
-
-from .vector_database_strategy import VectorDatabaseStrategy
-import uuid
-
-
-class MilvusStrategy(VectorDatabaseStrategy):
-    def __init__(self, collection_name: str):
-        self.collection_name = collection_name
-        self.collection = None
-
-    def connect(self, host: str = "localhost", port: str = "19530"):
-        connections.connect(alias="default", host=host, port=port)
-        # Define fields
-        fields = [
-            FieldSchema(
-                name="id", dtype=DataType.VARCHAR, max_length=36, is_primary=True
-            ),
-            FieldSchema(name="vector", dtype=DataType.FLOAT_VECTOR, dim=128),
-        ]
-        schema = CollectionSchema(
-            fields, description="Vector collection", enable_dynamic_field=True
-        )
-        self.collection = Collection(name=self.collection_name, schema=schema)
-
-        if not self.collection.has_index():
-             self.collection.create_index(field_name="vector", index_params={"index_type": "IVF_FLAT", "metric_type": "L2", "params": {"nlist": 128}})
-
-        self.collection.load()
-
-    def insert_vector(self, vector: np.ndarray, metadata: dict) -> str:
-        id = str(uuid.uuid4())
-        self.collection.insert(data={"id": id, "vector": vector.tolist()})
- 
-        return id
-
-    def search_vector(self, vector: np.ndarray, top_k: int):
-        search_params = {"metric_type": "L2", "params": {"nprobe": 10}}
-        results = self.collection.search(
-            [vector.tolist()], "vector", search_params, top_k
-        )
-        return results
-
-    def get_vector_by_id(self, id: str):
-        result = self.collection.query(expr=f"id=='{id}'",output_fields=["vector"])
-        return result
-
-    def delete_vector(self, id: str):
-        self.collection.delete(expr=f"id=='{id}'")