jan/openreplay
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

ui: swap socket path

Browse source
This commit is contained in:
nick-delirium 2025-05-13 13:53:39 +02:00
parent c4236cae6e
commit f742e6fff5
No known key found for this signature in database
GPG key ID: 93ABD695DF5FDBA0
1 changed files with 28 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

40
frontend/app/components/Kai/SocketManager.ts
View file

@ -4,13 +4,20 @@ export class ChatManager {
socket: ReturnType<typeof io>; socket: ReturnType<typeof io>;
threadId: string | null = null; threadId: string | null = null;
constructor({ projectId, threadId, token }: { projectId: string; threadId: string, token: string }) { constructor({
projectId,
threadId,
token,
}: {
projectId: string;
threadId: string;
token: string;
}) {
this.threadId = threadId; this.threadId = threadId;
const urlObject = new URL(window.env.API_EDP || window.location.origin); const urlObject = new URL(window.env.API_EDP || window.location.origin);
console.log(token, projectId, threadId) const socket = io(`${urlObject.origin}/kai/chat`, {
const socket = io(urlObject.origin, {
transports: ['websocket'], transports: ['websocket'],
path: '/kai/chat', path: '/kai/chat/socket.io',
autoConnect: true, autoConnect: true,
reconnection: true, reconnection: true,
reconnectionAttempts: 5, reconnectionAttempts: 5,
@ -23,8 +30,8 @@ export class ChatManager {
thread_id: threadId, thread_id: threadId,
}, },
auth: { auth: {
token, token: `Bearer ${token}`,
} },
}); });
socket.on('connect', () => { socket.on('connect', () => {
console.log('Connected to server'); console.log('Connected to server');
@ -34,7 +41,7 @@ export class ChatManager {
}); });
socket.on('error', (err) => { socket.on('error', (err) => {
console.error('Socket error:', err); console.error('Socket error:', err);
}) });
socket.onAny((e) => console.log('event', e)); socket.onAny((e) => console.log('event', e));
this.socket = socket; this.socket = socket;
@ -55,7 +62,9 @@ export class ChatManager {
msgCallback, msgCallback,
titleCallback, titleCallback,
}: { }: {
msgCallback: (msg: BotChunk | { state: string, type: 'state', start_time?: number }) => void; msgCallback: (
msg: BotChunk | { state: string; type: 'state'; start_time?: number },
) => void;
titleCallback: (title: string) => void; titleCallback: (title: string) => void;
}) => { }) => {
this.socket.on('chunk', (msg: BotChunk) => { this.socket.on('chunk', (msg: BotChunk) => {
@ -66,9 +75,16 @@ export class ChatManager {
console.log('Received title:', msg); console.log('Received title:', msg);
titleCallback(msg.content); titleCallback(msg.content);
}); });
this.socket.on('state', (state: { message: 'idle' | 'running', start_time: number }) => { this.socket.on(
msgCallback({ state: state.message, type: 'state', start_time: state.start_time }) 'state',
}) (state: { message: 'idle' | 'running'; start_time: number }) => {
msgCallback({
state: state.message,
type: 'state',
start_time: state.start_time,
});
},
);
}; };
disconnect = () => { disconnect = () => {
@ -90,5 +106,5 @@ export interface Message {
} }
export interface SentMessage extends Message { export interface SentMessage extends Message {
replace: boolean replace: boolean;
} }