File tree 1 file changed +5
-1
lines changed
1 file changed +5
-1
lines changed Original file line number Diff line number Diff line change 253
253
" \n " ,
254
254
" system_message = \" You are a helpful recipe assistant. You are to extract the generic ingredients from each of the recipes provided.\"\n " ,
255
255
" \n " ,
256
+ " def create_user_message(row):\n " ,
257
+ " return f\"\"\" Title: {row['title']}\\ n\\ nIngredients: {row['ingredients']}\\ n\\ nGeneric ingredients: \"\"\"\n " ,
258
+ " \n " ,
256
259
" def prepare_example_conversation(row):\n " ,
257
260
" messages = []\n " ,
258
261
" messages.append({\" role\" : \" system\" , \" content\" : system_message})\n " ,
259
262
" \n " ,
260
- " user_message = f \"\"\" Title: { row['title']} \\ n \\ nIngredients: {row['ingredients']} \\ n \\ nGeneric ingredients: \"\"\" \n" ,
263
+ " user_message = create_user_message( row) \n " ,
261
264
" messages.append({\" role\" : \" user\" , \" content\" : user_message})\n " ,
262
265
" \n " ,
263
266
" messages.append({\" role\" : \" assistant\" , \" content\" : row[\" NER\" ]})\n " ,
642
645
}
643
646
],
644
647
"source" : [
648
+ " test_df = recipe_df.loc[201:300]\n " ,
645
649
" test_row = test_df.iloc[0]\n " ,
646
650
" test_messages = []\n " ,
647
651
" test_messages.append({\" role\" : \" system\" , \" content\" : system_message})\n " ,
You can’t perform that action at this time.
0 commit comments