litellm-mirror/litellm/llms/openrouter.py

from typing import List, Dict
import types

class OpenrouterConfig():
    """
    Reference: https://openrouter.ai/docs#format

    """
    # OpenRouter-only parameters
    extra_body: Dict[str, List[str]] = {
        'transforms': [] # default transforms to []
    }


    def __init__(self, 
                 transforms: List[str] = [],
                 models: List[str] = [],
                 route: str = '',
                ) -> None:
        locals_ = locals()
        for key, value in locals_.items():
            if key != 'self' and value is not None:
                setattr(self.__class__, key, value)

    @classmethod
    def get_config(cls):
        return {k: v for k, v in cls.__dict__.items() 
                if not k.startswith('__') 
                and not isinstance(v, (types.FunctionType, types.BuiltinFunctionType, classmethod, staticmethod)) 
                and v is not None}