projectdavid 1.33.17__py3-none-any.whl → 1.33.19__py3-none-any.whl

projectdavid/clients/vector_store_manager.py

@@ -7,6 +7,7 @@ details from the higher‑level SDK.
 
 from __future__ import annotations
 
+import inspect
 import time
 import uuid
 from pathlib import Path
@@ -129,25 +130,45 @@ class VectorStoreManager(BaseVectorStore):
             log.error("Store info failed: %s", e)
             raise VectorStoreError(f"Info retrieval failed: {e}") from e
 
-    # ------------------------------------------------------------------ #
-    # ingestion helpers
-    # ------------------------------------------------------------------ #
     def add_to_store(
         self,
+        *,
         store_name: str,
         texts: List[str],
         vectors: List[List[float]],
         metadata: List[dict],
-        # *,
-        # vector_name: Optional[str] = None,  # NEW
-    ):
+        vector_name: Optional[str] = None,
+    ) -> Dict[str, Any]:
+        """Upsert vectors + payloads into *store_name*.
+
+        If *vector_name* is omitted, the manager auto‑detects the sole vector
+        field. If multiple fields exist it raises a clear error.
+        """
+
+        # ---- validate input ------------------------------------------------
         if not vectors:
             raise ValueError("Empty vectors list")
         expected = len(vectors[0])
         for i, vec in enumerate(vectors):
             if len(vec) != expected or not all(isinstance(v, float) for v in vec):
-                raise ValueError(f"Vector {i} malformed")
+                raise ValueError(f"Vector {i} malformed: expected {expected} floats")
+
+        # ---- auto‑detect vector field --------------------------------------
+        if vector_name is None:
+            coll_info = self.client.get_collection(collection_name=store_name)
+            v_cfg = coll_info.config.params.vectors
+            vector_fields = list(v_cfg.keys()) if isinstance(v_cfg, dict) else [v_cfg]
+            if len(vector_fields) == 1:
+                vector_name = vector_fields[0]
+                log.debug(
+                    "Auto‑detected vector_name=%r for store=%s", vector_name, store_name
+                )
+            else:
+                raise ValueError(
+                    f"Multiple vector fields {vector_fields}; please specify vector_name"
+                )
 
+        # ---- build points ---------------------------------------------------
         points = [
             qdrant.PointStruct(
                 id=self._generate_vector_id(),
@@ -156,18 +177,25 @@ class VectorStoreManager(BaseVectorStore):
             )
             for txt, vec, meta in zip(texts, vectors, metadata)
         ]
+
+        # ---- upsert with backward‑compat for old clients --------------------
+        upsert_sig = inspect.signature(self.client.upsert)
+        supports_vector_name = "vector_name" in upsert_sig.parameters
+
+        upsert_kwargs: Dict[str, Any] = {
+            "collection_name": store_name,
+            "points": points,
+            "wait": True,
+        }
+        if supports_vector_name:
+            upsert_kwargs["vector_name"] = vector_name
+
         try:
-            # pass vector_name if multi-column
-            self.client.upsert(
-                collection_name=store_name,
-                points=points,
-                wait=True,
-                # vector_name=vector_name,  # ignored if None
-            )
+            self.client.upsert(**upsert_kwargs)
             return {"status": "success", "points_inserted": len(points)}
-        except Exception as e:
-            log.error("Add‑to‑store failed: %s", e)
-            raise VectorStoreError(f"Insertion failed: {e}") from e
+        except Exception as exc:  # noqa: BLE001
+            log.error("Add‑to‑store failed: %s", exc, exc_info=True)
+            raise VectorStoreError(f"Insertion failed: {exc}") from exc
 
     # ------------------------------------------------------------------ #
     # search / query

{projectdavid-1.33.17.dist-info → projectdavid-1.33.19.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: projectdavid
-Version: 1.33.17
+Version: 1.33.19
 Summary: Python SDK for interacting with the Entities Assistant API.
 Author-email: Francis Neequaye Armah <francis.neequaye@projectdavid.co.uk>
 License: PolyForm Noncommercial License 1.0.0

{projectdavid-1.33.17.dist-info → projectdavid-1.33.19.dist-info}/RECORD

@@ -20,7 +20,7 @@ projectdavid/clients/synchronous_inference_wrapper.py,sha256=qh94rtNlLqgIxiA_ZbQ
 projectdavid/clients/threads_client.py,sha256=ekzU5w14zftmtmFkiec3NC90Of-_KVSUY1qH9cmfSFg,6771
 projectdavid/clients/tools_client.py,sha256=GkCVOmwpAoPqVt6aYmH0G1HIFha3iEwR9IIf9teR0j8,11487
 projectdavid/clients/users_client.py,sha256=eCuUb9qvyH1GUFhZu6TRL9zdoK-qzHSs8-Vmrk_0mmg,13729
-projectdavid/clients/vector_store_manager.py,sha256=78760RiuKuGD0aQ6I7mSnQN9KjatlE95MY4JsVXUlBI,12871
+projectdavid/clients/vector_store_manager.py,sha256=4uedMZ7foNr7s4Lsq5-FSN5kl4lR-Y8-nUlkoO2ff8M,14119
 projectdavid/clients/vectors.py,sha256=ubkB4rzcfDSYHFwOpa99Y4nHSJan_0jKnzyCYFJ15ck,31665
 projectdavid/clients/vision-file_processor.py,sha256=19ft9IUeY5x9_22vC4JqndiFlpDYyUn6z1ygv-EV2NE,16852
 projectdavid/clients/vision_vectors.py,sha256=cysPVbUzW3byB82MTqG2X1Iz5ZAe82WTS1JfQcoqVhE,40229
@@ -37,8 +37,8 @@ projectdavid/utils/monitor_launcher.py,sha256=3YAgJdeuaUvq3JGvpA4ymqFsAnk29nH5q9
 projectdavid/utils/peek_gate.py,sha256=5whMRnDOQjATRpThWDJkvY9ScXuJ7Sd_-9rvGgXeTAQ,2532
 projectdavid/utils/run_monitor.py,sha256=F_WkqIP-qnWH-4llIbileWWLfRj2Q1Cg-ni23SR1rec,3786
 projectdavid/utils/vector_search_formatter.py,sha256=YTe3HPGec26qGY7uxY8_GS8lc4QaN6aNXMzkl29nZpI,1735
-projectdavid-1.33.17.dist-info/licenses/LICENSE,sha256=_8yjiEGttpS284BkfhXxfERqTRZW_tUaHiBB0GTJTMg,4563
-projectdavid-1.33.17.dist-info/METADATA,sha256=BcZJ6aDRkRmfMs8XVdUkFeHAfLxjJpWr8ElVgxyh820,11555
-projectdavid-1.33.17.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-projectdavid-1.33.17.dist-info/top_level.txt,sha256=kil8GU4s7qYRfNnzGnFHhZnSNRSxgNG-J4HLgQMmMtw,13
-projectdavid-1.33.17.dist-info/RECORD,,
+projectdavid-1.33.19.dist-info/licenses/LICENSE,sha256=_8yjiEGttpS284BkfhXxfERqTRZW_tUaHiBB0GTJTMg,4563
+projectdavid-1.33.19.dist-info/METADATA,sha256=eXDpkgLi7akRbAzqob6TB1B4veUNoMY0EOwf7QH_Rpo,11555
+projectdavid-1.33.19.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+projectdavid-1.33.19.dist-info/top_level.txt,sha256=kil8GU4s7qYRfNnzGnFHhZnSNRSxgNG-J4HLgQMmMtw,13
+projectdavid-1.33.19.dist-info/RECORD,,