From 1e0d345fa4fc46c7e4c0f51d3de660e8e16aa50a Mon Sep 17 00:00:00 2001
From: MrBounty <adrien.bouvais.pro@gmail.com>
Date: Sun, 4 Aug 2024 22:16:52 +0200
Subject: [PATCH] Make lastSelectedModels var per user

---
 Request.go | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/Request.go b/Request.go
index fc00dae..9d65fbc 100644
--- a/Request.go
+++ b/Request.go
@@ -17,7 +17,7 @@ type RequestMessage struct {
 	Content string `json:"content"`
 }
 
-var lastSelectedLLMs []LLM
+var lastSelectedLLMs = make(map[string][]LLM)
 
 func GeneratePlaceholderHandler(c *fiber.Ctx) error {
 	// Step 1 I create a User message and send it as output with a placeholder
@@ -68,7 +68,7 @@ func GeneratePlaceholderHTML(c *fiber.Ctx, message string, selectedLLMIds []stri
 		}
 		selectedLLMs = append(selectedLLMs, selectedLLM)
 	}
-	lastSelectedLLMs = selectedLLMs
+	lastSelectedLLMs[c.Cookies("jade-edgedb-auth-token")] = selectedLLMs
 
 	_, position := insertArea(c)
 
@@ -105,7 +105,7 @@ func GenerateMultipleMessagesHandler(c *fiber.Ctx) error {
 	// Step 2 generate multiple messages
 	// And send them one by one using events
 	insertArea(c)
-	selectedLLMs := lastSelectedLLMs
+	selectedLLMs, _ := lastSelectedLLMs[c.Cookies("jade-edgedb-auth-token")]
 
 	var user User
 	err := edgeGlobalClient.WithGlobals(map[string]interface{}{"ext::auth::client_token": c.Cookies("jade-edgedb-auth-token")}).QuerySingle(context.Background(), `
@@ -238,6 +238,8 @@ func GenerateMultipleMessagesHandler(c *fiber.Ctx) error {
 	// Wait for all goroutines to finish
 	wg.Wait()
 
+	delete(lastSelectedLLMs, c.Cookies("jade-edgedb-auth-token"))
+
 	return c.SendString("")
 }