Verified Owner
Top Inners Fields Constructors Methods
org.jabref.gui.ai.chat

public Class AiChatStatusViewModel

extends AbstractViewModel
Class Inheritance
Imports
java.io.IOException, java.nio.file.Files, .StandardOpenOption, java.time.Instant, java.util.HashMap, .List, .Map, java.util.stream.Stream, javafx.beans.Observable, javafx.beans.property.ListProperty, .ObjectProperty, .ReadOnlyStringWrapper, .SimpleListProperty, .SimpleObjectProperty, javafx.beans.value.ChangeListener, javafx.collections.FXCollections, .ObservableList, org.jabref.gui.AbstractViewModel, .DialogService, org.jabref.gui.util.BindingsHelper, .FileDialogConfiguration, .UiTaskExecutor, org.jabref.logic.FilePreferences, org.jabref.logic.ai.chatting.ChatModel, org.jabref.logic.ai.chatting.exporters.AiChatJsonExporter, .AiChatMarkdownExporter, org.jabref.logic.ai.chatting.util.ChatModelFactory, org.jabref.logic.ai.embedding.AsyncEmbeddingModel, .EmbeddingModelCache, .EmbeddingModelFactory, org.jabref.logic.ai.ingestion.tasks.generateembeddings.GenerateEmbeddingsTask, org.jabref.logic.ai.preferences.AiPreferences, org.jabref.logic.ai.rag.logic.ResponseEngine, org.jabref.logic.ai.rag.util.ResponseEngineFactory, org.jabref.logic.ai.util.TrackedBackgroundTask, org.jabref.logic.bibtex.FieldPreferences, org.jabref.logic.l10n.Localization, org.jabref.logic.util.Directories, .ObservablesHelper, .StandardFileType, org.jabref.model.ai.AiMetadata, org.jabref.model.ai.chatting.ChatMessage, org.jabref.model.ai.identifiers.FullBibEntry, org.jabref.model.ai.pipeline.ResponseEngineKind, org.jabref.model.database.BibDatabaseMode, org.jabref.model.entry.BibEntry, .BibEntryTypesManager, .LinkedFile, dev.langchain4j.data.segment.TextSegment, dev.langchain4j.store.embedding.EmbeddingStore, org.slf4j.Logger, .LoggerFactory

Nested and Inner Type Summary

Modifier and TypeClass and Description
public static enum
AiChatStatusViewModel.FileStatus

public static class
AiChatStatusViewModel.IngestionStatusRow

Field Summary

Modifier and TypeField and Description
private final AiPreferences
aiPreferences

private final ListProperty<ChatMessage>
chatHistory

private final ObjectProperty<ChatModel>
chatModel

private final DialogService
dialogService

private final ObjectProperty<AsyncEmbeddingModel>
embeddingModel

private final EmbeddingModelCache
embeddingModelCache

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

private final ListProperty<FullBibEntry>
entries

private final BibEntryTypesManager
entryTypesManager

private final FieldPreferences
fieldPreferences

private final FilePreferences
filePreferences

private final ListProperty<GenerateEmbeddingsTask>
generateEmbeddingsTasks

private final ObservableList<AiChatStatusViewModel.IngestionStatusRow>
ingestionStatuses

private static final Logger
LOGGER

private final ObjectProperty<ResponseEngine>
responseEngine

private final ListProperty<ResponseEngineKind>
responseEngineKinds

private final ObjectProperty<ResponseEngineKind>
selectedResponseEngineKind

private final Map<GenerateEmbeddingsTask, ChangeListener<? super Status>>
taskListeners

Constructor Summary

AccessConstructor and Description
public
AiChatStatusViewModel(AiPreferences aiPreferences, FilePreferences filePreferences, FieldPreferences fieldPreferences, BibEntryTypesManager entryTypesManager, DialogService dialogService, EmbeddingModelCache embeddingModelCache, dev.langchain4j.store.embedding.EmbeddingStore<dev.langchain4j.data.segment.TextSegment> embeddingStore)

Method Summary

Modifier and TypeMethod and Description
private AiMetadata
buildMetadata()

public ListProperty<ChatMessage>
chatHistoryProperty()

public ObjectProperty<ChatModel>
chatModelProperty()

public void
clearChatHistory()

public ListProperty<FullBibEntry>
entriesProperty()

public void
exportJson()

public void
exportMarkdown()

public ListProperty<GenerateEmbeddingsTask>
generateEmbeddingsTasksProperty()

private List<BibEntry>
getBibEntriesFromFullEntries()

private BibDatabaseMode
getDatabaseModeOrDefault()

public ObservableList<AiChatStatusViewModel.IngestionStatusRow>
getIngestionStatuses()

private AiChatStatusViewModel.IngestionStatusRow
getOrCreateRow(LinkedFile file)

private void
processTask(GenerateEmbeddingsTask task)

public ListProperty<ResponseEngineKind>
responseEngineKindsProperty()

public ObjectProperty<ResponseEngine>
responseEngineProperty()

public ObjectProperty<ResponseEngineKind>
selectedResponseEngineKindProperty()

public void
setResponseEngine(ResponseEngine responseEngine)

private void
setupBindings()

private void
setupListeners()

private void
setupValues()

private void
unwireTask(GenerateEmbeddingsTask task)

private void
wireTask(GenerateEmbeddingsTask task)

Field Detail

aiPreferencesback to summary
private final AiPreferences aiPreferences
chatHistoryback to summary
private final ListProperty<ChatMessage> chatHistory
chatModelback to summary
private final ObjectProperty<ChatModel> chatModel
dialogServiceback to summary
private final DialogService dialogService
embeddingModelback to summary
private final ObjectProperty<AsyncEmbeddingModel> embeddingModel
embeddingModelCacheback to summary
private final EmbeddingModelCache embeddingModelCache
embeddingStoreback to summary
private final dev.langchain4j.store.embedding.EmbeddingStore<dev.langchain4j.data.segment.TextSegment> embeddingStore
entriesback to summary
private final ListProperty<FullBibEntry> entries
entryTypesManagerback to summary
private final BibEntryTypesManager entryTypesManager
fieldPreferencesback to summary
private final FieldPreferences fieldPreferences
filePreferencesback to summary
private final FilePreferences filePreferences
generateEmbeddingsTasksback to summary
private final ListProperty<GenerateEmbeddingsTask> generateEmbeddingsTasks
ingestionStatusesback to summary
private final ObservableList<AiChatStatusViewModel.IngestionStatusRow> ingestionStatuses
LOGGERback to summary
private static final Logger LOGGER
responseEngineback to summary
private final ObjectProperty<ResponseEngine> responseEngine
responseEngineKindsback to summary
private final ListProperty<ResponseEngineKind> responseEngineKinds
selectedResponseEngineKindback to summary
private final ObjectProperty<ResponseEngineKind> selectedResponseEngineKind
taskListenersback to summary
private final Map<GenerateEmbeddingsTask, ChangeListener<? super Status>> taskListeners

Constructor Detail

AiChatStatusViewModelback to summary
public AiChatStatusViewModel(AiPreferences aiPreferences, FilePreferences filePreferences, FieldPreferences fieldPreferences, BibEntryTypesManager entryTypesManager, DialogService dialogService, EmbeddingModelCache embeddingModelCache, dev.langchain4j.store.embedding.EmbeddingStore<dev.langchain4j.data.segment.TextSegment> embeddingStore)

Method Detail

buildMetadataback to summary
private AiMetadata buildMetadata()
chatHistoryPropertyback to summary
public ListProperty<ChatMessage> chatHistoryProperty()
chatModelPropertyback to summary
public ObjectProperty<ChatModel> chatModelProperty()
clearChatHistoryback to summary
public void clearChatHistory()
entriesPropertyback to summary
public ListProperty<FullBibEntry> entriesProperty()
exportJsonback to summary
public void exportJson()
exportMarkdownback to summary
public void exportMarkdown()
generateEmbeddingsTasksPropertyback to summary
public ListProperty<GenerateEmbeddingsTask> generateEmbeddingsTasksProperty()
getBibEntriesFromFullEntriesback to summary
private List<BibEntry> getBibEntriesFromFullEntries()
getDatabaseModeOrDefaultback to summary
private BibDatabaseMode getDatabaseModeOrDefault()
getIngestionStatusesback to summary
public ObservableList<AiChatStatusViewModel.IngestionStatusRow> getIngestionStatuses()
getOrCreateRowback to summary
private AiChatStatusViewModel.IngestionStatusRow getOrCreateRow(LinkedFile file)
processTaskback to summary
private void processTask(GenerateEmbeddingsTask task)
responseEngineKindsPropertyback to summary
public ListProperty<ResponseEngineKind> responseEngineKindsProperty()
responseEnginePropertyback to summary
public ObjectProperty<ResponseEngine> responseEngineProperty()
selectedResponseEngineKindPropertyback to summary
public ObjectProperty<ResponseEngineKind> selectedResponseEngineKindProperty()
setResponseEngineback to summary
public void setResponseEngine(ResponseEngine responseEngine)
setupBindingsback to summary
private void setupBindings()
setupListenersback to summary
private void setupListeners()
setupValuesback to summary
private void setupValues()
unwireTaskback to summary
private void unwireTask(GenerateEmbeddingsTask task)
wireTaskback to summary
private void wireTask(GenerateEmbeddingsTask task)
org.jabref.gui.ai.chat back to summary

public final Enum AiChatStatusViewModel.FileStatus

extends Enum<AiChatStatusViewModel.FileStatus>
Class Inheritance

Field Summary

Modifier and TypeField and Description
public static final AiChatStatusViewModel.FileStatus
CANCELLED

private final String
displayName

public static final AiChatStatusViewModel.FileStatus
ERROR_WHILE_PROCESSING

public static final AiChatStatusViewModel.FileStatus
INGESTED

public static final AiChatStatusViewModel.FileStatus
PENDING

public static final AiChatStatusViewModel.FileStatus
PROCESSING

Constructor Summary

AccessConstructor and Description
private
FileStatus(String displayName)

Method Summary

Modifier and TypeMethod and Description
public String
getDisplayName()

public static AiChatStatusViewModel.FileStatus
valueOf(String name)

public static AiChatStatusViewModel.FileStatus[]
values()

Inherited from java.lang.Enum:
clonecompareTodescribeConstableequalsfinalizegetDeclaringClasshashCodenameordinaltoStringvalueOf

Field Detail

CANCELLEDback to summary
public static final AiChatStatusViewModel.FileStatus CANCELLED
displayNameback to summary
private final String displayName
ERROR_WHILE_PROCESSINGback to summary
public static final AiChatStatusViewModel.FileStatus ERROR_WHILE_PROCESSING
INGESTEDback to summary
public static final AiChatStatusViewModel.FileStatus INGESTED
PENDINGback to summary
public static final AiChatStatusViewModel.FileStatus PENDING
PROCESSINGback to summary
public static final AiChatStatusViewModel.FileStatus PROCESSING

Constructor Detail

FileStatusback to summary
private FileStatus(String displayName)

Method Detail

getDisplayNameback to summary
public String getDisplayName()
valueOfback to summary
public static AiChatStatusViewModel.FileStatus valueOf(String name)
valuesback to summary
public static AiChatStatusViewModel.FileStatus[] values()
org.jabref.gui.ai.chat back to summary

public Class AiChatStatusViewModel.IngestionStatusRow

extends Object
Class Inheritance
  • java.lang.Object
  • org.jabref.gui.ai.chat.AiChatStatusViewModel.IngestionStatusRow

Field Summary

Modifier and TypeField and Description
private final ObjectProperty<Exception>
error

private final LinkedFile
linkedFile

private final ReadOnlyStringWrapper
name

private final ObjectProperty<AiChatStatusViewModel.FileStatus>
status

Constructor Summary

AccessConstructor and Description
public
IngestionStatusRow(LinkedFile linkedFile)

Method Summary

Modifier and TypeMethod and Description
public ObjectProperty<Exception>
errorProperty()

public Exception
getError()

public LinkedFile
getLinkedFile()

public AiChatStatusViewModel.FileStatus
getStatus()

public ReadOnlyStringWrapper
nameProperty()

public ObjectProperty<AiChatStatusViewModel.FileStatus>
statusProperty()

Inherited from java.lang.Object:
cloneequalsfinalizegetClasshashCodenotifynotifyAlltoStringwaitwaitwait

Field Detail

errorback to summary
private final ObjectProperty<Exception> error
linkedFileback to summary
private final LinkedFile linkedFile
nameback to summary
private final ReadOnlyStringWrapper name
statusback to summary
private final ObjectProperty<AiChatStatusViewModel.FileStatus> status

Constructor Detail

IngestionStatusRowback to summary
public IngestionStatusRow(LinkedFile linkedFile)

Method Detail

errorPropertyback to summary
public ObjectProperty<Exception> errorProperty()
getErrorback to summary
public Exception getError()
getLinkedFileback to summary
public LinkedFile getLinkedFile()
getStatusback to summary
public AiChatStatusViewModel.FileStatus getStatus()
namePropertyback to summary
public ReadOnlyStringWrapper nameProperty()
statusPropertyback to summary
public ObjectProperty<AiChatStatusViewModel.FileStatus> statusProperty()