adding coverage for openrouter in exception mapping

This commit is contained in:
Krrish Dholakia 2023-08-29 15:30:24 -07:00
parent 9646c03fe5
commit efbde50a03
6 changed files with 41 additions and 15 deletions

View file

@ -592,6 +592,7 @@ def completion(
model_response["choices"][0]["message"]["content"] = str(completion_response)
model_response["created"] = time.time()
model_response["model"] = model
response = model_response
elif model in litellm.vertex_text_models:
import vertexai
from vertexai.language_models import TextGenerationModel