def get_llm_provider(
model: str,
custom_llm_provider: Optional[str] = None,
api_base: Optional[str] = None,
api_key: Optional[str] = None,
):
try:
# ... more code ...
if (
model.split("/", 1)[0] in litellm.provider_list
and model.split("/", 1)[0] not in litellm.model_list
and len(model.split("/"))
> 1
):
custom_llm_provider = model.split("/", 1)[0]
model = model.split("/", 1)[1]
# ... more code ...
return model, custom_llm_provider, dynamic_api_key, api_base