UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-16 05:24:02 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #4431 from Nowheresly/fix4158_reconnect

Browse Source 
feat: #4158 allow reconnection when websocket is closed
This commit is contained in:
Timothy Jaeryang Baek 2024-08-07 17:13:57 +02:00 committed by GitHub
commit 70a8f6e707
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 20 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/lib/apis/index.ts
View File

@ -69,6 +69,7 @@ type ChatCompletedForm = {
model: string;
messages: string[];
chat_id: string;
session_id: string;
};
export const chatCompleted = async (token: string, body: ChatCompletedForm) => {

19
src/routes/+layout.svelte
View File

@ -111,6 +111,10 @@
if ($config) {
const _socket = io(`${WEBUI_BASE_URL}` || undefined, {
reconnection: true,
reconnectionDelay: 1000,
reconnectionDelayMax: 5000,
randomizationFactor: 0.5
path: '/ws/socket.io',
auth: { token: localStorage.token }
});
@ -119,6 +123,21 @@
console.log('connected');
});
_socket.on("reconnect_attempt", (attempt) => {
console.log('reconnect_attempt', attempt);
});
_socket.on("reconnect_failed", () => {
console.log('reconnect_failed');
});
_socket.on("disconnect", (reason, details) => {
console.log(`Socket ${socket.id} disconnected due to ${reason}`);
if (details) {
console.log('Additional details:', details);
}
});
await socket.set(_socket);
_socket.on('user-count', (data) => {