We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 77ce77c commit b063012Copy full SHA for b063012
1 file changed
tests/llm/test_google_llm.py
@@ -157,6 +157,7 @@ def test_raw_gen_stream_emits_thought_events(monkeypatch):
157
)
158
monkeypatch.setattr(FakeModels, "generate_content_stream", lambda self, *a, **kw: [chunk])
159
out = list(llm._raw_gen_stream(llm, model="gemini", messages=msgs, stream=True))
160
+
161
assert "answer token" in out
162
if len(out) > 1:
163
assert {"type": "thought", "thought": "thinking token"} in out
0 commit comments