from langchain.agents import create_agent
from langchain_deepseek import ChatDeepSeek
from pydantic import BaseModel,Field
from env_utils import DEEPSEEK_API_KEY,DEEPSEEK_BASE_URL
from langchain.agents import create_agent,AgentState
from langchain_core.tools import tool
from langgraph.checkpoint.memory import InMemorySaver
from langgraph.prebuilt import ToolRuntime
deepseek_llm = ChatDeepSeek(model='deepseek-chat',
                            api_key=DEEPSEEK_API_KEY,
                            base_url=DEEPSEEK_BASE_URL)
res=deepseek_llm.invoke("你是谁？")
print(res)

@tool
def get_wweather(location:str) -> str:
    """获取天气信息
    """
    print('1111111111111111111111')
    return f"{location}的天气是晴朗的"

agent=create_agent(
    model=deepseek_llm,
    tools=[get_wweather],
    system_prompt= "你是一个天气助手，你可以根据用户的问题，获取天气信息"
)

resp=agent.invoke({"messages":[{"role":"user","content":"杭州的天气"}]})
print(resp)

{'messages': [

HumanMessage(content='杭州的天气', additional_kwargs={}, response_metadata={}, id='e76ebce6-74a0-458c-9be9-c6936160061d'), 

AIMessage(content='好的，我来查一下杭州的天气信息。', additional_kwargs={'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 54, 'prompt_tokens': 287, 'total_tokens': 341, 'completion_tokens_details': None, 'prompt_tokens_details': {'audio_tokens': None, 'cached_tokens': 256}, 'prompt_cache_hit_tokens': 256, 'prompt_cache_miss_tokens': 31}, 'model_provider': 'deepseek', 'model_name': 'deepseek-v4-flash', 'system_fingerprint': 'fp_8b330d02d0_prod0820_fp8_kvcache_20260402', 'id': '1127627a-f2bd-484e-9dd6-0077b76d7c25', 'finish_reason': 'tool_calls', 'logprobs': None}, id='lc_run--019e1a29-4d60-7ac0-8a87-09d221f7ed51-0', tool_calls=[{'name': 'get_wweather', 'args': {'location': '杭州'}, 'id': 'call_00_GWC3Cem86tM2t2wm178U2284', 'type': 'tool_call'}], invalid_tool_calls=[], usage_metadata={'input_tokens': 287, 'output_tokens': 54, 'total_tokens': 341, 'input_token_details': {'cache_read': 256}, 'output_token_details': {}}),

 ToolMessage(content='杭州的天气是晴朗的', name='get_wweather', id='05b469c4-8afc-4dc7-b1cc-a38330c25a74', tool_call_id='call_00_GWC3Cem86tM2t2wm178U2284'), 

AIMessage(content='🌤️ **杭州天气信息**\n\n目前杭州的天气为：**晴朗** ☀️\n\n天气不错，适合外出活动哦！如果您需要更详细的信息（比如温度、风力等），也可以告诉我～', additional_kwargs={'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 44, 'prompt_tokens': 358, 'total_tokens': 402, 'completion_tokens_details': None, 'prompt_tokens_details': {'audio_tokens': None, 'cached_tokens': 256}, 'prompt_cache_hit_tokens': 256, 'prompt_cache_miss_tokens': 102}, 'model_provider': 'deepseek', 'model_name': 'deepseek-v4-flash', 'system_fingerprint': 'fp_8b330d02d0_prod0820_fp8_kvcache_20260402', 'id': '58d7a120-ba04-4ed4-a7fe-714e6f3c51ab', 'finish_reason': 'stop', 'logprobs': None}, id='lc_run--019e1a29-52ca-7ef1-b497-36999a93b8e2-0', tool_calls=[], invalid_tool_calls=[], usage_metadata={'input_tokens': 358, 'output_tokens': 44, 'total_tokens': 402, 'input_token_details': {'cache_read': 256}, 'output_token_details': {}})]}