Function generateFollowUpTasksFromResult

src/modules/taskWorker.ts:253–314

(
  finishedTask: LLMTask,
  chatState: ChatStateType
)

}

async function generateFollowUpTasksFromResult(
  finishedTask: LLMTask,
  chatState: ChatStateType
) {
  console.log('generate follow up task');
  if (finishedTask.result) {
    const taskDraftList: partialTaskDraft[] = [];
    let execute = false;
    if (finishedTask.result.type === 'ChatAnswer') {
      const choice = finishedTask.result.chatResponse?.choices[0];
      if (choice) {
        taskDraftList.push({
          state: 'Completed',
          role: choice.message.role,
          content: choice.message.content,
        });
      }
    } else if (finishedTask.result.type === 'AssistantAnswer') {
      const newTaskDraftList = createNewAssistantResponseTask(finishedTask);
      for (const td of newTaskDraftList) {
        taskDraftList.push({
          state: 'Completed',
          ...td,
        });
      }
    } else if (finishedTask.result.type === 'ToolChatResult') {
      const choice = finishedTask.result.chatResponse?.choices[0];
      let taskDraft = undefined;
      ({ taskDraft, execute } = await parseChatResponse(
        choice?.message.content || '',
        choice?.message.role
      ));
      if (taskDraft) {
        taskDraftList.push(taskDraft);
      }
    } else if (finishedTask.result.type === 'ToolCall') {
      const choice = finishedTask.result.chatResponse?.choices[0];
      if (choice) {
        const functionCall = extractOpenAIFunctions(choice);
        if (functionCall) {
          taskDraftList.push({
            role: 'function',
            content: null,
            context: { function: functionCall[0] },
          });
          execute = true;
        }
      }
    }
    let parentTask = finishedTask;
    const childCosts = {
      promptTokens: finishedTask.debugging.resultTokens,
      taskTokens: finishedTask.debugging.taskTokens,
      taskCosts: finishedTask.debugging.taskCosts,
    };
    for (const taskDraft of taskDraftList) {
      taskDraft.debugging = { ...taskDraft.debugging, ...childCosts };
      const newId = addTask2Tree(taskDraft, parentTask, chatState, execute);