com.github.hakenadu.javalangchains.chains.data.retrieval.RetrievalChain Maven / Gradle / Ivy

package com.github.hakenadu.javalangchains.chains.data.retrieval;

import java.util.Map;
import java.util.stream.Stream;

import com.github.hakenadu.javalangchains.chains.Chain;

/**
 * {@link Chain} which is utilized for retrieving documents in a QA context
 */
public abstract class RetrievalChain implements Chain>> {

	/**
	 * maximum count of retrieved documents
	 */
	private final int maxDocumentCount;

	/**
	 * @param maxDocumentCount {@link #maxDocumentCount}
	 */
	protected RetrievalChain(final int maxDocumentCount) {
		this.maxDocumentCount = maxDocumentCount;
	}

	/**
	 * @return {@link #maxDocumentCount}
	 */
	protected final int getMaxDocumentCount() {
		return this.maxDocumentCount;
	}
}