Coverage for src / lilbee / providers / local_servers / config_urls.py: 100%

1"""Resolve each local server's configured base URL, with spec-default fallback.""" 

2 

3from __future__ import annotations 

4 

5from collections.abc import Callable 

6 

7from lilbee.core.config.model import cfg 

8from lilbee.providers.local_servers.lm_studio import LM_STUDIO 

9from lilbee.providers.local_servers.ollama import OLLAMA 

10from lilbee.providers.local_servers.registry import LOCAL_SERVERS, local_server_for_key 

11from lilbee.providers.local_servers.spec import LocalServerSpec 

12 

13_URL_ACCESSOR_BY_KEY: dict[str, Callable[[], str]] = { 

14 OLLAMA.key: lambda: cfg.ollama_base_url, 

15 LM_STUDIO.key: lambda: cfg.lm_studio_base_url, 

16} 

17 

18 

19def _configured_url(spec: LocalServerSpec) -> str: 

20 return _URL_ACCESSOR_BY_KEY[spec.key]().strip() or spec.default_base_url 

21 

22 

23def base_url_for(server_key: str) -> str: 

24 """Configured URL for a server key, falling back to its spec default.""" 

25 spec = local_server_for_key(server_key) 

26 if spec is None: 

27 raise KeyError(f"Unknown local server: {server_key!r}") 

28 return _configured_url(spec) 

29 

30 

31def configured_local_servers() -> list[tuple[LocalServerSpec, str]]: 

32 """Return (spec, resolved-url) for every known local server.""" 

33 return [(spec, _configured_url(spec)) for spec in LOCAL_SERVERS]