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 |
|