phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 19:24:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix request_timeout_extra arg

Browse source
This commit is contained in:
ishaan-jaff 2023-08-14 19:57:11 -07:00
parent bba65a609b
commit 33fbf3d701
3 changed files with 10 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/main.py
View file

@ -51,7 +51,7 @@ def completion(
*, return_async=False, api_key=None, force_timeout=600, logger_fn=None, verbose=False, azure=False, custom_llm_provider=None, custom_api_base=None, *, return_async=False, api_key=None, force_timeout=600, logger_fn=None, verbose=False, azure=False, custom_llm_provider=None, custom_api_base=None,
# model specific optional params # model specific optional params
# used by text-bison only # used by text-bison only
top_k=40, top_k=40, request_timeout=0, # unused var for old version of OpenAI API
): ):
try: try:
global new_response global new_response

16
litellm/tests/test_completion.py
View file

@ -214,14 +214,14 @@ def test_completion_together_ai_stream():
pytest.fail(f"Error occurred: {e}") pytest.fail(f"Error occurred: {e}")
#### Test A121 ################### #### Test A121 ###################
def test_completion_ai21(): # def test_completion_ai21():
model_name = "j2-light" # model_name = "j2-light"
try: # try:
response = completion(model=model_name, messages=messages) # response = completion(model=model_name, messages=messages)
# Add any assertions here to check the response # # Add any assertions here to check the response
print(response) # print(response)
except Exception as e: # except Exception as e:
pytest.fail(f"Error occurred: {e}") # pytest.fail(f"Error occurred: {e}")
# test config file with completion # # test config file with completion #
# def test_completion_openai_config(): # def test_completion_openai_config():

2
pyproject.toml
View file

@ -1,6 +1,6 @@
[tool.poetry] [tool.poetry]
name = "litellm" name = "litellm"
version = "0.1.391" version = "0.1.392"
description = "Library to easily interface with LLM API providers" description = "Library to easily interface with LLM API providers"
authors = ["BerriAI"] authors = ["BerriAI"]
license = "MIT License" license = "MIT License"