charieswei
/
TradingAgents
mirror of https://github.com/TauricResearch/TradingAgents.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Normalize model validation inputs

This commit is contained in:
nornen0202 2026-04-07 01:32:42 +09:00
parent e17db7bd35
commit 83ad742dec
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/test_model_validation.py
View File

@ -50,3 +50,6 @@ class ModelValidationTests(unittest.TestCase):
client.get_llm() client.get_llm()
self.assertEqual(caught, []) self.assertEqual(caught, [])
def test_validator_accepts_known_model_with_surrounding_whitespace(self):
self.assertTrue(validate_model(" openai ", " gpt-5.4 "))

5
tradingagents/llm_clients/validators.py
View File

@ -15,7 +15,8 @@ def validate_model(provider: str, model: str) -> bool:
For ollama, openrouter - any model is accepted. For ollama, openrouter - any model is accepted.
""" """
provider_lower = provider.lower() provider_lower = provider.lower().strip()
model_name = model.strip()
if provider_lower in ("ollama", "openrouter"): if provider_lower in ("ollama", "openrouter"):
return True return True
@ -23,4 +24,4 @@ def validate_model(provider: str, model: str) -> bool:
if provider_lower not in VALID_MODELS: if provider_lower not in VALID_MODELS:
return True return True
return model in VALID_MODELS[provider_lower] return model_name in VALID_MODELS[provider_lower]