small nit on step decorator in notebook (#17486)

run-llama · Jan 11, 2025 · 2d399b9 · 2d399b9
1 parent 935b14e
commit 2d399b9
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/docs/docs/examples/workflow/multi_step_query_engine.ipynb b/docs/docs/examples/workflow/multi_step_query_engine.ipynb
@@ -189,7 +189,7 @@
     "\n",
     "        return \"none\" in query_bundle.query_str.lower()\n",
     "\n",
-    "    @step(pass_context=True)\n",
+    "    @step\n",
     "    async def query_multistep(\n",
     "        self, ctx: Context, ev: StartEvent\n",
     "    ) -> QueryMultiStepEvent:\n",
@@ -268,7 +268,7 @@
     "            final_response_metadata=final_response_metadata,\n",
     "        )\n",
     "\n",
-    "    @step(pass_context=True)\n",
+    "    @step\n",
     "    async def synthesize(\n",
     "        self, ctx: Context, ev: QueryMultiStepEvent\n",
     "    ) -> StopEvent:\n",
@@ -528,9 +528,9 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "llamaindex",
+   "display_name": "llama-index-caVs7DDe-py3.10",
    "language": "python",
-   "name": "llamaindex"
+   "name": "python3"
   },
   "language_info": {
    "codemirror_mode": {