raul-padua commited on
Commit
a311879
Β·
1 Parent(s): 2c6d573

Answer message added

Browse files

Integrated new message object.

Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -66,13 +66,18 @@ async def main(message):
66
  logger.info(f"Received message: {message}")
67
  response = await cl.make_async(query_engine.query)(message)
68
  response_message = cl.Message(content="")
69
-
 
70
  for token in response.response_gen:
71
  await response_message.stream_token(token=token)
72
 
73
  if response.response_txt:
74
  response_message.content = response.response_txt
75
 
 
 
 
 
76
  await response_message.send()
77
  logger.info(f"Response sent: {response.response_txt}")
78
 
 
66
  logger.info(f"Received message: {message}")
67
  response = await cl.make_async(query_engine.query)(message)
68
  response_message = cl.Message(content="")
69
+
70
+ # Logic to prepare answer and source_elements
71
  for token in response.response_gen:
72
  await response_message.stream_token(token=token)
73
 
74
  if response.response_txt:
75
  response_message.content = response.response_txt
76
 
77
+ # Integrated new message object
78
+ if answer: # conditional to when is not None
79
+ await cl.Message(content=answer, elements=source_elements).send()
80
+
81
  await response_message.send()
82
  logger.info(f"Response sent: {response.response_txt}")
83