rename validator (#25)

more appropriate name
This commit is contained in:
Harrison Chase 2022-10-25 20:56:48 -07:00 committed by GitHub
parent d2fdcba29d
commit 316aae8223
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -56,7 +56,7 @@ class Cohere(BaseModel, LLM):
extra = Extra.forbid extra = Extra.forbid
@root_validator() @root_validator()
def template_is_valid(cls, values: Dict) -> Dict: def validate_environment(cls, values: Dict) -> Dict:
"""Validate that api key and python package exists in environment.""" """Validate that api key and python package exists in environment."""
if "COHERE_API_KEY" not in os.environ: if "COHERE_API_KEY" not in os.environ:
raise ValueError( raise ValueError(