exception_type work

This commit is contained in:
ishaan-jaff 2023-09-19 21:29:51 -07:00
parent e9899db545
commit fc10cf5eeb
4 changed files with 53 additions and 19 deletions

View file

@ -1046,7 +1046,7 @@ def completion(
except Exception as e:
## Map to OpenAI Exception
raise exception_type(
model=model, custom_llm_provider=custom_llm_provider, original_exception=e
model=model, custom_llm_provider=custom_llm_provider, original_exception=e, completion_kwargs=args,
)