Coverage for packages/transform/src/langgate/transform/protocol.py: 100%

1"""Protocol definitions for parameter transformers.""" 

2 

3from typing import Any, Protocol 

4 

5from langgate.core.logging import get_logger 

6 

7logger = get_logger(__name__) 

8 

9 

10class TransformerClientProtocol(Protocol): 

11 """Protocol for transformer clients.""" 

12 

13 async def get_params( 

14 self, model_id: str, input_params: dict[str, Any] 

15 ) -> dict[str, Any]: 

16 """Get transformed parameters for the specified model. 

17 

18 Args: 

19 model_id: The ID of the model to get transformed parameters for 

20 input_params: The parameters to transform 

21 

22 Returns: 

23 The transformed parameters 

24 """ 

25 ...