Verified Owner
Top Fields Constructors Methods
org.jabref.logic.ai.rag.logic

public Class EmbeddingsSearchResponseEngine

extends Object
implements ResponseEngine
Class Inheritance
All Implemented Interfaces
org.jabref.logic.ai.rag.logic.ResponseEngine
Imports
java.nio.file.Path, java.util.List, .Optional, org.jabref.logic.FilePreferences, org.jabref.logic.ai.embedding.EmbeddingInputPrefixes, org.jabref.logic.ai.ingestion.logic.EmbeddingsCleaner, org.jabref.logic.ai.ingestion.logic.ingestion.FileIngestor, org.jabref.logic.ai.ingestion.util.FileHasher, org.jabref.model.ai.identifiers.FullBibEntry, org.jabref.model.ai.pipeline.RelevantInformation, .ResponseEngineKind, org.jabref.model.entry.BibEntry, .LinkedFile, org.jabref.model.util.ListUtil, dev.langchain4j.data.segment.TextSegment, dev.langchain4j.model.embedding.EmbeddingModel, dev.langchain4j.store.embedding.EmbeddingMatch, dev.langchain4j.store.embedding.EmbeddingSearchRequest, dev.langchain4j.store.embedding.EmbeddingSearchResult, dev.langchain4j.store.embedding.EmbeddingStore, dev.langchain4j.store.embedding.filter.Filter, dev.langchain4j.store.embedding.filter.MetadataFilterBuilder, org.slf4j.Logger, .LoggerFactory

Field Summary

Modifier and TypeField and Description
private final dev.langchain4j.model.embedding.EmbeddingModel
embeddingModel

private final dev.langchain4j.store.embedding.EmbeddingStore<dev.langchain4j.data.segment.TextSegment>
embeddingStore

private final FilePreferences
filePreferences

private static final Logger
LOGGER

private final int
maximumResultsCount

private final double
minimumScore

Constructor Summary

AccessConstructor and Description
public
EmbeddingsSearchResponseEngine(FilePreferences filePreferences, dev.langchain4j.model.embedding.EmbeddingModel embeddingModel, dev.langchain4j.store.embedding.EmbeddingStore<dev.langchain4j.data.segment.TextSegment> embeddingStore, double minimumScore, int maximumResultsCount)

Method Summary

Modifier and TypeMethod and Description
private Optional<BibEntry>

Returns:

the entry if found
findEntryByFileHash(List<FullBibEntry>
the entries to search
entries, String
the SHA-256 hash of the file
fileHash)

Finds a BibEntry that has a LinkedFile with the given file hash.

public ResponseEngineKind
getKind()

Implements org.jabref.logic.ai.rag.logic.ResponseEngine.getKind.

public List<RelevantInformation>
process(String query, List<FullBibEntry> entriesFilter)

Implements org.jabref.logic.ai.rag.logic.ResponseEngine.process.

Inherited from java.lang.Object:
cloneequalsfinalizegetClasshashCodenotifynotifyAlltoStringwaitwaitwait

Field Detail

embeddingModelback to summary
private final dev.langchain4j.model.embedding.EmbeddingModel embeddingModel
embeddingStoreback to summary
private final dev.langchain4j.store.embedding.EmbeddingStore<dev.langchain4j.data.segment.TextSegment> embeddingStore
filePreferencesback to summary
private final FilePreferences filePreferences
LOGGERback to summary
private static final Logger LOGGER
maximumResultsCountback to summary
private final int maximumResultsCount
minimumScoreback to summary
private final double minimumScore

Constructor Detail

EmbeddingsSearchResponseEngineback to summary
public EmbeddingsSearchResponseEngine(FilePreferences filePreferences, dev.langchain4j.model.embedding.EmbeddingModel embeddingModel, dev.langchain4j.store.embedding.EmbeddingStore<dev.langchain4j.data.segment.TextSegment> embeddingStore, double minimumScore, int maximumResultsCount)

Method Detail

findEntryByFileHashback to summary
private Optional<BibEntry> findEntryByFileHash(List<FullBibEntry> entries, String fileHash)

Finds a BibEntry that has a LinkedFile with the given file hash.

Parameters
entries:List<FullBibEntry>

the entries to search

fileHash:String

the SHA-256 hash of the file

Returns:Optional<BibEntry>

the entry if found

getKindback to summary
public ResponseEngineKind getKind()

Implements org.jabref.logic.ai.rag.logic.ResponseEngine.getKind.

Annotations
@Override
processback to summary
public List<RelevantInformation> process(String query, List<FullBibEntry> entriesFilter)

Implements org.jabref.logic.ai.rag.logic.ResponseEngine.process.

Annotations
@Override