feat: customized tool-use

astrbot/message/handler.py

@@ -214,7 +214,9 @@ class MessageHandler():
                         return MessageResult("AstrBot Function-calling 异常：未找到请求的函数调用。")
                     try:
                         args = json.loads(llm_result.arguments)
-                        function_invoked_ret = await func_obj(**args)
+                        args['ame'] = message
+                        args['context'] = self.context
+                        llm_result = await func_obj(**args)
                         has_func = True
                     except BaseException as e:
                         traceback.print_exc()

util/agent/func_call.py

@@ -1,6 +1,5 @@
 from model.provider.provider import Provider
 import json
-import time
 import textwrap
 
 class FuncCallJsonFormatError(Exception):

util/agent/web_searcher.py

@@ -1,6 +1,4 @@
-import traceback
 import random
-import json
 import aiohttp
 import os