projectdavid 1.29.9__py3-none-any.whl → 1.38.1__py3-none-any.whl

projectdavid/clients/threads_client.py

@@ -1,4 +1,4 @@
-# projectdavid/clients/threads_client.py
+#! projectdavid/clients/threads_client.py
 from typing import Any, Dict, List, Optional
 
 import httpx
@@ -104,16 +104,30 @@ class ThreadsClient(BaseAPIClient):
             logging_utility.error("Unexpected error retrieving thread: %s", str(e))
             raise
 
-    def update_thread(self, thread_id: str, **updates) -> validator.ThreadReadDetailed:
+    def update_thread(
+        self,
+        thread_id: str,
+        *,
+        participant_ids: Optional[List[str]] = None,
+        meta_data: Optional[Dict[str, Any]] = None,
+        tool_resources: Optional[Dict[str, Any]] = None,
+    ) -> validator.ThreadReadDetailed:
         logging_utility.info("Updating thread with id: %s", thread_id)
         try:
-            validated_updates = validator.ThreadUpdate(**updates)
-            response = self.client.post(
-                f"/v1/threads/{thread_id}", json=validated_updates.model_dump()
+            validated_updates = validator.ThreadUpdate(
+                participant_ids=participant_ids,
+                meta_data=meta_data,
+                tool_resources=tool_resources,
+            )
+
+            response = self.client.put(
+                f"/v1/threads/{thread_id}",
+                json=validated_updates.model_dump(),
             )
             response.raise_for_status()
             updated_thread = response.json()
             return validator.ThreadReadDetailed(**updated_thread)
+
         except httpx.HTTPStatusError as e:
             logging_utility.error("HTTP error updating thread: %s", str(e))
             raise
@@ -154,17 +168,23 @@ class ThreadsClient(BaseAPIClient):
             logging_utility.error("Unexpected error listing threads: %s", str(e))
             raise
 
-    def delete_thread(self, thread_id: str) -> bool:
+    def delete_thread(self, thread_id: str) -> validator.ThreadDeleted | None:
+        """
+        Delete a thread.
+        ▶ On success:  returns ThreadDeleted(id=…, object='thread.deleted', deleted=True)
+        ▶ If not found: returns None
+        """
         logging_utility.info("Deleting thread with id: %s", thread_id)
+
         try:
-            response = self.client.delete(f"/v1/threads/{thread_id}")
-            response.raise_for_status()
-            return True
+            resp = self.client.delete(f"/v1/threads/{thread_id}")
+            resp.raise_for_status()  # 2xx → OK
+            return validator.ThreadDeleted(**resp.json())  # ← parse envelope
         except httpx.HTTPStatusError as e:
             if e.response.status_code == 404:
-                return False
-            logging_utility.error("HTTP error deleting thread: %s", str(e))
+                return None  # thread not found
+            logging_utility.error("HTTP error deleting thread: %s", e)
             raise
         except Exception as e:
-            logging_utility.error("Unexpected error deleting thread: %s", str(e))
+            logging_utility.error("Unexpected error deleting thread: %s", e)
             raise

projectdavid/clients/vector_store_manager.py

@@ -9,7 +9,6 @@ from __future__ import annotations
 
 import time
 import uuid
-from pathlib import Path
 from typing import Any, Dict, List, Optional
 
 from dotenv import load_dotenv
@@ -50,11 +49,20 @@ class VectorStoreManager(BaseVectorStore):
     def create_store(
         self,
         collection_name: str,
+        *,
         vector_size: int = 384,
         distance: str = "COSINE",
+        vectors_config: Optional[Dict[str, qdrant.VectorParams]] = None,
     ) -> dict:
+        """
+        Create or recreate a Qdrant collection.
+
+        • If *vectors_config* is provided → use it verbatim (multi-vector schema).
+        • Otherwise create a classic single-vector collection *without* naming the
+          vector field – so upserts can omit ``vector_name``.
+        """
         try:
-            # quick existence check
+            # ── pre-existence check ────────────────────────────────────────────
             if any(
                 col.name == collection_name
                 for col in self.client.get_collections().collections
@@ -65,16 +73,32 @@ class VectorStoreManager(BaseVectorStore):
             if dist not in qdrant.Distance.__members__:
                 raise ValueError(f"Invalid distance metric '{distance}'")
 
+            # ── choose schema ──────────────────────────────────────────────────
+            if vectors_config:  # caller supplied full mapping
+                config = vectors_config  # e.g. {"text_vec": ..., "img_vec": ...}
+            else:  # default = single unnamed vector
+                config = qdrant.VectorParams(
+                    size=vector_size,
+                    distance=qdrant.Distance[dist],
+                )
+
+            # ── (re)create collection ─────────────────────────────────────────
             self.client.recreate_collection(
                 collection_name=collection_name,
-                vectors_config=qdrant.VectorParams(
-                    size=vector_size, distance=qdrant.Distance[dist]
-                ),
+                vectors_config=config,
             )
+
+            # ── bookkeeping ───────────────────────────────────────────────────
+            if isinstance(config, dict):
+                fields = list(config.keys())
+            else:  # unnamed field
+                fields = [None]
+
             self.active_stores[collection_name] = {
                 "created_at": int(time.time()),
                 "vector_size": vector_size,
                 "distance": dist,
+                "fields": fields,
             }
             log.info("Created Qdrant collection %s", collection_name)
             return {"collection_name": collection_name, "status": "created"}
@@ -103,30 +127,68 @@ class VectorStoreManager(BaseVectorStore):
                 "name": store_name,
                 "status": "active",
                 "vectors_count": info.points_count,
-                "configuration": info.config.params["default"],
+                "configuration": info.config.params,
                 "created_at": self.active_stores[store_name]["created_at"],
+                "fields": self.active_stores[store_name].get("fields"),
             }
         except Exception as e:
             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,
+    ) -> Dict[str, Any]:
+        """
+        Upsert vectors + payloads into *store_name*.
+
+        If *vector_name* is omitted the manager:
+
+        • auto-detects the single vector field for classic (unnamed) collections
+        • auto-detects the sole key for named-vector collections with exactly one field
+        • raises if multiple named fields exist.
+        """
+
+        # ─── input validation ───────────────────────────────────────────────
         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
+
+            if isinstance(v_cfg, dict):  # modern named-vector schema
+                vector_fields = list(v_cfg.keys())
+                if len(vector_fields) == 1:  # exactly one → safe default
+                    vector_name = vector_fields[0]
+                    log.debug(
+                        "Auto-detected vector_name=%r for store=%s",
+                        vector_name,
+                        store_name,
+                    )
+                else:  # >1 named fields → ambiguous
+                    raise ValueError(
+                        f"Multiple vector fields {vector_fields}; please specify vector_name"
+                    )
+            else:
+                # legacy single-vector schema → leave vector_name as None
+                log.debug(
+                    "Collection %s uses legacy single-vector schema; "
+                    "upserting without vector_name",
+                    store_name,
+                )
+
+        # ─── build points payload ───────────────────────────────────────────
         points = [
             qdrant.PointStruct(
                 id=self._generate_vector_id(),
@@ -135,12 +197,27 @@ class VectorStoreManager(BaseVectorStore):
             )
             for txt, vec, meta in zip(texts, vectors, metadata)
         ]
+
+        # ─── upsert with backward-compat for old qdrant-client builds ───────
+        import inspect  # keep local to avoid top-level dependency if absent elsewhere
+
+        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 and vector_name is not None:
+            upsert_kwargs["vector_name"] = vector_name
+
         try:
-            self.client.upsert(collection_name=store_name, points=points, wait=True)
+            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
@@ -189,15 +266,25 @@ class VectorStoreManager(BaseVectorStore):
         query_vector: List[float],
         top_k: int = 5,
         filters: Optional[dict] = None,
+        *,
+        vector_field: Optional[str] = None,  # ← NEW
         score_threshold: float = 0.0,
         offset: int = 0,
         limit: Optional[int] = None,
     ) -> List[dict]:
-        """Run a similarity search that works with any 1.x qdrant‑client."""
+        """
+        Run a similarity search against *store_name*.
+
+        • Works with any Qdrant-client ≥ 1.0
+        • `vector_field` lets you target a non-default vector column
+          (e.g. ``\"caption_vector\"`` for image stores).  Pass **None**
+          to use the collection’s default vector.
+        """
 
         limit = limit or top_k
         flt = self._dict_to_filter(filters) if filters else None
 
+        # ── shared kwargs ----------------------------------------------------
         common: Dict[str, Any] = dict(
             collection_name=store_name,
             query_vector=query_vector,
@@ -208,19 +295,21 @@ class VectorStoreManager(BaseVectorStore):
             with_vectors=False,
         )
 
+        # if vector_field:  # ← inject when requested
+        #    common["vector_name"] = vector_field
+
+        # ── call search (new client first, fallback to old) ------------------
         try:
-            # Newer clients (≥ 1.6) use `filter=`
-            res = self.client.search(**common, filter=flt)  # type: ignore[arg-type]
+            res = self.client.search(**common, filter=flt)  # ≥ 1.6
         except AssertionError as ae:
             if "Unknown arguments" not in str(ae):
                 raise
-            # Older clients use `query_filter=`
-            res = self.client.search(**common, query_filter=flt)  # type: ignore[arg-type]
-
+            res = self.client.search(**common, query_filter=flt)  # < 1.6
         except Exception as e:
             log.error("Query failed: %s", e)
             raise VectorStoreError(f"Query failed: {e}") from e
 
+        # ── normalise result -------------------------------------------------
         return [
             {
                 "id": p.id,