chore: use context.user_token instead of manual query

Author: mwbrooks

pydantic-ai/listeners/events/app_home_opened.py

@@ -16,14 +16,7 @@ def handle_app_home_opened(client: WebClient, context: BoltContext, logger: Logg
         is_connected = False
 
         if os.environ.get("SLACK_CLIENT_ID"):
-            from oauth import installation_store
-
-            installation = installation_store.find_installation(
-                enterprise_id=context.enterprise_id or "",
-                team_id=context.team_id or "",
-                user_id=user_id,
-            )
-            if installation and installation.user_token:
+            if context.user_token:
                 is_connected = True
             else:
                 redirect_uri = os.environ.get("SLACK_REDIRECT_URI", "")

pydantic-ai/listeners/events/app_mentioned.py

@@ -7,7 +7,6 @@
 from agent import CaseyDeps, run_casey
 from thread_context import conversation_store
 from listeners.views.feedback_builder import build_feedback_blocks
-from oauth import installation_store
 
 
 def handle_app_mentioned(
@@ -59,25 +58,14 @@ def handle_app_mentioned(
         # Get conversation history
         history = conversation_store.get_history(channel_id, thread_ts)
 
-        # Look up the user token directly from the installation store rather
-        # than context.user_token. Bolt's InstallationStoreAuthorize may fail
-        # to resolve the user token when installer-latest is missing or the
-        # requesting user differs from the original installer.
-        installation = installation_store.find_installation(
-            enterprise_id=context.enterprise_id,
-            team_id=context.team_id,
-            user_id=user_id,
-        )
-        user_token = installation.user_token if installation else None
-
         # Run the agent
         deps = CaseyDeps(
             client=client,
             user_id=user_id,
             channel_id=channel_id,
             thread_ts=thread_ts,
             message_ts=event["ts"],
-            user_token=user_token,
+            user_token=context.user_token,
         )
         result = run_casey(cleaned_text, deps, message_history=history)
 

pydantic-ai/listeners/events/message.py

@@ -6,7 +6,6 @@
 from agent import CaseyDeps, run_casey
 from thread_context import conversation_store
 from listeners.views.feedback_builder import build_feedback_blocks
-from oauth import installation_store
 
 
 def handle_message(
@@ -82,25 +81,14 @@ def handle_message(
             ],
         )
 
-        # Look up the user token directly from the installation store rather
-        # than context.user_token. Bolt's InstallationStoreAuthorize may fail
-        # to resolve the user token when installer-latest is missing or the
-        # requesting user differs from the original installer.
-        installation = installation_store.find_installation(
-            enterprise_id=context.enterprise_id,
-            team_id=context.team_id,
-            user_id=user_id,
-        )
-        user_token = installation.user_token if installation else None
-
         # Run the agent
         deps = CaseyDeps(
             client=client,
             user_id=user_id,
             channel_id=channel_id,
             thread_ts=thread_ts,
             message_ts=event["ts"],
-            user_token=user_token,
+            user_token=context.user_token,
         )
         result = run_casey(text, deps, message_history=history)