From 22038b6de0879f0d9804de68df1f474a589c10c7 Mon Sep 17 00:00:00 2001 From: voidborne-d Date: Sat, 4 Apr 2026 10:13:02 +0000 Subject: [PATCH] fix: restore sys.modules after loading streaming_parser stub Address review feedback from greptile-apps: save/restore strix.llm.utils in sys.modules to avoid clobbering the real module when the full test suite runs. --- tests/interface/test_streaming_parser.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/tests/interface/test_streaming_parser.py b/tests/interface/test_streaming_parser.py index e364c663..9ffc3e9e 100644 --- a/tests/interface/test_streaming_parser.py +++ b/tests/interface/test_streaming_parser.py @@ -30,9 +30,16 @@ def _load_streaming_parser(): fake_llm_utils.normalize_tool_format = lambda s: s # type: ignore[attr-defined] sys.modules.setdefault("strix", types.ModuleType("strix")) sys.modules.setdefault("strix.llm", types.ModuleType("strix.llm")) + _original_llm_utils = sys.modules.get("strix.llm.utils") sys.modules["strix.llm.utils"] = fake_llm_utils - spec.loader.exec_module(module) + try: + spec.loader.exec_module(module) + finally: + if _original_llm_utils is None: + sys.modules.pop("strix.llm.utils", None) + else: + sys.modules["strix.llm.utils"] = _original_llm_utils return module