chore: fix cli UT

PiperOrigin-RevId: 762734348
This commit is contained in:
Xiang (Sean) Zhou 2025-05-24 00:09:57 -07:00 committed by Copybara-Service
parent a134ba6f7a
commit a380d70ac7

View File

@ -80,6 +80,9 @@ def _patch_types_and_runner(monkeypatch: pytest.MonkeyPatch) -> None:
response = _Content("assistant", [_Part(f"echo:{text}")])
yield types.SimpleNamespace(author="assistant", content=response)
async def close(self, *a: Any, **k: Any) -> None:
...
monkeypatch.setattr(cli, "Runner", _FakeRunner)