Skip to content

Commit 4025aae

Browse files
author
vbarda
committed
add small sleep to to the stopStream implementation
1 parent bc9dad8 commit 4025aae

File tree

2 files changed

+8
-7
lines changed

2 files changed

+8
-7
lines changed

frontend/app/components/ChatWindow.tsx

Lines changed: 2 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -46,8 +46,6 @@ const MODEL_TYPES = [
4646
"cohere_command",
4747
];
4848

49-
const THREAD_NAME_PLACEHOLDER = "New chat"
50-
5149
const defaultLlmValue =
5250
MODEL_TYPES[Math.floor(Math.random() * MODEL_TYPES.length)];
5351

@@ -107,11 +105,8 @@ export function ChatWindow() {
107105
tags: ["model:" + llm],
108106
};
109107

110-
const getThreadName = (messageValue: string) => (
111-
messageValue.length > 20
112-
? messageValue.slice(0, 20) + "..."
113-
: messageValue
114-
)
108+
const getThreadName = (messageValue: string) =>
109+
messageValue.length > 20 ? messageValue.slice(0, 20) + "..." : messageValue;
115110

116111
const renameThread = async (messageValue: string) => {
117112
// NOTE: we're only setting this on the first message

frontend/app/hooks/useThreadMessages.ts

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,10 @@ function getMessagesWithSources(messages?: Message[], documents?: Document[]) {
2020
}));
2121
}
2222

23+
function sleep(ms: number) {
24+
return new Promise((resolve) => setTimeout(resolve, ms));
25+
}
26+
2327
export function getSources(documents: Document[]) {
2428
const sources = documents.map((doc) => ({
2529
url: doc.metadata.source,
@@ -63,6 +67,8 @@ export function useThreadMessages(
6367
useEffect(() => {
6468
async function fetchMessages() {
6569
if (threadId) {
70+
// sleep a tiny bit to make sure the stop
71+
await sleep(10);
6672
const { values, next } = await client.threads.getState<{
6773
messages: Message[];
6874
documents: Document[];

0 commit comments

Comments
 (0)