A few more minor agent fixes

Author: frikky

ai.go

@@ -8636,6 +8636,13 @@ You are the Action Execution Agent for the Shuffle platform. You receive tools (
 							Result: string(marshalledDecision),
 						}
 
+						for _, action := range execution.Workflow.Actions {
+							if action.ID == actionResult.Action.ID {
+								actionResult.Action = action
+								break
+							}
+						}
+
 						// This is required as the result for the agent isn't set yet on the first run. Minor delay to wait up a bit
 						if decisionIndex == 0 {
 							go func() {

cloudSync.go

@@ -2368,6 +2368,13 @@ func RunAgentDecisionAction(execution WorkflowExecution, agentOutput AgentOutput
 		Result: string(marshalledDecision),
 	}
 
+	for _, action := range execution.Workflow.Actions {
+		if action.ID == parsedAction.Action.ID {
+			parsedAction.Action = action
+			break
+		}
+	}
+
 	marshalledAction, err := json.Marshal(parsedAction)
 	if err != nil {
 		log.Printf("[ERROR][%s] AI Agent: Failed marshalling action in agent decision: %s", execution.ExecutionId, err)

db-connector.go

@@ -1999,6 +1999,11 @@ func Fixexecution(ctx context.Context, workflowExecution WorkflowExecution) (Wor
 		workflowExecution.Workflow.Actions[actionIndex].LargeImage = ""
 		workflowExecution.Workflow.Actions[actionIndex].SmallImage = ""
 		for resultIndex, innerresult := range workflowExecution.Results {
+			// There was some WAITING issue here. This is a hotfix from agent issues.
+			if innerresult.Status == "WAITING" && innerresult.Action.AppName == "Shuffle Tools" && innerresult.CompletedAt > 0 { 
+				workflowExecution.Results[resultIndex].Status = "SUCCESS"
+			}
+
 			if innerresult.Action.ID != action.ID {
 				continue
 			}

shared.go

@@ -22511,7 +22511,7 @@ func PrepareSingleAction(ctx context.Context, user User, appId string, body []by
 					})
 				}
 
-				log.Printf("[AUDIT] Injected system AI credentials (fallback) for org %s", user.ActiveOrg.Id)
+				//log.Printf("[AUDIT] Injected system AI credentials (fallback) for org %s", user.ActiveOrg.Id)
 
 				// Mapping to internal so the execution itself is not referencable
 				if project.Environment == "cloud" {