Update Cohere client setting (#118)

nabsabraham · web-flow · commit 6c7f16e69191 · 2024-02-13T10:28:29.000-05:00
Adding a single flag parameter that allows Cohere to track traffic from
this specific python client.
diff --git a/redisvl/utils/vectorize/text/cohere.py b/redisvl/utils/vectorize/text/cohere.py
@@ -79,7 +79,7 @@ def __init__(
                 "Provide it in api_config or set the COHERE_API_KEY environment variable."
             )
 
-        client = cohere.Client(api_key)
+        client = cohere.Client(api_key, client_name="redisvl")
         dims = self._set_model_dims(client, model)
         super().__init__(model=model, dims=dims, client=client)
 

Original file line number	Diff line number	Diff line change
`@@ -79,7 +79,7 @@ def __init__(`
`79`	`79`	`"Provide it in api_config or set the COHERE_API_KEY environment variable."`
`80`	`80`	`)`
`81`	`81`
`82`		`- client = cohere.Client(api_key)`
	`82`	`+ client = cohere.Client(api_key, client_name="redisvl")`
`83`	`83`	`dims = self._set_model_dims(client, model)`
`84`	`84`	`super().__init__(model=model, dims=dims, client=client)`
`85`	`85`