LibreChat/src/utils/handleSubmit.js

61 lines
1.4 KiB
JavaScript
Raw Normal View History

2023-02-06 21:17:46 -05:00
import { SSE } from '../../app/sse';
const endpoint = 'http://localhost:3050/ask';
2023-02-06 21:17:46 -05:00
export default function handleSubmit({
model,
text,
convo,
messageHandler,
convoHandler,
errorHandler
}) {
let payload = { model, text };
2023-02-06 21:17:46 -05:00
if (convo.conversationId && convo.parentMessageId) {
payload = {
...payload,
conversationId: convo.conversationId,
parentMessageId: convo.parentMessageId
};
}
2023-02-20 21:52:08 -05:00
if (model === 'bingai' && convo.conversationId) {
payload = {
...payload,
conversationId: convo.conversationId,
conversationSignature: convo.conversationSignature,
clientId: convo.clientId,
invocationId: convo.invocationId,
};
}
const server = model === 'bingai' ? endpoint + '/bing' : endpoint;
const events = new SSE(server, {
2023-02-06 21:17:46 -05:00
payload: JSON.stringify(payload),
headers: { 'Content-Type': 'application/json' }
});
events.onopen = function () {
console.log('connection is opened');
};
events.onmessage = function (e) {
const data = JSON.parse(e.data);
const text = data.text || data.response;
if (data.message) {
messageHandler(text);
} else if (data.final) {
2023-02-06 21:17:46 -05:00
console.log(data);
convoHandler(data);
} else {
console.log('initial', data);
}
};
events.onerror = function (e) {
console.log('error in opening conn.');
2023-02-06 21:17:46 -05:00
events.close();
errorHandler(e);
2023-02-06 21:17:46 -05:00
};
events.stream();
}