chore: Gracefully exit logfire trace

Author: andrewginns

agents_mcp_usage/multi_mcp/multi_mcp_use/adk_mcp.py

@@ -1,6 +1,7 @@
 import asyncio
 import contextlib
 import os
+import time
 
 import logfire
 from dotenv import load_dotenv
@@ -123,6 +124,13 @@ async def main(query: str = "Hi!", request_limit: int = 5) -> None:
     await exit_stack.aclose()
     print("Cleanup complete.")
 
+    # Give Logfire time to complete any pending exports
+    print("Shutting down Logfire...")
+    logfire.shutdown()
+    # Small delay to ensure export completes
+    time.sleep(0.5)
+    print("Logfire shutdown complete.")
+
 
 if __name__ == "__main__":
     query = f"Add the current time and fix the mermaid diagram syntax using the validator: {invalid_mermaid_diagram_easy}. Return only the fixed mermaid diagram between backticks."