diff --git a/rootfs/agents/math_teacher/agent.toml b/rootfs/agents/math_teacher/agent.toml index 871b29a..3f84b7a 100644 --- a/rootfs/agents/math_teacher/agent.toml +++ b/rootfs/agents/math_teacher/agent.toml @@ -1,6 +1,6 @@ instance_id = "math_teacher" fullname = "the one" -llm_model_name = "gpt-4-0613" +llm_model_name = "70B" [[prompt]] role = "system" content = "You are a teacher who is proficient in mathematics" diff --git a/src/aios_kernel/compute_kernel.py b/src/aios_kernel/compute_kernel.py index 3f97708..3cc5cf2 100644 --- a/src/aios_kernel/compute_kernel.py +++ b/src/aios_kernel/compute_kernel.py @@ -123,7 +123,7 @@ async def check_timer(): if task_req.state == ComputeTaskState.ERROR: break - if check_times >= 120: + if check_times >= 12000: task_req.state = ComputeTaskState.ERROR break diff --git a/src/aios_kernel/workflow.py b/src/aios_kernel/workflow.py index f1e0064..5cf0bbf 100644 --- a/src/aios_kernel/workflow.py +++ b/src/aios_kernel/workflow.py @@ -544,8 +544,9 @@ async def _do_process_msg(): async def _get_prompt_from_session(self,the_role:AIRole,chatsession:AIChatSession) -> AgentPrompt: messages = chatsession.read_history(the_role.history_len) # read last 10 message result_prompt = AgentPrompt() + for msg in reversed(messages): - if msg.sender == chatsession.owner_id: + if msg.sender == the_role.role_id: result_prompt.messages.append({"role":"assistant","content":msg.body}) else: result_prompt.messages.append({"role":"user","content":f"{msg.body}"}) diff --git a/src/service/aios_shell/aios_shell.py b/src/service/aios_shell/aios_shell.py index b10ac84..d42096e 100644 --- a/src/service/aios_shell/aios_shell.py +++ b/src/service/aios_shell/aios_shell.py @@ -414,8 +414,8 @@ async def handle_node_commands(self, args): if len(args) < 4: return show_text - model_name = args[3] - url = args[4] + model_name = args[2] + url = args[3] ComputeNodeConfig.get_instance().remove_node("llama", url, model_name) ComputeNodeConfig.get_instance().save() else: @@ -747,6 +747,7 @@ async def main(): '/disable $feature', '/node add llama $model_name $url', '/node rm llama $model_name $url', + '/node list', '/show', '/exit', '/help'], ignore_case=True)