mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-26 03:04:13 +00:00
b242c66a3b
* use anthropic transformation for bedrock/invoke * use anthropic transforms for bedrock invoke claude * TestBedrockInvokeClaudeJson * add AmazonAnthropicClaudeStreamDecoder * pass bedrock_invoke_provider to make_call * fix _get_base_bedrock_model * fix get_bedrock_route * fix bedrock routing * fixes for bedrock invoke * test_all_model_configs * fix AWSEventStreamDecoder linting * fix code qa * test_bedrock_get_base_model * test_get_model_info_bedrock_models * test_bedrock_base_model_helper * test_bedrock_route_detection
35 lines
966 B
Python
35 lines
966 B
Python
from typing import List, Optional
|
|
|
|
from litellm.secret_managers.main import get_secret_str
|
|
|
|
from ..base_llm.base_utils import BaseLLMModelInfo
|
|
from ..base_llm.chat.transformation import BaseLLMException
|
|
|
|
|
|
class TopazException(BaseLLMException):
|
|
pass
|
|
|
|
|
|
class TopazModelInfo(BaseLLMModelInfo):
|
|
def get_models(self) -> List[str]:
|
|
return [
|
|
"topaz/Standard V2",
|
|
"topaz/Low Resolution V2",
|
|
"topaz/CGI",
|
|
"topaz/High Resolution V2",
|
|
"topaz/Text Refine",
|
|
]
|
|
|
|
@staticmethod
|
|
def get_api_key(api_key: Optional[str] = None) -> Optional[str]:
|
|
return api_key or get_secret_str("TOPAZ_API_KEY")
|
|
|
|
@staticmethod
|
|
def get_api_base(api_base: Optional[str] = None) -> Optional[str]:
|
|
return (
|
|
api_base or get_secret_str("TOPAZ_API_BASE") or "https://api.topazlabs.com"
|
|
)
|
|
|
|
@staticmethod
|
|
def get_base_model(model: str) -> str:
|
|
return model
|