fix(assistant): get_thread_context falls through to store.find() for user_message events

Author: srtaalej

slack_bolt/context/get_thread_context/async_get_thread_context.py

@@ -31,14 +31,10 @@ async def __call__(self) -> Optional[AssistantThreadContext]:
         if self.thread_context_loaded is True:
             return self._thread_context
 
-        if self.payload.get("assistant_thread") is not None:
+        thread = self.payload.get("assistant_thread")
+        if thread is not None and thread.get("context", {}).get("channel_id") is not None:
             # assistant_thread_started
-            thread = self.payload["assistant_thread"]
-            self._thread_context = (
-                AssistantThreadContext(thread["context"])
-                if thread.get("context", {}).get("channel_id") is not None
-                else None
-            )
+            self._thread_context = AssistantThreadContext(thread["context"])
             # for this event, the context will never be changed
             self.thread_context_loaded = True
         elif self.payload.get("channel") is not None and self.payload.get("thread_ts") is not None:

slack_bolt/context/get_thread_context/get_thread_context.py

@@ -31,14 +31,10 @@ def __call__(self) -> Optional[AssistantThreadContext]:
         if self.thread_context_loaded is True:
             return self._thread_context
 
-        if self.payload.get("assistant_thread") is not None:
+        thread = self.payload.get("assistant_thread")
+        if thread is not None and thread.get("context", {}).get("channel_id") is not None:
             # assistant_thread_started
-            thread = self.payload["assistant_thread"]
-            self._thread_context = (
-                AssistantThreadContext(thread["context"])
-                if thread.get("context", {}).get("channel_id") is not None
-                else None
-            )
+            self._thread_context = AssistantThreadContext(thread["context"])
             # for this event, the context will never be changed
             self.thread_context_loaded = True
         elif self.payload.get("channel") is not None and self.payload.get("thread_ts") is not None:

tests/scenario_tests/test_events_assistant.py

@@ -186,7 +186,7 @@ def build_payload(event: dict) -> dict:
         "channel": "D111",
         "event_ts": "1726133700.887259",
         "channel_type": "im",
-        "assistant_thread": {"XXX": "YYY"},
+        "assistant_thread": {"action_token": "10647138185092.960436384805.afce3599"},
     }
 )
 

tests/scenario_tests_async/test_events_assistant.py

@@ -200,7 +200,7 @@ def build_payload(event: dict) -> dict:
         "channel": "D111",
         "event_ts": "1726133700.887259",
         "channel_type": "im",
-        "assistant_thread": {"XXX": "YYY"},
+        "assistant_thread": {"action_token": "10647138185092.960436384805.afce3599"},
     }
 )