From b117b36e5bea44891bd23071de3c544a5bdbb798 Mon Sep 17 00:00:00 2001
From: streetycat <305190374@qq.com>
Date: Mon, 16 Oct 2023 10:00:53 +0000
Subject: [PATCH 1/2] fix bugs

---
 rootfs/agents/math_teacher/agent.toml | 2 +-
 src/aios_kernel/compute_kernel.py     | 2 +-
 src/aios_kernel/workflow.py           | 3 ++-
 src/service/aios_shell/aios_shell.py  | 5 +++--
 4 files changed, 7 insertions(+), 5 deletions(-)

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 @@ class ComputeKernel:
                 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 @@ class Workflow:
     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 @@ class AIOS_Shell:
                 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)

From d945796f7e539d547777cac8d769decab80aaccb Mon Sep 17 00:00:00 2001
From: streetycat <305190374@qq.com>
Date: Mon, 16 Oct 2023 10:02:59 +0000
Subject: [PATCH 2/2] remove test

---
 rootfs/agents/math_teacher/agent.toml | 2 +-
 src/aios_kernel/compute_kernel.py     | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/rootfs/agents/math_teacher/agent.toml b/rootfs/agents/math_teacher/agent.toml
index 3f84b7a..871b29a 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 = "70B"
+llm_model_name = "gpt-4-0613"
 [[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 3cc5cf2..3f97708 100644
--- a/src/aios_kernel/compute_kernel.py
+++ b/src/aios_kernel/compute_kernel.py
@@ -123,7 +123,7 @@ class ComputeKernel:
                 if task_req.state == ComputeTaskState.ERROR:
                     break
 
-                if check_times >= 12000:
+                if check_times >= 120:
                     task_req.state = ComputeTaskState.ERROR
                     break