feat: 调整提示词

This commit is contained in:
2026-09-02 19:41:03 +08:00
parent e71a26e761
commit 03139ffdbd
2 changed files with 36 additions and 53 deletions

View File

@@ -1,7 +1,8 @@
import json
from datetime import datetime
from typing import Dict, Any
from langchain_core.messages import HumanMessage
from langchain_core.messages import HumanMessage, SystemMessage
from agent.health import tip_agent, recipe_agent
@@ -10,11 +11,15 @@ memory_store: Dict[str, Any] = {
"recipe": []
}
def query_tip_agent():
try:
print("开始生成今日5篇健康科普文章")
print("开始生成今日健康资讯")
today = datetime.today().strftime("%Y-%m-%d")
resp = tip_agent.invoke({
"messages": [HumanMessage(content="请生成今日5篇健康科普文章")]
"messages": [SystemMessage(content=f"当前系统日期:{today} \n\n"),
HumanMessage(content="请生成今日健康资讯")]
})
last_msg = resp["messages"][-1]
raw_text = last_msg.content.strip()
@@ -23,7 +28,7 @@ def query_tip_agent():
articles = json.loads(raw_text)
# 写入内存缓存
memory_store["articles"] = articles
print("结束生成今日5篇健康科普文章")
print("结束生成今日健康资讯")
return True
except json.JSONDecodeError as je:
print(f"[JSON解析错误] {str(je)}")
@@ -35,14 +40,19 @@ def query_tip_agent():
memory_store["articles"] = []
return False
def get_latest_tip():
return memory_store["articles"]
def query_recipe_agent():
try:
print("开始生成今日")
print("开始生成今日养生食")
today = datetime.today().strftime("%Y-%m-%d")
resp = recipe_agent.invoke({
"messages": [HumanMessage(content="请生成今日菜谱")]
"messages": [SystemMessage(content=f"当前系统日期:{today} \n\n"),
HumanMessage(content="请生成今日养生食谱")]
})
last_msg = resp["messages"][-1]
raw_text = last_msg.content.strip()
@@ -51,7 +61,7 @@ def query_recipe_agent():
recipes = json.loads(raw_text)
# 写入内存缓存
memory_store["recipe"] = recipes
print("结束生成今日")
print("结束生成今日养生食")
return True
except json.JSONDecodeError as je:
print(f"[JSON解析错误] {str(je)}")
@@ -63,5 +73,6 @@ def query_recipe_agent():
memory_store["recipe"] = []
return False
def get_latest_recipe():
return memory_store["recipe"]
return memory_store["recipe"]