@athenaintel/sdk 4.3.308 → 4.3.310

package/dist/esm/api/resources/agents/resources/general/client/Client.mjs

@@ -8,20 +8,21 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
         step((generator = generator.apply(thisArg, _arguments || [])).next());
     });
 };
-import * as environments from "../../../../../../environments.mjs";
+import { normalizeClientOptionsWithAuth } from "../../../../../../BaseClient.mjs";
+import { mergeHeaders } from "../../../../../../core/headers.mjs";
 import * as core from "../../../../../../core/index.mjs";
-import * as AthenaIntelligence from "../../../../../index.mjs";
-import { mergeHeaders, mergeOnlyDefinedHeaders } from "../../../../../../core/headers.mjs";
+import * as environments from "../../../../../../environments.mjs";
 import * as errors from "../../../../../../errors/index.mjs";
-export class General {
-    constructor(_options) {
-        this._options = _options;
+import * as AthenaIntelligence from "../../../../../index.mjs";
+export class GeneralClient {
+    constructor(options) {
+        this._options = 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}
      *
@@ -41,8 +42,9 @@ export 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 = mergeHeaders((_a = this._options) === null || _a === void 0 ? void 0 : _a.headers, 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 = 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",
@@ -54,6 +56,8 @@ export 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 {
@@ -80,6 +84,11 @@ export 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":
@@ -97,7 +106,7 @@ export 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}
      *
@@ -117,8 +126,9 @@ export 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 = mergeHeaders((_a = this._options) === null || _a === void 0 ? void 0 : _a.headers, 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 = 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",
@@ -130,6 +140,8 @@ export 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 {
@@ -156,6 +168,11 @@ export 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":
@@ -170,7 +187,7 @@ export 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}
      *
@@ -190,8 +207,9 @@ export 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 = mergeHeaders((_a = this._options) === null || _a === void 0 ? void 0 : _a.headers, 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 = 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",
@@ -203,6 +221,8 @@ export 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 {
@@ -229,6 +249,11 @@ export 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":
@@ -239,10 +264,4 @@ export 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 };
-        });
-    }
 }

package/dist/esm/api/resources/agents/resources/index.d.mts

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

package/dist/esm/api/resources/agents/resources/index.mjs

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

package/dist/esm/api/resources/agents/resources/research/client/Client.d.mts

@@ -1,20 +1,21 @@
 import type { BaseClientOptions, BaseRequestOptions } from "../../../../../../BaseClient.mjs";
+import { type NormalizedClientOptionsWithAuth } from "../../../../../../BaseClient.mjs";
 import * as core from "../../../../../../core/index.mjs";
 import * as AthenaIntelligence from "../../../../../index.mjs";
-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/esm/api/resources/agents/resources/research/client/Client.mjs

@@ -8,20 +8,21 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
         step((generator = generator.apply(thisArg, _arguments || [])).next());
     });
 };
-import * as environments from "../../../../../../environments.mjs";
+import { normalizeClientOptionsWithAuth } from "../../../../../../BaseClient.mjs";
+import { mergeHeaders } from "../../../../../../core/headers.mjs";
 import * as core from "../../../../../../core/index.mjs";
-import * as AthenaIntelligence from "../../../../../index.mjs";
-import { mergeHeaders, mergeOnlyDefinedHeaders } from "../../../../../../core/headers.mjs";
+import * as environments from "../../../../../../environments.mjs";
 import * as errors from "../../../../../../errors/index.mjs";
-export class Research {
-    constructor(_options) {
-        this._options = _options;
+import * as AthenaIntelligence from "../../../../../index.mjs";
+export class ResearchClient {
+    constructor(options) {
+        this._options = 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}
      *
@@ -40,8 +41,9 @@ export 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 = mergeHeaders((_a = this._options) === null || _a === void 0 ? void 0 : _a.headers, 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 = 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",
@@ -53,6 +55,8 @@ export 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 {
@@ -79,6 +83,11 @@ export 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":
@@ -89,10 +98,4 @@ export 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 };
-        });
-    }
 }

package/dist/esm/api/resources/agents/resources/research/client/index.d.mts

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

package/dist/esm/api/resources/agents/resources/research/client/index.mjs

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

package/dist/esm/api/resources/agents/resources/research/client/requests/index.d.mts

@@ -1 +1 @@
-export { type ResearchAgentRequest } from "./ResearchAgentRequest.mjs";
+export type { ResearchAgentRequest } from "./ResearchAgentRequest.mjs";

package/dist/esm/api/resources/agents/resources/research/client/requests/index.mjs

@@ -1 +1 @@
-export {} from "./ResearchAgentRequest.mjs";
+export {};

package/dist/esm/api/resources/agents/resources/sql/client/Client.d.mts

@@ -1,20 +1,21 @@
 import type { BaseClientOptions, BaseRequestOptions } from "../../../../../../BaseClient.mjs";
+import { type NormalizedClientOptionsWithAuth } from "../../../../../../BaseClient.mjs";
 import * as core from "../../../../../../core/index.mjs";
 import * as AthenaIntelligence from "../../../../../index.mjs";
-export declare namespace Sql {
+export declare namespace SqlClient {
     interface Options extends BaseClientOptions {
     }
     interface RequestOptions extends BaseRequestOptions {
     }
 }
-export declare class Sql {
-    protected readonly _options: Sql.Options;
-    constructor(_options: Sql.Options);
+export declare class SqlClient {
+    protected readonly _options: NormalizedClientOptionsWithAuth<SqlClient.Options>;
+    constructor(options: SqlClient.Options);
     /**
      * Coming soon! Generate, execute, and test SQL queries. Returns an asset ID for the query object.
      *
      * @param {AthenaIntelligence.agents.SqlAgentRequest} request
-     * @param {Sql.RequestOptions} requestOptions - Request-specific configuration.
+     * @param {SqlClient.RequestOptions} requestOptions - Request-specific configuration.
      *
      * @throws {@link AthenaIntelligence.UnprocessableEntityError}
      *
@@ -28,7 +29,6 @@ export declare class Sql {
      *             }]
      *     })
      */
-    invoke(request: AthenaIntelligence.agents.SqlAgentRequest, requestOptions?: Sql.RequestOptions): core.HttpResponsePromise<AthenaIntelligence.SqlAgentResponse>;
+    invoke(request: AthenaIntelligence.agents.SqlAgentRequest, requestOptions?: SqlClient.RequestOptions): core.HttpResponsePromise<AthenaIntelligence.SqlAgentResponse>;
     private __invoke;
-    protected _getCustomAuthorizationHeaders(): Promise<Record<string, string | undefined>>;
 }

package/dist/esm/api/resources/agents/resources/sql/client/Client.mjs

@@ -8,20 +8,21 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
         step((generator = generator.apply(thisArg, _arguments || [])).next());
     });
 };
-import * as environments from "../../../../../../environments.mjs";
+import { normalizeClientOptionsWithAuth } from "../../../../../../BaseClient.mjs";
+import { mergeHeaders } from "../../../../../../core/headers.mjs";
 import * as core from "../../../../../../core/index.mjs";
-import * as AthenaIntelligence from "../../../../../index.mjs";
-import { mergeHeaders, mergeOnlyDefinedHeaders } from "../../../../../../core/headers.mjs";
+import * as environments from "../../../../../../environments.mjs";
 import * as errors from "../../../../../../errors/index.mjs";
-export class Sql {
-    constructor(_options) {
-        this._options = _options;
+import * as AthenaIntelligence from "../../../../../index.mjs";
+export class SqlClient {
+    constructor(options) {
+        this._options = normalizeClientOptionsWithAuth(options);
     }
     /**
      * Coming soon! Generate, execute, and test SQL queries. Returns an asset ID for the query object.
      *
      * @param {AthenaIntelligence.agents.SqlAgentRequest} request
-     * @param {Sql.RequestOptions} requestOptions - Request-specific configuration.
+     * @param {SqlClient.RequestOptions} requestOptions - Request-specific configuration.
      *
      * @throws {@link AthenaIntelligence.UnprocessableEntityError}
      *
@@ -40,8 +41,9 @@ export class Sql {
     }
     __invoke(request, requestOptions) {
         return __awaiter(this, void 0, void 0, function* () {
-            var _a, _b, _c, _d, _e, _f, _g, _h, _j;
-            let _headers = mergeHeaders((_a = this._options) === null || _a === void 0 ? void 0 : _a.headers, 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 = 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/sql/invoke"),
                 method: "POST",
@@ -53,6 +55,8 @@ export class Sql {
                 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 { data: _response.body, rawResponse: _response.rawResponse };
@@ -76,6 +80,11 @@ export class Sql {
                         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/sql/invoke.");
                 case "unknown":
@@ -86,10 +95,4 @@ export class Sql {
             }
         });
     }
-    _getCustomAuthorizationHeaders() {
-        return __awaiter(this, void 0, void 0, function* () {
-            const apiKeyValue = yield core.Supplier.get(this._options.apiKey);
-            return { "X-API-KEY": apiKeyValue };
-        });
-    }
 }

package/dist/esm/api/resources/agents/resources/sql/client/index.d.mts

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

package/dist/esm/api/resources/agents/resources/sql/client/index.mjs

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

package/dist/esm/api/resources/agents/resources/sql/client/requests/index.d.mts

@@ -1 +1 @@
-export { type SqlAgentRequest } from "./SqlAgentRequest.mjs";
+export type { SqlAgentRequest } from "./SqlAgentRequest.mjs";

package/dist/esm/api/resources/agents/resources/sql/client/requests/index.mjs

@@ -1 +1 @@
-export {} from "./SqlAgentRequest.mjs";
+export {};

package/dist/esm/api/resources/aop/client/Client.d.mts

@@ -1,20 +1,21 @@
 import type { BaseClientOptions, BaseRequestOptions } from "../../../../BaseClient.mjs";
+import { type NormalizedClientOptionsWithAuth } from "../../../../BaseClient.mjs";
 import * as core from "../../../../core/index.mjs";
 import * as AthenaIntelligence from "../../../index.mjs";
-export declare namespace Aop {
+export declare namespace AopClient {
     interface Options extends BaseClientOptions {
     }
     interface RequestOptions extends BaseRequestOptions {
     }
 }
-export declare class Aop {
-    protected readonly _options: Aop.Options;
-    constructor(_options: Aop.Options);
+export declare class AopClient {
+    protected readonly _options: NormalizedClientOptionsWithAuth<AopClient.Options>;
+    constructor(options: AopClient.Options);
     /**
      * DEPRECATED: This endpoint is deprecated. Please use /aop/execute-async instead for better performance and reliability. Execute an existing Agent Operating Procedure (AOP) asset with optional user inputs. AOPs are pre-configured AI workflows that can perform complex tasks like research, analysis, and content generation.
      *
      * @param {AthenaIntelligence.AopExecuteRequestIn} request
-     * @param {Aop.RequestOptions} requestOptions - Request-specific configuration.
+     * @param {AopClient.RequestOptions} requestOptions - Request-specific configuration.
      *
      * @throws {@link AthenaIntelligence.BadRequestError}
      * @throws {@link AthenaIntelligence.UnauthorizedError}
@@ -31,13 +32,13 @@ export declare class Aop {
      *         }
      *     })
      */
-    execute(request: AthenaIntelligence.AopExecuteRequestIn, requestOptions?: Aop.RequestOptions): core.HttpResponsePromise<AthenaIntelligence.AopExecuteResponseOut>;
+    execute(request: AthenaIntelligence.AopExecuteRequestIn, requestOptions?: AopClient.RequestOptions): core.HttpResponsePromise<AthenaIntelligence.AopExecuteResponseOut>;
     private __execute;
     /**
      * Start execution of an Agent Operating Procedure (AOP) asset asynchronously. Returns immediately with a thread_id for tracking execution progress without waiting for completion.
      *
      * @param {AthenaIntelligence.AopExecuteRequestIn} request
-     * @param {Aop.RequestOptions} requestOptions - Request-specific configuration.
+     * @param {AopClient.RequestOptions} requestOptions - Request-specific configuration.
      *
      * @throws {@link AthenaIntelligence.BadRequestError}
      * @throws {@link AthenaIntelligence.UnauthorizedError}
@@ -54,7 +55,6 @@ export declare class Aop {
      *         }
      *     })
      */
-    executeAsync(request: AthenaIntelligence.AopExecuteRequestIn, requestOptions?: Aop.RequestOptions): core.HttpResponsePromise<AthenaIntelligence.AopAsyncExecuteResponseOut>;
+    executeAsync(request: AthenaIntelligence.AopExecuteRequestIn, requestOptions?: AopClient.RequestOptions): core.HttpResponsePromise<AthenaIntelligence.AopAsyncExecuteResponseOut>;
     private __executeAsync;
-    protected _getCustomAuthorizationHeaders(): Promise<Record<string, string | undefined>>;
 }

package/dist/esm/api/resources/aop/client/Client.mjs

@@ -8,20 +8,21 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
         step((generator = generator.apply(thisArg, _arguments || [])).next());
     });
 };
-import * as environments from "../../../../environments.mjs";
+import { normalizeClientOptionsWithAuth } from "../../../../BaseClient.mjs";
+import { mergeHeaders } from "../../../../core/headers.mjs";
 import * as core from "../../../../core/index.mjs";
-import * as AthenaIntelligence from "../../../index.mjs";
-import { mergeHeaders, mergeOnlyDefinedHeaders } from "../../../../core/headers.mjs";
+import * as environments from "../../../../environments.mjs";
 import * as errors from "../../../../errors/index.mjs";
-export class Aop {
-    constructor(_options) {
-        this._options = _options;
+import * as AthenaIntelligence from "../../../index.mjs";
+export class AopClient {
+    constructor(options) {
+        this._options = normalizeClientOptionsWithAuth(options);
     }
     /**
      * DEPRECATED: This endpoint is deprecated. Please use /aop/execute-async instead for better performance and reliability. Execute an existing Agent Operating Procedure (AOP) asset with optional user inputs. AOPs are pre-configured AI workflows that can perform complex tasks like research, analysis, and content generation.
      *
      * @param {AthenaIntelligence.AopExecuteRequestIn} request
-     * @param {Aop.RequestOptions} requestOptions - Request-specific configuration.
+     * @param {AopClient.RequestOptions} requestOptions - Request-specific configuration.
      *
      * @throws {@link AthenaIntelligence.BadRequestError}
      * @throws {@link AthenaIntelligence.UnauthorizedError}
@@ -43,8 +44,9 @@ export class Aop {
     }
     __execute(request, requestOptions) {
         return __awaiter(this, void 0, void 0, function* () {
-            var _a, _b, _c, _d, _e, _f, _g, _h, _j;
-            let _headers = mergeHeaders((_a = this._options) === null || _a === void 0 ? void 0 : _a.headers, 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 = 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/aop/execute"),
                 method: "POST",
@@ -56,6 +58,8 @@ export class Aop {
                 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 {
@@ -90,6 +94,11 @@ export class Aop {
                         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/aop/execute.");
                 case "unknown":
@@ -104,7 +113,7 @@ export class Aop {
      * Start execution of an Agent Operating Procedure (AOP) asset asynchronously. Returns immediately with a thread_id for tracking execution progress without waiting for completion.
      *
      * @param {AthenaIntelligence.AopExecuteRequestIn} request
-     * @param {Aop.RequestOptions} requestOptions - Request-specific configuration.
+     * @param {AopClient.RequestOptions} requestOptions - Request-specific configuration.
      *
      * @throws {@link AthenaIntelligence.BadRequestError}
      * @throws {@link AthenaIntelligence.UnauthorizedError}
@@ -126,8 +135,9 @@ export class Aop {
     }
     __executeAsync(request, requestOptions) {
         return __awaiter(this, void 0, void 0, function* () {
-            var _a, _b, _c, _d, _e, _f, _g, _h, _j;
-            let _headers = mergeHeaders((_a = this._options) === null || _a === void 0 ? void 0 : _a.headers, 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 = 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/aop/execute-async"),
                 method: "POST",
@@ -139,6 +149,8 @@ export class Aop {
                 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 {
@@ -173,6 +185,11 @@ export class Aop {
                         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/aop/execute-async.");
                 case "unknown":
@@ -183,10 +200,4 @@ export class Aop {
             }
         });
     }
-    _getCustomAuthorizationHeaders() {
-        return __awaiter(this, void 0, void 0, function* () {
-            const apiKeyValue = yield core.Supplier.get(this._options.apiKey);
-            return { "X-API-KEY": apiKeyValue };
-        });
-    }
 }