Skip to content

Commit 3505e85

Browse files
committed
Nicer AI streaming
1 parent deafc3f commit 3505e85

File tree

1 file changed

+16
-6
lines changed

1 file changed

+16
-6
lines changed

apps/webapp/app/components/code/AIQueryInput.tsx

Lines changed: 16 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -180,9 +180,19 @@ export function AIQueryInput({
180180
break;
181181
case "tool_call":
182182
if (event.tool === "setTimeFilter") {
183-
setThinking((prev) => prev + `\nSetting time filter...\n`);
183+
setThinking((prev) => {
184+
if (prev.trimEnd().endsWith("Setting time filter...")) {
185+
return prev;
186+
}
187+
return prev + `\nSetting time filter...\n`;
188+
});
184189
} else {
185-
setThinking((prev) => prev + `\nValidating query...\n`);
190+
setThinking((prev) => {
191+
if (prev.trimEnd().endsWith("Validating query...")) {
192+
return prev;
193+
}
194+
return prev + `\nValidating query...\n`;
195+
});
186196
}
187197
break;
188198
case "time_filter":
@@ -376,10 +386,10 @@ export function AIQueryInput({
376386
{isLoading
377387
? "AI is thinking..."
378388
: lastResult === "success"
379-
? "Query generated"
380-
: lastResult === "error"
381-
? "Generation failed"
382-
: "AI response"}
389+
? "Query generated"
390+
: lastResult === "error"
391+
? "Generation failed"
392+
: "AI response"}
383393
</span>
384394
</div>
385395
{isLoading ? (

0 commit comments

Comments
 (0)