diff --git a/src/langbot/pkg/plugin/handler.py b/src/langbot/pkg/plugin/handler.py index bdcd024d..645b28de 100644 --- a/src/langbot/pkg/plugin/handler.py +++ b/src/langbot/pkg/plugin/handler.py @@ -562,9 +562,7 @@ class RuntimeConnectionHandler(handler.Handler): limit = data.get('limit', 20) offset = data.get('offset', 0) try: - items, total = await self.ap.rag_runtime_service.vector_list( - collection_id, filters, limit, offset - ) + items, total = await self.ap.rag_runtime_service.vector_list(collection_id, filters, limit, offset) return handler.ActionResponse.success(data={'items': items, 'total': total}) except Exception as e: return _make_rag_error_response(e, 'VectorStoreError', collection_id=collection_id) diff --git a/src/langbot/pkg/vector/vdbs/chroma.py b/src/langbot/pkg/vector/vdbs/chroma.py index 4b880c0e..d0ee58fc 100644 --- a/src/langbot/pkg/vector/vdbs/chroma.py +++ b/src/langbot/pkg/vector/vdbs/chroma.py @@ -244,11 +244,13 @@ class ChromaVectorDatabase(VectorDatabase): items = [] for i, vid in enumerate(ids): - items.append({ - 'id': vid, - 'document': documents[i] if i < len(documents) else None, - 'metadata': metadatas[i] if i < len(metadatas) else {}, - }) + items.append( + { + 'id': vid, + 'document': documents[i] if i < len(documents) else None, + 'metadata': metadatas[i] if i < len(metadatas) else {}, + } + ) # Chroma col.count() gives total in collection; filtered count not available total = await asyncio.to_thread(col.count) if not filter else -1 diff --git a/src/langbot/pkg/vector/vdbs/milvus.py b/src/langbot/pkg/vector/vdbs/milvus.py index ec41722c..e9c66b09 100644 --- a/src/langbot/pkg/vector/vdbs/milvus.py +++ b/src/langbot/pkg/vector/vdbs/milvus.py @@ -368,15 +368,17 @@ class MilvusVectorDatabase(VectorDatabase): items = [] for row in results: - items.append({ - 'id': row.get('id', ''), - 'document': row.get('text'), - 'metadata': { - 'text': row.get('text', ''), - 'file_id': row.get('file_id', ''), - 'uuid': row.get('chunk_uuid', ''), - }, - }) + items.append( + { + 'id': row.get('id', ''), + 'document': row.get('text'), + 'metadata': { + 'text': row.get('text', ''), + 'file_id': row.get('file_id', ''), + 'uuid': row.get('chunk_uuid', ''), + }, + } + ) # Milvus query with count(*) total = -1 diff --git a/src/langbot/pkg/vector/vdbs/pgvector_db.py b/src/langbot/pkg/vector/vdbs/pgvector_db.py index dccf388b..35d05447 100644 --- a/src/langbot/pkg/vector/vdbs/pgvector_db.py +++ b/src/langbot/pkg/vector/vdbs/pgvector_db.py @@ -338,9 +338,7 @@ class PgVectorDatabase(VectorDatabase): ) count_stmt = ( - select(func.count()) - .select_from(PgVectorEntry) - .filter(PgVectorEntry.collection == collection) + select(func.count()).select_from(PgVectorEntry).filter(PgVectorEntry.collection == collection) ) if filter: @@ -356,15 +354,17 @@ class PgVectorDatabase(VectorDatabase): items = [] for row in rows: - items.append({ - 'id': row.id, - 'document': row.text or '', - 'metadata': { - 'text': row.text or '', - 'file_id': row.file_id or '', - 'uuid': row.chunk_uuid or '', - }, - }) + items.append( + { + 'id': row.id, + 'document': row.text or '', + 'metadata': { + 'text': row.text or '', + 'file_id': row.file_id or '', + 'uuid': row.chunk_uuid or '', + }, + } + ) return items, total except Exception as e: diff --git a/src/langbot/pkg/vector/vdbs/qdrant.py b/src/langbot/pkg/vector/vdbs/qdrant.py index 5df3a39a..f944846b 100644 --- a/src/langbot/pkg/vector/vdbs/qdrant.py +++ b/src/langbot/pkg/vector/vdbs/qdrant.py @@ -196,11 +196,13 @@ class QdrantVectorDatabase(VectorDatabase): break # Re-fetch payload if we skipped it during the skip phase payload = point.payload or {} - collected.append({ - 'id': str(point.id), - 'document': payload.get('text') or payload.get('document'), - 'metadata': payload, - }) + collected.append( + { + 'id': str(point.id), + 'document': payload.get('text') or payload.get('document'), + 'metadata': payload, + } + ) remaining_to_collect -= 1 if next_cursor is None: diff --git a/src/langbot/pkg/vector/vdbs/seekdb.py b/src/langbot/pkg/vector/vdbs/seekdb.py index ac3f8d04..676bc0d7 100644 --- a/src/langbot/pkg/vector/vdbs/seekdb.py +++ b/src/langbot/pkg/vector/vdbs/seekdb.py @@ -373,11 +373,13 @@ class SeekDBVectorDatabase(VectorDatabase): items = [] for i, vid in enumerate(ids): - items.append({ - 'id': vid, - 'document': documents[i] if i < len(documents) else None, - 'metadata': metadatas[i] if i < len(metadatas) else {}, - }) + items.append( + { + 'id': vid, + 'document': documents[i] if i < len(documents) else None, + 'metadata': metadatas[i] if i < len(metadatas) else {}, + } + ) total = await asyncio.to_thread(coll.count) if not filter else -1 return items, total