scriptoriumadmin/ReMe
1
0
Fork 0
mirror of https://github.com/agentscope-ai/ReMe.git synced 2026-10-08 03:10:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

add db action: delete_ids & insert

Browse source
This commit is contained in:
Dengjiaji 2025-08-13 14:53:57 +08:00
parent 8cf10453fe
commit 98bb937efe
2 changed files with 7 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
experiencemaker/experiencemaker/op/vector_store/vector_store_action_op.py
View file

@ -20,6 +20,12 @@ class VectorStoreActionOp(BaseOp):
elif request.action == "delete":
result = self.vector_store.delete_workspace(workspace_id=request.workspace_id)
elif request.action == "delete_ids":
result = self.vector_store.delete(workspace_id=request.workspace_id,node_ids = request.experience_ids)
elif request.action == "insert":
result = self.vector_store.insert(workspace_id=request.workspace_id, nodes = request.experience_ids)
elif request.action == "dump":
def node_to_experience(node: VectorNode) -> dict:
return vector_node_to_experience(node).model_dump()

1
experiencemaker/experiencemaker/schema/request.py
View file

@ -24,6 +24,7 @@ class VectorStoreRequest(BaseRequest):
action: str = Field(default="")
src_workspace_id: str = Field(default="")
path: str = Field(default="")
experience_ids: List[str] = Field(default_factory=list)
class AgentRequest(BaseRequest):