dave/storkit
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 7 Wiki Activity

story-kit: merge 168_bug_agent_message_queue_limited_to_one_line

Browse Source
This commit is contained in:
Dave
2026-02-24 19:17:33 +00:00
parent a80091c6f9
commit 74eeb308e1
2 changed files with 100 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

66
frontend/src/components/Chat.tsx
View File

@@ -91,9 +91,12 @@ export function Chat({ projectPath, onCloseProject }: ChatProps) {
const [agentConfigVersion, setAgentConfigVersion] = useState(0);
const [needsOnboarding, setNeedsOnboarding] = useState(false);
const onboardingTriggeredRef = useRef(false);
const [queuedMessage, setQueuedMessage] = useState<string | null>(null);
// Ref so stale WebSocket callbacks can read the current queued message
const queuedMessageRef = useRef<string | null>(null);
const [queuedMessages, setQueuedMessages] = useState<
{ id: string; text: string }[]
>([]);
// Ref so stale WebSocket callbacks can read the current queued messages
const queuedMessagesRef = useRef<{ id: string; text: string }[]>([]);
const queueIdCounterRef = useRef(0);
// Trigger state: set to a message string to fire auto-send after loading ends
const [pendingAutoSend, setPendingAutoSend] = useState<string | null>(null);
@@ -210,11 +213,10 @@ export function Chat({ projectPath, onCloseProject }: ChatProps) {
if (last?.role === "assistant" && !last.tool_calls) {
setLoading(false);
setActivityStatus(null);
if (queuedMessageRef.current) {
const msg = queuedMessageRef.current;
queuedMessageRef.current = null;
setQueuedMessage(null);
setPendingAutoSend(msg);
const nextQueued = queuedMessagesRef.current.shift();
if (nextQueued !== undefined) {
setQueuedMessages([...queuedMessagesRef.current]);
setPendingAutoSend(nextQueued.text);
}
}
},
@@ -225,11 +227,10 @@ export function Chat({ projectPath, onCloseProject }: ChatProps) {
console.error("WebSocket error:", message);
setLoading(false);
setActivityStatus(null);
if (queuedMessageRef.current) {
const msg = queuedMessageRef.current;
queuedMessageRef.current = null;
setQueuedMessage(null);
setPendingAutoSend(msg);
const nextQueued = queuedMessagesRef.current.shift();
if (nextQueued !== undefined) {
setQueuedMessages([...queuedMessagesRef.current]);
setPendingAutoSend(nextQueued.text);
}
},
onPipelineState: (state) => {
@@ -330,9 +331,9 @@ export function Chat({ projectPath, onCloseProject }: ChatProps) {
}, []);
const cancelGeneration = async () => {
// Discard any queued message — do not auto-send after cancel
queuedMessageRef.current = null;
setQueuedMessage(null);
// Discard any queued messages — do not auto-send after cancel
queuedMessagesRef.current = [];
setQueuedMessages([]);
try {
wsRef.current?.cancel();
await api.cancelChat();
@@ -358,8 +359,12 @@ export function Chat({ projectPath, onCloseProject }: ChatProps) {
// Agent is busy — queue the message instead of dropping it
if (loading) {
queuedMessageRef.current = messageToSend;
setQueuedMessage(messageToSend);
const newItem = {
id: String(queueIdCounterRef.current++),
text: messageToSend,
};
queuedMessagesRef.current = [...queuedMessagesRef.current, newItem];
setQueuedMessages([...queuedMessagesRef.current]);
if (!messageOverride || messageOverride === input) {
setInput("");
}
@@ -890,9 +895,10 @@ export function Chat({ projectPath, onCloseProject }: ChatProps) {
gap: "8px",
}}
>
{/* Queued message indicator */}
{queuedMessage && (
{/* Queued message indicators */}
{queuedMessages.map(({ id, text }) => (
<div
key={id}
data-testid="queued-message-indicator"
style={{
display: "flex",
@@ -926,15 +932,18 @@ export function Chat({ projectPath, onCloseProject }: ChatProps) {
whiteSpace: "nowrap",
}}
>
{queuedMessage}
{text}
</span>
<button
type="button"
title="Edit queued message"
onClick={() => {
setInput(queuedMessage);
queuedMessageRef.current = null;
setQueuedMessage(null);
setInput(text);
queuedMessagesRef.current =
queuedMessagesRef.current.filter(
(item) => item.id !== id,
);
setQueuedMessages([...queuedMessagesRef.current]);
inputRef.current?.focus();
}}
style={{
@@ -954,8 +963,11 @@ export function Chat({ projectPath, onCloseProject }: ChatProps) {
type="button"
title="Cancel queued message"
onClick={() => {
queuedMessageRef.current = null;
setQueuedMessage(null);
queuedMessagesRef.current =
queuedMessagesRef.current.filter(
(item) => item.id !== id,
);
setQueuedMessages([...queuedMessagesRef.current]);
}}
style={{
background: "none",
@@ -971,7 +983,7 @@ export function Chat({ projectPath, onCloseProject }: ChatProps) {
✕
</button>
</div>
)}
))}
{/* Input row */}
<div
style={{