add vertex_credentials in router param

This commit is contained in:
Ishaan Jaff 2024-08-02 16:58:17 -07:00
parent 7fa25c443a
commit cff7050147

View file

@ -143,6 +143,7 @@ class GenericLiteLLMParams(BaseModel):
## VERTEX AI ##
vertex_project: Optional[str] = None
vertex_location: Optional[str] = None
vertex_credentials: Optional[str] = None
## AWS BEDROCK / SAGEMAKER ##
aws_access_key_id: Optional[str] = None
aws_secret_access_key: Optional[str] = None
@ -486,6 +487,11 @@ class AssistantsTypedDict(TypedDict):
litellm_params: LiteLLMParamsTypedDict
class FineTuningConfig(BaseModel):
custom_llm_provider: Literal["azure", "openai"]
class CustomRoutingStrategyBase:
async def async_get_available_deployment(
self,