LLM/DocsGPT
1
0
Fork 0
mirror of https://github.com/arc53/DocsGPT.git synced 2026-03-07 14:24:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'main' into thinking-stream

Browse Source
This commit is contained in:
Alex
2026-02-09 14:22:46 +00:00
committed by GitHub
parent 6ab5eacfa6 fea94379d7
commit e215a73ed7
1 changed files with 4 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tests/llm/test_google_llm.py
View File

@@ -141,9 +141,7 @@ def test_raw_gen_stream_does_not_set_thinking_config_by_default(monkeypatch):
assert captured["config"].thinking_config is None
def test_raw_gen_stream_accepts_include_thoughts_without_setting_thinking_config(
monkeypatch,
):
def test_raw_gen_stream_sets_thinking_config_when_explicitly_requested(monkeypatch):
captured = {}
def fake_stream(self, *args, **kwargs):
@@ -154,7 +152,7 @@ def test_raw_gen_stream_accepts_include_thoughts_without_setting_thinking_config
llm = GoogleLLM(api_key="key")
msgs = [{"role": "user", "content": "hello"}]
out = list(
list(
llm._raw_gen_stream(
llm,
model="gemini",
@@ -164,8 +162,8 @@ def test_raw_gen_stream_accepts_include_thoughts_without_setting_thinking_config
)
)
assert out == ["a"]
assert captured["config"].thinking_config is None
assert captured["config"].thinking_config is not None
assert captured["config"].thinking_config.include_thoughts is True
def test_raw_gen_stream_emits_thought_events(monkeypatch):