scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-06 08:16:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 79

fixing mypy lint errors

Browse source
This commit is contained in:
Eddie Richter 2025-09-24 09:23:20 -06:00
parent 3d62596daa
commit bbfa00c61b
1 changed files with 6 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
litellm/llms/lemonade/chat/transformation.py
View file

@ -26,7 +26,7 @@ class LemonadeChatConfig(OpenAILikeChatConfig):
presence_penalty: Optional[int] = None
stop: Optional[Union[str, list]] = None
temperature: Optional[int] = None
top_p: Optional[float] = None
top_p: Optional[int] = None
top_k: Optional[int] = None
response_format: Optional[dict] = None
tools: Optional[list] = None
@ -42,7 +42,7 @@ class LemonadeChatConfig(OpenAILikeChatConfig):
presence_penalty: Optional[int] = None,
stop: Optional[Union[str, list]] = None,
temperature: Optional[int] = None,
top_p: Optional[float] = None,
top_p: Optional[int] = None,
top_k: Optional[int] = None,
response_format: Optional[dict] = None,
tools: Optional[list] = None,
@ -89,13 +89,16 @@ class LemonadeChatConfig(OpenAILikeChatConfig):
)
# Returning the model if it was found in lemonade. Currently there is no mechanism to report
# if the model supports function calling or the max tokens so we leave those out
# the models max input or output tokens so leaving those as None
return ModelInfoBase(
key=model,
litellm_provider="lemonade",
mode="chat",
input_cost_per_token=0.0,
output_cost_per_token=0.0,
max_input_tokens=None,
max_output_tokens=None,
max_tokens=None,
)
def _get_openai_compatible_provider_info(