Fix load_model

huggingface · Mar 3, 2025 · 55cf04f · 55cf04f
1 parent e40a6a0
commit 55cf04f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/smolagents/cli.py b/src/smolagents/cli.py
@@ -79,7 +79,7 @@ def parse_arguments(description):
     return parser.parse_args()
 
 
-def load_model(model_type: str, model_id: str, api_base: str | None, api_key: str | None) -> Model:
+def load_model(model_type: str, model_id: str, api_base: str | None = None, api_key: str | None = None) -> Model:
     if model_type == "OpenAIServerModel":
         return OpenAIServerModel(
             api_key=api_key or os.getenv("FIREWORKS_API_KEY"),