feat:增加移动端提示词
This commit is contained in:
12
services.py
12
services.py
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user