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

refactor(utils.py): refactor Logging to it's own class. Cut down utils.py to <10k lines.

Browse source 
Easier debugging

 Reference: https://github.com/BerriAI/litellm/issues/4206
This commit is contained in:
Krrish Dholakia 2024-06-15 10:57:20 -07:00
parent 40efbc6c37
commit 9d7f5d503c
20 changed files with 4517 additions and 2963 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/llms/databricks.py
View file

@ -10,10 +10,10 @@ from typing import Callable, Optional, List, Union, Tuple, Literal
from litellm.utils import (
ModelResponse,
Usage,
map_finish_reason,
CustomStreamWrapper,
EmbeddingResponse,
)
from litellm.litellm_core_utils.core_helpers import map_finish_reason
import litellm
from .prompt_templates.factory import prompt_factory, custom_prompt
from litellm.llms.custom_httpx.http_handler import AsyncHTTPHandler, HTTPHandler
@ -289,7 +289,7 @@ class DatabricksChatCompletion(BaseLLM):
response: Union[requests.Response, httpx.Response],
model_response: ModelResponse,
stream: bool,
logging_obj: litellm.utils.Logging,
logging_obj: litellm.litellm_core_utils.litellm_logging.Logging,
optional_params: dict,
api_key: str,
data: Union[dict, str],