@athenaintel/sdk 4.3.308 → 4.3.310

package/dist/cjs/api/resources/agents/client/Client.js

@@ -43,35 +43,36 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
     });
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.Agents = void 0;
-const environments = __importStar(require("../../../../environments.js"));
-const core = __importStar(require("../../../../core/index.js"));
-const AthenaIntelligence = __importStar(require("../../../index.js"));
+exports.AgentsClient = void 0;
+const BaseClient_js_1 = require("../../../../BaseClient.js");
 const headers_js_1 = require("../../../../core/headers.js");
+const core = __importStar(require("../../../../core/index.js"));
+const environments = __importStar(require("../../../../environments.js"));
 const errors = __importStar(require("../../../../errors/index.js"));
+const AthenaIntelligence = __importStar(require("../../../index.js"));
 const Client_js_1 = require("../resources/drive/client/Client.js");
 const Client_js_2 = require("../resources/general/client/Client.js");
 const Client_js_3 = require("../resources/research/client/Client.js");
 const Client_js_4 = require("../resources/sql/client/Client.js");
-class Agents {
-    constructor(_options) {
-        this._options = _options;
+class AgentsClient {
+    constructor(options) {
+        this._options = (0, BaseClient_js_1.normalizeClientOptionsWithAuth)(options);
     }
     get drive() {
         var _a;
-        return ((_a = this._drive) !== null && _a !== void 0 ? _a : (this._drive = new Client_js_1.Drive(this._options)));
+        return ((_a = this._drive) !== null && _a !== void 0 ? _a : (this._drive = new Client_js_1.DriveClient(this._options)));
     }
     get general() {
         var _a;
-        return ((_a = this._general) !== null && _a !== void 0 ? _a : (this._general = new Client_js_2.General(this._options)));
+        return ((_a = this._general) !== null && _a !== void 0 ? _a : (this._general = new Client_js_2.GeneralClient(this._options)));
     }
     get research() {
         var _a;
-        return ((_a = this._research) !== null && _a !== void 0 ? _a : (this._research = new Client_js_3.Research(this._options)));
+        return ((_a = this._research) !== null && _a !== void 0 ? _a : (this._research = new Client_js_3.ResearchClient(this._options)));
     }
     get sql() {
         var _a;
-        return ((_a = this._sql) !== null && _a !== void 0 ? _a : (this._sql = new Client_js_4.Sql(this._options)));
+        return ((_a = this._sql) !== null && _a !== void 0 ? _a : (this._sql = new Client_js_4.SqlClient(this._options)));
     }
     /**
      * Coming soon!
@@ -82,9 +83,9 @@ class Agents {
      * Refer to the specific agent's documentation for details on configuration options
      * and expected responses.
      *
-     * @param {string} agentId - The ID of the custom agent to invoke. Create custom agents in [spaces](https://resources.athenaintel.com/docs/agents/create-your-agent).
+     * @param {string} agent_id - The ID of the custom agent to invoke. Create custom agents in [spaces](https://resources.athenaintel.com/docs/agents/create-your-agent).
      * @param {AthenaIntelligence.CustomAgentRequest} request
-     * @param {Agents.RequestOptions} requestOptions - Request-specific configuration.
+     * @param {AgentsClient.RequestOptions} requestOptions - Request-specific configuration.
      *
      * @throws {@link AthenaIntelligence.UnprocessableEntityError}
      *
@@ -98,15 +99,16 @@ class Agents {
      *             }]
      *     })
      */
-    invokeById(agentId, request, requestOptions) {
-        return core.HttpResponsePromise.fromPromise(this.__invokeById(agentId, request, requestOptions));
+    invokeById(agent_id, request, requestOptions) {
+        return core.HttpResponsePromise.fromPromise(this.__invokeById(agent_id, request, requestOptions));
     }
-    __invokeById(agentId, request, requestOptions) {
+    __invokeById(agent_id, request, requestOptions) {
         return __awaiter(this, void 0, void 0, function* () {
-            var _a, _b, _c, _d, _e, _f, _g, _h, _j;
-            let _headers = (0, headers_js_1.mergeHeaders)((_a = this._options) === null || _a === void 0 ? void 0 : _a.headers, (0, headers_js_1.mergeOnlyDefinedHeaders)(Object.assign({}, (yield this._getCustomAuthorizationHeaders()))), requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.headers);
+            var _a, _b, _c, _d, _e, _f, _g, _h, _j, _k;
+            const _authRequest = yield this._options.authProvider.getAuthRequest();
+            const _headers = (0, headers_js_1.mergeHeaders)(_authRequest.headers, (_a = this._options) === null || _a === void 0 ? void 0 : _a.headers, requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.headers);
             const _response = yield ((_b = this._options.fetcher) !== null && _b !== void 0 ? _b : core.fetcher)({
-                url: core.url.join((_d = (_c = (yield core.Supplier.get(this._options.baseUrl))) !== null && _c !== void 0 ? _c : (yield core.Supplier.get(this._options.environment))) !== null && _d !== void 0 ? _d : environments.AthenaIntelligenceEnvironment.Production, `api/v0/agents/${encodeURIComponent(agentId)}/invoke`),
+                url: core.url.join((_d = (_c = (yield core.Supplier.get(this._options.baseUrl))) !== null && _c !== void 0 ? _c : (yield core.Supplier.get(this._options.environment))) !== null && _d !== void 0 ? _d : environments.AthenaIntelligenceEnvironment.Production, `api/v0/agents/${core.url.encodePathParam(agent_id)}/invoke`),
                 method: "POST",
                 headers: _headers,
                 contentType: "application/json",
@@ -116,6 +118,8 @@ class Agents {
                 timeoutMs: ((_g = (_e = requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.timeoutInSeconds) !== null && _e !== void 0 ? _e : (_f = this._options) === null || _f === void 0 ? void 0 : _f.timeoutInSeconds) !== null && _g !== void 0 ? _g : 60) * 1000,
                 maxRetries: (_h = requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.maxRetries) !== null && _h !== void 0 ? _h : (_j = this._options) === null || _j === void 0 ? void 0 : _j.maxRetries,
                 abortSignal: requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.abortSignal,
+                fetchFn: (_k = this._options) === null || _k === void 0 ? void 0 : _k.fetch,
+                logging: this._options.logging,
             });
             if (_response.ok) {
                 return {
@@ -142,6 +146,11 @@ class Agents {
                         body: _response.error.rawBody,
                         rawResponse: _response.rawResponse,
                     });
+                case "body-is-null":
+                    throw new errors.AthenaIntelligenceError({
+                        statusCode: _response.error.statusCode,
+                        rawResponse: _response.rawResponse,
+                    });
                 case "timeout":
                     throw new errors.AthenaIntelligenceTimeoutError("Timeout exceeded when calling POST /api/v0/agents/{agent_id}/invoke.");
                 case "unknown":
@@ -152,11 +161,5 @@ class Agents {
             }
         });
     }
-    _getCustomAuthorizationHeaders() {
-        return __awaiter(this, void 0, void 0, function* () {
-            const apiKeyValue = yield core.Supplier.get(this._options.apiKey);
-            return { "X-API-KEY": apiKeyValue };
-        });
-    }
 }
-exports.Agents = Agents;
+exports.AgentsClient = AgentsClient;

package/dist/cjs/api/resources/agents/client/index.d.ts

@@ -1,2 +1 @@
-export {};
 export * from "./requests/index.js";

package/dist/cjs/api/resources/agents/client/requests/index.d.ts

@@ -1 +1 @@
-export { type CustomAgentRequest } from "./CustomAgentRequest.js";
+export type { CustomAgentRequest } from "./CustomAgentRequest.js";

package/dist/cjs/api/resources/agents/resources/drive/client/Client.d.ts

@@ -1,20 +1,21 @@
 import type { BaseClientOptions, BaseRequestOptions } from "../../../../../../BaseClient.js";
+import { type NormalizedClientOptionsWithAuth } from "../../../../../../BaseClient.js";
 import * as core from "../../../../../../core/index.js";
 import * as AthenaIntelligence from "../../../../../index.js";
-export declare namespace Drive {
+export declare namespace DriveClient {
     interface Options extends BaseClientOptions {
     }
     interface RequestOptions extends BaseRequestOptions {
     }
 }
-export declare class Drive {
-    protected readonly _options: Drive.Options;
-    constructor(_options: Drive.Options);
+export declare class DriveClient {
+    protected readonly _options: NormalizedClientOptionsWithAuth<DriveClient.Options>;
+    constructor(options: DriveClient.Options);
     /**
      * Coming soon! Manage folders and search for files in the internal drive.
      *
      * @param {AthenaIntelligence.agents.DriveAgentRequest} request
-     * @param {Drive.RequestOptions} requestOptions - Request-specific configuration.
+     * @param {DriveClient.RequestOptions} requestOptions - Request-specific configuration.
      *
      * @throws {@link AthenaIntelligence.UnprocessableEntityError}
      *
@@ -28,7 +29,6 @@ export declare class Drive {
      *             }]
      *     })
      */
-    invoke(request: AthenaIntelligence.agents.DriveAgentRequest, requestOptions?: Drive.RequestOptions): core.HttpResponsePromise<AthenaIntelligence.DriveAgentResponse>;
+    invoke(request: AthenaIntelligence.agents.DriveAgentRequest, requestOptions?: DriveClient.RequestOptions): core.HttpResponsePromise<AthenaIntelligence.DriveAgentResponse>;
     private __invoke;
-    protected _getCustomAuthorizationHeaders(): Promise<Record<string, string | undefined>>;
 }

package/dist/cjs/api/resources/agents/resources/drive/client/Client.js

@@ -43,21 +43,22 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
     });
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.Drive = void 0;
-const environments = __importStar(require("../../../../../../environments.js"));
-const core = __importStar(require("../../../../../../core/index.js"));
-const AthenaIntelligence = __importStar(require("../../../../../index.js"));
+exports.DriveClient = void 0;
+const BaseClient_js_1 = require("../../../../../../BaseClient.js");
 const headers_js_1 = require("../../../../../../core/headers.js");
+const core = __importStar(require("../../../../../../core/index.js"));
+const environments = __importStar(require("../../../../../../environments.js"));
 const errors = __importStar(require("../../../../../../errors/index.js"));
-class Drive {
-    constructor(_options) {
-        this._options = _options;
+const AthenaIntelligence = __importStar(require("../../../../../index.js"));
+class DriveClient {
+    constructor(options) {
+        this._options = (0, BaseClient_js_1.normalizeClientOptionsWithAuth)(options);
     }
     /**
      * Coming soon! Manage folders and search for files in the internal drive.
      *
      * @param {AthenaIntelligence.agents.DriveAgentRequest} request
-     * @param {Drive.RequestOptions} requestOptions - Request-specific configuration.
+     * @param {DriveClient.RequestOptions} requestOptions - Request-specific configuration.
      *
      * @throws {@link AthenaIntelligence.UnprocessableEntityError}
      *
@@ -76,8 +77,9 @@ class Drive {
     }
     __invoke(request, requestOptions) {
         return __awaiter(this, void 0, void 0, function* () {
-            var _a, _b, _c, _d, _e, _f, _g, _h, _j;
-            let _headers = (0, headers_js_1.mergeHeaders)((_a = this._options) === null || _a === void 0 ? void 0 : _a.headers, (0, headers_js_1.mergeOnlyDefinedHeaders)(Object.assign({}, (yield this._getCustomAuthorizationHeaders()))), requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.headers);
+            var _a, _b, _c, _d, _e, _f, _g, _h, _j, _k;
+            const _authRequest = yield this._options.authProvider.getAuthRequest();
+            const _headers = (0, headers_js_1.mergeHeaders)(_authRequest.headers, (_a = this._options) === null || _a === void 0 ? void 0 : _a.headers, requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.headers);
             const _response = yield ((_b = this._options.fetcher) !== null && _b !== void 0 ? _b : core.fetcher)({
                 url: core.url.join((_d = (_c = (yield core.Supplier.get(this._options.baseUrl))) !== null && _c !== void 0 ? _c : (yield core.Supplier.get(this._options.environment))) !== null && _d !== void 0 ? _d : environments.AthenaIntelligenceEnvironment.Production, "api/v0/agents/drive/invoke"),
                 method: "POST",
@@ -89,6 +91,8 @@ class Drive {
                 timeoutMs: ((_g = (_e = requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.timeoutInSeconds) !== null && _e !== void 0 ? _e : (_f = this._options) === null || _f === void 0 ? void 0 : _f.timeoutInSeconds) !== null && _g !== void 0 ? _g : 60) * 1000,
                 maxRetries: (_h = requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.maxRetries) !== null && _h !== void 0 ? _h : (_j = this._options) === null || _j === void 0 ? void 0 : _j.maxRetries,
                 abortSignal: requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.abortSignal,
+                fetchFn: (_k = this._options) === null || _k === void 0 ? void 0 : _k.fetch,
+                logging: this._options.logging,
             });
             if (_response.ok) {
                 return {
@@ -115,6 +119,11 @@ class Drive {
                         body: _response.error.rawBody,
                         rawResponse: _response.rawResponse,
                     });
+                case "body-is-null":
+                    throw new errors.AthenaIntelligenceError({
+                        statusCode: _response.error.statusCode,
+                        rawResponse: _response.rawResponse,
+                    });
                 case "timeout":
                     throw new errors.AthenaIntelligenceTimeoutError("Timeout exceeded when calling POST /api/v0/agents/drive/invoke.");
                 case "unknown":
@@ -125,11 +134,5 @@ class Drive {
             }
         });
     }
-    _getCustomAuthorizationHeaders() {
-        return __awaiter(this, void 0, void 0, function* () {
-            const apiKeyValue = yield core.Supplier.get(this._options.apiKey);
-            return { "X-API-KEY": apiKeyValue };
-        });
-    }
 }
-exports.Drive = Drive;
+exports.DriveClient = DriveClient;

package/dist/cjs/api/resources/agents/resources/drive/client/index.d.ts

@@ -1,2 +1 @@
-export {};
 export * from "./requests/index.js";

package/dist/cjs/api/resources/agents/resources/drive/client/requests/index.d.ts

@@ -1 +1 @@
-export { type DriveAgentRequest } from "./DriveAgentRequest.js";
+export type { DriveAgentRequest } from "./DriveAgentRequest.js";

package/dist/cjs/api/resources/agents/resources/general/client/Client.d.ts

@@ -1,20 +1,21 @@
 import type { BaseClientOptions, BaseRequestOptions } from "../../../../../../BaseClient.js";
+import { type NormalizedClientOptionsWithAuth } from "../../../../../../BaseClient.js";
 import * as core from "../../../../../../core/index.js";
 import * as AthenaIntelligence from "../../../../../index.js";
-export declare namespace General {
+export declare namespace GeneralClient {
     interface Options extends BaseClientOptions {
     }
     interface RequestOptions extends BaseRequestOptions {
     }
 }
-export declare class General {
-    protected readonly _options: General.Options;
-    constructor(_options: General.Options);
+export declare class GeneralClient {
+    protected readonly _options: NormalizedClientOptionsWithAuth<GeneralClient.Options>;
+    constructor(options: GeneralClient.Options);
     /**
      * Coming soon! Call the general agent with batched requests and return the results.
      *
      * @param {AthenaIntelligence.GeneralAgentRequest[]} request
-     * @param {General.RequestOptions} requestOptions - Request-specific configuration.
+     * @param {GeneralClient.RequestOptions} requestOptions - Request-specific configuration.
      *
      * @throws {@link AthenaIntelligence.UnprocessableEntityError}
      *
@@ -29,7 +30,7 @@ export declare class General {
      *                 }]
      *         }])
      */
-    batch(request: AthenaIntelligence.GeneralAgentRequest[], requestOptions?: General.RequestOptions): core.HttpResponsePromise<AthenaIntelligence.GeneralAgentResponse[]>;
+    batch(request: AthenaIntelligence.GeneralAgentRequest[], requestOptions?: GeneralClient.RequestOptions): core.HttpResponsePromise<AthenaIntelligence.GeneralAgentResponse[]>;
     private __batch;
     /**
      * Call the general Athena agent synchronously.
@@ -38,7 +39,7 @@ export declare class General {
      * and return the result.
      *
      * @param {AthenaIntelligence.GeneralAgentRequest} request
-     * @param {General.RequestOptions} requestOptions - Request-specific configuration.
+     * @param {GeneralClient.RequestOptions} requestOptions - Request-specific configuration.
      *
      * @throws {@link AthenaIntelligence.UnprocessableEntityError}
      *
@@ -53,13 +54,13 @@ export declare class General {
      *             }]
      *     })
      */
-    invoke(request: AthenaIntelligence.GeneralAgentRequest, requestOptions?: General.RequestOptions): core.HttpResponsePromise<AthenaIntelligence.GeneralAgentResponse>;
+    invoke(request: AthenaIntelligence.GeneralAgentRequest, requestOptions?: GeneralClient.RequestOptions): core.HttpResponsePromise<AthenaIntelligence.GeneralAgentResponse>;
     private __invoke;
     /**
      * Coming soon! Call the general agent and stream events for real-time chat applications.
      *
      * @param {AthenaIntelligence.GeneralAgentRequest} request
-     * @param {General.RequestOptions} requestOptions - Request-specific configuration.
+     * @param {GeneralClient.RequestOptions} requestOptions - Request-specific configuration.
      *
      * @throws {@link AthenaIntelligence.UnprocessableEntityError}
      *
@@ -74,7 +75,6 @@ export declare class General {
      *             }]
      *     })
      */
-    streamEvents(request: AthenaIntelligence.GeneralAgentRequest, requestOptions?: General.RequestOptions): core.HttpResponsePromise<AthenaIntelligence.GeneralAgentResponse>;
+    streamEvents(request: AthenaIntelligence.GeneralAgentRequest, requestOptions?: GeneralClient.RequestOptions): core.HttpResponsePromise<AthenaIntelligence.GeneralAgentResponse>;
     private __streamEvents;
-    protected _getCustomAuthorizationHeaders(): Promise<Record<string, string | undefined>>;
 }

package/dist/cjs/api/resources/agents/resources/general/client/Client.js

@@ -43,21 +43,22 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
     });
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.General = void 0;
-const environments = __importStar(require("../../../../../../environments.js"));
-const core = __importStar(require("../../../../../../core/index.js"));
-const AthenaIntelligence = __importStar(require("../../../../../index.js"));
+exports.GeneralClient = void 0;
+const BaseClient_js_1 = require("../../../../../../BaseClient.js");
 const headers_js_1 = require("../../../../../../core/headers.js");
+const core = __importStar(require("../../../../../../core/index.js"));
+const environments = __importStar(require("../../../../../../environments.js"));
 const errors = __importStar(require("../../../../../../errors/index.js"));
-class General {
-    constructor(_options) {
-        this._options = _options;
+const AthenaIntelligence = __importStar(require("../../../../../index.js"));
+class GeneralClient {
+    constructor(options) {
+        this._options = (0, BaseClient_js_1.normalizeClientOptionsWithAuth)(options);
     }
     /**
      * Coming soon! Call the general agent with batched requests and return the results.
      *
      * @param {AthenaIntelligence.GeneralAgentRequest[]} request
-     * @param {General.RequestOptions} requestOptions - Request-specific configuration.
+     * @param {GeneralClient.RequestOptions} requestOptions - Request-specific configuration.
      *
      * @throws {@link AthenaIntelligence.UnprocessableEntityError}
      *
@@ -77,8 +78,9 @@ class General {
     }
     __batch(request, requestOptions) {
         return __awaiter(this, void 0, void 0, function* () {
-            var _a, _b, _c, _d, _e, _f, _g, _h, _j;
-            let _headers = (0, headers_js_1.mergeHeaders)((_a = this._options) === null || _a === void 0 ? void 0 : _a.headers, (0, headers_js_1.mergeOnlyDefinedHeaders)(Object.assign({}, (yield this._getCustomAuthorizationHeaders()))), requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.headers);
+            var _a, _b, _c, _d, _e, _f, _g, _h, _j, _k;
+            const _authRequest = yield this._options.authProvider.getAuthRequest();
+            const _headers = (0, headers_js_1.mergeHeaders)(_authRequest.headers, (_a = this._options) === null || _a === void 0 ? void 0 : _a.headers, requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.headers);
             const _response = yield ((_b = this._options.fetcher) !== null && _b !== void 0 ? _b : core.fetcher)({
                 url: core.url.join((_d = (_c = (yield core.Supplier.get(this._options.baseUrl))) !== null && _c !== void 0 ? _c : (yield core.Supplier.get(this._options.environment))) !== null && _d !== void 0 ? _d : environments.AthenaIntelligenceEnvironment.Production, "api/v0/agents/general/batch"),
                 method: "POST",
@@ -90,6 +92,8 @@ class General {
                 timeoutMs: ((_g = (_e = requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.timeoutInSeconds) !== null && _e !== void 0 ? _e : (_f = this._options) === null || _f === void 0 ? void 0 : _f.timeoutInSeconds) !== null && _g !== void 0 ? _g : 60) * 1000,
                 maxRetries: (_h = requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.maxRetries) !== null && _h !== void 0 ? _h : (_j = this._options) === null || _j === void 0 ? void 0 : _j.maxRetries,
                 abortSignal: requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.abortSignal,
+                fetchFn: (_k = this._options) === null || _k === void 0 ? void 0 : _k.fetch,
+                logging: this._options.logging,
             });
             if (_response.ok) {
                 return {
@@ -116,6 +120,11 @@ class General {
                         body: _response.error.rawBody,
                         rawResponse: _response.rawResponse,
                     });
+                case "body-is-null":
+                    throw new errors.AthenaIntelligenceError({
+                        statusCode: _response.error.statusCode,
+                        rawResponse: _response.rawResponse,
+                    });
                 case "timeout":
                     throw new errors.AthenaIntelligenceTimeoutError("Timeout exceeded when calling POST /api/v0/agents/general/batch.");
                 case "unknown":
@@ -133,7 +142,7 @@ class General {
      * and return the result.
      *
      * @param {AthenaIntelligence.GeneralAgentRequest} request
-     * @param {General.RequestOptions} requestOptions - Request-specific configuration.
+     * @param {GeneralClient.RequestOptions} requestOptions - Request-specific configuration.
      *
      * @throws {@link AthenaIntelligence.UnprocessableEntityError}
      *
@@ -153,8 +162,9 @@ class General {
     }
     __invoke(request, requestOptions) {
         return __awaiter(this, void 0, void 0, function* () {
-            var _a, _b, _c, _d, _e, _f, _g, _h, _j;
-            let _headers = (0, headers_js_1.mergeHeaders)((_a = this._options) === null || _a === void 0 ? void 0 : _a.headers, (0, headers_js_1.mergeOnlyDefinedHeaders)(Object.assign({}, (yield this._getCustomAuthorizationHeaders()))), requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.headers);
+            var _a, _b, _c, _d, _e, _f, _g, _h, _j, _k;
+            const _authRequest = yield this._options.authProvider.getAuthRequest();
+            const _headers = (0, headers_js_1.mergeHeaders)(_authRequest.headers, (_a = this._options) === null || _a === void 0 ? void 0 : _a.headers, requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.headers);
             const _response = yield ((_b = this._options.fetcher) !== null && _b !== void 0 ? _b : core.fetcher)({
                 url: core.url.join((_d = (_c = (yield core.Supplier.get(this._options.baseUrl))) !== null && _c !== void 0 ? _c : (yield core.Supplier.get(this._options.environment))) !== null && _d !== void 0 ? _d : environments.AthenaIntelligenceEnvironment.Production, "api/v0/agents/general/invoke"),
                 method: "POST",
@@ -166,6 +176,8 @@ class General {
                 timeoutMs: ((_g = (_e = requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.timeoutInSeconds) !== null && _e !== void 0 ? _e : (_f = this._options) === null || _f === void 0 ? void 0 : _f.timeoutInSeconds) !== null && _g !== void 0 ? _g : 60) * 1000,
                 maxRetries: (_h = requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.maxRetries) !== null && _h !== void 0 ? _h : (_j = this._options) === null || _j === void 0 ? void 0 : _j.maxRetries,
                 abortSignal: requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.abortSignal,
+                fetchFn: (_k = this._options) === null || _k === void 0 ? void 0 : _k.fetch,
+                logging: this._options.logging,
             });
             if (_response.ok) {
                 return {
@@ -192,6 +204,11 @@ class General {
                         body: _response.error.rawBody,
                         rawResponse: _response.rawResponse,
                     });
+                case "body-is-null":
+                    throw new errors.AthenaIntelligenceError({
+                        statusCode: _response.error.statusCode,
+                        rawResponse: _response.rawResponse,
+                    });
                 case "timeout":
                     throw new errors.AthenaIntelligenceTimeoutError("Timeout exceeded when calling POST /api/v0/agents/general/invoke.");
                 case "unknown":
@@ -206,7 +223,7 @@ class General {
      * Coming soon! Call the general agent and stream events for real-time chat applications.
      *
      * @param {AthenaIntelligence.GeneralAgentRequest} request
-     * @param {General.RequestOptions} requestOptions - Request-specific configuration.
+     * @param {GeneralClient.RequestOptions} requestOptions - Request-specific configuration.
      *
      * @throws {@link AthenaIntelligence.UnprocessableEntityError}
      *
@@ -226,8 +243,9 @@ class General {
     }
     __streamEvents(request, requestOptions) {
         return __awaiter(this, void 0, void 0, function* () {
-            var _a, _b, _c, _d, _e, _f, _g, _h, _j;
-            let _headers = (0, headers_js_1.mergeHeaders)((_a = this._options) === null || _a === void 0 ? void 0 : _a.headers, (0, headers_js_1.mergeOnlyDefinedHeaders)(Object.assign({}, (yield this._getCustomAuthorizationHeaders()))), requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.headers);
+            var _a, _b, _c, _d, _e, _f, _g, _h, _j, _k;
+            const _authRequest = yield this._options.authProvider.getAuthRequest();
+            const _headers = (0, headers_js_1.mergeHeaders)(_authRequest.headers, (_a = this._options) === null || _a === void 0 ? void 0 : _a.headers, requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.headers);
             const _response = yield ((_b = this._options.fetcher) !== null && _b !== void 0 ? _b : core.fetcher)({
                 url: core.url.join((_d = (_c = (yield core.Supplier.get(this._options.baseUrl))) !== null && _c !== void 0 ? _c : (yield core.Supplier.get(this._options.environment))) !== null && _d !== void 0 ? _d : environments.AthenaIntelligenceEnvironment.Production, "api/v0/agents/general/stream_events"),
                 method: "POST",
@@ -239,6 +257,8 @@ class General {
                 timeoutMs: ((_g = (_e = requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.timeoutInSeconds) !== null && _e !== void 0 ? _e : (_f = this._options) === null || _f === void 0 ? void 0 : _f.timeoutInSeconds) !== null && _g !== void 0 ? _g : 60) * 1000,
                 maxRetries: (_h = requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.maxRetries) !== null && _h !== void 0 ? _h : (_j = this._options) === null || _j === void 0 ? void 0 : _j.maxRetries,
                 abortSignal: requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.abortSignal,
+                fetchFn: (_k = this._options) === null || _k === void 0 ? void 0 : _k.fetch,
+                logging: this._options.logging,
             });
             if (_response.ok) {
                 return {
@@ -265,6 +285,11 @@ class General {
                         body: _response.error.rawBody,
                         rawResponse: _response.rawResponse,
                     });
+                case "body-is-null":
+                    throw new errors.AthenaIntelligenceError({
+                        statusCode: _response.error.statusCode,
+                        rawResponse: _response.rawResponse,
+                    });
                 case "timeout":
                     throw new errors.AthenaIntelligenceTimeoutError("Timeout exceeded when calling POST /api/v0/agents/general/stream_events.");
                 case "unknown":
@@ -275,11 +300,5 @@ class General {
             }
         });
     }
-    _getCustomAuthorizationHeaders() {
-        return __awaiter(this, void 0, void 0, function* () {
-            const apiKeyValue = yield core.Supplier.get(this._options.apiKey);
-            return { "X-API-KEY": apiKeyValue };
-        });
-    }
 }
-exports.General = General;
+exports.GeneralClient = GeneralClient;

package/dist/cjs/api/resources/agents/resources/index.d.ts

@@ -1,7 +1,7 @@
+export * from "./drive/client/requests/index.js";
 export * as drive from "./drive/index.js";
 export * as general from "./general/index.js";
-export * as research from "./research/index.js";
-export * as sql from "./sql/index.js";
-export * from "./drive/client/requests/index.js";
 export * from "./research/client/requests/index.js";
+export * as research from "./research/index.js";
 export * from "./sql/client/requests/index.js";
+export * as sql from "./sql/index.js";

package/dist/cjs/api/resources/agents/resources/index.js

@@ -15,6 +15,9 @@ var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (
 }) : function(o, v) {
     o["default"] = v;
 });
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
 var __importStar = (this && this.__importStar) || (function () {
     var ownKeys = function(o) {
         ownKeys = Object.getOwnPropertyNames || function (o) {
@@ -32,15 +35,12 @@ var __importStar = (this && this.__importStar) || (function () {
         return result;
     };
 })();
-var __exportStar = (this && this.__exportStar) || function(m, exports) {
-    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
-};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.sql = exports.research = exports.general = exports.drive = void 0;
+__exportStar(require("./drive/client/requests/index.js"), exports);
 exports.drive = __importStar(require("./drive/index.js"));
 exports.general = __importStar(require("./general/index.js"));
-exports.research = __importStar(require("./research/index.js"));
-exports.sql = __importStar(require("./sql/index.js"));
-__exportStar(require("./drive/client/requests/index.js"), exports);
 __exportStar(require("./research/client/requests/index.js"), exports);
+exports.research = __importStar(require("./research/index.js"));
 __exportStar(require("./sql/client/requests/index.js"), exports);
+exports.sql = __importStar(require("./sql/index.js"));

package/dist/cjs/api/resources/agents/resources/research/client/Client.d.ts

@@ -1,20 +1,21 @@
 import type { BaseClientOptions, BaseRequestOptions } from "../../../../../../BaseClient.js";
+import { type NormalizedClientOptionsWithAuth } from "../../../../../../BaseClient.js";
 import * as core from "../../../../../../core/index.js";
 import * as AthenaIntelligence from "../../../../../index.js";
-export declare namespace Research {
+export declare namespace ResearchClient {
     interface Options extends BaseClientOptions {
     }
     interface RequestOptions extends BaseRequestOptions {
     }
 }
-export declare class Research {
-    protected readonly _options: Research.Options;
-    constructor(_options: Research.Options);
+export declare class ResearchClient {
+    protected readonly _options: NormalizedClientOptionsWithAuth<ResearchClient.Options>;
+    constructor(options: ResearchClient.Options);
     /**
      * Coming soon! Conduct research using web and other sources.
      *
      * @param {AthenaIntelligence.agents.ResearchAgentRequest} request
-     * @param {Research.RequestOptions} requestOptions - Request-specific configuration.
+     * @param {ResearchClient.RequestOptions} requestOptions - Request-specific configuration.
      *
      * @throws {@link AthenaIntelligence.UnprocessableEntityError}
      *
@@ -28,7 +29,6 @@ export declare class Research {
      *             }]
      *     })
      */
-    invoke(request: AthenaIntelligence.agents.ResearchAgentRequest, requestOptions?: Research.RequestOptions): core.HttpResponsePromise<AthenaIntelligence.ResearchAgentResponse>;
+    invoke(request: AthenaIntelligence.agents.ResearchAgentRequest, requestOptions?: ResearchClient.RequestOptions): core.HttpResponsePromise<AthenaIntelligence.ResearchAgentResponse>;
     private __invoke;
-    protected _getCustomAuthorizationHeaders(): Promise<Record<string, string | undefined>>;
 }

package/dist/cjs/api/resources/agents/resources/research/client/Client.js

@@ -43,21 +43,22 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
     });
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.Research = void 0;
-const environments = __importStar(require("../../../../../../environments.js"));
-const core = __importStar(require("../../../../../../core/index.js"));
-const AthenaIntelligence = __importStar(require("../../../../../index.js"));
+exports.ResearchClient = void 0;
+const BaseClient_js_1 = require("../../../../../../BaseClient.js");
 const headers_js_1 = require("../../../../../../core/headers.js");
+const core = __importStar(require("../../../../../../core/index.js"));
+const environments = __importStar(require("../../../../../../environments.js"));
 const errors = __importStar(require("../../../../../../errors/index.js"));
-class Research {
-    constructor(_options) {
-        this._options = _options;
+const AthenaIntelligence = __importStar(require("../../../../../index.js"));
+class ResearchClient {
+    constructor(options) {
+        this._options = (0, BaseClient_js_1.normalizeClientOptionsWithAuth)(options);
     }
     /**
      * Coming soon! Conduct research using web and other sources.
      *
      * @param {AthenaIntelligence.agents.ResearchAgentRequest} request
-     * @param {Research.RequestOptions} requestOptions - Request-specific configuration.
+     * @param {ResearchClient.RequestOptions} requestOptions - Request-specific configuration.
      *
      * @throws {@link AthenaIntelligence.UnprocessableEntityError}
      *
@@ -76,8 +77,9 @@ class Research {
     }
     __invoke(request, requestOptions) {
         return __awaiter(this, void 0, void 0, function* () {
-            var _a, _b, _c, _d, _e, _f, _g, _h, _j;
-            let _headers = (0, headers_js_1.mergeHeaders)((_a = this._options) === null || _a === void 0 ? void 0 : _a.headers, (0, headers_js_1.mergeOnlyDefinedHeaders)(Object.assign({}, (yield this._getCustomAuthorizationHeaders()))), requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.headers);
+            var _a, _b, _c, _d, _e, _f, _g, _h, _j, _k;
+            const _authRequest = yield this._options.authProvider.getAuthRequest();
+            const _headers = (0, headers_js_1.mergeHeaders)(_authRequest.headers, (_a = this._options) === null || _a === void 0 ? void 0 : _a.headers, requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.headers);
             const _response = yield ((_b = this._options.fetcher) !== null && _b !== void 0 ? _b : core.fetcher)({
                 url: core.url.join((_d = (_c = (yield core.Supplier.get(this._options.baseUrl))) !== null && _c !== void 0 ? _c : (yield core.Supplier.get(this._options.environment))) !== null && _d !== void 0 ? _d : environments.AthenaIntelligenceEnvironment.Production, "api/v0/agents/research/invoke"),
                 method: "POST",
@@ -89,6 +91,8 @@ class Research {
                 timeoutMs: ((_g = (_e = requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.timeoutInSeconds) !== null && _e !== void 0 ? _e : (_f = this._options) === null || _f === void 0 ? void 0 : _f.timeoutInSeconds) !== null && _g !== void 0 ? _g : 60) * 1000,
                 maxRetries: (_h = requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.maxRetries) !== null && _h !== void 0 ? _h : (_j = this._options) === null || _j === void 0 ? void 0 : _j.maxRetries,
                 abortSignal: requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.abortSignal,
+                fetchFn: (_k = this._options) === null || _k === void 0 ? void 0 : _k.fetch,
+                logging: this._options.logging,
             });
             if (_response.ok) {
                 return {
@@ -115,6 +119,11 @@ class Research {
                         body: _response.error.rawBody,
                         rawResponse: _response.rawResponse,
                     });
+                case "body-is-null":
+                    throw new errors.AthenaIntelligenceError({
+                        statusCode: _response.error.statusCode,
+                        rawResponse: _response.rawResponse,
+                    });
                 case "timeout":
                     throw new errors.AthenaIntelligenceTimeoutError("Timeout exceeded when calling POST /api/v0/agents/research/invoke.");
                 case "unknown":
@@ -125,11 +134,5 @@ class Research {
             }
         });
     }
-    _getCustomAuthorizationHeaders() {
-        return __awaiter(this, void 0, void 0, function* () {
-            const apiKeyValue = yield core.Supplier.get(this._options.apiKey);
-            return { "X-API-KEY": apiKeyValue };
-        });
-    }
 }
-exports.Research = Research;
+exports.ResearchClient = ResearchClient;