Refactor vector query methods to support optional ID filtering

- Updated BaseVectorStorage query method signature to accept optional IDs
- Modified operate.py to pass query parameter IDs to vector storage queries
- Updated PostgreSQL vector storage SQL templates to filter results by document IDs
- Removed unused parameters and simplified query logic across multiple files
This commit is contained in:
Roy
2025-03-08 15:43:17 +00:00
parent bbe139cfeb
commit 528fb11364
4 changed files with 85 additions and 34 deletions

View File

@@ -108,9 +108,8 @@ class BaseVectorStorage(StorageNameSpace, ABC):
embedding_func: EmbeddingFunc
cosine_better_than_threshold: float = field(default=0.2)
meta_fields: set[str] = field(default_factory=set)
@abstractmethod
async def query(self, query: str, top_k: int, ids: list[str] = None) -> list[dict[str, Any]]:
async def query(self, query: str, top_k: int, ids: list[str] | None = None) -> list[dict[str, Any]]:
"""Query the vector storage and retrieve top_k results."""
@abstractmethod