@chainfuse/types 1.7.2 → 2.0.0

package/dist/ai-tools/index.d.ts

@@ -1,18 +1,10 @@
-import { type cloudflareFilteredModelPossibilities, type cloudflareModelPossibilities, type cloudflareModelTypes } from '../super-ai/index.js';
-export interface RawCoordinate {
+import { type CloudflareFunctionModelsEnum, type CloudflareModelsEnum } from './workers-ai/index.js';
+export * from './azure/index.js';
+export * from './workers-ai/index.js';
+export interface Coordinate {
     lat: string;
     lon: string;
 }
-export declare const enabledCloudflareLlmImageProviders: cloudflareModelPossibilities<'Text-to-Image'>[];
-export type CloudflareModelsEnum<M extends cloudflareModelTypes = cloudflareModelTypes> = {
-    [K in cloudflareModelPossibilities<M>]: `workersai:${K}`;
-};
-export type CloudflareFunctionModelsEnum = {
-    [K in cloudflareFilteredModelPossibilities<'Text Generation', 'function_calling', true>]: `workersai:${K}`;
-};
-export type AzureChatModels = 'gpt-35-turbo' | 'gpt-4-turbo' | 'gpt-4o' | 'gpt-4o-mini' | 'o1' | 'o3-mini';
-export type AzureImageModels = 'dall-e-3';
-export type AzureEmbeddingModels = 'text-embedding-3-small' | 'text-embedding-3-large';
 export declare namespace AiModels {
     namespace LanguageModels {
         enum Azure {

package/dist/ai-tools/index.js

@@ -1,6 +1,6 @@
-import { enabledCloudflareLlmEmbeddingProviders, enabledCloudflareLlmFunctionProviders, enabledCloudflareLlmProviders } from '../super-ai/index.js';
-import { workersAiCatalog } from './workers-ai-catalog.js';
-export const enabledCloudflareLlmImageProviders = workersAiCatalog.modelGroups['Text-to-Image'].models.map((model) => model.name);
+import { enabledCloudflareLlmEmbeddingProviders, enabledCloudflareLlmFunctionProviders, enabledCloudflareLlmImageProviders, enabledCloudflareLlmProviders } from './workers-ai/index.js';
+export * from './azure/index.js';
+export * from './workers-ai/index.js';
 export var AiModels;
 (function (AiModels) {
     let LanguageModels;

package/dist/ai-tools/{workers-ai-catalog.d.ts → workers-ai/catalog.d.ts}

@@ -271,6 +271,7 @@ export declare const workersAiCatalog: {
                 readonly tags: readonly [];
                 readonly properties: {
                     readonly context_window: 24000;
+                    readonly function_calling: true;
                     readonly terms: "https://github.com/meta-llama/llama-models/blob/main/models/llama3_3/LICENSE";
                 };
             }, {
@@ -471,6 +472,54 @@ export declare const workersAiCatalog: {
                 };
             }];
         };
+        readonly 'Text Embeddings': {
+            readonly id: "0137cdcf-162a-4108-94f2-1ca59e8c65ee";
+            readonly description: "Feature extraction models transform raw data into numerical features that can be processed while preserving the information in the original dataset. These models are ideal as part of building vector search applications or Retrieval Augmented Generation workflows with Large Language Models (LLM).";
+            readonly models: readonly [{
+                readonly id: "eed32bc1-8775-4985-89ce-dd1405508ad8";
+                readonly source: 1;
+                readonly name: "@cf/baai/bge-m3";
+                readonly description: "Multi-Functionality, Multi-Linguality, and Multi-Granularity embeddings model.";
+                readonly tags: readonly [];
+                readonly properties: {};
+            }, {
+                readonly id: "57fbd08a-a4c4-411c-910d-b9459ff36c20";
+                readonly source: 1;
+                readonly name: "@cf/baai/bge-small-en-v1.5";
+                readonly description: "BAAI general embedding (Small) model that transforms any given text into a 384-dimensional vector";
+                readonly tags: readonly [];
+                readonly properties: {
+                    readonly beta: true;
+                    readonly info: "https://huggingface.co/BAAI/bge-base-en-v1.5";
+                    readonly max_input_tokens: 512;
+                    readonly output_dimensions: 384;
+                };
+            }, {
+                readonly id: "429b9e8b-d99e-44de-91ad-706cf8183658";
+                readonly source: 1;
+                readonly name: "@cf/baai/bge-base-en-v1.5";
+                readonly description: "BAAI general embedding (Base) model that transforms any given text into a 768-dimensional vector";
+                readonly tags: readonly [];
+                readonly properties: {
+                    readonly beta: true;
+                    readonly info: "https://huggingface.co/BAAI/bge-base-en-v1.5";
+                    readonly max_input_tokens: 512;
+                    readonly output_dimensions: 768;
+                };
+            }, {
+                readonly id: "01bc2fb0-4bca-4598-b985-d2584a3f46c0";
+                readonly source: 1;
+                readonly name: "@cf/baai/bge-large-en-v1.5";
+                readonly description: "BAAI general embedding (Large) model that transforms any given text into a 1024-dimensional vector";
+                readonly tags: readonly [];
+                readonly properties: {
+                    readonly beta: true;
+                    readonly info: "https://huggingface.co/BAAI/bge-base-en-v1.5";
+                    readonly max_input_tokens: 512;
+                    readonly output_dimensions: 1024;
+                };
+            }];
+        };
         readonly 'Text Classification': {
             readonly id: "19606750-23ed-4371-aab2-c20349b53a60";
             readonly description: "Sentiment analysis or text classification is a common NLP task that classifies a text input into labels or classes.";
@@ -484,6 +533,13 @@ export declare const workersAiCatalog: {
                     readonly beta: true;
                     readonly info: "https://huggingface.co/Intel/distilbert-base-uncased-finetuned-sst-2-english-int8-static";
                 };
+            }, {
+                readonly id: "145337e7-cec3-4ebb-8e78-16ddfc75e580";
+                readonly source: 1;
+                readonly name: "@cf/baai/bge-reranker-base";
+                readonly description: "Different from embedding model, reranker uses question and document as input and directly output similarity instead of embedding. You can get a relevance score by inputting query and passage to the reranker. And the score can be mapped to a float value in [0,1] by sigmoid function.\n\n";
+                readonly tags: readonly [];
+                readonly properties: {};
             }];
         };
         readonly 'Object Detection': {
@@ -500,6 +556,18 @@ export declare const workersAiCatalog: {
                 };
             }];
         };
+        readonly 'Text-to-Speech': {
+            readonly id: "b52660a1-9a95-4ab2-8b1d-f232be34604a";
+            readonly description: "Text-to-Speech (TTS) is the task of generating natural sounding speech given text input. TTS models can be extended to have a single model that generates speech for multiple speakers and multiple languages.";
+            readonly models: readonly [{
+                readonly id: "c837b2ac-4d9b-4d37-8811-34de60f0c44f";
+                readonly source: 1;
+                readonly name: "@cf/myshell-ai/melotts";
+                readonly description: "MeloTTS is a high-quality multi-lingual text-to-speech library by MyShell.ai.";
+                readonly tags: readonly [];
+                readonly properties: {};
+            }];
+        };
         readonly 'Automatic Speech Recognition': {
             readonly id: "dfce1c48-2a81-462e-a7fd-de97ce985207";
             readonly description: "Automatic speech recognition (ASR) models convert a speech signal, typically an audio input, to text.";
@@ -528,9 +596,7 @@ export declare const workersAiCatalog: {
                 readonly name: "@cf/openai/whisper-large-v3-turbo";
                 readonly description: "Whisper is a pre-trained model for automatic speech recognition (ASR) and speech translation. ";
                 readonly tags: readonly [];
-                readonly properties: {
-                    readonly beta: true;
-                };
+                readonly properties: {};
             }];
         };
         readonly 'Image-to-Text': {
@@ -654,47 +720,6 @@ export declare const workersAiCatalog: {
                 };
             }];
         };
-        readonly 'Text Embeddings': {
-            readonly id: "0137cdcf-162a-4108-94f2-1ca59e8c65ee";
-            readonly description: "Feature extraction models transform raw data into numerical features that can be processed while preserving the information in the original dataset. These models are ideal as part of building vector search applications or Retrieval Augmented Generation workflows with Large Language Models (LLM).";
-            readonly models: readonly [{
-                readonly id: "57fbd08a-a4c4-411c-910d-b9459ff36c20";
-                readonly source: 1;
-                readonly name: "@cf/baai/bge-small-en-v1.5";
-                readonly description: "BAAI general embedding (Small) model that transforms any given text into a 384-dimensional vector";
-                readonly tags: readonly [];
-                readonly properties: {
-                    readonly beta: true;
-                    readonly info: "https://huggingface.co/BAAI/bge-base-en-v1.5";
-                    readonly max_input_tokens: 512;
-                    readonly output_dimensions: 384;
-                };
-            }, {
-                readonly id: "429b9e8b-d99e-44de-91ad-706cf8183658";
-                readonly source: 1;
-                readonly name: "@cf/baai/bge-base-en-v1.5";
-                readonly description: "BAAI general embedding (Base) model that transforms any given text into a 768-dimensional vector";
-                readonly tags: readonly [];
-                readonly properties: {
-                    readonly beta: true;
-                    readonly info: "https://huggingface.co/BAAI/bge-base-en-v1.5";
-                    readonly max_input_tokens: 512;
-                    readonly output_dimensions: 768;
-                };
-            }, {
-                readonly id: "01bc2fb0-4bca-4598-b985-d2584a3f46c0";
-                readonly source: 1;
-                readonly name: "@cf/baai/bge-large-en-v1.5";
-                readonly description: "BAAI general embedding (Large) model that transforms any given text into a 1024-dimensional vector";
-                readonly tags: readonly [];
-                readonly properties: {
-                    readonly beta: true;
-                    readonly info: "https://huggingface.co/BAAI/bge-base-en-v1.5";
-                    readonly max_input_tokens: 512;
-                    readonly output_dimensions: 1024;
-                };
-            }];
-        };
         readonly Summarization: {
             readonly id: "6f4e65d8-da0f-40d2-9aa4-db582a5a04fd";
             readonly description: "Summarization is the task of producing a shorter version of a document while preserving its important information. Some models can extract text from the original input, while other models can generate entirely new text.";

package/dist/ai-tools/{workers-ai-catalog.js → workers-ai/catalog.js}

@@ -296,6 +296,7 @@ export const workersAiCatalog = {
                     tags: [],
                     properties: {
                         context_window: 24000,
+                        function_calling: true,
                         terms: 'https://github.com/meta-llama/llama-models/blob/main/models/llama3_3/LICENSE',
                     },
                 },
@@ -515,6 +516,59 @@ export const workersAiCatalog = {
                 },
             ],
         },
+        'Text Embeddings': {
+            id: '0137cdcf-162a-4108-94f2-1ca59e8c65ee',
+            description: 'Feature extraction models transform raw data into numerical features that can be processed while preserving the information in the original dataset. These models are ideal as part of building vector search applications or Retrieval Augmented Generation workflows with Large Language Models (LLM).',
+            models: [
+                {
+                    id: 'eed32bc1-8775-4985-89ce-dd1405508ad8',
+                    source: 1,
+                    name: '@cf/baai/bge-m3',
+                    description: 'Multi-Functionality, Multi-Linguality, and Multi-Granularity embeddings model.',
+                    tags: [],
+                    properties: {},
+                },
+                {
+                    id: '57fbd08a-a4c4-411c-910d-b9459ff36c20',
+                    source: 1,
+                    name: '@cf/baai/bge-small-en-v1.5',
+                    description: 'BAAI general embedding (Small) model that transforms any given text into a 384-dimensional vector',
+                    tags: [],
+                    properties: {
+                        beta: true,
+                        info: 'https://huggingface.co/BAAI/bge-base-en-v1.5',
+                        max_input_tokens: 512,
+                        output_dimensions: 384,
+                    },
+                },
+                {
+                    id: '429b9e8b-d99e-44de-91ad-706cf8183658',
+                    source: 1,
+                    name: '@cf/baai/bge-base-en-v1.5',
+                    description: 'BAAI general embedding (Base) model that transforms any given text into a 768-dimensional vector',
+                    tags: [],
+                    properties: {
+                        beta: true,
+                        info: 'https://huggingface.co/BAAI/bge-base-en-v1.5',
+                        max_input_tokens: 512,
+                        output_dimensions: 768,
+                    },
+                },
+                {
+                    id: '01bc2fb0-4bca-4598-b985-d2584a3f46c0',
+                    source: 1,
+                    name: '@cf/baai/bge-large-en-v1.5',
+                    description: 'BAAI general embedding (Large) model that transforms any given text into a 1024-dimensional vector',
+                    tags: [],
+                    properties: {
+                        beta: true,
+                        info: 'https://huggingface.co/BAAI/bge-base-en-v1.5',
+                        max_input_tokens: 512,
+                        output_dimensions: 1024,
+                    },
+                },
+            ],
+        },
         'Text Classification': {
             id: '19606750-23ed-4371-aab2-c20349b53a60',
             description: 'Sentiment analysis or text classification is a common NLP task that classifies a text input into labels or classes.',
@@ -530,6 +584,14 @@ export const workersAiCatalog = {
                         info: 'https://huggingface.co/Intel/distilbert-base-uncased-finetuned-sst-2-english-int8-static',
                     },
                 },
+                {
+                    id: '145337e7-cec3-4ebb-8e78-16ddfc75e580',
+                    source: 1,
+                    name: '@cf/baai/bge-reranker-base',
+                    description: 'Different from embedding model, reranker uses question and document as input and directly output similarity instead of embedding. You can get a relevance score by inputting query and passage to the reranker. And the score can be mapped to a float value in [0,1] by sigmoid function.\n\n',
+                    tags: [],
+                    properties: {},
+                },
             ],
         },
         'Object Detection': {
@@ -548,6 +610,20 @@ export const workersAiCatalog = {
                 },
             ],
         },
+        'Text-to-Speech': {
+            id: 'b52660a1-9a95-4ab2-8b1d-f232be34604a',
+            description: 'Text-to-Speech (TTS) is the task of generating natural sounding speech given text input. TTS models can be extended to have a single model that generates speech for multiple speakers and multiple languages.',
+            models: [
+                {
+                    id: 'c837b2ac-4d9b-4d37-8811-34de60f0c44f',
+                    source: 1,
+                    name: '@cf/myshell-ai/melotts',
+                    description: 'MeloTTS is a high-quality multi-lingual text-to-speech library by MyShell.ai.',
+                    tags: [],
+                    properties: {},
+                },
+            ],
+        },
         'Automatic Speech Recognition': {
             id: 'dfce1c48-2a81-462e-a7fd-de97ce985207',
             description: 'Automatic speech recognition (ASR) models convert a speech signal, typically an audio input, to text.',
@@ -579,9 +655,7 @@ export const workersAiCatalog = {
                     name: '@cf/openai/whisper-large-v3-turbo',
                     description: 'Whisper is a pre-trained model for automatic speech recognition (ASR) and speech translation. ',
                     tags: [],
-                    properties: {
-                        beta: true,
-                    },
+                    properties: {},
                 },
             ],
         },
@@ -720,51 +794,6 @@ export const workersAiCatalog = {
                 },
             ],
         },
-        'Text Embeddings': {
-            id: '0137cdcf-162a-4108-94f2-1ca59e8c65ee',
-            description: 'Feature extraction models transform raw data into numerical features that can be processed while preserving the information in the original dataset. These models are ideal as part of building vector search applications or Retrieval Augmented Generation workflows with Large Language Models (LLM).',
-            models: [
-                {
-                    id: '57fbd08a-a4c4-411c-910d-b9459ff36c20',
-                    source: 1,
-                    name: '@cf/baai/bge-small-en-v1.5',
-                    description: 'BAAI general embedding (Small) model that transforms any given text into a 384-dimensional vector',
-                    tags: [],
-                    properties: {
-                        beta: true,
-                        info: 'https://huggingface.co/BAAI/bge-base-en-v1.5',
-                        max_input_tokens: 512,
-                        output_dimensions: 384,
-                    },
-                },
-                {
-                    id: '429b9e8b-d99e-44de-91ad-706cf8183658',
-                    source: 1,
-                    name: '@cf/baai/bge-base-en-v1.5',
-                    description: 'BAAI general embedding (Base) model that transforms any given text into a 768-dimensional vector',
-                    tags: [],
-                    properties: {
-                        beta: true,
-                        info: 'https://huggingface.co/BAAI/bge-base-en-v1.5',
-                        max_input_tokens: 512,
-                        output_dimensions: 768,
-                    },
-                },
-                {
-                    id: '01bc2fb0-4bca-4598-b985-d2584a3f46c0',
-                    source: 1,
-                    name: '@cf/baai/bge-large-en-v1.5',
-                    description: 'BAAI general embedding (Large) model that transforms any given text into a 1024-dimensional vector',
-                    tags: [],
-                    properties: {
-                        beta: true,
-                        info: 'https://huggingface.co/BAAI/bge-base-en-v1.5',
-                        max_input_tokens: 512,
-                        output_dimensions: 1024,
-                    },
-                },
-            ],
-        },
         Summarization: {
             id: '6f4e65d8-da0f-40d2-9aa4-db582a5a04fd',
             description: 'Summarization is the task of producing a shorter version of a document while preserving its important information. Some models can extract text from the original input, while other models can generate entirely new text.',

package/dist/ai-tools/workers-ai/index.d.ts

@@ -0,0 +1,23 @@
+import { workersAiCatalog } from './catalog.js';
+type cloudflareModelTypes = keyof typeof workersAiCatalog.modelGroups;
+type cloudflareModelPossibilitiesRaw<M extends cloudflareModelTypes = cloudflareModelTypes> = (typeof workersAiCatalog.modelGroups)[M]['models'][number];
+export type cloudflareModelPossibilities<M extends cloudflareModelTypes = cloudflareModelTypes> = cloudflareModelPossibilitiesRaw<M>['name'];
+type cloudflareModelProperties<Model> = Model extends {
+    properties: infer Props;
+} ? keyof Props : never;
+type cloudflareModelPossibilitiesProperties<M extends cloudflareModelTypes = cloudflareModelTypes> = cloudflareModelProperties<cloudflareModelPossibilitiesRaw<M>>;
+type cloudflareFilteredModelPossibilitiesRaw<M extends cloudflareModelTypes = cloudflareModelTypes, K extends cloudflareModelPossibilitiesProperties<M> = cloudflareModelPossibilitiesProperties<M>, V extends cloudflareModelPossibilitiesRaw<M>['properties'][K] = any> = cloudflareModelPossibilitiesRaw<M> extends infer Model ? (Model extends {
+    properties: Record<K, V>;
+} ? Model : never) : never;
+type cloudflareFilteredModelPossibilities<M extends cloudflareModelTypes = cloudflareModelTypes, K extends cloudflareModelPossibilitiesProperties<M> = cloudflareModelPossibilitiesProperties<M>, V extends cloudflareModelPossibilitiesRaw<M>['properties'][K] = any> = cloudflareFilteredModelPossibilitiesRaw<M, K, V>['name'];
+export declare const enabledCloudflareLlmEmbeddingProviders: cloudflareModelPossibilities<'Text Embeddings'>[];
+export declare const enabledCloudflareLlmProviders: cloudflareModelPossibilities<'Text Generation'>[];
+export declare const enabledCloudflareLlmFunctionProviders: ("@cf/meta/llama-3.3-70b-instruct-fp8-fast" | "@hf/nousresearch/hermes-2-pro-mistral-7b")[];
+export declare const enabledCloudflareLlmImageProviders: cloudflareModelPossibilities<'Text-to-Image'>[];
+export type CloudflareModelsEnum<M extends cloudflareModelTypes = cloudflareModelTypes> = {
+    [K in cloudflareModelPossibilities<M>]: `workersai:${K}`;
+};
+export type CloudflareFunctionModelsEnum = {
+    [K in cloudflareFilteredModelPossibilities<'Text Generation', 'function_calling', true>]: `workersai:${K}`;
+};
+export {};

package/dist/ai-tools/workers-ai/index.js

@@ -0,0 +1,7 @@
+import { workersAiCatalog } from './catalog.js';
+// const enabledCloudflareLlmSummaryProviders: cloudflareModelPossibilities<'Summarization'>[] = workersAiCatalog.modelGroups.Summarization.models.map((model) => model.name);
+// const enabledCloudflareLlmClassificationProviders: cloudflareModelPossibilities<'Text Classification'>[] = workersAiCatalog.modelGroups['Text Classification'].models.map((model) => model.name);
+export const enabledCloudflareLlmEmbeddingProviders = workersAiCatalog.modelGroups['Text Embeddings'].models.map((model) => model.name);
+export const enabledCloudflareLlmProviders = workersAiCatalog.modelGroups['Text Generation'].models.map((model) => model.name);
+export const enabledCloudflareLlmFunctionProviders = workersAiCatalog.modelGroups['Text Generation'].models.filter((model) => 'function_calling' in model.properties && model.properties.function_calling).map((model) => model.name);
+export const enabledCloudflareLlmImageProviders = workersAiCatalog.modelGroups['Text-to-Image'].models.map((model) => model.name);

package/dist/index.d.ts

@@ -1,7 +1,6 @@
 export * from './ai-tools/index.js';
 export * from './d1/index.js';
 export * from './discourse/index.js';
-export * from './super-ai/index.js';
 type MethodKeys<T> = {
     [P in keyof T]: T[P] extends (...args: any[]) => any ? P : never;
 }[keyof T];

package/dist/index.js

@@ -1,4 +1,3 @@
 export * from './ai-tools/index.js';
 export * from './d1/index.js';
 export * from './discourse/index.js';
-export * from './super-ai/index.js';

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@chainfuse/types",
-	"version": "1.7.2",
+	"version": "2.0.0",
 	"description": "",
 	"author": "ChainFuse",
 	"homepage": "https://github.com/ChainFuse/packages/tree/main/packages/types#readme",
@@ -24,7 +24,7 @@
 	"scripts": {
 		"fmt": "prettier --check .",
 		"fmt:fix": "prettier --write .",
-		"lint": "eslint .",
+		"lint": "eslint src",
 		"lint:fix": "npm run lint -- --fix",
 		"clean": "npx -y rimraf@latest ./dist ./.tsbuildinfo",
 		"build": "tsc",
@@ -67,27 +67,22 @@
 			"import": "./dist/discourse/topic/index.js",
 			"types": "./dist/discourse/topic/index.d.ts"
 		},
-		"./super-ai": {
-			"import": "./dist/super-ai/index.js",
-			"types": "./dist/super-ai/index.d.ts"
-		},
-		"./super-ai/catalog/workers-ai": {
-			"import": "./dist/ai-tools/workers-ai-catalog.js",
-			"types": "./dist/ai-tools/workers-ai-catalog.d.ts"
-		},
 		"./ai-tools": {
 			"import": "./dist/ai-tools/index.js",
 			"types": "./dist/ai-tools/index.d.ts"
 		},
 		"./ai-tools/catalog/workers-ai": {
-			"import": "./dist/ai-tools/workers-ai-catalog.js",
-			"types": "./dist/ai-tools/workers-ai-catalog.d.ts"
+			"import": "./dist/ai-tools/workers-ai/catalog.js",
+			"types": "./dist/ai-tools/workers-ai/catalog.d.ts"
+		},
+		"./ai-tools/catalog/azure": {
+			"import": "./dist/ai-tools/azure/catalog.js",
+			"types": "./dist/ai-tools/azure/catalog.d.ts"
 		}
 	},
 	"prettier": "@demosjarco/prettier-config",
 	"devDependencies": {
-		"@cloudflare/workers-types": "^4.20250312.0",
-		"@types/json-schema": "^7.0.15"
+		"@cloudflare/workers-types": "^4.20250320.0"
 	},
-	"gitHead": "d79fa61a8407c01c134b9190f9f787260ba4add1"
+	"gitHead": "60babb051a4005d8495655784427e446fc7c7bcb"
 }