Update main.py

main.py

@@ -590,8 +590,6 @@ def followup_agent(query: FollowupQueryModel, background_tasks: BackgroundTasks,
         
         yield "<json>" + json.dumps(result)
         
-        # Add the assistant's response to the conversation history
-        conversations[query.conversation_id].append({"role": "assistant", "content": full_response})
         
         # Process tool if present
         if tools and len(tools) > 0:
@@ -605,7 +603,9 @@ def followup_agent(query: FollowupQueryModel, background_tasks: BackgroundTasks,
                     yield content
                     full_response += content
                 yield "</report>"
-                
+        
+        # Add the assistant's response to the conversation history
+        conversations[query.conversation_id].append({"role": "assistant", "content": full_response})
         background_tasks.add_task(update_db, query.user_id, query.conversation_id, query.query, full_response)
         logger.info(f"Completed followup agent response for query: {query.query}, send result: {result}")