fix undefined variables in fine tune example (#660)

`create_user_message` and `test_df` are not defined.
This commit is contained in:
Viet Hoang Tran Duong 2023-08-25 13:03:06 -07:00 committed by GitHub
parent 35b7123faf
commit 1ae3bf631b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -253,11 +253,14 @@
"\n", "\n",
"system_message = \"You are a helpful recipe assistant. You are to extract the generic ingredients from each of the recipes provided.\"\n", "system_message = \"You are a helpful recipe assistant. You are to extract the generic ingredients from each of the recipes provided.\"\n",
"\n", "\n",
"def create_user_message(row):\n",
" return f\"\"\"Title: {row['title']}\\n\\nIngredients: {row['ingredients']}\\n\\nGeneric ingredients: \"\"\"\n",
"\n",
"def prepare_example_conversation(row):\n", "def prepare_example_conversation(row):\n",
" messages = []\n", " messages = []\n",
" messages.append({\"role\": \"system\", \"content\": system_message})\n", " messages.append({\"role\": \"system\", \"content\": system_message})\n",
"\n", "\n",
" user_message = f\"\"\"Title: {row['title']}\\n\\nIngredients: {row['ingredients']}\\n\\nGeneric ingredients: \"\"\"\n", " user_message = create_user_message(row)\n",
" messages.append({\"role\": \"user\", \"content\": user_message})\n", " messages.append({\"role\": \"user\", \"content\": user_message})\n",
"\n", "\n",
" messages.append({\"role\": \"assistant\", \"content\": row[\"NER\"]})\n", " messages.append({\"role\": \"assistant\", \"content\": row[\"NER\"]})\n",
@ -642,6 +645,7 @@
} }
], ],
"source": [ "source": [
"test_df = recipe_df.loc[201:300]\n",
"test_row = test_df.iloc[0]\n", "test_row = test_df.iloc[0]\n",
"test_messages = []\n", "test_messages = []\n",
"test_messages.append({\"role\": \"system\", \"content\": system_message})\n", "test_messages.append({\"role\": \"system\", \"content\": system_message})\n",