pvanand commited on
Commit
9df08b9
1 Parent(s): f029274

output format

Browse files
Files changed (1) hide show
  1. main.py +3 -1
main.py CHANGED
@@ -600,9 +600,11 @@ def followup_agent(query: FollowupQueryModel, background_tasks: BackgroundTasks,
600
  search_query = tool["input"]
601
  search_response = search_assistant_api(search_query, tool["name"], model=query.model_id)
602
 
 
603
  for content in search_response():
604
  yield content
605
-
 
606
  background_tasks.add_task(update_db, query.user_id, query.conversation_id, query.query, full_response)
607
  logger.info(f"Completed followup agent response for query: {query.query}, send result: {result}")
608
 
 
600
  search_query = tool["input"]
601
  search_response = search_assistant_api(search_query, tool["name"], model=query.model_id)
602
 
603
+ yield "<report>"
604
  for content in search_response():
605
  yield content
606
+ yield "</report>"
607
+
608
  background_tasks.add_task(update_db, query.user_id, query.conversation_id, query.query, full_response)
609
  logger.info(f"Completed followup agent response for query: {query.query}, send result: {result}")
610