• Home
• io.quarkiverse.langchain4j
• quarkus-langchain4j-llama3-java
• 0.23.0.CR2
• source code
• MessageMapper.java

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

io.quarkiverse.langchain4j.llama3.MessageMapper Maven / Gradle / Ivy

package io.quarkiverse.langchain4j.llama3;

import dev.langchain4j.data.message.ChatMessage;
import dev.langchain4j.data.message.ChatMessageType;
import io.quarkiverse.langchain4j.llama3.copy.ChatFormat;

final class MessageMapper {

    static ChatFormat.Message toLlama3Message(ChatMessage langchainMessage) {
        ChatFormat.Role role = toJllamaRole(langchainMessage.type());
        return new ChatFormat.Message(role, langchainMessage.text());
    }

    private static ChatFormat.Role toJllamaRole(ChatMessageType chatMessageType) {
        return switch (chatMessageType) {
            case SYSTEM -> ChatFormat.Role.SYSTEM;
            case USER -> ChatFormat.Role.USER;
            case AI -> ChatFormat.Role.ASSISTANT;
            default -> throw new IllegalArgumentException("Unsupported chat message type: " + chatMessageType);
        };
    }
}