@reititin/client 0.0.6 → 0.0.8

package/index.js

@@ -1,10 +1,9 @@
 import http2 from 'http2';
 import axios from 'axios';
 
-export const ReititinClient = ({ agentId, onMessage }) => {
-  const MESSAGE_ENDPOINT = `/api/agents/messages?agentId=${agentId}`;
-  const FINALIZE_URL = "https://reititin.com/api/finalizeMessage";
-  let stayIdle = null;
+export const ReititinClient = ({ token, onMessage }) => {
+  const MESSAGE_ENDPOINT = `/api/agents/messages`;
+  const FINALIZE_URL = "https://test.reititin.com/api/finalizeMessage";
   let buffer = "";
 
   const connectSSE = () => {
@@ -12,14 +11,13 @@ export const ReititinClient = ({ agentId, onMessage }) => {
     
     console.log("Waiting for messages.");
 
-    const client = http2.connect('https://reititin.com');
-    const req = client.request({ ':method': 'GET', ':path': MESSAGE_ENDPOINT, 'accept': 'text/event-stream' })
-      .setEncoding('utf8');
-
-    stayIdle = setTimeout(() => {
-      req.close();
-      client.close();
-    }, 55 * 60 * 1000);
+    const client = http2.connect('https://test.reititin.com');
+    const req = client.request({ 
+      ':method': 'GET', 
+      ':path': MESSAGE_ENDPOINT, 
+      'accept': 'text/event-stream',
+      'authorization': `Bearer ${token}`
+    }).setEncoding('utf8');
 
     req.on('data', chunk => buffer += chunk);
 
@@ -29,17 +27,24 @@ export const ReititinClient = ({ agentId, onMessage }) => {
         const chat_id = lines.find(line => line.startsWith('chat_id:'))?.split(': ')[1];
         const agent_id = lines.find(line => line.startsWith('agent_id:'))?.split(': ')[1];
         const rawData = lines.find(line => line.startsWith('data:'))?.replace('data: ', '');
-        if (onMessage) {
+        if (onMessage && rawData && agent_id && chat_id) {
           console.log("Message received. Processing.");
           const processedMessage = await onMessage(JSON.parse(rawData));
-          await axios.post(FINALIZE_URL, { chat_id: chat_id, agent_id: agent_id, message: processedMessage });
+          await axios.post(
+            FINALIZE_URL, 
+            { chat_id, agent_id, message: processedMessage }, 
+            {
+              headers: {
+                Authorization: `Bearer ${token}`
+              }
+            }
+          );
           console.log(`Response sent.`);
         }
         buffer = "";
       } catch(err) {
         console.log("Reconnecting.");
       }
-      stayIdle = null;
       scheduleReconnect();
     });
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@reititin/client",
-  "version": "0.0.6",
+  "version": "0.0.8",
   "description": "Reititin client allows you to connect your LLM system to Reititin platform.",
   "keywords": [
     "AI",