feat: 调整提示词
This commit is contained in:
@@ -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"]
|
||||
|
||||
Reference in New Issue
Block a user