Minor fixes to cookie due to cross-origin access

Author: frikky

ai.go

@@ -1674,6 +1674,7 @@ func AutofixAppLabels(app WorkflowApp, label string, keys []string) (WorkflowApp
 
 	var guessedAction WorkflowAppAction
 	type ActionStruct struct {
+		Success bool `json:"success"`
 		Action string `json:"action"`
 	}
 
@@ -1708,7 +1709,12 @@ func AutofixAppLabels(app WorkflowApp, label string, keys []string) (WorkflowApp
 	//userMessage := "The available actions are as follows:\n"
 
 	if cacheGeterr != nil {
-		systemMessage := `Your goal is to find the most correct action for a specific label from the actions. You have to pick the most likely action. Synonyms are accepted, and you should be very critical to not make mistakes. A synonym example can be something like: case = alert = ticket = issue = task, or message = chat = communication. Be extra careful of not confusing LIST and GET operations, based on the user query, respond with the most likely action name. If it exists, return {"success": true, "action": "<action>"} where <action> is replaced with the action found. If it does not exist, Last case scenario is return {"success": false, "action": ""}. Output as JSON."`
+		systemMessage := `Your goal is to find the most correct action for a specific label from the actions. You have to pick the most likely action. Synonyms are accepted, and you should be very critical to not make mistakes. A synonym example can be something like: case = alert = ticket = issue = task, or message = chat = communication. Be extra careful of not confusing LIST and GET operations, based on the user query, respond with the most likely action name. If it exists, return {"success": true, "action": "<action>"} where <action> is replaced with the action found. If it does not exist, Last case scenario is return {"success": false, "action": ""}. Output as JSON with JUST the action name."`
+
+		if label == "app_validation" || label == "test" || label == "test_api" {
+			systemMessage = fmt.Sprintf(`Your goal is to find the most correct action for TESTING an API. You have to pick the most likely action. Synonyms are accepted. A synonym example can be something like: case = alert = ticket = issue = task, or message = chat = communication. E.g. for ITSM it should be to list tickets. For email, to list emails. For EDR, to list alerts etc. Find a simple API. The current app is '%s'. Our goal is to get a 200 OK or similar. If it exists, return {"success": true, "action": "<action>"} where <action> is replaced with the action found. If it does not exist, Last case scenario is return {"success": false, "action": ""}. Output as JSON, with JUST the action name.`, app.Name)
+		}
+
 		userMessage := fmt.Sprintf("Out of the following actions, which action matches '%s'?\n", label)
 
 		//changedNames := map[string]string{}
@@ -1744,6 +1750,12 @@ func AutofixAppLabels(app WorkflowApp, label string, keys []string) (WorkflowApp
 				method = "DELETE"
 			} 
 
+			if label == "app_validation" || label == "test" || label == "test_api" {
+				if method != "GET" { 
+					continue
+				}
+			}
+
 			// We need to parse out the url from description to help
 			parsedDescriptionUrlPath := ""
 			for _, line := range strings.Split(action.Description, "\n") {
@@ -1787,7 +1799,7 @@ func AutofixAppLabels(app WorkflowApp, label string, keys []string) (WorkflowApp
 				parsedEnding = ""
 			}
 
-			userMessage += fmt.Sprintf("%s %s\n", action.Name, parsedEnding)
+			userMessage += fmt.Sprintf("- %s %s\n", action.Name, parsedEnding)
 		}
 
 		if len(keys) > 0 {
@@ -1803,7 +1815,24 @@ func AutofixAppLabels(app WorkflowApp, label string, keys []string) (WorkflowApp
 
 		}
 
-		output, err := RunAiQuery(systemMessage, userMessage)
+		chatCompletion := openai.ChatCompletionRequest{
+			Model:     model,
+			Messages:  []openai.ChatCompletionMessage{
+				openai.ChatCompletionMessage{
+					Role:	openai.ChatMessageRoleSystem,
+					Content: systemMessage,
+				},
+				openai.ChatCompletionMessage{
+					Role:    openai.ChatMessageRoleUser,
+					Content: userMessage,
+				},
+			},
+			MaxTokens: maxTokens,
+			Temperature: 0,
+			ReasoningEffort: "low",
+		}
+
+		output, err := RunAiQuery(systemMessage, userMessage, chatCompletion)
 		if err != nil {
 			log.Printf("[ERROR] Failed to run AI query in AutofixAppLabels for app %s (%s): %s", app.Name, app.ID, err)
 			return app, WorkflowAppAction{}
@@ -1819,6 +1848,22 @@ func AutofixAppLabels(app WorkflowApp, label string, keys []string) (WorkflowApp
 			log.Printf("[ERROR] FAILED action mapping parsed output: %s", output)
 		}
 
+		// Strip anything after the first space.
+		if strings.Contains(actionStruct.Action, "(") {
+			// Split and only keep everything based on first space
+			splitAction := strings.Split(actionStruct.Action, "(")
+			newAction := actionStruct.Action
+			if len(splitAction) > 0 {
+				newAction = strings.TrimSpace(splitAction[0])
+			}
+
+			if debug {
+				log.Printf("[DEBUG] Changing action from '%s' to '%s' based on parsing", actionStruct.Action, newAction)
+			}
+
+			actionStruct.Action = newAction
+		}
+
 	}
 
 	if len(actionStruct.Action) == 0 && cacheGeterr == nil {

shared.go

@@ -123,6 +123,7 @@ func HandleCors(resp http.ResponseWriter, request *http.Request) bool {
 
 			// For a new test project
 			"https://cases.shuffler.io",
+			"https://security.shuffler.io",
 			"https://83c56bc8-506d-4dc5-a245-6b57e03ff019.lovableproject.com",
 			"https://id-preview--83c56bc8-506d-4dc5-a245-6b57e03ff019.lovable.app",
 			"https://preview--shuffle-cases.lovable.app",
@@ -249,7 +250,8 @@ func ConstructSessionCookie(value string, expires time.Time) *http.Cookie {
 	if project.Environment == "cloud" {
 		c.Domain = ".shuffler.io"
 		c.Secure = true
-		c.SameSite = http.SameSiteLaxMode
+		//c.SameSite = http.SameSiteLaxMode
+		c.SameSite = http.SameSiteNoneMode
 	}
 
 	return &c
@@ -32148,6 +32150,12 @@ func checkExecutionStatus(ctx context.Context, exec *WorkflowExecution) *Workflo
 	handledAuth := []string{}
 	timenow := time.Now().Unix() * 1000
 	runtimeLocationName := ""
+	for _, action := range exec.Workflow.Actions {
+		if len(action.Environment) > 0 { 
+			runtimeLocationName = action.Environment
+			break
+		}
+	}
 
 	//log.Printf("\n\n[DEBUG][%s] STARTING VALIDATION WITH %d results and %d actions\n\n", exec.ExecutionId, len(exec.Results), len(workflow.Actions))
 	for _, result := range exec.Results {
@@ -32156,7 +32164,9 @@ func checkExecutionStatus(ctx context.Context, exec *WorkflowExecution) *Workflo
 			continue
 		}
 
-		runtimeLocationName = result.Action.Environment
+		if len(runtimeLocationName) == 0 && len(result.Action.Environment) > 0 {
+			runtimeLocationName = result.Action.Environment
+		}
 
 		found := false
 		foundAction := Action{}