mirror of
https://github.com/bytedance/deer-flow.git
synced 2026-10-26 21:26:39 +00:00
513c03e8c0
* feat: add API Route setup wizard preset Signed-off-by: dennyho0917 <dennyho0917@gmail.com> * fix: address API Route wizard review feedback Signed-off-by: dennyho0917 <dennyho0917@gmail.com> --------- Signed-off-by: dennyho0917 <dennyho0917@gmail.com> Co-authored-by: Willem Jiang <willem.jiang@gmail.com>
58 lines
2.3 KiB
Python
58 lines
2.3 KiB
Python
"""API Route setup uses the existing OpenAI-compatible wizard path."""
|
|
|
|
import yaml
|
|
from wizard.providers import LLM_PROVIDERS
|
|
from wizard.steps import llm as llm_step
|
|
from wizard.writer import build_minimal_config
|
|
|
|
|
|
def api_route_provider():
|
|
provider = next((p for p in LLM_PROVIDERS if p.name == "api_route"), None)
|
|
assert provider is not None
|
|
return provider
|
|
|
|
|
|
def test_api_route_uses_versioned_chat_endpoint_and_bare_model_id():
|
|
provider = api_route_provider()
|
|
assert provider.use == "langchain_openai:ChatOpenAI"
|
|
assert provider.package == "langchain-openai"
|
|
assert provider.env_var == "API_ROUTE_API_KEY"
|
|
assert provider.models == ["gpt-6.1-sol"]
|
|
assert provider.default_model == "gpt-6.1-sol"
|
|
assert provider.extra_config["base_url"] == "https://global.api-route.com/v1"
|
|
|
|
|
|
def test_api_route_generated_config_preserves_key_reference_and_model():
|
|
provider = api_route_provider()
|
|
config = yaml.safe_load(
|
|
build_minimal_config(
|
|
provider_use=provider.use,
|
|
model_name=provider.default_model,
|
|
display_name=provider.display_name,
|
|
api_key_field=provider.api_key_field,
|
|
env_var=provider.env_var,
|
|
extra_model_config=provider.extra_config_for(provider.default_model),
|
|
)
|
|
)
|
|
model = config["models"][0]
|
|
assert model["model"] == "gpt-6.1-sol"
|
|
assert model["api_key"] == "$API_ROUTE_API_KEY"
|
|
assert model["base_url"] == "https://global.api-route.com/v1"
|
|
assert model["use"] == "langchain_openai:ChatOpenAI"
|
|
assert model["temperature"] == 0.7
|
|
assert "use_responses_api" not in model
|
|
assert not model.get("supports_vision", False)
|
|
assert not model.get("supports_thinking", False)
|
|
|
|
|
|
def test_api_route_wizard_selects_preset_and_collects_key(monkeypatch):
|
|
provider = api_route_provider()
|
|
monkeypatch.setattr(llm_step, "LLM_PROVIDERS", [provider])
|
|
monkeypatch.setattr(llm_step, "ask_choice", lambda *_args, **_kwargs: 0)
|
|
monkeypatch.setattr(llm_step, "ask_secret", lambda prompt: "test-api-route-key" if prompt == "API_ROUTE_API_KEY" else None)
|
|
result = llm_step.run_llm_step()
|
|
assert result.model_name == "gpt-6.1-sol"
|
|
assert result.provider.extra_config["base_url"] == "https://global.api-route.com/v1"
|
|
assert result.provider.env_var == "API_ROUTE_API_KEY"
|
|
assert result.api_key == "test-api-route-key"
|