aboutsummaryrefslogtreecommitdiffstats
path: root/test_llamachat.py
diff options
context:
space:
mode:
authorDanilo M. <danix@danix.xyz>2026-08-25 18:34:04 +0200
committerDanilo M. <danix@danix.xyz>2026-08-25 18:34:04 +0200
commit8776e6fd87207d0bd76f1567851f554b7eb3a70e (patch)
treedb7500ba1ab8f519004635cbb625b819fc116513 /test_llamachat.py
parent3bbd3b6e95ee06b8d4171196ee4e4cfb6f43ba9e (diff)
downloadllamachat-8776e6fd87207d0bd76f1567851f554b7eb3a70e.tar.gz
llamachat-8776e6fd87207d0bd76f1567851f554b7eb3a70e.zip
feat: stream skill_loaded events and show loaded-skill chips
Diffstat (limited to 'test_llamachat.py')
-rwxr-xr-xtest_llamachat.py4
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