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

Cohere envvar fix
open-webui · Jun 4, 2024 · 06ed4bc · 06ed4bc
2 parents 3f7a44d + 5743db5
commit 06ed4bc
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/examples/pipelines/providers/cohere_manifold_pipeline.py b/examples/pipelines/providers/cohere_manifold_pipeline.py
@@ -34,7 +34,9 @@ def __init__(self):
 
         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()