OllamaChatStreamObserver.java

Go to the documentation of this file.

package io.github.ollama4j.models.chat;
 
import io.github.ollama4j.models.generate.OllamaStreamHandler;
 
import java.util.ArrayList;
import java.util.List;
 
public class OllamaChatStreamObserver {
 
    private OllamaStreamHandler streamHandler;
 
    private List<OllamaChatResponseModel> responseParts = new ArrayList<>();
 
    private String message = "";
 
    public OllamaChatStreamObserver(OllamaStreamHandler streamHandler) {
        this.streamHandler = streamHandler;
    }
 
    public void notify(OllamaChatResponseModel currentResponsePart) {
        responseParts.add(currentResponsePart);
        handleCurrentResponsePart(currentResponsePart);
    }
 
    protected void handleCurrentResponsePart(OllamaChatResponseModel currentResponsePart) {
        message = message + currentResponsePart.getMessage().getContent();
        streamHandler.accept(message);
    }
 
 
}