56 lines
1.3 KiB
JavaScript
56 lines
1.3 KiB
JavaScript
import FormData from 'form-data';
|
|
import defineAction from '../../../../helpers/define-action.js';
|
|
|
|
export default defineAction({
|
|
name: 'New chat',
|
|
key: 'newChat',
|
|
description: 'Create a new chat session for Helix AI.',
|
|
arguments: [
|
|
{
|
|
label: 'Input',
|
|
key: 'input',
|
|
type: 'string',
|
|
required: true,
|
|
description: 'Prompt to start the chat with.',
|
|
variables: true,
|
|
},
|
|
],
|
|
|
|
async run($) {
|
|
const formData = new FormData();
|
|
formData.append('input', $.step.parameters.input);
|
|
formData.append('mode', 'inference');
|
|
formData.append('type', 'text');
|
|
|
|
const sessionResponse = await $.http.post('/api/v1/sessions', formData, {
|
|
headers: {
|
|
...formData.getHeaders(),
|
|
},
|
|
});
|
|
|
|
const sessionId = sessionResponse.data.id;
|
|
|
|
let attempts = 0;
|
|
|
|
while (attempts < 10) {
|
|
const response = await $.http.get(`/api/v1/sessions/${sessionId}`);
|
|
|
|
const message =
|
|
response.data.interactions[response.data.interactions.length - 1];
|
|
|
|
if (message.creator === 'system' && message.state === 'complete') {
|
|
$.setActionItem({
|
|
raw: message,
|
|
});
|
|
|
|
return;
|
|
}
|
|
|
|
await new Promise((resolve) => setTimeout(resolve, 1000));
|
|
attempts++;
|
|
}
|
|
|
|
throw new Error('Failed to start a new chat session for Helix API!');
|
|
},
|
|
});
|