mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-25 10:44:24 +00:00
f770dd0c95
* test(test_utils.py): initial test for valid models Addresses https://github.com/BerriAI/litellm/issues/7525 * fix: test * feat(fireworks_ai/transformation.py): support retrieving valid models from fireworks ai endpoint * refactor(fireworks_ai/): support checking model info on `/v1/models` route * docs(set_keys.md): update docs to clarify check llm provider api usage * fix(watsonx/common_utils.py): support 'WATSONX_ZENAPIKEY' for iam auth * fix(watsonx): read in watsonx token from env var * fix: fix linting errors * fix(utils.py): fix provider config check * style: cleanup unused imports
18 lines
407 B
Python
18 lines
407 B
Python
from abc import ABC, abstractmethod
|
|
from typing import List, Optional
|
|
|
|
from litellm.types.utils import ModelInfoBase
|
|
|
|
|
|
class BaseLLMModelInfo(ABC):
|
|
@abstractmethod
|
|
def get_model_info(
|
|
self,
|
|
model: str,
|
|
existing_model_info: Optional[ModelInfoBase] = None,
|
|
) -> Optional[ModelInfoBase]:
|
|
pass
|
|
|
|
@abstractmethod
|
|
def get_models(self) -> List[str]:
|
|
pass
|