scriptoriumadmin/ReMe
1
0
Fork 0
mirror of https://github.com/agentscope-ai/ReMe.git synced 2026-08-28 05:25:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 11

feat(summary): support list input for tool names

Browse source
This commit is contained in:
jinli.yl 2025-10-27 21:56:20 +08:00
parent 7e82a3179e
commit 0039cf8bd9
1 changed files with 10 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
reme_ai/summary/tool/summary_tool_memory_op.py
View file

@ -1,5 +1,5 @@
import asyncio
from typing import List
from typing import List, Union
from flowllm import C, BaseAsyncOp
from flowllm.enumeration.role import Role
@ -124,7 +124,7 @@ class SummaryToolMemoryOp(BaseAsyncOp):
return result
async def async_execute(self):
tool_names: str = self.context.get("tool_names", "")
tool_names: Union[str, List[str]] = self.context.get("tool_names", "")
workspace_id: str = self.context.workspace_id
if not tool_names:
@ -133,8 +133,14 @@ class SummaryToolMemoryOp(BaseAsyncOp):
self.context.response.success = False
return
# Split tool names by comma
tool_name_list = [name.strip() for name in tool_names.split(",") if name.strip()]
# Normalize tool_names to list
if isinstance(tool_names, str):
# Split tool names by comma
tool_name_list = [name.strip() for name in tool_names.split(",") if name.strip()]
else:
# Already a list
tool_name_list = [name.strip() for name in tool_names if name.strip()]
logger.info(f"workspace_id={workspace_id} processing {len(tool_name_list)} tools: {tool_name_list}")
# Search for each tool in the vector store