gyt/opendan
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add proxy in config

Browse Source
This commit is contained in:
streetycat
2023-09-19 08:29:55 +00:00
committed by zhangzhen
parent d5073cc21b
commit c9af0cfc0d
4 changed files with 59 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/service/aios_shell/aios_shell.py
+14 -8
View File
@@ -23,7 +23,7 @@ directory = os.path.dirname(__file__)
sys.path.append(directory + '/../../')
from aios_kernel import AIOS_Version,UserConfigItem,AIStorage,Workflow,AIAgent,AgentMsg,AgentMsgStatus,ComputeKernel,OpenAI_ComputeNode,AIBus,AIChatSession,AgentTunnel,TelegramTunnel,CalenderEnvironment,Environment,EmailTunnel,LocalLlama_ComputeNode
import proxy
sys.path.append(directory + '/../../component/')
from agent_manager import AgentManager
@@ -55,6 +55,7 @@ class AIOS_Shell:
openai_node.declare_user_config()
user_config.add_user_config("shell.current","last opened target and topic",True,"default@Jarvis")
proxy.declare_user_config()
async def _handle_no_target_msg(self,bus:AIBus,msg:AgentMsg) -> bool:
@@ -95,7 +96,7 @@ class AIOS_Shell:
llama_ai_node = LocalLlama_ComputeNode()
await llama_ai_node.start()
#ComputeKernel.get_instance().add_compute_node(llama_ai_node)
# ComputeKernel.get_instance().add_compute_node(llama_ai_node)
await ComputeKernel.get_instance().start()
@@ -192,13 +193,16 @@ class AIOS_Shell:
show_text = FormattedText([("class:title", f"set config failed!")])
if len(args) == 1:
key = args[0]
config_item = AIStorage.get_instance().get_user_config().get_config_item(key)
old_value = AIStorage.get_instance().get_user_config().get_value(key)
config_item = AIStorage.get_instance().get_user_config().get_user_config(key)
old_value = None
if config_item is not None:
value = await session.prompt_async(f"{key} : {config_item.desc} \nCurrent : {old_value}\nPlease input new value:",style=shell_style)
AIStorage.get_instance().get_user_config().set_value(key,value)
await AIStorage.get_instance().get_user_config().save_to_user_config()
show_text = FormattedText([("class:title", f"set {key} to {value} success!")])
old_value = config_item.value
value = await session.prompt_async(f"{key} : {config_item.desc} \nCurrent : {old_value}\nPlease input new value:",style=shell_style)
AIStorage.get_instance().get_user_config().set_user_config(key,value)
await AIStorage.get_instance().get_user_config().save_value_to_user_config()
show_text = FormattedText([("class:title", f"set {key} to {value} success!")])
return show_text
case 'connect':
@@ -390,6 +394,8 @@ async def main():
if is_daemon:
return await main_daemon_loop(shell)
proxy.apply_storage()
#TODO: read last input config
completer = WordCompleter(['/send $target $msg $topic',
'/open $target $topic',