feat(error-object): add `key`

This commit is contained in:
ItzCrazyKns 2024-05-05 16:28:46 +05:30
parent ba7b92ffde
commit 6e61c88c9e
No known key found for this signature in database
GPG Key ID: 8162927C7CCE3065
2 changed files with 28 additions and 5 deletions

View File

@ -70,7 +70,8 @@ export const handleConnection = async (
ws.send( ws.send(
JSON.stringify({ JSON.stringify({
type: 'error', type: 'error',
data: 'Invalid LLM or embeddings model selected', data: 'Invalid LLM or embeddings model selected, please refresh the page and try again.',
key: 'INVALID_MODEL_SELECTED',
}), }),
); );
ws.close(); ws.close();

View File

@ -57,7 +57,13 @@ const handleEmitterEvents = (
}); });
emitter.on('error', (data) => { emitter.on('error', (data) => {
const parsedData = JSON.parse(data); const parsedData = JSON.parse(data);
ws.send(JSON.stringify({ type: 'error', data: parsedData.data })); ws.send(
JSON.stringify({
type: 'error',
data: parsedData.data,
key: 'CHAIN_ERROR',
}),
);
}); });
}; };
@ -73,7 +79,11 @@ export const handleMessage = async (
if (!parsedMessage.content) if (!parsedMessage.content)
return ws.send( return ws.send(
JSON.stringify({ type: 'error', data: 'Invalid message format' }), JSON.stringify({
type: 'error',
data: 'Invalid message format',
key: 'INVALID_FORMAT',
}),
); );
const history: BaseMessage[] = parsedMessage.history.map((msg) => { const history: BaseMessage[] = parsedMessage.history.map((msg) => {
@ -99,11 +109,23 @@ export const handleMessage = async (
); );
handleEmitterEvents(emitter, ws, id); handleEmitterEvents(emitter, ws, id);
} else { } else {
ws.send(JSON.stringify({ type: 'error', data: 'Invalid focus mode' })); ws.send(
JSON.stringify({
type: 'error',
data: 'Invalid focus mode',
key: 'INVALID_FOCUS_MODE',
}),
);
} }
} }
} catch (err) { } catch (err) {
ws.send(JSON.stringify({ type: 'error', data: 'Invalid message format' })); ws.send(
JSON.stringify({
type: 'error',
data: 'Invalid message format',
key: 'INVALID_FORMAT',
}),
);
logger.error(`Failed to handle message: ${err}`); logger.error(`Failed to handle message: ${err}`);
} }
}; };