|
@@ -1,23 +1,20 @@ |
|
|
import json |
|
|
|
|
|
import os |
|
|
|
|
|
|
|
|
|
|
|
from chatgpt_tool_hub.apps import AppFactory |
|
|
from chatgpt_tool_hub.apps import AppFactory |
|
|
from chatgpt_tool_hub.apps.app import App |
|
|
from chatgpt_tool_hub.apps.app import App |
|
|
from chatgpt_tool_hub.tools.all_tool_list import get_all_tool_names |
|
|
|
|
|
|
|
|
from chatgpt_tool_hub.tools.tool_register import main_tool_register |
|
|
|
|
|
|
|
|
import plugins |
|
|
import plugins |
|
|
from bridge.bridge import Bridge |
|
|
from bridge.bridge import Bridge |
|
|
from bridge.context import ContextType |
|
|
from bridge.context import ContextType |
|
|
from bridge.reply import Reply, ReplyType |
|
|
from bridge.reply import Reply, ReplyType |
|
|
from common import const |
|
|
from common import const |
|
|
from config import conf |
|
|
|
|
|
|
|
|
from config import conf, get_appdata_dir |
|
|
from plugins import * |
|
|
from plugins import * |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@plugins.register( |
|
|
@plugins.register( |
|
|
name="tool", |
|
|
name="tool", |
|
|
desc="Arming your ChatGPT bot with various tools", |
|
|
desc="Arming your ChatGPT bot with various tools", |
|
|
version="0.4", |
|
|
|
|
|
|
|
|
version="0.5", |
|
|
author="goldfishh", |
|
|
author="goldfishh", |
|
|
desire_priority=0, |
|
|
desire_priority=0, |
|
|
) |
|
|
) |
|
@@ -26,6 +23,8 @@ class Tool(Plugin): |
|
|
super().__init__() |
|
|
super().__init__() |
|
|
self.handlers[Event.ON_HANDLE_CONTEXT] = self.on_handle_context |
|
|
self.handlers[Event.ON_HANDLE_CONTEXT] = self.on_handle_context |
|
|
|
|
|
|
|
|
|
|
|
self.tool_config = self._read_json() |
|
|
|
|
|
self.app_kwargs = self._build_tool_kwargs(self.tool_config.get("kwargs", {})) |
|
|
self.app = self._reset_app() |
|
|
self.app = self._reset_app() |
|
|
|
|
|
|
|
|
logger.info("[tool] inited") |
|
|
logger.info("[tool] inited") |
|
@@ -80,6 +79,8 @@ class Tool(Plugin): |
|
|
elif len(content_list) > 1: |
|
|
elif len(content_list) > 1: |
|
|
if content_list[1].strip() == "reset": |
|
|
if content_list[1].strip() == "reset": |
|
|
logger.debug("[tool]: reset config") |
|
|
logger.debug("[tool]: reset config") |
|
|
|
|
|
self.tool_config = self._read_json() |
|
|
|
|
|
self.app_kwargs = self._build_tool_kwargs(self.tool_config.get("kwargs", {})) |
|
|
self.app = self._reset_app() |
|
|
self.app = self._reset_app() |
|
|
reply.content = "重置工具成功" |
|
|
reply.content = "重置工具成功" |
|
|
e_context["reply"] = reply |
|
|
e_context["reply"] = reply |
|
@@ -91,17 +92,28 @@ class Tool(Plugin): |
|
|
|
|
|
|
|
|
e_context.action = EventAction.BREAK |
|
|
e_context.action = EventAction.BREAK |
|
|
return |
|
|
return |
|
|
|
|
|
|
|
|
query = content_list[1].strip() |
|
|
query = content_list[1].strip() |
|
|
|
|
|
|
|
|
|
|
|
use_one_tool = False |
|
|
|
|
|
for tool_name in main_tool_register.get_registered_tool_names(): |
|
|
|
|
|
if query.startswith(tool_name): |
|
|
|
|
|
use_one_tool = True |
|
|
|
|
|
query = query[len(tool_name):] |
|
|
|
|
|
break |
|
|
|
|
|
|
|
|
# Don't modify bot name |
|
|
# Don't modify bot name |
|
|
all_sessions = Bridge().get_bot("chat").sessions |
|
|
all_sessions = Bridge().get_bot("chat").sessions |
|
|
user_session = all_sessions.session_query(query, e_context["context"]["session_id"]).messages |
|
|
user_session = all_sessions.session_query(query, e_context["context"]["session_id"]).messages |
|
|
|
|
|
|
|
|
# chatgpt-tool-hub will reply you with many tools |
|
|
|
|
|
logger.debug("[tool]: just-go") |
|
|
logger.debug("[tool]: just-go") |
|
|
try: |
|
|
try: |
|
|
_reply = self.app.ask(query, user_session) |
|
|
|
|
|
|
|
|
if use_one_tool: |
|
|
|
|
|
_func, _ = main_tool_register.get_registered_tool()[tool_name] |
|
|
|
|
|
tool = _func(**self.app_kwargs) |
|
|
|
|
|
_reply = tool.run(query) |
|
|
|
|
|
else: |
|
|
|
|
|
# chatgpt-tool-hub will reply you with many tools |
|
|
|
|
|
_reply = self.app.ask(query, user_session) |
|
|
e_context.action = EventAction.BREAK_PASS |
|
|
e_context.action = EventAction.BREAK_PASS |
|
|
all_sessions.session_reply(_reply, e_context["context"]["session_id"]) |
|
|
all_sessions.session_reply(_reply, e_context["context"]["session_id"]) |
|
|
except Exception as e: |
|
|
except Exception as e: |
|
@@ -126,53 +138,108 @@ class Tool(Plugin): |
|
|
request_timeout = kwargs.get("request_timeout") |
|
|
request_timeout = kwargs.get("request_timeout") |
|
|
|
|
|
|
|
|
return { |
|
|
return { |
|
|
"debug": kwargs.get("debug", False), |
|
|
|
|
|
"openai_api_key": conf().get("open_ai_api_key", ""), |
|
|
|
|
|
"open_ai_api_base": conf().get("open_ai_api_base", "https://api.openai.com/v1"), |
|
|
|
|
|
"deployment_id": conf().get("azure_deployment_id", ""), |
|
|
|
|
|
"proxy": conf().get("proxy", ""), |
|
|
|
|
|
|
|
|
# 全局配置相关 |
|
|
|
|
|
"log": True, # tool 日志开关 |
|
|
|
|
|
"debug": kwargs.get("debug", False), # 输出更多日志 |
|
|
|
|
|
"no_default": kwargs.get("no_default", False), # 不要默认的工具,只加载自己导入的工具 |
|
|
|
|
|
"think_depth": kwargs.get("think_depth", 2), # 一个问题最多使用多少次工具 |
|
|
|
|
|
"proxy": conf().get("proxy", ""), # 科学上网 |
|
|
"request_timeout": request_timeout if request_timeout else conf().get("request_timeout", 120), |
|
|
"request_timeout": request_timeout if request_timeout else conf().get("request_timeout", 120), |
|
|
|
|
|
"temperature": kwargs.get("temperature", 0), # llm 温度,建议设置0 |
|
|
|
|
|
# LLM配置相关 |
|
|
|
|
|
"llm_api_key": conf().get("open_ai_api_key", ""), # 如果llm api用key鉴权,传入这里 |
|
|
|
|
|
"llm_api_base_url": conf().get("open_ai_api_base", "https://api.openai.com/v1"), # 支持openai接口的llm服务地址前缀 |
|
|
|
|
|
"deployment_id": conf().get("azure_deployment_id", ""), # azure openai会用到 |
|
|
# note: 目前tool暂未对其他模型测试,但这里仍对配置来源做了优先级区分,一般插件配置可覆盖全局配置 |
|
|
# note: 目前tool暂未对其他模型测试,但这里仍对配置来源做了优先级区分,一般插件配置可覆盖全局配置 |
|
|
"model_name": tool_model_name if tool_model_name else conf().get("model", "gpt-3.5-turbo"), |
|
|
|
|
|
"no_default": kwargs.get("no_default", False), |
|
|
|
|
|
"top_k_results": kwargs.get("top_k_results", 3), |
|
|
|
|
|
# for news tool |
|
|
|
|
|
"news_api_key": kwargs.get("news_api_key", ""), |
|
|
|
|
|
|
|
|
"model_name": tool_model_name if tool_model_name else conf().get("model", const.GPT35), |
|
|
|
|
|
# 工具配置相关 |
|
|
|
|
|
# for arxiv tool |
|
|
|
|
|
"arxiv_simple": kwargs.get("arxiv_simple", True), # 返回内容更精简 |
|
|
|
|
|
"arxiv_top_k_results": kwargs.get("arxiv_top_k_results", 2), # 只返回前k个搜索结果 |
|
|
|
|
|
"arxiv_sort_by": kwargs.get("arxiv_sort_by", "relevance"), # 搜索排序方式 ["relevance","lastUpdatedDate","submittedDate"] |
|
|
|
|
|
"arxiv_sort_order": kwargs.get("arxiv_sort_order", "descending"), # 搜索排序方式 ["ascending", "descending"] |
|
|
|
|
|
"arxiv_output_type": kwargs.get("arxiv_output_type", "text"), # 搜索结果类型 ["text", "pdf", "all"] |
|
|
# for bing-search tool |
|
|
# for bing-search tool |
|
|
"bing_subscription_key": kwargs.get("bing_subscription_key", ""), |
|
|
"bing_subscription_key": kwargs.get("bing_subscription_key", ""), |
|
|
|
|
|
"bing_search_url": kwargs.get("bing_search_url", "https://api.bing.microsoft.com/v7.0/search"), # 必应搜索的endpoint地址,无需修改 |
|
|
|
|
|
"bing_search_top_k_results": kwargs.get("bing_search_top_k_results", 2), # 只返回前k个搜索结果 |
|
|
|
|
|
"bing_search_simple": kwargs.get("bing_search_simple", True), # 返回内容更精简 |
|
|
|
|
|
"bing_search_output_type": kwargs.get("bing_search_output_type", "text"), # 搜索结果类型 ["text", "json"] |
|
|
|
|
|
# for email tool |
|
|
|
|
|
"email_nickname_mapping": kwargs.get("email_nickname_mapping", "{}"), # 关于人的代号对应的邮箱地址,可以不输入邮箱地址发送邮件。键为代号值为邮箱地址 |
|
|
|
|
|
"email_smtp_host": kwargs.get("email_smtp_host", ""), # 例如 'smtp.qq.com' |
|
|
|
|
|
"email_smtp_port": kwargs.get("email_smtp_port", ""), # 例如 587 |
|
|
|
|
|
"email_sender": kwargs.get("email_sender", ""), # 发送者的邮件地址 |
|
|
|
|
|
"email_authorization_code": kwargs.get("email_authorization_code", ""), # 发送者验证秘钥(可能不是登录密码) |
|
|
# for google-search tool |
|
|
# for google-search tool |
|
|
"google_api_key": kwargs.get("google_api_key", ""), |
|
|
"google_api_key": kwargs.get("google_api_key", ""), |
|
|
"google_cse_id": kwargs.get("google_cse_id", ""), |
|
|
"google_cse_id": kwargs.get("google_cse_id", ""), |
|
|
|
|
|
"google_simple": kwargs.get("google_simple", True), # 返回内容更精简 |
|
|
|
|
|
"google_output_type": kwargs.get("google_output_type", "text"), # 搜索结果类型 ["text", "json"] |
|
|
|
|
|
# for finance-news tool |
|
|
|
|
|
"finance_news_filter": kwargs.get("finance_news_filter", False), # 是否开启过滤 |
|
|
|
|
|
"finance_news_filter_list": kwargs.get("finance_news_filter_list", []), # 过滤词列表 |
|
|
|
|
|
"finance_news_simple": kwargs.get("finance_news_simple", True), # 返回内容更精简 |
|
|
|
|
|
"finance_news_repeat_news": kwargs.get("finance_news_repeat_news", False), # 是否过滤不返回。该tool每次返回约50条新闻,可能有重复新闻 |
|
|
|
|
|
# for morning-news tool |
|
|
|
|
|
"morning_news_api_key": kwargs.get("morning_news_api_key", ""), # api-key |
|
|
|
|
|
"morning_news_simple": kwargs.get("morning_news_simple", True), # 返回内容更精简 |
|
|
|
|
|
"morning_news_output_type": kwargs.get("morning_news_output_type", "text"), # 搜索结果类型 ["text", "image"] |
|
|
|
|
|
# for news-api tool |
|
|
|
|
|
"news_api_key": kwargs.get("news_api_key", ""), |
|
|
# for searxng-search tool |
|
|
# for searxng-search tool |
|
|
"searx_search_host": kwargs.get("searx_search_host", ""), |
|
|
|
|
|
|
|
|
"searxng_search_host": kwargs.get("searxng_search_host", ""), |
|
|
|
|
|
"searxng_search_top_k_results": kwargs.get("searxng_search_top_k_results", 2), # 只返回前k个搜索结果 |
|
|
|
|
|
"searxng_search_output_type": kwargs.get("searxng_search_output_type", "text"), # 搜索结果类型 ["text", "json"] |
|
|
|
|
|
# for sms tool |
|
|
|
|
|
"sms_nickname_mapping": kwargs.get("sms_nickname_mapping", "{}"), # 关于人的代号对应的手机号,可以不输入手机号发送sms。键为代号值为手机号 |
|
|
|
|
|
"sms_username": kwargs.get("sms_username", ""), # smsbao用户名 |
|
|
|
|
|
"sms_apikey": kwargs.get("sms_apikey", ""), # smsbao |
|
|
|
|
|
# for stt tool |
|
|
|
|
|
"stt_api_key": kwargs.get("stt_api_key", ""), # azure |
|
|
|
|
|
"stt_api_region": kwargs.get("stt_api_region", ""), # azure |
|
|
|
|
|
"stt_recognition_language": kwargs.get("stt_recognition_language", "zh-CN"), # 识别的语言类型 部分:en-US ja-JP ko-KR yue-CN zh-CN |
|
|
|
|
|
# for tts tool |
|
|
|
|
|
"tts_api_key": kwargs.get("tts_api_key", ""), # azure |
|
|
|
|
|
"tts_api_region": kwargs.get("tts_api_region", ""), # azure |
|
|
|
|
|
"tts_auto_detect": kwargs.get("tts_auto_detect", True), # 是否自动检测语音的语言 |
|
|
|
|
|
"tts_speech_id": kwargs.get("tts_speech_id", "zh-CN-XiaozhenNeural"), # 输出语音ID |
|
|
|
|
|
# for summary tool |
|
|
|
|
|
"summary_max_segment_length": kwargs.get("summary_max_segment_length", 2500), # 每2500tokens分段,多段触发总结tool |
|
|
|
|
|
# for terminal tool |
|
|
|
|
|
"terminal_nsfc_filter": kwargs.get("terminal_nsfc_filter", True), # 是否过滤llm输出的危险命令 |
|
|
|
|
|
"terminal_return_err_output": kwargs.get("terminal_return_err_output", True), # 是否输出错误信息 |
|
|
|
|
|
"terminal_timeout": kwargs.get("terminal_timeout", 20), # 允许命令最长执行时间 |
|
|
|
|
|
# for visual tool |
|
|
|
|
|
"caption_api_key": kwargs.get("caption_api_key", ""), # ali dashscope apikey |
|
|
|
|
|
# for browser tool |
|
|
|
|
|
"browser_use_summary": kwargs.get("browser_use_summary", True), # 是否对返回结果使用tool功能 |
|
|
|
|
|
# for url-get tool |
|
|
|
|
|
"url_get_use_summary": kwargs.get("url_get_use_summary", True), # 是否对返回结果使用tool功能 |
|
|
|
|
|
# for wechat tool |
|
|
|
|
|
"wechat_hot_reload": kwargs.get("wechat_hot_reload", True), # 是否使用热重载的方式发送wechat |
|
|
|
|
|
"wechat_cpt_path": kwargs.get("wechat_cpt_path", os.path.join(get_appdata_dir(), "itchat.pkl")), # wechat 配置文件(`itchat.pkl`) |
|
|
|
|
|
"wechat_send_group": kwargs.get("wechat_send_group", False), # 是否向群组发送消息 |
|
|
|
|
|
"wechat_nickname_mapping": kwargs.get("wechat_nickname_mapping", "{}"), # 关于人的代号映射关系。键为代号值为微信名(昵称、备注名均可) |
|
|
|
|
|
# for wikipedia tool |
|
|
|
|
|
"wikipedia_top_k_results": kwargs.get("wikipedia_top_k_results", 2), # 只返回前k个搜索结果 |
|
|
# for wolfram-alpha tool |
|
|
# for wolfram-alpha tool |
|
|
"wolfram_alpha_appid": kwargs.get("wolfram_alpha_appid", ""), |
|
|
"wolfram_alpha_appid": kwargs.get("wolfram_alpha_appid", ""), |
|
|
# for morning-news tool |
|
|
|
|
|
"morning_news_api_key": kwargs.get("morning_news_api_key", ""), |
|
|
|
|
|
# for visual_dl tool |
|
|
|
|
|
"cuda_device": kwargs.get("cuda_device", "cpu"), |
|
|
|
|
|
"think_depth": kwargs.get("think_depth", 3), |
|
|
|
|
|
"arxiv_summary": kwargs.get("arxiv_summary", True), |
|
|
|
|
|
"morning_news_use_llm": kwargs.get("morning_news_use_llm", False), |
|
|
|
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
def _filter_tool_list(self, tool_list: list): |
|
|
def _filter_tool_list(self, tool_list: list): |
|
|
valid_list = [] |
|
|
valid_list = [] |
|
|
for tool in tool_list: |
|
|
for tool in tool_list: |
|
|
if tool in get_all_tool_names(): |
|
|
|
|
|
|
|
|
if tool in main_tool_register.get_registered_tool_names(): |
|
|
valid_list.append(tool) |
|
|
valid_list.append(tool) |
|
|
else: |
|
|
else: |
|
|
logger.warning("[tool] filter invalid tool: " + repr(tool)) |
|
|
logger.warning("[tool] filter invalid tool: " + repr(tool)) |
|
|
return valid_list |
|
|
return valid_list |
|
|
|
|
|
|
|
|
def _reset_app(self) -> App: |
|
|
def _reset_app(self) -> App: |
|
|
tool_config = self._read_json() |
|
|
|
|
|
app_kwargs = self._build_tool_kwargs(tool_config.get("kwargs", {})) |
|
|
|
|
|
|
|
|
|
|
|
app = AppFactory() |
|
|
app = AppFactory() |
|
|
app.init_env(**app_kwargs) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
app.init_env(**self.app_kwargs) |
|
|
# filter not support tool |
|
|
# filter not support tool |
|
|
tool_list = self._filter_tool_list(tool_config.get("tools", [])) |
|
|
|
|
|
|
|
|
tool_list = self._filter_tool_list(self.tool_config.get("tools", [])) |
|
|
|
|
|
|
|
|
return app.create_app(tools_list=tool_list, **app_kwargs) |
|
|
|
|
|
|
|
|
return app.create_app(tools_list=tool_list, **self.app_kwargs) |