ragit 0.6__py3-none-any.whl → 0.7__py3-none-any.whl

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.dist-info → ragit-0.7.dist-info}/METADATA

@@ -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.7.dist-info/RECORD

@@ -0,0 +1,6 @@
+ragit/__init__.py,sha256=GECJxYFL_0PMy6tbcVFpW9Fhe1JiI2uXH4iJWhUHpKs,48
+ragit/main.py,sha256=f2kDfZPxP26DBvzmP7aF6VhnNAE1hC-ZONU5ZH6RVBM,11774
+ragit-0.7.dist-info/METADATA,sha256=xYd5dnzwTFkXCGTCHMDakiER_xfDr8VjQBntOFlwL6M,5165
+ragit-0.7.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+ragit-0.7.dist-info/top_level.txt,sha256=pkPbG7yrw61wt9_y_xcLE2vq2a55fzockASD0yq0g4s,6
+ragit-0.7.dist-info/RECORD,,

ragit-0.6.dist-info/RECORD

@@ -1,6 +0,0 @@
-ragit/__init__.py,sha256=GECJxYFL_0PMy6tbcVFpW9Fhe1JiI2uXH4iJWhUHpKs,48
-ragit/main.py,sha256=L5jId4Cwrd99tV10QzokvF3XWarOj3vPEOWLzPrTMN0,12721
-ragit-0.6.dist-info/METADATA,sha256=tlkKRahw3UUvvG_bC9ADGrvfz5homuLxQkV7i2gkzq8,5422
-ragit-0.6.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-ragit-0.6.dist-info/top_level.txt,sha256=pkPbG7yrw61wt9_y_xcLE2vq2a55fzockASD0yq0g4s,6
-ragit-0.6.dist-info/RECORD,,