chore: change task function to be async in notebook (#3845)

Arize-ai · Jul 4, 2024 · a939569 · a939569
1 parent 8aed7ba
commit a939569
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tutorials/experiments/llama-index/guideline_eval.ipynb b/tutorials/experiments/llama-index/guideline_eval.ipynb
@@ -200,8 +200,8 @@
     "llm = OpenAI(model=\"gpt-3.5-turbo\")\n",
     "\n",
     "\n",
-    "def task(input):\n",
-    "    return llm.complete(input[\"document\"] + \"\\n\\n\" + input[\"messages\"][-1][\"content\"]).text"
+    "async def task(input):\n",
+    "    return (await llm.acomplete(input[\"document\"] + \"\\n\\n\" + input[\"messages\"][-1][\"content\"])).text"
    ]
   },
   {
@@ -218,7 +218,7 @@
    "outputs": [],
    "source": [
     "example = dataset[0]\n",
-    "task_output = task(example.input)\n",
+    "task_output = await task(example.input)\n",
     "print(shorten(json.dumps(task_output), width=80))"
    ]
   },