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

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()