transformToChatPrompt method

String transformToChatPrompt({
  1. ModelType type = ModelType.general,
})

Implementation

String transformToChatPrompt({ModelType type = ModelType.general}) {
  switch (type) {
    case ModelType.general:
      return text;

    case ModelType.gemmaIt:
      return isUser
          ? '$startTurn$userPrefix\n$text$endTurn\n$startTurn$modelPrefix\n'
          : '$text$endTurn\n';

    case ModelType.deepSeek:
      if (isUser) {
        return '$deepseekStart$deepseekUser$text$deepseekAssistant';
      } else {
        return text;
      }
  }
}