暂时移除插件锁定

This commit is contained in:
binary-husky 2023-06-01 09:39:00 +08:00
parent 2b620ccf2e
commit 2f037f30d5
2 changed files with 94 additions and 76 deletions

View File

@ -126,7 +126,7 @@ def get_crazy_functions():
###################### 第二组插件 ########################### ###################### 第二组插件 ###########################
# [第二组插件]: 经过充分测试 # [第二组插件]: 经过充分测试
from crazy_functions.批量总结PDF文档 import 批量总结PDF文档 from crazy_functions.批量总结PDF文档 import 批量总结PDF文档
from crazy_functions.批量总结PDF文档pdfminer import 批量总结PDF文档pdfminer # from crazy_functions.批量总结PDF文档pdfminer import 批量总结PDF文档pdfminer
from crazy_functions.批量翻译PDF文档_多线程 import 批量翻译PDF文档 from crazy_functions.批量翻译PDF文档_多线程 import 批量翻译PDF文档
from crazy_functions.谷歌检索小助手 import 谷歌检索小助手 from crazy_functions.谷歌检索小助手 import 谷歌检索小助手
from crazy_functions.理解PDF文档内容 import 理解PDF文档内容标准文件输入 from crazy_functions.理解PDF文档内容 import 理解PDF文档内容标准文件输入
@ -152,17 +152,16 @@ def get_crazy_functions():
# HotReload 的意思是热更新,修改函数插件代码后,不需要重启程序,代码直接生效 # HotReload 的意思是热更新,修改函数插件代码后,不需要重启程序,代码直接生效
"Function": HotReload(批量总结PDF文档) "Function": HotReload(批量总结PDF文档)
}, },
"[测试功能] 批量总结PDF文档pdfminer": { # "[测试功能] 批量总结PDF文档pdfminer": {
"Color": "stop", # "Color": "stop",
"AsButton": False, # 加入下拉菜单中 # "AsButton": False, # 加入下拉菜单中
"Function": HotReload(批量总结PDF文档pdfminer) # "Function": HotReload(批量总结PDF文档pdfminer)
}, # },
"谷歌学术检索助手输入谷歌学术搜索页url": { "谷歌学术检索助手输入谷歌学术搜索页url": {
"Color": "stop", "Color": "stop",
"AsButton": False, # 加入下拉菜单中 "AsButton": False, # 加入下拉菜单中
"Function": HotReload(谷歌检索小助手) "Function": HotReload(谷歌检索小助手)
}, },
"理解PDF文档内容 模仿ChatPDF": { "理解PDF文档内容 模仿ChatPDF": {
# HotReload 的意思是热更新,修改函数插件代码后,不需要重启程序,代码直接生效 # HotReload 的意思是热更新,修改函数插件代码后,不需要重启程序,代码直接生效
"Color": "stop", "Color": "stop",
@ -181,7 +180,7 @@ def get_crazy_functions():
"AsButton": False, # 加入下拉菜单中 "AsButton": False, # 加入下拉菜单中
"Function": HotReload(Latex英文纠错) "Function": HotReload(Latex英文纠错)
}, },
"[测试功能] 中文Latex项目全文润色输入路径或上传压缩包": { "中文Latex项目全文润色输入路径或上传压缩包": {
# HotReload 的意思是热更新,修改函数插件代码后,不需要重启程序,代码直接生效 # HotReload 的意思是热更新,修改函数插件代码后,不需要重启程序,代码直接生效
"Color": "stop", "Color": "stop",
"AsButton": False, # 加入下拉菜单中 "AsButton": False, # 加入下拉菜单中
@ -210,7 +209,9 @@ def get_crazy_functions():
}) })
###################### 第三组插件 ########################### ###################### 第三组插件 ###########################
# [第三组插件]: 尚未充分测试的函数插件,放在这里 # [第三组插件]: 尚未充分测试的函数插件
try:
from crazy_functions.下载arxiv论文翻译摘要 import 下载arxiv论文并翻译摘要 from crazy_functions.下载arxiv论文翻译摘要 import 下载arxiv论文并翻译摘要
function_plugins.update({ function_plugins.update({
"一键下载arxiv论文并翻译摘要先在input输入编号如1812.10695": { "一键下载arxiv论文并翻译摘要先在input输入编号如1812.10695": {
@ -219,7 +220,10 @@ def get_crazy_functions():
"Function": HotReload(下载arxiv论文并翻译摘要) "Function": HotReload(下载arxiv论文并翻译摘要)
} }
}) })
except:
print('Load function plugin failed')
try:
from crazy_functions.联网的ChatGPT import 连接网络回答问题 from crazy_functions.联网的ChatGPT import 连接网络回答问题
function_plugins.update({ function_plugins.update({
"连接网络回答问题(先输入问题,再点击按钮,需要访问谷歌)": { "连接网络回答问题(先输入问题,再点击按钮,需要访问谷歌)": {
@ -228,7 +232,10 @@ def get_crazy_functions():
"Function": HotReload(连接网络回答问题) "Function": HotReload(连接网络回答问题)
} }
}) })
except:
print('Load function plugin failed')
try:
from crazy_functions.解析项目源代码 import 解析任意code项目 from crazy_functions.解析项目源代码 import 解析任意code项目
function_plugins.update({ function_plugins.update({
"解析项目源代码(手动指定和筛选源代码文件类型)": { "解析项目源代码(手动指定和筛选源代码文件类型)": {
@ -239,6 +246,10 @@ def get_crazy_functions():
"Function": HotReload(解析任意code项目) "Function": HotReload(解析任意code项目)
}, },
}) })
except:
print('Load function plugin failed')
try:
from crazy_functions.询问多个大语言模型 import 同时问询_指定模型 from crazy_functions.询问多个大语言模型 import 同时问询_指定模型
function_plugins.update({ function_plugins.update({
"询问多个GPT模型手动指定询问哪些模型": { "询问多个GPT模型手动指定询问哪些模型": {
@ -249,6 +260,10 @@ def get_crazy_functions():
"Function": HotReload(同时问询_指定模型) "Function": HotReload(同时问询_指定模型)
}, },
}) })
except:
print('Load function plugin failed')
try:
from crazy_functions.图片生成 import 图片生成 from crazy_functions.图片生成 import 图片生成
function_plugins.update({ function_plugins.update({
"图片生成先切换模型到openai或api2d": { "图片生成先切换模型到openai或api2d": {
@ -259,6 +274,10 @@ def get_crazy_functions():
"Function": HotReload(图片生成) "Function": HotReload(图片生成)
}, },
}) })
except:
print('Load function plugin failed')
try:
from crazy_functions.总结音视频 import 总结音视频 from crazy_functions.总结音视频 import 总结音视频
function_plugins.update({ function_plugins.update({
"批量总结音视频(输入路径或上传压缩包)": { "批量总结音视频(输入路径或上传压缩包)": {
@ -269,6 +288,9 @@ def get_crazy_functions():
"Function": HotReload(总结音视频) "Function": HotReload(总结音视频)
} }
}) })
except:
print('Load function plugin failed')
try: try:
from crazy_functions.数学动画生成manim import 动画生成 from crazy_functions.数学动画生成manim import 动画生成
function_plugins.update({ function_plugins.update({
@ -298,7 +320,7 @@ def get_crazy_functions():
try: try:
from crazy_functions.Langchain知识库 import 知识库问答 from crazy_functions.Langchain知识库 import 知识库问答
function_plugins.update({ function_plugins.update({
"构建知识库(请先上传文件素材)": { "[功能尚不稳定] 构建知识库(请先上传文件素材)": {
"Color": "stop", "Color": "stop",
"AsButton": False, "AsButton": False,
"AdvancedArgs": True, "AdvancedArgs": True,
@ -306,9 +328,13 @@ def get_crazy_functions():
"Function": HotReload(知识库问答) "Function": HotReload(知识库问答)
} }
}) })
except:
print('Load function plugin failed')
try:
from crazy_functions.Langchain知识库 import 读取知识库作答 from crazy_functions.Langchain知识库 import 读取知识库作答
function_plugins.update({ function_plugins.update({
"知识库问答": { "[功能尚不稳定] 知识库问答": {
"Color": "stop", "Color": "stop",
"AsButton": False, "AsButton": False,
"AdvancedArgs": True, "AdvancedArgs": True,

View File

@ -59,14 +59,6 @@ def ArgsGeneralWrapper(f):
} }
chatbot_with_cookie = ChatBotWithCookies(cookies) chatbot_with_cookie = ChatBotWithCookies(cookies)
chatbot_with_cookie.write_list(chatbot) chatbot_with_cookie.write_list(chatbot)
if 'lock_plugin' in cookies and cookies['lock_plugin'] is not None:
# 处理插件锁定状态
module, fn_name = cookies['lock_plugin'].split('->')
f_hot_reload = getattr(importlib.import_module(module, fn_name), fn_name)
yield from HotReload(f_hot_reload)(txt_passon, llm_kwargs, plugin_kwargs, chatbot_with_cookie, history, system_prompt, *args)
else:
# 正常状态
yield from f(txt_passon, llm_kwargs, plugin_kwargs, chatbot_with_cookie, history, system_prompt, *args) yield from f(txt_passon, llm_kwargs, plugin_kwargs, chatbot_with_cookie, history, system_prompt, *args)
return decorated return decorated