From 48faad3bdd1f4717709d48828ef92c6e68227ec8 Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Thu, 24 Jul 2025 14:15:49 -0700
Subject: [PATCH] add auto_router_embedding_model

---
 litellm/types/router.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/litellm/types/router.py b/litellm/types/router.py
index 02647c543a7..60e03d39183 100644
--- a/litellm/types/router.py
+++ b/litellm/types/router.py
@@ -209,6 +209,12 @@ class GenericLiteLLMParams(CredentialLiteLLMParams, CustomPricingLiteLLMParams):
     model_info: Optional[Dict] = None
     mock_response: Optional[Union[str, ModelResponse, Exception, Any]] = None
 
+
+    # auto-router params
+    auto_router_config_path: Optional[str] = None
+    auto_router_default_model: Optional[str] = None
+    auto_router_embedding_model: Optional[str] = None
+
     def __init__(
         self,
         custom_llm_provider: Optional[str] = None,
@@ -253,6 +259,10 @@ class GenericLiteLLMParams(CredentialLiteLLMParams, CustomPricingLiteLLMParams):
         merge_reasoning_content_in_choices: Optional[bool] = False,
         model_info: Optional[Dict] = None,
         mock_response: Optional[Union[str, ModelResponse, Exception, Any]] = None,
+        # auto-router params
+        auto_router_config_path: Optional[str] = None,
+        auto_router_default_model: Optional[str] = None,
+        auto_router_embedding_model: Optional[str] = None,
         **params,
     ):
         args = locals()