Skip to content

Commit b063012

Browse files
committed
final pytest check
1 parent 77ce77c commit b063012

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

tests/llm/test_google_llm.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -157,6 +157,7 @@ def test_raw_gen_stream_emits_thought_events(monkeypatch):
157157
)
158158
monkeypatch.setattr(FakeModels, "generate_content_stream", lambda self, *a, **kw: [chunk])
159159
out = list(llm._raw_gen_stream(llm, model="gemini", messages=msgs, stream=True))
160+
160161
assert "answer token" in out
161162
if len(out) > 1:
162163
assert {"type": "thought", "thought": "thinking token"} in out

0 commit comments

Comments
 (0)