from langgraph.prebuilt import create_react_agent
from langchain_ollama import ChatOllama
from langgraph.checkpoint.memory import InMemorySaver

def getId() -> str:
    """explain your identity"""
    return f"我是凯瑞汽车训练的智能销售助理"

def getPriceByItem(item):
    """根据车名查询价格"""
    priceTable= {"月光女神":"135000",
                 "黑猫":"243000",
                 "飞扬":"175000"}
    if item in priceTable:
        return item+"的单价是："+priceTable[item]

checkpointer = InMemorySaver()

agent = create_react_agent(
    model=ChatOllama(model="qwen2.5:7b",temperature = 0.1),
    tools=[getId,getPriceByItem],
    checkpointer=checkpointer
)


config = {"configurable": {"thread_id": "1"}}
prompt1 = "月光女神这款车多少钱？"
print("user:"+prompt1)
result = agent.invoke(
    {"messages": [{"role": "user", "content": prompt1}]},
    config
)
print("assistant:"+result["messages"][-1].content)

prompt2 = "那飞扬呢？"
print("user:"+prompt2)
result = agent.invoke(
    {"messages": [{"role": "user", "content":prompt2 }]},
    config
)
print("assistant:"+result["messages"][-1].content)
print(result["messages"])