Update lib/@randydev/together/cohere.js

lib/@randydev/together/cohere.js

@@ -1,20 +1,23 @@
-import got from 'got';
+import { CohereClient } from "cohere-ai";
 
-async function CohereAI(message) {
-    const response = await got.post('https://api.cohere.com/v1/chat', {
-        headers: {
-            'Authorization': 'Bearer ' + process.env['COHERE_API_KEY'],
-            'Content-Type': 'application/json'
-        },
-        json: {
-            "model": "command-r-08-2024",
-            'message': `${message}`,
-            "temperature": 0.3,
-            "chat_history": [],
-            "prompt_truncation": "AUTO",
-            "stream": false
-        }
-    }).json();
-    return response;
+const cohere = new CohereClient({
+  token: process.env["COHERE_API_KEY"],
+});
+
+function CohereAI(message) {
+  const stream = await cohere.chatStream({
+    model: "command-r-08-2024",
+    message: `${message}`,
+    temperature: 0.3,
+    chatHistory: [],
+    promptTruncation: "AUTO"
+  });
+  let answer = "";
+  for await (const chat of stream) {
+      if (chat.eventType === "text-generation") {
+          answer += chat.text;
+      }
+  }
+  return answer
 }
 export { CohereAI };