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

refactor: replace dbrx with 'openai_like'

Browse source
This commit is contained in:
Krrish Dholakia 2024-12-02 20:57:46 -08:00
parent f7772f2147
commit 5acf221484
3 changed files with 7 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/llms/sagemaker/sagemaker.py
View file

@ -273,9 +273,9 @@ class SagemakerLLM(BaseAWSLLM):
model_id = optional_params.get("model_id", None)
if use_messages_api is True:
from litellm.llms.databricks.chat.handler import DatabricksChatCompletion
from litellm.llms.openai_like.chat.handler import OpenAILikeChatHandler
openai_like_chat_completions = DatabricksChatCompletion()
openai_like_chat_completions = OpenAILikeChatHandler()
inference_params["stream"] = True if stream is True else False
_data: Dict[str, Any] = {
"model": model,