Upload patch_chat_input.py

patch_chat_input.py

@@ -1,9 +1,10 @@
-"""Patch ChatInput.tsx for auto-continue buttons."""
+"""Patch ChatInput.tsx - Fixed version with all required props."""
 
 CHAT_INPUT = "/source/frontend/src/components/Chat/ChatInput.tsx"
 
-def patch_chat_input():
+def patch():
     import os
+    import re
     if not os.path.exists(CHAT_INPUT):
         print(f"SKIP: {CHAT_INPUT} not found")
         return
@@ -18,27 +19,40 @@ def patch_chat_input():
             "import { Box, IconButton, Stack, Tooltip, Button } from '@mui/material';"
         )
     
-    # Add props
-    props_code = '''
-interface ChatInputProps {
-  isProcessing?: boolean;
-  disabled?: boolean;
-  onUndoTurn?: () => void;
+    # Add import for DatasetUploadResponse if needed
+    if "DatasetUploadResponse" not in content:
+        content = content.replace(
+            "import { apiFetch } from '@/utils/api';",
+            "import { apiFetch } from '@/utils/api';\nimport type { DatasetUploadResponse } from '@/types/agent';"
+        )
+    
+    # Replace interface ChatInputProps with complete version including auto-continue props
+    new_props = '''interface ChatInputProps {
+  sessionId: string;
+  initialModelPath: string | null | undefined;
+  onSend: (text: string) => Promise<void>;
+  onStop: () => void;
+  onDatasetUploaded: () => Promise<boolean>;
+  isProcessing: boolean;
+  disabled: boolean;
+  placeholder?: string;
   showAutoContinue?: boolean;
   taskIncompleteInfo?: { incompletePlan: Array<{ id: string; content: string; status: string }> } | null;
   startAutoContinue?: () => void;
   cancelAutoContinue?: () => void;
-}
-'''
+}'''
+    
     if "showAutoContinue" not in content:
         # Find existing interface
-        match = re.search(r'interface ChatInputProps[^}]*[}][^}]*[}]', content, re.DOTALL)
+        match = re.search(r'interface ChatInputProps\s*\{[^}]*\}', content, re.DOTALL)
         if match:
-            content = content.replace(match.group(0), props_code)
+            content = content.replace(match.group(0), new_props)
+            print("OK: Replaced ChatInputProps interface")
+        else:
+            print("WARN: Could not find ChatInputProps interface")
     
-    # Add auto-continue buttons before the bottom bar
-    buttons_code = '''
-      {/* Auto-continue controls for free models */}
+    # Add auto-continue buttons before the flex:1 Box
+    buttons_code = '''      {/* Auto-continue controls for free models */}
       {showAutoContinue && taskIncompleteInfo && (
         <Stack direction="row" spacing={1} sx={{ mb: 1, px: 1 }}>
           <Button
@@ -60,21 +74,19 @@ interface ChatInputProps {
             Tạm dừng
           </Button>
         </Stack>
-      )}
-'''
+      )}'''
     
-    # Find a good insertion point - after the message area, before flex box
     if "Tự động tiếp tục" not in content:
         # Insert before <Box sx={{ flex: 1 }}
         content = content.replace(
-            "<Box sx={{ flex",
-            buttons_code + "\n        <Box sx={{"
+            "<Box sx={{ flex: 1 }} />",
+            buttons_code + "\n      <Box sx={{ flex: 1 }} />"
         )
+        print("OK: Added auto-continue buttons")
     
     with open(CHAT_INPUT, "w", encoding="utf-8") as f:
         f.write(content)
     print(f"OK: Patched {CHAT_INPUT}")
 
 if __name__ == "__main__":
-    import re
-    patch_chat_input()
+    patch()