正常

2025-12-12 18:22:26 +08:00
parent 968268b368
commit 8141c6cad2
11 changed files with 544 additions and 279 deletions
--- a/Source/WulaFallenEmpire/EventSystem/AI/SimpleAIClient.cs
+++ b/Source/WulaFallenEmpire/EventSystem/AI/SimpleAIClient.cs
@@ -0,0 +1,173 @@
+using System;
+using System.Collections.Generic;
+using System.Text;
+using System.Threading.Tasks;
+using UnityEngine.Networking;
+using Verse;
+
+namespace WulaFallenEmpire.EventSystem.AI
+{
+    public class SimpleAIClient
+    {
+        private readonly string _apiKey;
+        private readonly string _baseUrl;
+        private readonly string _model;
+
+        public SimpleAIClient(string apiKey, string baseUrl, string model)
+        {
+            _apiKey = apiKey;
+            _baseUrl = baseUrl?.TrimEnd('/');
+            _model = model;
+        }
+
+        public async Task<string> GetChatCompletionAsync(string instruction, List<(string role, string message)> messages)
+        {
+            if (string.IsNullOrEmpty(_baseUrl))
+            {
+                Log.Error("[WulaAI] Base URL is missing.");
+                return null;
+            }
+
+            string endpoint = $"{_baseUrl}/chat/completions";
+            // Handle cases where baseUrl already includes /v1 or full path
+            if (_baseUrl.EndsWith("/chat/completions")) endpoint = _baseUrl;
+            else if (!_baseUrl.EndsWith("/v1")) endpoint = $"{_baseUrl}/v1/chat/completions";
+
+            // Build JSON manually to avoid dependencies
+            StringBuilder jsonBuilder = new StringBuilder();
+            jsonBuilder.Append("{");
+            jsonBuilder.Append($"\"model\": \"{_model}\",");
+            jsonBuilder.Append("\"stream\": false,");
+            jsonBuilder.Append("\"messages\": [");
+            
+            // System instruction
+            if (!string.IsNullOrEmpty(instruction))
+            {
+                jsonBuilder.Append($"{{\"role\": \"system\", \"content\": \"{EscapeJson(instruction)}\"}},");
+            }
+
+            // Messages
+            for (int i = 0; i < messages.Count; i++)
+            {
+                var msg = messages[i];
+                string role = msg.role.ToLower();
+                if (role == "ai") role = "assistant";
+                // Map other roles if needed
+                
+                jsonBuilder.Append($"{{\"role\": \"{role}\", \"content\": \"{EscapeJson(msg.message)}\"}}");
+                if (i < messages.Count - 1) jsonBuilder.Append(",");
+            }
+            
+            jsonBuilder.Append("]");
+            jsonBuilder.Append("}");
+
+            string jsonBody = jsonBuilder.ToString();
+            Log.Message($"[WulaAI] Sending request to {endpoint}");
+
+            using (UnityWebRequest request = new UnityWebRequest(endpoint, "POST"))
+            {
+                byte[] bodyRaw = Encoding.UTF8.GetBytes(jsonBody);
+                request.uploadHandler = new UploadHandlerRaw(bodyRaw);
+                request.downloadHandler = new DownloadHandlerBuffer();
+                request.SetRequestHeader("Content-Type", "application/json");
+                if (!string.IsNullOrEmpty(_apiKey))
+                {
+                    request.SetRequestHeader("Authorization", $"Bearer {_apiKey}");
+                }
+
+                var operation = request.SendWebRequest();
+
+                while (!operation.isDone)
+                {
+                    await Task.Delay(50);
+                }
+
+#if UNITY_2020_2_OR_NEWER
+                if (request.result == UnityWebRequest.Result.ConnectionError || request.result == UnityWebRequest.Result.ProtocolError)
+#else
+                if (request.isNetworkError || request.isHttpError)
+#endif
+                {
+                    Log.Error($"[WulaAI] API Error: {request.error}\nResponse: {request.downloadHandler.text}");
+                    return null;
+                }
+
+                string responseText = request.downloadHandler.text;
+                Log.Message($"[WulaAI] Raw Response: {responseText}");
+                return ExtractContent(responseText);
+            }
+        }
+
+        private string EscapeJson(string s)
+        {
+            if (s == null) return "";
+            return s.Replace("\\", "\\\\")
+                    .Replace("\"", "\\\"")
+                    .Replace("\n", "\\n")
+                    .Replace("\r", "\\r")
+                    .Replace("\t", "\\t");
+        }
+
+        private string ExtractContent(string json)
+        {
+            try
+            {
+                // Robust parsing for "content": "..." allowing for whitespace variations
+                int contentIndex = json.IndexOf("\"content\"");
+                if (contentIndex == -1) return null;
+
+                // Find the opening quote after "content"
+                int openQuoteIndex = -1;
+                for (int i = contentIndex + 9; i < json.Length; i++)
+                {
+                    if (json[i] == '"')
+                    {
+                        openQuoteIndex = i;
+                        break;
+                    }
+                }
+                if (openQuoteIndex == -1) return null;
+
+                int startIndex = openQuoteIndex + 1;
+                StringBuilder content = new StringBuilder();
+                bool escaped = false;
+                
+                for (int i = startIndex; i < json.Length; i++)
+                {
+                    char c = json[i];
+                    if (escaped)
+                    {
+                        if (c == 'n') content.Append('\n');
+                        else if (c == 'r') content.Append('\r');
+                        else if (c == 't') content.Append('\t');
+                        else if (c == '"') content.Append('"');
+                        else if (c == '\\') content.Append('\\');
+                        else content.Append(c); // Literal
+                        escaped = false;
+                    }
+                    else
+                    {
+                        if (c == '\\')
+                        {
+                            escaped = true;
+                        }
+                        else if (c == '"')
+                        {
+                            // End of string
+                            return content.ToString();
+                        }
+                        else
+                        {
+                            content.Append(c);
+                        }
+                    }
+                }
+            }
+            catch (Exception ex)
+            {
+                Log.Error($"[WulaAI] Error parsing response: {ex}");
+            }
+            return null;
+        }
+    }
+}