[WIP] Add search summary function

README.md

@@ -45,7 +45,8 @@ After the first deployment, properly configure the settings below. Follow [the o
   - `app_mention`
   - `message.channels`
   - `file_shared`
-- OAuth scops
+- OAuth scopes
   - `chat:write`
   - `files:write`
   - `files:read`
+  - `search:read`

app/.chalice/config.template.json

@@ -8,6 +8,7 @@
       "automatic_layer": true,
       "environment_variables": {
         "SLACK_BOT_TOKEN": "{{ SLACK_BOT_TOKEN_DEV }}",
+        "SLACK_USER_TOKEN": "{{ SLACK_USER_TOKEN_DEV }}",
         "SLACK_SIGNING_SECRET": "{{ SLACK_SIGNING_SECRET_DEV }}",
         "OPENAI_ORGANIZATION": "{{ OPENAI_ORGANIZATION }}",
         "OPENAI_API_KEY": "{{ OPENAI_API_KEY }}",
@@ -20,6 +21,7 @@
       "automatic_layer": true,
       "environment_variables": {
         "SLACK_BOT_TOKEN": "{{ SLACK_BOT_TOKEN }}",
+        "SLACK_USER_TOKEN": "{{ SLACK_USER_TOKEN }}",
         "SLACK_SIGNING_SECRET": "{{ SLACK_SIGNING_SECRET }}",
         "OPENAI_ORGANIZATION": "{{ OPENAI_ORGANIZATION }}",
         "OPENAI_API_KEY": "{{ OPENAI_API_KEY }}",

app/chalicelib/service.py

@@ -1,8 +1,9 @@
 # Services are responsible for generating the reply message and additional content (e.g. images).
 # They must be easily testable and shoud not depend on Slack SDK.
 
+import json
 import tempfile
-from typing import Optional
+from typing import Optional, Sequence
 
 import langchain
 from langchain import LLMChain, PromptTemplate
@@ -162,3 +163,9 @@ def generate_image_variation(image_file: str) -> tuple[str, Optional[list[str]]]
             f.write(image)
         image_files.append(image_file)
     return reply, image_files
+
+
+def summarize_slack_messages(messages: Sequence[dict]) -> str:
+    # TODO: implement it
+    reply = "```\n" + json.dumps(messages, indent=2, ensure_ascii=False) + "\n```"
+    return reply

app/chalicelib/slack/__init__.py

@@ -12,22 +12,27 @@
     command_help,
     command_image,
     command_image_variation,
+    command_slacksearch,
     command_text,
     command_textedit,
     command_textinsert,
     parse,
 )
 from .matcher import match_file_share, match_message_replied
 
+SLACK_BOT_TOKEN = os.environ.get("SLACK_BOT_TOKEN")
+SLACK_USER_TOKEN = os.environ.get("SLACK_USER_TOKEN")
+SLACK_SIGNING_SECRET = os.environ.get("SLACK_SIGNING_SECRET")
+
 bolt_app = App(
-    token=os.environ.get("SLACK_BOT_TOKEN"),
-    signing_secret=os.environ.get("SLACK_SIGNING_SECRET"),
+    token=SLACK_BOT_TOKEN,
+    signing_secret=SLACK_SIGNING_SECRET,
     process_before_response=True,
 )
 
 
 @bolt_app.event("app_mention")
-def reply_mention(event, say):
+def reply_mention(event, context, say):
     ts = event["ts"]
     channel = event["channel"]
     user = event["user"]
@@ -78,6 +83,18 @@ def reply_mention(event, say):
         )
         return
 
+    if command == "slacksearch":
+        command_slacksearch(
+            query=args[0],
+            count=10,
+            client=bolt_app.client,
+            user=user,
+            channel=channel,
+            say=say,
+            user_token=SLACK_USER_TOKEN,
+        )
+        return
+
     # must be unreachable
     reply = "私、バグっているみたいです！"
     say(f"<@{user}> {reply}")

app/chalicelib/slack/command.py

@@ -21,6 +21,7 @@
     generate_text_completion,
     generate_text_edit,
     generate_text_insertion,
+    summarize_slack_messages,
 )
 from ..util import download_file
 
@@ -34,6 +35,7 @@
     "codeedit",
     "codeinsert",
     "image",
+    "slacksearch",
 )
 
 Command = Literal[
@@ -46,6 +48,7 @@
     "codeedit",
     "codeinsert",
     "image",
+    "slacksearch",
 ]
 Args = Sequence[str]
 
@@ -108,7 +111,8 @@ def command_help(say: Say):
         "codeedit: コードを編集します\n"
         "codeinsert: コードを挿入します\n"
         "image: 文章から画像を生成します\n"
-        "画像アップロード: 画像のバリエーションを生成します"
+        "画像アップロード: 画像のバリエーションを生成します\n"
+        "slacksearch: Slackの検索結果を要約します\n"
         "```"
     )
 
@@ -205,3 +209,32 @@ def command_image_variation(
                 os.remove(file)
         else:
             say(f"<@{user}> {reply}")
+
+
+def command_slacksearch(
+    query: str,
+    count: int,
+    client: WebClient,
+    user: str,
+    channel: str,
+    say: Say,
+    user_token: str,
+):
+    resp = client.search_messages(
+        query=query,
+        count=count,
+        token=user_token,
+    )
+    messages = [
+        {
+            "channel": m["channel"]["id"],
+            "channel_name": m["channel"]["name"],
+            "user": m["user"],
+            "text": m["text"],
+            "url": m["permalink"],
+        }
+        for m in resp["messages"]["matches"]
+        if m["channel"]["is_private"] is False
+    ]
+    reply = summarize_slack_messages(messages)
+    say(f"<@{user}> {reply}")

deploy.sh

@@ -8,8 +8,10 @@ stage="${1:-dev}"  # dev or prod
 poetry export -f requirements.txt --output requirements.txt --only main
 jinja2 .chalice/config.template.json \
     -D SLACK_BOT_TOKEN="${SLACK_BOT_TOKEN}" \
+    -D SLACK_USER_TOKEN="${SLACK_USER_TOKEN}" \
     -D SLACK_SIGNING_SECRET="${SLACK_SIGNING_SECRET}" \
     -D SLACK_BOT_TOKEN_DEV="${SLACK_BOT_TOKEN_DEV}" \
+    -D SLACK_USER_TOKEN_DEV="${SLACK_USER_TOKEN_DEV}" \
     -D SLACK_SIGNING_SECRET_DEV="${SLACK_SIGNING_SECRET_DEV}" \
     -D OPENAI_ORGANIZATION="${OPENAI_ORGANIZATION}" \
     -D OPENAI_API_KEY="${OPENAI_API_KEY}" \