mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-26 03:04:13 +00:00
5bbf906c83
All checks were successful
Read Version from pyproject.toml / read-version (push) Successful in 44s
* feat(base_llm): initial commit for common base config class Addresses code qa critique https://github.com/andrewyng/aisuite/issues/113#issuecomment-2512369132 * feat(base_llm/): add transform request/response abstract methods to base config class * feat(cohere-+-clarifai): refactor integrations to use common base config class * fix: fix linting errors * refactor(anthropic/): move anthropic + vertex anthropic to use base config * test: fix xai test * test: fix tests * fix: fix linting errors * test: comment out WIP test * fix(transformation.py): fix is pdf used check * fix: fix linting error
19 lines
565 B
Python
19 lines
565 B
Python
from litellm.llms.base_llm.transformation import BaseLLMException
|
|
|
|
|
|
class CohereError(BaseLLMException):
|
|
def __init__(self, status_code, message):
|
|
super().__init__(status_code=status_code, message=message)
|
|
|
|
|
|
def validate_environment(*, api_key: str, headers: dict) -> dict:
|
|
headers.update(
|
|
{
|
|
"Request-Source": "unspecified:litellm",
|
|
"accept": "application/json",
|
|
"content-type": "application/json",
|
|
}
|
|
)
|
|
if api_key:
|
|
headers["Authorization"] = f"Bearer {api_key}"
|
|
return headers
|