h2ogpt-chatbot2

Runtime error

pseudotensor commited on Feb 25, 2024

Commit

2927206

1 Parent(s): 2ce2260

Fix

Files changed (1) hide show

src/gen.py CHANGED Viewed

@@ -2696,7 +2696,7 @@ def evaluate(
  res = gr_client.predict(str(dict(client_kwargs)), api_name=api_name)
  res_dict = ast.literal_eval(res)
  text = res_dict['response']
- sources = res_dict['sources']
  response = prompter.get_response(prompt + text, prompt=prompt,
  sanitize_bot_response=sanitize_bot_response)
  yield dict(response=response, sources=sources, save_dict=dict())
@@ -2715,7 +2715,7 @@ def evaluate(
  res = job.communicator.job.outputs[-1]
  res_dict = ast.literal_eval(res)
  text = res_dict['response']
- sources = res_dict['sources']
  if gr_prompt_type == 'plain':
  # then gradio server passes back full prompt + text
  prompt_and_text = text
@@ -2736,7 +2736,7 @@ def evaluate(
  res = res_all[-1]
  res_dict = ast.literal_eval(res)
  text = res_dict['response']
- sources = res_dict['sources']
  else:
  # go with old text if last call didn't work
  e = job.future._exception

  res = gr_client.predict(str(dict(client_kwargs)), api_name=api_name)
  res_dict = ast.literal_eval(res)
  text = res_dict['response']
+ sources = res_dict.get('sources', '')
  response = prompter.get_response(prompt + text, prompt=prompt,
  sanitize_bot_response=sanitize_bot_response)
  yield dict(response=response, sources=sources, save_dict=dict())
  res = job.communicator.job.outputs[-1]
  res_dict = ast.literal_eval(res)
  text = res_dict['response']
+ sources = res_dict.get('sources', '')
  if gr_prompt_type == 'plain':
  # then gradio server passes back full prompt + text
  prompt_and_text = text
  res = res_all[-1]
  res_dict = ast.literal_eval(res)
  text = res_dict['response']
+ sources = res_dict.get('sources', '')
  else:
  # go with old text if last call didn't work
  e = job.future._exception