Refactor slack setup

Signed-off-by: Di Wang <[email protected]>

.env.example

@@ -3,4 +3,7 @@ OLLAMA_CHAT_MODEL=mistral
 OLLAMA_URL=http://host.docker.internal:11434
 OPENAI_API_KEY=
 OPENAI_CHAT_MODEL='gpt-3.5-turbo-0125'
+SLACK_APP_TOKEN=
+SLACK_BOT_TOKEN=
+SLACK_SIGNING_SECRET=
 VERBOSE=0

README.md

@@ -43,3 +43,9 @@ langchain serve
 ```shell
 docker compose up --build
 ```
+
+## Start Slack socket mode
+
+```shell
+slack_socket_mode
+```

app/config.py

@@ -6,10 +6,13 @@
 load_dotenv()
 
 DEBUG = os.getenv("DEBUG", "0") == "1"
-VERBOSE = os.getenv("VERBOSE", "0") == "1"
 OLLAMA_CHAT_MODEL = os.getenv("OLLAMA_CHAT_MODEL", "mistral")
 OLLAMA_URL = os.getenv("OLLAMA_URL", "http://localhost:11434")
 OPENAI_CHAT_MODEL = os.getenv("OPENAI_CHAT_MODEL", "gpt-3.5-turbo")
+SLACK_APP_TOKEN = os.getenv("SLACK_APP_TOKEN")
+SLACK_BOT_TOKEN = os.getenv("SLACK_BOT_TOKEN")
+SLACK_SIGNING_SECRET = os.getenv("SLACK_SIGNING_SECRET")
+VERBOSE = os.getenv("VERBOSE", "0") == "1"
 
 set_debug(DEBUG)
 set_verbose(VERBOSE)

app/dependencies/slack.py

@@ -0,0 +1,38 @@
+from typing import Annotated
+
+from fastapi import Depends
+from slack_bolt import App
+from slack_bolt.adapter.fastapi import SlackRequestHandler
+from slack_bolt.adapter.socket_mode import SocketModeHandler
+
+from app import config
+
+app = App(
+    token=config.SLACK_BOT_TOKEN,
+    signing_secret=config.SLACK_SIGNING_SECRET
+)
+
+slack_request_handler = SlackRequestHandler(app)
+
+slack_socket_model_handler = SocketModeHandler(app, config.SLACK_APP_TOKEN)
+
+
+def get_slack_request_handler() -> SlackRequestHandler:
+    return slack_request_handler
+
+
+SlackRequestHandlerDep = Annotated[SlackRequestHandler, Depends(get_slack_request_handler)]
+
+@app.event("message")
+def handle_message_events(body, say, logger):
+    logger.info(body)
+    text = body.get('event', {}).get('text', '')
+    if 'hello' in text.lower():
+        say("Hello there! :wave:")
+
+
+# Define an event listener for "app_mention" events
+@app.event("app_mention")
+def handle_app_mention_events(body, say, logger):
+    logger.info(body)
+    say("Hello SRE slackers! :blush:")

app/routers/slack.py

@@ -0,0 +1,10 @@
+from fastapi import APIRouter, Request
+
+from app.dependencies.slack import SlackRequestHandlerDep
+
+router = APIRouter()
+
+
+@router.post("/slack/events")
+async def slack_events(request: Request, handler: SlackRequestHandlerDep):
+    return await handler.handle(request)

app/server.py

@@ -6,6 +6,7 @@
 from app.chains.extraction import extraction_chain
 from app.chains.supervisor import build_supervisor_chain
 from app.dependencies.ollama_chat_model import ollama_chat_model
+from app.routers import slack
 
 app = FastAPI()
 
@@ -39,6 +40,8 @@ async def redirect_root_to_docs():
     path="/supervisor",
 )
 
+app.include_router(slack.router)
+
 if __name__ == "__main__":
     import uvicorn
 

app/slack_socket_mode.py

@@ -0,0 +1,15 @@
+#!/usr/bin/env python3
+
+from app import config
+from app.dependencies.slack import slack_socket_model_handler
+
+
+def run():
+    if config.SLACK_APP_TOKEN:
+        slack_socket_model_handler.start()
+    else:
+        print("SLACK_APP_TOKEN not found.")
+
+
+if __name__ == '__main__':
+    run()

pyproject.toml

@@ -24,12 +24,16 @@ langserve = {extras = ["server"], version = ">=0.0.41"}
 pydantic = "<2"
 python = "^3.11"
 python-dotenv = "^1.0.1"
+slack-bolt = "^1.18.1"
 uvicorn = "^0.23.2"
 
 [tool.poetry.group.dev.dependencies]
 langchain-cli = ">=0.0.15"
 ruff = "^0.2.1"
 
+[tool.poetry.scripts]
+slack_socket_mode = "app.slack_socket_mode:run"
+
 [build-system]
 requires = ["poetry-core"]
 build-backend = "poetry.core.masonry.api"