Part of an agent framework

agent_repl.py

@@ -1,20 +1,15 @@
-import argparse
+import warnings
 from typing import List, Tuple
 
-from fastapi import Depends
+from dotenv import load_dotenv
 
 from aios.hooks.modules.agent import useFactory
 from aios.hooks.modules.llm import useCore
 from aios.hooks.modules.scheduler import useFIFOScheduler
-from aios.utils.utils import delete_directories, humanify_agent, parse_global_args
 from aios.utils.state import useGlobalState
-
-from pyopenagi.agents.interact import Interactor
+from aios.utils.utils import delete_directories, parse_global_args
 from pyopenagi.manager.manager import AgentManager
 
-from dotenv import load_dotenv
-import warnings
-
 # load the args
 warnings.filterwarnings("ignore")
 parser = parse_global_args()

agenthub/app/agents/ContentLayout/DatasetsHeader.tsx

@@ -40,4 +40,4 @@ export function DatasetsHeader({ filteredCount = 0 }: DatasetsHeaderProps) {
       </div>
     </div>
   )
-}
+}

agenthub/app/agents/const.ts

@@ -12,7 +12,7 @@ export const DatasetList: DatasetItem[] = [...Array(30)].map(() => ({
 }))
 
 export const AgentList: AgentItem[] = [];
-// export const AgentList = 
+// export const AgentList =
 
 export const AgentListGenerator: () => Promise<AgentItem[]> = async () => {
     try {
@@ -395,4 +395,4 @@ export const DatasetLanguages = [
   'xx',
   'us',
   'ua',
-]
+]

agenthub/app/agents/page.tsx

@@ -8,4 +8,4 @@ export default function Datasets() {
       </div>
     </div>
   )
-}
+}

agenthub/app/agents/type.ts

@@ -22,4 +22,4 @@ export interface AgentItem {
     createdAt: string;
     numDownloads: number;
     numFavorites: number;
-}
+}

agenthub/app/chat/page.tsx

@@ -83,7 +83,7 @@ const ChatInterface: React.FC = () => {
         thinking: false
       };
 
-      setChats(prevChats => prevChats.map(chat => 
+      setChats(prevChats => prevChats.map(chat =>
         chat.id === activeChat
           ? { ...chat, messages: [...chat.messages, newMessage] }
           : chat
@@ -98,19 +98,19 @@ const ChatInterface: React.FC = () => {
         thinking: true
       };
 
-      setChats(prevChats => prevChats.map(chat => 
+      setChats(prevChats => prevChats.map(chat =>
         chat.id === activeChat
           ? { ...chat, messages: [...chat.messages, botMessage] }
           : chat
       ));
 
       const res = await processAgentCommand(parseNamedContent(parseText(content))[0] as AgentCommand);
 
-      setChats(prevChats => prevChats.map(chat => 
+      setChats(prevChats => prevChats.map(chat =>
         chat.id === activeChat
           ? {
               ...chat,
-              messages: chat.messages.map(message => 
+              messages: chat.messages.map(message =>
                 message.id === messageId
                   ? { ...message, thinking: false, text: res.content }
                   : message
@@ -186,8 +186,8 @@ const ChatInterface: React.FC = () => {
 
   // Add updateChatName function
   const updateChatName = (chatId: number, newName: string) => {
-    setChats(prevChats => 
-      prevChats.map(chat => 
+    setChats(prevChats =>
+      prevChats.map(chat =>
         chat.id === chatId ? { ...chat, name: newName } : chat
       )
     );
@@ -217,10 +217,10 @@ const ChatInterface: React.FC = () => {
         darkMode={darkMode}
       />
       <div className="flex flex-col flex-grow pb-4">
-        <Header 
-          darkMode={darkMode} 
+        <Header
+          darkMode={darkMode}
           setDarkMode={setDarkMode}
-          title={chats.find(chat => chat.id === activeChat)?.name || 'Chat'} 
+          title={chats.find(chat => chat.id === activeChat)?.name || 'Chat'}
         />
         <MessageList messages={activeMessages} darkMode={darkMode} />
         <div className='w-full flex h-fit justify-center'>

agenthub/components/agentchat/Header.tsx

@@ -29,4 +29,4 @@ export const Header: React.FC<HeaderProps> = ({ darkMode, setDarkMode, title })
         </div>
       </div>
     );
-  };
+  };

agenthub/components/agentchat/Sidebar.tsx

@@ -18,49 +18,49 @@ export interface SidebarProps {
 export const Sidebar: React.FC<SidebarProps> = ({ chats, activeChat, setActiveChat, addChat, updateChatName, deleteChat, darkMode }) => {
     const [editingId, setEditingId] = useState<number | null>(null);
     const [editingName, setEditingName] = useState('');
-  
+
     const categoryStyle = "text-xs font-semibold uppercase tracking-wide text-gray-500 mb-2 mt-4 px-2 flex justify-between items-center";
     const channelStyle = `flex items-center justify-between rounded px-2 py-1.5 text-sm font-medium transition-colors duration-200 ease-in-out cursor-pointer`;
     const activeChannelStyle = darkMode ? 'bg-gray-700 text-white' : 'bg-gray-300 text-gray-900';
     const inactiveChannelStyle = darkMode ? 'text-gray-400 hover:bg-gray-700 hover:text-gray-200' : 'text-gray-700 hover:bg-gray-200 hover:text-gray-900';
-  
+
     const startEditing = (chat: Chat) => {
       setEditingId(chat.id);
       setEditingName(chat.name);
     };
-  
+
     const cancelEditing = () => {
       setEditingId(null);
       setEditingName('');
     };
-  
+
     const saveEditing = () => {
       if (editingId !== null && editingName.trim() !== '') {
         updateChatName(editingId, editingName.trim());
         setEditingId(null);
       }
     };
-  
+
     const handleDelete = (chatId: number, e: React.MouseEvent) => {
       e.stopPropagation();
       if (window.confirm('Are you sure you want to delete this channel?')) {
         deleteChat(chatId);
       }
     };
-  
+
     return (
       <div className={`w-60 flex-shrink-0 ${darkMode ? 'bg-gray-800' : 'bg-gray-100'} p-3 flex flex-col`}>
         <div className={`p-4 ${darkMode ? 'bg-gray-700' : 'bg-gray-200'} rounded-lg mb-4`}>
           <h2 className={`font-bold ${darkMode ? 'text-white' : 'text-gray-800'}`}>Your AIOS Workspace</h2>
         </div>
-        
+
         <div className="flex-grow overflow-y-auto">
           <div className={categoryStyle}>
             Channels
             <Tooltip label="Add Channel" position="right">
-              <ActionIcon 
-                onClick={addChat} 
-                variant="subtle" 
+              <ActionIcon
+                onClick={addChat}
+                variant="subtle"
                 color={darkMode ? "gray" : "dark"}
                 className="hover:bg-gray-600 cursor-pointer"
               >
@@ -99,32 +99,32 @@ export const Sidebar: React.FC<SidebarProps> = ({ chats, activeChat, setActiveCh
                   </div>
                   <div className="flex items-center">
                     <Tooltip label="Rename Channel" position="right">
-                      <ActionIcon 
+                      <ActionIcon
                         onClick={(e) => {
                           e.stopPropagation();
                           startEditing(chat);
-                        }} 
-                        variant="subtle" 
+                        }}
+                        variant="subtle"
                         color={darkMode ? "gray" : "dark"}
                         size="sm"
                         className="ml-1"
                       >
-                        <Edit2 
-                          size={14} 
-                          className={darkMode ? "text-blue-400 hover:text-blue-300" : "text-blue-600 hover:text-blue-500"} 
+                        <Edit2
+                          size={14}
+                          className={darkMode ? "text-blue-400 hover:text-blue-300" : "text-blue-600 hover:text-blue-500"}
                         />
                       </ActionIcon>
                     </Tooltip>
                     <Tooltip label="Delete Channel" position="right">
-                      <ActionIcon 
-                        onClick={(e) => handleDelete(chat.id, e)} 
-                        variant="subtle" 
+                      <ActionIcon
+                        onClick={(e) => handleDelete(chat.id, e)}
+                        variant="subtle"
                         size="sm"
                         className="ml-1"
                       >
-                        <Trash2 
-                          size={14} 
-                          className="text-red-400 hover:text-red-300" 
+                        <Trash2
+                          size={14}
+                          className="text-red-400 hover:text-red-300"
                         />
                       </ActionIcon>
                     </Tooltip>

agenthub/components/homepage/Products.tsx

@@ -49,7 +49,7 @@ const BuiltWithSupabase = () => {
         <div className="text-2xl pt-6">
           Try out our many features
         </div>
-     
+
       </div>
       <div className="mt-16 grid grid-cols-12 gap-5">
         {Examples.slice(0, 2).map((example: any, i: number) => {

agenthub/interfaces/agentchat.ts

@@ -11,4 +11,4 @@ export interface Chat {
     id: number;
     name: string;
     messages: Message[];
-} 
+}

agenthub/lib/env.ts

@@ -1,6 +1,9 @@
 export const inDevEnvironment = !!process && process.env.NODE_ENV === 'development';
 // export const serverUrl = inDevEnvironment ? 'http://localhost:8000' : 'https://myapp-y5z35kuonq-uk.a.run.app'
-export const baseUrl = inDevEnvironment ? 'http://localhost:3000' : 'https://my.aios.foundation'
+export const baseUrl = process.env.NODE_ENV === 'development'
+  ? 'http://localhost:3000'
+  : 'https://my.aios.foundation';
 // export const serverUrl = inDevEnvironment ? 'http://localhost:8000' : 'http://35.232.56.61:8000'
-export const serverUrl = 'http://35.232.56.61:8000';
-
+export const serverUrl = process.env.NODE_ENV === 'development'
+  ? 'http://localhost:8000'
+  : 'https://api.aios.chat';

agenthub/public/MentionListV2Light.scss

@@ -10,15 +10,15 @@
     overflow: auto;
     padding: 0.4rem;
     position: relative;
-  
+
     button {
       align-items: center;
       background-color: transparent;
       display: flex;
       gap: 0.25rem;
       text-align: left;
       width: 100%;
-  
+
       .item-text {
         color: #000000 !important;
       }
@@ -27,7 +27,7 @@
       &:hover.is-selected {
         background-color: var(--gray-3);
       }
-  
+
       &.is-selected {
         background-color: var(--gray-2);
       }
@@ -38,7 +38,7 @@
     :first-child {
       margin-top: 0;
     }
-  
+
     .mention {
       background-color: var(--purple-light);
       border-radius: 0.4rem;
@@ -54,4 +54,4 @@
     box-decoration-break: clone;
     color: var(--purple);
     padding: 0.1rem 0.3rem;
-  }
+  }

aios/context/simple_context.py

@@ -3,12 +3,8 @@
 
 from aios.context.base import BaseContextManager
 
-import os
-
-import torch
 
 # import shutil
-from threading import Lock
 
 class SimpleContextManager(BaseContextManager):
     def __init__(self):
@@ -21,7 +17,7 @@ def start(self):
     def gen_snapshot(self, pid, context):
         # file_path = os.path.join(self.context_dir, f"process-{pid}.pt")
         # torch.save(context, file_path)
-        self.context_dict[str(pid)] = context
+        self.context_dic[str(pid)] = context
 
     def gen_recover(self, pid):
         # file_path = os.path.join(self.context_dir, f"process-{pid}.pt")

aios/hooks/llm.py

@@ -139,7 +139,9 @@ def aios_starter(
     max_gpu_memory,
     eval_device,
     max_new_tokens,
-    log_mode,
+    scheduler_log_mode,
+    agent_log_mode,
+    llm_kernel_log_mode,
     use_backend
 ):
     """
@@ -151,7 +153,9 @@ def aios_starter(
         max_gpu_memory (str): The maximum amount of GPU memory to use.
         eval_device (str): The device to evaluate the LLM on.
         max_new_tokens (int): The maximum number of new tokens to generate.
-        log_mode (str): The log mode.
+        agent_log_mode:
+        llm_kernel_log_mode:
+        scheduler_log_mode:
         use_backend (str): The backend to use for running the LLM kernel.
 
     Yields:
@@ -164,15 +168,15 @@ def aios_starter(
         max_gpu_memory=max_gpu_memory,
         eval_device=eval_device,
         max_new_tokens=max_new_tokens,
-        log_mode=log_mode,
+        log_mode=llm_kernel_log_mode,
         use_backend=use_backend
     )
 
     # run agents concurrently for maximum efficiency using a scheduler
     submit_agent, await_agent_execution = useFactory(
-        log_mode=log_mode,
+        log_mode=agent_log_mode,
         max_workers=64
     )
 
-    with fifo_scheduler(llm=llm, log_mode=log_mode, get_queue_message=None):
+    with fifo_scheduler(llm=llm, log_mode=scheduler_log_mode, get_queue_message=None):
         yield submit_agent, await_agent_execution