framework code has been completed basicly. Through the use of aios_shell, we are now able to get agents run able (at openai compute node)

2023-08-27 18:07:33 -07:00
parent 1a6cf1ad7a
commit ccbef2104b
25 changed files with 1011 additions and 198 deletions
@@ -0,0 +1,128 @@
+# aiso shell like bash for linux
+import asyncio
+import sys
+import os
+import logging
+
+from typing import Any, Optional, TypeVar, Tuple, Sequence
+import argparse
+
+from prompt_toolkit.formatted_text.base import AnyFormattedText
+from prompt_toolkit import Application, PromptSession, prompt
+from prompt_toolkit.selection import SelectionState
+from prompt_toolkit.history import FileHistory
+from prompt_toolkit.auto_suggest import AutoSuggestFromHistory
+from prompt_toolkit.completion import WordCompleter
+
+
+directory = os.path.dirname(__file__)
+sys.path.append(directory + '/../../')
+from aios_kernel import Workflow,AIAgent,AgentMsg,AgentMsgState,ComputeKernel,OpenAI_ComputeNode
+
+sys.path.append(directory + '/../../component/')
+from agent_manager import AgentManager
+from workflow_manager import WorkflowManager
+
+class AIOS_Shell:
+    def __init__(self,username:str) -> None:
+        self.username = username
+        self.user_chatsession = {}
+
+    async def initial(self) -> bool:
+        AgentManager().initial(directory + "/../../../rootfs/")
+        WorkflowManager().initial(directory + "/../../../../rootfs/workflows/workflows.cfg")
+        open_ai_node = OpenAI_ComputeNode()
+        open_ai_node.start()
+        ComputeKernel().add_compute_node(open_ai_node)
+        return True 
+        
+
+    def get_version(self) -> str:
+        return "0.0.1"
+
+    async def send_msg(self,msg:str,target_id:str,sender:str = None) -> str:
+        agent_msg = AgentMsg()
+        agent_msg.set(sender,target_id,msg)
+
+        agent : AIAgent = await AgentManager().get(target_id)
+        if agent is not None:
+           agent.post_msg(agent_msg)
+        
+        a_workflow = WorkflowManager().get_workflow(target_id)
+        if a_workflow is not None:
+            a_workflow.post_msg(agent_msg)
+
+
+        async def check_timer():
+            check_times = 0
+            while True:
+                if agent_msg.state == AgentMsgState.RESPONSED:
+                    break
+
+                if agent_msg.state == AgentMsgState.ERROR:
+                    break
+
+                if check_times >=  20:
+                    agent_msg.state = AgentMsgState.ERROR
+                    break
+
+                await asyncio.sleep(0.5)
+                check_times += 1
+            
+        await asyncio.create_task(check_timer())
+        if agent_msg.state == AgentMsgState.RESPONSED:
+            return agent_msg.resp_msg.body
+        
+        return "error!"
+
+
+    async def install_workflow(self,workflow_id:Workflow) -> None:
+        pass
+    
+
+#######################################################################################    
+def proc_input_by_agent():
+    pass
+
+def show_help():
+    print("this is help")
+
+
+async def main():
+    print("aios shell prepareing...")
+    logging.basicConfig(level=logging.INFO,
+                    format='%(asctime)s %(name)s %(levelname)s %(message)s')
+    
+    shell = AIOS_Shell("user")
+    await shell.initial()
+    print(f"aios shell {shell.get_version()} ready.")
+
+
+    completer = WordCompleter(['list agent', 'list workflow', 'exit', 'help'], ignore_case=True)
+    #history = FileHistory('history.txt')
+    while True:
+        user_input = await PromptSession().prompt_async('>>> ',completer=completer)
+        match user_input:
+            case "list agent":
+                print(AgentManager().list_agent())
+            case "list workflow":
+                print(WorkflowManager().list_workflow())
+            case "help":
+                show_help()
+            case "exit":
+                break
+        
+
+        if user_input.startswith("send"):
+            args = user_input.split(" ")
+            if len(args) < 3:
+                print("send msg failed, usage: send target_id msg_content")
+                continue
+            target_id = args[1]
+            msg_content = args[2]
+            resp = await shell.send_msg(msg_content,target_id,shell.username)
+            print(f"<<< {resp}")
+
+if __name__ == "__main__":
+    asyncio.run(main())
+