Slackbot + Open WebUI MVP Integration

bot.py

#!/usr/bin/env python3

import os
from collections.abc import Callable
from slack_bolt import App
from slack_bolt.adapter.socket_mode import SocketModeHandler
from dotenv import load_dotenv
from openwebui_client.client import OpenWebUIClient

load_dotenv()

class InteractiveBot:

    def __init__(self):
        self.app = App(token=os.environ["SLACK_BOT_TOKEN"])
        self.app.event("message")(self.handle_message)
        self.llm = OpenWebUIClient(
            base_url=os.environ["OUI_BASE_URL"],
            api_key=os.environ["OUI_API_KEY"],
        )

    def handle_message(self, body: dict, say: Callable):
        print(f"Received body: {body}")
        text = body.get("event", {}).get("text", "")
        if text:
            try:
                # Call the OpenWebUIClient for chat completion (non-streaming)
                response = self.llm.chat.completions.create(
                    messages=[{"role": "user", "content": text}],
                    model=self.llm.default_model or "gpt-4o-mini",
                    stream=False
                )
                # Extract the response text (adjust as per your client’s response structure)
                reply = response.choices[0].message.content if response.choices else "Sorry, I couldn't generate a response."
            except Exception as e:
                reply = f"Error: {e}"
            say(reply)

    @staticmethod
    def main():
        bot = InteractiveBot()
        SocketModeHandler(bot.app, os.environ["SLACK_APP_TOKEN"]).start()

if __name__ == "__main__":
    InteractiveBot.main()