from langchain.agents import create_agent
from langchain.agents.middleware import wrap_model_call, ModelRequest, ModelResponse
from langchain.chat_models import init_chat_model
from langchain_openai import ChatOpenAI
from langchain_core.tools import tool
from langchain_core.messages import HumanMessage, AIMessage, ToolMessage, function
from init_llm import deepseek_llm
定义股票查询工具
from langchain.agents import create_agent
from langchain.tools import tool
from langgraph.graph.state import CompiledStateGraph
from init_llm import deepseek_llm
@tool
def get_weather(city: str) -> str:
"""获取指定城市的天气信息。"""
return f"{city}的天气为晴朗,25°C。"
agent = create_agent(
model=deepseek_llm,
tools=get_weather)
resp = agent.invoke( {
"messages": [
{"role": "system", "content": "你是一个天气查询助手,只回答天气相关的问题,其他问题请直接回答:我不清楚这问题答案。"},
{"role": "user", "content": "北京天气如何?"}
]
})
print(resp)
print(resp"messages"-1.content)
for x in resp"messages":
x.pretty_print()
D:\Users\msi\miniconda3\python.exe D:\nanobot-main\langchain-demo\bg16.py
{'messages': [
SystemMessage(content='你是一个天气查询助手,只回答天气相关的问题,其他问题请直接回答:我不清楚这问题答案。', additional_kwargs={}, response_metadata={}, id='c970d5d9-db58-4a87-b84b-e5054fab6480'),
HumanMessage(content='北京天气如何?', additional_kwargs={}, response_metadata={}, id='71422678-3692-4162-98e6-60d07d984e2a'),
AIMessage(content='好的,我来查询北京当前的天气情况。', additional_kwargs={'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 53, 'prompt_tokens': 301, 'total_tokens': 354, 'completion_tokens_details': None, 'prompt_tokens_details': {'audio_tokens': None, 'cached_tokens': 256}, 'prompt_cache_hit_tokens': 256, 'prompt_cache_miss_tokens': 45}, 'model_provider': 'deepseek', 'model_name': 'deepseek-v4-flash', 'system_fingerprint': 'fp_8b330d02d0_prod0820_fp8_kvcache_20260402', 'id': '0843ea56-571f-4216-95f8-09cae866b0a8', 'finish_reason': 'tool_calls', 'logprobs': None}, id='lc_run--019e97fa-d7f3-75c1-beef-7b91be5a8c51-0', tool_calls={'name': 'get_weather', 'args': {'city': '北京'}, 'id': 'call_00_fCu7OgPUVyZ8WNa1HUhZ7618', 'type': 'tool_call'}, invalid_tool_calls=\[\], usage_metadata={'input_tokens': 301, 'output_tokens': 53, 'total_tokens': 354, 'input_token_details': {'cache_read': 256}, 'output_token_details': {}}),
ToolMessage(content='北京的天气为晴朗,25°C。', name='get_weather', id='ff42e71d-bd3a-42c8-9dbc-033a4e01e9ec', tool_call_id='call_00_fCu7OgPUVyZ8WNa1HUhZ7618'),
AIMessage(content='北京当前天气为**晴朗**,气温 **25°C**,天气不错,适合外出活动!', additional_kwargs={'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 22, 'prompt_tokens': 373, 'total_tokens': 395, 'completion_tokens_details': None, 'prompt_tokens_details': {'audio_tokens': None, 'cached_tokens': 256}, 'prompt_cache_hit_tokens': 256, 'prompt_cache_miss_tokens': 117}, 'model_provider': 'deepseek', 'model_name': 'deepseek-v4-flash', 'system_fingerprint': 'fp_8b330d02d0_prod0820_fp8_kvcache_20260402', 'id': 'e68f0121-7a58-4769-80da-4e80d6ea11bc', 'finish_reason': 'stop', 'logprobs': None}, id='lc_run--019e97fa-dd04-7402-92a2-cddd3a8b62b5-0', tool_calls=\[\], invalid_tool_calls=\[\], usage_metadata={'input_tokens': 373, 'output_tokens': 22, 'total_tokens': 395, 'input_token_details': {'cache_read': 256}, 'output_token_details': {}})]}