ragit 0.6__tar.gz → 0.7__tar.gz

{ragit-0.6 → ragit-0.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: ragit
-Version: 0.6
+Version: 0.7
 Home-page: https://github.com/stsfaroz/ragit
 Author: Salman Faroz
 License: MIT
@@ -72,13 +72,15 @@ db_manager.create_database(
 ```
 ### Reloading Your Database
 
-Once your database is created and your data is added, simply load it for later use by specifying the same folder:
+To reuse your existing vector database, initialize VectorDBManager with the same parameters that were used when creating the database.
 
 ```python
 from ragit import VectorDBManager
 
 db_manager = VectorDBManager(
     persist_directory="./my_vector_db",
+    provider="sentence_transformer",
+    model_name="all-mpnet-base-v2"
 )
 ```
 
@@ -131,17 +133,7 @@ for item in results:
     print(f"Distance ({item['metric']}): {item['raw_distance']}")
 ```
 
-### 7. Deleting an Entry
-Remove an entry from the collection by its ID:
-
-```python
-db_manager.delete_entry_by_id(
-    id_="1",
-    collection_name="my_collection"
-)
-```
-
-### 8. Fetching Texts by IDs
+### 7. Fetching Texts by IDs
 Retrieve text entries for a list of IDs:
 
 ```python
@@ -150,18 +142,18 @@ texts = db_manager.get_by_ids(ids_to_fetch, "my_collection")
 print("Texts:", texts)
 ```
 
-### 9. Fetching IDs by Texts
-For an exact text match, get the corresponding IDs:
+### 8. Deleting a Row / Collection
+
+Remove an entry from the collection by its ID:
 
 ```python
-texts_to_fetch = [
-    "Plato was an ancient Greek philosopher of the Classical period who is considered a foundational thinker in Western philosophy"
-]
-ids = db_manager.get_by_texts(texts_to_fetch, "my_collection")
-print("IDs:", ids)
+db_manager.delete_entry_by_id(
+    id_="1",
+    collection_name="my_collection"
+)
 ```
 
-### 10. Deleting a Collection
+
 Delete an entire collection. **Note:** You must pass `confirmation="yes"` to proceed with deletion.
 
 ```python

{ragit-0.6 → ragit-0.7}/README.md

@@ -52,13 +52,15 @@ db_manager.create_database(
 ```
 ### Reloading Your Database
 
-Once your database is created and your data is added, simply load it for later use by specifying the same folder:
+To reuse your existing vector database, initialize VectorDBManager with the same parameters that were used when creating the database.
 
 ```python
 from ragit import VectorDBManager
 
 db_manager = VectorDBManager(
     persist_directory="./my_vector_db",
+    provider="sentence_transformer",
+    model_name="all-mpnet-base-v2"
 )
 ```
 
@@ -111,17 +113,7 @@ for item in results:
     print(f"Distance ({item['metric']}): {item['raw_distance']}")
 ```
 
-### 7. Deleting an Entry
-Remove an entry from the collection by its ID:
-
-```python
-db_manager.delete_entry_by_id(
-    id_="1",
-    collection_name="my_collection"
-)
-```
-
-### 8. Fetching Texts by IDs
+### 7. Fetching Texts by IDs
 Retrieve text entries for a list of IDs:
 
 ```python
@@ -130,18 +122,18 @@ texts = db_manager.get_by_ids(ids_to_fetch, "my_collection")
 print("Texts:", texts)
 ```
 
-### 9. Fetching IDs by Texts
-For an exact text match, get the corresponding IDs:
+### 8. Deleting a Row / Collection
+
+Remove an entry from the collection by its ID:
 
 ```python
-texts_to_fetch = [
-    "Plato was an ancient Greek philosopher of the Classical period who is considered a foundational thinker in Western philosophy"
-]
-ids = db_manager.get_by_texts(texts_to_fetch, "my_collection")
-print("IDs:", ids)
+db_manager.delete_entry_by_id(
+    id_="1",
+    collection_name="my_collection"
+)
 ```
 
-### 10. Deleting a Collection
+
 Delete an entire collection. **Note:** You must pass `confirmation="yes"` to proceed with deletion.
 
 ```python

{ragit-0.6 → ragit-0.7}/ragit/main.py

@@ -5,7 +5,6 @@ from sentence_transformers import SentenceTransformer
 from typing import List, Dict, Optional, Union
 import os
 
-
 class VectorDBManager:
     def __init__(
         self,
@@ -353,32 +352,3 @@ class VectorDBManager:
         except Exception as e:
             self.logger.error(f"Error getting texts by IDs: {str(e)}")
             return {}
-
-    def get_by_texts(self, texts: List[str], collection_name: str) -> Dict[str, str]:
-        """
-        Get IDs for given texts in batch.
-        Note: For exact text matching. For similar texts, use find_nearby_texts.
-
-        Args:
-            texts (List[str]): List of texts to fetch
-            collection_name (str): Name of the collection
-
-        Returns:
-            Dict[str, str]: Dictionary mapping texts to their corresponding IDs
-        """
-        try:
-            collection = self.client.get_collection(collection_name)
-
-            all_data = collection.get()
-
-            text_to_id = {
-                text: id_
-                for text, id_ in zip(all_data["documents"], all_data["ids"])
-                if text in texts
-            }
-
-            return text_to_id
-
-        except Exception as e:
-            self.logger.error(f"Error getting IDs by texts: {str(e)}")
-            return {}

{ragit-0.6 → ragit-0.7}/ragit.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: ragit
-Version: 0.6
+Version: 0.7
 Home-page: https://github.com/stsfaroz/ragit
 Author: Salman Faroz
 License: MIT
@@ -72,13 +72,15 @@ db_manager.create_database(
 ```
 ### Reloading Your Database
 
-Once your database is created and your data is added, simply load it for later use by specifying the same folder:
+To reuse your existing vector database, initialize VectorDBManager with the same parameters that were used when creating the database.
 
 ```python
 from ragit import VectorDBManager
 
 db_manager = VectorDBManager(
     persist_directory="./my_vector_db",
+    provider="sentence_transformer",
+    model_name="all-mpnet-base-v2"
 )
 ```
 
@@ -131,17 +133,7 @@ for item in results:
     print(f"Distance ({item['metric']}): {item['raw_distance']}")
 ```
 
-### 7. Deleting an Entry
-Remove an entry from the collection by its ID:
-
-```python
-db_manager.delete_entry_by_id(
-    id_="1",
-    collection_name="my_collection"
-)
-```
-
-### 8. Fetching Texts by IDs
+### 7. Fetching Texts by IDs
 Retrieve text entries for a list of IDs:
 
 ```python
@@ -150,18 +142,18 @@ texts = db_manager.get_by_ids(ids_to_fetch, "my_collection")
 print("Texts:", texts)
 ```
 
-### 9. Fetching IDs by Texts
-For an exact text match, get the corresponding IDs:
+### 8. Deleting a Row / Collection
+
+Remove an entry from the collection by its ID:
 
 ```python
-texts_to_fetch = [
-    "Plato was an ancient Greek philosopher of the Classical period who is considered a foundational thinker in Western philosophy"
-]
-ids = db_manager.get_by_texts(texts_to_fetch, "my_collection")
-print("IDs:", ids)
+db_manager.delete_entry_by_id(
+    id_="1",
+    collection_name="my_collection"
+)
 ```
 
-### 10. Deleting a Collection
+
 Delete an entire collection. **Note:** You must pass `confirmation="yes"` to proceed with deletion.
 
 ```python

{ragit-0.6 → ragit-0.7}/setup.py

@@ -5,7 +5,7 @@ with open("README.md", "r") as f:
     
 setup(
     name="ragit",
-    version="0.6",
+    version="0.7",
     packages=find_packages(),
     install_requires = ['sentence-transformers>=3.4.1', 
                         'pandas>=2.2.3', 'chromadb>=0.6.3',