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

refactor(huggingface,-anthropic,-replicate,-sagemaker): making huggingface, anthropic, replicate, sagemaker compatible openai v1 sdk

Browse source
This commit is contained in:
Krrish Dholakia 2023-11-11 17:38:15 -08:00
parent 547598a134
commit 4f42beb9d9
4 changed files with 11 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/llms/sagemaker.py
View file

@ -8,11 +8,14 @@ import litellm
from litellm.utils import ModelResponse, get_secret
import sys
from copy import deepcopy
import httpx
class SagemakerError(Exception):
def __init__(self, status_code, message):
self.status_code = status_code
self.message = message
self.request = httpx.Request(method="POST", url="https://us-west-2.console.aws.amazon.com/sagemaker")
self.response = httpx.Response(status_code=status_code, request=self.request)
super().__init__(
self.message
) # Call the base class constructor with the parameters it needs