feat:增加移动端提示词

This commit is contained in:
2026-07-04 20:25:10 +08:00
parent 20e45d5c8b
commit 900d5bd774
5 changed files with 71 additions and 38 deletions

View File

@@ -1,9 +1,10 @@
from datetime import datetime
from langchain_core.messages import HumanMessage, AIMessageChunk
from langchain_core.messages import HumanMessage, AIMessageChunk, SystemMessage, BaseMessage
from agent.bill import bill_agent
from modes.agent import QueryRequest
from prompt.mobile import mobile_prompt
async def query_bill_agent(query: QueryRequest):
@@ -11,19 +12,22 @@ async def query_bill_agent(query: QueryRequest):
print(query)
today = datetime.today().strftime("%Y-%m-%d")
message_content = (
date_content = (
f"[当前系统日期: {today}]\n\n"
f"{query.message}"
)
message = HumanMessage(content=message_content)
messages: list[BaseMessage] = [HumanMessage(content=date_content)]
if query.platform == "mobile":
messages.insert(0, SystemMessage(content=mobile_prompt))
tool_called = False
content_generate = False
# 流式调用Agent
for chunk, metadata in bill_agent.stream(
{"messages": [message]},
{"messages": messages},
stream_mode="messages"
):
# print(chunk)