Cleanup test, at least pass with openai

This commit is contained in:
yujong-lee 2023-08-31 20:33:34 +09:00
parent 119e75655f
commit 2cc9074d09
2 changed files with 5 additions and 39 deletions

View file

@ -1361,7 +1361,7 @@ def exception_type(model, original_exception, custom_llm_provider):
original_exception.llm_provider = "azure"
else:
original_exception.llm_provider = "openai"
if "This model's maximum context length is" in original_exception.message:
if "This model's maximum context length is" in original_exception._message:
raise ContextWindowExceededError(
message=str(original_exception),
model=model,