Refactored the way variables pass to ask_ai to get to slack channels

Author: itisallgood

src/robusta/core/playbooks/internal/ai_integration.py

@@ -115,8 +115,10 @@ def ask_holmes(event: ExecutionBaseEvent, params: AIInvestigateParams):
             runner_context = getattr(params, "robusta_context", None) # Safely get the context dict
             if runner_context and "thread_ts" in runner_context:
                 original_thread_ts = runner_context.get("thread_ts")
+                original_channel_id = runner_context.get("channel_id")
                 if original_thread_ts:                    
                     finding.robusta_context["thread_ts"] = original_thread_ts
+                    finding.robusta_context["channel_id"] = original_channel_id
                     logging.info(f"Added message_ts={original_thread_ts} to finding {finding.id} annotations.")
                 else:
                     logging.warning(f"message_ts found in robusta_context for finding {finding.id} but it is empty.")

src/robusta/integrations/receiver.py

@@ -178,7 +178,8 @@ def __exec_external_request(self, action_request: ExternalActionRequest, validat
 
         if action_request.slack_container and action_request.slack_container.message_ts:
             thread_ts = action_request.slack_container.message_ts
-            action_request.body.action_params["robusta_context"] = {"thread_ts": thread_ts}
+            channel_id = action_request.slack_container.channel_id
+            action_request.body.action_params["robusta_context"] = {"thread_ts": thread_ts, "channel_id": channel_id}
 
         response = self.event_handler.run_external_action(
             action_request.body.action_name,

src/robusta/integrations/slack/sender.py

@@ -589,16 +589,16 @@ def send_finding_to_slack(
         )
         robusta_context = getattr(finding, "robusta_context", None)
         effective_thread_ts = thread_ts # Default to the one passed in (e.g., from grouping)
+        effective_channel_id = slack_channel
         if robusta_context:
             annotation_ts = robusta_context.get("thread_ts")
+            channel_id = robusta_context.get("channel_id")
             if annotation_ts: # Make sure it's not None or empty
                 effective_thread_ts = annotation_ts # Prioritize the annotation!
-                logging.info(f"Using thread_ts from annotation for AI analysis finding {finding.id}: {effective_thread_ts}")
-            else:
-                logging.warning(f"Found empty message_ts annotation for AI analysis finding {finding.id}, using original thread_ts: {thread_ts}")
+                effective_channel_id = channel_id
         if finding.finding_type == FindingType.AI_ANALYSIS:
             # holmes analysis message needs special handling
-            self.send_holmes_analysis(finding, slack_channel, platform_enabled, effective_thread_ts)
+            self.send_holmes_analysis(finding, effective_channel_id, platform_enabled, effective_thread_ts)
             return ""  # [arik] Looks like the return value here is not used, needs to be removed
 
         status: FindingStatus = (