Added Estimate quality translation action

bb-io · Feb 5, 2025 · 355a7de · 355a7de
1 parent efa6b28
commit 355a7de
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/Apps.Widn/Actions/QualityActions.cs b/Apps.Widn/Actions/QualityActions.cs
@@ -106,7 +106,8 @@ public async Task<QualityResponse> EstimateQuality([ActionParameter] LanguageOpt
 
 
         [Action("Estimate XLIFF translation quality", Description = "Estimates the quality of a translation from an XLIFF file")]
-        public async Task<QualityResponse> EstimateQualityXliff([ActionParameter] FileRequest input, [ActionParameter][StaticDataSource(typeof(EstimateModelDataHandler))] string model)
+        public async Task<QualityResponse> EstimateQualityXliff([ActionParameter] FileRequest input, 
+            [ActionParameter][StaticDataSource(typeof(EstimateModelDataHandler))][Display("Model")] string model)
         {
             if (input.File == null)
                 throw new PluginMisconfigurationException("XLIFF file cannot be null. Please provide a valid file.");

diff --git a/Apps.Widn/DataSources/Enums/EstimateModelDataHandler.cs b/Apps.Widn/DataSources/Enums/EstimateModelDataHandler.cs
@@ -12,7 +12,7 @@ public class EstimateModelDataHandler : IStaticDataSourceItemHandler
     {
         public IEnumerable<DataSourceItem> GetData() => new List<DataSourceItem>()
         {
-            new("mqm-qe", "Anthill"),
+            new("mqm-qe", "Mqm-qe"),
             new("xcomet-xl", "Xcomet-xl")
         };
     }

diff --git a/Tests.Widn/TranslateTests.cs b/Tests.Widn/TranslateTests.cs
@@ -70,7 +70,7 @@ public async Task GetQualityXLIFF_ReturnsValues()
         {
             File = new FileReference { Name = "translated.xliff" },
         };
-        var result = await action.EstimateQualityXliff(input);
+        var result = await action.EstimateQualityXliff(input, "mqm-qe");
         Assert.IsNotNull(result);
         Console.WriteLine($"Final Score: {result.Score}");
         Assert.IsTrue(result.Score > 0);