Commit afda82ab by tinywell

补充当前时间

parent 5cb5d5df
from typing import Any, List, Sequence, Union
from datetime import datetime
import langchain_core
from langchain.tools import BaseTool
from langchain_core.prompts import PromptTemplate,ChatPromptTemplate,SystemMessagePromptTemplate,MessagesPlaceholder,HumanMessagePromptTemplate
......@@ -123,15 +123,18 @@ Action:
你的回复格式为 Action:```$JSON_BLOB```然后 Observation。
"""
PROMPT_AGENT_HUMAN = """{input}\n\n {agent_scratchpad}\n (请注意,无论如何都要以 JSON 对象回复。工具返回的数据必须使用表格展示,包含在最终输出中,并且要保证数据的完整性)"""
PROMPT_AGENT_SYS_VARS = [ "tool_names", "tools"]
class RateAgentV2:
def __init__(self, llm, tools: List[BaseTool],prompt: PromptTemplate = None, verbose: bool = False,**args):
date_now = datetime.now().strftime("%Y-%m-%d")
prompt_human = f"{PROMPT_AGENT_HUMAN}\n\n今天是{date_now}"
prompt = ChatPromptTemplate.from_messages([
SystemMessagePromptTemplate.from_template(ONLINE_RATE_SYSTEM_PROMPT),
MessagesPlaceholder(variable_name="chat_history", optional=True),
HumanMessagePromptTemplate.from_template(PROMPT_AGENT_HUMAN)
HumanMessagePromptTemplate.from_template(prompt_human)
])
agent = create_structured_chat_agent(llm, tools, prompt)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment