Ling0925
/
suna
mirror of https://github.com/kortix-ai/suna.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

AI: How can we stream the edit_file tool when it generating like create_file ? Also the edit_file tool show this 

"""Invalid File Edit

Could not extract the file changes from the tool result."""

Check the state of code base and make to sure implement fully
This commit is contained in:
LE Quoc Dat 2025-07-28 21:52:15 +02:00
parent 66194d03d3
commit 4a19e1e521
2 changed files with 25 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
frontend/src/components/thread/content/ShowToolStream.tsx
View File

@ -5,10 +5,8 @@ import { getToolIcon, getUserFriendlyToolName, extractPrimaryParam } from '@/com
// Only show streaming for file operation tools
const FILE_OPERATION_TOOLS = new Set([
'Create File',
'Delete File',
'Full File Rewrite',
'Read File',
'Creating File',
'Rewriting File',
'AI File Edit',
]);
@ -41,26 +39,30 @@ export const ShowToolStream: React.FC<ShowToolStreamProps> = ({
const rawToolName = extractToolNameFromStream(content);
const toolName = getUserFriendlyToolName(rawToolName || '');
const isEditFile = toolName === 'AI File Edit';
const isCreateFile = toolName === 'Creating File';
const isFullFileRewrite = toolName === 'Rewriting File';
// Extract code_edit content for streaming
const codeEditContent = React.useMemo(() => {
if (!isEditFile || !content) return '';
// New regex for <parameter name="code_edit">
const newMatch = content.match(/<parameter\s+name=["']code_edit["']>([\s\S]*)/i);
if (newMatch && newMatch[1]) {
// Remove closing tags if present
return newMatch[1].replace(/<\/parameter>[\s\S]*$/, '');
}
const streamingFileContent = React.useMemo(() => {
if (!content) return '';
let paramName: string | null = null;
if (isEditFile) paramName = 'code_edit';
else if (isCreateFile || isFullFileRewrite) paramName = 'file_contents';
// Fallback for old format <code_edit>
const oldMatch = content.match(/<code_edit>([\s\S]*)/i);
if (oldMatch && oldMatch[1]) {
return oldMatch[1].replace(/<\/code_edit>[\s\S]*$/, '');
if (paramName) {
const newMatch = content.match(new RegExp(`<parameter\\s+name=["']${paramName}["']>([\\s\\S]*)`, 'i'));
if (newMatch && newMatch[1]) {
return newMatch[1].replace(/<\/parameter>[\s\S]*$/, '');
}
// Fallback for old formats
if (isEditFile) {
const oldMatch = content.match(/<code_edit>([\s\S]*)/i);
if (oldMatch && oldMatch[1]) {
return oldMatch[1].replace(/<\/code_edit>[\s\S]*$/, '');
}
}
}
return '';
}, [content, isEditFile]);
return content; // fallback to full content
}, [content, isEditFile, isCreateFile, isFullFileRewrite]);
// Time-based logic - show streaming content after 1500ms
useEffect(() => {
@ -149,7 +151,7 @@ export const ShowToolStream: React.FC<ShowToolStreamProps> = ({
WebkitMaskImage: 'linear-gradient(to bottom, transparent 0%, black 8%, black 92%, transparent 100%)'
}}
>
{isEditFile ? codeEditContent : content}
{isEditFile || isCreateFile || isFullFileRewrite ? streamingFileContent : content}
</div>
{/* Top gradient */}
<div className={`absolute top-0 left-0 right-0 h-8 pointer-events-none transition-all duration-500 ease-in-out ${shouldShowContent

2
frontend/src/components/thread/tool-views/file-operation/FileEditToolView.tsx
View File

@ -142,7 +142,7 @@ export function FileEditToolView({
const lineDiff = originalContent && updatedContent ? generateLineDiff(originalContent, updatedContent) : [];
const stats: DiffStats = calculateDiffStats(lineDiff);
const shouldShowError = !isStreaming && !actualIsSuccess;
const shouldShowError = !isStreaming && (!actualIsSuccess || (actualIsSuccess && (!originalContent || !updatedContent)));
return (
<Card className="gap-0 flex border shadow-none border-t border-b-0 border-x-0 p-0 rounded-none flex-col h-full overflow-hidden bg-card">