This commit is contained in:
yujong-lee 2023-08-31 20:18:17 +09:00
parent 7f27bd6435
commit 119e75655f

View file

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