fix: custom target language doesn't work in llm

SakiRinn · Feb 22, 2025 · 402203f · 402203f
1 parent 5ba3893
commit 402203f
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 7 deletions.
diff --git a/Properties/AssemblyInfo.cs b/Properties/AssemblyInfo.cs
@@ -1,4 +1,4 @@
-
+
 //
 // This code was generated by a tool. Any changes made manually will be lost
 // the next time this code is regenerated.
@@ -8,5 +8,5 @@
 
 
 [assembly: AssemblyTitle("LiveCaptions Translator")]
-[assembly: AssemblyVersion("1.1.1251.2215")]
-[assembly: AssemblyFileVersion("1.1.1251.2215")]
+[assembly: AssemblyVersion("1.1.1251.2220")]
+[assembly: AssemblyFileVersion("1.1.1251.2220")]
diff --git a/src/models/TranslateAPI.cs b/src/models/TranslateAPI.cs
@@ -30,8 +30,9 @@ public static string Prompt
         public static async Task<string> OpenAI(string text)
         {
             var config = App.Settings.CurrentAPIConfig as OpenAIConfig;
-            var language = config?.SupportedLanguages[App.Settings.TargetLanguage];
-
+            string language = config.SupportedLanguages.TryGetValue(App.Settings.TargetLanguage, out var langValue) 
+                ? langValue 
+                : App.Settings.TargetLanguage; 
             var requestData = new
             {
                 model = config?.ModelName,
@@ -75,8 +76,9 @@ public static async Task<string> Ollama(string text)
             var apiUrl = $"http://localhost:{OLLAMA_PORT}/api/chat";
 
             var config = App.Settings.CurrentAPIConfig as OllamaConfig;
-            var language = config?.SupportedLanguages[App.Settings.TargetLanguage];
-
+            string language = config.SupportedLanguages.TryGetValue(App.Settings.TargetLanguage, out var langValue) 
+                ? langValue 
+                : App.Settings.TargetLanguage; 
             var requestData = new
             {
                 model = config?.ModelName,