projectdavid 1.33.0__py3-none-any.whl → 1.33.2__py3-none-any.whl

projectdavid/clients/vectors.py

@@ -72,8 +72,9 @@ class VectorStoreClient:
         api_key: Optional[str] = None,
         *,
         vector_store_host: str = "localhost",
-        file_processor_kwargs: Optional[dict] = None,  # 🔶 add arg
+        file_processor_kwargs: Optional[dict] = None,
     ):
+
         self.base_url = (base_url or os.getenv("BASE_URL", "")).rstrip("/")
         self.api_key = api_key or os.getenv("API_KEY")
         if not self.base_url:
@@ -94,7 +95,18 @@ class VectorStoreClient:
         self.identifier_service = UtilsInterface.IdentifierService()
 
         # 🔶 forward kwargs into the upgraded FileProcessor
-        self.file_processor = FileProcessor(**(file_processor_kwargs or {}))
+
+        self.file_processor = FileProcessor(
+            **(
+                file_processor_kwargs
+                or {
+                    "use_gpu": False,
+                    "use_detection": True,
+                    "use_geo": True,
+                    "use_ocr": True,
+                }
+            )
+        )
 
         log.info("VectorStoreClient → %s", self.base_url)
 
@@ -494,27 +506,89 @@ class VectorStoreClient:
 
     def create_vector_vision_store(
         self,
-        name: str = "vision",
-    ):
+        name: str,
+        *,
+        vector_size: int = 384,
+        distance_metric: str = "Cosine",
+        config: Optional[Dict[str, Any]] = None,
+        vectors_config: Optional[Dict[str, qdrant.VectorParams]] = None,  # ← NEW
+    ) -> ValidationInterface.VectorStoreRead:
 
-        vectors_config = {
-            # Raw visual embeddings (OpenCLIP ViT-H/14 → 1024-D)
-            "image_vector": qdrant.VectorParams(
-                size=1024, distance=qdrant.Distance.COSINE
-            ),
-            # Language embeddings of your BLIP-2 captions → 1024-D
-            "caption_vector": qdrant.VectorParams(
-                size=1024, distance=qdrant.Distance.COSINE
-            ),
-            # Object-region embeddings (YOLO crop + Sentence-BERT) → 1024-D
-            "region_vector": qdrant.VectorParams(
-                size=1024, distance=qdrant.Distance.COSINE
-            ),
-            # Geo-location unit vectors (RegioNet) → 3-D
-            "geo_vector": qdrant.VectorParams(size=3, distance=qdrant.Distance.COSINE),
-        }
+        if not vectors_config:
+            vectors_config = {
+                # Raw visual embeddings (OpenCLIP ViT-H/14 → 1024-D)
+                "image_vector": qdrant.VectorParams(
+                    size=1024, distance=qdrant.Distance.COSINE
+                ),
+                # Language embeddings of your BLIP-2 captions → 1024-D
+                "caption_vector": qdrant.VectorParams(
+                    size=1024, distance=qdrant.Distance.COSINE
+                ),
+                # Object-region embeddings (YOLO crop + Sentence-BERT) → 1024-D
+                "region_vector": qdrant.VectorParams(
+                    size=1024, distance=qdrant.Distance.COSINE
+                ),
+                # Geo-location unit vectors (RegioNet) → 3-D
+                "geo_vector": qdrant.VectorParams(
+                    size=3, distance=qdrant.Distance.COSINE
+                ),
+            }
 
-        return self.create_vector_store(name=name, vectors_config=vectors_config)
+        return self._run_sync(
+            self._create_vs_async(
+                name,
+                vector_size,
+                distance_metric,
+                config,
+                vectors_config,
+            )
+        )
+
+    def create_vector_vision_store_for_user(
+        self,
+        owner_id: str,
+        name: str,
+        *,
+        vector_size: int = 384,
+        distance_metric: str = "Cosine",
+        config: Optional[Dict[str, Any]] = None,
+        vectors_config: Optional[Dict[str, qdrant.VectorParams]] = None,  # ← NEW
+    ) -> ValidationInterface.VectorStoreRead:
+        """
+        Admin-only: create a store on behalf of another user.
+        Pass `vectors_config` to define a multi-vector schema.
+        """
+        if not vectors_config:
+
+            vectors_config = {
+                # Raw visual embeddings (OpenCLIP ViT-H/14 → 1024-D)
+                "image_vector": qdrant.VectorParams(
+                    size=1024, distance=qdrant.Distance.COSINE
+                ),
+                # Language embeddings of your BLIP-2 captions → 1024-D
+                "caption_vector": qdrant.VectorParams(
+                    size=1024, distance=qdrant.Distance.COSINE
+                ),
+                # Object-region embeddings (YOLO crop + Sentence-BERT) → 1024-D
+                "region_vector": qdrant.VectorParams(
+                    size=1024, distance=qdrant.Distance.COSINE
+                ),
+                # Geo-location unit vectors (RegioNet) → 3-D
+                "geo_vector": qdrant.VectorParams(
+                    size=3, distance=qdrant.Distance.COSINE
+                ),
+            }
+
+        return self._run_sync(
+            self._create_vs_for_user_async(
+                owner_id,
+                name,
+                vector_size,
+                distance_metric,
+                config,
+                vectors_config,
+            )
+        )
 
     def create_vector_store_for_user(
         self,

projectdavid/entity.py

@@ -30,11 +30,15 @@ class Entity:
         self,
         base_url: Optional[str] = None,
         api_key: Optional[str] = None,
+        file_processor_kwargs: Optional[dict] = None,
     ):
         """
         Initialize the main client with configuration.
         Optionally, a configuration object can be injected to decouple from environment variables.
         """
+
+        self.file_processor_kwargs = file_processor_kwargs
+
         self.base_url = base_url or os.getenv(
             "ENTITIES_BASE_URL", "http://localhost:9000/"
         )
@@ -145,7 +149,9 @@ class Entity:
     def vectors(self) -> VectorStoreClient:
         if self._vectors_client is None:
             self._vectors_client = VectorStoreClient(
-                base_url=self.base_url, api_key=self.api_key
+                base_url=self.base_url,
+                api_key=self.api_key,
+                file_processor_kwargs=self.file_processor_kwargs,
             )
 
         return self._vectors_client

{projectdavid-1.33.0.dist-info → projectdavid-1.33.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: projectdavid
-Version: 1.33.0
+Version: 1.33.2
 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.0.dist-info → projectdavid-1.33.2.dist-info}/RECORD

@@ -1,6 +1,6 @@
 projectdavid/__init__.py,sha256=QKjYL0-e4ifiRWyp05lK_dGm0u_jaxo36fyF-c0qqeI,224
 projectdavid/_version.py,sha256=_qiy4uXbgWEU-in9CM9h6tk4Ddhzpvv-HnoAYSuiGg0,100
-projectdavid/entity.py,sha256=uqRggs1zLWYnKnycPwO5pUhVegOJN8nCC0R9F2RYROI,5634
+projectdavid/entity.py,sha256=SRZg3E9Cifz3O0Ax4FNeWXWSgOZFKUBzLYtdjbH-e_w,5832
 projectdavid/events.py,sha256=m_vu5BgphrM6dpBmnkTXK8jczwyYXO1UwXXI9340vjQ,720
 projectdavid/serializers.py,sha256=OdipJGXGGjRoZBcPkpmHSDgsNptXUg0CMn2TQoCZIOI,1723
 projectdavid/clients/actions_client.py,sha256=9DVkQzRA3wQ7DIHR1ab6VFQYcIkhExAQz1-EK8MEt2c,17006
@@ -20,7 +20,7 @@ projectdavid/clients/threads_client.py,sha256=ekzU5w14zftmtmFkiec3NC90Of-_KVSUY1
 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=q-ZgRQVX_S3nMrKYhmvkVrDjDRzM3ZFzUF55HBGRTe8,12861
-projectdavid/clients/vectors.py,sha256=sDbT1GbsIuwzvInuDDcyX4scuuLJDWWgL7Fmc-pguRQ,37672
+projectdavid/clients/vectors.py,sha256=co6CStRnsWk0yO5EyH5AV2c9vMYaGk3VAwIQD4Y-N20,40108
 projectdavid/constants/platform.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 projectdavid/services/logging_service.py,sha256=jdoRL46E42Ar8JFTDOV-xVD67CulcHSN-xhcEqA5CXQ,2643
 projectdavid/synthesis/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -34,8 +34,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.0.dist-info/licenses/LICENSE,sha256=_8yjiEGttpS284BkfhXxfERqTRZW_tUaHiBB0GTJTMg,4563
-projectdavid-1.33.0.dist-info/METADATA,sha256=1Yhr24U_T9-Wv6Ladi67ZjTqwvcmpP4C241DtcCJMRk,11554
-projectdavid-1.33.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-projectdavid-1.33.0.dist-info/top_level.txt,sha256=kil8GU4s7qYRfNnzGnFHhZnSNRSxgNG-J4HLgQMmMtw,13
-projectdavid-1.33.0.dist-info/RECORD,,
+projectdavid-1.33.2.dist-info/licenses/LICENSE,sha256=_8yjiEGttpS284BkfhXxfERqTRZW_tUaHiBB0GTJTMg,4563
+projectdavid-1.33.2.dist-info/METADATA,sha256=AXinjpf8D5k2ySQ7SE_roXMsr8SaPHlklAYwXW6cjG0,11554
+projectdavid-1.33.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+projectdavid-1.33.2.dist-info/top_level.txt,sha256=kil8GU4s7qYRfNnzGnFHhZnSNRSxgNG-J4HLgQMmMtw,13
+projectdavid-1.33.2.dist-info/RECORD,,