Compare commits
3 Commits
a82a80c56e
...
59488ae459
| Author | SHA1 | Date | |
|---|---|---|---|
| 59488ae459 | |||
| 9eb8142a0b | |||
| d52174532e |
@@ -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
|
||||||
|
|||||||
@@ -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)
|
||||||
|
|
||||||
|
|||||||
@@ -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__)
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user