fix request_timeout_extra arg

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

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,
# model specific optional params
# used by text-bison only
top_k=40,
top_k=40, request_timeout=0, # unused var for old version of OpenAI API
):
try:
global new_response

View file

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

View file

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