@qbraid-core/chat 0.7.0 → 0.8.0

package/dist/src/client.d.ts

@@ -2,6 +2,6 @@ import { QbraidSession, QbraidClient } from '@qbraid-core/base';
 import { ChatModel, ChatResponse } from './types';
 export declare class ChatClient extends QbraidClient {
     constructor(session?: QbraidSession);
-    createMessage(prompt: string, model?: string): Promise<ChatResponse>;
+    createMessage(prompt: string, model?: string, stream?: boolean): Promise<ChatResponse | string>;
     getModels(): Promise<ChatModel[]>;
 }

package/dist/src/client.js

@@ -8,12 +8,34 @@ class ChatClient extends base_1.QbraidClient {
     constructor(session) {
         super(session);
     }
-    async createMessage(prompt, model = 'gpt-4o-mini') {
+    async createMessage(prompt, model = 'gpt-4o-mini', stream = false) {
         const response = await this.session.client.post('/chat', {
             prompt,
             model,
+            stream,
+        }, {
+            responseType: stream ? 'stream' : 'json',
         });
-        return response.data;
+        // stream: false, return type: ChatResponse
+        // stream: true, return type: string
+        if (stream) {
+            return new Promise((resolve, reject) => {
+                let data = '';
+                response.data.on('data', (chunk) => {
+                    data += chunk.toString();
+                });
+                response.data.on('end', () => {
+                    console.log('Stream ended');
+                    resolve(data);
+                });
+                response.data.on('error', (err) => {
+                    reject(err);
+                });
+            });
+        }
+        else {
+            return response.data;
+        }
     }
     async getModels() {
         const response = await this.session.client.get('/chat/models');

package/dist/src/client.js.map

@@ -1 +1 @@
-{"version":3,"file":"client.js","sourceRoot":"","sources":["../../src/client.ts"],"names":[],"mappings":";AAAA,8CAA8C;AAC9C,uBAAuB;;;AAEvB,4CAAgE;AAIhE,MAAa,UAAW,SAAQ,mBAAY;IAC1C,YAAY,OAAuB;QACjC,KAAK,CAAC,OAAO,CAAC,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,MAAc,EAAE,KAAK,GAAG,aAAa;QACvD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAe,OAAO,EAAE;YACrE,MAAM;YACN,KAAK;SACN,CAAC,CAAC;QACH,OAAO,QAAQ,CAAC,IAAI,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,SAAS;QACb,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,CAAc,cAAc,CAAC,CAAC;QAC5E,OAAO,QAAQ,CAAC,IAAI,CAAC;IACvB,CAAC;CACF;AAjBD,gCAiBC"}
+{"version":3,"file":"client.js","sourceRoot":"","sources":["../../src/client.ts"],"names":[],"mappings":";AAAA,8CAA8C;AAC9C,uBAAuB;;;AAEvB,4CAAgE;AAIhE,MAAa,UAAW,SAAQ,mBAAY;IAC1C,YAAY,OAAuB;QACjC,KAAK,CAAC,OAAO,CAAC,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,aAAa,CACjB,MAAc,EACd,QAAgB,aAAa,EAC7B,SAAkB,KAAK;QAEvB,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAC7C,OAAO,EACP;YACE,MAAM;YACN,KAAK;YACL,MAAM;SACP,EACD;YACE,YAAY,EAAE,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM;SACzC,CACF,CAAC;QACF,2CAA2C;QAC3C,oCAAoC;QAEpC,IAAI,MAAM,EAAE,CAAC;YACX,OAAO,IAAI,OAAO,CAAS,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;gBAC7C,IAAI,IAAI,GAAG,EAAE,CAAC;gBACd,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAa,EAAE,EAAE;oBACzC,IAAI,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;gBAC3B,CAAC,CAAC,CAAC;gBACH,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE;oBAC3B,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;oBAC5B,OAAO,CAAC,IAAI,CAAC,CAAC;gBAChB,CAAC,CAAC,CAAC;gBACH,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAU,EAAE,EAAE;oBACvC,MAAM,CAAC,GAAG,CAAC,CAAC;gBACd,CAAC,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;QACL,CAAC;aAAM,CAAC;YACN,OAAO,QAAQ,CAAC,IAAI,CAAC;QACvB,CAAC;IACH,CAAC;IAED,KAAK,CAAC,SAAS;QACb,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,CAAc,cAAc,CAAC,CAAC;QAC5E,OAAO,QAAQ,CAAC,IAAI,CAAC;IACvB,CAAC;CACF;AA/CD,gCA+CC"}

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@qbraid-core/chat",
   "description": "Client for the qBraid AI Chat service.",
-  "version": "0.7.0",
+  "version": "0.8.0",
   "main": "dist/src/index.js",
   "types": "dist/src/index.d.ts",
   "author": "qBraid Development Team",
@@ -32,7 +32,7 @@
   },
   "homepage": "https://qbraid.github.io/qbraid-core-js/modules/chat.html",
   "dependencies": {
-    "@qbraid-core/base": "0.7.0"
+    "@qbraid-core/base": "0.8.0"
   },
   "scripts": {
     "clean": "rimraf dist tsconfig.tsbuildinfo src/*.d.ts src/*.js",