coder/ai-chat-system
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: WebSocket history 消息添加 extra_data 字段，搜索结果可正确显示

Browse Source
This commit is contained in:
hubian
2026-04-13 23:10:13 +08:00
parent 49e15130f1
commit d1431ac521
2 changed files with 14 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
main_v2.py
View File

@@ -743,6 +743,7 @@ async def websocket_endpoint(websocket: WebSocket, user_id: str):
"role": m.role,
"content": m.content,
"thinking_content": m.thinking_content,
"extra_data": m.extra_data, # 包含搜索结果等
"agent_id": m.agent_id, # 每条消息的Agent ID
"source": m.source,
"created_at": m.created_at.isoformat()

19
templates/index.html
View File

@@ -373,12 +373,18 @@
div.innerHTML = html;
// 如果是用户消息且有搜索结果，在设置innerHTML后追加
if (role === 'user' && extraData && extraData.search_results && extraData.search_results.length > 0) {
const searchHtml = buildSearchResultsHtml(extraData.search_results, extraData.search_query || content);
// 直接追加到 div 的 innerHTML
const bodyDiv = div.querySelector('.message-body');
if (bodyDiv) {
bodyDiv.insertAdjacentHTML('beforeend', searchHtml);
if (role === 'user' && extraData) {
console.log('Processing extraData for user message:', extraData);
console.log('search_results exists:', extraData.search_results);
if (extraData.search_results && extraData.search_results.length > 0) {
console.log('Building search results HTML for', extraData.search_results.length, 'results');
const searchHtml = buildSearchResultsHtml(extraData.search_results, extraData.search_query || content);
const bodyDiv = div.querySelector('.message-body');
console.log('bodyDiv found:', bodyDiv != null);
if (bodyDiv) {
bodyDiv.insertAdjacentHTML('beforeend', searchHtml);
console.log('Search results HTML inserted');
}
}
}
@@ -634,6 +640,7 @@
const container = document.getElementById('messagesContainer');
container.innerHTML = '';
messages.forEach(m => {
console.log('displayHistory message:', m.role, 'extra_data:', m.extra_data);
appendMessage(m.role, m.content, m.thinking_content, null, m.extra_data);
});
}