making cohere encoder optional (#44)

superagent-ai · Feb 15, 2024 · 68074e1 · 68074e1
1 parent 3f20dd8
commit 68074e1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/service/router.py b/service/router.py
@@ -24,7 +24,8 @@ def create_route_layer() -> RouteLayer:
             score_threshold=0.5,
         )
     ]
-    encoder = CohereEncoder(cohere_api_key=config("COHERE_API_KEY"))
+    cohere_api_key = config("COHERE_API_KEY", None)
+    encoder = CohereEncoder(cohere_api_key=cohere_api_key) if cohere_api_key else None
     return RouteLayer(encoder=encoder, routes=routes)