OllamaChatMessage.java

Go to the documentation of this file.

package io.github.ollama4j.models.chat;
 
import com.fasterxml.jackson.annotation.JsonIgnoreProperties;
import com.fasterxml.jackson.annotation.JsonProperty;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.annotation.JsonSerialize;
import io.github.ollama4j.utils.FileToBase64Serializer;
import lombok.*;
 
import java.util.List;
 
import static io.github.ollama4j.utils.Utils.getObjectMapper;

@Data
@AllArgsConstructor
@RequiredArgsConstructor
@NoArgsConstructor
@JsonIgnoreProperties(ignoreUnknown = true)
public class OllamaChatMessage {
 
    @NonNull
    private OllamaChatMessageRole role;
 
    @NonNull
    private String content;
 
    private String thinking;
 
    private @JsonProperty("tool_calls") List<OllamaChatToolCalls> toolCalls;
 
    @JsonSerialize(using = FileToBase64Serializer.class)
    private List<byte[]> images;
 
    @Override
    public String toString() {
        try {
            return getObjectMapper().writerWithDefaultPrettyPrinter().writeValueAsString(this);
        } catch (JsonProcessingException e) {
            throw new RuntimeException(e);
        }
    }
}