qinyong/sales-assistant-py-new
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: 意图识别优化

Browse Source
This commit is contained in:
qinyong@9artedu.com 2026-08-13 11:27:20 +08:00
parent 928085ecfc
commit 50bf80803b
2 changed files with 23 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
app/chat_qa/classification/history_intent.py
View File

File diff suppressed because one or more lines are too long

40
app/summary/service.py
View File

@ -125,23 +125,23 @@ class SummaryService:
logger.error(f"{date_user_a}和{date_user_b}对话embeddings重试后仍然失败: {retry_error}")
if not batch_embeddings or len(batch_embeddings) < 2:
batch_embeddings=[[[0]],[[1]]]
else:
try:
self.summary_repository.insert(
msg_time=[day_date],
msg_type=[0],
from_user=[date_a.from_user],
to_user=[date_user_b_clean],
room_id=[date_a.roomid or ""],
message_context=[filter_gossip_message_text],
summary=[summary_text],
message_dense_vector=[batch_embeddings[1]],
summary_dense_vector=[batch_embeddings[0]]
)
logger.info(f"查询{date_user_a}和{date_user_b}对话成功入库")
except Exception as e:
logger.error(f"{date_user_a}和{date_user_b}对话操作milvus失败: {e}")
batch_embeddings = [[0.0] * 1024, [0.0] * 1024]
try:
self.summary_repository.insert(
msg_time=[day_date],
msg_type=[0],
from_user=[date_a.from_user],
to_user=[date_user_b_clean],
room_id=[date_a.roomid or ""],
message_context=[filter_gossip_message_text],
summary=[summary_text],
message_dense_vector=[batch_embeddings[1]],
summary_dense_vector=[batch_embeddings[0]]
)
logger.info(f"查询{date_user_a}和{date_user_b}对话成功入库")
except Exception as e:
logger.error(f"{date_user_a}和{date_user_b}对话操作milvus失败: {e}")
logger.info(f"运行结束：{(time.time()-stime)}")
async def filter_gossip_message(self, coro_list: List[ArchiveMessages]) -> str:
@ -238,7 +238,7 @@ async def build(day:str):
archive_messages_repository = ArchiveMessagesRepository(db_assistant)
summary_repository = SummaryRepository()
service = SummaryService(archive_messages_repository, embedding_client, summary_repository)
await service.real_time_summary("BuShouShiJinBuGaiMing","wmI1AkDQAADMN3I0RScMP_g4R8jGoVfQ")
await service.get_message_summary(day)
finally:
await embedding_client.close()
await db_assistant_mysql_client_manager.close()
@ -247,8 +247,8 @@ async def build(day:str):
if __name__ == '__main__':
#
start = datetime(2026, 6, 24)
end = datetime(2026, 6, 24)
start = datetime(2026, 8, 12)
end = datetime(2026, 8, 12)
result = []
temp = start