镜像自地址
https://github.com/binary-husky/gpt_academic.git
已同步 2025-12-06 22:46:48 +00:00
67 行
3.3 KiB
Python
67 行
3.3 KiB
Python
from pydantic import BaseModel, Field
|
|
from typing import List
|
|
from toolbox import update_ui_lastest_msg
|
|
from request_llm.bridge_all import predict_no_ui_long_connection
|
|
from crazy_functions.json_fns.pydantic_io import GptJsonIO
|
|
import copy, json, pickle, os, sys
|
|
|
|
|
|
def modify_configuration_hot(txt, llm_kwargs, plugin_kwargs, chatbot, history, system_prompt, user_intention):
|
|
import config
|
|
|
|
# ⭐ ⭐ ⭐ 读取可配置项目条目
|
|
from enum import Enum
|
|
names = {}
|
|
for k, v in config.__dict__.items():
|
|
if k.startswith('__'): continue
|
|
names.update({k:k})
|
|
# if len(names) > 20: break # 限制最多前10个配置项，如果太多了会导致gpt无法理解
|
|
|
|
ConfigOptions = Enum('ConfigOptions', names)
|
|
class ModifyConfigurationIntention(BaseModel):
|
|
which_config_to_modify: ConfigOptions = Field(description="the name of the configuration to modify, you must choose from one of the ConfigOptions enum.", default=None)
|
|
new_option_value: str = Field(description="the new value of the option", default=None)
|
|
|
|
# ⭐ ⭐ ⭐ 分析用户意图
|
|
yield from update_ui_lastest_msg(lastmsg=f"正在执行任务: {txt}\n\n读取新配置中", chatbot=chatbot, history=history, delay=0)
|
|
gpt_json_io = GptJsonIO(ModifyConfigurationIntention)
|
|
inputs = "Analyze how to change configuration according to following user input, answer me with json: \n\n" + \
|
|
">>" + txt + '\n\n' + \
|
|
gpt_json_io.format_instructions
|
|
|
|
run_gpt_fn = lambda inputs, sys_prompt: predict_no_ui_long_connection(
|
|
inputs=inputs, llm_kwargs=llm_kwargs, history=[], sys_prompt=sys_prompt, observe_window=[])
|
|
user_intention = gpt_json_io.generate_output_auto_repair(run_gpt_fn(inputs, ""), run_gpt_fn)
|
|
|
|
explicit_conf = user_intention.which_config_to_modify.value
|
|
|
|
ok = (explicit_conf in txt)
|
|
if ok:
|
|
yield from update_ui_lastest_msg(
|
|
lastmsg=f"正在执行任务: {txt}\n\n新配置{explicit_conf}={user_intention.new_option_value}",
|
|
chatbot=chatbot, history=history, delay=1
|
|
)
|
|
yield from update_ui_lastest_msg(
|
|
lastmsg=f"正在执行任务: {txt}\n\n新配置{explicit_conf}={user_intention.new_option_value}\n\n正在修改配置中",
|
|
chatbot=chatbot, history=history, delay=2
|
|
)
|
|
|
|
# ⭐ ⭐ ⭐ 立即应用配置
|
|
from toolbox import set_conf
|
|
set_conf(explicit_conf, user_intention.new_option_value)
|
|
|
|
yield from update_ui_lastest_msg(
|
|
lastmsg=f"正在执行任务: {txt}\n\n配置修改完成，重新页面即可生效。", chatbot=chatbot, history=history, delay=1
|
|
)
|
|
else:
|
|
yield from update_ui_lastest_msg(
|
|
lastmsg=f"失败，如果需要配置{explicit_conf}，您需要明确说明并在指令中提到它。", chatbot=chatbot, history=history, delay=5
|
|
)
|
|
|
|
def modify_configuration_reboot(txt, llm_kwargs, plugin_kwargs, chatbot, history, system_prompt, user_intention):
|
|
yield from modify_configuration_hot(txt, llm_kwargs, plugin_kwargs, chatbot, history, system_prompt, user_intention)
|
|
yield from update_ui_lastest_msg(
|
|
lastmsg=f"正在执行任务: {txt}\n\n配置修改完成，五秒后即将重启！若出现报错请无视即可。", chatbot=chatbot, history=history, delay=5
|
|
)
|
|
os.execl(sys.executable, sys.executable, *sys.argv)
|