Function taskWorker

src/modules/taskWorker.ts:334–446

(chatState: ChatStateType, db: TaskyonDatabase)

}

async function taskWorker(chatState: ChatStateType, db: TaskyonDatabase) {
  let cancelAllTasks = false;
  let cancelImmediately = false;
  let cancelCurrenTask = false;

  const cancelCurrentTaskListener = () => {
    cancelCurrenTask = true;
    console.log('Received signal to cancel current task');
  };

  const cancelTasksListener = () => {
    cancelAllTasks = true;
    console.log('Received signal to cancel all queued tasks');
  };

  const cancelImmediatelyListener = () => {
    cancelImmediately = true;
    console.log('Received signal for immediate cancellation');
    document.removeEventListener(
      RUNNING_TASK_WORKER_CANCELLATION_EVENT,
      cancelImmediatelyListener
    );
    throw new Error('Task Worker immediately cancelled');
  };

  document.addEventListener(
    CURRENT_TASK_CANCELLATION_EVENT,
    cancelCurrentTaskListener
  );
  document.addEventListener(TASK_CANCELLATION_EVENT, cancelTasksListener);
  document.addEventListener(
    RUNNING_TASK_WORKER_CANCELLATION_EVENT,
    cancelImmediatelyListener
  );

  window.addEventListener('beforeunload', () => {
    processTasksQueue.clear().forEach((t) => (t.state = 'Cancelled'));
  });

  console.log('entering task worker loop...');
  while (!cancelImmediately) {
    console.log('waiting for next task!');
    let task = await processTasksQueue.pop();
    if (!processTasksQueue.count()) {
      cancelAllTasks = false;
      cancelCurrenTask = false;
    } else if (cancelAllTasks) {
      task.state = 'Cancelled';
      continue;
    }
    task.state = 'In Progress';
    console.log('processing task:', task);
    try {
      if (task.role == 'user') {
        task = await processChatTask(task, chatState, db);
        task.state = 'Completed';
      } else if (task.role == 'function') {
        // in the case of 'FunctionCall' result, we run it twice: