Update ChatCohere.py

ChatCohere.py

@@ -16,10 +16,11 @@ def chat_completion(message: str) -> str:
     return response.message.content[0].text
 
 def summarize(message, system_prompt="You are an helpful assistant whose job is to summarize the text you are given in less than 900 charachters (including spaces) in such a way that it would constitute an effective and engaging description of a pokemon card package"):
-        response = chat_completion(
-            message_history=[
+    response = co.chat(
+        model="command-r-plus-08-2024",
+        messages=[
                 {"role": "system", "content": system_prompt},
                 {"role": "user", "content": message},
             ]
-        )
-        return response
+    )
+    return response.message.content[0].text