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"]