ragit 0.6__py3-none-any.whl → 0.8__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.8.dist-info/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Salman Faroz
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

{ragit-0.6.dist-info → ragit-0.8.dist-info}/METADATA

@@ -1,10 +1,11 @@
 Metadata-Version: 2.2
 Name: ragit
-Version: 0.6
+Version: 0.8
 Home-page: https://github.com/stsfaroz/ragit
 Author: Salman Faroz
 License: MIT
 Description-Content-Type: text/markdown
+License-File: LICENSE
 Requires-Dist: sentence-transformers>=3.4.1
 Requires-Dist: pandas>=2.2.3
 Requires-Dist: chromadb>=0.6.3
@@ -22,6 +23,11 @@ Dynamic: requires-dist
 # Ragit
 🚀 Smart, Fast, Scalable Search 🚀
 
+## Installation
+```
+pip install ragit
+```
+
 **ragit** is a lightweight Python library that simplifies the management of vector databases. With **ragit**, you can easily create, update, query, and manage your vector database, all from CSV files containing text data.
 
 ## Features
@@ -72,13 +78,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 +139,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 +148,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.8.dist-info/RECORD

@@ -0,0 +1,7 @@
+ragit/__init__.py,sha256=GECJxYFL_0PMy6tbcVFpW9Fhe1JiI2uXH4iJWhUHpKs,48
+ragit/main.py,sha256=f2kDfZPxP26DBvzmP7aF6VhnNAE1hC-ZONU5ZH6RVBM,11774
+ragit-0.8.dist-info/LICENSE,sha256=L8f7hg7lQm80qoZhSCoW1ACAKph-FpJaNaa9MyNDBqo,1069
+ragit-0.8.dist-info/METADATA,sha256=H73xJZU_viExL4wcb-knClT7-BmMvhtlAaeQ07gliXM,5230
+ragit-0.8.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+ragit-0.8.dist-info/top_level.txt,sha256=pkPbG7yrw61wt9_y_xcLE2vq2a55fzockASD0yq0g4s,6
+ragit-0.8.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,,