notebook nits (#17628)

run-llama · Jan 25, 2025 · d71bedb · d71bedb
1 parent 957bde1
commit d71bedb
Show file tree

Hide file tree

Showing 2 changed files with 59 additions and 82 deletions.
diff --git a/docs/docs/examples/agent/agent_workflow_basic.ipynb b/docs/docs/examples/agent/agent_workflow_basic.ipynb
@@ -332,7 +332,9 @@
     "\n",
     "\n",
     "async def set_name(ctx: Context, name: str) -> str:\n",
-    "    await ctx.set(\"name\", name)\n",
+    "    state = await ctx.get(\"state\")\n",
+    "    state[\"name\"] = name\n",
+    "    await ctx.set(\"state\", state)\n",
     "    return f\"Name set to {name}\"\n",
     "\n",
     "\n",
@@ -348,8 +350,8 @@
     "response = await workflow.run(user_msg=\"My name is Logan\", ctx=ctx)\n",
     "print(str(response))\n",
     "\n",
-    "name = await ctx.get(\"name\")\n",
-    "print(name)"
+    "state = await ctx.get(\"state\")\n",
+    "print(state[\"name\"])"
    ]
   },
   {