diff --git a/client/main.tsx b/client/main.tsx index 5639b3c..08aa7fb 100644 --- a/client/main.tsx +++ b/client/main.tsx @@ -35,9 +35,12 @@ function conversationUI(id: string) { let socket: WebSocket; let connected = false; const connect = () => { - socket = new WebSocket( - `/api/conversation/${id}/connect` + (connected ? "?continue=1" : ""), - ); + const u = new URL(`/api/conversation/${id}/connect`, window.location.href); + if (connected) u.searchParams.set("continue", "1"); + if ("llm_model" in globalThis && Reflect.get(globalThis, "llm_model")) + u.searchParams.set("model", Reflect.get(globalThis, "llm_model")); + + socket = new WebSocket(u); socket.addEventListener("open", () => (connected = true)); socket.addEventListener("close", () => (socket = connect())); socket.addEventListener("error", ev => {