YchKhan commited on
Commit
9aa7c1c
1 Parent(s): 4c80563

Update excel_chat.py

Browse files
Files changed (1) hide show
  1. excel_chat.py +1 -1
excel_chat.py CHANGED
@@ -105,7 +105,7 @@ def chat_with_mistral(source_cols, dest_col, prompt, excel_file, url, search_col
105
  concatenated_content = "\n\n".join(f"{column_name}: {str(row[column_name])}" for column_name in source_cols)
106
  if not concatenated_content == "\n\n".join(f"{column_name}: nan" for column_name in source_cols):
107
  llm_answer = ask_llm(prompt[0], concatenated_content, client, user)
108
- print(f"{index+1}/{len(df)}\nQUERY:\n{prompt[0]}\nCONTENT:\n{concatenated_content[:200]}...\n\nANSWER:\n{llm_answer}")
109
  df.at[index, dest_col] = llm_answer
110
  # progress((index+1)/len(df),desc=f'Request {index+1}/{len(df)}')
111
 
 
105
  concatenated_content = "\n\n".join(f"{column_name}: {str(row[column_name])}" for column_name in source_cols)
106
  if not concatenated_content == "\n\n".join(f"{column_name}: nan" for column_name in source_cols):
107
  llm_answer = ask_llm(prompt[0], concatenated_content, client, user)
108
+ print(f"{index+1}/{len(filtred_df)}\nQUERY:\n{prompt[0]}\nCONTENT:\n{concatenated_content[:200]}...\n\nANSWER:\n{llm_answer}")
109
  df.at[index, dest_col] = llm_answer
110
  # progress((index+1)/len(df),desc=f'Request {index+1}/{len(df)}')
111