mirror of
https://github.com/RYDE-WORK/Langchain-Chatchat.git
synced 2026-01-19 21:37:20 +08:00
9c525b7fa5
新功能： - 优化 PDF 文件的 OCR，过滤无意义的小图片 by @liunux4odoo #2525 - 支持 Gemini 在线模型 by @yhfgyyf #2630 - 支持 GLM4 在线模型 by @zRzRzRzRzRzRzR - elasticsearch更新https连接 by @xldistance #2390 - 增强对PPT、DOC知识库文件的OCR识别 by @596192804 #2013 - 更新 Agent 对话功能 by @zRzRzRzRzRzRzR - 每次创建对象时从连接池获取连接，避免每次执行方法时都新建连接 by @Lijia0 #2480 - 实现 ChatOpenAI 判断token有没有超过模型的context上下文长度 by @glide-the - 更新运行数据库报错和项目里程碑 by @zRzRzRzRzRzRzR #2659 - 更新配置文件/文档/依赖 by @imClumsyPanda @zRzRzRzRzRzRzR - 添加日文版 readme by @eltociear #2787 修复： - langchain 更新后，PGVector 向量库连接错误 by @HALIndex #2591 - Minimax's model worker 错误 by @xyhshen - ES库无法向量检索.添加mappings创建向量索引 by MSZheng20 #2688
32 lines
1.0 KiB
Python
32 lines
1.0 KiB
Python
from typing import Any, Dict, List
|
||
|
||
from langchain.callbacks.base import BaseCallbackHandler
|
||
from langchain.schema import LLMResult
|
||
from server.db.repository import update_message
|
||
|
||
|
||
class ConversationCallbackHandler(BaseCallbackHandler):
|
||
raise_error: bool = True
|
||
|
||
def __init__(self, conversation_id: str, message_id: str, chat_type: str, query: str):
|
||
self.conversation_id = conversation_id
|
||
self.message_id = message_id
|
||
self.chat_type = chat_type
|
||
self.query = query
|
||
self.start_at = None
|
||
|
||
@property
|
||
def always_verbose(self) -> bool:
|
||
"""Whether to call verbose callbacks even if verbose is False."""
|
||
return True
|
||
|
||
def on_llm_start(
|
||
self, serialized: Dict[str, Any], prompts: List[str], **kwargs: Any
|
||
) -> None:
|
||
# 如果想存更多信息，则prompts 也需要持久化
|
||
pass
|
||
|
||
def on_llm_end(self, response: LLMResult, **kwargs: Any) -> None:
|
||
answer = response.generations[0][0].text
|
||
update_message(self.message_id, answer)
|