Spaces:
Sleeping
Sleeping
Update format_transplant.py
Browse files- format_transplant.py +7 -1
format_transplant.py
CHANGED
|
@@ -335,9 +335,15 @@ def llm_config_from_args(
|
|
| 335 |
f"No API key for provider '{provider_str}'. "
|
| 336 |
f"Set env var {defaults.get('env', '?')} or pass --llm-key."
|
| 337 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 338 |
return LLMConfig(
|
| 339 |
provider=LLMProvider(provider_str),
|
| 340 |
-
model=
|
| 341 |
api_key=resolved_key,
|
| 342 |
base_url=defaults.get("base_url"),
|
| 343 |
)
|
|
|
|
| 335 |
f"No API key for provider '{provider_str}'. "
|
| 336 |
f"Set env var {defaults.get('env', '?')} or pass --llm-key."
|
| 337 |
)
|
| 338 |
+
|
| 339 |
+
# Handle 'auto' or 'default' markers from UI/CLI
|
| 340 |
+
resolved_model = model
|
| 341 |
+
if resolved_model and resolved_model.lower() in ("auto", "default"):
|
| 342 |
+
resolved_model = None
|
| 343 |
+
|
| 344 |
return LLMConfig(
|
| 345 |
provider=LLMProvider(provider_str),
|
| 346 |
+
model=resolved_model or defaults.get("model", ""),
|
| 347 |
api_key=resolved_key,
|
| 348 |
base_url=defaults.get("base_url"),
|
| 349 |
)
|