@chainfuse/ai-tools 0.3.0 → 0.3.1

package/dist/models.d.mts

@@ -1,5 +1,5 @@
-import type { ImageModelValues, LanguageModelValues, TextEmbeddingModelValues } from '@chainfuse/types';
-import { wrapLanguageModel, type embed, type embedMany, type experimental_generateImage as generateImage } from 'ai';
+import type { LanguageModelValues, TextEmbeddingModelValues } from '@chainfuse/types';
+import { wrapLanguageModel, type embed, type embedMany } from 'ai';
 import { AiBase } from './base.mjs';
 import { AiRegistry } from './registry.mjs';
 import type { AiRequestConfig } from './types.mjs';
@@ -8,22 +8,12 @@ type ValidProviders = keyof ProvidersReturnType;
 type ProviderLanguageModels = {
     [P in ValidProviders]: Parameters<ProvidersReturnType[P]['languageModel']>[0];
 };
-type ValidImageProviders = {
-    [P in ValidProviders]: ProvidersReturnType[P] extends {
-        imageModel: (...args: any[]) => any;
-    } ? P : never;
-}[ValidProviders];
-type ProviderImageModels = {
-    [P in Extract<ValidImageProviders, ValidProviders>]: Parameters<ProvidersReturnType[P]['imageModel']>[0];
-};
 type ProviderTextEmbeddingModels = {
     [P in ValidProviders]: Parameters<ProvidersReturnType[P]['textEmbeddingModel']>[0];
 };
 export declare class AiModel extends AiBase {
     wrappedLanguageModel<P extends ValidProviders>(args: AiRequestConfig, provider: P, model: ProviderLanguageModels[P]): Promise<ReturnType<typeof wrapLanguageModel>>;
     wrappedLanguageModel(args: AiRequestConfig, model: LanguageModelValues): Promise<ReturnType<typeof wrapLanguageModel>>;
-    wrappedImageModel<P extends ValidImageProviders>(args: AiRequestConfig, provider: P, model: ProviderImageModels[P]): Promise<Parameters<typeof generateImage>[0]['model']>;
-    wrappedImageModel(args: AiRequestConfig, model: ImageModelValues): Promise<Parameters<typeof generateImage>[0]['model']>;
     wrappedTextEmbeddingModel<P extends ValidProviders>(args: AiRequestConfig, provider: P, model: ProviderTextEmbeddingModels[P]): Promise<Parameters<typeof embed | typeof embedMany>[0]['model']>;
     wrappedTextEmbeddingModel(args: AiRequestConfig, model: TextEmbeddingModelValues): Promise<Parameters<typeof embed | typeof embedMany>[0]['model']>;
     private get middleware();

package/dist/models.mjs

@@ -8,9 +8,6 @@ export class AiModel extends AiBase {
             middleware: this.middleware,
         }));
     }
-    wrappedImageModel(args, modelOrProvider, model) {
-        return new AiRegistry(this.config).registry(args).then((registry) => registry.imageModel(model ? `${modelOrProvider}:${model}` : modelOrProvider));
-    }
     wrappedTextEmbeddingModel(args, modelOrProvider, model) {
         return new AiRegistry(this.config).registry(args).then((registry) => registry.textEmbeddingModel(model ? `${modelOrProvider}:${model}` : modelOrProvider));
     }

package/dist/providers/customProviders.mjs

@@ -66,13 +66,15 @@ export class AiCustomProviders extends AiBase {
                 });
                 return acc;
             }, Promise.resolve({})),
-            // @ts-expect-error override for types
-            imageModels: await server.imageModelAvailability.reduce(async (accPromise, model) => {
-                const acc = await accPromise;
-                // @ts-expect-error override for types
-                acc[model] = (await raw.azOpenai(args, server)).imageModel(model);
-                return acc;
-            }, Promise.resolve({})),
+            // imageModels: await server!.imageModelAvailability.reduce(
+            // 	async (accPromise, model) => {
+            // 		const acc = await accPromise;
+            // 		// @ts-expect-error override for types
+            // 		acc[model as AzureImageModels] = (await raw.azOpenai(args, server!)).imageModel(model);
+            // 		return acc;
+            // 	},
+            // 	Promise.resolve({} as Record<AzureImageModels, Awaited<ReturnType<AiRawProviders['azOpenai']>>>),
+            // ),
             // @ts-expect-error override for types
             textEmbeddingModels: await server.textEmbeddingModelAvailability.reduce(async (accPromise, model) => {
                 const acc = await accPromise;

package/dist/providers/types.d.mts

@@ -1,7 +1,6 @@
-import type { OpenAIProvider } from '@ai-sdk/openai';
 import type { OpenAIChatSettings, OpenAIEmbeddingSettings } from '@ai-sdk/openai/internal';
-import type { EmbeddingModelV1, ImageModelV1, LanguageModelV1 } from '@ai-sdk/provider';
-import type { AzureChatModels, AzureEmbeddingModels, AzureImageModels } from '@chainfuse/types';
+import type { EmbeddingModelV1, LanguageModelV1 } from '@ai-sdk/provider';
+import type { AzureChatModels, AzureEmbeddingModels } from '@chainfuse/types';
 import type { Provider } from 'ai';
 export interface AzureOpenAIProvider extends Provider {
     (deploymentId: AzureChatModels, settings?: OpenAIChatSettings): LanguageModelV1;
@@ -13,8 +12,4 @@ export interface AzureOpenAIProvider extends Provider {
      * Creates an Azure OpenAI model for text embeddings.
      */
     textEmbeddingModel(deploymentId: AzureEmbeddingModels, settings?: OpenAIEmbeddingSettings): EmbeddingModelV1<string>;
-    /**
-     * Creates a model for image generation.
-     */
-    imageModel(modelId: AzureImageModels, settings?: Parameters<OpenAIProvider['imageModel']>[1]): ImageModelV1;
 }

package/dist/registry.mjs

@@ -13,8 +13,6 @@ export class AiRegistry extends AiBase {
         });
     }
     async registry(args) {
-        return createProviderRegistry(
-        // @ts-expect-error override for types
-        await this.providers(args));
+        return createProviderRegistry(await this.providers(args));
     }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@chainfuse/ai-tools",
-	"version": "0.3.0",
+	"version": "0.3.1",
 	"description": "",
 	"author": "ChainFuse",
 	"homepage": "https://github.com/ChainFuse/packages/tree/main/packages/ai-tools#readme",
@@ -64,5 +64,5 @@
 		"@cloudflare/workers-types": "^4.20250129.0",
 		"openai": "^4.82.0"
 	},
-	"gitHead": "5c8d059af28f814e48549d079a7b7f01dc3ad1db"
+	"gitHead": "5798b5c3f0920b026ba9e4fc0f3d3eac8ea3fe7f"
 }