Do fast env validation if models start with openai/ anthropic/

This commit is contained in:
Paul Gauthier 2024-08-06 15:30:43 -03:00
parent ed9ed895d0
commit f059505edf

View file

@ -566,9 +566,9 @@ class Model:
"""Fast path for common models. Avoids forcing litellm import.""" """Fast path for common models. Avoids forcing litellm import."""
model = self.name model = self.name
if model in OPENAI_MODELS: if model in OPENAI_MODELS or model.startswith("openai/"):
var = "OPENAI_API_KEY" var = "OPENAI_API_KEY"
elif model in ANTHROPIC_MODELS: elif model in ANTHROPIC_MODELS or model.startswith("anthropic/"):
var = "ANTHROPIC_API_KEY" var = "ANTHROPIC_API_KEY"
else: else:
return return