Spaces:
Running
Running
output format
Browse files
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 |
|