leantext 1.0.5__cp38-abi3-win_amd64.whl → 1.0.6__cp38-abi3-win_amd64.whl

{leantext-1.0.5.dist-info → leantext-1.0.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: leantext
-Version: 1.0.5
+Version: 1.0.6
 Classifier: Programming Language :: Rust
 Classifier: Programming Language :: Python :: Implementation :: CPython
 Classifier: Programming Language :: Python :: Implementation :: PyPy

leantext-1.0.6.dist-info/RECORD

@@ -0,0 +1,6 @@
+leantext\__init__.py,sha256=gc6BKLkTF6zMXaN3OGq2DJkxElZCxkvsVBd0G2IG8jc,115
+leantext\leantext.pyd,sha256=df0xphI176EI6ghW97-tGtzACD3QDjg_ZYnTAqqk9Rk,2405376
+leantext-1.0.6.dist-info\METADATA,sha256=ma0e0yLdVifvrHo-P-JKquLBQg_8kUnUf88BQgMBtgA,628
+leantext-1.0.6.dist-info\WHEEL,sha256=gPqN4EsdiAyGvmfrYy_ONrF276O8o0hPitI2CKZrEFA,95
+leantext_engine.py,sha256=YNpsg2vNR-5NW3LgvtxzW3I-kKKQrZwAqMKlFT_id3g,25997
+leantext-1.0.6.dist-info\RECORD,,

leantext_engine.py

@@ -256,13 +256,14 @@ class LeanTextEngine:
                     cols.add(name)
         return list(cols)
     
-    def add(self, content: str, metadata: Optional[Dict[str, Any]] = None, doc_id: Optional[str] = None, collection: str = "default") -> str:
+    def add(self, content: str, title: Optional[str] = None, metadata: Optional[Dict[str, Any]] = None, doc_id: Optional[str] = None, collection: str = "default") -> str:
         """Adds a single document. Returns the document ID."""
-        return self.add_batch([content], [metadata] if metadata else None, [doc_id] if doc_id else None, collection)[0]
+        return self.add_batch([content], [title] if title else None, [metadata] if metadata else None, [doc_id] if doc_id else None, collection)[0]
     
     def add_batch(
         self,
         contents: List[str],
+        titles: Optional[List[str]] = None,
         metadatas: Optional[List[Dict[str, Any]]] = None,
         ids: Optional[List[str]] = None,
         collection: str = "default"
@@ -294,6 +295,9 @@ class LeanTextEngine:
                 meta = copy.deepcopy(meta_orig) if meta_orig else {}
                 meta = self._sanitize_metadata(meta)
                 
+                # Title handling
+                title = titles[i] if titles and i < len(titles) else None
+
                 # ID handling
                 provided_id = ids[i] if i < len(ids) else None
                 doc_id = str(provided_id or meta.get("id") or meta.get("_id") or uuid.uuid4())
@@ -305,7 +309,8 @@ class LeanTextEngine:
                 except orjson.JSONEncodeError:
                     meta_json = orjson.dumps({"id": doc_id}).decode('utf-8')
                 
-                db.add(doc_id, text, meta_json)
+                # Update: Pass title to Rust
+                db.add(doc_id, text, title, meta_json)
                 result_ids.append(doc_id)
             
             # Persist and version tracking (only if thread_safe)
@@ -426,7 +431,7 @@ class LeanTextEngine:
                 db = self._ensure_collection(collection)
                 filter_str = orjson.dumps(filters).decode('utf-8') if filters else None
                 
-                # Returns: Vec<(String, f32, String, Option<String>)>
+                # Returns: Vec<(String, f32, String, String, Option<String>)>
                 raw_results = db.search(
                     query, 
                     filter_str, 
@@ -453,7 +458,7 @@ class LeanTextEngine:
                 )
         
         results = []
-        for doc_id, score, meta_str, content in raw_results:
+        for doc_id, score, title, meta_str, content in raw_results:
             try:
                 meta = orjson.loads(meta_str)
             except (TypeError, orjson.JSONDecodeError):
@@ -462,6 +467,7 @@ class LeanTextEngine:
             res = {
                 "id": doc_id,
                 "score": score,
+                "title": title,
                 "metadata": meta
             }
             if return_content:

leantext-1.0.5.dist-info/RECORD

@@ -1,6 +0,0 @@
-leantext\__init__.py,sha256=gc6BKLkTF6zMXaN3OGq2DJkxElZCxkvsVBd0G2IG8jc,115
-leantext\leantext.pyd,sha256=h_N4r94imTEIBy3wWKoXRum5yDWoMOoq6tljVasQZmU,2356224
-leantext-1.0.5.dist-info\METADATA,sha256=n60uNaI7lqtsV4EaFGNrApFr5pLDCYHb31uwTjhXs2E,628
-leantext-1.0.5.dist-info\WHEEL,sha256=gPqN4EsdiAyGvmfrYy_ONrF276O8o0hPitI2CKZrEFA,95
-leantext_engine.py,sha256=j616JL1hf89lboKF0g8wJ9QU_D6McbqSBDUnCuxzoho,25683
-leantext-1.0.5.dist-info\RECORD,,