phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

add context window exceeded error mapping to openai

Browse source
This commit is contained in:
Krrish Dholakia 2023-08-29 16:08:00 -07:00
parent d7fe4f6c90
commit f7955d52b5
4 changed files with 10 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/tests/test_exceptions.py
View file

@ -35,12 +35,12 @@ litellm.failure_callback = ["sentry"]
# Approach: Run each model through the test -> assert if the correct error (always the same one) is triggered
# models = ["gpt-3.5-turbo", "chatgpt-test", "claude-instant-1", "command-nightly"]
test_model = "claude-instant-1"
models = ["claude-instant-1"]
test_model = "gpt-3.5-turbo"
models = ["gpt-3.5-turbo"]
def logging_fn(model_call_dict):
return
# return
if "model" in model_call_dict:
print(f"model_call_dict: {model_call_dict['model']}")
else:
@ -59,7 +59,6 @@ def test_context_window(model):
messages=messages,
logger_fn=logging_fn,
)
print(f"response: {response}")
except ContextWindowExceededError as e:
print(f"ContextWindowExceededError: {e.llm_provider}")
return