Add OpenWebUI bot integration for channels feature

2025-05-21 09:07:10 +01:00
parent 1c75eece93
commit 4ee0da8db6
8 changed files with 760 additions and 154 deletions
@@ -0,0 +1,101 @@
+#!/usr/bin/env python3
+"""
+Script to send a test message to an OpenWebUI channel.
+"""
+
+import asyncio
+import aiohttp
+import json
+import sys
+
+# Configuration
+OPENWEBUI_URL = "http://104.225.217.215:8080"
+API_KEY = "GdCU4ieYDqHsLfH2"
+
+async def create_channel():
+    """Create a test channel."""
+    print("Creating a test channel...")
+    
+    async with aiohttp.ClientSession() as session:
+        headers = {"Authorization": f"Bearer {API_KEY}"}
+        channel_data = {
+            "name": "Bot Test Channel",
+            "description": "Channel for testing the bot"
+        }
+        
+        try:
+            async with session.post(
+                f"{OPENWEBUI_URL}/api/channels",
+                headers=headers,
+                json=channel_data
+            ) as response:
+                if response.status == 200:
+                    channel = await response.json()
+                    channel_id = channel["id"]
+                    print(f"Created channel with ID: {channel_id}")
+                    return channel_id
+                else:
+                    print(f"Failed to create channel: {response.status} - {await response.text()}")
+                    return None
+        except Exception as e:
+            print(f"Error creating channel: {str(e)}")
+            return None
+
+async def send_message(channel_id, message):
+    """Send a message to a channel."""
+    print(f"Sending message to channel {channel_id}: {message}")
+    
+    async with aiohttp.ClientSession() as session:
+        headers = {"Authorization": f"Bearer {API_KEY}"}
+        message_data = {
+            "content": message
+        }
+        
+        try:
+            async with session.post(
+                f"{OPENWEBUI_URL}/api/v1/channels/{channel_id}/messages/post",
+                headers=headers,
+                json=message_data
+            ) as response:
+                if response.status == 200:
+                    print("Message sent successfully!")
+                    return True
+                else:
+                    print(f"Failed to send message: {response.status} - {await response.text()}")
+                    return False
+        except Exception as e:
+            print(f"Error sending message: {str(e)}")
+            return False
+
+async def main():
+    """Main function."""
+    # Create a channel
+    channel_id = await create_channel()
+    if not channel_id:
+        print("Failed to create a channel. Exiting.")
+        return False
+    
+    # Send a test message
+    success = await send_message(channel_id, "@ai Hello, are you working?")
+    if success:
+        print(f"Test complete! Check the channel at {OPENWEBUI_URL}/channels/{channel_id}")
+        return True
+    else:
+        print("Failed to send test message.")
+        return False
+
+if __name__ == "__main__":
+    try:
+        result = asyncio.run(main())
+        if result:
+            print("Test successful!")
+            sys.exit(0)
+        else:
+            print("Test failed!")
+            sys.exit(1)
+    except KeyboardInterrupt:
+        print("Test interrupted by user.")
+        sys.exit(130)
+    except Exception as e:
+        print(f"Unexpected error: {str(e)}")
+        sys.exit(1)