05_human_in_the_loop.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "id": "53d49299-d23c-4174-96a5-de27d60fc3e1",
   "metadata": {},
   "outputs": [],
   "source": [
    "from dotenv import load_dotenv\n",
    "\n",
    "_ = load_dotenv()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "id": "f1919a49-216f-49f5-8d15-fea3d3e8f3ba",
   "metadata": {},
   "outputs": [],
   "source": [
    "from langgraph.graph import StateGraph, END\n",
    "from typing import TypedDict, Annotated\n",
    "import operator\n",
    "import sqlite3\n",
    "from langchain_core.messages import AnyMessage, SystemMessage, HumanMessage, ToolMessage\n",
    "from langchain_openai import ChatOpenAI\n",
    "from langchain_community.tools.tavily_search import TavilySearchResults\n",
    "from langgraph.checkpoint.sqlite import SqliteSaver\n",
    "\n",
    "\n",
    "db_path = \"checkpoints.db\"\n",
    "conn = sqlite3.connect(db_path, check_same_thread=False)\n",
    "\n",
    "memory = SqliteSaver(conn)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "id": "16f7dd64-44e9-485e-b8de-ed941b48f6b0",
   "metadata": {},
   "outputs": [],
   "source": [
    "from uuid import uuid4\n",
    "from langchain_core.messages import AnyMessage, SystemMessage, HumanMessage, AIMessage\n",
    "\n",
    "\"\"\"\n",
    "In previous examples we've annotated the `messages` state key\n",
    "with the default `operator.add` or `+` reducer, which always\n",
    "appends new messages to the end of the existing messages array.\n",
    "\n",
    "Now, to support replacing existing messages, we annotate the\n",
    "`messages` key with a customer reducer function, which replaces\n",
    "messages with the same `id`, and appends them otherwise.\n",
    "\"\"\"\n",
    "def reduce_messages(left: list[AnyMessage], right: list[AnyMessage]) -> list[AnyMessage]:\n",
    "    # assign ids to messages that don't have them\n",
    "    for message in right:\n",
    "        if not message.id:\n",
    "            message.id = str(uuid4())\n",
    "    # merge the new messages with the existing messages\n",
    "    merged = left.copy()\n",
    "    for message in right:\n",
    "        for i, existing in enumerate(merged):\n",
    "            # replace any existing messages with the same id\n",
    "            if existing.id == message.id:\n",
    "                merged[i] = message\n",
    "                break\n",
    "        else:\n",
    "            # append any new messages to the end\n",
    "            merged.append(message)\n",
    "    return merged\n",
    "\n",
    "class AgentState(TypedDict):\n",
    "    messages: Annotated[list[AnyMessage], reduce_messages]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "id": "24fd3103-9d06-4289-b5de-1a338b8050d5",
   "metadata": {},
   "outputs": [],
   "source": [
    "tool = TavilySearchResults(max_results=2)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "id": "a6aafcb4-e8cc-469d-a65d-2f4b33307904",
   "metadata": {},
   "outputs": [],
   "source": [
    "# Manual Human Approval\n",
    "\n",
    "\n",
    "class Agent:\n",
    "    def __init__(self, model, tools, system=\"\", checkpointer=None):\n",
    "        self.system = system\n",
    "        graph = StateGraph(AgentState)\n",
    "        graph.add_node(\"llm\", self.call_openai)\n",
    "        graph.add_node(\"action\", self.take_action)\n",
    "        graph.add_conditional_edges(\"llm\", self.exists_action, {True: \"action\", False: END})\n",
    "        graph.add_edge(\"action\", \"llm\")\n",
    "        graph.set_entry_point(\"llm\")\n",
    "        self.graph = graph.compile(\n",
    "            checkpointer=checkpointer,\n",
    "            interrupt_before=[\"action\"]\n",
    "        )\n",
    "        self.tools = {t.name: t for t in tools}\n",
    "        self.model = model.bind_tools(tools)\n",
    "\n",
    "    def call_openai(self, state: AgentState):\n",
    "        messages = state['messages']\n",
    "        if self.system:\n",
    "            messages = [SystemMessage(content=self.system)] + messages\n",
    "        message = self.model.invoke(messages)\n",
    "        return {'messages': [message]}\n",
    "\n",
    "    def exists_action(self, state: AgentState):\n",
    "        print(state)\n",
    "        result = state['messages'][-1]\n",
    "        return len(result.tool_calls) > 0\n",
    "\n",
    "    def take_action(self, state: AgentState):\n",
    "        tool_calls = state['messages'][-1].tool_calls\n",
    "        results = []\n",
    "        for t in tool_calls:\n",
    "            print(f\"Calling: {t}\")\n",
    "            result = self.tools[t['name']].invoke(t['args'])\n",
    "            results.append(ToolMessage(tool_call_id=t['id'], name=t['name'], content=str(result)))\n",
    "        print(\"Back to the model!\")\n",
    "        return {'messages': results}"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "id": "a869f9b9-dd60-4b25-b9d7-51bccb06b1a8",
   "metadata": {},
   "outputs": [],
   "source": [
    "prompt = \"\"\"You are a smart research assistant. Use the search engine to look up information. \\\n",
    "You are allowed to make multiple calls (either together or in sequence). \\\n",
    "Only look up information when you are sure of what you want. \\\n",
    "If you need to look up some information before asking a follow up question, you are allowed to do that!\n",
    "\"\"\"\n",
    "model = ChatOpenAI(model=\"gpt-3.5-turbo\")\n",
    "abot = Agent(model, [tool], system=prompt, checkpointer=memory)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 13,
   "id": "3d4eeedb-4767-4374-9152-c59f417fb6f9",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "{'messages': [HumanMessage(content='What is the weather in sf?', additional_kwargs={}, response_metadata={}), AIMessage(content='', additional_kwargs={'tool_calls': [{'id': 'call_Skc1tN346NFsZWtEmFNeOaBF', 'function': {'arguments': '{\"query\":\"San Francisco weather today\"}', 'name': 'tavily_search_results_json'}, 'type': 'function'}], 'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 22, 'prompt_tokens': 151, 'total_tokens': 173, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-4o-2024-08-06', 'system_fingerprint': 'fp_50cad350e4', 'finish_reason': 'tool_calls', 'logprobs': None}, id='run-99627f6c-ec6a-4e1e-8917-db7abcf1b5ee-0', tool_calls=[{'name': 'tavily_search_results_json', 'args': {'query': 'San Francisco weather today'}, 'id': 'call_Skc1tN346NFsZWtEmFNeOaBF', 'type': 'tool_call'}], usage_metadata={'input_tokens': 151, 'output_tokens': 22, 'total_tokens': 173, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}}), ToolMessage(content='[{\\'url\\': \\'https://www.weatherapi.com/\\', \\'content\\': \"{\\'location\\': {\\'name\\': \\'San Francisco\\', \\'region\\': \\'California\\', \\'country\\': \\'United States of America\\', \\'lat\\': 37.775, \\'lon\\': -122.4183, \\'tz_id\\': \\'America/Los_Angeles\\', \\'localtime_epoch\\': 1738378619, \\'localtime\\': \\'2025-01-31 18:56\\'}, \\'current\\': {\\'last_updated_epoch\\': 1738377900, \\'last_updated\\': \\'2025-01-31 18:45\\', \\'temp_c\\': 13.3, \\'temp_f\\': 55.9, \\'is_day\\': 0, \\'condition\\': {\\'text\\': \\'Moderate rain\\', \\'icon\\': \\'//cdn.weatherapi.com/weather/64x64/night/302.png\\', \\'code\\': 1189}, \\'wind_mph\\': 11.4, \\'wind_kph\\': 18.4, \\'wind_degree\\': 154, \\'wind_dir\\': \\'SSE\\', \\'pressure_mb\\': 1022.0, \\'pressure_in\\': 30.19, \\'precip_mm\\': 0.0, \\'precip_in\\': 0.0, \\'humidity\\': 80, \\'cloud\\': 100, \\'feelslike_c\\': 11.7, \\'feelslike_f\\': 53.0, \\'windchill_c\\': 8.6, \\'windchill_f\\': 47.5, \\'heatindex_c\\': 10.7, \\'heatindex_f\\': 51.2, \\'dewpoint_c\\': 10.2, \\'dewpoint_f\\': 50.4, \\'vis_km\\': 16.0, \\'vis_miles\\': 9.0, \\'uv\\': 0.0, \\'gust_mph\\': 17.5, \\'gust_kph\\': 28.2}}\"}, {\\'url\\': \\'https://world-weather.info/forecast/usa/san_francisco/january-2025/\\', \\'content\\': \"Weather in San Francisco in January 2025 (California) - Detailed Weather Forecast for a Month Weather World Weather in San Francisco Weather in San Francisco in January 2025 San Francisco Weather Forecast for January 2025, is based on previous years\\' statistical data. +55°+46° +54°+45° +57°+48° +55°+48° +55°+48° +55°+46° +57°+50° +59°+50° +57°+50° +59°+50° +55°+48° +57°+48° +57°+48° +57°+50° +55°+46° +55°+48° +57°+48° +57°+48° +57°+48° +55°+48° +57°+48° +55°+48° +55°+48° +55°+46° +57°+48° +57°+48° +59°+48° +59°+48° +59°+48° +59°+48° +61°+50° Extended weather forecast in San Francisco HourlyWeek10-Day14-Day30-DayYear Weather in Washington, D.C.+30° Sacramento+45° Pleasanton+50° Redwood City+54° San Leandro+54° San Mateo+54° San Rafael+50° San Ramon+48° South San Francisco+55° Vallejo+50° Palo Alto+52° Pacifica+59° Berkeley+48° Castro Valley+54° Concord+48° Daly City+55° Leona Heights+54° Centennial+54° world\\'s temperature today day day Temperature units\"}]', name='tavily_search_results_json', tool_call_id='call_Skc1tN346NFsZWtEmFNeOaBF'), AIMessage(content='The current weather in San Francisco is moderate rain with a temperature of 13.3°C (55.9°F). The wind is coming from the SSE at 11.4 mph (18.4 kph), and the humidity level is at 80%. The conditions are quite cloudy with a cloud cover of 100%.', additional_kwargs={'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 68, 'prompt_tokens': 975, 'total_tokens': 1043, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-4o-2024-08-06', 'system_fingerprint': 'fp_50cad350e4', 'finish_reason': 'stop', 'logprobs': None}, id='run-c73fd64e-a209-4478-9e53-71aa860675bb-0', usage_metadata={'input_tokens': 975, 'output_tokens': 68, 'total_tokens': 1043, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}}), HumanMessage(content='What about in la?', additional_kwargs={}, response_metadata={}), AIMessage(content='', additional_kwargs={'tool_calls': [{'id': 'call_YEOGS21lpACUuv2X5FxWKN2T', 'function': {'arguments': '{\"query\":\"Los Angeles weather today\"}', 'name': 'tavily_search_results_json'}, 'type': 'function'}], 'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 22, 'prompt_tokens': 1054, 'total_tokens': 1076, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-4o-2024-08-06', 'system_fingerprint': 'fp_50cad350e4', 'finish_reason': 'tool_calls', 'logprobs': None}, id='run-8c40c7fb-b445-4fc9-bf80-5bb2fbdb5011-0', tool_calls=[{'name': 'tavily_search_results_json', 'args': {'query': 'Los Angeles weather today'}, 'id': 'call_YEOGS21lpACUuv2X5FxWKN2T', 'type': 'tool_call'}], usage_metadata={'input_tokens': 1054, 'output_tokens': 22, 'total_tokens': 1076, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}}), ToolMessage(content='[{\\'url\\': \\'https://www.weatherapi.com/\\', \\'content\\': \"{\\'location\\': {\\'name\\': \\'Los Angeles\\', \\'region\\': \\'California\\', \\'country\\': \\'United States of America\\', \\'lat\\': 34.0522, \\'lon\\': -118.2428, \\'tz_id\\': \\'America/Los_Angeles\\', \\'localtime_epoch\\': 1738378720, \\'localtime\\': \\'2025-01-31 18:58\\'}, \\'current\\': {\\'last_updated_epoch\\': 1738377900, \\'last_updated\\': \\'2025-01-31 18:45\\', \\'temp_c\\': 13.3, \\'temp_f\\': 55.9, \\'is_day\\': 0, \\'condition\\': {\\'text\\': \\'Clear\\', \\'icon\\': \\'//cdn.weatherapi.com/weather/64x64/night/113.png\\', \\'code\\': 1000}, \\'wind_mph\\': 6.3, \\'wind_kph\\': 10.1, \\'wind_degree\\': 243, \\'wind_dir\\': \\'WSW\\', \\'pressure_mb\\': 1023.0, \\'pressure_in\\': 30.21, \\'precip_mm\\': 0.0, \\'precip_in\\': 0.0, \\'humidity\\': 75, \\'cloud\\': 0, \\'feelslike_c\\': 12.6, \\'feelslike_f\\': 54.6, \\'windchill_c\\': 16.4, \\'windchill_f\\': 61.6, \\'heatindex_c\\': 16.4, \\'heatindex_f\\': 61.6, \\'dewpoint_c\\': 10.2, \\'dewpoint_f\\': 50.4, \\'vis_km\\': 16.0, \\'vis_miles\\': 9.0, \\'uv\\': 0.0, \\'gust_mph\\': 7.8, \\'gust_kph\\': 12.6}}\"}, {\\'url\\': \\'https://www.easeweather.com/north-america/united-states/california/los-angeles-county/los-angeles/february\\', \\'content\\': \\'Weather in Los Angeles in February 2025 - Detailed Forecast Weather in Los Angeles for February 2025 Your guide to Los Angeles weather in February - trends and predictions The forecast for the first days of February 2025 in Los Angeles predicts temperatures to be around 18\\\\xa0°F, close to the historical average. In general, the average temperature in Los Angeles at the beginning of February is 18.1\\\\xa0°F. Los Angeles experiences moderate rainfall in February, with an average of 10 rainy days and a total precipitation of 79.7\\\\xa0mm. Los Angeles in February average weather Temperatures trend during February in Los Angeles Los Angeles in February - FAQ Get accurate weather forecasts for Los Angeles, located at latitude 34.052 and longitude -118.244.\\'}]', name='tavily_search_results_json', tool_call_id='call_YEOGS21lpACUuv2X5FxWKN2T'), AIMessage(content='The current weather in Los Angeles is clear with a temperature of 13.3°C (55.9°F). The wind is coming from the WSW at 6.3 mph (10.1 kph), and the humidity level is at 75%. The sky is clear with no cloud cover.', additional_kwargs={'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 64, 'prompt_tokens': 1689, 'total_tokens': 1753, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 1024}}, 'model_name': 'gpt-4o-2024-08-06', 'system_fingerprint': 'fp_50cad350e4', 'finish_reason': 'stop', 'logprobs': None}, id='run-ecd5771a-302a-46b9-a27f-4796a1494eca-0', usage_metadata={'input_tokens': 1689, 'output_tokens': 64, 'total_tokens': 1753, 'input_token_details': {'audio': 0, 'cache_read': 1024}, 'output_token_details': {'audio': 0, 'reasoning': 0}}), HumanMessage(content='Which one is warmer?', additional_kwargs={}, response_metadata={}), AIMessage(content='Both San Francisco and Los Angeles currently have the same temperature of 13.3°C (55.9°F), so neither is warmer than the other at this time. However, Los Angeles has clear skies while San Francisco experiences moderate rain, which can affect the perception of warmth.', additional_kwargs={'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 58, 'prompt_tokens': 1764, 'total_tokens': 1822, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 1664}}, 'model_name': 'gpt-4o-2024-08-06', 'system_fingerprint': 'fp_50cad350e4', 'finish_reason': 'stop', 'logprobs': None}, id='run-06951198-6c86-4013-8d34-74bbcfa11d26-0', usage_metadata={'input_tokens': 1764, 'output_tokens': 58, 'total_tokens': 1822, 'input_token_details': {'audio': 0, 'cache_read': 1664}, 'output_token_details': {'audio': 0, 'reasoning': 0}}), HumanMessage(content='Whats the weather in SF?', additional_kwargs={}, response_metadata={}, id='5535cb89-c952-424e-ae44-4eb3a0d13d07'), AIMessage(content='', additional_kwargs={'tool_calls': [{'id': 'call_hrtrcRj5Irr0gvQyJrtYFaTs', 'function': {'arguments': '{\"query\":\"San Francisco weather today\"}', 'name': 'tavily_search_results_json'}, 'type': 'function'}], 'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 22, 'prompt_tokens': 1857, 'total_tokens': 1879, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'tool_calls', 'logprobs': None}, id='run-a562ed94-83d6-407c-bc2e-e9b23483f3fc-0', tool_calls=[{'name': 'tavily_search_results_json', 'args': {'query': 'San Francisco weather today'}, 'id': 'call_hrtrcRj5Irr0gvQyJrtYFaTs', 'type': 'tool_call'}], usage_metadata={'input_tokens': 1857, 'output_tokens': 22, 'total_tokens': 1879, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}})]}\n",
      "{'messages': [AIMessage(content='', additional_kwargs={'tool_calls': [{'id': 'call_hrtrcRj5Irr0gvQyJrtYFaTs', 'function': {'arguments': '{\"query\":\"San Francisco weather today\"}', 'name': 'tavily_search_results_json'}, 'type': 'function'}], 'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 22, 'prompt_tokens': 1857, 'total_tokens': 1879, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'tool_calls', 'logprobs': None}, id='run-a562ed94-83d6-407c-bc2e-e9b23483f3fc-0', tool_calls=[{'name': 'tavily_search_results_json', 'args': {'query': 'San Francisco weather today'}, 'id': 'call_hrtrcRj5Irr0gvQyJrtYFaTs', 'type': 'tool_call'}], usage_metadata={'input_tokens': 1857, 'output_tokens': 22, 'total_tokens': 1879, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}})]}\n",
      "()\n"
     ]
    }
   ],
   "source": [
    "messages = [HumanMessage(content=\"Whats the weather in SF?\")]\n",
    "thread = {\"configurable\": {\"thread_id\": \"1\"}}\n",
    "for event in abot.graph.stream({\"messages\": messages}, thread):\n",
    "    for v in event.values():\n",
    "        print(v)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 14,
   "id": "c5ccf605-6795-4cd3-b6a3-a6c07a2a95df",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "StateSnapshot(values={'messages': [HumanMessage(content='What is the weather in sf?', additional_kwargs={}, response_metadata={}), AIMessage(content='', additional_kwargs={'tool_calls': [{'id': 'call_Skc1tN346NFsZWtEmFNeOaBF', 'function': {'arguments': '{\"query\":\"San Francisco weather today\"}', 'name': 'tavily_search_results_json'}, 'type': 'function'}], 'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 22, 'prompt_tokens': 151, 'total_tokens': 173, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-4o-2024-08-06', 'system_fingerprint': 'fp_50cad350e4', 'finish_reason': 'tool_calls', 'logprobs': None}, id='run-99627f6c-ec6a-4e1e-8917-db7abcf1b5ee-0', tool_calls=[{'name': 'tavily_search_results_json', 'args': {'query': 'San Francisco weather today'}, 'id': 'call_Skc1tN346NFsZWtEmFNeOaBF', 'type': 'tool_call'}], usage_metadata={'input_tokens': 151, 'output_tokens': 22, 'total_tokens': 173, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}}), ToolMessage(content='[{\\'url\\': \\'https://www.weatherapi.com/\\', \\'content\\': \"{\\'location\\': {\\'name\\': \\'San Francisco\\', \\'region\\': \\'California\\', \\'country\\': \\'United States of America\\', \\'lat\\': 37.775, \\'lon\\': -122.4183, \\'tz_id\\': \\'America/Los_Angeles\\', \\'localtime_epoch\\': 1738378619, \\'localtime\\': \\'2025-01-31 18:56\\'}, \\'current\\': {\\'last_updated_epoch\\': 1738377900, \\'last_updated\\': \\'2025-01-31 18:45\\', \\'temp_c\\': 13.3, \\'temp_f\\': 55.9, \\'is_day\\': 0, \\'condition\\': {\\'text\\': \\'Moderate rain\\', \\'icon\\': \\'//cdn.weatherapi.com/weather/64x64/night/302.png\\', \\'code\\': 1189}, \\'wind_mph\\': 11.4, \\'wind_kph\\': 18.4, \\'wind_degree\\': 154, \\'wind_dir\\': \\'SSE\\', \\'pressure_mb\\': 1022.0, \\'pressure_in\\': 30.19, \\'precip_mm\\': 0.0, \\'precip_in\\': 0.0, \\'humidity\\': 80, \\'cloud\\': 100, \\'feelslike_c\\': 11.7, \\'feelslike_f\\': 53.0, \\'windchill_c\\': 8.6, \\'windchill_f\\': 47.5, \\'heatindex_c\\': 10.7, \\'heatindex_f\\': 51.2, \\'dewpoint_c\\': 10.2, \\'dewpoint_f\\': 50.4, \\'vis_km\\': 16.0, \\'vis_miles\\': 9.0, \\'uv\\': 0.0, \\'gust_mph\\': 17.5, \\'gust_kph\\': 28.2}}\"}, {\\'url\\': \\'https://world-weather.info/forecast/usa/san_francisco/january-2025/\\', \\'content\\': \"Weather in San Francisco in January 2025 (California) - Detailed Weather Forecast for a Month Weather World Weather in San Francisco Weather in San Francisco in January 2025 San Francisco Weather Forecast for January 2025, is based on previous years\\' statistical data. +55°+46° +54°+45° +57°+48° +55°+48° +55°+48° +55°+46° +57°+50° +59°+50° +57°+50° +59°+50° +55°+48° +57°+48° +57°+48° +57°+50° +55°+46° +55°+48° +57°+48° +57°+48° +57°+48° +55°+48° +57°+48° +55°+48° +55°+48° +55°+46° +57°+48° +57°+48° +59°+48° +59°+48° +59°+48° +59°+48° +61°+50° Extended weather forecast in San Francisco HourlyWeek10-Day14-Day30-DayYear Weather in Washington, D.C.+30° Sacramento+45° Pleasanton+50° Redwood City+54° San Leandro+54° San Mateo+54° San Rafael+50° San Ramon+48° South San Francisco+55° Vallejo+50° Palo Alto+52° Pacifica+59° Berkeley+48° Castro Valley+54° Concord+48° Daly City+55° Leona Heights+54° Centennial+54° world\\'s temperature today day day Temperature units\"}]', name='tavily_search_results_json', tool_call_id='call_Skc1tN346NFsZWtEmFNeOaBF'), AIMessage(content='The current weather in San Francisco is moderate rain with a temperature of 13.3°C (55.9°F). The wind is coming from the SSE at 11.4 mph (18.4 kph), and the humidity level is at 80%. The conditions are quite cloudy with a cloud cover of 100%.', additional_kwargs={'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 68, 'prompt_tokens': 975, 'total_tokens': 1043, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-4o-2024-08-06', 'system_fingerprint': 'fp_50cad350e4', 'finish_reason': 'stop', 'logprobs': None}, id='run-c73fd64e-a209-4478-9e53-71aa860675bb-0', usage_metadata={'input_tokens': 975, 'output_tokens': 68, 'total_tokens': 1043, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}}), HumanMessage(content='What about in la?', additional_kwargs={}, response_metadata={}), AIMessage(content='', additional_kwargs={'tool_calls': [{'id': 'call_YEOGS21lpACUuv2X5FxWKN2T', 'function': {'arguments': '{\"query\":\"Los Angeles weather today\"}', 'name': 'tavily_search_results_json'}, 'type': 'function'}], 'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 22, 'prompt_tokens': 1054, 'total_tokens': 1076, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-4o-2024-08-06', 'system_fingerprint': 'fp_50cad350e4', 'finish_reason': 'tool_calls', 'logprobs': None}, id='run-8c40c7fb-b445-4fc9-bf80-5bb2fbdb5011-0', tool_calls=[{'name': 'tavily_search_results_json', 'args': {'query': 'Los Angeles weather today'}, 'id': 'call_YEOGS21lpACUuv2X5FxWKN2T', 'type': 'tool_call'}], usage_metadata={'input_tokens': 1054, 'output_tokens': 22, 'total_tokens': 1076, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}}), ToolMessage(content='[{\\'url\\': \\'https://www.weatherapi.com/\\', \\'content\\': \"{\\'location\\': {\\'name\\': \\'Los Angeles\\', \\'region\\': \\'California\\', \\'country\\': \\'United States of America\\', \\'lat\\': 34.0522, \\'lon\\': -118.2428, \\'tz_id\\': \\'America/Los_Angeles\\', \\'localtime_epoch\\': 1738378720, \\'localtime\\': \\'2025-01-31 18:58\\'}, \\'current\\': {\\'last_updated_epoch\\': 1738377900, \\'last_updated\\': \\'2025-01-31 18:45\\', \\'temp_c\\': 13.3, \\'temp_f\\': 55.9, \\'is_day\\': 0, \\'condition\\': {\\'text\\': \\'Clear\\', \\'icon\\': \\'//cdn.weatherapi.com/weather/64x64/night/113.png\\', \\'code\\': 1000}, \\'wind_mph\\': 6.3, \\'wind_kph\\': 10.1, \\'wind_degree\\': 243, \\'wind_dir\\': \\'WSW\\', \\'pressure_mb\\': 1023.0, \\'pressure_in\\': 30.21, \\'precip_mm\\': 0.0, \\'precip_in\\': 0.0, \\'humidity\\': 75, \\'cloud\\': 0, \\'feelslike_c\\': 12.6, \\'feelslike_f\\': 54.6, \\'windchill_c\\': 16.4, \\'windchill_f\\': 61.6, \\'heatindex_c\\': 16.4, \\'heatindex_f\\': 61.6, \\'dewpoint_c\\': 10.2, \\'dewpoint_f\\': 50.4, \\'vis_km\\': 16.0, \\'vis_miles\\': 9.0, \\'uv\\': 0.0, \\'gust_mph\\': 7.8, \\'gust_kph\\': 12.6}}\"}, {\\'url\\': \\'https://www.easeweather.com/north-america/united-states/california/los-angeles-county/los-angeles/february\\', \\'content\\': \\'Weather in Los Angeles in February 2025 - Detailed Forecast Weather in Los Angeles for February 2025 Your guide to Los Angeles weather in February - trends and predictions The forecast for the first days of February 2025 in Los Angeles predicts temperatures to be around 18\\\\xa0°F, close to the historical average. In general, the average temperature in Los Angeles at the beginning of February is 18.1\\\\xa0°F. Los Angeles experiences moderate rainfall in February, with an average of 10 rainy days and a total precipitation of 79.7\\\\xa0mm. Los Angeles in February average weather Temperatures trend during February in Los Angeles Los Angeles in February - FAQ Get accurate weather forecasts for Los Angeles, located at latitude 34.052 and longitude -118.244.\\'}]', name='tavily_search_results_json', tool_call_id='call_YEOGS21lpACUuv2X5FxWKN2T'), AIMessage(content='The current weather in Los Angeles is clear with a temperature of 13.3°C (55.9°F). The wind is coming from the WSW at 6.3 mph (10.1 kph), and the humidity level is at 75%. The sky is clear with no cloud cover.', additional_kwargs={'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 64, 'prompt_tokens': 1689, 'total_tokens': 1753, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 1024}}, 'model_name': 'gpt-4o-2024-08-06', 'system_fingerprint': 'fp_50cad350e4', 'finish_reason': 'stop', 'logprobs': None}, id='run-ecd5771a-302a-46b9-a27f-4796a1494eca-0', usage_metadata={'input_tokens': 1689, 'output_tokens': 64, 'total_tokens': 1753, 'input_token_details': {'audio': 0, 'cache_read': 1024}, 'output_token_details': {'audio': 0, 'reasoning': 0}}), HumanMessage(content='Which one is warmer?', additional_kwargs={}, response_metadata={}), AIMessage(content='Both San Francisco and Los Angeles currently have the same temperature of 13.3°C (55.9°F), so neither is warmer than the other at this time. However, Los Angeles has clear skies while San Francisco experiences moderate rain, which can affect the perception of warmth.', additional_kwargs={'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 58, 'prompt_tokens': 1764, 'total_tokens': 1822, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 1664}}, 'model_name': 'gpt-4o-2024-08-06', 'system_fingerprint': 'fp_50cad350e4', 'finish_reason': 'stop', 'logprobs': None}, id='run-06951198-6c86-4013-8d34-74bbcfa11d26-0', usage_metadata={'input_tokens': 1764, 'output_tokens': 58, 'total_tokens': 1822, 'input_token_details': {'audio': 0, 'cache_read': 1664}, 'output_token_details': {'audio': 0, 'reasoning': 0}}), HumanMessage(content='Whats the weather in SF?', additional_kwargs={}, response_metadata={}, id='5535cb89-c952-424e-ae44-4eb3a0d13d07'), AIMessage(content='', additional_kwargs={'tool_calls': [{'id': 'call_hrtrcRj5Irr0gvQyJrtYFaTs', 'function': {'arguments': '{\"query\":\"San Francisco weather today\"}', 'name': 'tavily_search_results_json'}, 'type': 'function'}], 'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 22, 'prompt_tokens': 1857, 'total_tokens': 1879, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'tool_calls', 'logprobs': None}, id='run-a562ed94-83d6-407c-bc2e-e9b23483f3fc-0', tool_calls=[{'name': 'tavily_search_results_json', 'args': {'query': 'San Francisco weather today'}, 'id': 'call_hrtrcRj5Irr0gvQyJrtYFaTs', 'type': 'tool_call'}], usage_metadata={'input_tokens': 1857, 'output_tokens': 22, 'total_tokens': 1879, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}})]}, next=('action',), config={'configurable': {'thread_id': '1', 'checkpoint_ns': '', 'checkpoint_id': '1efe0f6c-6ba3-6e6b-800e-41898c1626e5'}}, metadata={'source': 'loop', 'writes': {'llm': {'messages': [AIMessage(content='', additional_kwargs={'tool_calls': [{'id': 'call_hrtrcRj5Irr0gvQyJrtYFaTs', 'function': {'arguments': '{\"query\":\"San Francisco weather today\"}', 'name': 'tavily_search_results_json'}, 'type': 'function'}], 'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 22, 'prompt_tokens': 1857, 'total_tokens': 1879, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'tool_calls', 'logprobs': None}, id='run-a562ed94-83d6-407c-bc2e-e9b23483f3fc-0', tool_calls=[{'name': 'tavily_search_results_json', 'args': {'query': 'San Francisco weather today'}, 'id': 'call_hrtrcRj5Irr0gvQyJrtYFaTs', 'type': 'tool_call'}], usage_metadata={'input_tokens': 1857, 'output_tokens': 22, 'total_tokens': 1879, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}})]}}, 'thread_id': '1', 'step': 14, 'parents': {}}, created_at='2025-02-01T23:46:37.244538+00:00', parent_config={'configurable': {'thread_id': '1', 'checkpoint_ns': '', 'checkpoint_id': '1efe0f6c-2a99-6e96-800d-cde0eae2173a'}}, tasks=(PregelTask(id='60da94e7-439c-bc84-9e3f-43d550681de6', name='action', path=('__pregel_pull', 'action'), error=None, interrupts=(), state=None, result=None),))"
      ]
     },
     "execution_count": 14,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "abot.graph.get_state(thread)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 15,
   "id": "c0fa39b1-d0e0-471b-99e7-9f827329382d",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "('action',)"
      ]
     },
     "execution_count": 15,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "abot.graph.get_state(thread).next"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 16,
   "id": "30ee8da4-9184-47ad-b2ad-eb5f0554489e",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Calling: {'name': 'tavily_search_results_json', 'args': {'query': 'San Francisco weather today'}, 'id': 'call_hrtrcRj5Irr0gvQyJrtYFaTs', 'type': 'tool_call'}\n",
      "Back to the model!\n",
      "{'messages': [ToolMessage(content='[{\\'url\\': \\'https://www.weatherapi.com/\\', \\'content\\': \"{\\'location\\': {\\'name\\': \\'San Francisco\\', \\'region\\': \\'California\\', \\'country\\': \\'United States of America\\', \\'lat\\': 37.775, \\'lon\\': -122.4183, \\'tz_id\\': \\'America/Los_Angeles\\', \\'localtime_epoch\\': 1738453598, \\'localtime\\': \\'2025-02-01 15:46\\'}, \\'current\\': {\\'last_updated_epoch\\': 1738453500, \\'last_updated\\': \\'2025-02-01 15:45\\', \\'temp_c\\': 15.6, \\'temp_f\\': 60.1, \\'is_day\\': 1, \\'condition\\': {\\'text\\': \\'Overcast\\', \\'icon\\': \\'//cdn.weatherapi.com/weather/64x64/day/122.png\\', \\'code\\': 1009}, \\'wind_mph\\': 19.0, \\'wind_kph\\': 30.6, \\'wind_degree\\': 207, \\'wind_dir\\': \\'SSW\\', \\'pressure_mb\\': 1018.0, \\'pressure_in\\': 30.06, \\'precip_mm\\': 0.0, \\'precip_in\\': 0.0, \\'humidity\\': 86, \\'cloud\\': 100, \\'feelslike_c\\': 15.6, \\'feelslike_f\\': 60.1, \\'windchill_c\\': 10.6, \\'windchill_f\\': 51.0, \\'heatindex_c\\': 12.7, \\'heatindex_f\\': 54.8, \\'dewpoint_c\\': 12.4, \\'dewpoint_f\\': 54.2, \\'vis_km\\': 13.0, \\'vis_miles\\': 8.0, \\'uv\\': 0.1, \\'gust_mph\\': 31.3, \\'gust_kph\\': 50.4}}\"}, {\\'url\\': \\'https://www.easeweather.com/north-america/united-states/california/city-and-county-of-san-francisco/february\\', \\'content\\': \\'February 2025 Weather - City and County of San Francisco Weather in City and County of San Francisco for February 2025 Your guide to City and County of San Francisco weather in February - trends and predictions The forecast for the first days of February 2025 in City and County of San Francisco predicts temperatures to be around 12\\\\xa0°F, close to the historical average. In general, the average temperature in City and County of San Francisco at the beginning of February is 12.6\\\\xa0°F. City and County of San Francisco in February average weather Temperatures trend during February in City and County of San Francisco City and County of San Francisco in February - FAQ\\'}]', name='tavily_search_results_json', id='d8d8d49b-ce93-4dea-83bd-7aa79263666e', tool_call_id='call_hrtrcRj5Irr0gvQyJrtYFaTs')]}\n",
      "{'messages': [HumanMessage(content='What is the weather in sf?', additional_kwargs={}, response_metadata={}), AIMessage(content='', additional_kwargs={'tool_calls': [{'id': 'call_Skc1tN346NFsZWtEmFNeOaBF', 'function': {'arguments': '{\"query\":\"San Francisco weather today\"}', 'name': 'tavily_search_results_json'}, 'type': 'function'}], 'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 22, 'prompt_tokens': 151, 'total_tokens': 173, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-4o-2024-08-06', 'system_fingerprint': 'fp_50cad350e4', 'finish_reason': 'tool_calls', 'logprobs': None}, id='run-99627f6c-ec6a-4e1e-8917-db7abcf1b5ee-0', tool_calls=[{'name': 'tavily_search_results_json', 'args': {'query': 'San Francisco weather today'}, 'id': 'call_Skc1tN346NFsZWtEmFNeOaBF', 'type': 'tool_call'}], usage_metadata={'input_tokens': 151, 'output_tokens': 22, 'total_tokens': 173, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}}), ToolMessage(content='[{\\'url\\': \\'https://www.weatherapi.com/\\', \\'content\\': \"{\\'location\\': {\\'name\\': \\'San Francisco\\', \\'region\\': \\'California\\', \\'country\\': \\'United States of America\\', \\'lat\\': 37.775, \\'lon\\': -122.4183, \\'tz_id\\': \\'America/Los_Angeles\\', \\'localtime_epoch\\': 1738378619, \\'localtime\\': \\'2025-01-31 18:56\\'}, \\'current\\': {\\'last_updated_epoch\\': 1738377900, \\'last_updated\\': \\'2025-01-31 18:45\\', \\'temp_c\\': 13.3, \\'temp_f\\': 55.9, \\'is_day\\': 0, \\'condition\\': {\\'text\\': \\'Moderate rain\\', \\'icon\\': \\'//cdn.weatherapi.com/weather/64x64/night/302.png\\', \\'code\\': 1189}, \\'wind_mph\\': 11.4, \\'wind_kph\\': 18.4, \\'wind_degree\\': 154, \\'wind_dir\\': \\'SSE\\', \\'pressure_mb\\': 1022.0, \\'pressure_in\\': 30.19, \\'precip_mm\\': 0.0, \\'precip_in\\': 0.0, \\'humidity\\': 80, \\'cloud\\': 100, \\'feelslike_c\\': 11.7, \\'feelslike_f\\': 53.0, \\'windchill_c\\': 8.6, \\'windchill_f\\': 47.5, \\'heatindex_c\\': 10.7, \\'heatindex_f\\': 51.2, \\'dewpoint_c\\': 10.2, \\'dewpoint_f\\': 50.4, \\'vis_km\\': 16.0, \\'vis_miles\\': 9.0, \\'uv\\': 0.0, \\'gust_mph\\': 17.5, \\'gust_kph\\': 28.2}}\"}, {\\'url\\': \\'https://world-weather.info/forecast/usa/san_francisco/january-2025/\\', \\'content\\': \"Weather in San Francisco in January 2025 (California) - Detailed Weather Forecast for a Month Weather World Weather in San Francisco Weather in San Francisco in January 2025 San Francisco Weather Forecast for January 2025, is based on previous years\\' statistical data. +55°+46° +54°+45° +57°+48° +55°+48° +55°+48° +55°+46° +57°+50° +59°+50° +57°+50° +59°+50° +55°+48° +57°+48° +57°+48° +57°+50° +55°+46° +55°+48° +57°+48° +57°+48° +57°+48° +55°+48° +57°+48° +55°+48° +55°+48° +55°+46° +57°+48° +57°+48° +59°+48° +59°+48° +59°+48° +59°+48° +61°+50° Extended weather forecast in San Francisco HourlyWeek10-Day14-Day30-DayYear Weather in Washington, D.C.+30° Sacramento+45° Pleasanton+50° Redwood City+54° San Leandro+54° San Mateo+54° San Rafael+50° San Ramon+48° South San Francisco+55° Vallejo+50° Palo Alto+52° Pacifica+59° Berkeley+48° Castro Valley+54° Concord+48° Daly City+55° Leona Heights+54° Centennial+54° world\\'s temperature today day day Temperature units\"}]', name='tavily_search_results_json', tool_call_id='call_Skc1tN346NFsZWtEmFNeOaBF'), AIMessage(content='The current weather in San Francisco is moderate rain with a temperature of 13.3°C (55.9°F). The wind is coming from the SSE at 11.4 mph (18.4 kph), and the humidity level is at 80%. The conditions are quite cloudy with a cloud cover of 100%.', additional_kwargs={'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 68, 'prompt_tokens': 975, 'total_tokens': 1043, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-4o-2024-08-06', 'system_fingerprint': 'fp_50cad350e4', 'finish_reason': 'stop', 'logprobs': None}, id='run-c73fd64e-a209-4478-9e53-71aa860675bb-0', usage_metadata={'input_tokens': 975, 'output_tokens': 68, 'total_tokens': 1043, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}}), HumanMessage(content='What about in la?', additional_kwargs={}, response_metadata={}), AIMessage(content='', additional_kwargs={'tool_calls': [{'id': 'call_YEOGS21lpACUuv2X5FxWKN2T', 'function': {'arguments': '{\"query\":\"Los Angeles weather today\"}', 'name': 'tavily_search_results_json'}, 'type': 'function'}], 'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 22, 'prompt_tokens': 1054, 'total_tokens': 1076, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-4o-2024-08-06', 'system_fingerprint': 'fp_50cad350e4', 'finish_reason': 'tool_calls', 'logprobs': None}, id='run-8c40c7fb-b445-4fc9-bf80-5bb2fbdb5011-0', tool_calls=[{'name': 'tavily_search_results_json', 'args': {'query': 'Los Angeles weather today'}, 'id': 'call_YEOGS21lpACUuv2X5FxWKN2T', 'type': 'tool_call'}], usage_metadata={'input_tokens': 1054, 'output_tokens': 22, 'total_tokens': 1076, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}}), ToolMessage(content='[{\\'url\\': \\'https://www.weatherapi.com/\\', \\'content\\': \"{\\'location\\': {\\'name\\': \\'Los Angeles\\', \\'region\\': \\'California\\', \\'country\\': \\'United States of America\\', \\'lat\\': 34.0522, \\'lon\\': -118.2428, \\'tz_id\\': \\'America/Los_Angeles\\', \\'localtime_epoch\\': 1738378720, \\'localtime\\': \\'2025-01-31 18:58\\'}, \\'current\\': {\\'last_updated_epoch\\': 1738377900, \\'last_updated\\': \\'2025-01-31 18:45\\', \\'temp_c\\': 13.3, \\'temp_f\\': 55.9, \\'is_day\\': 0, \\'condition\\': {\\'text\\': \\'Clear\\', \\'icon\\': \\'//cdn.weatherapi.com/weather/64x64/night/113.png\\', \\'code\\': 1000}, \\'wind_mph\\': 6.3, \\'wind_kph\\': 10.1, \\'wind_degree\\': 243, \\'wind_dir\\': \\'WSW\\', \\'pressure_mb\\': 1023.0, \\'pressure_in\\': 30.21, \\'precip_mm\\': 0.0, \\'precip_in\\': 0.0, \\'humidity\\': 75, \\'cloud\\': 0, \\'feelslike_c\\': 12.6, \\'feelslike_f\\': 54.6, \\'windchill_c\\': 16.4, \\'windchill_f\\': 61.6, \\'heatindex_c\\': 16.4, \\'heatindex_f\\': 61.6, \\'dewpoint_c\\': 10.2, \\'dewpoint_f\\': 50.4, \\'vis_km\\': 16.0, \\'vis_miles\\': 9.0, \\'uv\\': 0.0, \\'gust_mph\\': 7.8, \\'gust_kph\\': 12.6}}\"}, {\\'url\\': \\'https://www.easeweather.com/north-america/united-states/california/los-angeles-county/los-angeles/february\\', \\'content\\': \\'Weather in Los Angeles in February 2025 - Detailed Forecast Weather in Los Angeles for February 2025 Your guide to Los Angeles weather in February - trends and predictions The forecast for the first days of February 2025 in Los Angeles predicts temperatures to be around 18\\\\xa0°F, close to the historical average. In general, the average temperature in Los Angeles at the beginning of February is 18.1\\\\xa0°F. Los Angeles experiences moderate rainfall in February, with an average of 10 rainy days and a total precipitation of 79.7\\\\xa0mm. Los Angeles in February average weather Temperatures trend during February in Los Angeles Los Angeles in February - FAQ Get accurate weather forecasts for Los Angeles, located at latitude 34.052 and longitude -118.244.\\'}]', name='tavily_search_results_json', tool_call_id='call_YEOGS21lpACUuv2X5FxWKN2T'), AIMessage(content='The current weather in Los Angeles is clear with a temperature of 13.3°C (55.9°F). The wind is coming from the WSW at 6.3 mph (10.1 kph), and the humidity level is at 75%. The sky is clear with no cloud cover.', additional_kwargs={'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 64, 'prompt_tokens': 1689, 'total_tokens': 1753, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 1024}}, 'model_name': 'gpt-4o-2024-08-06', 'system_fingerprint': 'fp_50cad350e4', 'finish_reason': 'stop', 'logprobs': None}, id='run-ecd5771a-302a-46b9-a27f-4796a1494eca-0', usage_metadata={'input_tokens': 1689, 'output_tokens': 64, 'total_tokens': 1753, 'input_token_details': {'audio': 0, 'cache_read': 1024}, 'output_token_details': {'audio': 0, 'reasoning': 0}}), HumanMessage(content='Which one is warmer?', additional_kwargs={}, response_metadata={}), AIMessage(content='Both San Francisco and Los Angeles currently have the same temperature of 13.3°C (55.9°F), so neither is warmer than the other at this time. However, Los Angeles has clear skies while San Francisco experiences moderate rain, which can affect the perception of warmth.', additional_kwargs={'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 58, 'prompt_tokens': 1764, 'total_tokens': 1822, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 1664}}, 'model_name': 'gpt-4o-2024-08-06', 'system_fingerprint': 'fp_50cad350e4', 'finish_reason': 'stop', 'logprobs': None}, id='run-06951198-6c86-4013-8d34-74bbcfa11d26-0', usage_metadata={'input_tokens': 1764, 'output_tokens': 58, 'total_tokens': 1822, 'input_token_details': {'audio': 0, 'cache_read': 1664}, 'output_token_details': {'audio': 0, 'reasoning': 0}}), HumanMessage(content='Whats the weather in SF?', additional_kwargs={}, response_metadata={}, id='5535cb89-c952-424e-ae44-4eb3a0d13d07'), AIMessage(content='', additional_kwargs={'tool_calls': [{'id': 'call_hrtrcRj5Irr0gvQyJrtYFaTs', 'function': {'arguments': '{\"query\":\"San Francisco weather today\"}', 'name': 'tavily_search_results_json'}, 'type': 'function'}], 'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 22, 'prompt_tokens': 1857, 'total_tokens': 1879, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'tool_calls', 'logprobs': None}, id='run-a562ed94-83d6-407c-bc2e-e9b23483f3fc-0', tool_calls=[{'name': 'tavily_search_results_json', 'args': {'query': 'San Francisco weather today'}, 'id': 'call_hrtrcRj5Irr0gvQyJrtYFaTs', 'type': 'tool_call'}], usage_metadata={'input_tokens': 1857, 'output_tokens': 22, 'total_tokens': 1879, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}}), ToolMessage(content='[{\\'url\\': \\'https://www.weatherapi.com/\\', \\'content\\': \"{\\'location\\': {\\'name\\': \\'San Francisco\\', \\'region\\': \\'California\\', \\'country\\': \\'United States of America\\', \\'lat\\': 37.775, \\'lon\\': -122.4183, \\'tz_id\\': \\'America/Los_Angeles\\', \\'localtime_epoch\\': 1738453598, \\'localtime\\': \\'2025-02-01 15:46\\'}, \\'current\\': {\\'last_updated_epoch\\': 1738453500, \\'last_updated\\': \\'2025-02-01 15:45\\', \\'temp_c\\': 15.6, \\'temp_f\\': 60.1, \\'is_day\\': 1, \\'condition\\': {\\'text\\': \\'Overcast\\', \\'icon\\': \\'//cdn.weatherapi.com/weather/64x64/day/122.png\\', \\'code\\': 1009}, \\'wind_mph\\': 19.0, \\'wind_kph\\': 30.6, \\'wind_degree\\': 207, \\'wind_dir\\': \\'SSW\\', \\'pressure_mb\\': 1018.0, \\'pressure_in\\': 30.06, \\'precip_mm\\': 0.0, \\'precip_in\\': 0.0, \\'humidity\\': 86, \\'cloud\\': 100, \\'feelslike_c\\': 15.6, \\'feelslike_f\\': 60.1, \\'windchill_c\\': 10.6, \\'windchill_f\\': 51.0, \\'heatindex_c\\': 12.7, \\'heatindex_f\\': 54.8, \\'dewpoint_c\\': 12.4, \\'dewpoint_f\\': 54.2, \\'vis_km\\': 13.0, \\'vis_miles\\': 8.0, \\'uv\\': 0.1, \\'gust_mph\\': 31.3, \\'gust_kph\\': 50.4}}\"}, {\\'url\\': \\'https://www.easeweather.com/north-america/united-states/california/city-and-county-of-san-francisco/february\\', \\'content\\': \\'February 2025 Weather - City and County of San Francisco Weather in City and County of San Francisco for February 2025 Your guide to City and County of San Francisco weather in February - trends and predictions The forecast for the first days of February 2025 in City and County of San Francisco predicts temperatures to be around 12\\\\xa0°F, close to the historical average. In general, the average temperature in City and County of San Francisco at the beginning of February is 12.6\\\\xa0°F. City and County of San Francisco in February average weather Temperatures trend during February in City and County of San Francisco City and County of San Francisco in February - FAQ\\'}]', name='tavily_search_results_json', id='d8d8d49b-ce93-4dea-83bd-7aa79263666e', tool_call_id='call_hrtrcRj5Irr0gvQyJrtYFaTs'), AIMessage(content='The current weather in San Francisco is overcast with a temperature of 15.6°C (60.1°F). The wind is coming from the SSW at 19.0 mph (30.6 kph), and the humidity level is at 86%. The sky is completely covered with clouds at 100% cloud cover.', additional_kwargs={'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 70, 'prompt_tokens': 2476, 'total_tokens': 2546, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'stop', 'logprobs': None}, id='run-12f99469-8e00-4d97-9d02-c8488c44e329-0', usage_metadata={'input_tokens': 2476, 'output_tokens': 70, 'total_tokens': 2546, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}})]}\n",
      "{'messages': [AIMessage(content='The current weather in San Francisco is overcast with a temperature of 15.6°C (60.1°F). The wind is coming from the SSW at 19.0 mph (30.6 kph), and the humidity level is at 86%. The sky is completely covered with clouds at 100% cloud cover.', additional_kwargs={'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 70, 'prompt_tokens': 2476, 'total_tokens': 2546, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'stop', 'logprobs': None}, id='run-12f99469-8e00-4d97-9d02-c8488c44e329-0', usage_metadata={'input_tokens': 2476, 'output_tokens': 70, 'total_tokens': 2546, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}})]}\n"
     ]
    }
   ],
   "source": [
    "# Continue after interruption\n",
    "\n",
    "for event in abot.graph.stream(None, thread):\n",
    "    for v in event.values():\n",
    "        print(v)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 17,
   "id": "e6564258-7abd-4ed9-9675-96d4b9edf943",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "StateSnapshot(values={'messages': [HumanMessage(content='What is the weather in sf?', additional_kwargs={}, response_metadata={}), AIMessage(content='', additional_kwargs={'tool_calls': [{'id': 'call_Skc1tN346NFsZWtEmFNeOaBF', 'function': {'arguments': '{\"query\":\"San Francisco weather today\"}', 'name': 'tavily_search_results_json'}, 'type': 'function'}], 'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 22, 'prompt_tokens': 151, 'total_tokens': 173, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-4o-2024-08-06', 'system_fingerprint': 'fp_50cad350e4', 'finish_reason': 'tool_calls', 'logprobs': None}, id='run-99627f6c-ec6a-4e1e-8917-db7abcf1b5ee-0', tool_calls=[{'name': 'tavily_search_results_json', 'args': {'query': 'San Francisco weather today'}, 'id': 'call_Skc1tN346NFsZWtEmFNeOaBF', 'type': 'tool_call'}], usage_metadata={'input_tokens': 151, 'output_tokens': 22, 'total_tokens': 173, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}}), ToolMessage(content='[{\\'url\\': \\'https://www.weatherapi.com/\\', \\'content\\': \"{\\'location\\': {\\'name\\': \\'San Francisco\\', \\'region\\': \\'California\\', \\'country\\': \\'United States of America\\', \\'lat\\': 37.775, \\'lon\\': -122.4183, \\'tz_id\\': \\'America/Los_Angeles\\', \\'localtime_epoch\\': 1738378619, \\'localtime\\': \\'2025-01-31 18:56\\'}, \\'current\\': {\\'last_updated_epoch\\': 1738377900, \\'last_updated\\': \\'2025-01-31 18:45\\', \\'temp_c\\': 13.3, \\'temp_f\\': 55.9, \\'is_day\\': 0, \\'condition\\': {\\'text\\': \\'Moderate rain\\', \\'icon\\': \\'//cdn.weatherapi.com/weather/64x64/night/302.png\\', \\'code\\': 1189}, \\'wind_mph\\': 11.4, \\'wind_kph\\': 18.4, \\'wind_degree\\': 154, \\'wind_dir\\': \\'SSE\\', \\'pressure_mb\\': 1022.0, \\'pressure_in\\': 30.19, \\'precip_mm\\': 0.0, \\'precip_in\\': 0.0, \\'humidity\\': 80, \\'cloud\\': 100, \\'feelslike_c\\': 11.7, \\'feelslike_f\\': 53.0, \\'windchill_c\\': 8.6, \\'windchill_f\\': 47.5, \\'heatindex_c\\': 10.7, \\'heatindex_f\\': 51.2, \\'dewpoint_c\\': 10.2, \\'dewpoint_f\\': 50.4, \\'vis_km\\': 16.0, \\'vis_miles\\': 9.0, \\'uv\\': 0.0, \\'gust_mph\\': 17.5, \\'gust_kph\\': 28.2}}\"}, {\\'url\\': \\'https://world-weather.info/forecast/usa/san_francisco/january-2025/\\', \\'content\\': \"Weather in San Francisco in January 2025 (California) - Detailed Weather Forecast for a Month Weather World Weather in San Francisco Weather in San Francisco in January 2025 San Francisco Weather Forecast for January 2025, is based on previous years\\' statistical data. +55°+46° +54°+45° +57°+48° +55°+48° +55°+48° +55°+46° +57°+50° +59°+50° +57°+50° +59°+50° +55°+48° +57°+48° +57°+48° +57°+50° +55°+46° +55°+48° +57°+48° +57°+48° +57°+48° +55°+48° +57°+48° +55°+48° +55°+48° +55°+46° +57°+48° +57°+48° +59°+48° +59°+48° +59°+48° +59°+48° +61°+50° Extended weather forecast in San Francisco HourlyWeek10-Day14-Day30-DayYear Weather in Washington, D.C.+30° Sacramento+45° Pleasanton+50° Redwood City+54° San Leandro+54° San Mateo+54° San Rafael+50° San Ramon+48° South San Francisco+55° Vallejo+50° Palo Alto+52° Pacifica+59° Berkeley+48° Castro Valley+54° Concord+48° Daly City+55° Leona Heights+54° Centennial+54° world\\'s temperature today day day Temperature units\"}]', name='tavily_search_results_json', tool_call_id='call_Skc1tN346NFsZWtEmFNeOaBF'), AIMessage(content='The current weather in San Francisco is moderate rain with a temperature of 13.3°C (55.9°F). The wind is coming from the SSE at 11.4 mph (18.4 kph), and the humidity level is at 80%. The conditions are quite cloudy with a cloud cover of 100%.', additional_kwargs={'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 68, 'prompt_tokens': 975, 'total_tokens': 1043, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-4o-2024-08-06', 'system_fingerprint': 'fp_50cad350e4', 'finish_reason': 'stop', 'logprobs': None}, id='run-c73fd64e-a209-4478-9e53-71aa860675bb-0', usage_metadata={'input_tokens': 975, 'output_tokens': 68, 'total_tokens': 1043, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}}), HumanMessage(content='What about in la?', additional_kwargs={}, response_metadata={}), AIMessage(content='', additional_kwargs={'tool_calls': [{'id': 'call_YEOGS21lpACUuv2X5FxWKN2T', 'function': {'arguments': '{\"query\":\"Los Angeles weather today\"}', 'name': 'tavily_search_results_json'}, 'type': 'function'}], 'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 22, 'prompt_tokens': 1054, 'total_tokens': 1076, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-4o-2024-08-06', 'system_fingerprint': 'fp_50cad350e4', 'finish_reason': 'tool_calls', 'logprobs': None}, id='run-8c40c7fb-b445-4fc9-bf80-5bb2fbdb5011-0', tool_calls=[{'name': 'tavily_search_results_json', 'args': {'query': 'Los Angeles weather today'}, 'id': 'call_YEOGS21lpACUuv2X5FxWKN2T', 'type': 'tool_call'}], usage_metadata={'input_tokens': 1054, 'output_tokens': 22, 'total_tokens': 1076, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}}), ToolMessage(content='[{\\'url\\': \\'https://www.weatherapi.com/\\', \\'content\\': \"{\\'location\\': {\\'name\\': \\'Los Angeles\\', \\'region\\': \\'California\\', \\'country\\': \\'United States of America\\', \\'lat\\': 34.0522, \\'lon\\': -118.2428, \\'tz_id\\': \\'America/Los_Angeles\\', \\'localtime_epoch\\': 1738378720, \\'localtime\\': \\'2025-01-31 18:58\\'}, \\'current\\': {\\'last_updated_epoch\\': 1738377900, \\'last_updated\\': \\'2025-01-31 18:45\\', \\'temp_c\\': 13.3, \\'temp_f\\': 55.9, \\'is_day\\': 0, \\'condition\\': {\\'text\\': \\'Clear\\', \\'icon\\': \\'//cdn.weatherapi.com/weather/64x64/night/113.png\\', \\'code\\': 1000}, \\'wind_mph\\': 6.3, \\'wind_kph\\': 10.1, \\'wind_degree\\': 243, \\'wind_dir\\': \\'WSW\\', \\'pressure_mb\\': 1023.0, \\'pressure_in\\': 30.21, \\'precip_mm\\': 0.0, \\'precip_in\\': 0.0, \\'humidity\\': 75, \\'cloud\\': 0, \\'feelslike_c\\': 12.6, \\'feelslike_f\\': 54.6, \\'windchill_c\\': 16.4, \\'windchill_f\\': 61.6, \\'heatindex_c\\': 16.4, \\'heatindex_f\\': 61.6, \\'dewpoint_c\\': 10.2, \\'dewpoint_f\\': 50.4, \\'vis_km\\': 16.0, \\'vis_miles\\': 9.0, \\'uv\\': 0.0, \\'gust_mph\\': 7.8, \\'gust_kph\\': 12.6}}\"}, {\\'url\\': \\'https://www.easeweather.com/north-america/united-states/california/los-angeles-county/los-angeles/february\\', \\'content\\': \\'Weather in Los Angeles in February 2025 - Detailed Forecast Weather in Los Angeles for February 2025 Your guide to Los Angeles weather in February - trends and predictions The forecast for the first days of February 2025 in Los Angeles predicts temperatures to be around 18\\\\xa0°F, close to the historical average. In general, the average temperature in Los Angeles at the beginning of February is 18.1\\\\xa0°F. Los Angeles experiences moderate rainfall in February, with an average of 10 rainy days and a total precipitation of 79.7\\\\xa0mm. Los Angeles in February average weather Temperatures trend during February in Los Angeles Los Angeles in February - FAQ Get accurate weather forecasts for Los Angeles, located at latitude 34.052 and longitude -118.244.\\'}]', name='tavily_search_results_json', tool_call_id='call_YEOGS21lpACUuv2X5FxWKN2T'), AIMessage(content='The current weather in Los Angeles is clear with a temperature of 13.3°C (55.9°F). The wind is coming from the WSW at 6.3 mph (10.1 kph), and the humidity level is at 75%. The sky is clear with no cloud cover.', additional_kwargs={'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 64, 'prompt_tokens': 1689, 'total_tokens': 1753, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 1024}}, 'model_name': 'gpt-4o-2024-08-06', 'system_fingerprint': 'fp_50cad350e4', 'finish_reason': 'stop', 'logprobs': None}, id='run-ecd5771a-302a-46b9-a27f-4796a1494eca-0', usage_metadata={'input_tokens': 1689, 'output_tokens': 64, 'total_tokens': 1753, 'input_token_details': {'audio': 0, 'cache_read': 1024}, 'output_token_details': {'audio': 0, 'reasoning': 0}}), HumanMessage(content='Which one is warmer?', additional_kwargs={}, response_metadata={}), AIMessage(content='Both San Francisco and Los Angeles currently have the same temperature of 13.3°C (55.9°F), so neither is warmer than the other at this time. However, Los Angeles has clear skies while San Francisco experiences moderate rain, which can affect the perception of warmth.', additional_kwargs={'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 58, 'prompt_tokens': 1764, 'total_tokens': 1822, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 1664}}, 'model_name': 'gpt-4o-2024-08-06', 'system_fingerprint': 'fp_50cad350e4', 'finish_reason': 'stop', 'logprobs': None}, id='run-06951198-6c86-4013-8d34-74bbcfa11d26-0', usage_metadata={'input_tokens': 1764, 'output_tokens': 58, 'total_tokens': 1822, 'input_token_details': {'audio': 0, 'cache_read': 1664}, 'output_token_details': {'audio': 0, 'reasoning': 0}}), HumanMessage(content='Whats the weather in SF?', additional_kwargs={}, response_metadata={}, id='5535cb89-c952-424e-ae44-4eb3a0d13d07'), AIMessage(content='', additional_kwargs={'tool_calls': [{'id': 'call_hrtrcRj5Irr0gvQyJrtYFaTs', 'function': {'arguments': '{\"query\":\"San Francisco weather today\"}', 'name': 'tavily_search_results_json'}, 'type': 'function'}], 'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 22, 'prompt_tokens': 1857, 'total_tokens': 1879, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'tool_calls', 'logprobs': None}, id='run-a562ed94-83d6-407c-bc2e-e9b23483f3fc-0', tool_calls=[{'name': 'tavily_search_results_json', 'args': {'query': 'San Francisco weather today'}, 'id': 'call_hrtrcRj5Irr0gvQyJrtYFaTs', 'type': 'tool_call'}], usage_metadata={'input_tokens': 1857, 'output_tokens': 22, 'total_tokens': 1879, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}}), ToolMessage(content='[{\\'url\\': \\'https://www.weatherapi.com/\\', \\'content\\': \"{\\'location\\': {\\'name\\': \\'San Francisco\\', \\'region\\': \\'California\\', \\'country\\': \\'United States of America\\', \\'lat\\': 37.775, \\'lon\\': -122.4183, \\'tz_id\\': \\'America/Los_Angeles\\', \\'localtime_epoch\\': 1738453598, \\'localtime\\': \\'2025-02-01 15:46\\'}, \\'current\\': {\\'last_updated_epoch\\': 1738453500, \\'last_updated\\': \\'2025-02-01 15:45\\', \\'temp_c\\': 15.6, \\'temp_f\\': 60.1, \\'is_day\\': 1, \\'condition\\': {\\'text\\': \\'Overcast\\', \\'icon\\': \\'//cdn.weatherapi.com/weather/64x64/day/122.png\\', \\'code\\': 1009}, \\'wind_mph\\': 19.0, \\'wind_kph\\': 30.6, \\'wind_degree\\': 207, \\'wind_dir\\': \\'SSW\\', \\'pressure_mb\\': 1018.0, \\'pressure_in\\': 30.06, \\'precip_mm\\': 0.0, \\'precip_in\\': 0.0, \\'humidity\\': 86, \\'cloud\\': 100, \\'feelslike_c\\': 15.6, \\'feelslike_f\\': 60.1, \\'windchill_c\\': 10.6, \\'windchill_f\\': 51.0, \\'heatindex_c\\': 12.7, \\'heatindex_f\\': 54.8, \\'dewpoint_c\\': 12.4, \\'dewpoint_f\\': 54.2, \\'vis_km\\': 13.0, \\'vis_miles\\': 8.0, \\'uv\\': 0.1, \\'gust_mph\\': 31.3, \\'gust_kph\\': 50.4}}\"}, {\\'url\\': \\'https://www.easeweather.com/north-america/united-states/california/city-and-county-of-san-francisco/february\\', \\'content\\': \\'February 2025 Weather - City and County of San Francisco Weather in City and County of San Francisco for February 2025 Your guide to City and County of San Francisco weather in February - trends and predictions The forecast for the first days of February 2025 in City and County of San Francisco predicts temperatures to be around 12\\\\xa0°F, close to the historical average. In general, the average temperature in City and County of San Francisco at the beginning of February is 12.6\\\\xa0°F. City and County of San Francisco in February average weather Temperatures trend during February in City and County of San Francisco City and County of San Francisco in February - FAQ\\'}]', name='tavily_search_results_json', id='d8d8d49b-ce93-4dea-83bd-7aa79263666e', tool_call_id='call_hrtrcRj5Irr0gvQyJrtYFaTs'), AIMessage(content='The current weather in San Francisco is overcast with a temperature of 15.6°C (60.1°F). The wind is coming from the SSW at 19.0 mph (30.6 kph), and the humidity level is at 86%. The sky is completely covered with clouds at 100% cloud cover.', additional_kwargs={'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 70, 'prompt_tokens': 2476, 'total_tokens': 2546, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'stop', 'logprobs': None}, id='run-12f99469-8e00-4d97-9d02-c8488c44e329-0', usage_metadata={'input_tokens': 2476, 'output_tokens': 70, 'total_tokens': 2546, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}})]}, next=(), config={'configurable': {'thread_id': '1', 'checkpoint_ns': '', 'checkpoint_id': '1efe0f6d-98a7-6be1-8010-321069525346'}}, metadata={'source': 'loop', 'writes': {'llm': {'messages': [AIMessage(content='The current weather in San Francisco is overcast with a temperature of 15.6°C (60.1°F). The wind is coming from the SSW at 19.0 mph (30.6 kph), and the humidity level is at 86%. The sky is completely covered with clouds at 100% cloud cover.', additional_kwargs={'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 70, 'prompt_tokens': 2476, 'total_tokens': 2546, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'stop', 'logprobs': None}, id='run-12f99469-8e00-4d97-9d02-c8488c44e329-0', usage_metadata={'input_tokens': 2476, 'output_tokens': 70, 'total_tokens': 2546, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}})]}}, 'thread_id': '1', 'step': 16, 'parents': {}}, created_at='2025-02-01T23:47:08.808241+00:00', parent_config={'configurable': {'thread_id': '1', 'checkpoint_ns': '', 'checkpoint_id': '1efe0f6d-864d-6eb6-800f-f9d309817c1a'}}, tasks=())"
      ]
     },
     "execution_count": 17,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "abot.graph.get_state(thread)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 18,
   "id": "d376353c-9b1d-4214-9a29-78bb28ef3400",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "()"
      ]
     },
     "execution_count": 18,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "abot.graph.get_state(thread).next"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 19,
   "id": "98c2d49e-27fc-4220-b9cc-783bd473af20",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "{'messages': [HumanMessage(content='Which one is warmer?', additional_kwargs={}, response_metadata={}), AIMessage(content='Could you please provide more context or specify what you are comparing in terms of warmth? For example, are you asking about weather in different locations, materials, clothing items, or something else?', additional_kwargs={'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 40, 'prompt_tokens': 149, 'total_tokens': 189, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-4o-2024-08-06', 'system_fingerprint': 'fp_50cad350e4', 'finish_reason': 'stop', 'logprobs': None}, id='run-96cbe194-e107-4d6c-a8c0-e19440527413-0', usage_metadata={'input_tokens': 149, 'output_tokens': 40, 'total_tokens': 189, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}}), HumanMessage(content='Whats the weather in LA?', additional_kwargs={}, response_metadata={}, id='6569b98f-d178-4237-b65a-47a181ae6696'), AIMessage(content='', additional_kwargs={'tool_calls': [{'id': 'call_k3zVV3FtFnKrkj6P4AFNBF36', 'function': {'arguments': '{\"query\":\"Los Angeles current weather\"}', 'name': 'tavily_search_results_json'}, 'type': 'function'}], 'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 22, 'prompt_tokens': 203, 'total_tokens': 225, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'tool_calls', 'logprobs': None}, id='run-bb348a61-e292-4f67-a136-89a9e2fcb156-0', tool_calls=[{'name': 'tavily_search_results_json', 'args': {'query': 'Los Angeles current weather'}, 'id': 'call_k3zVV3FtFnKrkj6P4AFNBF36', 'type': 'tool_call'}], usage_metadata={'input_tokens': 203, 'output_tokens': 22, 'total_tokens': 225, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}})]}\n",
      "{'messages': [AIMessage(content='', additional_kwargs={'tool_calls': [{'id': 'call_k3zVV3FtFnKrkj6P4AFNBF36', 'function': {'arguments': '{\"query\":\"Los Angeles current weather\"}', 'name': 'tavily_search_results_json'}, 'type': 'function'}], 'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 22, 'prompt_tokens': 203, 'total_tokens': 225, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'tool_calls', 'logprobs': None}, id='run-bb348a61-e292-4f67-a136-89a9e2fcb156-0', tool_calls=[{'name': 'tavily_search_results_json', 'args': {'query': 'Los Angeles current weather'}, 'id': 'call_k3zVV3FtFnKrkj6P4AFNBF36', 'type': 'tool_call'}], usage_metadata={'input_tokens': 203, 'output_tokens': 22, 'total_tokens': 225, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}})]}\n",
      "()\n",
      "\n",
      " StateSnapshot(values={'messages': [HumanMessage(content='Which one is warmer?', additional_kwargs={}, response_metadata={}), AIMessage(content='Could you please provide more context or specify what you are comparing in terms of warmth? For example, are you asking about weather in different locations, materials, clothing items, or something else?', additional_kwargs={'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 40, 'prompt_tokens': 149, 'total_tokens': 189, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-4o-2024-08-06', 'system_fingerprint': 'fp_50cad350e4', 'finish_reason': 'stop', 'logprobs': None}, id='run-96cbe194-e107-4d6c-a8c0-e19440527413-0', usage_metadata={'input_tokens': 149, 'output_tokens': 40, 'total_tokens': 189, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}}), HumanMessage(content='Whats the weather in LA?', additional_kwargs={}, response_metadata={}, id='6569b98f-d178-4237-b65a-47a181ae6696'), AIMessage(content='', additional_kwargs={'tool_calls': [{'id': 'call_k3zVV3FtFnKrkj6P4AFNBF36', 'function': {'arguments': '{\"query\":\"Los Angeles current weather\"}', 'name': 'tavily_search_results_json'}, 'type': 'function'}], 'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 22, 'prompt_tokens': 203, 'total_tokens': 225, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'tool_calls', 'logprobs': None}, id='run-bb348a61-e292-4f67-a136-89a9e2fcb156-0', tool_calls=[{'name': 'tavily_search_results_json', 'args': {'query': 'Los Angeles current weather'}, 'id': 'call_k3zVV3FtFnKrkj6P4AFNBF36', 'type': 'tool_call'}], usage_metadata={'input_tokens': 203, 'output_tokens': 22, 'total_tokens': 225, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}})]}, next=('action',), config={'configurable': {'thread_id': '2', 'checkpoint_ns': '', 'checkpoint_id': '1efe0f6d-de30-69b4-8004-dc6990c246cd'}}, metadata={'source': 'loop', 'writes': {'llm': {'messages': [AIMessage(content='', additional_kwargs={'tool_calls': [{'id': 'call_k3zVV3FtFnKrkj6P4AFNBF36', 'function': {'arguments': '{\"query\":\"Los Angeles current weather\"}', 'name': 'tavily_search_results_json'}, 'type': 'function'}], 'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 22, 'prompt_tokens': 203, 'total_tokens': 225, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'tool_calls', 'logprobs': None}, id='run-bb348a61-e292-4f67-a136-89a9e2fcb156-0', tool_calls=[{'name': 'tavily_search_results_json', 'args': {'query': 'Los Angeles current weather'}, 'id': 'call_k3zVV3FtFnKrkj6P4AFNBF36', 'type': 'tool_call'}], usage_metadata={'input_tokens': 203, 'output_tokens': 22, 'total_tokens': 225, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}})]}}, 'thread_id': '2', 'step': 4, 'parents': {}}, created_at='2025-02-01T23:47:16.099474+00:00', parent_config={'configurable': {'thread_id': '2', 'checkpoint_ns': '', 'checkpoint_id': '1efe0f6d-d0e4-6cb6-8003-104b3eb787cd'}}, tasks=(PregelTask(id='bf32a1ab-b5c2-3ca3-2af2-8bfeeedefb18', name='action', path=('__pregel_pull', 'action'), error=None, interrupts=(), state=None, result=None),)) \n",
      "\n"
     ]
    },
    {
     "name": "stdin",
     "output_type": "stream",
     "text": [
      "proceed? y\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Calling: {'name': 'tavily_search_results_json', 'args': {'query': 'Los Angeles current weather'}, 'id': 'call_k3zVV3FtFnKrkj6P4AFNBF36', 'type': 'tool_call'}\n",
      "Back to the model!\n",
      "{'messages': [ToolMessage(content='[{\\'url\\': \\'https://www.weatherapi.com/\\', \\'content\\': \"{\\'location\\': {\\'name\\': \\'Los Angeles\\', \\'region\\': \\'California\\', \\'country\\': \\'United States of America\\', \\'lat\\': 34.0522, \\'lon\\': -118.2428, \\'tz_id\\': \\'America/Los_Angeles\\', \\'localtime_epoch\\': 1738453981, \\'localtime\\': \\'2025-02-01 15:53\\'}, \\'current\\': {\\'last_updated_epoch\\': 1738453500, \\'last_updated\\': \\'2025-02-01 15:45\\', \\'temp_c\\': 17.8, \\'temp_f\\': 64.0, \\'is_day\\': 1, \\'condition\\': {\\'text\\': \\'Sunny\\', \\'icon\\': \\'//cdn.weatherapi.com/weather/64x64/day/113.png\\', \\'code\\': 1000}, \\'wind_mph\\': 6.7, \\'wind_kph\\': 10.8, \\'wind_degree\\': 248, \\'wind_dir\\': \\'WSW\\', \\'pressure_mb\\': 1019.0, \\'pressure_in\\': 30.09, \\'precip_mm\\': 0.0, \\'precip_in\\': 0.0, \\'humidity\\': 56, \\'cloud\\': 0, \\'feelslike_c\\': 17.8, \\'feelslike_f\\': 64.0, \\'windchill_c\\': 20.1, \\'windchill_f\\': 68.3, \\'heatindex_c\\': 20.1, \\'heatindex_f\\': 68.3, \\'dewpoint_c\\': 9.0, \\'dewpoint_f\\': 48.1, \\'vis_km\\': 16.0, \\'vis_miles\\': 9.0, \\'uv\\': 1.1, \\'gust_mph\\': 7.7, \\'gust_kph\\': 12.4}}\"}, {\\'url\\': \\'https://world-weather.info/forecast/usa/los_angeles/january-2025/\\', \\'content\\': \"Weather in Los Angeles in January 2025 (California) - Detailed Weather Forecast for a Month Weather Weather in Los Angeles Weather in Los Angeles in January 2025 Los Angeles Weather Forecast for January 2025, is based on previous years\\' statistical data. +61°+50° +63°+52° +63°+52° +63°+52° +64°+55° +64°+54° +64°+55° +64°+55° +64°+55° +64°+54° +64°+54° +63°+52° +64°+54° +66°+54° +66°+54° +64°+55° +66°+55° +66°+55° +64°+54° +63°+54° +63°+54° +64°+54° +64°+54° +63°+52° +64°+54° +64°+54° +68°+55° +68°+54° +66°+54° +66°+54° +66°+55° Extended weather forecast in Los Angeles HourlyWeek10 days14 days30 daysYear Weather in large and nearby cities Weather in Washington, D.C.+64° Sacramento+66° Monterey Park+61° North Glendale+63° Norwalk+61° Pasadena+63° Rosemead+61° Santa Monica+61° Manhattan Beach+63° Inglewood+61° Bellflower+61° Beverly Hills+59° Burbank+63° Compton+61° Rancho Dominguez+61° world\\'s temperature today day day Temperature units\"}]', name='tavily_search_results_json', id='02069852-f89c-456a-87e6-7fd09542fba2', tool_call_id='call_k3zVV3FtFnKrkj6P4AFNBF36')]}\n",
      "{'messages': [HumanMessage(content='Which one is warmer?', additional_kwargs={}, response_metadata={}), AIMessage(content='Could you please provide more context or specify what you are comparing in terms of warmth? For example, are you asking about weather in different locations, materials, clothing items, or something else?', additional_kwargs={'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 40, 'prompt_tokens': 149, 'total_tokens': 189, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-4o-2024-08-06', 'system_fingerprint': 'fp_50cad350e4', 'finish_reason': 'stop', 'logprobs': None}, id='run-96cbe194-e107-4d6c-a8c0-e19440527413-0', usage_metadata={'input_tokens': 149, 'output_tokens': 40, 'total_tokens': 189, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}}), HumanMessage(content='Whats the weather in LA?', additional_kwargs={}, response_metadata={}, id='6569b98f-d178-4237-b65a-47a181ae6696'), AIMessage(content='', additional_kwargs={'tool_calls': [{'id': 'call_k3zVV3FtFnKrkj6P4AFNBF36', 'function': {'arguments': '{\"query\":\"Los Angeles current weather\"}', 'name': 'tavily_search_results_json'}, 'type': 'function'}], 'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 22, 'prompt_tokens': 203, 'total_tokens': 225, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'tool_calls', 'logprobs': None}, id='run-bb348a61-e292-4f67-a136-89a9e2fcb156-0', tool_calls=[{'name': 'tavily_search_results_json', 'args': {'query': 'Los Angeles current weather'}, 'id': 'call_k3zVV3FtFnKrkj6P4AFNBF36', 'type': 'tool_call'}], usage_metadata={'input_tokens': 203, 'output_tokens': 22, 'total_tokens': 225, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}}), ToolMessage(content='[{\\'url\\': \\'https://www.weatherapi.com/\\', \\'content\\': \"{\\'location\\': {\\'name\\': \\'Los Angeles\\', \\'region\\': \\'California\\', \\'country\\': \\'United States of America\\', \\'lat\\': 34.0522, \\'lon\\': -118.2428, \\'tz_id\\': \\'America/Los_Angeles\\', \\'localtime_epoch\\': 1738453981, \\'localtime\\': \\'2025-02-01 15:53\\'}, \\'current\\': {\\'last_updated_epoch\\': 1738453500, \\'last_updated\\': \\'2025-02-01 15:45\\', \\'temp_c\\': 17.8, \\'temp_f\\': 64.0, \\'is_day\\': 1, \\'condition\\': {\\'text\\': \\'Sunny\\', \\'icon\\': \\'//cdn.weatherapi.com/weather/64x64/day/113.png\\', \\'code\\': 1000}, \\'wind_mph\\': 6.7, \\'wind_kph\\': 10.8, \\'wind_degree\\': 248, \\'wind_dir\\': \\'WSW\\', \\'pressure_mb\\': 1019.0, \\'pressure_in\\': 30.09, \\'precip_mm\\': 0.0, \\'precip_in\\': 0.0, \\'humidity\\': 56, \\'cloud\\': 0, \\'feelslike_c\\': 17.8, \\'feelslike_f\\': 64.0, \\'windchill_c\\': 20.1, \\'windchill_f\\': 68.3, \\'heatindex_c\\': 20.1, \\'heatindex_f\\': 68.3, \\'dewpoint_c\\': 9.0, \\'dewpoint_f\\': 48.1, \\'vis_km\\': 16.0, \\'vis_miles\\': 9.0, \\'uv\\': 1.1, \\'gust_mph\\': 7.7, \\'gust_kph\\': 12.4}}\"}, {\\'url\\': \\'https://world-weather.info/forecast/usa/los_angeles/january-2025/\\', \\'content\\': \"Weather in Los Angeles in January 2025 (California) - Detailed Weather Forecast for a Month Weather Weather in Los Angeles Weather in Los Angeles in January 2025 Los Angeles Weather Forecast for January 2025, is based on previous years\\' statistical data. +61°+50° +63°+52° +63°+52° +63°+52° +64°+55° +64°+54° +64°+55° +64°+55° +64°+55° +64°+54° +64°+54° +63°+52° +64°+54° +66°+54° +66°+54° +64°+55° +66°+55° +66°+55° +64°+54° +63°+54° +63°+54° +64°+54° +64°+54° +63°+52° +64°+54° +64°+54° +68°+55° +68°+54° +66°+54° +66°+54° +66°+55° Extended weather forecast in Los Angeles HourlyWeek10 days14 days30 daysYear Weather in large and nearby cities Weather in Washington, D.C.+64° Sacramento+66° Monterey Park+61° North Glendale+63° Norwalk+61° Pasadena+63° Rosemead+61° Santa Monica+61° Manhattan Beach+63° Inglewood+61° Bellflower+61° Beverly Hills+59° Burbank+63° Compton+61° Rancho Dominguez+61° world\\'s temperature today day day Temperature units\"}]', name='tavily_search_results_json', id='02069852-f89c-456a-87e6-7fd09542fba2', tool_call_id='call_k3zVV3FtFnKrkj6P4AFNBF36'), AIMessage(content='The current weather in Los Angeles is as follows:\\n- Temperature: 64.0°F (17.8°C)\\n- Condition: Sunny\\n- Wind: 6.7 mph (10.8 kph) from WSW\\n- Humidity: 56%\\n- Feels like: 64.0°F (17.8°C)\\n- Dewpoint: 48.1°F (9.0°C)\\n\\nIf you have any specific questions or need more details, feel free to ask!', additional_kwargs={'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 104, 'prompt_tokens': 1028, 'total_tokens': 1132, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'stop', 'logprobs': None}, id='run-56fe7067-d172-46cf-8f89-c6523a8eb91f-0', usage_metadata={'input_tokens': 1028, 'output_tokens': 104, 'total_tokens': 1132, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}})]}\n",
      "{'messages': [AIMessage(content='The current weather in Los Angeles is as follows:\\n- Temperature: 64.0°F (17.8°C)\\n- Condition: Sunny\\n- Wind: 6.7 mph (10.8 kph) from WSW\\n- Humidity: 56%\\n- Feels like: 64.0°F (17.8°C)\\n- Dewpoint: 48.1°F (9.0°C)\\n\\nIf you have any specific questions or need more details, feel free to ask!', additional_kwargs={'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 104, 'prompt_tokens': 1028, 'total_tokens': 1132, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'stop', 'logprobs': None}, id='run-56fe7067-d172-46cf-8f89-c6523a8eb91f-0', usage_metadata={'input_tokens': 1028, 'output_tokens': 104, 'total_tokens': 1132, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}})]}\n"
     ]
    }
   ],
   "source": [
    "messages = [HumanMessage(\"Whats the weather in LA?\")]\n",
    "thread = {\"configurable\": {\"thread_id\": \"2\"}}\n",
    "for event in abot.graph.stream({\"messages\": messages}, thread):\n",
    "    for v in event.values():\n",
    "        print(v)\n",
    "\n",
    "while abot.graph.get_state(thread).next:\n",
    "    print(\"\\n\", abot.graph.get_state(thread),\"\\n\")\n",
    "    _input = input(\"proceed?\")\n",
    "    if _input != \"y\":\n",
    "        print(\"aborting\")\n",
    "        break\n",
    "    for event in abot.graph.stream(None, thread):\n",
    "        for v in event.values():\n",
    "            print(v)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 20,
   "id": "f14d85c6-1058-44de-95bb-bea382310e2a",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "{'messages': [HumanMessage(content='Whats the weather in LA?', additional_kwargs={}, response_metadata={}, id='cab22332-f4eb-4762-aca6-8d5c130362b1'), AIMessage(content='', additional_kwargs={'tool_calls': [{'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'function': {'arguments': '{\"query\":\"weather in Los Angeles\"}', 'name': 'tavily_search_results_json'}, 'type': 'function'}], 'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 22, 'prompt_tokens': 152, 'total_tokens': 174, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'tool_calls', 'logprobs': None}, id='run-116bf1b3-6030-400e-a1d0-8339e7088e31-0', tool_calls=[{'name': 'tavily_search_results_json', 'args': {'query': 'weather in Los Angeles'}, 'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'type': 'tool_call'}], usage_metadata={'input_tokens': 152, 'output_tokens': 22, 'total_tokens': 174, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}})]}\n",
      "{'messages': [AIMessage(content='', additional_kwargs={'tool_calls': [{'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'function': {'arguments': '{\"query\":\"weather in Los Angeles\"}', 'name': 'tavily_search_results_json'}, 'type': 'function'}], 'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 22, 'prompt_tokens': 152, 'total_tokens': 174, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'tool_calls', 'logprobs': None}, id='run-116bf1b3-6030-400e-a1d0-8339e7088e31-0', tool_calls=[{'name': 'tavily_search_results_json', 'args': {'query': 'weather in Los Angeles'}, 'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'type': 'tool_call'}], usage_metadata={'input_tokens': 152, 'output_tokens': 22, 'total_tokens': 174, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}})]}\n",
      "()\n"
     ]
    }
   ],
   "source": [
    "messages = [HumanMessage(\"Whats the weather in LA?\")]\n",
    "thread = {\"configurable\": {\"thread_id\": \"3\"}}\n",
    "for event in abot.graph.stream({\"messages\": messages}, thread):\n",
    "    for v in event.values():\n",
    "        print(v)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 21,
   "id": "49b39bb1-6c13-409f-9821-5d56cc637de9",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "StateSnapshot(values={'messages': [HumanMessage(content='Whats the weather in LA?', additional_kwargs={}, response_metadata={}, id='cab22332-f4eb-4762-aca6-8d5c130362b1'), AIMessage(content='', additional_kwargs={'tool_calls': [{'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'function': {'arguments': '{\"query\":\"weather in Los Angeles\"}', 'name': 'tavily_search_results_json'}, 'type': 'function'}], 'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 22, 'prompt_tokens': 152, 'total_tokens': 174, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'tool_calls', 'logprobs': None}, id='run-116bf1b3-6030-400e-a1d0-8339e7088e31-0', tool_calls=[{'name': 'tavily_search_results_json', 'args': {'query': 'weather in Los Angeles'}, 'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'type': 'tool_call'}], usage_metadata={'input_tokens': 152, 'output_tokens': 22, 'total_tokens': 174, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}})]}, next=('action',), config={'configurable': {'thread_id': '3', 'checkpoint_ns': '', 'checkpoint_id': '1efe0f7b-a224-6387-8001-d8242dd7ad14'}}, metadata={'source': 'loop', 'writes': {'llm': {'messages': [AIMessage(content='', additional_kwargs={'tool_calls': [{'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'function': {'arguments': '{\"query\":\"weather in Los Angeles\"}', 'name': 'tavily_search_results_json'}, 'type': 'function'}], 'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 22, 'prompt_tokens': 152, 'total_tokens': 174, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'tool_calls', 'logprobs': None}, id='run-116bf1b3-6030-400e-a1d0-8339e7088e31-0', tool_calls=[{'name': 'tavily_search_results_json', 'args': {'query': 'weather in Los Angeles'}, 'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'type': 'tool_call'}], usage_metadata={'input_tokens': 152, 'output_tokens': 22, 'total_tokens': 174, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}})]}}, 'thread_id': '3', 'step': 1, 'parents': {}}, created_at='2025-02-01T23:53:25.612578+00:00', parent_config={'configurable': {'thread_id': '3', 'checkpoint_ns': '', 'checkpoint_id': '1efe0f7b-9878-68ff-8000-838b1e05c511'}}, tasks=(PregelTask(id='4469c5af-8ee9-3033-98d7-c09acc4c7551', name='action', path=('__pregel_pull', 'action'), error=None, interrupts=(), state=None, result=None),))"
      ]
     },
     "execution_count": 21,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "abot.graph.get_state(thread)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 22,
   "id": "495757ae-ed33-43a2-9b75-eacc3b24c415",
   "metadata": {},
   "outputs": [],
   "source": [
    "current_values = abot.graph.get_state(thread)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 23,
   "id": "addc83c3-3f22-4063-9e1e-21e13b8fef00",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "AIMessage(content='', additional_kwargs={'tool_calls': [{'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'function': {'arguments': '{\"query\":\"weather in Los Angeles\"}', 'name': 'tavily_search_results_json'}, 'type': 'function'}], 'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 22, 'prompt_tokens': 152, 'total_tokens': 174, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'tool_calls', 'logprobs': None}, id='run-116bf1b3-6030-400e-a1d0-8339e7088e31-0', tool_calls=[{'name': 'tavily_search_results_json', 'args': {'query': 'weather in Los Angeles'}, 'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'type': 'tool_call'}], usage_metadata={'input_tokens': 152, 'output_tokens': 22, 'total_tokens': 174, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}})"
      ]
     },
     "execution_count": 23,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "current_values.values['messages'][-1]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 24,
   "id": "6049e54d-eabd-4f2a-8a6e-ae0eddac54f8",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "[{'name': 'tavily_search_results_json',\n",
       "  'args': {'query': 'weather in Los Angeles'},\n",
       "  'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv',\n",
       "  'type': 'tool_call'}]"
      ]
     },
     "execution_count": 24,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "current_values.values['messages'][-1].tool_calls"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 25,
   "id": "d714253c-1e84-46b8-81eb-729d75daa5d6",
   "metadata": {},
   "outputs": [],
   "source": [
    "_id = current_values.values['messages'][-1].tool_calls[0]['id']\n",
    "current_values.values['messages'][-1].tool_calls = [\n",
    "    {'name': 'tavily_search_results_json',\n",
    "  'args': {'query': 'current weather in Louisiana'},\n",
    "  'id': _id}\n",
    "]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 26,
   "id": "a14c1c1b-d035-4eaa-a1bc-58c3af81d9b0",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "{'messages': [HumanMessage(content='Whats the weather in LA?', additional_kwargs={}, response_metadata={}, id='cab22332-f4eb-4762-aca6-8d5c130362b1'), AIMessage(content='', additional_kwargs={'tool_calls': [{'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'function': {'arguments': '{\"query\":\"weather in Los Angeles\"}', 'name': 'tavily_search_results_json'}, 'type': 'function'}], 'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 22, 'prompt_tokens': 152, 'total_tokens': 174, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'tool_calls', 'logprobs': None}, id='run-116bf1b3-6030-400e-a1d0-8339e7088e31-0', tool_calls=[{'name': 'tavily_search_results_json', 'args': {'query': 'current weather in Louisiana'}, 'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv'}], usage_metadata={'input_tokens': 152, 'output_tokens': 22, 'total_tokens': 174, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}})]}\n"
     ]
    },
    {
     "data": {
      "text/plain": [
       "{'configurable': {'thread_id': '3',\n",
       "  'checkpoint_ns': '',\n",
       "  'checkpoint_id': '1efe0f7d-74aa-69bb-8002-36803ed0c072'}}"
      ]
     },
     "execution_count": 26,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "abot.graph.update_state(thread, current_values.values)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 27,
   "id": "da29632a-1e55-4231-8f9c-81ca28065e2d",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "StateSnapshot(values={'messages': [HumanMessage(content='Whats the weather in LA?', additional_kwargs={}, response_metadata={}, id='cab22332-f4eb-4762-aca6-8d5c130362b1'), AIMessage(content='', additional_kwargs={'tool_calls': [{'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'function': {'arguments': '{\"query\":\"weather in Los Angeles\"}', 'name': 'tavily_search_results_json'}, 'type': 'function'}], 'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 22, 'prompt_tokens': 152, 'total_tokens': 174, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'tool_calls', 'logprobs': None}, id='run-116bf1b3-6030-400e-a1d0-8339e7088e31-0', tool_calls=[{'name': 'tavily_search_results_json', 'args': {'query': 'current weather in Louisiana'}, 'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'type': 'tool_call'}], usage_metadata={'input_tokens': 152, 'output_tokens': 22, 'total_tokens': 174, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}})]}, next=('action',), config={'configurable': {'thread_id': '3', 'checkpoint_ns': '', 'checkpoint_id': '1efe0f7d-74aa-69bb-8002-36803ed0c072'}}, metadata={'source': 'update', 'writes': {'llm': {'messages': [HumanMessage(content='Whats the weather in LA?', additional_kwargs={}, response_metadata={}, id='cab22332-f4eb-4762-aca6-8d5c130362b1'), AIMessage(content='', additional_kwargs={'tool_calls': [{'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'function': {'arguments': '{\"query\":\"weather in Los Angeles\"}', 'name': 'tavily_search_results_json'}, 'type': 'function'}], 'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 22, 'prompt_tokens': 152, 'total_tokens': 174, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'tool_calls', 'logprobs': None}, id='run-116bf1b3-6030-400e-a1d0-8339e7088e31-0', tool_calls=[{'name': 'tavily_search_results_json', 'args': {'query': 'current weather in Louisiana'}, 'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'type': 'tool_call'}], usage_metadata={'input_tokens': 152, 'output_tokens': 22, 'total_tokens': 174, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}})]}}, 'thread_id': '3', 'step': 2, 'parents': {}}, created_at='2025-02-01T23:54:14.531265+00:00', parent_config={'configurable': {'thread_id': '3', 'checkpoint_ns': '', 'checkpoint_id': '1efe0f7b-a224-6387-8001-d8242dd7ad14'}}, tasks=(PregelTask(id='f959be95-7373-11ab-0e38-a29600e70525', name='action', path=('__pregel_pull', 'action'), error=None, interrupts=(), state=None, result=None),))"
      ]
     },
     "execution_count": 27,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "abot.graph.get_state(thread)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 28,
   "id": "26d0d679-e50e-4f9b-bab2-53b083c4eeb5",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Calling: {'name': 'tavily_search_results_json', 'args': {'query': 'current weather in Louisiana'}, 'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'type': 'tool_call'}\n",
      "Back to the model!\n",
      "{'messages': [ToolMessage(content='[{\\'url\\': \\'https://www.weatherapi.com/\\', \\'content\\': \"{\\'location\\': {\\'name\\': \\'Louisiana\\', \\'region\\': \\'Missouri\\', \\'country\\': \\'USA United States of America\\', \\'lat\\': 39.4411, \\'lon\\': -91.0551, \\'tz_id\\': \\'America/Chicago\\', \\'localtime_epoch\\': 1738454236, \\'localtime\\': \\'2025-02-01 17:57\\'}, \\'current\\': {\\'last_updated_epoch\\': 1738453500, \\'last_updated\\': \\'2025-02-01 17:45\\', \\'temp_c\\': 4.7, \\'temp_f\\': 40.5, \\'is_day\\': 0, \\'condition\\': {\\'text\\': \\'Clear\\', \\'icon\\': \\'//cdn.weatherapi.com/weather/64x64/night/113.png\\', \\'code\\': 1000}, \\'wind_mph\\': 14.5, \\'wind_kph\\': 23.4, \\'wind_degree\\': 143, \\'wind_dir\\': \\'SE\\', \\'pressure_mb\\': 1018.0, \\'pressure_in\\': 30.06, \\'precip_mm\\': 0.0, \\'precip_in\\': 0.0, \\'humidity\\': 77, \\'cloud\\': 0, \\'feelslike_c\\': 0.3, \\'feelslike_f\\': 32.5, \\'windchill_c\\': 0.2, \\'windchill_f\\': 32.4, \\'heatindex_c\\': 4.5, \\'heatindex_f\\': 40.1, \\'dewpoint_c\\': 1.5, \\'dewpoint_f\\': 34.8, \\'vis_km\\': 16.0, \\'vis_miles\\': 9.0, \\'uv\\': 0.0, \\'gust_mph\\': 22.2, \\'gust_kph\\': 35.8}}\"}, {\\'url\\': \\'https://forecast.weather.gov/zipcity.php?inputstring=Baton+Rouge,LA\\', \\'content\\': \\'FORECAST  Last update 19 Nov 4:53 am CST Hourly Weather Forecast Low: 71 °F High: 80 °F Tuesday Night High: 68 °F Wednesday Night High: 66 °F Thursday Night High: 63 °F Friday Night Tuesday Night Wednesday Night Thursday Night Friday Night Saturday Night Sunday Night Additional Forecasts and Information ZONE AREA FORECAST FOR EAST BATON ROUGE PARISH, LA Hourly Weather Forecast Air Quality Forecasts Marine Weather Aviation Weather NWS New Orleans/Baton Rouge Home ABOUT THIS FORECAST Hourly Weather Forecast Winter Weather Forecasts Fire Weather Outlooks Surface Weather FORECAST Local Forecast International Forecasts Fire Weather Current Graphical Forecast Maps Aviation Weather Fire Weather Tsunami Warning System International Weather WEATHER SAFETY NWS Education Home National Weather Service Please Contact Us. Disclaimer\\'}]', name='tavily_search_results_json', id='82af75cb-d790-4f3f-805e-98d48e7abe2f', tool_call_id='call_S8IFcTbfn4n10pd8XkXiPQSv')]}\n",
      "{'messages': [HumanMessage(content='Whats the weather in LA?', additional_kwargs={}, response_metadata={}, id='cab22332-f4eb-4762-aca6-8d5c130362b1'), AIMessage(content='', additional_kwargs={'tool_calls': [{'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'function': {'arguments': '{\"query\":\"weather in Los Angeles\"}', 'name': 'tavily_search_results_json'}, 'type': 'function'}], 'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 22, 'prompt_tokens': 152, 'total_tokens': 174, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'tool_calls', 'logprobs': None}, id='run-116bf1b3-6030-400e-a1d0-8339e7088e31-0', tool_calls=[{'name': 'tavily_search_results_json', 'args': {'query': 'current weather in Louisiana'}, 'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'type': 'tool_call'}], usage_metadata={'input_tokens': 152, 'output_tokens': 22, 'total_tokens': 174, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}}), ToolMessage(content='[{\\'url\\': \\'https://www.weatherapi.com/\\', \\'content\\': \"{\\'location\\': {\\'name\\': \\'Louisiana\\', \\'region\\': \\'Missouri\\', \\'country\\': \\'USA United States of America\\', \\'lat\\': 39.4411, \\'lon\\': -91.0551, \\'tz_id\\': \\'America/Chicago\\', \\'localtime_epoch\\': 1738454236, \\'localtime\\': \\'2025-02-01 17:57\\'}, \\'current\\': {\\'last_updated_epoch\\': 1738453500, \\'last_updated\\': \\'2025-02-01 17:45\\', \\'temp_c\\': 4.7, \\'temp_f\\': 40.5, \\'is_day\\': 0, \\'condition\\': {\\'text\\': \\'Clear\\', \\'icon\\': \\'//cdn.weatherapi.com/weather/64x64/night/113.png\\', \\'code\\': 1000}, \\'wind_mph\\': 14.5, \\'wind_kph\\': 23.4, \\'wind_degree\\': 143, \\'wind_dir\\': \\'SE\\', \\'pressure_mb\\': 1018.0, \\'pressure_in\\': 30.06, \\'precip_mm\\': 0.0, \\'precip_in\\': 0.0, \\'humidity\\': 77, \\'cloud\\': 0, \\'feelslike_c\\': 0.3, \\'feelslike_f\\': 32.5, \\'windchill_c\\': 0.2, \\'windchill_f\\': 32.4, \\'heatindex_c\\': 4.5, \\'heatindex_f\\': 40.1, \\'dewpoint_c\\': 1.5, \\'dewpoint_f\\': 34.8, \\'vis_km\\': 16.0, \\'vis_miles\\': 9.0, \\'uv\\': 0.0, \\'gust_mph\\': 22.2, \\'gust_kph\\': 35.8}}\"}, {\\'url\\': \\'https://forecast.weather.gov/zipcity.php?inputstring=Baton+Rouge,LA\\', \\'content\\': \\'FORECAST  Last update 19 Nov 4:53 am CST Hourly Weather Forecast Low: 71 °F High: 80 °F Tuesday Night High: 68 °F Wednesday Night High: 66 °F Thursday Night High: 63 °F Friday Night Tuesday Night Wednesday Night Thursday Night Friday Night Saturday Night Sunday Night Additional Forecasts and Information ZONE AREA FORECAST FOR EAST BATON ROUGE PARISH, LA Hourly Weather Forecast Air Quality Forecasts Marine Weather Aviation Weather NWS New Orleans/Baton Rouge Home ABOUT THIS FORECAST Hourly Weather Forecast Winter Weather Forecasts Fire Weather Outlooks Surface Weather FORECAST Local Forecast International Forecasts Fire Weather Current Graphical Forecast Maps Aviation Weather Fire Weather Tsunami Warning System International Weather WEATHER SAFETY NWS Education Home National Weather Service Please Contact Us. Disclaimer\\'}]', name='tavily_search_results_json', id='82af75cb-d790-4f3f-805e-98d48e7abe2f', tool_call_id='call_S8IFcTbfn4n10pd8XkXiPQSv'), AIMessage(content='I found the current weather in Louisiana. The temperature is 4.7°C (40.5°F) with clear skies. The wind speed is 14.5 mph coming from the southeast direction.', additional_kwargs={'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 43, 'prompt_tokens': 790, 'total_tokens': 833, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'stop', 'logprobs': None}, id='run-0920c15d-6d58-4353-bd19-cbbf2296461f-0', usage_metadata={'input_tokens': 790, 'output_tokens': 43, 'total_tokens': 833, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}})]}\n",
      "{'messages': [AIMessage(content='I found the current weather in Louisiana. The temperature is 4.7°C (40.5°F) with clear skies. The wind speed is 14.5 mph coming from the southeast direction.', additional_kwargs={'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 43, 'prompt_tokens': 790, 'total_tokens': 833, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'stop', 'logprobs': None}, id='run-0920c15d-6d58-4353-bd19-cbbf2296461f-0', usage_metadata={'input_tokens': 790, 'output_tokens': 43, 'total_tokens': 833, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}})]}\n"
     ]
    }
   ],
   "source": [
    "for event in abot.graph.stream(None, thread):\n",
    "    for v in event.values():\n",
    "        print(v)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 29,
   "id": "c80025ce-5b56-43a9-80ad-059f951b15f5",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "StateSnapshot(values={'messages': [HumanMessage(content='Whats the weather in LA?', additional_kwargs={}, response_metadata={}, id='cab22332-f4eb-4762-aca6-8d5c130362b1'), AIMessage(content='', additional_kwargs={'tool_calls': [{'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'function': {'arguments': '{\"query\":\"weather in Los Angeles\"}', 'name': 'tavily_search_results_json'}, 'type': 'function'}], 'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 22, 'prompt_tokens': 152, 'total_tokens': 174, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'tool_calls', 'logprobs': None}, id='run-116bf1b3-6030-400e-a1d0-8339e7088e31-0', tool_calls=[{'name': 'tavily_search_results_json', 'args': {'query': 'current weather in Louisiana'}, 'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'type': 'tool_call'}], usage_metadata={'input_tokens': 152, 'output_tokens': 22, 'total_tokens': 174, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}}), ToolMessage(content='[{\\'url\\': \\'https://www.weatherapi.com/\\', \\'content\\': \"{\\'location\\': {\\'name\\': \\'Louisiana\\', \\'region\\': \\'Missouri\\', \\'country\\': \\'USA United States of America\\', \\'lat\\': 39.4411, \\'lon\\': -91.0551, \\'tz_id\\': \\'America/Chicago\\', \\'localtime_epoch\\': 1738454236, \\'localtime\\': \\'2025-02-01 17:57\\'}, \\'current\\': {\\'last_updated_epoch\\': 1738453500, \\'last_updated\\': \\'2025-02-01 17:45\\', \\'temp_c\\': 4.7, \\'temp_f\\': 40.5, \\'is_day\\': 0, \\'condition\\': {\\'text\\': \\'Clear\\', \\'icon\\': \\'//cdn.weatherapi.com/weather/64x64/night/113.png\\', \\'code\\': 1000}, \\'wind_mph\\': 14.5, \\'wind_kph\\': 23.4, \\'wind_degree\\': 143, \\'wind_dir\\': \\'SE\\', \\'pressure_mb\\': 1018.0, \\'pressure_in\\': 30.06, \\'precip_mm\\': 0.0, \\'precip_in\\': 0.0, \\'humidity\\': 77, \\'cloud\\': 0, \\'feelslike_c\\': 0.3, \\'feelslike_f\\': 32.5, \\'windchill_c\\': 0.2, \\'windchill_f\\': 32.4, \\'heatindex_c\\': 4.5, \\'heatindex_f\\': 40.1, \\'dewpoint_c\\': 1.5, \\'dewpoint_f\\': 34.8, \\'vis_km\\': 16.0, \\'vis_miles\\': 9.0, \\'uv\\': 0.0, \\'gust_mph\\': 22.2, \\'gust_kph\\': 35.8}}\"}, {\\'url\\': \\'https://forecast.weather.gov/zipcity.php?inputstring=Baton+Rouge,LA\\', \\'content\\': \\'FORECAST  Last update 19 Nov 4:53 am CST Hourly Weather Forecast Low: 71 °F High: 80 °F Tuesday Night High: 68 °F Wednesday Night High: 66 °F Thursday Night High: 63 °F Friday Night Tuesday Night Wednesday Night Thursday Night Friday Night Saturday Night Sunday Night Additional Forecasts and Information ZONE AREA FORECAST FOR EAST BATON ROUGE PARISH, LA Hourly Weather Forecast Air Quality Forecasts Marine Weather Aviation Weather NWS New Orleans/Baton Rouge Home ABOUT THIS FORECAST Hourly Weather Forecast Winter Weather Forecasts Fire Weather Outlooks Surface Weather FORECAST Local Forecast International Forecasts Fire Weather Current Graphical Forecast Maps Aviation Weather Fire Weather Tsunami Warning System International Weather WEATHER SAFETY NWS Education Home National Weather Service Please Contact Us. Disclaimer\\'}]', name='tavily_search_results_json', id='82af75cb-d790-4f3f-805e-98d48e7abe2f', tool_call_id='call_S8IFcTbfn4n10pd8XkXiPQSv'), AIMessage(content='I found the current weather in Louisiana. The temperature is 4.7°C (40.5°F) with clear skies. The wind speed is 14.5 mph coming from the southeast direction.', additional_kwargs={'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 43, 'prompt_tokens': 790, 'total_tokens': 833, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'stop', 'logprobs': None}, id='run-0920c15d-6d58-4353-bd19-cbbf2296461f-0', usage_metadata={'input_tokens': 790, 'output_tokens': 43, 'total_tokens': 833, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}})]}, next=(), config={'configurable': {'thread_id': '3', 'checkpoint_ns': '', 'checkpoint_id': '1efe0f84-5609-6bf8-8004-3835882a6fbc'}}, metadata={'source': 'loop', 'writes': {'llm': {'messages': [AIMessage(content='I found the current weather in Louisiana. The temperature is 4.7°C (40.5°F) with clear skies. The wind speed is 14.5 mph coming from the southeast direction.', additional_kwargs={'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 43, 'prompt_tokens': 790, 'total_tokens': 833, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'stop', 'logprobs': None}, id='run-0920c15d-6d58-4353-bd19-cbbf2296461f-0', usage_metadata={'input_tokens': 790, 'output_tokens': 43, 'total_tokens': 833, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}})]}}, 'thread_id': '3', 'step': 4, 'parents': {}}, created_at='2025-02-01T23:57:19.224466+00:00', parent_config={'configurable': {'thread_id': '3', 'checkpoint_ns': '', 'checkpoint_id': '1efe0f84-441f-6e4d-8003-fc83ce64f323'}}, tasks=())\n",
      "--\n",
      "StateSnapshot(values={'messages': [HumanMessage(content='Whats the weather in LA?', additional_kwargs={}, response_metadata={}, id='cab22332-f4eb-4762-aca6-8d5c130362b1'), AIMessage(content='', additional_kwargs={'tool_calls': [{'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'function': {'arguments': '{\"query\":\"weather in Los Angeles\"}', 'name': 'tavily_search_results_json'}, 'type': 'function'}], 'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 22, 'prompt_tokens': 152, 'total_tokens': 174, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'tool_calls', 'logprobs': None}, id='run-116bf1b3-6030-400e-a1d0-8339e7088e31-0', tool_calls=[{'name': 'tavily_search_results_json', 'args': {'query': 'current weather in Louisiana'}, 'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'type': 'tool_call'}], usage_metadata={'input_tokens': 152, 'output_tokens': 22, 'total_tokens': 174, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}}), ToolMessage(content='[{\\'url\\': \\'https://www.weatherapi.com/\\', \\'content\\': \"{\\'location\\': {\\'name\\': \\'Louisiana\\', \\'region\\': \\'Missouri\\', \\'country\\': \\'USA United States of America\\', \\'lat\\': 39.4411, \\'lon\\': -91.0551, \\'tz_id\\': \\'America/Chicago\\', \\'localtime_epoch\\': 1738454236, \\'localtime\\': \\'2025-02-01 17:57\\'}, \\'current\\': {\\'last_updated_epoch\\': 1738453500, \\'last_updated\\': \\'2025-02-01 17:45\\', \\'temp_c\\': 4.7, \\'temp_f\\': 40.5, \\'is_day\\': 0, \\'condition\\': {\\'text\\': \\'Clear\\', \\'icon\\': \\'//cdn.weatherapi.com/weather/64x64/night/113.png\\', \\'code\\': 1000}, \\'wind_mph\\': 14.5, \\'wind_kph\\': 23.4, \\'wind_degree\\': 143, \\'wind_dir\\': \\'SE\\', \\'pressure_mb\\': 1018.0, \\'pressure_in\\': 30.06, \\'precip_mm\\': 0.0, \\'precip_in\\': 0.0, \\'humidity\\': 77, \\'cloud\\': 0, \\'feelslike_c\\': 0.3, \\'feelslike_f\\': 32.5, \\'windchill_c\\': 0.2, \\'windchill_f\\': 32.4, \\'heatindex_c\\': 4.5, \\'heatindex_f\\': 40.1, \\'dewpoint_c\\': 1.5, \\'dewpoint_f\\': 34.8, \\'vis_km\\': 16.0, \\'vis_miles\\': 9.0, \\'uv\\': 0.0, \\'gust_mph\\': 22.2, \\'gust_kph\\': 35.8}}\"}, {\\'url\\': \\'https://forecast.weather.gov/zipcity.php?inputstring=Baton+Rouge,LA\\', \\'content\\': \\'FORECAST  Last update 19 Nov 4:53 am CST Hourly Weather Forecast Low: 71 °F High: 80 °F Tuesday Night High: 68 °F Wednesday Night High: 66 °F Thursday Night High: 63 °F Friday Night Tuesday Night Wednesday Night Thursday Night Friday Night Saturday Night Sunday Night Additional Forecasts and Information ZONE AREA FORECAST FOR EAST BATON ROUGE PARISH, LA Hourly Weather Forecast Air Quality Forecasts Marine Weather Aviation Weather NWS New Orleans/Baton Rouge Home ABOUT THIS FORECAST Hourly Weather Forecast Winter Weather Forecasts Fire Weather Outlooks Surface Weather FORECAST Local Forecast International Forecasts Fire Weather Current Graphical Forecast Maps Aviation Weather Fire Weather Tsunami Warning System International Weather WEATHER SAFETY NWS Education Home National Weather Service Please Contact Us. Disclaimer\\'}]', name='tavily_search_results_json', id='82af75cb-d790-4f3f-805e-98d48e7abe2f', tool_call_id='call_S8IFcTbfn4n10pd8XkXiPQSv')]}, next=('llm',), config={'configurable': {'thread_id': '3', 'checkpoint_ns': '', 'checkpoint_id': '1efe0f84-441f-6e4d-8003-fc83ce64f323'}}, metadata={'source': 'loop', 'writes': {'action': {'messages': [ToolMessage(content='[{\\'url\\': \\'https://www.weatherapi.com/\\', \\'content\\': \"{\\'location\\': {\\'name\\': \\'Louisiana\\', \\'region\\': \\'Missouri\\', \\'country\\': \\'USA United States of America\\', \\'lat\\': 39.4411, \\'lon\\': -91.0551, \\'tz_id\\': \\'America/Chicago\\', \\'localtime_epoch\\': 1738454236, \\'localtime\\': \\'2025-02-01 17:57\\'}, \\'current\\': {\\'last_updated_epoch\\': 1738453500, \\'last_updated\\': \\'2025-02-01 17:45\\', \\'temp_c\\': 4.7, \\'temp_f\\': 40.5, \\'is_day\\': 0, \\'condition\\': {\\'text\\': \\'Clear\\', \\'icon\\': \\'//cdn.weatherapi.com/weather/64x64/night/113.png\\', \\'code\\': 1000}, \\'wind_mph\\': 14.5, \\'wind_kph\\': 23.4, \\'wind_degree\\': 143, \\'wind_dir\\': \\'SE\\', \\'pressure_mb\\': 1018.0, \\'pressure_in\\': 30.06, \\'precip_mm\\': 0.0, \\'precip_in\\': 0.0, \\'humidity\\': 77, \\'cloud\\': 0, \\'feelslike_c\\': 0.3, \\'feelslike_f\\': 32.5, \\'windchill_c\\': 0.2, \\'windchill_f\\': 32.4, \\'heatindex_c\\': 4.5, \\'heatindex_f\\': 40.1, \\'dewpoint_c\\': 1.5, \\'dewpoint_f\\': 34.8, \\'vis_km\\': 16.0, \\'vis_miles\\': 9.0, \\'uv\\': 0.0, \\'gust_mph\\': 22.2, \\'gust_kph\\': 35.8}}\"}, {\\'url\\': \\'https://forecast.weather.gov/zipcity.php?inputstring=Baton+Rouge,LA\\', \\'content\\': \\'FORECAST  Last update 19 Nov 4:53 am CST Hourly Weather Forecast Low: 71 °F High: 80 °F Tuesday Night High: 68 °F Wednesday Night High: 66 °F Thursday Night High: 63 °F Friday Night Tuesday Night Wednesday Night Thursday Night Friday Night Saturday Night Sunday Night Additional Forecasts and Information ZONE AREA FORECAST FOR EAST BATON ROUGE PARISH, LA Hourly Weather Forecast Air Quality Forecasts Marine Weather Aviation Weather NWS New Orleans/Baton Rouge Home ABOUT THIS FORECAST Hourly Weather Forecast Winter Weather Forecasts Fire Weather Outlooks Surface Weather FORECAST Local Forecast International Forecasts Fire Weather Current Graphical Forecast Maps Aviation Weather Fire Weather Tsunami Warning System International Weather WEATHER SAFETY NWS Education Home National Weather Service Please Contact Us. Disclaimer\\'}]', name='tavily_search_results_json', id='82af75cb-d790-4f3f-805e-98d48e7abe2f', tool_call_id='call_S8IFcTbfn4n10pd8XkXiPQSv')]}}, 'thread_id': '3', 'step': 3, 'parents': {}}, created_at='2025-02-01T23:57:17.346113+00:00', parent_config={'configurable': {'thread_id': '3', 'checkpoint_ns': '', 'checkpoint_id': '1efe0f7d-74aa-69bb-8002-36803ed0c072'}}, tasks=(PregelTask(id='27ab2217-4a60-2e49-598c-39c587262c22', name='llm', path=('__pregel_pull', 'llm'), error=None, interrupts=(), state=None, result={'messages': [AIMessage(content='I found the current weather in Louisiana. The temperature is 4.7°C (40.5°F) with clear skies. The wind speed is 14.5 mph coming from the southeast direction.', additional_kwargs={'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 43, 'prompt_tokens': 790, 'total_tokens': 833, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'stop', 'logprobs': None}, id='run-0920c15d-6d58-4353-bd19-cbbf2296461f-0', usage_metadata={'input_tokens': 790, 'output_tokens': 43, 'total_tokens': 833, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}})]}),))\n",
      "--\n",
      "StateSnapshot(values={'messages': [HumanMessage(content='Whats the weather in LA?', additional_kwargs={}, response_metadata={}, id='cab22332-f4eb-4762-aca6-8d5c130362b1'), AIMessage(content='', additional_kwargs={'tool_calls': [{'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'function': {'arguments': '{\"query\":\"weather in Los Angeles\"}', 'name': 'tavily_search_results_json'}, 'type': 'function'}], 'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 22, 'prompt_tokens': 152, 'total_tokens': 174, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'tool_calls', 'logprobs': None}, id='run-116bf1b3-6030-400e-a1d0-8339e7088e31-0', tool_calls=[{'name': 'tavily_search_results_json', 'args': {'query': 'current weather in Louisiana'}, 'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'type': 'tool_call'}], usage_metadata={'input_tokens': 152, 'output_tokens': 22, 'total_tokens': 174, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}})]}, next=('action',), config={'configurable': {'thread_id': '3', 'checkpoint_ns': '', 'checkpoint_id': '1efe0f7d-74aa-69bb-8002-36803ed0c072'}}, metadata={'source': 'update', 'writes': {'llm': {'messages': [HumanMessage(content='Whats the weather in LA?', additional_kwargs={}, response_metadata={}, id='cab22332-f4eb-4762-aca6-8d5c130362b1'), AIMessage(content='', additional_kwargs={'tool_calls': [{'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'function': {'arguments': '{\"query\":\"weather in Los Angeles\"}', 'name': 'tavily_search_results_json'}, 'type': 'function'}], 'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 22, 'prompt_tokens': 152, 'total_tokens': 174, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'tool_calls', 'logprobs': None}, id='run-116bf1b3-6030-400e-a1d0-8339e7088e31-0', tool_calls=[{'name': 'tavily_search_results_json', 'args': {'query': 'current weather in Louisiana'}, 'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'type': 'tool_call'}], usage_metadata={'input_tokens': 152, 'output_tokens': 22, 'total_tokens': 174, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}})]}}, 'thread_id': '3', 'step': 2, 'parents': {}}, created_at='2025-02-01T23:54:14.531265+00:00', parent_config={'configurable': {'thread_id': '3', 'checkpoint_ns': '', 'checkpoint_id': '1efe0f7b-a224-6387-8001-d8242dd7ad14'}}, tasks=(PregelTask(id='f959be95-7373-11ab-0e38-a29600e70525', name='action', path=('__pregel_pull', 'action'), error=None, interrupts=(), state=None, result={'messages': [ToolMessage(content='[{\\'url\\': \\'https://www.weatherapi.com/\\', \\'content\\': \"{\\'location\\': {\\'name\\': \\'Louisiana\\', \\'region\\': \\'Missouri\\', \\'country\\': \\'USA United States of America\\', \\'lat\\': 39.4411, \\'lon\\': -91.0551, \\'tz_id\\': \\'America/Chicago\\', \\'localtime_epoch\\': 1738454236, \\'localtime\\': \\'2025-02-01 17:57\\'}, \\'current\\': {\\'last_updated_epoch\\': 1738453500, \\'last_updated\\': \\'2025-02-01 17:45\\', \\'temp_c\\': 4.7, \\'temp_f\\': 40.5, \\'is_day\\': 0, \\'condition\\': {\\'text\\': \\'Clear\\', \\'icon\\': \\'//cdn.weatherapi.com/weather/64x64/night/113.png\\', \\'code\\': 1000}, \\'wind_mph\\': 14.5, \\'wind_kph\\': 23.4, \\'wind_degree\\': 143, \\'wind_dir\\': \\'SE\\', \\'pressure_mb\\': 1018.0, \\'pressure_in\\': 30.06, \\'precip_mm\\': 0.0, \\'precip_in\\': 0.0, \\'humidity\\': 77, \\'cloud\\': 0, \\'feelslike_c\\': 0.3, \\'feelslike_f\\': 32.5, \\'windchill_c\\': 0.2, \\'windchill_f\\': 32.4, \\'heatindex_c\\': 4.5, \\'heatindex_f\\': 40.1, \\'dewpoint_c\\': 1.5, \\'dewpoint_f\\': 34.8, \\'vis_km\\': 16.0, \\'vis_miles\\': 9.0, \\'uv\\': 0.0, \\'gust_mph\\': 22.2, \\'gust_kph\\': 35.8}}\"}, {\\'url\\': \\'https://forecast.weather.gov/zipcity.php?inputstring=Baton+Rouge,LA\\', \\'content\\': \\'FORECAST  Last update 19 Nov 4:53 am CST Hourly Weather Forecast Low: 71 °F High: 80 °F Tuesday Night High: 68 °F Wednesday Night High: 66 °F Thursday Night High: 63 °F Friday Night Tuesday Night Wednesday Night Thursday Night Friday Night Saturday Night Sunday Night Additional Forecasts and Information ZONE AREA FORECAST FOR EAST BATON ROUGE PARISH, LA Hourly Weather Forecast Air Quality Forecasts Marine Weather Aviation Weather NWS New Orleans/Baton Rouge Home ABOUT THIS FORECAST Hourly Weather Forecast Winter Weather Forecasts Fire Weather Outlooks Surface Weather FORECAST Local Forecast International Forecasts Fire Weather Current Graphical Forecast Maps Aviation Weather Fire Weather Tsunami Warning System International Weather WEATHER SAFETY NWS Education Home National Weather Service Please Contact Us. Disclaimer\\'}]', name='tavily_search_results_json', tool_call_id='call_S8IFcTbfn4n10pd8XkXiPQSv')]}),))\n",
      "--\n",
      "StateSnapshot(values={'messages': [HumanMessage(content='Whats the weather in LA?', additional_kwargs={}, response_metadata={}, id='cab22332-f4eb-4762-aca6-8d5c130362b1'), AIMessage(content='', additional_kwargs={'tool_calls': [{'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'function': {'arguments': '{\"query\":\"weather in Los Angeles\"}', 'name': 'tavily_search_results_json'}, 'type': 'function'}], 'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 22, 'prompt_tokens': 152, 'total_tokens': 174, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'tool_calls', 'logprobs': None}, id='run-116bf1b3-6030-400e-a1d0-8339e7088e31-0', tool_calls=[{'name': 'tavily_search_results_json', 'args': {'query': 'weather in Los Angeles'}, 'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'type': 'tool_call'}], usage_metadata={'input_tokens': 152, 'output_tokens': 22, 'total_tokens': 174, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}})]}, next=('action',), config={'configurable': {'thread_id': '3', 'checkpoint_ns': '', 'checkpoint_id': '1efe0f7b-a224-6387-8001-d8242dd7ad14'}}, metadata={'source': 'loop', 'writes': {'llm': {'messages': [AIMessage(content='', additional_kwargs={'tool_calls': [{'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'function': {'arguments': '{\"query\":\"weather in Los Angeles\"}', 'name': 'tavily_search_results_json'}, 'type': 'function'}], 'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 22, 'prompt_tokens': 152, 'total_tokens': 174, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'tool_calls', 'logprobs': None}, id='run-116bf1b3-6030-400e-a1d0-8339e7088e31-0', tool_calls=[{'name': 'tavily_search_results_json', 'args': {'query': 'weather in Los Angeles'}, 'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'type': 'tool_call'}], usage_metadata={'input_tokens': 152, 'output_tokens': 22, 'total_tokens': 174, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}})]}}, 'thread_id': '3', 'step': 1, 'parents': {}}, created_at='2025-02-01T23:53:25.612578+00:00', parent_config={'configurable': {'thread_id': '3', 'checkpoint_ns': '', 'checkpoint_id': '1efe0f7b-9878-68ff-8000-838b1e05c511'}}, tasks=(PregelTask(id='4469c5af-8ee9-3033-98d7-c09acc4c7551', name='action', path=('__pregel_pull', 'action'), error=None, interrupts=(), state=None, result=None),))\n",
      "--\n",
      "StateSnapshot(values={'messages': [HumanMessage(content='Whats the weather in LA?', additional_kwargs={}, response_metadata={}, id='cab22332-f4eb-4762-aca6-8d5c130362b1')]}, next=('llm',), config={'configurable': {'thread_id': '3', 'checkpoint_ns': '', 'checkpoint_id': '1efe0f7b-9878-68ff-8000-838b1e05c511'}}, metadata={'source': 'loop', 'writes': None, 'thread_id': '3', 'step': 0, 'parents': {}}, created_at='2025-02-01T23:53:24.598569+00:00', parent_config={'configurable': {'thread_id': '3', 'checkpoint_ns': '', 'checkpoint_id': '1efe0f7b-9872-651e-bfff-f3fbe69674c0'}}, tasks=(PregelTask(id='41da4042-8071-4486-c76e-b906cb872e81', name='llm', path=('__pregel_pull', 'llm'), error=None, interrupts=(), state=None, result={'messages': [AIMessage(content='', additional_kwargs={'tool_calls': [{'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'function': {'arguments': '{\"query\":\"weather in Los Angeles\"}', 'name': 'tavily_search_results_json'}, 'type': 'function'}], 'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 22, 'prompt_tokens': 152, 'total_tokens': 174, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'tool_calls', 'logprobs': None}, id='run-116bf1b3-6030-400e-a1d0-8339e7088e31-0', tool_calls=[{'name': 'tavily_search_results_json', 'args': {'query': 'weather in Los Angeles'}, 'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'type': 'tool_call'}], usage_metadata={'input_tokens': 152, 'output_tokens': 22, 'total_tokens': 174, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}})]}),))\n",
      "--\n",
      "StateSnapshot(values={'messages': []}, next=('__start__',), config={'configurable': {'thread_id': '3', 'checkpoint_ns': '', 'checkpoint_id': '1efe0f7b-9872-651e-bfff-f3fbe69674c0'}}, metadata={'source': 'input', 'writes': {'__start__': {'messages': [HumanMessage(content='Whats the weather in LA?', additional_kwargs={}, response_metadata={})]}}, 'thread_id': '3', 'step': -1, 'parents': {}}, created_at='2025-02-01T23:53:24.596031+00:00', parent_config=None, tasks=(PregelTask(id='915b0bae-5dce-2114-4ac0-2d6bc0a5a8a2', name='__start__', path=('__pregel_pull', '__start__'), error=None, interrupts=(), state=None, result={'messages': [HumanMessage(content='Whats the weather in LA?', additional_kwargs={}, response_metadata={}, id='cab22332-f4eb-4762-aca6-8d5c130362b1')]}),))\n",
      "--\n"
     ]
    }
   ],
   "source": [
    "# Time Travel\n",
    "\n",
    "states = []\n",
    "for state in abot.graph.get_state_history(thread):\n",
    "    print(state)\n",
    "    print('--')\n",
    "    states.append(state)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 32,
   "id": "ffdef1a2-8d28-4024-89f1-d1cb2675cf69",
   "metadata": {},
   "outputs": [],
   "source": [
    "# To fetch the same state as was filmed, the offset below is changed \n",
    "# to -3 from -1. This accounts for the initial state __start__ and the \n",
    "# first state that are now stored to state memory with the latest version \n",
    "# of software.\n",
    "\n",
    "to_replay = states[-3]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 33,
   "id": "ea773197-23df-44ae-be83-97f2ef012f28",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "StateSnapshot(values={'messages': [HumanMessage(content='Whats the weather in LA?', additional_kwargs={}, response_metadata={}, id='cab22332-f4eb-4762-aca6-8d5c130362b1'), AIMessage(content='', additional_kwargs={'tool_calls': [{'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'function': {'arguments': '{\"query\":\"weather in Los Angeles\"}', 'name': 'tavily_search_results_json'}, 'type': 'function'}], 'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 22, 'prompt_tokens': 152, 'total_tokens': 174, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'tool_calls', 'logprobs': None}, id='run-116bf1b3-6030-400e-a1d0-8339e7088e31-0', tool_calls=[{'name': 'tavily_search_results_json', 'args': {'query': 'weather in Los Angeles'}, 'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'type': 'tool_call'}], usage_metadata={'input_tokens': 152, 'output_tokens': 22, 'total_tokens': 174, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}})]}, next=('action',), config={'configurable': {'thread_id': '3', 'checkpoint_ns': '', 'checkpoint_id': '1efe0f7b-a224-6387-8001-d8242dd7ad14'}}, metadata={'source': 'loop', 'writes': {'llm': {'messages': [AIMessage(content='', additional_kwargs={'tool_calls': [{'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'function': {'arguments': '{\"query\":\"weather in Los Angeles\"}', 'name': 'tavily_search_results_json'}, 'type': 'function'}], 'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 22, 'prompt_tokens': 152, 'total_tokens': 174, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'tool_calls', 'logprobs': None}, id='run-116bf1b3-6030-400e-a1d0-8339e7088e31-0', tool_calls=[{'name': 'tavily_search_results_json', 'args': {'query': 'weather in Los Angeles'}, 'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'type': 'tool_call'}], usage_metadata={'input_tokens': 152, 'output_tokens': 22, 'total_tokens': 174, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}})]}}, 'thread_id': '3', 'step': 1, 'parents': {}}, created_at='2025-02-01T23:53:25.612578+00:00', parent_config={'configurable': {'thread_id': '3', 'checkpoint_ns': '', 'checkpoint_id': '1efe0f7b-9878-68ff-8000-838b1e05c511'}}, tasks=(PregelTask(id='4469c5af-8ee9-3033-98d7-c09acc4c7551', name='action', path=('__pregel_pull', 'action'), error=None, interrupts=(), state=None, result=None),))"
      ]
     },
     "execution_count": 33,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "to_replay"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 34,
   "id": "38fd0c09-d692-478e-a726-862842cc4c80",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Calling: {'name': 'tavily_search_results_json', 'args': {'query': 'weather in Los Angeles'}, 'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'type': 'tool_call'}\n",
      "Back to the model!\n",
      "{'messages': [ToolMessage(content='[{\\'url\\': \\'https://www.weatherapi.com/\\', \\'content\\': \"{\\'location\\': {\\'name\\': \\'Los Angeles\\', \\'region\\': \\'California\\', \\'country\\': \\'United States of America\\', \\'lat\\': 34.0522, \\'lon\\': -118.2428, \\'tz_id\\': \\'America/Los_Angeles\\', \\'localtime_epoch\\': 1738454284, \\'localtime\\': \\'2025-02-01 15:58\\'}, \\'current\\': {\\'last_updated_epoch\\': 1738453500, \\'last_updated\\': \\'2025-02-01 15:45\\', \\'temp_c\\': 17.8, \\'temp_f\\': 64.0, \\'is_day\\': 1, \\'condition\\': {\\'text\\': \\'Sunny\\', \\'icon\\': \\'//cdn.weatherapi.com/weather/64x64/day/113.png\\', \\'code\\': 1000}, \\'wind_mph\\': 6.7, \\'wind_kph\\': 10.8, \\'wind_degree\\': 248, \\'wind_dir\\': \\'WSW\\', \\'pressure_mb\\': 1019.0, \\'pressure_in\\': 30.09, \\'precip_mm\\': 0.0, \\'precip_in\\': 0.0, \\'humidity\\': 56, \\'cloud\\': 0, \\'feelslike_c\\': 17.8, \\'feelslike_f\\': 64.0, \\'windchill_c\\': 20.1, \\'windchill_f\\': 68.3, \\'heatindex_c\\': 20.1, \\'heatindex_f\\': 68.3, \\'dewpoint_c\\': 9.0, \\'dewpoint_f\\': 48.1, \\'vis_km\\': 16.0, \\'vis_miles\\': 9.0, \\'uv\\': 1.1, \\'gust_mph\\': 7.7, \\'gust_kph\\': 12.4}}\"}, {\\'url\\': \\'https://weathershogun.com/weather/usa/ca/los-angeles/451/january/2025-01-02\\', \\'content\\': \"Thursday, January 2, 2025. Los Angeles, CA - Weather Forecast Los Angeles, CA Home Contact Browse States Privacy Policy Terms and Conditions °F) °C) Today Tomorrow Hourly 7 days 30 days January Los Angeles, California Weather: Air Quality Alert (Issued when air pollution levels are unhealthy, especially for sensitive groups.) Thursday, January 2, 2025 Day 77° Night 52° Precipitation 0 % Wind 4 mph UV Index (0 - 11+) 3 Friday Hourly Today Tomorrow 7 days 30 days Weather Forecast History Last Year\\'s Weather on This Day (January 2, 2024) Day 70° Night 52° Wind 4 mph UV Index Precipitation Please note that while we strive for accuracy, the information provided may not always be correct. Use at your own risk.\"}]', name='tavily_search_results_json', id='a670e710-41a8-4b2b-9229-de81d526c699', tool_call_id='call_S8IFcTbfn4n10pd8XkXiPQSv')]}\n",
      "{'messages': [HumanMessage(content='Whats the weather in LA?', additional_kwargs={}, response_metadata={}, id='cab22332-f4eb-4762-aca6-8d5c130362b1'), AIMessage(content='', additional_kwargs={'tool_calls': [{'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'function': {'arguments': '{\"query\":\"weather in Los Angeles\"}', 'name': 'tavily_search_results_json'}, 'type': 'function'}], 'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 22, 'prompt_tokens': 152, 'total_tokens': 174, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'tool_calls', 'logprobs': None}, id='run-116bf1b3-6030-400e-a1d0-8339e7088e31-0', tool_calls=[{'name': 'tavily_search_results_json', 'args': {'query': 'weather in Los Angeles'}, 'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'type': 'tool_call'}], usage_metadata={'input_tokens': 152, 'output_tokens': 22, 'total_tokens': 174, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}}), ToolMessage(content='[{\\'url\\': \\'https://www.weatherapi.com/\\', \\'content\\': \"{\\'location\\': {\\'name\\': \\'Los Angeles\\', \\'region\\': \\'California\\', \\'country\\': \\'United States of America\\', \\'lat\\': 34.0522, \\'lon\\': -118.2428, \\'tz_id\\': \\'America/Los_Angeles\\', \\'localtime_epoch\\': 1738454284, \\'localtime\\': \\'2025-02-01 15:58\\'}, \\'current\\': {\\'last_updated_epoch\\': 1738453500, \\'last_updated\\': \\'2025-02-01 15:45\\', \\'temp_c\\': 17.8, \\'temp_f\\': 64.0, \\'is_day\\': 1, \\'condition\\': {\\'text\\': \\'Sunny\\', \\'icon\\': \\'//cdn.weatherapi.com/weather/64x64/day/113.png\\', \\'code\\': 1000}, \\'wind_mph\\': 6.7, \\'wind_kph\\': 10.8, \\'wind_degree\\': 248, \\'wind_dir\\': \\'WSW\\', \\'pressure_mb\\': 1019.0, \\'pressure_in\\': 30.09, \\'precip_mm\\': 0.0, \\'precip_in\\': 0.0, \\'humidity\\': 56, \\'cloud\\': 0, \\'feelslike_c\\': 17.8, \\'feelslike_f\\': 64.0, \\'windchill_c\\': 20.1, \\'windchill_f\\': 68.3, \\'heatindex_c\\': 20.1, \\'heatindex_f\\': 68.3, \\'dewpoint_c\\': 9.0, \\'dewpoint_f\\': 48.1, \\'vis_km\\': 16.0, \\'vis_miles\\': 9.0, \\'uv\\': 1.1, \\'gust_mph\\': 7.7, \\'gust_kph\\': 12.4}}\"}, {\\'url\\': \\'https://weathershogun.com/weather/usa/ca/los-angeles/451/january/2025-01-02\\', \\'content\\': \"Thursday, January 2, 2025. Los Angeles, CA - Weather Forecast Los Angeles, CA Home Contact Browse States Privacy Policy Terms and Conditions °F) °C) Today Tomorrow Hourly 7 days 30 days January Los Angeles, California Weather: Air Quality Alert (Issued when air pollution levels are unhealthy, especially for sensitive groups.) Thursday, January 2, 2025 Day 77° Night 52° Precipitation 0 % Wind 4 mph UV Index (0 - 11+) 3 Friday Hourly Today Tomorrow 7 days 30 days Weather Forecast History Last Year\\'s Weather on This Day (January 2, 2024) Day 70° Night 52° Wind 4 mph UV Index Precipitation Please note that while we strive for accuracy, the information provided may not always be correct. Use at your own risk.\"}]', name='tavily_search_results_json', id='a670e710-41a8-4b2b-9229-de81d526c699', tool_call_id='call_S8IFcTbfn4n10pd8XkXiPQSv'), AIMessage(content='The current weather in Los Angeles is sunny with a temperature of 64.0°F (17.8°C). The wind is blowing at 10.8 km/h coming from the WSW direction. The humidity is at 56%, and there is no precipitation at the moment. The visibility is 16.0 km, and the UV index is 1.1.', additional_kwargs={'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 78, 'prompt_tokens': 812, 'total_tokens': 890, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'stop', 'logprobs': None}, id='run-d1df1b5f-9513-442d-b721-b757c5aab2de-0', usage_metadata={'input_tokens': 812, 'output_tokens': 78, 'total_tokens': 890, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}})]}\n",
      "{'messages': [AIMessage(content='The current weather in Los Angeles is sunny with a temperature of 64.0°F (17.8°C). The wind is blowing at 10.8 km/h coming from the WSW direction. The humidity is at 56%, and there is no precipitation at the moment. The visibility is 16.0 km, and the UV index is 1.1.', additional_kwargs={'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 78, 'prompt_tokens': 812, 'total_tokens': 890, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'stop', 'logprobs': None}, id='run-d1df1b5f-9513-442d-b721-b757c5aab2de-0', usage_metadata={'input_tokens': 812, 'output_tokens': 78, 'total_tokens': 890, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}})]}\n"
     ]
    }
   ],
   "source": [
    "for event in abot.graph.stream(None, to_replay.config):\n",
    "    for k, v in event.items():\n",
    "        print(v)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 35,
   "id": "66f9dfe3-d89d-4972-a444-b839e10dcf58",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "StateSnapshot(values={'messages': [HumanMessage(content='Whats the weather in LA?', additional_kwargs={}, response_metadata={}, id='cab22332-f4eb-4762-aca6-8d5c130362b1'), AIMessage(content='', additional_kwargs={'tool_calls': [{'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'function': {'arguments': '{\"query\":\"weather in Los Angeles\"}', 'name': 'tavily_search_results_json'}, 'type': 'function'}], 'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 22, 'prompt_tokens': 152, 'total_tokens': 174, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'tool_calls', 'logprobs': None}, id='run-116bf1b3-6030-400e-a1d0-8339e7088e31-0', tool_calls=[{'name': 'tavily_search_results_json', 'args': {'query': 'weather in Los Angeles'}, 'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'type': 'tool_call'}], usage_metadata={'input_tokens': 152, 'output_tokens': 22, 'total_tokens': 174, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}})]}, next=('action',), config={'configurable': {'thread_id': '3', 'checkpoint_ns': '', 'checkpoint_id': '1efe0f7b-a224-6387-8001-d8242dd7ad14'}}, metadata={'source': 'loop', 'writes': {'llm': {'messages': [AIMessage(content='', additional_kwargs={'tool_calls': [{'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'function': {'arguments': '{\"query\":\"weather in Los Angeles\"}', 'name': 'tavily_search_results_json'}, 'type': 'function'}], 'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 22, 'prompt_tokens': 152, 'total_tokens': 174, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'tool_calls', 'logprobs': None}, id='run-116bf1b3-6030-400e-a1d0-8339e7088e31-0', tool_calls=[{'name': 'tavily_search_results_json', 'args': {'query': 'weather in Los Angeles'}, 'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'type': 'tool_call'}], usage_metadata={'input_tokens': 152, 'output_tokens': 22, 'total_tokens': 174, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}})]}}, 'thread_id': '3', 'step': 1, 'parents': {}}, created_at='2025-02-01T23:53:25.612578+00:00', parent_config={'configurable': {'thread_id': '3', 'checkpoint_ns': '', 'checkpoint_id': '1efe0f7b-9878-68ff-8000-838b1e05c511'}}, tasks=(PregelTask(id='4469c5af-8ee9-3033-98d7-c09acc4c7551', name='action', path=('__pregel_pull', 'action'), error=None, interrupts=(), state=None, result=None),))"
      ]
     },
     "execution_count": 35,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# GO back in time and edit\n",
    "\n",
    "to_replay"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 36,
   "id": "85673be9-5e17-4c58-ab25-6a29e17014dd",
   "metadata": {},
   "outputs": [],
   "source": [
    "_id = to_replay.values['messages'][-1].tool_calls[0]['id']\n",
    "to_replay.values['messages'][-1].tool_calls = [{'name': 'tavily_search_results_json',\n",
    "  'args': {'query': 'current weather in LA, accuweather'},\n",
    "  'id': _id}]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 37,
   "id": "e041221d-87d0-4d5a-a5c4-ced1ccf0c3a3",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "'call_S8IFcTbfn4n10pd8XkXiPQSv'"
      ]
     },
     "execution_count": 37,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "_id"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 38,
   "id": "e8cd7bc9-8c14-421c-9a86-2fccdef17ba6",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "{'messages': [HumanMessage(content='Whats the weather in LA?', additional_kwargs={}, response_metadata={}, id='cab22332-f4eb-4762-aca6-8d5c130362b1'), AIMessage(content='', additional_kwargs={'tool_calls': [{'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'function': {'arguments': '{\"query\":\"weather in Los Angeles\"}', 'name': 'tavily_search_results_json'}, 'type': 'function'}], 'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 22, 'prompt_tokens': 152, 'total_tokens': 174, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'tool_calls', 'logprobs': None}, id='run-116bf1b3-6030-400e-a1d0-8339e7088e31-0', tool_calls=[{'name': 'tavily_search_results_json', 'args': {'query': 'current weather in LA, accuweather'}, 'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv'}], usage_metadata={'input_tokens': 152, 'output_tokens': 22, 'total_tokens': 174, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}})]}\n"
     ]
    }
   ],
   "source": [
    "branch_state = abot.graph.update_state(to_replay.config, to_replay.values)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 40,
   "id": "d07b0970-3108-46f5-91ed-776e986c00eb",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Calling: {'name': 'tavily_search_results_json', 'args': {'query': 'current weather in LA, accuweather'}, 'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'type': 'tool_call'}\n",
      "Back to the model!\n",
      "{'messages': [ToolMessage(content='[{\\'url\\': \\'https://www.weatherapi.com/\\', \\'content\\': \"{\\'location\\': {\\'name\\': \\'Los Angeles\\', \\'region\\': \\'California\\', \\'country\\': \\'United States of America\\', \\'lat\\': 34.0522, \\'lon\\': -118.2428, \\'tz_id\\': \\'America/Los_Angeles\\', \\'localtime_epoch\\': 1738454330, \\'localtime\\': \\'2025-02-01 15:58\\'}, \\'current\\': {\\'last_updated_epoch\\': 1738453500, \\'last_updated\\': \\'2025-02-01 15:45\\', \\'temp_c\\': 17.8, \\'temp_f\\': 64.0, \\'is_day\\': 1, \\'condition\\': {\\'text\\': \\'Sunny\\', \\'icon\\': \\'//cdn.weatherapi.com/weather/64x64/day/113.png\\', \\'code\\': 1000}, \\'wind_mph\\': 6.7, \\'wind_kph\\': 10.8, \\'wind_degree\\': 248, \\'wind_dir\\': \\'WSW\\', \\'pressure_mb\\': 1019.0, \\'pressure_in\\': 30.09, \\'precip_mm\\': 0.0, \\'precip_in\\': 0.0, \\'humidity\\': 56, \\'cloud\\': 0, \\'feelslike_c\\': 17.8, \\'feelslike_f\\': 64.0, \\'windchill_c\\': 20.1, \\'windchill_f\\': 68.3, \\'heatindex_c\\': 20.1, \\'heatindex_f\\': 68.3, \\'dewpoint_c\\': 9.0, \\'dewpoint_f\\': 48.1, \\'vis_km\\': 16.0, \\'vis_miles\\': 9.0, \\'uv\\': 1.1, \\'gust_mph\\': 7.7, \\'gust_kph\\': 12.4}}\"}, {\\'url\\': \\'https://www.accuweather.com/en/us/los-angeles/90012/current-weather/347625\\', \\'content\\': \\'Los Angeles, CA Current Weather | AccuWeather Winter Weather Storm to deliver snow, rain and wind to Midwest, Northeast 1 hour ago Weather Forecasts Bomb cyclone with atmospheric river to blast Northwest 1 hour ago Astronomy Moon to align with Jupiter, Mars on Monday night 8 hours ago Severe Weather Severe thunderstorms continue to rumble through southern Plains 1 hour ago Hurricane Tropical Rainstorm Sara, now in Gulf, to drench southern US coast 1 hour ago  More Stories AccuWeather Ready Business Health Hurricane Leisure and Recreation Severe Weather Space and Astronomy Sports Travel Weather News Winter Center AccuWeather Ready Business Health Hurricane Leisure and Recreation Severe Weather Space and Astronomy Sports Travel Weather News Winter Center\\'}]', name='tavily_search_results_json', id='5ee5918e-5226-4b86-b3f2-72b3f9a704b7', tool_call_id='call_S8IFcTbfn4n10pd8XkXiPQSv')]}\n",
      "\n",
      "{'messages': [HumanMessage(content='Whats the weather in LA?', additional_kwargs={}, response_metadata={}, id='cab22332-f4eb-4762-aca6-8d5c130362b1'), AIMessage(content='', additional_kwargs={'tool_calls': [{'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'function': {'arguments': '{\"query\":\"weather in Los Angeles\"}', 'name': 'tavily_search_results_json'}, 'type': 'function'}], 'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 22, 'prompt_tokens': 152, 'total_tokens': 174, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'tool_calls', 'logprobs': None}, id='run-116bf1b3-6030-400e-a1d0-8339e7088e31-0', tool_calls=[{'name': 'tavily_search_results_json', 'args': {'query': 'current weather in LA, accuweather'}, 'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'type': 'tool_call'}], usage_metadata={'input_tokens': 152, 'output_tokens': 22, 'total_tokens': 174, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}}), ToolMessage(content='[{\\'url\\': \\'https://www.weatherapi.com/\\', \\'content\\': \"{\\'location\\': {\\'name\\': \\'Los Angeles\\', \\'region\\': \\'California\\', \\'country\\': \\'United States of America\\', \\'lat\\': 34.0522, \\'lon\\': -118.2428, \\'tz_id\\': \\'America/Los_Angeles\\', \\'localtime_epoch\\': 1738454330, \\'localtime\\': \\'2025-02-01 15:58\\'}, \\'current\\': {\\'last_updated_epoch\\': 1738453500, \\'last_updated\\': \\'2025-02-01 15:45\\', \\'temp_c\\': 17.8, \\'temp_f\\': 64.0, \\'is_day\\': 1, \\'condition\\': {\\'text\\': \\'Sunny\\', \\'icon\\': \\'//cdn.weatherapi.com/weather/64x64/day/113.png\\', \\'code\\': 1000}, \\'wind_mph\\': 6.7, \\'wind_kph\\': 10.8, \\'wind_degree\\': 248, \\'wind_dir\\': \\'WSW\\', \\'pressure_mb\\': 1019.0, \\'pressure_in\\': 30.09, \\'precip_mm\\': 0.0, \\'precip_in\\': 0.0, \\'humidity\\': 56, \\'cloud\\': 0, \\'feelslike_c\\': 17.8, \\'feelslike_f\\': 64.0, \\'windchill_c\\': 20.1, \\'windchill_f\\': 68.3, \\'heatindex_c\\': 20.1, \\'heatindex_f\\': 68.3, \\'dewpoint_c\\': 9.0, \\'dewpoint_f\\': 48.1, \\'vis_km\\': 16.0, \\'vis_miles\\': 9.0, \\'uv\\': 1.1, \\'gust_mph\\': 7.7, \\'gust_kph\\': 12.4}}\"}, {\\'url\\': \\'https://www.accuweather.com/en/us/los-angeles/90012/current-weather/347625\\', \\'content\\': \\'Los Angeles, CA Current Weather | AccuWeather Winter Weather Storm to deliver snow, rain and wind to Midwest, Northeast 1 hour ago Weather Forecasts Bomb cyclone with atmospheric river to blast Northwest 1 hour ago Astronomy Moon to align with Jupiter, Mars on Monday night 8 hours ago Severe Weather Severe thunderstorms continue to rumble through southern Plains 1 hour ago Hurricane Tropical Rainstorm Sara, now in Gulf, to drench southern US coast 1 hour ago  More Stories AccuWeather Ready Business Health Hurricane Leisure and Recreation Severe Weather Space and Astronomy Sports Travel Weather News Winter Center AccuWeather Ready Business Health Hurricane Leisure and Recreation Severe Weather Space and Astronomy Sports Travel Weather News Winter Center\\'}]', name='tavily_search_results_json', id='5ee5918e-5226-4b86-b3f2-72b3f9a704b7', tool_call_id='call_S8IFcTbfn4n10pd8XkXiPQSv'), AIMessage(content='The current weather in Los Angeles is sunny with a temperature of 64.0°F (17.8°C). The wind speed is 6.7 mph (10.8 kph) coming from the WSW direction. The humidity is at 56%, and there is no precipitation.', additional_kwargs={'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 61, 'prompt_tokens': 777, 'total_tokens': 838, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'stop', 'logprobs': None}, id='run-9af55cf6-7e6c-4bb6-b894-85e5f313bee2-0', usage_metadata={'input_tokens': 777, 'output_tokens': 61, 'total_tokens': 838, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}})]}\n",
      "{'messages': [AIMessage(content='The current weather in Los Angeles is sunny with a temperature of 64.0°F (17.8°C). The wind speed is 6.7 mph (10.8 kph) coming from the WSW direction. The humidity is at 56%, and there is no precipitation.', additional_kwargs={'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 61, 'prompt_tokens': 777, 'total_tokens': 838, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'stop', 'logprobs': None}, id='run-9af55cf6-7e6c-4bb6-b894-85e5f313bee2-0', usage_metadata={'input_tokens': 777, 'output_tokens': 61, 'total_tokens': 838, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}})]}\n",
      "\n"
     ]
    }
   ],
   "source": [
    "for event in abot.graph.stream(None, branch_state):\n",
    "    for k, v in event.items():\n",
    "        if k != \"__end__\":\n",
    "            print(v, end=\"\\n\\n\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 41,
   "id": "270b3de5-33ef-4d4c-aa3b-cf990b90a47e",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "StateSnapshot(values={'messages': [HumanMessage(content='Whats the weather in LA?', additional_kwargs={}, response_metadata={}, id='cab22332-f4eb-4762-aca6-8d5c130362b1'), AIMessage(content='', additional_kwargs={'tool_calls': [{'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'function': {'arguments': '{\"query\":\"weather in Los Angeles\"}', 'name': 'tavily_search_results_json'}, 'type': 'function'}], 'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 22, 'prompt_tokens': 152, 'total_tokens': 174, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'tool_calls', 'logprobs': None}, id='run-116bf1b3-6030-400e-a1d0-8339e7088e31-0', tool_calls=[{'name': 'tavily_search_results_json', 'args': {'query': 'current weather in LA, accuweather'}, 'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv'}], usage_metadata={'input_tokens': 152, 'output_tokens': 22, 'total_tokens': 174, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}})]}, next=('action',), config={'configurable': {'thread_id': '3', 'checkpoint_ns': '', 'checkpoint_id': '1efe0f7b-a224-6387-8001-d8242dd7ad14'}}, metadata={'source': 'loop', 'writes': {'llm': {'messages': [AIMessage(content='', additional_kwargs={'tool_calls': [{'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'function': {'arguments': '{\"query\":\"weather in Los Angeles\"}', 'name': 'tavily_search_results_json'}, 'type': 'function'}], 'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 22, 'prompt_tokens': 152, 'total_tokens': 174, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'tool_calls', 'logprobs': None}, id='run-116bf1b3-6030-400e-a1d0-8339e7088e31-0', tool_calls=[{'name': 'tavily_search_results_json', 'args': {'query': 'weather in Los Angeles'}, 'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'type': 'tool_call'}], usage_metadata={'input_tokens': 152, 'output_tokens': 22, 'total_tokens': 174, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}})]}}, 'thread_id': '3', 'step': 1, 'parents': {}}, created_at='2025-02-01T23:53:25.612578+00:00', parent_config={'configurable': {'thread_id': '3', 'checkpoint_ns': '', 'checkpoint_id': '1efe0f7b-9878-68ff-8000-838b1e05c511'}}, tasks=(PregelTask(id='4469c5af-8ee9-3033-98d7-c09acc4c7551', name='action', path=('__pregel_pull', 'action'), error=None, interrupts=(), state=None, result=None),))"
      ]
     },
     "execution_count": 41,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# Add message to a state at a given time\n",
    "\n",
    "to_replay"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 42,
   "id": "da451bd2-f202-4d32-9332-514f8021277a",
   "metadata": {},
   "outputs": [],
   "source": [
    "_id = to_replay.values['messages'][-1].tool_calls[0]['id']"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 43,
   "id": "1e3231c7-1b7f-48de-a3ed-2e005ed153d9",
   "metadata": {},
   "outputs": [],
   "source": [
    "state_update = {\"messages\": [ToolMessage(\n",
    "    tool_call_id=_id,\n",
    "    name=\"tavily_search_results_json\",\n",
    "    content=\"54 degree celcius\",\n",
    ")]}"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 44,
   "id": "b3ba3496-9b78-4a43-a20b-6ee7a0755341",
   "metadata": {},
   "outputs": [],
   "source": [
    "branch_and_add = abot.graph.update_state(\n",
    "    to_replay.config, \n",
    "    state_update, \n",
    "    as_node=\"action\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 45,
   "id": "b259ed5d-0397-4229-b2f1-a60c8ef59d53",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "{'messages': [HumanMessage(content='Whats the weather in LA?', additional_kwargs={}, response_metadata={}, id='cab22332-f4eb-4762-aca6-8d5c130362b1'), AIMessage(content='', additional_kwargs={'tool_calls': [{'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'function': {'arguments': '{\"query\":\"weather in Los Angeles\"}', 'name': 'tavily_search_results_json'}, 'type': 'function'}], 'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 22, 'prompt_tokens': 152, 'total_tokens': 174, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'tool_calls', 'logprobs': None}, id='run-116bf1b3-6030-400e-a1d0-8339e7088e31-0', tool_calls=[{'name': 'tavily_search_results_json', 'args': {'query': 'weather in Los Angeles'}, 'id': 'call_S8IFcTbfn4n10pd8XkXiPQSv', 'type': 'tool_call'}], usage_metadata={'input_tokens': 152, 'output_tokens': 22, 'total_tokens': 174, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}}), ToolMessage(content='54 degree celcius', name='tavily_search_results_json', id='ee570a92-25e7-4b75-9435-37781f15f6e5', tool_call_id='call_S8IFcTbfn4n10pd8XkXiPQSv'), AIMessage(content='The weather in Los Angeles is currently 54 degrees Celsius.', additional_kwargs={'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 14, 'prompt_tokens': 190, 'total_tokens': 204, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'stop', 'logprobs': None}, id='run-db3f7ac5-77d7-4b61-b354-31cacb8a160e-0', usage_metadata={'input_tokens': 190, 'output_tokens': 14, 'total_tokens': 204, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}})]}\n",
      "{'messages': [AIMessage(content='The weather in Los Angeles is currently 54 degrees Celsius.', additional_kwargs={'refusal': None}, response_metadata={'token_usage': {'completion_tokens': 14, 'prompt_tokens': 190, 'total_tokens': 204, 'completion_tokens_details': {'accepted_prediction_tokens': 0, 'audio_tokens': 0, 'reasoning_tokens': 0, 'rejected_prediction_tokens': 0}, 'prompt_tokens_details': {'audio_tokens': 0, 'cached_tokens': 0}}, 'model_name': 'gpt-3.5-turbo-0125', 'system_fingerprint': None, 'finish_reason': 'stop', 'logprobs': None}, id='run-db3f7ac5-77d7-4b61-b354-31cacb8a160e-0', usage_metadata={'input_tokens': 190, 'output_tokens': 14, 'total_tokens': 204, 'input_token_details': {'audio': 0, 'cache_read': 0}, 'output_token_details': {'audio': 0, 'reasoning': 0}})]}\n"
     ]
    }
   ],
   "source": [
    "for event in abot.graph.stream(None, branch_and_add):\n",
    "    for k, v in event.items():\n",
    "        print(v)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "b3e67208-8f36-4db0-af90-620ec2a6e8bd",
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.12.7"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}