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