添加了 AI 助手接口

2025-04-14 11:15:40 +08:00
parent 8305c5b3d6
commit a7210e527e
4 changed files with 124 additions and 3 deletions
@@ -0,0 +1,102 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+using CSharpToJsonSchema;
+using Ollama;
+using RhythmBase.Components;
+using System.ComponentModel;
+
+namespace RDTKEditor.AIAssistant
+{
+	public class Assistant
+	{
+		public const string Deepseek = "deepseek-r1:8b";
+		public const string Smollm = "smollm2:135m";
+		public const string Qwen_0_5 = "qwen2.5-coder:0.5b";
+		public const string Qwen_3 = "qwen2.5-coder:3b";
+		public OllamaApiClient Ollama { get; } = new(baseUri: new Uri("http://127.0.0.1:11434/api"));
+		public Chat Chat { get; }
+		public Assistant(string modelName)
+		{
+			Chat = Ollama.Chat(
+				model: modelName,
+				autoCallTools: true
+				);
+			var utils = new RhythmBaseUtils();
+			Chat.AddToolService(utils.AsTools().AsOllamaTools(), utils.AsCalls());
+		}
+		public async IAsyncEnumerable<GenerateCompletionResponse> GetResponse(string modelName, string userInput)
+		{
+			await foreach (var responce in Ollama.Completions.GenerateCompletionAsync(modelName, userInput))
+			{
+				if (responce != null)
+				{
+					yield return responce;
+				}
+			}
+		}
+	}
+	[GenerateJsonSchema]
+	public interface IRhythmBaseUtils
+	{
+		[Description("创建空关卡")]
+		public void CreateEmptyLevel();
+		[Description("创建关卡模板（默认关卡）")]
+		public void CreateLevelTemplate();
+		[Description("打开关卡")]
+		public void OpenLevel([Description("关卡路径")] string filepath);
+		[Description("添加轨道")]
+		public void AddRow(
+			[Description("轨道所在的房间")] RDRoomIndex roomIndex,
+			[Description("轨道名称")] string trackName
+			);
+		[Description("保存关卡")]
+		public void SaveLevel([Description("关卡路径，后缀为 .rdlevel")] string filepath);
+	}
+	public class RhythmBaseUtils : IRhythmBaseUtils
+	{
+		private RDLevel? level;
+		public void CreateEmptyLevel()
+		{
+			Console.BackgroundColor = ConsoleColor.Green;
+			Console.WriteLine($"Called CreateEmptyLevel");
+			Console.ResetColor();
+			level = [];
+		}
+		public void CreateLevelTemplate()
+		{
+			Console.BackgroundColor = ConsoleColor.Green;
+			Console.WriteLine($"Called CreateLevelTemplate");
+			Console.ResetColor();
+			level = RDLevel.Default;
+		}
+		public void OpenLevel(string filepath)
+		{
+			Console.BackgroundColor = ConsoleColor.Green;
+			Console.WriteLine($"Called OpenLevel: {filepath}");
+			Console.ResetColor();
+			level = RDLevel.Read(filepath);
+		}
+		public void AddRow(RDRoomIndex roomIndex, string trackName)
+		{
+			Console.BackgroundColor = ConsoleColor.Green;
+			Console.WriteLine($"Called AddRow: [{roomIndex}] {trackName}");
+			Console.ResetColor();
+			level?.CreateRow(roomIndex, trackName);
+		}
+		public void SaveLevel(string filepath)
+		{
+			Console.BackgroundColor = ConsoleColor.Green;
+			Console.WriteLine($"Called SaveLevel: {filepath}");
+			Console.ResetColor();
+			string dir = Path.GetDirectoryName(filepath) ?? "";
+			if (!string.IsNullOrEmpty(dir))
+			{
+				Directory.CreateDirectory(dir);
+			}
+			level?.Write(filepath);
+		}
+	}
+}