diff --git a/examples/customer-support/customer-support.ipynb b/examples/customer-support/customer-support.ipynb index 1bda11bd7..ee7c57086 100644 --- a/examples/customer-support/customer-support.ipynb +++ b/examples/customer-support/customer-support.ipynb @@ -1023,7 +1023,8 @@ "\n", " def __call__(self, state: State, config: RunnableConfig):\n", " while True:\n", - " passenger_id = config.get(\"passenger_id\", None)\n", + " configuration = config.get(\"configurable\", {})\n", + " passenger_id = configuration.get(\"passenger_id\", None)\n", " state = {**state, \"user_info\": passenger_id}\n", " result = self.runnable.invoke(state)\n", " # If the LLM happens to return an empty response, we will re-prompt it\n", @@ -1840,7 +1841,6 @@ "\n", " def __call__(self, state: State, config: RunnableConfig):\n", " while True:\n", - " passenger_id = config.get(\"passenger_id\", None)\n", " result = self.runnable.invoke(state)\n", " # If the LLM happens to return an empty response, we will re-prompt it\n", " # for an actual response.\n",