Cohere envvar fix

This commit is contained in:
Justin Hayes 2024-06-04 13:16:14 -04:00 committed by GitHub
parent 3f7a44d4b8
commit 5743db5d03
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -34,7 +34,9 @@ class Pipeline:
self.name = "cohere/"
self.valves = self.Valves(**{"COHERE_API_KEY": os.getenv("COHERE_API_KEY")})
self.valves = self.Valves(
**{"COHERE_API_KEY": os.getenv("COHERE_API_KEY", "your-api-key-here")}
)
self.pipelines = self.get_cohere_models()