Merge pull request #342 from microsoft/block-deletion-of-global-assis…

…tants

block deletion of global assistant assets

src/AzureAIProxy.Shared/Database/Assistant.cs

@@ -1,7 +1,17 @@
 namespace AzureAIProxy.Shared.Database;
+using System.ComponentModel.DataAnnotations.Schema;
+
+public enum Scope
+{
+    Personal,
+    Global
+}
 
 public partial class Assistant
 {
     public string ApiKey { get; set; } = null!;
     public string Id { get; set; } = null!;
+
+    [NotMapped]
+    public Scope Scope { get; set; } = Scope.Personal; // Default to Personal
 }

src/AzureAIProxy/Routes/AzureOpenAIAssistants.cs

@@ -134,6 +134,8 @@ private static async Task<IResult> CreateThreadAsync(
                 var assistant = await assistantService.GetIdAsync(requestContext.ApiKey, assistantId.Split("/").First());
                 if (assistant is null)
                     return OpenAIResult.Unauthorized("Unauthorized assistant access.");
+                else if (method == HttpMethod.Delete.Method && assistant.Scope == Scope.Global)
+                    return OpenAIResult.Unauthorized("Unauthorized assistant deletion.");
             }
             else if (threadId is not null)
             {

src/AzureAIProxy/Services/AssistantService.cs

@@ -110,7 +110,8 @@ public async Task DeleteIdAsync(string apiKey, string responseContent)
                     result = new Assistant
                     {
                         ApiKey = apiKey,
-                        Id = id
+                        Id = id,
+                        Scope = Scope.Global
                     };
                     memoryCache.Set(cacheKey, result, TimeSpan.FromMinutes(10));
                 }