mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-25 18:54:30 +00:00
LiteLLM Common Base LLM Config (pt.4): Move Ollama to Base LLM Config (#7157)
* 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
This commit is contained in:
parent
3b5485a14e
commit
40a22eb4c6
11 changed files with 322 additions and 234 deletions
12
litellm/llms/ollama/common_utils.py
Normal file
12
litellm/llms/ollama/common_utils.py
Normal file
|
@ -0,0 +1,12 @@
|
|||
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)
|
Loading…
Add table
Add a link
Reference in a new issue