Package com.tnsai.server.rag.retrieval

Class KnowledgeBaseStream

java.lang.Object

com.tnsai.server.rag.retrieval.KnowledgeBaseStream

All Implemented Interfaces:

RetrievalStream

public final class KnowledgeBaseStream
extends Object
implements RetrievalStream

Wraps the existing KnowledgeBase TF-IDF search as a RetrievalStream.

Constructor Summary

Constructors

Constructor	Description
KnowledgeBaseStream(com.tnsai.knowledge.KnowledgeBase knowledgeBase)

Method Summary

Modifier and Type	Method	Description
boolean	isAvailable()
String	name()
List<com.tnsai.knowledge.SearchResult>	retrieve(String query, int topK)

Methods inherited from class java.lang.Object

equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Details

KnowledgeBaseStream

public KnowledgeBaseStream(com.tnsai.knowledge.KnowledgeBase knowledgeBase)

Method Details

name

public String name()

Specified by:

name in interface RetrievalStream

retrieve

public List<com.tnsai.knowledge.SearchResult> retrieve(String query,
 int topK)

Specified by:

retrieve in interface RetrievalStream

isAvailable

public boolean isAvailable()

Specified by:

isAvailable in interface RetrievalStream