phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 02:34:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(common_utils.py): fix linting error
Some checks failed
Read Version from pyproject.toml / read-version (push) Successful in 19s
Details
Helm unit test / unit-test (push) Successful in 20s
Details
Publish Prisma Migrations / publish-migrations (push) Failing after 1m15s
Details

Browse source
This commit is contained in:
Krrish Dholakia 2025-03-27 23:31:58 -07:00
parent 3226477464
commit 34a1a48af1
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
litellm/llms/anthropic/common_utils.py
View file

@ -24,7 +24,7 @@ class AnthropicError(BaseLLMException):
class AnthropicModelInfo(BaseLLMModelInfo):
@staticmethod
def get_api_base(api_base: Optional[str] = None) -> str | None:
def get_api_base(api_base: Optional[str] = None) -> Optional[str]:
return (
api_base
or get_secret_str("ANTHROPIC_API_BASE")
@ -32,12 +32,12 @@ class AnthropicModelInfo(BaseLLMModelInfo):
)
@staticmethod
def get_api_key(api_key: str | None = None) -> str | None:
def get_api_key(api_key: Optional[str] = None) -> Optional[str]:
return api_key or get_secret_str("ANTHROPIC_API_KEY")
@staticmethod
def get_base_model(model: str) -> str | None:
return model.replace("anthropic/", "")
def get_base_model(model: Optional[str] = None) -> Optional[str]:
return model.replace("anthropic/", "") if model else None
def get_models(
self, api_key: Optional[str] = None, api_base: Optional[str] = None