Save Plugins opened

PromptPlayground/Messages/PluginOpenMessage.cs

@@ -6,4 +6,9 @@ public class PluginOpenMessage : FileOrFolderOpenMessage
     {
         public PluginOpenMessage(string path) : base(path) { }
     }
+
+    public class PluginCloseMessage : FileOrFolderOpenMessage
+    {
+        public PluginCloseMessage(string path) : base(path) { }
+    }
 }

PromptPlayground/Services/ProfileService.cs

@@ -0,0 +1,50 @@
+using System;
+using System.Collections.Generic;
+using System.IO;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+
+namespace PromptPlayground.Services
+{
+    internal class ProfileService<T>
+    {
+        private readonly string _profile;
+
+        public ProfileService(string profile = nameof(T))
+        {
+            this._profile = profile;
+        }
+
+        public string ProfilePath()
+        {
+            var userProfile = Environment.GetFolderPath(Environment.SpecialFolder.UserProfile);
+            var profileFolder = Path.Combine(userProfile, ".prompt_playground");
+            if (!Directory.Exists(profileFolder))
+            {
+                Directory.CreateDirectory(profileFolder);
+            }
+            return Path.Combine(profileFolder, $"{_profile}.json");
+        }
+
+
+
+        public T? Get()
+        {
+            var path = ProfilePath();
+            if (!File.Exists(path))
+            {
+                return default;
+            }
+            var json = File.ReadAllText(path);
+            return System.Text.Json.JsonSerializer.Deserialize<T>(json);
+        }
+
+        public void Save(T profile)
+        {
+            var path = ProfilePath();
+            var json = System.Text.Json.JsonSerializer.Serialize(profile);
+            File.WriteAllText(path, json, Encoding.UTF8);
+        }
+    }
+}

PromptPlayground/Services/PromptService.cs

@@ -10,88 +10,88 @@
 
 namespace PromptPlayground.Services
 {
-	public class PromptService
-	{
-		private readonly ILLMConfigViewModel model;
-
-		public PromptService(IConfigAttributesProvider provider)
-		{
-			this.model = provider.GetLLM() ?? throw new Exception("无法创建Kernel，请检查LLM配置");
-		}
-
-		private Kernel Build()
-		{
-
-			var builder = model.CreateKernelBuilder();
-
-			var _kernel = builder.Build();
-
-			_kernel.ImportPluginFromType<TimePlugin>();
-
-
-			return _kernel;
-		}
-
-		public async Task<GenerateResult> RunAsync(string prompt, PromptExecutionSettings? config, KernelArguments arguments, CancellationToken cancellationToken = default)
-		{
-			var _kernel = Build();
-			var promptFilter = new KernelFilter();
-			_kernel.PromptFilters.Add(promptFilter);
-			_kernel.FunctionFilters.Add(promptFilter);
-
-			var sw = Stopwatch.StartNew();
-			var func = _kernel.CreateFunctionFromPrompt(prompt, config);
-
-			var result = await func.InvokeAsync(_kernel, arguments, cancellationToken);
-			sw.Stop();
-			try
-			{
-				var usage = model.GetUsage(result);
-				return new GenerateResult()
-				{
-					Text = result.GetValue<string>() ?? "",
-					TokenUsage = usage,
-					Elapsed = sw.Elapsed,
-					PromptRendered = promptFilter.PromptRendered
-				};
-			}
-			catch (KernelException ex)
-			{
-				return new GenerateResult()
-				{
-					Text = ex.Message,
-					Elapsed = sw.Elapsed,
-					Error = ex.Message,
-				};
-			}
-		}
-
-		public KernelArguments CreateArguments() => new KernelArguments();
-	}
-
-	public class KernelFilter : IPromptFilter, IFunctionFilter
-	{
-		public string? PromptRendered { get; set; }
-		public void OnFunctionInvoked(FunctionInvokedContext context)
-		{
-
-		}
-
-		public void OnFunctionInvoking(FunctionInvokingContext context)
-		{
-
-		}
-
-		public void OnPromptRendered(PromptRenderedContext context)
-		{
-			PromptRendered = context.RenderedPrompt;
-		}
-
-		public void OnPromptRendering(PromptRenderingContext context)
-		{
-
-		}
-	}
+    public class PromptService
+    {
+        private readonly ILLMConfigViewModel model;
+
+        public PromptService(IConfigAttributesProvider provider)
+        {
+            this.model = provider.GetLLM() ?? throw new Exception("无法创建Kernel，请检查LLM配置");
+        }
+
+        private Kernel Build()
+        {
+
+            var builder = model.CreateKernelBuilder();
+
+            var _kernel = builder.Build();
+
+            _kernel.ImportPluginFromType<TimePlugin>();
+
+
+            return _kernel;
+        }
+
+        public async Task<GenerateResult> RunAsync(string prompt, PromptExecutionSettings? config, KernelArguments arguments, CancellationToken cancellationToken = default)
+        {
+            var _kernel = Build();
+            var promptFilter = new KernelFilter();
+            _kernel.PromptFilters.Add(promptFilter);
+            _kernel.FunctionFilters.Add(promptFilter);
+
+            var sw = Stopwatch.StartNew();
+            var func = _kernel.CreateFunctionFromPrompt(prompt, config);
+
+            var result = await func.InvokeAsync(_kernel, arguments, cancellationToken);
+            sw.Stop();
+            try
+            {
+                var usage = model.GetUsage(result);
+                return new GenerateResult()
+                {
+                    Text = result.GetValue<string>() ?? "",
+                    TokenUsage = usage,
+                    Elapsed = sw.Elapsed,
+                    PromptRendered = promptFilter.PromptRendered
+                };
+            }
+            catch (KernelException ex)
+            {
+                return new GenerateResult()
+                {
+                    Text = ex.Message,
+                    Elapsed = sw.Elapsed,
+                    Error = ex.Message,
+                };
+            }
+        }
+
+        public KernelArguments CreateArguments() => new KernelArguments();
+    }
+
+    public class KernelFilter : IPromptFilter, IFunctionFilter
+    {
+        public string? PromptRendered { get; set; }
+        public void OnFunctionInvoked(FunctionInvokedContext context)
+        {
+
+        }
+
+        public void OnFunctionInvoking(FunctionInvokingContext context)
+        {
+
+        }
+
+        public void OnPromptRendered(PromptRenderedContext context)
+        {
+            PromptRendered = context.RenderedPrompt;
+        }
+
+        public void OnPromptRendering(PromptRenderingContext context)
+        {
+
+        }
+    }
 
 
 }

PromptPlayground/ViewModels/ConfigViewModel.cs

@@ -22,7 +22,7 @@ public partial class ConfigViewModel : ViewModelBase, IConfigAttributesProvider,
     private string[] RequiredAttributes =
    [
        #region LLM Config
-        ConfigAttribute.AzureDeployment,
+       ConfigAttribute.AzureDeployment,
        ConfigAttribute.AzureEndpoint,
        ConfigAttribute.AzureSecret,
        ConfigAttribute.BaiduClientId,

PromptPlayground/ViewModels/GenerateResult.cs

@@ -5,35 +5,35 @@
 
 namespace PromptPlayground.Services
 {
-	public partial class GenerateResult : ViewModelBase
-	{
-		public string Text { get; set; } = string.Empty;
-		public TimeSpan Elapsed { get; set; }
-		public string? Error { get; set; } = string.Empty;
-		public string? PromptRendered { get; set; }
-		public ResultTokenUsage? TokenUsage { get; set; }
+    public partial class GenerateResult : ViewModelBase
+    {
+        public string Text { get; set; } = string.Empty;
+        public TimeSpan Elapsed { get; set; }
+        public string? Error { get; set; } = string.Empty;
+        public string? PromptRendered { get; set; }
+        public ResultTokenUsage? TokenUsage { get; set; }
 
-		public bool HasError => !string.IsNullOrWhiteSpace(Error);
+        public bool HasError => !string.IsNullOrWhiteSpace(Error);
 
-		[RelayCommand]
-		public void CopyPrompt()
-		{
-			if (!string.IsNullOrWhiteSpace(PromptRendered))
-			{
-				WeakReferenceMessenger.Default.Send(new CopyTextMessage(PromptRendered));
-			}
-		}
+        [RelayCommand]
+        public void CopyPrompt()
+        {
+            if (!string.IsNullOrWhiteSpace(PromptRendered))
+            {
+                WeakReferenceMessenger.Default.Send(new CopyTextMessage(PromptRendered));
+            }
+        }
 
-		[RelayCommand]
-		public void CopyText()
-		{
-			if (!string.IsNullOrWhiteSpace(Text) && !HasError)
-			{
-				WeakReferenceMessenger.Default.Send(new CopyTextMessage(Text));
-			}
-		}
+        [RelayCommand]
+        public void CopyText()
+        {
+            if (!string.IsNullOrWhiteSpace(Text) && !HasError)
+            {
+                WeakReferenceMessenger.Default.Send(new CopyTextMessage(Text));
+            }
+        }
 
-	}
+    }
 
-	public record ResultTokenUsage(int Total, int Prompt, int Completion);
+    public record ResultTokenUsage(int Total, int Prompt, int Completion);
 }

PromptPlayground/ViewModels/PluginViewModel.cs

@@ -69,6 +69,20 @@ private bool CanAddNewFunction()
             return !Directory.Exists(this.Folder);
         }
 
+        [RelayCommand(CanExecute = nameof(CanRemovePlugin))]
+        public void RemovePlugin()
+        {
+            if (this.Folder != null && Directory.Exists(this.Folder))
+            {
+                WeakReferenceMessenger.Default.Send(new PluginCloseMessage(this.Folder));
+            }
+        }
+
+        private bool CanRemovePlugin()
+        {
+            return Directory.Exists(this.Folder);
+        }
+
         public void AddNewFunction(SemanticFunctionViewModel function)
         {
             if (string.IsNullOrWhiteSpace(function.Name))
@@ -77,8 +91,8 @@ public void AddNewFunction(SemanticFunctionViewModel function)
             }
             this.Functions.Add(function);
         }
-        const string DefaultName = "New Function";
-        readonly Regex DefaultNamePattern = new(@"\[New Function (?<index>\d+)\]", RegexOptions.Compiled);
+        const string DefaultName = "Unsaved";
+        readonly Regex DefaultNamePattern = new(@"\[Unsaved (?<index>\d+)\]", RegexOptions.Compiled);
         private int FindLastFunctionIndex()
         {
             return this.Functions.Where(_ => DefaultNamePattern.IsMatch(_.Name))