@qvac/harness 1.0.0-rc.0 → 1.0.0-rc.2

package/dist-lib/spec/hrpc/index.js

@@ -71,7 +71,15 @@ const methods = new Map([
   ['@harness/load-model', 31],
   [31, '@harness/load-model'],
   ['@harness/unload-model', 32],
-  [32, '@harness/unload-model']
+  [32, '@harness/unload-model'],
+  ['@harness/knowledge-upsert', 33],
+  [33, '@harness/knowledge-upsert'],
+  ['@harness/knowledge-remove', 34],
+  [34, '@harness/knowledge-remove'],
+  ['@harness/knowledge-search', 35],
+  [35, '@harness/knowledge-search'],
+  ['@harness/knowledge-status', 36],
+  [36, '@harness/knowledge-status']
 ])
 
 class HRPC {
@@ -111,7 +119,11 @@ class HRPC {
       ['@harness/preview-model-source', getEncoding('@harness/preview-model-source-request')],
       ['@harness/set-model-idle-timeout', getEncoding('@harness/set-model-idle-timeout-request')],
       ['@harness/load-model', getEncoding('@harness/warm-model-request')],
-      ['@harness/unload-model', getEncoding('@harness/unload-model-request')]
+      ['@harness/unload-model', getEncoding('@harness/unload-model-request')],
+      ['@harness/knowledge-upsert', getEncoding('@harness/knowledge-upsert-request')],
+      ['@harness/knowledge-remove', getEncoding('@harness/knowledge-remove-request')],
+      ['@harness/knowledge-search', getEncoding('@harness/knowledge-search-request')],
+      ['@harness/knowledge-status', getEncoding('@harness/empty')]
     ])
     this._responseEncodings = new Map([
       ['@harness/completion', getEncoding('@harness/completion-event')],
@@ -146,7 +158,11 @@ class HRPC {
       ['@harness/preview-model-source', getEncoding('@harness/preview-model-source-response')],
       ['@harness/set-model-idle-timeout', getEncoding('@harness/ok-response')],
       ['@harness/load-model', getEncoding('@harness/load-model-response')],
-      ['@harness/unload-model', getEncoding('@harness/unload-model-response')]
+      ['@harness/unload-model', getEncoding('@harness/unload-model-response')],
+      ['@harness/knowledge-upsert', getEncoding('@harness/knowledge-upsert-response')],
+      ['@harness/knowledge-remove', getEncoding('@harness/knowledge-remove-response')],
+      ['@harness/knowledge-search', getEncoding('@harness/knowledge-search-response')],
+      ['@harness/knowledge-status', getEncoding('@harness/knowledge-status-response')]
     ])
     this._rpc = new RPC(stream, async (req) => {
       const command = methods.get(req.command)
@@ -376,6 +392,22 @@ class HRPC {
     return this._call('@harness/unload-model', args)
   }
 
+  async knowledgeUpsert(args) {
+    return this._call('@harness/knowledge-upsert', args)
+  }
+
+  async knowledgeRemove(args) {
+    return this._call('@harness/knowledge-remove', args)
+  }
+
+  async knowledgeSearch(args) {
+    return this._call('@harness/knowledge-search', args)
+  }
+
+  async knowledgeStatus(args) {
+    return this._call('@harness/knowledge-status', args)
+  }
+
   onCompletion(responseFn) {
     this._handlers['@harness/completion'] = responseFn
   }
@@ -508,6 +540,22 @@ class HRPC {
     this._handlers['@harness/unload-model'] = responseFn
   }
 
+  onKnowledgeUpsert(responseFn) {
+    this._handlers['@harness/knowledge-upsert'] = responseFn
+  }
+
+  onKnowledgeRemove(responseFn) {
+    this._handlers['@harness/knowledge-remove'] = responseFn
+  }
+
+  onKnowledgeSearch(responseFn) {
+    this._handlers['@harness/knowledge-search'] = responseFn
+  }
+
+  onKnowledgeStatus(responseFn) {
+    this._handlers['@harness/knowledge-status'] = responseFn
+  }
+
   _requestIsStream(command) {
     return [
       '@harness/completion',