scriptoriumadmin/strix
1
0
Fork 0
mirror of https://github.com/usestrix/strix.git synced 2026-09-13 23:11:07 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: restore sys.modules after loading streaming_parser stub

Browse source 
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.
This commit is contained in:
voidborne-d 2026-04-04 10:13:02 +00:00
parent 30cc9ecd47
commit 22038b6de0
1 changed files with 8 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
tests/interface/test_streaming_parser.py
View file

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