@chainfuse/ai-tools 1.0.16 → 1.0.17

package/dist/base.mjs

@@ -8,6 +8,14 @@ export class AiBase {
     get config() {
         return {
             ...Object.freeze(this._config),
+            gateway: {
+                ...Object.freeze(this._config.gateway),
+                binding: this._config.gateway.binding,
+            },
+            providers: {
+                ...Object.freeze(this._config.providers),
+                workersAi: this._config.providers.workersAi,
+            },
             backgroundContext: this._config.backgroundContext,
         };
     }

package/dist/providers/rawProviders.mjs

@@ -24,8 +24,8 @@ export class AiRawProviders extends AiBase {
         };
         if (logId) {
             const updateMetadata = (() => {
-                if ('gateway' in this.config.providers.workersAi && typeof this.config.providers.workersAi.gateway === 'function') {
-                    return this.config.providers.workersAi
+                if ('binding' in this.config.gateway && typeof this.config.gateway.binding?.gateway === 'function') {
+                    return this.config.gateway.binding
                         .gateway(this.gatewayName)
                         .patchLog(logId, {
                         metadata: {
@@ -220,8 +220,8 @@ export class AiRawProviders extends AiBase {
                 if (args.logging ?? this.gatewayLog)
                     console.info(new Date().toISOString(), this.chalk.rgb(...Helpers.uniqueIdColor(metadataHeader.idempotencyId))(`[${metadataHeader.idempotencyId}]`), this.chalk.magenta(rawInit?.method), this.chalk.magenta(new URL(new Request(input).url).pathname));
                 return (() => {
-                    if ('gateway' in this.config.providers.workersAi && typeof this.config.providers.workersAi.gateway === 'function') {
-                        return this.config.providers.workersAi.gateway(this.gatewayName).run(fallbackedBody, {
+                    if ('binding' in this.config.gateway && typeof this.config.gateway.binding?.gateway === 'function') {
+                        return this.config.gateway.binding.gateway(this.gatewayName).run(fallbackedBody, {
                             extraHeaders: (() => {
                                 // Prevent duplicates
                                 headers.delete('cf-aig-authorization');

package/dist/types.d.mts

@@ -1,12 +1,13 @@
 import type { Coordinate } from '@chainfuse/types/ai-tools';
 import type { azureCatalog } from '@chainfuse/types/ai-tools/azure/catalog';
 import type { PrefixedUuid, UuidExport } from '@chainfuse/types/d1';
-import type { Ai, ExecutionContext, IncomingRequestCfProperties } from '@cloudflare/workers-types/experimental';
+import type { Ai, DurableObjectState, ExecutionContext, IncomingRequestCfProperties } from '@cloudflare/workers-types/experimental';
 import type haversine from 'haversine-distance';
-export interface AiConfig {
+export interface AiConfig<C extends Pick<ExecutionContext | DurableObjectState, 'waitUntil'> = Pick<ExecutionContext | DurableObjectState, 'waitUntil'>> {
     gateway: {
         accountId: string;
         apiToken: string;
+        binding?: AiConfigWorkersAiBinding;
     };
     geoRouting?: {
         userCoordinate?: Coordinate;
@@ -26,7 +27,7 @@ export interface AiConfig {
         action: 'active';
     }));
     providers: AiConfigProviders;
-    backgroundContext?: ExecutionContext;
+    backgroundContext?: C;
 }
 export interface AiConfigProviders {
     anthropic: AiConfigAnthropic;
@@ -62,11 +63,11 @@ export interface AiConfigOaiOpenai {
     apiToken: `sk-${string}`;
     organization: `org-${string}`;
 }
-export type AiConfigWorkersai = AiConfigWorkersaiRest | AiConfigWorkersaiBinding;
+export type AiConfigWorkersai = AiConfigWorkersaiRest | AiConfigWorkersAiBinding;
 export interface AiConfigWorkersaiRest {
     apiToken: string;
 }
-export type AiConfigWorkersaiBinding<T extends Ai = Ai> = T;
+export type AiConfigWorkersAiBinding<T extends Ai = Ai> = T;
 export type AzureServers = typeof azureCatalog;
 export type Servers = AzureServers;
 export type PrivacyRegion = Extract<Servers[number], {

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@chainfuse/ai-tools",
-	"version": "1.0.16",
+	"version": "1.0.17",
 	"description": "",
 	"author": "ChainFuse",
 	"homepage": "https://github.com/ChainFuse/packages/tree/main/packages/ai-tools#readme",
@@ -66,5 +66,5 @@
 		"@cloudflare/workers-types": "^4.20251127.0",
 		"openai": "^6.9.1"
 	},
-	"gitHead": "a7ba85aa01dd1c54356ef54b1af743cd9f6a767e"
+	"gitHead": "018281c52a87f850cd721399db3785b65c1cf39f"
 }