phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

linting type fixes

Browse source
This commit is contained in:
ishaan-jaff 2023-08-18 11:36:06 -07:00
parent fbcb6d1c20
commit 5e7d22512d
6 changed files with 13 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/llms/huggingface_restapi.py
View file

@ -6,6 +6,7 @@ from litellm import logging
import time
from typing import Callable
from litellm.utils import ModelResponse
from typing import Optional
class HuggingfaceError(Exception):
@ -45,7 +46,7 @@ class HuggingfaceRestAPILLM:
logger_fn=None,
): # logic for parsing in - calling - parsing out model completion calls
if custom_api_base:
completion_url = custom_api_base
completion_url: Optional[str] = custom_api_base
elif "HF_API_BASE" in os.environ:
completion_url = os.getenv("HF_API_BASE")
else:
@ -136,5 +137,5 @@ class HuggingfaceRestAPILLM:
return model_response
pass
def embedding(): # logic for parsing in - calling - parsing out model embedding calls
def embedding(self): # logic for parsing in - calling - parsing out model embedding calls
pass