Change rocket.chat admin logic

This commit is contained in:
ChenKaiLiuG
2026-02-01 08:15:45 +08:00
parent 44ba70efe6
commit f105a7e748
3 changed files with 358 additions and 215 deletions

133
INTRO.md
View File

@@ -7,18 +7,22 @@ TobiichiGPT 將 Open WebUI 這類 AI 對話系統的後端改造成由真人管
### 基本原理 ### 基本原理
1. **API 中間層** - 提供 OpenAI API 相容端點(`/v1/chat/completions` 1. **API 中間層** - 提供 OpenAI API 相容端點(`/v1/chat/completions`
2. **訊息轉發** - 將用戶訊息存入資料庫的 `reply_queue` 表格 2. **訊息轉發** - 將用戶訊息轉發到 Rocket.Chat 頻道
3. **管理員介面** - 使用對話軟體讓真人管理員查看並回覆 3. **管理員介面** - 使用 Rocket.Chat 讓真人管理員查看並回覆
4. **輪詢機制** - API 等待管理員在資料庫中填入回覆後返回給用戶 4. **輪詢機制** - API 使用時間戳比較等待管理員在 Rocket.Chat 中回覆
### 技術架構 ### 技術架構
``` ```
用戶 → Open WebUI → API (FastAPI) → PostgreSQL ← 管理介面 ← 管理員 用戶 → Open WebUI → API (FastAPI) → Rocket.Chat ← 管理員
↓ ↓
等待回覆 填寫回覆 等待回覆 管理員查看訊息
↓ ↓
返回給用戶 ←──────────────────────────┘ 輪詢檢查 管理員回覆訊息
↓ ↓
返回給用戶 ←─────────┘
(PostgreSQL 用於記錄追蹤)
``` ```
## 需求分析 ## 需求分析
@@ -48,21 +52,23 @@ Open WebUI 採用**雙層結構**來組織對話:
理想的管理介面應該呈現: 理想的管理介面應該呈現:
``` ```
📂 所有對話(按用戶分組 📂 所有對話(每個對話獨立頻道
├── 👤 張三 (User A) ├── 💬 張三-chat001 (教我 Python)
│ ├── 💬 對話: "教我 Python" (3 則訊息) │ ├── 💬 張三: "Python 怎麼學?"
── 💬 對話: "推薦餐廳" (5 則訊息) ── 💬 管理員: "建議從基礎開始..."
│ └── 💬 對話: "旅遊攻略" (2 則訊息)
── 👤 李四 (User B) ── 💬 張三-chat002 (推薦餐廳)
├── 💬 對話: "程式問題" (1 則訊息) ⚠️ 待回覆 ├── 💬 張三: "台北有什麼好吃的?"
└── 💬 對話: "健康諮詢" (4 則訊息) └── 💬 管理員: "推薦鼎泰豐..."
└── 💬 李四-chat004 (程式問題) ⚠️ 待回覆
└── 💬 李四: "React Hook 怎麼用?"
``` ```
**核心需求** **核心需求**
-用戶分組 - 能看到每個用戶的所有對話 -對話隔離 - 每個對話一個獨立頻道
-對話追蹤 - 同一對話的多輪訊息要串在一起 -用戶識別 - 頻道名稱包含用戶名和對話 ID
-多層結構 - 支援「用戶 → 對話 → 訊息」的層級關係 -簡單直觀 - 不使用 Thread直接在頻道內問答
- ✅ 狀態管理 - 清楚標示哪些對話待回覆 - ✅ 狀態管理 - 清楚標示哪些對話待回覆
## Rocket.Chat 整合架構 ## Rocket.Chat 整合架構
@@ -83,23 +89,26 @@ Open WebUI 採用**雙層結構**來組織對話:
├─────────────────────────────────────────────────────────┤ ├─────────────────────────────────────────────────────────┤
│ 2. 解析 Headers 提取 user_id, chat_id, user_name │ │ 2. 解析 Headers 提取 user_id, chat_id, user_name │
│ 3. 登入 Rocket.Chat 取得認證 Token │ │ 3. 登入 Rocket.Chat 取得認證 Token │
│ 4. 創建/取得用戶頻道 (#user-{user_id}) │ 4. 創建/取得對話頻道 ({user_name}-{chat_id[:8]})
│ 5. 在頻道中創建執行緒訊息 │ 5. 發送用戶訊息到頻道,記錄時間戳
│ 6. 輪詢等待管理員在執行緒中回覆 │ 6. 輪詢頻道訊息,用時間戳比較等待管理員回覆
│ 7. 將回覆返回給 Open WebUI │ 7. 將管理員回覆返回給 Open WebUI │
└──────┬──────────────────────────┬───────────────────────┘ └──────┬──────────────────────────┬───────────────────────┘
│ │ │ │
▼ ▼ ▼ ▼
┌─────────────┐ ┌─────────────────────────┐ ┌─────────────┐ ┌─────────────────────────┐
│ PostgreSQL │ │ Rocket.Chat + MongoDB │ │ PostgreSQL │ │ Rocket.Chat + MongoDB │
│ reply_queue │ ├─────────────────────────┤ │ reply_queue │ ├─────────────────────────┤
└─────────────┘ │ 頻道: #user-abc123 │ (記錄追蹤) │ 頻道: ckliu-68a77282
── 🧵 chat-001 └─────────────┘── 💬 ckliu: 測試訊息
├── 用戶訊息 └── 💬 管理員: ...
└── 管理員回覆
│ 頻道: ckliu-a8224bcf │
│ ├── 💬 ckliu: 測試訊息 │
│ └── 💬 管理員: ... │
└─────────────────────────┘ └─────────────────────────┘
│ 8. 管理員回覆 │ 8. 管理員直接在頻道回覆
┌─────┴──────┐ ┌─────┴──────┐
│ 管理員瀏覽器│ │ 管理員瀏覽器│
@@ -110,37 +119,45 @@ Open WebUI 採用**雙層結構**來組織對話:
| Open WebUI | Rocket.Chat | 說明 | | Open WebUI | Rocket.Chat | 說明 |
|-----------|-------------|------| |-----------|-------------|------|
| 用戶 (user_id) | 頻道 (Channel) | 每個用戶一個專屬頻道 `#user-{id}` | | 對話 (chat_id) | 頻道 (Channel) | 每個對話一個專屬頻道 `{user_name}-{chat_id[:8]}` |
| 對話 (chat_id) | 執行緒 (Thread) | 同一對話的多輪訊息在同一執行緒 | | 訊息 (message) | 訊息 (Message) | 直接在頻道中發送,不使用 Thread |
| 訊息 (message) | 訊息 (Message) | 用戶和管理員的對話內容 | | 用戶 (user_name) | 頻道名稱前綴 | 用於識別是誰的對話 |
**設計理念**
- 簡化架構,不使用 Thread執行緒
- 每個對話完全獨立,避免複雜的層級結構
- 頻道名稱直接顯示用戶和對話 ID方便識別
### Rocket.Chat 呈現效果 ### Rocket.Chat 呈現效果
``` ```
Rocket.Chat 介面: Rocket.Chat 介面:
├── 📂 頻道列表 ├── 📂 頻道列表
│ ├── #user-abc123 (張三) │ ├── ckliu-68a77282 (最近活躍)
│ └── 描述: 用戶: 張三 (ID: abc123...) ├── ckliu-a8224bcf
│ ├── #user-def456 (李四) │ ├── alice-3f91c2e1
│ └── 描述: 用戶: 李四 (ID: def456...) │ └── bob-7d4e9a23
│ └── #user-xyz789 (王五)
└── 📂 #user-abc123 (張三) 的內容 └── 📂 ckliu-68a77282 頻道內容
├── 🧵 [對話 chat-001] "如何學習 Python?" ├── 💬 **💬 ckliu:**
├── 💬 張三: "如何學習 Python?" NPU能吃嗎...
│ └── 💬 管理員: "推薦從基礎開始..." ── 💬 管理員回覆:
NPU 是神經處理單元,不能吃喔 😄
├── 🧵 [對話 chat-002] "推薦餐廳" ├── 💬 **💬 ckliu:**
├── 💬 張三: "台北有什麼好吃的?" 那有什麼用途?
└── 💬 管理員: "推薦鼎泰豐..." └── 💬 管理員回覆:
主要用於 AI 運算加速...
└── 🧵 [對話 chat-003] ⚠️ 待回覆
└── 💬 張三: "旅遊攻略"
``` ```
**特點**
- 每個頻道 = 一個完整對話
- 訊息直接在頻道中,不用點開 Thread
- 管理員看到新頻道就知道有新對話
- 頻道名稱清楚標示用戶和對話 ID
### 資料庫結構 ### 資料庫結構
`reply_queue` 表格結構: `reply_queue` 表格結構(用於記錄追蹤)
```sql ```sql
CREATE TABLE reply_queue ( CREATE TABLE reply_queue (
@@ -154,11 +171,16 @@ CREATE TABLE reply_queue (
status VARCHAR(20) DEFAULT 'pending', status VARCHAR(20) DEFAULT 'pending',
created_at TIMESTAMP DEFAULT NOW(), created_at TIMESTAMP DEFAULT NOW(),
replied_at TIMESTAMP, replied_at TIMESTAMP,
rocketchat_room_id VARCHAR(100), rocketchat_room_id VARCHAR(100), -- 儲存頻道 ID
rocketchat_thread_id VARCHAR(100) rocketchat_thread_id VARCHAR(100) -- 保留欄位(目前未使用)
); );
``` ```
**說明**
- PostgreSQL 僅用於記錄追蹤,非主要邏輯
- 實際對話內容和回覆檢測都在 Rocket.Chat 中進行
- 使用時間戳比較來判斷是否有新的管理員回覆
### 服務列表 ### 服務列表
| 服務 | 容器名 | Port | 說明 | | 服務 | 容器名 | Port | 說明 |
@@ -171,8 +193,9 @@ CREATE TABLE reply_queue (
### 技術特點 ### 技術特點
1. **雙層映射** - 完美對應 Open WebUI 的用戶+對話結構 1. **一對一映射** - 每個 Open WebUI 對話對應一個 Rocket.Chat 頻道
2. **執行緒隔離** - 每個對話獨立,不會混雜 2. **簡化架構** - 不使用 Thread減少複雜度和潛在問題
3. **即時通知** - Rocket.Chat 支援桌面和手機推送 3. **時間戳比較** - 使用 ISO 格式時間戳判斷新訊息,簡單可靠
4. **輕量部署** - 共 5 個容器(含初始化容器) 4. **即時通知** - Rocket.Chat 支援桌面和手機推送
5. **可擴展性** - 支援多管理員協作回覆 5. **輕量部署** - 共 5 個容器(含初始化容器)
6. **可擴展性** - 支援多管理員協作回覆

View File

@@ -82,18 +82,21 @@ async def rocketchat_login():
return None return None
async def get_or_create_channel(user_id: str, user_name: str = None): async def get_or_create_chat_channel(chat_id: str, user_name: str = None):
"""取得或創建用戶專屬頻道""" """為每個對話創建專屬頻道
新架構:每個 chat_id 對應一個 Channel不使用 Thread
"""
if not rocketchat_auth: if not rocketchat_auth:
await rocketchat_login() await rocketchat_login()
if not rocketchat_auth: if not rocketchat_auth:
return None return None
# 頻道名稱使用用戶名,如果沒有則用 user_id # 頻道名稱用戶名-對話ID前8位
display_name = user_name if user_name else f"User-{user_id[:8]}" display_name = user_name if user_name else "user"
# 清理頻道名稱移除空格和特殊字符Rocket.Chat 頻道名不支援) clean_name = display_name.replace(" ", "-").replace("@", "").lower()
channel_name = display_name.replace(" ", "-").replace("@", "").lower()[:50] # 限制長度 channel_name = f"{clean_name}-{chat_id[:8]}"[:50] # 限制長度
try: try:
async with httpx.AsyncClient(timeout=10.0) as client: async with httpx.AsyncClient(timeout=10.0) as client:
@@ -103,14 +106,14 @@ async def get_or_create_channel(user_id: str, user_name: str = None):
headers=rocketchat_auth, headers=rocketchat_auth,
json={ json={
"name": channel_name, "name": channel_name,
"members": [] # 空成員列表,只有管理員可見 "members": []
} }
) )
if resp.status_code == 200: if resp.status_code == 200:
data = resp.json() data = resp.json()
room_id = data["channel"]["_id"] room_id = data["channel"]["_id"]
print(f"✅ 創建頻道: {channel_name} ({display_name})") print(f"✅ 創建對話頻道: {channel_name}")
# 設置頻道描述 # 設置頻道描述
await client.post( await client.post(
@@ -118,7 +121,7 @@ async def get_or_create_channel(user_id: str, user_name: str = None):
headers=rocketchat_auth, headers=rocketchat_auth,
json={ json={
"roomId": room_id, "roomId": room_id,
"description": f"用戶: {display_name} (ID: {user_id})" "description": f"用戶: {display_name} | 對話: {chat_id[:8]}"
} }
) )
@@ -149,122 +152,50 @@ async def get_or_create_channel(user_id: str, user_name: str = None):
return None return None
async def get_or_create_thread(room_id: str, chat_id: str): async def send_user_message(room_id: str, user_message: str, user_name: str = None):
"""查找或創建對話的執行緒""" """發送用戶訊息到頻道(不使用 Thread"""
if not rocketchat_auth: if not rocketchat_auth:
await rocketchat_login() await rocketchat_login()
if not rocketchat_auth: if not rocketchat_auth:
return None return None
try: display_name = user_name if user_name else "用戶"
async with httpx.AsyncClient(timeout=10.0) as client:
# 搜尋頻道中是否已有此 chat_id 的執行緒搜尋最近50條消息
resp = await client.get(
f"{ROCKETCHAT_URL}/api/v1/channels.messages",
headers=rocketchat_auth,
params={
"roomId": room_id,
"count": 50
}
)
if resp.status_code == 200:
data = resp.json()
messages = data.get("messages", [])
# 找到現有的執行緒起始訊息
for msg in messages:
if f"對話 {chat_id[:8]}" in msg.get("msg", ""):
thread_id = msg["_id"]
print(f"✅ 找到現有執行緒: {thread_id}")
return thread_id
# 沒有找到,返回 None需要創建新執行緒
return None
except Exception as e:
print(f"❌ 查找執行緒錯誤: {e}")
return None
async def create_thread_message(room_id: str, chat_id: str, user_message: str, user_name: str = None):
"""在頻道中發送訊息並創建執行緒"""
if not rocketchat_auth:
await rocketchat_login()
if not rocketchat_auth:
return None
display_name = user_name if user_name else "User"
try: try:
async with httpx.AsyncClient(timeout=10.0) as client: async with httpx.AsyncClient(timeout=10.0) as client:
# 發送訊息(這會成為執行緒的起始訊息)
resp = await client.post( resp = await client.post(
f"{ROCKETCHAT_URL}/api/v1/chat.postMessage", f"{ROCKETCHAT_URL}/api/v1/chat.postMessage",
headers=rocketchat_auth, headers=rocketchat_auth,
json={ json={
"roomId": room_id, "roomId": room_id,
"text": f"**[對話 {chat_id[:8]}] - {display_name}**\n\n{user_message}" "text": f"**💬 {display_name}:**\n{user_message}"
} }
) )
if resp.status_code == 200: if resp.status_code == 200:
data = resp.json() data = resp.json()
message_id = data["message"]["_id"] message_id = data["message"]["_id"]
print(f"✅ 創建執行緒訊息: {message_id}") message_ts = data["message"]["ts"]
return message_id print(f"✅ 發送用戶訊息: {message_id}")
return {"message_id": message_id, "ts": message_ts}
else: else:
print(f"⚠️ 發送訊息失敗: {resp.status_code}") print(f"⚠️ 發送訊息失敗: {resp.status_code}")
print(f" 錯誤內容: {resp.text}")
return None return None
except Exception as e: except Exception as e:
print(f" Rocket.Chat 發送訊息錯誤: {e}") print(f"❌ 發送訊息錯誤: {e}")
return None return None
async def add_thread_reply(room_id: str, thread_id: str, user_message: str, user_name: str = None): async def wait_for_admin_reply(room_id: str, after_ts: str, exclude_msg_id: str = None, timeout: int = 600):
"""在現有執行緒中追加用戶訊息""" """等待管理員在頻道中的回覆
if not rocketchat_auth:
await rocketchat_login()
if not rocketchat_auth:
return False
display_name = user_name if user_name else "User"
try:
async with httpx.AsyncClient(timeout=10.0) as client:
# 在執行緒中回覆
resp = await client.post(
f"{ROCKETCHAT_URL}/api/v1/chat.postMessage",
headers=rocketchat_auth,
json={
"roomId": room_id,
"text": f"**{display_name}:**\n\n{user_message}",
"tmid": thread_id # 指定執行緒 ID
}
)
if resp.status_code == 200:
print(f"✅ 追加執行緒訊息")
return True
else:
print(f"⚠️ 追加訊息失敗: {resp.status_code}")
return False
except Exception as e:
print(f"❌ 追加訊息錯誤: {e}")
return False
async def wait_for_thread_reply(room_id: str, thread_id: str, existing_message_count: int = 0, timeout: int = 600):
"""輪詢等待執行緒中的管理員回覆
Args: Args:
existing_message_count: 發送訊息前執行緒中已有的訊息數量,只等待新增的回覆 room_id: 頻道 ID
after_ts: 用戶訊息的時間戳 (ISO 格式)
exclude_msg_id: 要排除的用戶訊息 ID (避免自己讀到自己)
timeout: 超時秒數
""" """
if not rocketchat_auth: if not rocketchat_auth:
await rocketchat_login() await rocketchat_login()
@@ -273,19 +204,23 @@ async def wait_for_thread_reply(room_id: str, thread_id: str, existing_message_c
return None return None
start_time = time.time() start_time = time.time()
check_interval = 3 # 每 3 秒檢查一次 check_interval = 2 # 每 2 秒檢查一次
print(f"🔄 開始輪詢執行緒回覆 (thread_id: {thread_id}, 現有訊息: {existing_message_count})") # 取得機器人用戶 ID
bot_user_id = rocketchat_auth.get("X-User-Id")
print(f"🔄 等待回覆 (room: {room_id[:8]}..., after: {after_ts}, exclude: {exclude_msg_id})")
try: try:
async with httpx.AsyncClient(timeout=10.0) as client: async with httpx.AsyncClient(timeout=10.0) as client:
while time.time() - start_time < timeout: while time.time() - start_time < timeout:
# 取得執行緒中的所有訊息 # 取得頻道中的最新訊息
resp = await client.get( resp = await client.get(
f"{ROCKETCHAT_URL}/api/v1/chat.getThreadMessages", f"{ROCKETCHAT_URL}/api/v1/channels.messages",
headers=rocketchat_auth, headers=rocketchat_auth,
params={ params={
"tmid": thread_id # Thread Message ID "roomId": room_id,
"count": 20 # 稍微增加數量以防萬一
} }
) )
@@ -293,33 +228,64 @@ async def wait_for_thread_reply(room_id: str, thread_id: str, existing_message_c
data = resp.json() data = resp.json()
messages = data.get("messages", []) messages = data.get("messages", [])
# 只看新增的訊息(比現有數量多的部分) found_new_reply = False
if len(messages) > existing_message_count + 1: # +1 是我們剛發的訊息
print(f" 檢查執行緒,找到 {len(messages)} 條訊息 (新增: {len(messages) - existing_message_count - 1})") for msg in messages:
msg_id = msg.get("_id")
msg_ts = msg.get("ts", "")
sender_id = msg.get("u", {}).get("_id", "")
sender_name = msg.get("u", {}).get("username", "?")
reply_text = msg.get("msg", "")
msg_type = msg.get("t")
# 按時間排序,取最新的訊息 # 1. 基本過濾
sorted_messages = sorted(messages, key=lambda m: m.get("ts", ""), reverse=True) if msg_type: # 忽略系統訊息
continue
# 尋找管理員的新回覆 if msg_id == exclude_msg_id: # 忽略剛發送的用戶訊息
for msg in sorted_messages: continue
# 跳過原始訊息
if msg["_id"] == thread_id:
continue
# 檢查是否為真人回覆(有 u 欄位且不是 bot # 2. 時間戳檢查 (放寬條件: 只要 >= 就考慮,因為可能有精度差)
if "u" in msg and not msg.get("bot"): # 但為了避免讀到舊訊息,我們仍需確保它"看起來"是新的
reply_text = msg.get("msg", "") if msg_ts < after_ts:
# 跳過用戶自己的訊息(檢查是否包含對話標記) continue
if reply_text and not reply_text.startswith("[對話"):
print(f"✅ 收到管理員回覆: {reply_text[:50]}...") # 3. 機器人自我發送過濾
return reply_text if sender_id == bot_user_id:
# 這是最關鍵的部分: 如何區分 "API轉發的用戶訊息" 與 "Admin用同一帳號的回覆"
# A. 如果這就是我們要排除的 ID (前面已處理)
# B. 檢查內容格式
# 格式為: "**💬 {display_name}:**\n{user_message}"
# 我們檢查幾個特徵: 開頭是 **💬, 包含 :**
is_forwarded_msg = False
if reply_text.strip().startswith("**💬") and ":**" in reply_text:
is_forwarded_msg = True
if is_forwarded_msg:
# 這是轉發的訊息,跳過
continue
# 如果不是轉發格式,那我們假設這是 Admin 的手動回覆
pass
# 4. 找到有效回覆
if reply_text:
print(f"✅ 收到管理員回覆 (from {sender_name}, id: {msg_id}): {reply_text[:50]}...")
return reply_text
# 沒找到
elapsed = int(time.time() - start_time)
if elapsed % 10 == 0:
print(f"🔄 等待中... ({elapsed}s)")
else: else:
print(f" ⚠️ API 回應異常: {resp.status_code}") print(f"⚠️ API 錯誤: {resp.status_code}")
await asyncio.sleep(check_interval) await asyncio.sleep(check_interval)
# 超時 print(f"⚠️ 等待回覆超時")
print(f"⚠️ 等待回覆超時 ({timeout}秒)")
return "抱歉,目前客服繁忙,請稍後再試。" return "抱歉,目前客服繁忙,請稍後再試。"
except Exception as e: except Exception as e:
@@ -502,13 +468,12 @@ async def chat_completions(
} }
print(f"📝 收到訊息") print(f"📝 收到訊息")
print(f" 用戶ID: {user_id}") print(f" 用戶: {user_name} ({user_id[:8]}...)")
print(f" 用戶名: {user_name}")
print(f" 對話: {chat_id[:8]}") print(f" 對話: {chat_id[:8]}")
print(f" 內容: {user_message[:50]}...") print(f" 內容: {user_message[:50]}...")
# 1. 取得或創建用戶頻道 # 1. 為此對話取得或創建專屬頻道(每個 chat_id = 一個 Channel
room_id = await get_or_create_channel(user_id, user_name) room_id = await get_or_create_chat_channel(chat_id, user_name)
if not room_id: if not room_id:
return JSONResponse( return JSONResponse(
@@ -516,39 +481,16 @@ async def chat_completions(
content={"error": "Failed to create Rocket.Chat channel"} content={"error": "Failed to create Rocket.Chat channel"}
) )
# 2. 查找是否已有此 chat_id 的執行緒 # 2. 發送用戶訊息到頻道
thread_id = await get_or_create_thread(room_id, chat_id) msg_result = await send_user_message(room_id, user_message, user_name)
existing_message_count = 0
if thread_id: if not msg_result:
# 已有執行緒,先取得現有訊息數量 return JSONResponse(
try: status_code=500,
async with httpx.AsyncClient(timeout=10.0) as client: content={"error": "Failed to send message to Rocket.Chat"}
resp = await client.get( )
f"{ROCKETCHAT_URL}/api/v1/chat.getThreadMessages",
headers=rocketchat_auth, user_message_ts = msg_result["ts"] # 用時間戳比較,更可靠
params={"tmid": thread_id}
)
if resp.status_code == 200:
existing_message_count = len(resp.json().get("messages", []))
except Exception as e:
print(f"⚠️ 取得現有訊息數量失敗: {e}")
# 追加訊息
success = await add_thread_reply(room_id, thread_id, user_message, user_name)
if not success:
return JSONResponse(
status_code=500,
content={"error": "Failed to add message to thread"}
)
else:
# 沒有執行緒,創建新的
thread_id = await create_thread_message(room_id, chat_id, user_message, user_name)
if not thread_id:
return JSONResponse(
status_code=500,
content={"error": "Failed to create Rocket.Chat thread"}
)
# 3. 記錄到資料庫 # 3. 記錄到資料庫
async with db_pool.acquire() as conn: async with db_pool.acquire() as conn:
@@ -560,11 +502,11 @@ async def chat_completions(
) )
VALUES ($1, $2, $3, $4, $5, 'pending', $6, $7) VALUES ($1, $2, $3, $4, $5, 'pending', $6, $7)
""", """,
message_id, user_id, chat_id, user_name, user_message, room_id, thread_id message_id, user_id, chat_id, user_name, user_message, room_id, msg_result["message_id"]
) )
# 4. 等待管理員在執行緒中回覆(傳入現有訊息數量,只等待新回覆 # 4. 等待管理員在頻道中回覆(使用時間戳比較
admin_reply = await wait_for_thread_reply(room_id, thread_id, existing_message_count) admin_reply = await wait_for_admin_reply(room_id, user_message_ts, exclude_msg_id=msg_result["message_id"])
# 5. 更新資料庫 # 5. 更新資料庫
async with db_pool.acquire() as conn: async with db_pool.acquire() as conn:

View File

@@ -109,4 +109,182 @@ CMD ["streamlit", "run", "admin.py", "--server.port=8501", "--server.address=0.0
2. 伺服器資源不足,無法負擔 Rocket.Chat + MongoDB。 2. 伺服器資源不足,無法負擔 Rocket.Chat + MongoDB。
3. 需要針對特定業務邏輯(如:查看用戶餘額、審核特定關鍵字)進行客製化開發。 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
```