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: fix linting errors

Browse source
This commit is contained in:
Krrish Dholakia 2023-12-06 11:46:09 -08:00
parent de58dcc016
commit 102de97960
2 changed files with 8 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/llms/sagemaker.py
View file

@ -5,7 +5,7 @@ import requests
import time
from typing import Callable, Optional
import litellm
from litellm.utils import ModelResponse, get_secret, Usage
from litellm.utils import ModelResponse, EmbeddingResponse, get_secret, Usage
import sys
from copy import deepcopy
import httpx
@ -198,7 +198,7 @@ def completion(
def embedding(model: str,
input: list,
model_response: ModelResponse,
model_response: EmbeddingResponse,
print_verbose: Callable,
encoding,
logging_obj,
@ -262,7 +262,7 @@ def embedding(model: str,
ContentType="application/json",
Body={data},
CustomAttributes="accept_eula=true",
)"""
)""" # type: ignore
logging_obj.pre_call(
input=input,
api_key="",