ddi-fw 0.0.96__tar.gz → 0.0.97__tar.gz

{ddi_fw-0.0.96 → ddi_fw-0.0.97}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ddi_fw
-Version: 0.0.96
+Version: 0.0.97
 Summary: Do not use :)
 Author-email: Kıvanç Bayraktar <bayraktarkivanc@gmail.com>
 Maintainer-email: Kıvanç Bayraktar <bayraktarkivanc@gmail.com>

{ddi_fw-0.0.96 → ddi_fw-0.0.97}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "ddi_fw"
-version = "0.0.96"
+version = "0.0.97"
 description = "Do not use :)"
 readme = "README.md"
 authors = [

{ddi_fw-0.0.96 → ddi_fw-0.0.97}/src/ddi_fw/pipeline/pipeline.py

@@ -47,7 +47,7 @@ class Pipeline:
         self.combinations = combinations
         self.model = model
 
-    def __create_or_update_embeddings__(self, embedding_dict, vector_db_persist_directory, vector_db_collection_name, column):
+    def __create_or_update_embeddings__(self, embedding_dict, vector_db_persist_directory, vector_db_collection_name, column=None):
         """
         Fetch embeddings and metadata from a persistent Chroma vector database and update the provided embedding_dict.
 
@@ -65,9 +65,11 @@ class Pipeline:
 
             # Fetch the embeddings and metadata
             if column == None:
-                dictionary = collection.get(include=['embeddings', 'metadatas'])
+                dictionary = collection.get(
+                    include=['embeddings', 'metadatas'])
             else:
-                dictionary = collection.get(include=['embeddings', 'metadatas'], where= {"type": {"$eq": f"{column}"}})
+                dictionary = collection.get(include=['embeddings', 'metadatas'], where={
+                                            "type": {"$eq": f"{column}"}})
             # Populate the embedding dictionary with embeddings from the vector database
             for metadata, embedding in zip(dictionary['metadatas'], dictionary['embeddings']):
                 embedding_dict[metadata["type"]
@@ -77,7 +79,7 @@ class Pipeline:
         else:
             raise ValueError(
                 "Persistent directory for the vector DB is not specified.")
-        
+
     def build(self):
         # 'enzyme','target','pathway','smile','all_text','indication', 'description','mechanism_of_action','pharmacodynamics', 'tui', 'cui', 'entities'
         kwargs = {"columns": self.columns}
@@ -89,14 +91,16 @@ class Pipeline:
             if self.vector_db_persist_directory:
                 self.__create_or_update_embeddings__(
                     embedding_dict, self.vector_db_persist_directory, self.vector_db_collection_name)
-                
+
             if self.column_embedding_configs:
                 for item in self.column_embedding_configs:
                     col = item["column"]
                     col_db_dir = item["vector_db_persist_directory"]
                     col_db_collection = item["vector_db_collection_name"]
-                    self.__create_or_update_embeddings__(embedding_dict, col_db_dir, col_db_collection, col)
-                    print(f"Embedings of {col} is calculated from {col_db_collection}")
+                    self.__create_or_update_embeddings__(
+                        embedding_dict, col_db_dir, col_db_collection, col)
+                    print(
+                        f"Embedings of {col} is calculated from {col_db_collection}")
 
         # if self.embedding_dict == None:
         #     if self.vector_db_persist_directory:

{ddi_fw-0.0.96 → ddi_fw-0.0.97}/src/ddi_fw.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ddi_fw
-Version: 0.0.96
+Version: 0.0.97
 Summary: Do not use :)
 Author-email: Kıvanç Bayraktar <bayraktarkivanc@gmail.com>
 Maintainer-email: Kıvanç Bayraktar <bayraktarkivanc@gmail.com>