wingbot 3.72.0 → 3.73.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "wingbot",
-  "version": "3.72.0",
+  "version": "3.73.0",
   "description": "Enterprise Messaging Bot Conversation Engine",
   "main": "index.js",
   "type": "commonjs",

package/src/LLM.js

@@ -110,6 +110,7 @@ const Ai = require('./Ai');
  * @typedef {object} PromptInfo
  * @prop {LLMMessage[]} prompt
  * @prop {LLMMessage} result
+ * @prop {VectorSearchResult} [vectorSearchResult]
  */
 
 /**
@@ -122,6 +123,20 @@ const Ai = require('./Ai');
  * @prop {LogPrompt} logPrompt
  */
 
+/**
+ * @typedef {object} VectorSearchDocument
+ * @property {string} id
+ * @property {string} name
+ * @property {string} text
+ * @property {number} cosineDistance
+ * @property {boolean} excludedByCosineDistanceThreshold
+ */
+/**
+ * @typedef {object} VectorSearchResult
+ * @property {number} maximalCosineDistanceThreshold
+ * @property {number} nearestNeighbourCount
+ * @property {VectorSearchDocument[]} resultDocuments
+ */
 /**
  * @class LLM
  */
@@ -233,7 +248,7 @@ class LLM {
 
         const prompt = session.toArray(true);
         const result = await this._provider.requestChat(prompt, opts);
-        this._logPrompt(prompt, result);
+        this.logPrompt(prompt, result);
         return result;
     }
 
@@ -241,11 +256,12 @@ class LLM {
      *
      * @param {LLMMessage[]} prompt
      * @param {LLMMessage} result
+     * @param {VectorSearchResult} [vectorSearchResult]
      */
-    _logPrompt (prompt, result) {
+    logPrompt (prompt, result, vectorSearchResult) {
         this._lastResult = result;
         this._configuration.logger.logPrompt({
-            prompt, result
+            prompt, result, vectorSearchResult
         });
     }
 

package/src/Processor.js

@@ -405,6 +405,7 @@ class Processor extends EventEmitter {
             return { status: 304 };
         }
 
+        /** @type {LLMConfiguration} */
         const llmOptions = {
             provider: new LLMMockProvider(),
             ...this.options.llm

package/src/graphApi/GraphApi.js

@@ -27,9 +27,6 @@ const DEFAULT_CACHE = 86400000; // 24 hours
 /** @typedef {import('../CallbackAuditLog')} AuditLog */
 /** @typedef {import('graphql')} GqlLib */
 
-/**
- * Experimental chatbot API
- */
 class GraphApi {
 
     /**

package/src/graphApi/schema.gql

@@ -109,9 +109,24 @@ type LLMMessage {
     toolCalls: [ToolCall!]
 }
 
+type VectorSearchDocument {
+  id: String!
+  name: String!
+  text: String!
+  cosineDistance: Float!
+  excludedByCosineDistanceThreshold: Boolean!
+}
+
+type VectorSearchResult {
+  maximalCosineDistanceThreshold: Float! 
+  nearestNeighbourCount: Int!
+  resultDocuments: [VectorSearchDocument!]!  
+}
+
 type PromptInfo {
     prompt: [LLMMessage!]!
     result: LLMMessage!
+    vectorSearchResult: VectorSearchResult
 }
 
 type UserInteraction {
@@ -340,4 +355,4 @@ type Mutation {
 
     "subscribe with metadata"
     subscribeWithData (subscriptions: [SubscriptionData!]!): Boolean
-}
+}

package/src/testTools/PromptAssert.js

@@ -59,6 +59,18 @@ class PromptAssert {
         return this;
     }
 
+    /**
+     * Check if vector search documents contain selected string
+     *
+     * @param {string} search
+     * @returns {this}
+     */
+    vectorSearchContains (search) {
+        const documents = this._flatVectorSearchDocuments();
+        this._promptContains(search, documents, false, 'in vector search documents');
+        return this;
+    }
+
     /**
      *
      * @param {(value: LLMMessage, index: number, array: LLMMessage[]) => unknown} filter
@@ -79,6 +91,16 @@ class PromptAssert {
             .map((prompt) => prompt.result);
     }
 
+    /**
+     *
+     * @returns {LLMMessage[]}
+     */
+    _flatVectorSearchDocuments () {
+        return this._prompts
+            .flatMap((prompt) => prompt.vectorSearchResult?.resultDocuments || [])
+            .map((doc) => ({ role: 'system', content: doc.text }));
+    }
+
     _promptContains (search, messages, notContains = false, addMessage = 'No LLM message found') {
         if (messages.length === 0) {
             PromptAssert.debug(this._prompts, true);