Use DeepAgents model resolver in reviewer graph
All checks were successful
build / build (push) Successful in 14s
All checks were successful
build / build (push) Successful in 14s
This commit is contained in:
@@ -84,16 +84,13 @@ def test_reviewer_graph_uses_forwarded_llm_credentials(
|
||||
) -> None:
|
||||
captured: dict = {}
|
||||
|
||||
class FakeChatOpenAI:
|
||||
def __init__(self, **kwargs: object) -> None:
|
||||
captured["model_kwargs"] = kwargs
|
||||
|
||||
def fake_create_deep_agent(**kwargs: object) -> dict[str, object]:
|
||||
def fake_create_a2a_deep_agent(ctx, *, creds, default_temperature=None, **kwargs) -> dict[str, object]:
|
||||
captured["creds"] = creds
|
||||
captured["default_temperature"] = default_temperature
|
||||
captured["graph_kwargs"] = kwargs
|
||||
return kwargs
|
||||
|
||||
monkeypatch.setattr(reviewer_builder, "ChatOpenAI", FakeChatOpenAI)
|
||||
monkeypatch.setattr(reviewer_builder, "create_deep_agent", fake_create_deep_agent)
|
||||
monkeypatch.setattr(reviewer_builder, "create_a2a_deep_agent", fake_create_a2a_deep_agent)
|
||||
|
||||
graph = build_reviewer_graph(
|
||||
ReviewerContext(
|
||||
@@ -111,13 +108,13 @@ def test_reviewer_graph_uses_forwarded_llm_credentials(
|
||||
)
|
||||
|
||||
assert graph is captured["graph_kwargs"]
|
||||
assert captured["model_kwargs"] == {
|
||||
"model": "platform-model",
|
||||
"base_url": "http://litellm.test/v1",
|
||||
"api_key": "scoped-grant-token",
|
||||
"stream_usage": True,
|
||||
"extra_body": {"extra_body": {"thinking": {"type": "disabled"}}},
|
||||
}
|
||||
creds = captured["creds"]
|
||||
assert getattr(creds, "model") == "platform-model"
|
||||
assert getattr(creds, "base_url") == "http://litellm.test/v1"
|
||||
assert getattr(creds, "api_key") == "scoped-grant-token"
|
||||
assert getattr(creds, "temperature_mode") == "omit"
|
||||
assert getattr(creds, "extra_body") == {"extra_body": {"thinking": {"type": "disabled"}}}
|
||||
assert captured["default_temperature"] == 0.0
|
||||
|
||||
|
||||
def test_reviewer_graph_rejects_missing_forwarded_llm_credentials(
|
||||
|
||||
Reference in New Issue
Block a user