Coverage for packages/server/src/langgate/server/api/api_v1/endpoints/models_ep.py: 79%

1from fastapi import APIRouter 

2 

3from langgate.core.logging import get_logger 

4from langgate.registry.models import LLMInfo 

5from langgate.server.api.services.registry_api import ModelRegistryAPI 

6 

7logger = get_logger(__name__) 

8router = APIRouter() 

9registry_service = ModelRegistryAPI() 

10 

11 

12@router.get("", response_model=list[LLMInfo]) 

13async def list_models(): 

14 return await registry_service.list_models() 

15 

16 

17@router.get("/{model_id:path}", response_model=LLMInfo) 

18async def get_model_info( 

19 *, 

20 model_id: str, 

21): 

22 model = model_id 

23 return await registry_service.get_model_info(model)