mirror of
https://github.com/MindWorkAI/AI-Studio.git
synced 2026-09-13 22:13:38 +00:00
46 lines
1.6 KiB
C#
46 lines
1.6 KiB
C#
|
using AIStudio.Settings.DataModel;
|
||
|
|
using AIStudio.Tools.PluginSystem;
|
||
|
|
|
||
|
|
namespace AIStudio.Tools.Services;
|
||
|
|
|
||
|
|
public sealed record DataSourceEmbeddingStatus(
|
||
|
|
string DataSourceId,
|
||
|
|
string DataSourceName,
|
||
|
|
DataSourceType DataSourceType,
|
||
|
|
DataSourceEmbeddingState State,
|
||
|
|
int TotalFiles,
|
||
|
|
int IndexedFiles,
|
||
|
|
int FailedFiles,
|
||
|
|
string CurrentFile,
|
||
|
|
string LastError,
|
||
|
|
IReadOnlyList<DataSourceEmbeddingFailure> Failures,
|
||
|
|
int PermanentlySkippedFiles = 0)
|
||
|
|
{
|
||
|
|
private static string TB(string fallbackEN) => I18N.I.T(fallbackEN, typeof(DataSourceEmbeddingStatus).Namespace, nameof(DataSourceEmbeddingStatus));
|
||
|
|
|
||
|
|
/// <remarks>
|
||
|
|
/// Files which were skipped for good are done, even though nothing was indexed of them.
|
||
|
|
/// Leaving them out would keep the bar short of the end for a data source which has nothing
|
||
|
|
/// left to do.
|
||
|
|
/// </remarks>
|
||
|
|
public int ProgressPercent => this.TotalFiles <= 0 ? 0 : Math.Clamp((int)Math.Round((this.IndexedFiles + this.PermanentlySkippedFiles) * 100d / this.TotalFiles), 0, 100);
|
||
|
|
|
||
|
|
public string StateLabel => this.State switch
|
||
|
|
{
|
||
|
|
DataSourceEmbeddingState.QUEUED => TB("Queued"),
|
||
|
|
DataSourceEmbeddingState.RUNNING => TB("Running"),
|
||
|
|
DataSourceEmbeddingState.COMPLETED => TB("Completed"),
|
||
|
|
DataSourceEmbeddingState.FAILED => TB("Needs attention"),
|
||
|
|
_ => TB("Idle")
|
||
|
|
};
|
||
|
|
|
||
|
|
public int SortOrder => this.State switch
|
||
|
|
{
|
||
|
|
DataSourceEmbeddingState.RUNNING => 0,
|
||
|
|
DataSourceEmbeddingState.QUEUED => 1,
|
||
|
|
DataSourceEmbeddingState.FAILED => 2,
|
||
|
|
DataSourceEmbeddingState.COMPLETED => 3,
|
||
|
|
_ => 4,
|
||
|
|
};
|
||
|
|
}
|