Function parseChatResponse

src/modules/taskWorker.ts:197–251

(
  message: string,
  role: partialTaskDraft['role'] | undefined
)

}

async function parseChatResponse(
  message: string,
  role: partialTaskDraft['role'] | undefined
): Promise<{ taskDraft: partialTaskDraft; execute: boolean }> {
  // parse the response and create a new task filled with the correct parameters
  let yamlContent = message.trim();
  // Use exec() to find a match
  const yamlBlockRegex = /```(yaml|YAML)\n?([\s\S]*?)\n?```/;
  const yamlMatch = yamlBlockRegex.exec(yamlContent);
  if (yamlMatch) {
    yamlContent = yamlMatch[1]; // Use the captured group
  }
  // Parse the extracted or original YAML content
  const parsedYaml = load(yamlContent);
  const toolChatResult = await yamlToolChatType.safeParseAsync(parsedYaml);
  if (toolChatResult.success) {
    console.log(toolChatResult);
    if (toolChatResult.data.toolCommand) {
      return {
        taskDraft: {
          role: 'function',
          content: null,
          context: {
            function: {
              name: toolChatResult.data.toolCommand.name,
              arguments: toolChatResult.data.toolCommand.args,
            },
          },
        },
        execute: true,
      };
    } else {
      return {
        taskDraft: {
          state: 'Completed',
          role: role || 'assistant',
          content:
            toolChatResult.data.answer ||
            toolChatResult.data.thought ||
            message,
        },
        execute: false,
      };
    }
  } else {
    return {
      taskDraft: {
        state: 'Completed',
        role: role || 'system',
        content: toolChatResult.error.toString(),
      },
      execute: false,
    };
  }
}

async function generateFollowUpTasksFromResult(
  finishedTask: LLMTask,