SemianiakaVY/BotSharp
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

Fix map key.

Browse source
This commit is contained in:
Haiping Chen 2024-05-14 14:47:36 -05:00
parent ed9e6457fe
commit 535e20ea90
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
src/Infrastructure/BotSharp.Core/Translation/TranslationService.cs
View file

@ -1,3 +1,4 @@
using Amazon.Runtime.Internal.Transform;
using BotSharp.Abstraction.Infrastructures.Enums; using BotSharp.Abstraction.Infrastructures.Enums;
using BotSharp.Abstraction.MLTasks; using BotSharp.Abstraction.MLTasks;
using BotSharp.Abstraction.Options; using BotSharp.Abstraction.Options;
@ -54,8 +55,9 @@ public class TranslationService : ITranslationService
model: _router?.LlmConfig?.Model); model: _router?.LlmConfig?.Model);
var template = _router.Templates.First(x => x.Name == "translation_prompt").Content; var template = _router.Templates.First(x => x.Name == "translation_prompt").Content;
var keys = unique.ToArray();
var texts = unique.ToArray() var texts = unique.ToArray()
.Select((text, i) => $"{i + 1}. {text}") .Select((text, i) => $"{i + 1}. \"{text}\"")
.ToList(); .ToList();
var translatedStringList = await InnerTranslate(texts, language, template); var translatedStringList = await InnerTranslate(texts, language, template);
@ -74,7 +76,7 @@ public class TranslationService : ITranslationService
for (var i = 0; i < texts.Count; i++) for (var i = 0; i < texts.Count; i++)
{ {
map.Add(texts[i], translatedTexts[i]); map.Add(keys[0], translatedTexts[i]);
} }
clonedData = Assign(clonedData, map); clonedData = Assign(clonedData, map);