[linsheng-li]

Description

Stopping a scheduler that has never initialized RabbitMQ logs AttributeError: 'OptimizedScheduler' object has no attribute '_io_loop_thread' during API shutdown. This includes local-queue deployments and initialization skipped in CI. No restart is required to trigger it.

The constructor initializes _rabbitmq_io_loop_thread, but both initialize_rabbitmq() and rabbitmq_close() use _io_loop_thread. Initialize the same field in the constructor. The production change is one line; connection cleanup, thread joins, shutdown ordering, and exception handling are unchanged. No new dependencies or configuration are required.

Related issue: N/A; the reproduction and regression tests are included here.

Type of change

• Bug fix (non-breaking)

How Has This Been Tested?

• Unit tests: pytest tests/mem_scheduler/test_rabbitmq_shutdown.py -q — 5 passed. Without the fix, 3 failed, 2 passed. Covers uninitialized/repeated close, CI initialization skip, connection cleanup with a real IO thread, native optimized-scheduler lifecycle cleanup, and propagation of unexpected join errors.
• Test steps: isolated API shutdown with RabbitMQ uninitialized, followed by restart and a second shutdown — both exits were clean; graph/vector/business snapshots were unchanged. No generation or embedding requests were made.
• make format — all Ruff checks passed; 631 files unchanged.
• Pre-commit checks on both changed files — passed.

Additional directory run: pytest tests/mem_scheduler/ -q — 70 passed, 1 skipped, 1 failed. The existing test_dynamic_cache_layers_access requires PyTorch, which is absent in the test image; the same failure was reproduced against the unmodified source. Pytest also warns that the image lacks the asyncio plugin. The full repository suite was not run.

Checklist

• Self-reviewed the one-line fix and focused tests.
• Added tests demonstrating the failure and preserving existing cleanup behavior.
• No documentation/API/schema changes are needed for this internal field-name correction.

[Memtensor-AI]

🤖 Open Code Review

Target: PR #2532
Task: c326249ca4d67364
Base: dev-v2.0.36
Head: fix/rabbitmq-shutdown-thread-init

✅ OpenCodeReview: Review complete: 0 finding(s) across 1 selected item(s).

Generated by cloud-assistant via Open Code Review.

[Memtensor-AI]

❌ Automated Test Results: FAILED

Auto-fix retry 1/2 triggered.

Failed tests:

• test_close_without_rabbitmq_initialization_is_safe_and_repeatable
• test_early_initialization_skip_still_closes
• test_real_io_thread_and_connection_cleanup_still_execute
• test_actual_optimized_scheduler_lifecycle_joins_threads_and_shuts_executor
• test_unexpected_join_error_is_not_hidden

Error details

Tests failed. Failed cases: test_close_without_rabbitmq_initialization_is_safe_and_repeatable, test_early_initialization_skip_still_closes, test_real_io_thread_and_connection_cleanup_still_execute, test_actual_optimized_scheduler_lifecycle_joins_threads_and_shuts_executor, test_unexpected_join_error_is_not_hidden

Branch: fix/rabbitmq-shutdown-thread-init