clean up gpt4 routers. add model_info base_model for azure gpt router (…

…#620)
Skyvern-AI · Jul 18, 2024 · fd9419b · fd9419b
1 parent 78a1bae
commit fd9419b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/skyvern/forge/sdk/api/llm/models.py b/skyvern/forge/sdk/api/llm/models.py
@@ -1,4 +1,4 @@
-from dataclasses import dataclass
+from dataclasses import dataclass, field
 from typing import Any, Awaitable, Literal, Protocol
 
 from skyvern.forge.sdk.models import Step
@@ -27,6 +27,7 @@ class LLMRouterModelConfig:
     model_name: str
     # https://litellm.vercel.app/docs/routing
     litellm_params: dict[str, Any]
+    model_info: dict[str, Any] = field(default_factory=dict)
     tpm: int | None = None
     rpm: int | None = None