This commit is contained in:
qinyong@9artedu.com 2026-06-15 16:15:46 +08:00
parent 586b529933
commit addbde2e26
2 changed files with 20 additions and 5 deletions

View File

@ -1,5 +1,6 @@
import asyncio import asyncio
import time import time
from datetime import datetime, timedelta
from typing import List from typing import List
from langchain_core.prompts import PromptTemplate from langchain_core.prompts import PromptTemplate
@ -159,7 +160,8 @@ class SummaryService:
return result return result
async def build(): async def build(day:str):
logger.info(f"执行日期:{day}")
db_assistant_mysql_client_manager.init() db_assistant_mysql_client_manager.init()
embedding_client.init() embedding_client.init()
milvus_client.init() milvus_client.init()
@ -168,7 +170,7 @@ async def build():
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.get_message_summary("2026-6-11") 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()
@ -176,5 +178,18 @@ async def build():
if __name__ == '__main__': if __name__ == '__main__':
asyncio.run(build()) #
start = datetime(2026, 5, 22)
end = datetime(2026, 6, 14)
result = []
temp = start
while temp <= end:
# 按 年-月-日 无补零格式输出
date_format = f"{temp.year}-{temp.month}-{temp.day}"
result.append(date_format)
temp += timedelta(days=1)
for date in result:
asyncio.run(build(date))

View File

@ -17,13 +17,13 @@ db_assistant:
database: ai_assistant database: ai_assistant
embedding: embedding:
host: localhost host: 8.159.132.53
port: 8081 port: 8081
model: BAAI/bge-large-zh-v1.5 model: BAAI/bge-large-zh-v1.5
milvus: milvus:
host: localhost host: 8.159.132.53
port: 19530 port: 19530
user: root user: root
password: Milvus password: Milvus