Tiny nits

This commit is contained in:
Yuhong Sun
2025-01-30 14:17:05 -08:00
committed by Evan Lohn
parent 01e6e9a2ba
commit c43c2320e7
2 changed files with 2 additions and 3 deletions

View File

@@ -440,14 +440,14 @@ def get_llm_max_tokens(
if "max_input_tokens" in model_obj:
max_tokens = model_obj["max_input_tokens"]
logger.info(
logger.debug(
f"Max tokens for {model_name}: {max_tokens} (from max_input_tokens)"
)
return max_tokens
if "max_tokens" in model_obj:
max_tokens = model_obj["max_tokens"]
logger.info(f"Max tokens for {model_name}: {max_tokens} (from max_tokens)")
logger.debug(f"Max tokens for {model_name}: {max_tokens} (from max_tokens)")
return max_tokens
logger.error(f"No max tokens found for LLM: {model_name}")