@docbrasil/api-systemmanager 1.1.88 → 1.1.90

package/api/ai/sessions.js

@@ -245,6 +245,89 @@ class AISession {
       throw ex;
     }
   }
+  /**
+   * @author Myndware <augusto.pissarra@myndware.com>
+   * @description Add documents to an existing agent session.
+   * The agent will handle the documentIds accordingly.
+   * @param {object} params Parameters
+   * @param {string} params.sessionId The session ID
+   * @param {array<string>} params.documentIds Array of document IDs to add
+   * @param {string} authorization Authorization token
+   * @return {Promise<object>} data The result from the agent
+   * @public
+   * @async
+   * @example
+   *
+   * const API = require('@docbrasil/api-systemmanager');
+   * const api = new API();
+   * const authorization = '...';
+   * const params = {
+   *   sessionId: 'session-abc-123',
+   *   documentIds: ['doc-123', 'doc-456']
+   * };
+   * const retData = await api.ai.sessions.addDocuments(params, authorization);
+   */
+  async addDocuments(params, authorization) {
+    const self = this;
+
+    try {
+      Joi.assert(params, Joi.object().required().error(new Error('params is required')));
+      Joi.assert(params.sessionId, Joi.string().required().error(new Error('sessionId is required')));
+      Joi.assert(params.documentIds, Joi.array().items(Joi.string()).min(1).required().error(new Error('documentIds is required and must be a non-empty array')));
+
+      const { sessionId, documentIds } = params;
+
+      const client = self.parent.dispatch.getAkamaiClient();
+      const apiCall = client
+        .post(`/agents/${sessionId}/documents/add`, { documentIds }, self._setHeader(authorization));
+
+      return self._returnData(await apiCall);
+    } catch (ex) {
+      throw ex;
+    }
+  }
+
+  /**
+   * @author Myndware <augusto.pissarra@myndware.com>
+   * @description Remove documents from an existing agent session.
+   * The agent will handle the documentIds accordingly.
+   * @param {object} params Parameters
+   * @param {string} params.sessionId The session ID
+   * @param {array<string>} params.documentIds Array of document IDs to remove
+   * @param {string} authorization Authorization token
+   * @return {Promise<object>} data The result from the agent
+   * @public
+   * @async
+   * @example
+   *
+   * const API = require('@docbrasil/api-systemmanager');
+   * const api = new API();
+   * const authorization = '...';
+   * const params = {
+   *   sessionId: 'session-abc-123',
+   *   documentIds: ['doc-123', 'doc-456']
+   * };
+   * const retData = await api.ai.sessions.removeDocuments(params, authorization);
+   */
+  async removeDocuments(params, authorization) {
+    const self = this;
+
+    try {
+      Joi.assert(params, Joi.object().required().error(new Error('params is required')));
+      Joi.assert(params.sessionId, Joi.string().required().error(new Error('sessionId is required')));
+      Joi.assert(params.documentIds, Joi.array().items(Joi.string()).min(1).required().error(new Error('documentIds is required and must be a non-empty array')));
+
+      const { sessionId, documentIds } = params;
+
+      const client = self.parent.dispatch.getAkamaiClient();
+      const apiCall = client
+        .post(`/agents/${sessionId}/documents/remove`, { documentIds }, self._setHeader(authorization));
+
+      return self._returnData(await apiCall);
+    } catch (ex) {
+      throw ex;
+    }
+  }
 }
 
 export default AISession;