Compare commits

..

3 Commits

Author SHA1 Message Date
ly 59488ae459 停止使用历史消息 2024-08-19 17:23:43 +08:00
ly 9eb8142a0b 修改导入关系 2024-08-19 17:23:01 +08:00
ly d52174532e 调整TOP_K参数值 2024-08-19 17:22:38 +08:00
3 changed files with 3 additions and 4 deletions
+1 -1
View File
@@ -4,7 +4,7 @@ SQL_DATABASE_URL=mysql+pymysql://zjinfo1:Dy2Bcr53Hm5xRkba@110.42.234.166:3306/zj
#SQL_DATABASE_URL=mysql+pymysql://zjinfo2:GSKcziSdBixDXwcd@110.42.234.166:3306/zjinfo2 #SQL_DATABASE_URL=mysql+pymysql://zjinfo2:GSKcziSdBixDXwcd@110.42.234.166:3306/zjinfo2
# The number of similar embeddings to return when retrieving documents. # The number of similar embeddings to return when retrieving documents.
TOP_K=5 TOP_K=10
#-------------------------- #--------------------------
# 是否启用检索重排功能 # 是否启用检索重排功能
RERANK_ENABLED=true RERANK_ENABLED=true
+2 -2
View File
@@ -48,6 +48,8 @@ async def chat(
): ):
try: try:
last_message_content = data.get_last_message_content() last_message_content = data.get_last_message_content()
# 由于基于历史消息的提示词没有调整好,所以暂时屏蔽历史消息
data.messages.clear()
messages = data.get_history_messages() messages = data.get_history_messages()
doc_ids = data.get_chat_document_ids() doc_ids = data.get_chat_document_ids()
@@ -59,8 +61,6 @@ async def chat(
event_handler = EventCallbackHandler() event_handler = EventCallbackHandler()
chat_engine.callback_manager.handlers.append(event_handler) # type: ignore chat_engine.callback_manager.handlers.append(event_handler) # type: ignore
# 由于基于历史消息的提示词没有调整好,所以暂时屏蔽历史消息
messages = None
response = await chat_engine.astream_chat(last_message_content, messages) response = await chat_engine.astream_chat(last_message_content, messages)
process_response_nodes(response.source_nodes, background_tasks) process_response_nodes(response.source_nodes, background_tasks)
-1
View File
@@ -13,7 +13,6 @@ from llama_index.core.instrumentation.events.rerank import ReRankStartEvent, ReR
from llama_index.core.postprocessor.types import BaseNodePostprocessor from llama_index.core.postprocessor.types import BaseNodePostprocessor
from llama_index.core.schema import ImageType, NodeWithScore, QueryBundle from llama_index.core.schema import ImageType, NodeWithScore, QueryBundle
from pydantic import Field from pydantic import Field
from win32comext.shell.demos.IUniformResourceLocator import new_sh
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)