|
@@ -127,12 +127,13 @@ class MessagePushAgent(SimpleOpenAICompatibleChatAgent):
|
|
|
tools: Optional[List[FunctionTool]] = None,
|
|
|
generate_cfg: Optional[dict] = None, max_run_step: Optional[int] = None):
|
|
|
system_prompt = system_prompt or DEFAULT_SYSTEM_PROMPT
|
|
|
- tools = tools or []
|
|
|
- tools = tools.copy()
|
|
|
- tools.extend([
|
|
|
- *ImageDescriber().get_tools(),
|
|
|
- *MessageNotifier().get_tools(),
|
|
|
- ])
|
|
|
+ if tools is None:
|
|
|
+ self.tools = [
|
|
|
+ *ImageDescriber().get_tools(),
|
|
|
+ *MessageNotifier().get_tools()
|
|
|
+ ]
|
|
|
+ else:
|
|
|
+ self.tools = [*tools]
|
|
|
super().__init__(model, system_prompt, tools, generate_cfg, max_run_step)
|
|
|
|
|
|
def generate_message(self, context: Dict, dialogue_history: List[Dict],
|