diff --git a/dsp/modules/cohere.py b/dsp/modules/cohere.py index 1100206d4b..6bfb609cdb 100644 --- a/dsp/modules/cohere.py +++ b/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, diff --git a/dsp/modules/sentence_vectorizer.py b/dsp/modules/sentence_vectorizer.py index b878420f70..61a87bc872 100644 --- a/dsp/modules/sentence_vectorizer.py +++ b/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)