crewplus 0.2.99__tar.gz → 0.3.1__tar.gz

{crewplus-0.2.99 → crewplus-0.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: crewplus
-Version: 0.2.99
+Version: 0.3.1
 Summary: Base services for CrewPlus AI applications
 Author-Email: Tim Liu <tim@opsmateai.com>
 License: MIT

{crewplus-0.2.99 → crewplus-0.3.1}/crewplus/vectorstores/milvus/vdb_service.py

@@ -9,7 +9,7 @@ import logging
 import time
 import uuid
 from collections import defaultdict
-from typing import List, Dict, Optional
+from typing import AsyncIterator, List, Dict, Optional, Any
 
 from langchain_core.documents import Document
 from langchain_core.embeddings import Embeddings
@@ -797,6 +797,81 @@ class VDBService(object):
         except Exception as e:
             raise RuntimeError(f"delete collection data failed: {str(e)}")
 
+    async def aquery_rows(
+        self,
+        *,
+        collection_name: str,
+        filter_expr: str,
+        output_fields: List[str],
+        limit: int,
+        offset: int = 0,
+    ) -> List[dict[str, Any]]:
+        if not collection_name:
+            raise ValueError("collection_name must be provided.")
+        if limit <= 0:
+            return []
+        store = await self.aget_vector_store(collection_name)
+        collection = getattr(store, "col", None)
+        if collection is None:
+            raise RuntimeError(
+                "The vector store instance does not expose a pymilvus Collection via `.col`."
+            )
+        return await asyncio.to_thread(
+            collection.query,
+            expr=filter_expr,
+            output_fields=output_fields,
+            limit=limit,
+            offset=offset,
+        )
+
+    async def aiter_query_rows(
+        self,
+        *,
+        collection_name: str,
+        filter_expr: str,
+        output_fields: List[str],
+        batch_size: int,
+    ) -> AsyncIterator[List[dict[str, Any]]]:
+        if not collection_name:
+            raise ValueError("collection_name must be provided.")
+        if batch_size <= 0:
+            return
+        store = await self.aget_vector_store(collection_name)
+        collection = getattr(store, "col", None)
+        if collection is None:
+            raise RuntimeError(
+                "The vector store instance does not expose a pymilvus Collection via `.col`."
+            )
+        iterator = await asyncio.to_thread(
+            collection.query_iterator,
+            expr=filter_expr,
+            output_fields=output_fields,
+            batch_size=batch_size,
+        )
+        try:
+            while True:
+                batch_rows = await asyncio.to_thread(iterator.next)
+                if not batch_rows:
+                    break
+                yield batch_rows
+        finally:
+            await asyncio.to_thread(iterator.close)
+
+    async def asimilarity_search_with_score(
+        self,
+        *,
+        collection_name: str,
+        query: str,
+        k: int,
+        expr: str | None = None,
+        embeddings: Embeddings | None = None,
+    ) -> list[tuple[Document, float]]:
+        store = await self.aget_vector_store(collection_name, embeddings)
+        kwargs: dict[str, Any] = {"k": k}
+        if expr:
+            kwargs["expr"] = expr
+        return await asyncio.to_thread(store.similarity_search_with_score, query, **kwargs)
+
     async def aget_docs_by_ids(self, collection_names: List[str], ids: List[str], embeddings: Embeddings, output_fields: List[str] = None) -> List[Document]:
         """
         Retrieves documents from multiple collections by their primary key IDs using a direct query.

{crewplus-0.2.99 → crewplus-0.3.1}/pyproject.toml

@@ -6,7 +6,7 @@ build-backend = "pdm.backend"
 
 [project]
 name = "crewplus"
-version = "0.2.99"
+version = "0.3.01"
 description = "Base services for CrewPlus AI applications"
 authors = [
     { name = "Tim Liu", email = "tim@opsmateai.com" },