@tiledesk/tiledesk-tybot-connector 2.0.48 → 2.0.50

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tiledesk/tiledesk-tybot-connector",
-  "version": "2.0.48",
+  "version": "2.0.50",
   "description": "Tiledesk Tybot connector",
   "main": "index.js",
   "scripts": {

package/tiledeskChatbotPlugs/directives/DirAskGPTV2.js

@@ -425,7 +425,11 @@ class DirAskGPTV2 {
             return;
 
           } else {
-            await this.#assignAttributes(action, resbody.answer, resbody.source, resbody.content_chunks);
+            let json_sources;
+            if (citations) {
+              json_sources = this.normalizeCitationSources(resbody.citations);
+            }
+            await this.#assignAttributes(action, resbody.answer, resbody.source, resbody.content_chunks, json_sources);
             let tokens = resbody.prompt_token_size;
             if (publicKey === true && !chunks_only) {
 
@@ -532,10 +536,11 @@ class DirAskGPTV2 {
     }
   }
 
-  async #assignAttributes(action, answer, source, chunks) {
+  async #assignAttributes(action, answer, source, chunks, json_sources) {
     winston.debug("DirAskGPTV2assignAttributes action: ", action)
     winston.debug("DirAskGPTV2assignAttributes answer: ", answer)
     winston.debug("DirAskGPTV2assignAttributes source: ", source)
+
     if (this.context.tdcache) {
       if (action.assignReplyTo && answer) {
         await TiledeskChatbot.addParameterStatic(this.context.tdcache, this.context.requestId, action.assignReplyTo, answer);
@@ -546,6 +551,9 @@ class DirAskGPTV2 {
       if (action.assignChunksTo && chunks) {
         await TiledeskChatbot.addParameterStatic(this.context.tdcache, this.context.requestId, action.assignChunksTo, chunks);
       }
+      if (action.assignJsonSourcesTo && json_sources) {
+        await TiledeskChatbot.addParameterStatic(this.context.tdcache, this.context.requestId, action.assignJsonSourcesTo, json_sources);
+      }
     }
   }
 
@@ -561,6 +569,16 @@ class DirAskGPTV2 {
     })
   }
 
+  normalizeCitationSources(citations) {
+    const uniqueMap = new Map();
+    for (const { source_id, ...source } of citations) {
+      if (!uniqueMap.has(source.source_name)) {
+        uniqueMap.set(source.source_name, source);
+      }
+    }
+    return Array.from(uniqueMap.values());
+  }
+
   /**
    * Transforms the transcirpt array in a dictionary like '0': { "question": "xxx", "answer":"xxx"}
    * merging consecutive messages with the same role in a single question or answer.