Skip to content

Commit 609d8f2

Browse files
committed
Fixed bug in send_finding_slack
1 parent 38c8ce4 commit 609d8f2

4 files changed

Lines changed: 8 additions & 10 deletions

File tree

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

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -113,12 +113,11 @@ def ask_holmes(event: ExecutionBaseEvent, params: AIInvestigateParams):
113113
[HolmesResultsBlock(holmes_result=holmes_result)], enrichment_type=EnrichmentType.ai_analysis
114114
)
115115
runner_context = getattr(params, "robusta_context", None)
116-
if runner_context and "thread_ts" in runner_context:
117-
original_thread_ts = runner_context.get("thread_ts")
118-
original_channel_id = runner_context.get("channel_id")
119-
if original_thread_ts:
120-
finding.robusta_context["thread_ts"] = original_thread_ts
121-
finding.robusta_context["channel_id"] = original_channel_id
116+
if runner_context:
117+
if "thread_ts" in runner_context:
118+
finding.robusta_context["thread_ts"] = runner_context.get("thread_ts")
119+
if "channel_id" in runner_context:
120+
finding.robusta_context["channel_id"] = runner_context.get("channel_id")
122121

123122
event.add_finding(finding)
124123

src/robusta/core/reporting/base.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,6 @@
99
from typing import Any, Dict, List, Optional, Union
1010
from urllib.parse import urlencode
1111

12-
from pydantic import Field
1312
from pydantic.main import BaseModel
1413
from strenum import StrEnum
1514

src/robusta/core/reporting/callbacks.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
from robusta.core.playbooks.actions_registry import Action
66
from robusta.core.reporting import CallbackChoice
77
from robusta.core.reporting.action_requests import ActionRequestBody, ExternalActionRequest, sign_action_request
8-
import logging
8+
99

1010
class ExternalActionRequestBuilder(BaseModel):
1111
@classmethod

src/robusta/integrations/slack/sender.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -603,13 +603,13 @@ def send_finding_to_slack(
603603
blocks: List[BaseBlock] = []
604604
attachment_blocks: List[BaseBlock] = []
605605

606-
channel_id, thread_ts = self._resolve_slack_thread(
606+
slack_channel, thread_ts = self._resolve_slack_thread(
607607
finding, sink_params, thread_ts
608608
)
609609

610610
if finding.finding_type == FindingType.AI_ANALYSIS:
611611
# holmes analysis message needs special handling
612-
self.send_holmes_analysis(finding, channel_id, platform_enabled, thread_ts)
612+
self.send_holmes_analysis(finding, slack_channel, platform_enabled, thread_ts)
613613
return "" # [arik] Looks like the return value here is not used, needs to be removed
614614

615615
status: FindingStatus = (

0 commit comments

Comments
 (0)