Merge pull request #242 from Yoko-0x0/feature/lmstudio

feat: Add LMStudio API integration with configuration and UI support

Author: SakiRinn

src/apis/ModelsApiService.cs

@@ -0,0 +1,134 @@
+using System.Net.Http;
+using System.Text.Json;
+using LiveCaptionsTranslator.utils;
+
+namespace LiveCaptionsTranslator.apis
+{
+    /// <summary>
+    /// Servicio para obtener listas de modelos desde APIs compatibles (LMStudio, Ollama, etc.)
+    /// </summary>
+    public static class ModelsApiService
+    {
+        private static readonly HttpClient client = new HttpClient()
+        {
+            Timeout = TimeSpan.FromSeconds(10)
+        };
+
+        /// <summary>
+        /// APIs que soportan obtener modelos desde un endpoint.
+        /// </summary>
+        public static readonly List<string> APIs_WITH_MODELS_ENDPOINT = new()
+        {
+            "LMStudio",
+            "Ollama"
+        };
+
+        /// <summary>
+        /// Obtiene la URL del endpoint de modelos para una API.
+        /// </summary>
+        public static string GetModelsEndpoint(string apiName, string baseUrl)
+        {
+            return apiName switch
+            {
+                "LMStudio" => TextUtil.NormalizeUrl(baseUrl) + "/models",
+                "Ollama" => TextUtil.NormalizeUrl(baseUrl) + "/api/tags",
+                _ => null
+            };
+        }
+
+        /// <summary>
+        /// Obtiene la lista de modelos disponibles desde la API.
+        /// </summary>
+        /// <param name="apiName">Nombre de la API (LMStudio, Ollama, etc.)</param>
+        /// <param name="baseUrl">URL base de la API</param>
+        /// <returns>Lista de identificadores de modelos para usar en el chat</returns>
+        public static async Task<List<ModelInfo>> FetchModelsAsync(string apiName, string baseUrl, CancellationToken token = default)
+        {
+            string endpoint = GetModelsEndpoint(apiName, baseUrl);
+            if (string.IsNullOrEmpty(endpoint))
+                return new List<ModelInfo>();
+
+            try
+            {
+                var response = await client.GetAsync(endpoint, token);
+                if (!response.IsSuccessStatusCode)
+                    return new List<ModelInfo>();
+
+                string json = await response.Content.ReadAsStringAsync(token);
+
+                return apiName switch
+                {
+                    "LMStudio" => ParseLMStudioModels(json),
+                    "Ollama" => ParseOllamaModels(json),
+                    _ => new List<ModelInfo>()
+                };
+            }
+            catch
+            {
+                return new List<ModelInfo>();
+            }
+        }
+
+        public class ModelInfo
+        {
+            public string Id { get; set; }
+            public string DisplayName { get; set; }
+        }
+
+        private static List<ModelInfo> ParseLMStudioModels(string json)
+        {
+            var result = new List<ModelInfo>();
+            try
+            {
+                using var doc = JsonDocument.Parse(json);
+                var root = doc.RootElement;
+
+                if (!root.TryGetProperty("models", out var modelsArray))
+                    return result;
+
+                foreach (var model in modelsArray.EnumerateArray())
+                {
+                    string type = model.TryGetProperty("type", out var typeProp) ? typeProp.GetString() : null;
+                    if (type != "llm")
+                        continue;
+
+                    string key = model.TryGetProperty("key", out var keyProp) ? keyProp.GetString() : null;
+                    if (string.IsNullOrEmpty(key))
+                        continue;
+
+                    string displayName = model.TryGetProperty("display_name", out var dnProp) ? dnProp.GetString() : key;
+
+                    result.Add(new ModelInfo { Id = key, DisplayName = displayName ?? key });
+                }
+            }
+            catch { }
+
+            return result;
+        }
+
+        private static List<ModelInfo> ParseOllamaModels(string json)
+        {
+            var result = new List<ModelInfo>();
+            try
+            {
+                using var doc = JsonDocument.Parse(json);
+                var root = doc.RootElement;
+
+                if (!root.TryGetProperty("models", out var modelsArray))
+                    return result;
+
+                foreach (var model in modelsArray.EnumerateArray())
+                {
+                    string name = model.TryGetProperty("name", out var nameProp) ? nameProp.GetString() : null;
+                    if (string.IsNullOrEmpty(name))
+                        continue;
+
+                    result.Add(new ModelInfo { Id = name, DisplayName = name });
+                }
+            }
+            catch { }
+
+            return result;
+        }
+    }
+}

src/apis/TranslateAPI.cs

@@ -1,4 +1,4 @@
-using System.Net;
+using System.Net;
 using System.Net.Http;
 using System.Text;
 using System.Text.Json;
@@ -23,6 +23,7 @@ public static readonly Dictionary<string, Func<string, CancellationToken, Task<s
             { "Google2", Google2 },
             { "Ollama", Ollama },
             { "OpenAI", OpenAI },
+            { "LMStudio", LMStudio },
             { "DeepL", DeepL },
             { "OpenRouter", OpenRouter },
             { "Youdao", Youdao },
@@ -32,7 +33,7 @@ public static readonly Dictionary<string, Func<string, CancellationToken, Task<s
         };
         public static readonly List<string> LLM_BASED_APIS = new()
         {
-            "Ollama", "OpenAI", "OpenRouter"
+            "Ollama", "OpenAI", "OpenRouter", "LMStudio"
         };
         public static readonly List<string> NO_CONFIG_APIS = new()
         {
@@ -191,6 +192,92 @@ public static async Task<string> Ollama(string text, CancellationToken token = d
                 return $"[ERROR] Translation Failed: HTTP Error - {response.StatusCode}";
         }
 
+        public static async Task<string> LMStudio(string text, CancellationToken token = default)
+        {
+            var config = Translator.Setting["LMStudio"] as LMStudioConfig;
+            string language = LMStudioConfig.SupportedLanguages.TryGetValue(
+                Translator.Setting.TargetLanguage, out var langValue) ? langValue : Translator.Setting.TargetLanguage;
+            string apiUrl = TextUtil.NormalizeUrl(config.ApiUrl) + "/chat";
+
+            string systemPrompt = string.Format(Prompt, language);
+
+            // Build input with optional context
+            string input = $"🔤 {text} 🔤";
+            if (Translator.Setting.ContextAware)
+            {
+                var contextLines = new List<string>();
+                foreach (var entry in Translator.Caption.AwareContexts)
+                {
+                    string translatedText = entry.TranslatedText;
+                    if (translatedText.Contains("[ERROR]") || translatedText.Contains("[WARNING]"))
+                        continue;
+                    translatedText = RegexPatterns.NoticePrefix().Replace(translatedText, "");
+                    contextLines.Add($"🔤 {entry.SourceText} 🔤 → {translatedText}");
+                }
+                if (contextLines.Count > 0)
+                    input = string.Join("\n", contextLines) + "\n" + input;
+            }
+
+            var requestData = new
+            {
+                model = config.ModelName,
+                system_prompt = systemPrompt,
+                input = input,
+                temperature = config.Temperature
+            };
+
+            string jsonContent = JsonSerializer.Serialize(requestData);
+            var content = new StringContent(jsonContent, Encoding.UTF8, "application/json");
+            client.DefaultRequestHeaders.Clear();
+
+            HttpResponseMessage response;
+            try
+            {
+                response = await client.PostAsync(apiUrl, content, token);
+            }
+            catch (OperationCanceledException ex)
+            {
+                if (ex.Message.StartsWith("The request"))
+                    return $"[ERROR] Translation Failed: The request was canceled due to timeout (> 8 seconds), " +
+                           $"please use a faster API or check network connection.";
+                throw;
+            }
+            catch (Exception ex)
+            {
+                return $"[ERROR] Translation Failed: {ex.Message}";
+            }
+
+            if (response.IsSuccessStatusCode)
+            {
+                string responseString = await response.Content.ReadAsStringAsync();
+                using var doc = JsonDocument.Parse(responseString);
+                var root = doc.RootElement;
+
+                // LMStudio native /api/v1/chat response:
+                // { "output": [ { "type": "message", "content": "..." }, ... ] }
+                if (root.TryGetProperty("output", out var outputArray) &&
+                    outputArray.ValueKind == JsonValueKind.Array)
+                {
+                    foreach (var item in outputArray.EnumerateArray())
+                    {
+                        if (item.TryGetProperty("type", out var typeProp) &&
+                            typeProp.GetString() == "message" &&
+                            item.TryGetProperty("content", out var contentProp))
+                        {
+                            return RegexPatterns.ModelThinking().Replace(contentProp.GetString() ?? "", "");
+                        }
+                    }
+                }
+
+                return "[ERROR] Translation Failed: Unexpected response format";
+            }
+            else
+            {
+                string body = await response.Content.ReadAsStringAsync();
+                return $"[ERROR] Translation Failed: HTTP Error - {response.StatusCode}: {body}";
+            }
+        }
+
         public static async Task<string> OpenRouter(string text, CancellationToken token = default)
         {
             var config = Translator.Setting["OpenRouter"] as OpenRouterConfig;

src/models/Setting.cs

@@ -1,4 +1,4 @@
-using System.ComponentModel;
+using System.ComponentModel;
 using System.IO;
 using System.Runtime.CompilerServices;
 using System.Text.Json;
@@ -196,6 +196,7 @@ public Setting()
                 { "Google2", [new TranslateAPIConfig()] },
                 { "Ollama", [new OllamaConfig()] },
                 { "OpenAI", [new OpenAIConfig()] },
+                { "LMStudio", [new LMStudioConfig()] },
                 { "OpenRouter", [new OpenRouterConfig()] },
                 { "DeepL", [new DeepLConfig()] },
                 { "Youdao", [new YoudaoConfig()] },
@@ -209,6 +210,7 @@ public Setting()
                 { "Google2", 0 },
                 { "Ollama", 0 },
                 { "OpenAI", 0 },
+                { "LMStudio", 0 },
                 { "OpenRouter", 0 },
                 { "DeepL", 0 },
                 { "Youdao", 0 },
@@ -265,6 +267,13 @@ public static Setting Load(string jsonPath)
                     setting.Configs[key] = [new TranslateAPIConfig()];
             }
 
+            // Ensure ConfigIndices has all keys (for upgrades from older setting.json)
+            foreach (string key in TranslateAPI.TRANSLATE_FUNCTIONS.Keys)
+            {
+                if (!setting.ConfigIndices.ContainsKey(key))
+                    setting.ConfigIndices[key] = 0;
+            }
+
             return setting;
         }
 

src/models/TranslateAPIConfig.cs

@@ -1,4 +1,4 @@
-using System.ComponentModel;
+using System.ComponentModel;
 using System.Runtime.CompilerServices;
 using System.Text.Json.Serialization;
 
@@ -164,6 +164,27 @@ public string ApiKey
         }
     }
 
+    public class LMStudioConfig : BaseLLMConfig
+    {
+        public class Response
+        {
+            public string model { get; set; }
+            public string output { get; set; }
+        }
+
+        private string apiUrl = "http://localhost:1234/api/v1";
+
+        public string ApiUrl
+        {
+            get => apiUrl;
+            set
+            {
+                apiUrl = value;
+                OnPropertyChanged("ApiUrl");
+            }
+        }
+    }
+
     public class DeepLConfig : TranslateAPIConfig
     {
         [JsonIgnore]