From d3c1c4bf28d2532e4421f0fa2b29b44f2a2a9f2a Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Mon, 18 Dec 2023 08:55:51 +0530 Subject: [PATCH] (feat) set default openrouter configs --- litellm/llms/openrouter.py | 11 +++++++++++ litellm/utils.py | 2 +- 2 files changed, 12 insertions(+), 1 deletion(-) create mode 100644 litellm/llms/openrouter.py diff --git a/litellm/llms/openrouter.py b/litellm/llms/openrouter.py new file mode 100644 index 0000000000..c1ae146c98 --- /dev/null +++ b/litellm/llms/openrouter.py @@ -0,0 +1,11 @@ +from typing import List + +class OpenrouterConfig(): + """ + Reference: https://openrouter.ai/docs#format + + """ + # OpenRouter-only parameters + transforms: List[str] = [] + models: List[str] = [] + route: str = '' diff --git a/litellm/utils.py b/litellm/utils.py index ab22c200dc..90e5d4f09c 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -3962,7 +3962,7 @@ def convert_to_model_response_object(response_object: Optional[dict]=None, model raise Exception("Error in response object format") if model_response_object is None: - model_response_object = EmbeddingResponse() + model_response_object = ImageResponse() if "created" in response_object: model_response_object.created = response_object["created"]