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

Add pyright to ci/cd + Fix remaining type-checking errors (#6082)

Browse source 
* fix: fix type-checking errors

* fix: fix additional type-checking errors

* fix: additional type-checking error fixes

* fix: fix additional type-checking errors

* fix: additional type-check fixes

* fix: fix all type-checking errors + add pyright to ci/cd

* fix: fix incorrect import

* ci(config.yml): use mypy on ci/cd

* fix: fix type-checking errors in utils.py

* fix: fix all type-checking errors on main.py

* fix: fix mypy linting errors

* fix(anthropic/cost_calculator.py): fix linting errors

* fix: fix mypy linting errors

* fix: fix linting errors
This commit is contained in:
Krish Dholakia 2024-10-05 17:04:00 -04:00 • committed by GitHub
parent f7ce1173f3
commit fac3b2ee42
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
65 changed files with 619 additions and 522 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/llms/huggingface_restapi.py
View file

@ -554,7 +554,7 @@ class Huggingface(BaseLLM):
else:
prompt = prompt_factory(model=model, messages=messages)
data = {
"inputs": prompt,
"inputs": prompt, # type: ignore
"parameters": optional_params,
"stream": ( # type: ignore
True
@ -589,7 +589,7 @@ class Huggingface(BaseLLM):
inference_params.pop("details")
inference_params.pop("return_full_text")
data = {
"inputs": prompt,
"inputs": prompt, # type: ignore
}
if task == "text-generation-inference":
data["parameters"] = inference_params