Merge branch 'main' of github.com:Shuffle/shuffle-shared

Author: 0x0elliot

ai.go

@@ -426,7 +426,7 @@ func RunKmsTranslation(ctx context.Context, fullBody []byte, authConfig, paramNa
 	}
 
 	// Added a filename_prefix to know which field each belongs to
-	schemalessOutput, err := schemaless.Translate(ctx, "get_kms_key", marshalledBody, authConfig, fmt.Sprintf("filename_prefix:%s-", paramName))
+	schemalessOutput, _, err := schemaless.Translate(ctx, "get_kms_key", marshalledBody, authConfig, fmt.Sprintf("filename_prefix:%s-", paramName))
 	if err != nil {
 		log.Printf("[ERROR] Failed to translate KMS response (2): %s", err)
 		return string(fullBody), err
@@ -8944,11 +8944,30 @@ func RunAiQuery(systemMessage, userMessage string, incomingRequest ...openai.Cha
 		if len(chatCompletion.Messages) == 0 {
 			return "", errors.New("No messages to send to OpenAI. Pass systemmessage, usermessage")
 		}
+	}
+
+	// Gotta cut it back down, as it can get into 50+ etc
+	if len(chatCompletion.Messages) > 10 {
+		// Ensures we keep the system messages
+		newMessages := []openai.ChatCompletionMessage{
+			chatCompletion.Messages[0],
+			chatCompletion.Messages[1],
+			chatCompletion.Messages[2],
+		}
+
+		for messageIndex, chat := range chatCompletion.Messages {
+			if messageIndex > len(chatCompletion.Messages)-7 {
+				newMessages = append(newMessages, chat)
+			}
+		}
 
-		//log.Printf("\n\n\nGot %d messages in chat completion (%s)\n\n\n", len(chatCompletion.Messages), cachedChat)
+		if len(newMessages) > 5 { 
+			chatCompletion.Messages = newMessages
+		}
 	}
 
 	if debug {
+
 		log.Printf("\n\n[DEBUG] Chatcompletion messages: %d\n\n", len(chatCompletion.Messages))
 	}
 

shared.go

@@ -33352,7 +33352,7 @@ func checkExecutionStatus(ctx context.Context, exec *WorkflowExecution) *Workflo
 	IncrementCache(ctx, exec.ExecutionOrg, "app_executions", amountFinished)
 
 	go RunCacheCleanup(ctx, *exec)
-	go RunIOCFinder(ctx, *exec)
+	//go RunIOCFinder(ctx, *exec)
 
 	//log.Printf("[DEBUG][%s] Running status fixing for workflow %#v to see if auth + workflow(s) are functional. Results: %d", exec.ExecutionId, exec.Workflow.ID, len(exec.Results))
 	orgId := exec.ExecutionOrg