making cohere encoder optional (#44)

elisalimli · web-flow · commit 68074e18d917 · 2024-02-15T13:11:51.000-08:00
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)
 
 

Original file line number	Diff line number	Diff line change
`@@ -24,7 +24,8 @@ def create_route_layer() -> RouteLayer:`
`24`	`24`	`score_threshold=0.5,`
`25`	`25`	`)`
`26`	`26`	`]`
`27`		`- encoder = CohereEncoder(cohere_api_key=config("COHERE_API_KEY"))`
	`27`	`+ cohere_api_key = config("COHERE_API_KEY", None)`
	`28`	`+ encoder = CohereEncoder(cohere_api_key=cohere_api_key) if cohere_api_key else None`
`28`	`29`	`return RouteLayer(encoder=encoder, routes=routes)`
`29`	`30`
`30`	`31`