diff options
Diffstat (limited to 'test_llamachat.py')
| -rwxr-xr-x | test_llamachat.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/test_llamachat.py b/test_llamachat.py index 85e69a6..d96b599 100755 --- a/test_llamachat.py +++ b/test_llamachat.py @@ -2157,7 +2157,7 @@ def test_usage_accumulation(): def wire_name(self, model): return model - def stream_chat(self, wire, messages, search_cfg=None): + def stream_chat(self, wire, messages, search_cfg=None, skills_cfg=None): # One usage block per round: a tool round, then the final answer. yield ("usage", '{"prompt_tokens": 100, "completion_tokens": 5,' ' "total_tokens": 105}') @@ -2177,7 +2177,7 @@ def test_usage_accumulation(): # A turn with no usage chunk at all (no include_usage support) yields # an empty array, which the window stores as NULL. class SilentClient(FakeClient): - def stream_chat(self, wire, messages, search_cfg=None): + def stream_chat(self, wire, messages, search_cfg=None, skills_cfg=None): yield ("content", "hi") return |