Change rocket.chat admin logic
This commit is contained in:
ChenKaiLiuG
180
exchange.md
180
exchange.md
@@ -109,4 +109,182 @@ CMD ["streamlit", "run", "admin.py", "--server.port=8501", "--server.address=0.0
|
||||
2. 伺服器資源不足，無法負擔 Rocket.Chat + MongoDB。
|
||||
3. 需要針對特定業務邏輯（如：查看用戶餘額、審核特定關鍵字）進行客製化開發。
|
||||
|
||||
## Rocket.chat實作紀錄
|
||||
## Rocket.Chat 實作紀錄
|
||||
|
||||
### 實作日期
|
||||
2026年2月1日
|
||||
|
||||
### 系統架構
|
||||
|
||||
```
|
||||
┌─────────────────┐ ┌──────────────────┐ ┌─────────────────┐
|
||||
│ Open WebUI │────▶│ API 轉接層 │────▶│ Rocket.Chat │
|
||||
│ (Port 10060) │ │ (Port 18000) │ │ (Port 13000) │
|
||||
│ 用戶前端 │◀────│ FastAPI │◀────│ 管理員後台 │
|
||||
└─────────────────┘ └──────────────────┘ └─────────────────┘
|
||||
│ │
|
||||
▼ ▼
|
||||
┌──────────────────┐ ┌─────────────────┐
|
||||
│ PostgreSQL │ │ MongoDB │
|
||||
│ (Port 5432) │ │ (Port 27017) │
|
||||
└──────────────────┘ └─────────────────┘
|
||||
```
|
||||
|
||||
### 工作流程
|
||||
|
||||
1. **用戶發送訊息** → Open WebUI 呼叫 `/v1/chat/completions`
|
||||
2. **API 轉接層接收** → 提取用戶資訊 (user_name, chat_id)
|
||||
3. **創建 Rocket.Chat 頻道** → 頻道名稱: `{用戶名}-{對話ID前8位}`
|
||||
4. **發送用戶訊息** → 顯示格式: `💬 用戶名: 訊息內容`
|
||||
5. **輪詢等待回覆** → 每 2 秒檢查是否有管理員新訊息
|
||||
6. **回傳給用戶** → 以 OpenAI 格式回傳管理員的回覆
|
||||
|
||||
### 已完成的修改
|
||||
|
||||
#### 1. docker-compose.yml
|
||||
|
||||
**新增服務：**
|
||||
- `mongo`: MongoDB 7.0 (Rocket.Chat 8.0.1 需要 7.0+)
|
||||
- `mongo-init-replica`: 初始化 MongoDB Replica Set
|
||||
- `rocketchat`: Rocket.Chat 8.0.1
|
||||
|
||||
**修改服務：**
|
||||
- `openwebui`: 新增 `ENABLE_FORWARD_USER_INFO_HEADERS=True` 環境變數
|
||||
|
||||
**關鍵配置：**
|
||||
```yaml
|
||||
mongo:
|
||||
image: mongo:7.0 # 從 5.0 升級，因為 Rocket.Chat 8.0.1 不支援 5.0
|
||||
command: mongod --oplogSize 128 --replSet rs0
|
||||
|
||||
rocketchat:
|
||||
image: registry.rocket.chat/rocketchat/rocket.chat:latest
|
||||
ports:
|
||||
- "13000:3000"
|
||||
environment:
|
||||
MONGO_URL: mongodb://mongo:27017/rocketchat?replicaSet=rs0
|
||||
MONGO_OPLOG_URL: mongodb://mongo:27017/local?replicaSet=rs0
|
||||
|
||||
openwebui:
|
||||
environment:
|
||||
- ENABLE_FORWARD_USER_INFO_HEADERS=True # 轉發用戶資訊到 API
|
||||
```
|
||||
|
||||
#### 2. api/server.py
|
||||
|
||||
**新增 imports：**
|
||||
```python
|
||||
import json
|
||||
import hashlib
|
||||
```
|
||||
|
||||
**核心函數重寫：**
|
||||
|
||||
| 函數名稱 | 功能說明 |
|
||||
|---------|---------|
|
||||
| `rocketchat_login()` | 使用帳密登入 Rocket.Chat，取得 Auth Token |
|
||||
| `get_or_create_chat_channel()` | 為每個對話創建專屬頻道 (不使用 Thread) |
|
||||
| `send_user_message()` | 發送用戶訊息到頻道，回傳訊息 ID |
|
||||
| `wait_for_admin_reply()` | 輪詢等待管理員回覆，過濾機器人訊息 |
|
||||
|
||||
**Headers 提取邏輯：**
|
||||
```python
|
||||
# Open WebUI 轉發的 Headers
|
||||
user_id = headers_dict.get("x-openwebui-user-id")
|
||||
chat_id = headers_dict.get("x-openwebui-chat-id")
|
||||
user_name = headers_dict.get("x-openwebui-user-name")
|
||||
user_email = headers_dict.get("x-openwebui-user-email")
|
||||
```
|
||||
|
||||
**系統任務過濾：**
|
||||
```python
|
||||
# 跳過 Open WebUI 的標題/標籤/後續問題生成請求
|
||||
if user_message.strip().startswith("### Task:"):
|
||||
return {"choices": [{"message": {"content": "{}"}}], ...}
|
||||
```
|
||||
|
||||
### 遇到的問題與解決方案
|
||||
|
||||
#### 問題 1: MongoDB 版本不相容
|
||||
- **錯誤**: `MongoServerSelectionError`
|
||||
- **原因**: Rocket.Chat 8.0.1 需要 MongoDB 7.0+，但配置的是 5.0
|
||||
- **解決**: 將 `mongo:5.0` 升級為 `mongo:7.0`
|
||||
|
||||
#### 問題 2: postMessage 返回 400 錯誤
|
||||
- **錯誤**: `alias` 參數需要特殊權限
|
||||
- **原因**: Rocket.Chat API 的 `alias` 欄位需要 bot 權限
|
||||
- **解決**: 移除 `alias` 參數，改用訊息內文標註用戶名
|
||||
|
||||
#### 問題 3: Open WebUI 不發送用戶資訊 Headers
|
||||
- **錯誤**: `user_name` 始終為 `None`
|
||||
- **原因**: Open WebUI 預設不轉發用戶資訊
|
||||
- **解決**: 設置環境變數 `ENABLE_FORWARD_USER_INFO_HEADERS=True`
|
||||
|
||||
#### 問題 4: 奇怪的系統訊息被發送到 Rocket.Chat
|
||||
- **錯誤**: `### Task: Generate a concise title...` 等訊息出現
|
||||
- **原因**: Open WebUI 會額外發送標題/標籤/後續問題生成請求
|
||||
- **解決**: 過濾以 `### Task:` 開頭的訊息，直接回傳空 JSON
|
||||
|
||||
#### 問題 5: 重複回傳相同的管理員回覆
|
||||
- **錯誤**: 每次用戶訊息都收到同一個回覆
|
||||
- **原因**: Thread 架構複雜，難以正確追蹤新回覆
|
||||
- **解決**: 改用簡化架構 (每個對話一個 Channel，不用 Thread)
|
||||
|
||||
### 最終架構決策
|
||||
|
||||
**放棄 Thread 架構，改用 Channel 架構：**
|
||||
|
||||
| 項目 | Thread 架構 (放棄) | Channel 架構 (採用) |
|
||||
|-----|-------------------|-------------------|
|
||||
| 結構 | 用戶→頻道→多個Thread | 對話→專屬頻道 |
|
||||
| 追蹤 | 需要追蹤 Thread ID | 只需追蹤訊息 ID |
|
||||
| 複雜度 | 高 | 低 |
|
||||
| 可讀性 | 中 (Thread 內對話) | 高 (直接看頻道) |
|
||||
|
||||
**頻道命名規則：**
|
||||
```
|
||||
{用戶名小寫}-{chat_id前8位}
|
||||
例: ckliu-67b4341e
|
||||
```
|
||||
|
||||
### 待辦事項
|
||||
|
||||
- [ ] 實作管理員回覆後的通知機制
|
||||
- [ ] 處理超時後的重試邏輯
|
||||
- [ ] 新增管理員身份驗證 (目前使用單一 admin 帳號)
|
||||
- [ ] 考慮切換到 Streamlit 方案 (更輕量、完全客製化)
|
||||
|
||||
### 相關檔案
|
||||
|
||||
```
|
||||
tobiichiGPT/
|
||||
├── docker-compose.yml # 容器編排配置
|
||||
├── api/
|
||||
│ ├── server.py # API 轉接層主程式
|
||||
│ └── requirements.txt # Python 依賴
|
||||
└── exchange.md # 本文件
|
||||
```
|
||||
|
||||
### 環境變數清單
|
||||
|
||||
| 變數名稱 | 用途 | 預設值 |
|
||||
|---------|------|-------|
|
||||
| `ROCKETCHAT_URL` | Rocket.Chat API 位址 | `http://rocketchat:3000` |
|
||||
| `ROCKETCHAT_USER` | 登入帳號 | `admin` |
|
||||
| `ROCKETCHAT_PASSWORD` | 登入密碼 | `admin` |
|
||||
| `DB_HOST` | PostgreSQL 主機 | `postgres` |
|
||||
| `DB_PASSWORD` | 資料庫密碼 | (必填) |
|
||||
|
||||
### 測試指令
|
||||
|
||||
```bash
|
||||
# 重建並重啟 API 容器
|
||||
cd /mnt/data/External/tobiichiGPT
|
||||
sudo docker-compose up -d --build api
|
||||
|
||||
# 查看 API 日誌
|
||||
sudo docker logs tobiichiGPT-api --tail 100
|
||||
|
||||
# 重啟 Open WebUI (套用 Headers 設定)
|
||||
sudo docker-compose up -d openwebui
|
||||
```
|
||||
|
||||
Reference in New Issue
Block a user