mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-26 11:14:04 +00:00
e9fbefca5d
* refactor(ollama/): refactor ollama `/api/generate` to use base llm config Addresses https://github.com/andrewyng/aisuite/issues/113#issuecomment-2512369132 * test: skip unresponsive test * test(test_secret_manager.py): mark flaky test * test: fix google sm test * fix: fix init.py
12 lines
336 B
Python
12 lines
336 B
Python
from typing import Union
|
|
|
|
import httpx
|
|
|
|
from litellm.llms.base_llm.transformation import BaseLLMException
|
|
|
|
|
|
class OllamaError(BaseLLMException):
|
|
def __init__(
|
|
self, status_code: int, message: str, headers: Union[dict, httpx.Headers]
|
|
):
|
|
super().__init__(status_code=status_code, message=message, headers=headers)
|