working version

Author: Avi-Robusta

src/robusta/core/model/runner_config.py

@@ -18,6 +18,7 @@
 from robusta.core.sinks.rocketchat.rocketchat_sink_params import RocketchatSinkConfigWrapper
 from robusta.core.sinks.servicenow.servicenow_sink_params import ServiceNowSinkConfigWrapper
 from robusta.core.sinks.slack.slack_sink_params import SlackSinkConfigWrapper
+from robusta.core.sinks.slack.preview.slack_sink_preview_params import SlackSinkPreviewConfigWrapper
 from robusta.core.sinks.mail.mail_sink_params import MailSinkConfigWrapper
 from robusta.core.sinks.telegram.telegram_sink_params import TelegramSinkConfigWrapper
 from robusta.core.sinks.victorops.victorops_sink_params import VictoropsConfigWrapper
@@ -55,6 +56,7 @@ class RunnerConfig(BaseModel):
             Union[
                 RobustaSinkConfigWrapper,
                 SlackSinkConfigWrapper,
+                SlackSinkPreviewConfigWrapper,
                 DataDogSinkConfigWrapper,
                 KafkaSinkConfigWrapper,
                 MsTeamsSinkConfigWrapper,

src/robusta/core/sinks/sink_factory.py

@@ -22,7 +22,8 @@
 from robusta.core.sinks.sink_base import SinkBase
 from robusta.core.sinks.sink_config import SinkConfigBase
 from robusta.core.sinks.slack import SlackSink, SlackSinkConfigWrapper
-from robusta.core.sinks.slack.preview import SlackSinkPreview, SlackSinkPreviewConfigWrapper
+from robusta.core.sinks.slack.preview.slack_sink_preview_params import SlackSinkPreviewConfigWrapper
+from robusta.core.sinks.slack.preview.slack_sink_preview import SlackSinkPreview
 from robusta.core.sinks.telegram import TelegramSink, TelegramSinkConfigWrapper
 from robusta.core.sinks.victorops import VictoropsConfigWrapper, VictoropsSink
 from robusta.core.sinks.webex import WebexSink, WebexSinkConfigWrapper

src/robusta/core/sinks/slack/__init__.py

@@ -1,2 +1,4 @@
+from robusta.core.sinks.slack.slack_sink_params import SlackSinkParams, SlackSinkConfigWrapper
 from robusta.core.sinks.slack.slack_sink import SlackSink
-from robusta.core.sinks.slack.slack_sink_params import SlackSinkConfigWrapper, SlackSinkParams
+
+__all__ = ["SlackSink", "SlackSinkParams", "SlackSinkConfigWrapper"]

src/robusta/core/sinks/slack/preview/__init__.py

@@ -1,11 +1,10 @@
 from robusta.core.sinks.slack.preview.slack_sink_preview_params import SlackSinkPreviewConfigWrapper, SlackSinkPreviewParams
 from robusta.core.sinks.slack.slack_sink import SlackSink
-from robusta.integrations import slack as slack_module
 
 
 class SlackSinkPreview(SlackSink):
     params: SlackSinkPreviewParams
 
     def __init__(self, sink_config: SlackSinkPreviewConfigWrapper, registry):
-        super().__init__(sink_config.slack_sink_preview, registry, is_preview=True)
+        super().__init__(sink_config, registry, is_preview=True)
 

src/robusta/core/sinks/slack/preview/slack_sink_preview.py

@@ -1,6 +1,6 @@
 from robusta.core.sinks.sink_base_params import SinkBaseParams
 from robusta.core.sinks.sink_config import SinkConfigBase
-from robusta.core.sinks.slack import SlackSinkParams
+from robusta.core.sinks.slack.slack_sink_params import SlackSinkParams
 from enum import Enum
 from typing import Optional, Dict
 

src/robusta/core/sinks/slack/preview/slack_sink_preview_params.py

@@ -11,9 +11,10 @@ class SlackSink(SinkBase):
     params: SlackSinkParams
 
     def __init__(self, sink_config: SlackSinkConfigWrapper, registry, is_preview=False):
-        super().__init__(sink_config.slack_sink, registry)
-        self.slack_channel = sink_config.slack_sink.slack_channel
-        self.api_key = sink_config.slack_sink.api_key
+        slack_sink_params = sink_config.get_params()
+        super().__init__(slack_sink_params, registry)
+        self.slack_channel = slack_sink_params.slack_channel
+        self.api_key = slack_sink_params.api_key
         self.slack_sender = slack_module.SlackSender(
             self.api_key, self.account_id, self.cluster_name, self.signing_key, self.slack_channel, is_preview
         )