mirror of
https://github.com/agentscope-ai/ReMe.git
synced 2026-09-28 01:31:46 +00:00
test(vector-store): enhance fake vector store with descriptive docstrings
- Add docstring to list method indicating it returns no existing history nodes - Add docstring to delete method explaining it records deleted vector ids - Add docstring to insert method describing it records inserted vector nodes - Add docstring to search method stating it returns one matching history chunk - Rename parameters in list method from filters/limit/sort_key/reverse to _filters/_limit/_sort_key/_reverse
This commit is contained in:
parent
6e9e4e43d7
commit
572ccd4e00
1 changed files with 5 additions and 1 deletions
|
|
@ -197,13 +197,16 @@ async def test_add_history_stores_parent_and_chunk_nodes():
|
|||
self.deleted_ids = None
|
||||
self.inserted_nodes = []
|
||||
|
||||
async def list(self, filters=None, limit=None, sort_key=None, reverse=True):
|
||||
async def list(self, _filters=None, _limit=None, _sort_key=None, _reverse=True):
|
||||
"""Return no existing history nodes."""
|
||||
return []
|
||||
|
||||
async def delete(self, vector_ids):
|
||||
"""Record deleted vector ids."""
|
||||
self.deleted_ids = vector_ids
|
||||
|
||||
async def insert(self, nodes):
|
||||
"""Record inserted vector nodes."""
|
||||
self.inserted_nodes = nodes
|
||||
|
||||
vector_store = FakeVectorStore()
|
||||
|
|
@ -244,6 +247,7 @@ async def test_retrieve_history_filters_by_optional_history_id():
|
|||
self.search_calls = []
|
||||
|
||||
async def search(self, query, limit, filters):
|
||||
"""Return one matching history chunk."""
|
||||
self.search_calls.append({"query": query, "limit": limit, "filters": filters})
|
||||
return [
|
||||
MemoryNode(
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue