Merge pull request stanfordnlp#925 from BeatrixCohere/beatrix/CohereClientName

Add the cohere client name to Cohere calls

Author: arnavsinghvi11

dsp/modules/cohere.py

@@ -58,7 +58,7 @@ def __init__(
             Additional arguments to pass to the API provider.
         """
         super().__init__(model)
-        self.co = cohere.Client(api_key)
+        self.co = cohere.Client(api_key, client_name='dspy')
         self.provider = "cohere"
         self.kwargs = {
             "model": model,

dsp/modules/sentence_vectorizer.py

@@ -128,7 +128,7 @@ def __init__(
         self.embedding_type = embedding_type
 
         import cohere
-        self.client = cohere.Client(api_key)
+        self.client = cohere.Client(api_key, client_name='dspy')
 
     def __call__(self, inp_examples: List["Example"]) -> np.ndarray:
         text_to_vectorize = self._extract_text_from_examples(inp_examples)