cstr commited on
Commit
2126cb1
·
verified ·
1 Parent(s): f345d03

Update format_transplant.py

Browse files
Files changed (1) hide show
  1. 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=model or defaults.get("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
  )