handleAssistantResponse method

List<LlmChatMessage> handleAssistantResponse({
  1. required ChatStreamResponse response,
  2. required List<LlmChatMessage> messages,
})

Implementation

List<LlmChatMessage> handleAssistantResponse({
  required ChatStreamResponse response,
  required List<LlmChatMessage> messages,
}) {
  var lastAssistantMessageIndex = messages.lastIndexWhere(
    (message) => message.type == 'assistant',
  );

  if (lastAssistantMessageIndex != -1) {
    messages[lastAssistantMessageIndex] = LlmChatMessage.assistant(
      message: (messages[lastAssistantMessageIndex].message ?? '') +
          response.response,
    );
  }
  return messages;
}