OllamaResultStream.java

Go to the documentation of this file.

/*
 * Ollama4j - Java library for interacting with Ollama server.
 * Copyright (c) 2025 Amith Koujalgi and contributors.
 *
 * Licensed under the MIT License (the "License");
 * you may not use this file except in compliance with the License.
 *
*/
package io.github.ollama4j.models.response;
 
import java.util.Iterator;
import java.util.LinkedList;
import java.util.Queue;
 
public class OllamaResultStream extends LinkedList<String> implements Queue<String> {
    @Override
    public String poll() {
        StringBuilder tokens = new StringBuilder();
        Iterator<String> iterator = this.listIterator();
        while (iterator.hasNext()) {
            tokens.append(iterator.next());
            iterator.remove();
        }
        return tokens.toString();
    }
}