Merge pull request #56 from justinh-rahb/cohere-fix

Cohere envvar fix
This commit is contained in:
Timothy Jaeryang Baek 2024-06-04 10:18:55 -07:00 committed by GitHub
commit 06ed4bc5f2
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()