From a9395698e5d9ceac5c136243b199a660ecf2cea7 Mon Sep 17 00:00:00 2001 From: Roger Yang <80478925+RogerHYang@users.noreply.github.com> Date: Wed, 3 Jul 2024 19:22:51 -0700 Subject: [PATCH] chore: change task function to be async in notebook (#3845) --- tutorials/experiments/llama-index/guideline_eval.ipynb | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tutorials/experiments/llama-index/guideline_eval.ipynb b/tutorials/experiments/llama-index/guideline_eval.ipynb index 0997c89ce4..408e7f34f9 100644 --- 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))" ] },