mirror of
https://github.com/RYDE-WORK/Langchain-Chatchat.git
synced 2026-02-02 12:46:56 +08:00
fa906b33a8
* 新功能：
- WEBUI 添加对话评分功能
- 增加 /chat/feedback 接口，用于接收对话评分
- /chat/chat 接口返回值由 str 改为 {"text":str, "chat_history_id": str}
- init_database.py 添加 --create-tables --clear-tables 参数
依赖：
- streamlit-chatbox==1.1.11
开发者：
- ChatHistoryModel 的 id 字段支持自动生成
- SAVE_CHAT_HISTORY 改到 basic_config.py
* 修复：点击反馈后页面未刷新
---------
Co-authored-by: liqiankun.1111 <liqiankun.1111@bytedance.com>
Co-authored-by: liunux4odoo <liunux@qq.com>
Co-authored-by: liunux4odoo <41217877+liunux4odoo@users.noreply.github.com>
76 lines
2.2 KiB
Python
76 lines
2.2 KiB
Python
from server.db.session import with_session
|
|
from server.db.models.chat_history_model import ChatHistoryModel
|
|
import re
|
|
import uuid
|
|
from typing import Dict, List
|
|
|
|
|
|
def _convert_query(query: str) -> str:
|
|
p = re.sub(r"\s+", "%", query)
|
|
return f"%{p}%"
|
|
|
|
|
|
@with_session
|
|
def add_chat_history_to_db(session, chat_type, query, response="", chat_history_id=None, metadata: Dict = {}):
|
|
"""
|
|
新增聊天记录
|
|
"""
|
|
if not chat_history_id:
|
|
chat_history_id = uuid.uuid4().hex
|
|
ch = ChatHistoryModel(id=chat_history_id, chat_type=chat_type, query=query, response=response,
|
|
metadata=metadata)
|
|
session.add(ch)
|
|
session.commit()
|
|
return ch.id
|
|
|
|
|
|
@with_session
|
|
def update_chat_history(session, chat_history_id, response: str = None, metadata: Dict = None):
|
|
"""
|
|
更新已有的聊天记录
|
|
"""
|
|
ch = get_chat_history_by_id(chat_history_id)
|
|
if ch is not None:
|
|
if response is not None:
|
|
ch.response = response
|
|
if isinstance(metadata, dict):
|
|
ch.meta_data = metadata
|
|
session.add(ch)
|
|
return ch.id
|
|
|
|
|
|
@with_session
|
|
def feedback_chat_history_to_db(session, chat_history_id, feedback_score, feedback_reason):
|
|
"""
|
|
反馈聊天记录
|
|
"""
|
|
ch = session.query(ChatHistoryModel).filter_by(id=chat_history_id).first()
|
|
if ch:
|
|
ch.feedback_score = feedback_score
|
|
ch.feedback_reason = feedback_reason
|
|
return ch.id
|
|
|
|
|
|
@with_session
|
|
def get_chat_history_by_id(session, chat_history_id) -> ChatHistoryModel:
|
|
"""
|
|
查询聊天记录
|
|
"""
|
|
ch = session.query(ChatHistoryModel).filter_by(id=chat_history_id).first()
|
|
return ch
|
|
|
|
|
|
@with_session
|
|
def filter_chat_history(session, query=None, response=None, score=None, reason=None) -> List[ChatHistoryModel]:
|
|
ch =session.query(ChatHistoryModel)
|
|
if query is not None:
|
|
ch = ch.filter(ChatHistoryModel.query.ilike(_convert_query(query)))
|
|
if response is not None:
|
|
ch = ch.filter(ChatHistoryModel.response.ilike(_convert_query(response)))
|
|
if score is not None:
|
|
ch = ch.filter_by(feedback_score=score)
|
|
if reason is not None:
|
|
ch = ch.filter(ChatHistoryModel.feedback_reason.ilike(_convert_query(reason)))
|
|
|
|
return ch
|