fix: update imports after file structure refactor

Author: srtaalej

listeners/assistant/assistant.py

@@ -5,7 +5,7 @@
 from slack_sdk import WebClient
 from slack_sdk.errors import SlackApiError
 
-from .llm_caller import call_llm
+from ..llm_caller import call_llm
 
 # Refer to https://tools.slack.dev/bolt-python/concepts/assistant/ for more details
 assistant = Assistant()
@@ -72,13 +72,17 @@ def respond_in_assistant_thread(
             thread_context = get_thread_context()
             referred_channel_id = thread_context.get("channel_id")
             try:
-                channel_history = client.conversations_history(channel=referred_channel_id, limit=50)
+                channel_history = client.conversations_history(
+                    channel=referred_channel_id, limit=50
+                )
             except SlackApiError as e:
                 if e.response["error"] == "not_in_channel":
                     # If this app's bot user is not in the public channel,
                     # we'll try joining the channel and then calling the same API again
                     client.conversations_join(channel=referred_channel_id)
-                    channel_history = client.conversations_history(channel=referred_channel_id, limit=50)
+                    channel_history = client.conversations_history(
+                        channel=referred_channel_id, limit=50
+                    )
                 else:
                     raise e
 

listeners/events/__init__.py

@@ -15,18 +15,25 @@
 def register(app: App):
     app.event("assistant_thread_started")(start_thread_with_suggested_prompts)
     app.event("assistant_thread_context_changed")(save_new_thread_context)
-    app.event("message", matchers=[is_user_message_event_in_assistant_thread])(respond_to_user_message)
+    app.event("message", matchers=[is_user_message_event_in_assistant_thread])(
+        respond_to_user_message
+    )
     app.event("message", matchers=[is_message_event_in_assistant_thread])(just_ack)
 
 
 def is_message_event_in_assistant_thread(body: Dict[str, Any]) -> bool:
     if is_event(body):
-        return body["event"]["type"] == "message" and body["event"].get("channel_type") == "im"
+        return (
+            body["event"]["type"] == "message"
+            and body["event"].get("channel_type") == "im"
+        )
     return False
 
 
 def is_user_message_event_in_assistant_thread(body: Dict[str, Any]) -> bool:
-    return is_message_event_in_assistant_thread(body) and body["event"].get("subtype") in (None, "file_share")
+    return is_message_event_in_assistant_thread(body) and body["event"].get(
+        "subtype"
+    ) in (None, "file_share")
 
 
 def just_ack():

listeners/events/thread_context_store.py

@@ -23,7 +23,10 @@ def _find_parent_message(
     )
     if response.get("messages"):
         for message in response.get("messages"):
-            if message.get("subtype") is None and message.get("user") == context.bot_user_id:
+            if (
+                message.get("subtype") is None
+                and message.get("user") == context.bot_user_id
+            ):
                 return message
 
 
@@ -34,7 +37,9 @@ def get_thread_context(
     channel_id: str,
     thread_ts: str,
 ) -> Optional[dict]:
-    parent_message = _find_parent_message(context=context, client=client, channel_id=channel_id, thread_ts=thread_ts)
+    parent_message = _find_parent_message(
+        context=context, client=client, channel_id=channel_id, thread_ts=thread_ts
+    )
     if parent_message is not None and parent_message.get("metadata") is not None:
         return parent_message["metadata"]["event_payload"]