cohere[patch]: Fix positional argument (langchain-ai#19678)

cohere: Fix positional argument Co-authored-by: Bagatur <[email protected]>
FalkorDB · Mar 30, 2024 · 474d6aa · 474d6aa
1 parent 99a8c66
commit 474d6aa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/libs/partners/cohere/langchain_cohere/chat_models.py b/libs/partners/cohere/langchain_cohere/chat_models.py
@@ -303,7 +303,7 @@ async def _agenerate(
 
     def get_num_tokens(self, text: str) -> int:
         """Calculate number of tokens."""
-        return len(self.client.tokenize(text).tokens)
+        return len(self.client.tokenize(text=text).tokens)
 
 
 def _format_cohere_tool_calls(