feat: sequential chat

Author: hughlv

api/agentok_api/templates/start_chat.j2

@@ -2,25 +2,33 @@
 
 {% if initial_chat_targets | length > 1 %}
 # Sequential Chats
-# TODO: The input messsage is ignored, I guess it should be in the first element?
-chat_result = node_{{ first_converser['id'] }}.initiate_chats(
+chat_results = node_{{ first_converser['id'] }}.initiate_chats(
     [
-      {%- for target in initial_chat_targets -%}
+      {%- for target in initial_chat_targets %}
       {
         "recipient": node_{{ target['node']['id'] }},
-        {%- if target['chat_options'].message %}
-        "message": "{{ target.chat_options.message }}",
+        {%- if loop.first %}
+        "message": args.message,
+        {%- elif target['chat_options'].instructions %}
+        "message": "{{ target.chat_options.instructions }}",
         {%- endif %}
         {%- if target['chat_options'].max_turns %}
         "max_turns": {{ target.chat_options.max_turns }},
         {%- endif %}
         {%- if target['chat_options'].summary_method %}
         "summary_method": "{{ target.chat_options.summary_method }}",
         {%- endif %}
+        {%- if target['chat_options'].summary_prompt %}
+        "summary_prompt": "{{ target.chat_options.summary_prompt }}",
+        {%- endif %}
       }{% if not loop.last %},{% endif %}
-      {%- endfor -%}
+      {%- endfor %}
     ]
 )
+
+# Sequential Chat Results
+
+
 {%- elif initial_chat_targets | length == 1 %}
 {%- if initial_chat_targets[0].node['type'] == 'nestedchat' %}
 # Nested Chat
@@ -31,7 +39,7 @@ reply =  node_{{ first_converser['id'] }}.generate_reply(
 # Talk to one single agent
 chat_result = node_{{ first_converser['id'] }}.initiate_chat(
     node_{{ initial_chat_targets[0].node['id'] }},
-    {%- if initial_chat_targets and initial_chat_targets[0] and 'data' in initial_chat_targets[0] and initial_chat_targets[0]['data'].class in ['RetrieveUserProxyAgent', 'MathUserProxyAgent']  %}
+    {%- if initial_chat_targets and initial_chat_targets[0] and 'data' in initial_chat_targets[0] and initial_chat_targets[0]['data'].class_type in ['RetrieveUserProxyAgent', 'MathUserProxyAgent']  %}
     problem=args.message,
     {%- else %}
     message=args.message,

frontend/src/components/chat/chat-pane.tsx

@@ -13,7 +13,6 @@ import { Loading } from '@/components/loader';
 import { ScrollArea } from '@/components/ui/scroll-area';
 import { Button } from '../ui/button';
 import { Badge } from '../ui/badge';
-import { nanoid } from 'nanoid';
 
 interface ChatPaneProps {
   projectId: number;
@@ -64,7 +63,7 @@ export const ChatPane = ({ projectId, chatId }: ChatPaneProps) => {
 
     // Subscribe to chat_messages
     const messagesChannel: RealtimeChannel = supabase
-      .channel(`chat_messasges_${nanoid()}`)
+      .channel(`chat_messasges_${genId()}`)
       .on(
         'postgres_changes',
         {

frontend/src/components/flow/config/assistant.tsx

@@ -7,7 +7,6 @@ export const AssistantConfig = ({ nodeId, data, className, ...props }: any) => {
       data={data}
       className={className}
       toolScene={'llm'}
-      optionsDisabled={['human_input_mode']}
       {...props}
     />
   );

frontend/src/components/flow/config/converse-config.tsx

@@ -78,6 +78,17 @@ export const ConverseConfig = ({ edgeId, data }: any) => {
     <ScrollArea>
       <div className="flex flex-col gap-4 p-2">
         <GenericOption
+          type="text"
+          key={`${edgeId}-instructions`}
+          nodeId={edgeId}
+          data={data}
+          rows={2}
+          name="instructions"
+          label="Instructions"
+          placeholder="Enter a prompt for the instructions. This can be overridden with command arguments."
+        />
+        <GenericOption
+          key={`${edgeId}-mode`}
           type="select"
           nodeId={edgeId}
           data={data}
@@ -89,12 +100,46 @@ export const ConverseConfig = ({ edgeId, data }: any) => {
           ]}
         />
         <GenericOption
+          key={`${edgeId}-allow-repeat`}
           type="check"
           nodeId={edgeId}
           data={data}
           name="allow_repeat"
           label="Allow Repeat"
         />
+        <GenericOption
+          key={`${edgeId}-summary-method`}
+          type="select"
+          nodeId={edgeId}
+          data={data}
+          name="summary_method"
+          label="Summary Method"
+          options={[
+            { value: 'last_msg', label: 'Last Message' },
+            { value: 'reflection_with_llm', label: 'Reflection with LLM' },
+          ]}
+        />
+        <GenericOption
+          key={`${edgeId}-summary-prompt`}
+          type="text"
+          nodeId={edgeId}
+          data={data}
+          rows={3}
+          name="summary_prompt"
+          label="Summary Prompt"
+          placeholder="Enter a prompt for the summary."
+        />
+        <GenericOption
+          key={`${edgeId}-max-turns`}
+          type="range"
+          nodeId={edgeId}
+          data={data}
+          name="max_turns"
+          label="Max Turns"
+          min={1}
+          max={50}
+          step={1}
+        />
         <div className="flex flex-col gap-2 w-full">
           <div className="flex items-center justify-between gap-2 w-full">
             <span className="text-sm font-medium">Tools</span>

frontend/src/components/flow/config/initializer.tsx

@@ -24,36 +24,6 @@ export const InitializerConfig = ({ nodeId, data }: any) => {
           label="Sample Message (One per line)"
           placeholder="Enter sample messages, one per line."
         />
-        <GenericOption
-          type="select"
-          nodeId={nodeId}
-          data={data}
-          name="summary_method"
-          label="Summary Method"
-          options={[
-            { value: 'last_msg', label: 'Last Message' },
-            { value: 'reflection_with_llm', label: 'Reflection with LLM' },
-          ]}
-        />
-        <GenericOption
-          type="text"
-          nodeId={nodeId}
-          data={data}
-          rows={3}
-          name="summary_prompt"
-          label="Summary Prompt"
-          placeholder="Enter a prompt for the summary."
-        />
-        <GenericOption
-          type="range"
-          nodeId={nodeId}
-          data={data}
-          name="max_turns"
-          label="Max Turns"
-          min={1}
-          max={50}
-          step={1}
-        />
       </div>
     </ScrollArea>
   );

frontend/src/components/flow/flow-canvas.tsx

@@ -28,9 +28,9 @@ import { NodeButton } from './node-button';
 import { PythonViewer } from './python';
 import { useProject } from '@/hooks';
 import { debounce } from 'lodash-es';
-import { nanoid } from 'nanoid';
 import { Button } from '../ui/button';
 import { toast } from '@/hooks/use-toast';
+import { genId } from '@/lib/id';
 
 const DEBOUNCE_DELAY = 500; // Adjust this value as needed
 
@@ -329,7 +329,7 @@ export const FlowCanvas = ({
       });
 
       const { offsetX, offsetY, width, height, ...cleanedData } = data;
-      const newId = nanoid(8);
+      const newId = genId();
       const newNode = {
         id: `${data.id}_${newId}`,
         type: data.id,
@@ -434,7 +434,7 @@ export const FlowCanvas = ({
       height?: number
     ) => {
       setNodes((nds) => {
-        const newId = nanoid(8);
+        const newId = genId();
         const bounds = flowParent.current?.getBoundingClientRect();
         if (!bounds) return nds;
         const center = {
@@ -471,7 +471,7 @@ export const FlowCanvas = ({
 
       setIsDirty(true);
     },
-    [screenToFlowPosition, setNodes, nanoid]
+    [screenToFlowPosition, setNodes]
   );
 
   const onGeneratePython = async () => {

frontend/src/components/flow/node/assistant.tsx

@@ -19,10 +19,7 @@ export const AssistantNode: ComponentType<NodeProps> = ({
       data={data}
       selected={selected}
       type={type}
-      ports={[
-        { type: 'target', name: '' },
-        { type: 'source', name: '' },
-      ]}
+      ports={[{ type: 'target', name: '' }]}
     />
   );
 };

frontend/src/hooks/use-chats.ts

@@ -9,7 +9,7 @@ import { useProjects } from './use-projects';
 import { useTemplates } from './use-templates';
 import { createClient } from '@/lib/supabase/client';
 import { RealtimePostgresChangesPayload } from '@supabase/supabase-js';
-import { nanoid } from 'nanoid';
+import { genId } from '@/lib/id';
 
 export function useChats() {
   const { data, error, mutate } = useSWR('/api/chats', fetcher);
@@ -198,7 +198,7 @@ export function useChat(chatId: number) {
     // Subscribe to chat status updates
     const supabase = createClient();
     const channel = supabase
-      .channel(`chat_status_${nanoid(6)}`)
+      .channel(`chat_status_${genId()}`)
       .on(
         'postgres_changes',
         {

frontend/src/lib/id.ts

@@ -1,5 +1,7 @@
-export const genId = (): number => {
-  const timestamp = Math.floor(Date.now() / 1000); // Unix timestamp (seconds)
-  const random = Math.floor(Math.random() * 100000); // 5-digit random number
-  return timestamp * 100000 + random;
+import { customAlphabet } from "nanoid";
+
+export const genId = (): string => {
+  // Create custom nanoid without hyphens
+  const nanoid = customAlphabet('0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ', 8);
+  return nanoid(6);
 };

frontend/src/lib/supabase/client.ts

@@ -1,6 +1,6 @@
 import { createBrowserClient } from '@supabase/ssr';
 import { SupabaseClient } from '@supabase/supabase-js';
-import { nanoid } from 'nanoid';
+import { genId } from '../id';
 
 export function createClient() {
   return createBrowserClient(
@@ -61,7 +61,7 @@ export async function uploadFile(
     );
   }
 
-  const path = `${nanoid()}.${ext}`;
+  const path = `${genId()}.${ext}`;
 
   try {
     const { data, error } = await supabase.storage