@cloudflare/tanstack-ai 0.1.6 → 0.1.7
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +6 -6
- package/dist/adapters/anthropic.cjs +32 -11
- package/dist/adapters/anthropic.cjs.map +1 -1
- package/dist/adapters/anthropic.d.cts +23 -5
- package/dist/adapters/anthropic.d.mts +23 -0
- package/dist/adapters/anthropic.mjs +30 -0
- package/dist/adapters/anthropic.mjs.map +1 -0
- package/dist/adapters/gemini.cjs +83 -20
- package/dist/adapters/gemini.cjs.map +1 -1
- package/dist/adapters/gemini.d.cts +52 -14
- package/dist/adapters/gemini.d.mts +52 -0
- package/dist/adapters/gemini.mjs +76 -0
- package/dist/adapters/gemini.mjs.map +1 -0
- package/dist/adapters/grok.cjs +44 -15
- package/dist/adapters/grok.cjs.map +1 -1
- package/dist/adapters/grok.d.cts +26 -9
- package/dist/adapters/grok.d.mts +26 -0
- package/dist/adapters/grok.mjs +40 -0
- package/dist/adapters/grok.mjs.map +1 -0
- package/dist/adapters/openai.cjs +76 -27
- package/dist/adapters/openai.cjs.map +1 -1
- package/dist/adapters/openai.d.cts +45 -17
- package/dist/adapters/openai.d.mts +45 -0
- package/dist/adapters/openai.mjs +66 -0
- package/dist/adapters/openai.mjs.map +1 -0
- package/dist/adapters/openrouter.cjs +83 -11
- package/dist/adapters/openrouter.cjs.map +1 -1
- package/dist/adapters/openrouter.d.cts +39 -7
- package/dist/adapters/openrouter.d.mts +39 -0
- package/dist/adapters/openrouter.mjs +81 -0
- package/dist/adapters/openrouter.mjs.map +1 -0
- package/dist/adapters/workers-ai-image.cjs +97 -11
- package/dist/adapters/workers-ai-image.cjs.map +1 -1
- package/dist/adapters/workers-ai-image.d.cts +41 -3
- package/dist/adapters/workers-ai-image.d.mts +41 -0
- package/dist/adapters/workers-ai-image.mjs +96 -0
- package/dist/adapters/workers-ai-image.mjs.map +1 -0
- package/dist/adapters/workers-ai-summarize.cjs +86 -10
- package/dist/adapters/workers-ai-summarize.cjs.map +1 -1
- package/dist/adapters/workers-ai-summarize.d.cts +42 -3
- package/dist/adapters/workers-ai-summarize.d.mts +42 -0
- package/dist/adapters/workers-ai-summarize.mjs +85 -0
- package/dist/adapters/workers-ai-summarize.mjs.map +1 -0
- package/dist/adapters/workers-ai-transcription.cjs +200 -11
- package/dist/adapters/workers-ai-transcription.cjs.map +1 -1
- package/dist/adapters/workers-ai-transcription.d.cts +74 -3
- package/dist/adapters/workers-ai-transcription.d.mts +74 -0
- package/dist/adapters/workers-ai-transcription.mjs +199 -0
- package/dist/adapters/workers-ai-transcription.mjs.map +1 -0
- package/dist/adapters/workers-ai-tts.cjs +112 -11
- package/dist/adapters/workers-ai-tts.cjs.map +1 -1
- package/dist/adapters/workers-ai-tts.d.cts +54 -3
- package/dist/adapters/workers-ai-tts.d.mts +54 -0
- package/dist/adapters/workers-ai-tts.mjs +111 -0
- package/dist/adapters/workers-ai-tts.mjs.map +1 -0
- package/dist/adapters/workers-ai.cjs +4 -11
- package/dist/adapters/workers-ai.d.cts +18 -3
- package/dist/adapters/workers-ai.d.mts +18 -0
- package/dist/adapters/workers-ai.mjs +470 -0
- package/dist/adapters/workers-ai.mjs.map +1 -0
- package/dist/binary-C9FAYwZj.cjs +70 -0
- package/dist/binary-C9FAYwZj.cjs.map +1 -0
- package/dist/binary-p4H_N_3M.mjs +59 -0
- package/dist/binary-p4H_N_3M.mjs.map +1 -0
- package/dist/create-fetcher-5iL34e6H.cjs +310 -0
- package/dist/create-fetcher-5iL34e6H.cjs.map +1 -0
- package/dist/create-fetcher-6p6heb85.d.mts +93 -0
- package/dist/create-fetcher-DY7wfYYy.mjs +281 -0
- package/dist/create-fetcher-DY7wfYYy.mjs.map +1 -0
- package/dist/create-fetcher-vAQ8WW-p.d.cts +93 -0
- package/dist/defineProperty-CbyrzcbA.mjs +39 -0
- package/dist/defineProperty-DQoAg20E.cjs +44 -0
- package/dist/index.cjs +50 -97
- package/dist/index.d.cts +12 -64
- package/dist/index.d.mts +12 -0
- package/dist/index.mjs +11 -0
- package/dist/workers-ai-Bm7Up4or.cjs +510 -0
- package/dist/workers-ai-Bm7Up4or.cjs.map +1 -0
- package/dist/workers-ai-rest-CkNCtBwv.cjs +85 -0
- package/dist/workers-ai-rest-CkNCtBwv.cjs.map +1 -0
- package/dist/workers-ai-rest-GKy2r7eG.mjs +74 -0
- package/dist/workers-ai-rest-GKy2r7eG.mjs.map +1 -0
- package/package.json +73 -40
- package/dist/_tsup-dts-rollup.d.cts +0 -837
- package/dist/_tsup-dts-rollup.d.ts +0 -837
- package/dist/adapters/anthropic.d.ts +0 -5
- package/dist/adapters/anthropic.js +0 -13
- package/dist/adapters/anthropic.js.map +0 -1
- package/dist/adapters/gemini.d.ts +0 -14
- package/dist/adapters/gemini.js +0 -22
- package/dist/adapters/gemini.js.map +0 -1
- package/dist/adapters/grok.d.ts +0 -9
- package/dist/adapters/grok.js +0 -17
- package/dist/adapters/grok.js.map +0 -1
- package/dist/adapters/openai.d.ts +0 -17
- package/dist/adapters/openai.js +0 -29
- package/dist/adapters/openai.js.map +0 -1
- package/dist/adapters/openrouter.d.ts +0 -7
- package/dist/adapters/openrouter.js +0 -13
- package/dist/adapters/openrouter.js.map +0 -1
- package/dist/adapters/workers-ai-image.d.ts +0 -3
- package/dist/adapters/workers-ai-image.js +0 -13
- package/dist/adapters/workers-ai-image.js.map +0 -1
- package/dist/adapters/workers-ai-summarize.d.ts +0 -3
- package/dist/adapters/workers-ai-summarize.js +0 -12
- package/dist/adapters/workers-ai-summarize.js.map +0 -1
- package/dist/adapters/workers-ai-transcription.d.ts +0 -3
- package/dist/adapters/workers-ai-transcription.js +0 -13
- package/dist/adapters/workers-ai-transcription.js.map +0 -1
- package/dist/adapters/workers-ai-tts.d.ts +0 -3
- package/dist/adapters/workers-ai-tts.js +0 -13
- package/dist/adapters/workers-ai-tts.js.map +0 -1
- package/dist/adapters/workers-ai.cjs.map +0 -1
- package/dist/adapters/workers-ai.d.ts +0 -3
- package/dist/adapters/workers-ai.js +0 -11
- package/dist/adapters/workers-ai.js.map +0 -1
- package/dist/chunk-2VII5BK2.js +0 -42
- package/dist/chunk-2VII5BK2.js.map +0 -1
- package/dist/chunk-3VQDXJLW.cjs +0 -46
- package/dist/chunk-3VQDXJLW.cjs.map +0 -1
- package/dist/chunk-3WK5ZD57.js +0 -57
- package/dist/chunk-3WK5ZD57.js.map +0 -1
- package/dist/chunk-4DE2IREA.cjs +0 -8
- package/dist/chunk-4DE2IREA.cjs.map +0 -1
- package/dist/chunk-53A2QIXV.js +0 -98
- package/dist/chunk-53A2QIXV.js.map +0 -1
- package/dist/chunk-6MEX2ZSM.cjs +0 -48
- package/dist/chunk-6MEX2ZSM.cjs.map +0 -1
- package/dist/chunk-7HSUHP63.cjs +0 -42
- package/dist/chunk-7HSUHP63.cjs.map +0 -1
- package/dist/chunk-AIAJANLC.js +0 -327
- package/dist/chunk-AIAJANLC.js.map +0 -1
- package/dist/chunk-CB3PSZY2.cjs +0 -218
- package/dist/chunk-CB3PSZY2.cjs.map +0 -1
- package/dist/chunk-FQE2UKUU.cjs +0 -57
- package/dist/chunk-FQE2UKUU.cjs.map +0 -1
- package/dist/chunk-GL3HHR4M.js +0 -523
- package/dist/chunk-GL3HHR4M.js.map +0 -1
- package/dist/chunk-J5DSSZTO.js +0 -71
- package/dist/chunk-J5DSSZTO.js.map +0 -1
- package/dist/chunk-JRFPCMSE.js +0 -57
- package/dist/chunk-JRFPCMSE.js.map +0 -1
- package/dist/chunk-KWYOUT2Y.js +0 -111
- package/dist/chunk-KWYOUT2Y.js.map +0 -1
- package/dist/chunk-MJ45L566.cjs +0 -57
- package/dist/chunk-MJ45L566.cjs.map +0 -1
- package/dist/chunk-N6U27RQW.cjs +0 -111
- package/dist/chunk-N6U27RQW.cjs.map +0 -1
- package/dist/chunk-NH3FF5JC.js +0 -86
- package/dist/chunk-NH3FF5JC.js.map +0 -1
- package/dist/chunk-OFUYHQZE.cjs +0 -31
- package/dist/chunk-OFUYHQZE.cjs.map +0 -1
- package/dist/chunk-OUEAAP25.cjs +0 -523
- package/dist/chunk-OUEAAP25.cjs.map +0 -1
- package/dist/chunk-Q33U5I5S.js +0 -31
- package/dist/chunk-Q33U5I5S.js.map +0 -1
- package/dist/chunk-V4LKHWJA.cjs +0 -71
- package/dist/chunk-V4LKHWJA.cjs.map +0 -1
- package/dist/chunk-V6TY7KAL.js +0 -8
- package/dist/chunk-V6TY7KAL.js.map +0 -1
- package/dist/chunk-WBTQKWFN.cjs +0 -327
- package/dist/chunk-WBTQKWFN.cjs.map +0 -1
- package/dist/chunk-XU7YEPML.js +0 -46
- package/dist/chunk-XU7YEPML.js.map +0 -1
- package/dist/chunk-YCGNMVQZ.js +0 -218
- package/dist/chunk-YCGNMVQZ.js.map +0 -1
- package/dist/chunk-Z5ZC5ZOQ.js +0 -48
- package/dist/chunk-Z5ZC5ZOQ.js.map +0 -1
- package/dist/chunk-ZFB72L57.cjs +0 -86
- package/dist/chunk-ZFB72L57.cjs.map +0 -1
- package/dist/chunk-ZGLSJM26.cjs +0 -98
- package/dist/chunk-ZGLSJM26.cjs.map +0 -1
- package/dist/index.cjs.map +0 -1
- package/dist/index.d.ts +0 -64
- package/dist/index.js +0 -97
- package/dist/index.js.map +0 -1
|
@@ -0,0 +1,39 @@
|
|
|
1
|
+
import { t as AiGatewayAdapterConfig } from "../create-fetcher-6p6heb85.mjs";
|
|
2
|
+
import { OpenRouterImageAdapter, OpenRouterSummarizeAdapter } from "@tanstack/ai-openrouter";
|
|
3
|
+
import { AnyTextAdapter } from "@tanstack/ai";
|
|
4
|
+
|
|
5
|
+
//#region src/adapters/openrouter.d.ts
|
|
6
|
+
type OpenRouterGatewayConfig = AiGatewayAdapterConfig;
|
|
7
|
+
/** OpenRouter chat model identifier. Accepts any string since OpenRouter proxies hundreds of models. */
|
|
8
|
+
type OpenRouterChatModel = string;
|
|
9
|
+
/** OpenRouter image model identifier. */
|
|
10
|
+
type OpenRouterImageModel = string;
|
|
11
|
+
/** OpenRouter summarize model identifier (same as chat models). */
|
|
12
|
+
type OpenRouterSummarizeModel = string;
|
|
13
|
+
/**
|
|
14
|
+
* Creates an OpenRouter chat adapter which uses Cloudflare AI Gateway.
|
|
15
|
+
* Supports both binding and credential-based configurations.
|
|
16
|
+
*
|
|
17
|
+
* @param model The model to use (e.g. "openai/gpt-4o", "anthropic/claude-sonnet-4-5")
|
|
18
|
+
* @param config Configuration options
|
|
19
|
+
*/
|
|
20
|
+
declare function createOpenRouterChat(model: OpenRouterChatModel, config: OpenRouterGatewayConfig): AnyTextAdapter;
|
|
21
|
+
/**
|
|
22
|
+
* Creates an OpenRouter image adapter which uses Cloudflare AI Gateway.
|
|
23
|
+
* Supports both binding and credential-based configurations.
|
|
24
|
+
*
|
|
25
|
+
* @param model The image model to use
|
|
26
|
+
* @param config Configuration options
|
|
27
|
+
*/
|
|
28
|
+
declare function createOpenRouterImage(model: OpenRouterImageModel, config: OpenRouterGatewayConfig): OpenRouterImageAdapter<any>;
|
|
29
|
+
/**
|
|
30
|
+
* Creates an OpenRouter summarize adapter which uses Cloudflare AI Gateway.
|
|
31
|
+
* Supports both binding and credential-based configurations.
|
|
32
|
+
*
|
|
33
|
+
* @param model The model to use for summarization
|
|
34
|
+
* @param config Configuration options
|
|
35
|
+
*/
|
|
36
|
+
declare function createOpenRouterSummarize(model: OpenRouterSummarizeModel, config: OpenRouterGatewayConfig): OpenRouterSummarizeAdapter<any>;
|
|
37
|
+
//#endregion
|
|
38
|
+
export { OpenRouterChatModel, OpenRouterGatewayConfig, OpenRouterImageModel, OpenRouterSummarizeModel, createOpenRouterChat, createOpenRouterImage, createOpenRouterSummarize };
|
|
39
|
+
//# sourceMappingURL=openrouter.d.mts.map
|
|
@@ -0,0 +1,81 @@
|
|
|
1
|
+
import { t as createGatewayFetch } from "../create-fetcher-DY7wfYYy.mjs";
|
|
2
|
+
import { OpenRouterImageAdapter, OpenRouterSummarizeAdapter, OpenRouterTextAdapter } from "@tanstack/ai-openrouter";
|
|
3
|
+
import { HTTPClient } from "@openrouter/sdk";
|
|
4
|
+
//#region src/adapters/openrouter.ts
|
|
5
|
+
/**
|
|
6
|
+
* Build OpenRouter config that routes requests through AI Gateway.
|
|
7
|
+
*
|
|
8
|
+
* The OpenRouter SDK accepts an `httpClient` with a custom `fetcher`,
|
|
9
|
+
* which we use to inject the AI Gateway fetch.
|
|
10
|
+
*/
|
|
11
|
+
function buildOpenRouterConfig(config) {
|
|
12
|
+
const httpClient = new HTTPClient({ fetcher: createGatewayFetch("openrouter", config) });
|
|
13
|
+
return {
|
|
14
|
+
apiKey: config.apiKey ?? "unused",
|
|
15
|
+
httpClient
|
|
16
|
+
};
|
|
17
|
+
}
|
|
18
|
+
/**
|
|
19
|
+
* Build OpenRouter image config that routes requests through AI Gateway.
|
|
20
|
+
*
|
|
21
|
+
* `OpenRouterImageConfig` extends `OpenRouterClientConfig` which declares
|
|
22
|
+
* `baseURL` and `apiKey` but not `httpClient`. However, the image adapter
|
|
23
|
+
* internally creates an `OpenRouter` SDK instance (which does accept
|
|
24
|
+
* `httpClient` via `SDKOptions`). The double-cast is needed because the
|
|
25
|
+
* declared config type is narrower than what the SDK constructor accepts.
|
|
26
|
+
*/
|
|
27
|
+
function buildOpenRouterImageConfig(config) {
|
|
28
|
+
const httpClient = new HTTPClient({ fetcher: createGatewayFetch("openrouter", config) });
|
|
29
|
+
return {
|
|
30
|
+
apiKey: config.apiKey ?? "unused",
|
|
31
|
+
httpClient
|
|
32
|
+
};
|
|
33
|
+
}
|
|
34
|
+
/**
|
|
35
|
+
* Build OpenRouter summarize config.
|
|
36
|
+
*
|
|
37
|
+
* `OpenRouterSummarizeConfig` extends `OpenRouterConfig` with optional
|
|
38
|
+
* `temperature` and `maxTokens` fields. Since those are optional, a plain
|
|
39
|
+
* `OpenRouterConfig` is structurally compatible — no cast needed.
|
|
40
|
+
*/
|
|
41
|
+
function buildOpenRouterSummarizeConfig(config) {
|
|
42
|
+
const httpClient = new HTTPClient({ fetcher: createGatewayFetch("openrouter", config) });
|
|
43
|
+
return {
|
|
44
|
+
apiKey: config.apiKey ?? "unused",
|
|
45
|
+
httpClient
|
|
46
|
+
};
|
|
47
|
+
}
|
|
48
|
+
/**
|
|
49
|
+
* Creates an OpenRouter chat adapter which uses Cloudflare AI Gateway.
|
|
50
|
+
* Supports both binding and credential-based configurations.
|
|
51
|
+
*
|
|
52
|
+
* @param model The model to use (e.g. "openai/gpt-4o", "anthropic/claude-sonnet-4-5")
|
|
53
|
+
* @param config Configuration options
|
|
54
|
+
*/
|
|
55
|
+
function createOpenRouterChat(model, config) {
|
|
56
|
+
return new OpenRouterTextAdapter(buildOpenRouterConfig(config), model);
|
|
57
|
+
}
|
|
58
|
+
/**
|
|
59
|
+
* Creates an OpenRouter image adapter which uses Cloudflare AI Gateway.
|
|
60
|
+
* Supports both binding and credential-based configurations.
|
|
61
|
+
*
|
|
62
|
+
* @param model The image model to use
|
|
63
|
+
* @param config Configuration options
|
|
64
|
+
*/
|
|
65
|
+
function createOpenRouterImage(model, config) {
|
|
66
|
+
return new OpenRouterImageAdapter(buildOpenRouterImageConfig(config), model);
|
|
67
|
+
}
|
|
68
|
+
/**
|
|
69
|
+
* Creates an OpenRouter summarize adapter which uses Cloudflare AI Gateway.
|
|
70
|
+
* Supports both binding and credential-based configurations.
|
|
71
|
+
*
|
|
72
|
+
* @param model The model to use for summarization
|
|
73
|
+
* @param config Configuration options
|
|
74
|
+
*/
|
|
75
|
+
function createOpenRouterSummarize(model, config) {
|
|
76
|
+
return new OpenRouterSummarizeAdapter(buildOpenRouterSummarizeConfig(config), model);
|
|
77
|
+
}
|
|
78
|
+
//#endregion
|
|
79
|
+
export { createOpenRouterChat, createOpenRouterImage, createOpenRouterSummarize };
|
|
80
|
+
|
|
81
|
+
//# sourceMappingURL=openrouter.mjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"openrouter.mjs","names":[],"sources":["../../src/adapters/openrouter.ts"],"sourcesContent":["import {\n\tOpenRouterTextAdapter,\n\tOpenRouterImageAdapter,\n\tOpenRouterSummarizeAdapter,\n\ttype OpenRouterConfig,\n\ttype OpenRouterImageConfig,\n\ttype OpenRouterSummarizeConfig,\n} from \"@tanstack/ai-openrouter\";\nimport { HTTPClient } from \"@openrouter/sdk\";\nimport { createGatewayFetch, type AiGatewayAdapterConfig } from \"../utils/create-fetcher\";\nimport type { AnyTextAdapter } from \"@tanstack/ai\";\n\nexport type OpenRouterGatewayConfig = AiGatewayAdapterConfig;\n\n/**\n * Build OpenRouter config that routes requests through AI Gateway.\n *\n * The OpenRouter SDK accepts an `httpClient` with a custom `fetcher`,\n * which we use to inject the AI Gateway fetch.\n */\nfunction buildOpenRouterConfig(config: OpenRouterGatewayConfig): OpenRouterConfig {\n\tconst httpClient = new HTTPClient({\n\t\tfetcher: createGatewayFetch(\"openrouter\", config),\n\t});\n\treturn {\n\t\tapiKey: config.apiKey ?? \"unused\",\n\t\t// Cast needed: the installed @openrouter/sdk version may differ from the\n\t\t// version @tanstack/ai-openrouter was built against. The HTTPClient interface\n\t\t// is structurally compatible but TypeScript sees them as separate declarations.\n\t\thttpClient: httpClient as unknown as OpenRouterConfig[\"httpClient\"],\n\t};\n}\n\n/**\n * Build OpenRouter image config that routes requests through AI Gateway.\n *\n * `OpenRouterImageConfig` extends `OpenRouterClientConfig` which declares\n * `baseURL` and `apiKey` but not `httpClient`. However, the image adapter\n * internally creates an `OpenRouter` SDK instance (which does accept\n * `httpClient` via `SDKOptions`). The double-cast is needed because the\n * declared config type is narrower than what the SDK constructor accepts.\n */\nfunction buildOpenRouterImageConfig(config: OpenRouterGatewayConfig): OpenRouterImageConfig {\n\tconst httpClient = new HTTPClient({\n\t\tfetcher: createGatewayFetch(\"openrouter\", config),\n\t});\n\treturn {\n\t\tapiKey: config.apiKey ?? \"unused\",\n\t\thttpClient,\n\t} as unknown as OpenRouterImageConfig;\n}\n\n/**\n * Build OpenRouter summarize config.\n *\n * `OpenRouterSummarizeConfig` extends `OpenRouterConfig` with optional\n * `temperature` and `maxTokens` fields. Since those are optional, a plain\n * `OpenRouterConfig` is structurally compatible — no cast needed.\n */\nfunction buildOpenRouterSummarizeConfig(\n\tconfig: OpenRouterGatewayConfig,\n): OpenRouterSummarizeConfig {\n\tconst httpClient = new HTTPClient({\n\t\tfetcher: createGatewayFetch(\"openrouter\", config),\n\t});\n\treturn {\n\t\tapiKey: config.apiKey ?? \"unused\",\n\t\thttpClient: httpClient as unknown as OpenRouterSummarizeConfig[\"httpClient\"],\n\t};\n}\n\n// ---------------------------------------------------------------------------\n// Chat model type — OpenRouter supports many models; we use a loose string type\n// ---------------------------------------------------------------------------\n\n/** OpenRouter chat model identifier. Accepts any string since OpenRouter proxies hundreds of models. */\nexport type OpenRouterChatModel = string;\n\n/** OpenRouter image model identifier. */\nexport type OpenRouterImageModel = string;\n\n/** OpenRouter summarize model identifier (same as chat models). */\nexport type OpenRouterSummarizeModel = string;\n\n// ---------------------------------------------------------------------------\n// Factory functions\n// ---------------------------------------------------------------------------\n\n/**\n * Creates an OpenRouter chat adapter which uses Cloudflare AI Gateway.\n * Supports both binding and credential-based configurations.\n *\n * @param model The model to use (e.g. \"openai/gpt-4o\", \"anthropic/claude-sonnet-4-5\")\n * @param config Configuration options\n */\nexport function createOpenRouterChat(\n\tmodel: OpenRouterChatModel,\n\tconfig: OpenRouterGatewayConfig,\n): AnyTextAdapter {\n\t// Cast needed: we accept any string model while upstream expects a literal union\n\treturn new OpenRouterTextAdapter(buildOpenRouterConfig(config), model as any);\n}\n\n/**\n * Creates an OpenRouter image adapter which uses Cloudflare AI Gateway.\n * Supports both binding and credential-based configurations.\n *\n * @param model The image model to use\n * @param config Configuration options\n */\nexport function createOpenRouterImage(\n\tmodel: OpenRouterImageModel,\n\tconfig: OpenRouterGatewayConfig,\n) {\n\t// Cast needed: we accept any string model while upstream expects a literal union\n\treturn new OpenRouterImageAdapter(buildOpenRouterImageConfig(config), model as any);\n}\n\n/**\n * Creates an OpenRouter summarize adapter which uses Cloudflare AI Gateway.\n * Supports both binding and credential-based configurations.\n *\n * @param model The model to use for summarization\n * @param config Configuration options\n */\nexport function createOpenRouterSummarize(\n\tmodel: OpenRouterSummarizeModel,\n\tconfig: OpenRouterGatewayConfig,\n) {\n\t// Cast needed: we accept any string model while upstream expects a literal union\n\treturn new OpenRouterSummarizeAdapter(buildOpenRouterSummarizeConfig(config), model as any);\n}\n"],"mappings":";;;;;;;;;;AAoBA,SAAS,sBAAsB,QAAmD;CACjF,MAAM,aAAa,IAAI,WAAW,EACjC,SAAS,mBAAmB,cAAc,OAAO,EACjD,CAAC;AACF,QAAO;EACN,QAAQ,OAAO,UAAU;EAIb;EACZ;;;;;;;;;;;AAYF,SAAS,2BAA2B,QAAwD;CAC3F,MAAM,aAAa,IAAI,WAAW,EACjC,SAAS,mBAAmB,cAAc,OAAO,EACjD,CAAC;AACF,QAAO;EACN,QAAQ,OAAO,UAAU;EACzB;EACA;;;;;;;;;AAUF,SAAS,+BACR,QAC4B;CAC5B,MAAM,aAAa,IAAI,WAAW,EACjC,SAAS,mBAAmB,cAAc,OAAO,EACjD,CAAC;AACF,QAAO;EACN,QAAQ,OAAO,UAAU;EACb;EACZ;;;;;;;;;AA2BF,SAAgB,qBACf,OACA,QACiB;AAEjB,QAAO,IAAI,sBAAsB,sBAAsB,OAAO,EAAE,MAAa;;;;;;;;;AAU9E,SAAgB,sBACf,OACA,QACC;AAED,QAAO,IAAI,uBAAuB,2BAA2B,OAAO,EAAE,MAAa;;;;;;;;;AAUpF,SAAgB,0BACf,OACA,QACC;AAED,QAAO,IAAI,2BAA2B,+BAA+B,OAAO,EAAE,MAAa"}
|
|
@@ -1,13 +1,99 @@
|
|
|
1
|
-
|
|
1
|
+
Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
|
|
2
|
+
require("../workers-ai-Bm7Up4or.cjs");
|
|
3
|
+
const require_create_fetcher = require("../create-fetcher-5iL34e6H.cjs");
|
|
4
|
+
const require_defineProperty = require("../defineProperty-DQoAg20E.cjs");
|
|
5
|
+
const require_workers_ai_rest = require("../workers-ai-rest-CkNCtBwv.cjs");
|
|
6
|
+
const require_binary = require("../binary-C9FAYwZj.cjs");
|
|
7
|
+
let _tanstack_ai_adapters = require("@tanstack/ai/adapters");
|
|
8
|
+
//#region src/adapters/workers-ai-image.ts
|
|
9
|
+
var WorkersAiImageAdapter = class extends _tanstack_ai_adapters.BaseImageAdapter {
|
|
10
|
+
constructor(config, model) {
|
|
11
|
+
super({}, model);
|
|
12
|
+
require_defineProperty._defineProperty(this, "name", "workers-ai-image");
|
|
13
|
+
require_defineProperty._defineProperty(this, "adapterConfig", void 0);
|
|
14
|
+
require_create_fetcher.validateWorkersAiConfig(config);
|
|
15
|
+
this.adapterConfig = config;
|
|
16
|
+
}
|
|
17
|
+
async generateImages(options) {
|
|
18
|
+
const { prompt, size, modelOptions } = options;
|
|
19
|
+
const extra = { ...modelOptions };
|
|
20
|
+
if (size) {
|
|
21
|
+
const [w, h] = size.split("x");
|
|
22
|
+
if (w) extra.width = Number(w);
|
|
23
|
+
if (h) extra.height = Number(h);
|
|
24
|
+
}
|
|
25
|
+
if (require_create_fetcher.isDirectBindingConfig(this.adapterConfig)) return this.generateViaBinding(prompt, extra);
|
|
26
|
+
if (require_create_fetcher.isDirectCredentialsConfig(this.adapterConfig)) return this.generateViaRest(prompt, extra);
|
|
27
|
+
return this.generateViaGateway(prompt, extra);
|
|
28
|
+
}
|
|
29
|
+
async generateViaBinding(prompt, options) {
|
|
30
|
+
const b64 = await require_binary.binaryToBase64(await this.adapterConfig.binding.run(this.model, {
|
|
31
|
+
prompt,
|
|
32
|
+
...options
|
|
33
|
+
}), "image");
|
|
34
|
+
return this.wrapResult(b64);
|
|
35
|
+
}
|
|
36
|
+
async generateViaRest(prompt, options) {
|
|
37
|
+
const config = this.adapterConfig;
|
|
38
|
+
const buffer = await (await require_workers_ai_rest.workersAiRestFetch(config, this.model, {
|
|
39
|
+
prompt,
|
|
40
|
+
...options
|
|
41
|
+
}, {
|
|
42
|
+
label: "Workers AI image",
|
|
43
|
+
signal: options.signal
|
|
44
|
+
})).arrayBuffer();
|
|
45
|
+
return this.wrapResult(require_binary.uint8ArrayToBase64(new Uint8Array(buffer)));
|
|
46
|
+
}
|
|
47
|
+
async generateViaGateway(prompt, options) {
|
|
48
|
+
const gatewayConfig = this.adapterConfig;
|
|
49
|
+
const response = await require_create_fetcher.createGatewayFetch("workers-ai", gatewayConfig)("https://api.cloudflare.com/v1/images/generations", {
|
|
50
|
+
method: "POST",
|
|
51
|
+
body: JSON.stringify({
|
|
52
|
+
model: this.model,
|
|
53
|
+
prompt,
|
|
54
|
+
...options
|
|
55
|
+
})
|
|
56
|
+
});
|
|
57
|
+
if (!response.ok) {
|
|
58
|
+
const errorText = await response.text();
|
|
59
|
+
throw new Error(`Workers AI image gateway request failed (${response.status}): ${errorText}`);
|
|
60
|
+
}
|
|
61
|
+
const buffer = await response.arrayBuffer();
|
|
62
|
+
return this.wrapResult(require_binary.uint8ArrayToBase64(new Uint8Array(buffer)));
|
|
63
|
+
}
|
|
64
|
+
/** Wrap a base64 image string into the standard ImageGenerationResult. */
|
|
65
|
+
wrapResult(b64) {
|
|
66
|
+
return {
|
|
67
|
+
id: this.generateId(),
|
|
68
|
+
model: this.model,
|
|
69
|
+
images: [{ b64Json: b64 }]
|
|
70
|
+
};
|
|
71
|
+
}
|
|
72
|
+
};
|
|
73
|
+
/**
|
|
74
|
+
* Creates a Workers AI image generation adapter.
|
|
75
|
+
*
|
|
76
|
+
* Works with TanStack AI's `generateImage()` activity function:
|
|
77
|
+
* ```ts
|
|
78
|
+
* import { generateImage } from "@tanstack/ai";
|
|
79
|
+
* import { createWorkersAiImage } from "@cloudflare/tanstack-ai";
|
|
80
|
+
*
|
|
81
|
+
* const adapter = createWorkersAiImage(
|
|
82
|
+
* "@cf/stabilityai/stable-diffusion-xl-base-1.0",
|
|
83
|
+
* { binding: env.AI },
|
|
84
|
+
* );
|
|
85
|
+
*
|
|
86
|
+
* const result = await generateImage({ adapter, prompt: "a cat in space" });
|
|
87
|
+
* ```
|
|
88
|
+
*
|
|
89
|
+
* Note: Factory takes `(model, config)` for ergonomics — the class constructor
|
|
90
|
+
* uses `(config, model)` to match TanStack AI's upstream convention.
|
|
91
|
+
*/
|
|
92
|
+
function createWorkersAiImage(model, config) {
|
|
93
|
+
return new WorkersAiImageAdapter(config, model);
|
|
94
|
+
}
|
|
95
|
+
//#endregion
|
|
96
|
+
exports.WorkersAiImageAdapter = WorkersAiImageAdapter;
|
|
97
|
+
exports.createWorkersAiImage = createWorkersAiImage;
|
|
2
98
|
|
|
3
|
-
|
|
4
|
-
var _chunkZGLSJM26cjs = require('../chunk-ZGLSJM26.cjs');
|
|
5
|
-
require('../chunk-3VQDXJLW.cjs');
|
|
6
|
-
require('../chunk-7HSUHP63.cjs');
|
|
7
|
-
require('../chunk-WBTQKWFN.cjs');
|
|
8
|
-
require('../chunk-4DE2IREA.cjs');
|
|
9
|
-
|
|
10
|
-
|
|
11
|
-
|
|
12
|
-
exports.WorkersAiImageAdapter = _chunkZGLSJM26cjs.WorkersAiImageAdapter; exports.createWorkersAiImage = _chunkZGLSJM26cjs.createWorkersAiImage;
|
|
13
99
|
//# sourceMappingURL=workers-ai-image.cjs.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"sources":["/
|
|
1
|
+
{"version":3,"file":"workers-ai-image.cjs","names":["BaseImageAdapter","isDirectBindingConfig","isDirectCredentialsConfig","binaryToBase64","workersAiRestFetch","uint8ArrayToBase64","createGatewayFetch"],"sources":["../../src/adapters/workers-ai-image.ts"],"sourcesContent":["import { BaseImageAdapter } from \"@tanstack/ai/adapters\";\nimport type { ImageGenerationOptions, ImageGenerationResult } from \"@tanstack/ai\";\nimport type { AiModels, BaseAiTextToImage } from \"@cloudflare/workers-types\";\nimport {\n\ttype WorkersAiAdapterConfig,\n\ttype WorkersAiDirectBindingConfig,\n\ttype AiGatewayAdapterConfig,\n\tcreateGatewayFetch,\n\tisDirectBindingConfig,\n\tisDirectCredentialsConfig,\n\tvalidateWorkersAiConfig,\n} from \"../utils/create-fetcher\";\nimport { workersAiRestFetch } from \"../utils/workers-ai-rest\";\nimport { binaryToBase64, uint8ArrayToBase64 } from \"../utils/binary\";\nimport type { WorkersAiDirectCredentialsConfig } from \"../utils/create-fetcher\";\n\n// ---------------------------------------------------------------------------\n// Model type derived from @cloudflare/workers-types\n// ---------------------------------------------------------------------------\n\nexport type WorkersAiImageModel =\n\t| { [K in keyof AiModels]: AiModels[K] extends BaseAiTextToImage ? K : never }[keyof AiModels]\n\t| (string & {});\n\n// ---------------------------------------------------------------------------\n// WorkersAiImageAdapter: image generation via Workers AI\n// Extends BaseImageAdapter so it works with TanStack AI's generateImage()\n// ---------------------------------------------------------------------------\n\nexport class WorkersAiImageAdapter extends BaseImageAdapter<WorkersAiImageModel> {\n\treadonly name = \"workers-ai-image\" as const;\n\tprivate adapterConfig: WorkersAiAdapterConfig;\n\n\tconstructor(config: WorkersAiAdapterConfig, model: WorkersAiImageModel) {\n\t\tsuper({}, model);\n\t\tvalidateWorkersAiConfig(config);\n\t\tthis.adapterConfig = config;\n\t}\n\n\tasync generateImages(options: ImageGenerationOptions): Promise<ImageGenerationResult> {\n\t\tconst { prompt, size, modelOptions } = options;\n\t\tconst extra: Record<string, unknown> = { ...modelOptions };\n\n\t\t// Note: Workers AI Stable Diffusion models only generate a single image\n\t\t// per request — there is no multi-image parameter. `numberOfImages` from\n\t\t// the TanStack AI options is intentionally not forwarded. Use\n\t\t// `modelOptions.num_steps` to control the number of diffusion steps.\n\n\t\tif (size) {\n\t\t\tconst [w, h] = size.split(\"x\");\n\t\t\tif (w) extra.width = Number(w);\n\t\t\tif (h) extra.height = Number(h);\n\t\t}\n\n\t\tif (isDirectBindingConfig(this.adapterConfig)) {\n\t\t\treturn this.generateViaBinding(prompt, extra);\n\t\t}\n\n\t\tif (isDirectCredentialsConfig(this.adapterConfig)) {\n\t\t\treturn this.generateViaRest(prompt, extra);\n\t\t}\n\n\t\t// Gateway mode\n\t\treturn this.generateViaGateway(prompt, extra);\n\t}\n\n\tprivate async generateViaBinding(\n\t\tprompt: string,\n\t\toptions: Record<string, unknown>,\n\t): Promise<ImageGenerationResult> {\n\t\tconst ai = (this.adapterConfig as WorkersAiDirectBindingConfig).binding;\n\t\tconst result = await ai.run(this.model, { prompt, ...options });\n\n\t\tconst b64 = await binaryToBase64(result, \"image\");\n\t\treturn this.wrapResult(b64);\n\t}\n\n\tprivate async generateViaRest(\n\t\tprompt: string,\n\t\toptions: Record<string, unknown>,\n\t): Promise<ImageGenerationResult> {\n\t\tconst config = this.adapterConfig as WorkersAiDirectCredentialsConfig;\n\t\tconst response = await workersAiRestFetch(\n\t\t\tconfig,\n\t\t\tthis.model,\n\t\t\t{ prompt, ...options },\n\t\t\t{ label: \"Workers AI image\", signal: (options as { signal?: AbortSignal }).signal },\n\t\t);\n\n\t\tconst buffer = await response.arrayBuffer();\n\t\treturn this.wrapResult(uint8ArrayToBase64(new Uint8Array(buffer)));\n\t}\n\n\tprivate async generateViaGateway(\n\t\tprompt: string,\n\t\toptions: Record<string, unknown>,\n\t): Promise<ImageGenerationResult> {\n\t\tconst gatewayConfig = this.adapterConfig as AiGatewayAdapterConfig;\n\t\tconst gatewayFetch = createGatewayFetch(\"workers-ai\", gatewayConfig);\n\n\t\t// The URL here is a placeholder — createGatewayFetch for \"workers-ai\" extracts\n\t\t// the model from the body, sets it as the endpoint, and routes through the gateway.\n\t\t// The actual URL path is not used.\n\t\tconst response = await gatewayFetch(\"https://api.cloudflare.com/v1/images/generations\", {\n\t\t\tmethod: \"POST\",\n\t\t\tbody: JSON.stringify({\n\t\t\t\tmodel: this.model,\n\t\t\t\tprompt,\n\t\t\t\t...options,\n\t\t\t}),\n\t\t});\n\n\t\tif (!response.ok) {\n\t\t\tconst errorText = await response.text();\n\t\t\tthrow new Error(\n\t\t\t\t`Workers AI image gateway request failed (${response.status}): ${errorText}`,\n\t\t\t);\n\t\t}\n\n\t\tconst buffer = await response.arrayBuffer();\n\t\treturn this.wrapResult(uint8ArrayToBase64(new Uint8Array(buffer)));\n\t}\n\n\t/** Wrap a base64 image string into the standard ImageGenerationResult. */\n\tprivate wrapResult(b64: string): ImageGenerationResult {\n\t\treturn {\n\t\t\tid: this.generateId(),\n\t\t\tmodel: this.model,\n\t\t\timages: [{ b64Json: b64 }],\n\t\t};\n\t}\n}\n\n// ---------------------------------------------------------------------------\n// Factory function\n// ---------------------------------------------------------------------------\n\n/**\n * Creates a Workers AI image generation adapter.\n *\n * Works with TanStack AI's `generateImage()` activity function:\n * ```ts\n * import { generateImage } from \"@tanstack/ai\";\n * import { createWorkersAiImage } from \"@cloudflare/tanstack-ai\";\n *\n * const adapter = createWorkersAiImage(\n * \"@cf/stabilityai/stable-diffusion-xl-base-1.0\",\n * { binding: env.AI },\n * );\n *\n * const result = await generateImage({ adapter, prompt: \"a cat in space\" });\n * ```\n *\n * Note: Factory takes `(model, config)` for ergonomics — the class constructor\n * uses `(config, model)` to match TanStack AI's upstream convention.\n */\nexport function createWorkersAiImage(model: WorkersAiImageModel, config: WorkersAiAdapterConfig) {\n\treturn new WorkersAiImageAdapter(config, model);\n}\n"],"mappings":";;;;;;;;AA6BA,IAAa,wBAAb,cAA2CA,sBAAAA,iBAAsC;CAIhF,YAAY,QAAgC,OAA4B;AACvE,QAAM,EAAE,EAAE,MAAM;+CAJR,QAAO,mBAA4B;+CACpC,iBAAA,KAAA,EAAsC;AAI7C,yBAAA,wBAAwB,OAAO;AAC/B,OAAK,gBAAgB;;CAGtB,MAAM,eAAe,SAAiE;EACrF,MAAM,EAAE,QAAQ,MAAM,iBAAiB;EACvC,MAAM,QAAiC,EAAE,GAAG,cAAc;AAO1D,MAAI,MAAM;GACT,MAAM,CAAC,GAAG,KAAK,KAAK,MAAM,IAAI;AAC9B,OAAI,EAAG,OAAM,QAAQ,OAAO,EAAE;AAC9B,OAAI,EAAG,OAAM,SAAS,OAAO,EAAE;;AAGhC,MAAIC,uBAAAA,sBAAsB,KAAK,cAAc,CAC5C,QAAO,KAAK,mBAAmB,QAAQ,MAAM;AAG9C,MAAIC,uBAAAA,0BAA0B,KAAK,cAAc,CAChD,QAAO,KAAK,gBAAgB,QAAQ,MAAM;AAI3C,SAAO,KAAK,mBAAmB,QAAQ,MAAM;;CAG9C,MAAc,mBACb,QACA,SACiC;EAIjC,MAAM,MAAM,MAAMC,eAAAA,eAFH,MADH,KAAK,cAA+C,QACxC,IAAI,KAAK,OAAO;GAAE;GAAQ,GAAG;GAAS,CAAC,EAEtB,QAAQ;AACjD,SAAO,KAAK,WAAW,IAAI;;CAG5B,MAAc,gBACb,QACA,SACiC;EACjC,MAAM,SAAS,KAAK;EAQpB,MAAM,SAAS,OAPE,MAAMC,wBAAAA,mBACtB,QACA,KAAK,OACL;GAAE;GAAQ,GAAG;GAAS,EACtB;GAAE,OAAO;GAAoB,QAAS,QAAqC;GAAQ,CACnF,EAE6B,aAAa;AAC3C,SAAO,KAAK,WAAWC,eAAAA,mBAAmB,IAAI,WAAW,OAAO,CAAC,CAAC;;CAGnE,MAAc,mBACb,QACA,SACiC;EACjC,MAAM,gBAAgB,KAAK;EAM3B,MAAM,WAAW,MALIC,uBAAAA,mBAAmB,cAAc,cAAc,CAKhC,oDAAoD;GACvF,QAAQ;GACR,MAAM,KAAK,UAAU;IACpB,OAAO,KAAK;IACZ;IACA,GAAG;IACH,CAAC;GACF,CAAC;AAEF,MAAI,CAAC,SAAS,IAAI;GACjB,MAAM,YAAY,MAAM,SAAS,MAAM;AACvC,SAAM,IAAI,MACT,4CAA4C,SAAS,OAAO,KAAK,YACjE;;EAGF,MAAM,SAAS,MAAM,SAAS,aAAa;AAC3C,SAAO,KAAK,WAAWD,eAAAA,mBAAmB,IAAI,WAAW,OAAO,CAAC,CAAC;;;CAInE,WAAmB,KAAoC;AACtD,SAAO;GACN,IAAI,KAAK,YAAY;GACrB,OAAO,KAAK;GACZ,QAAQ,CAAC,EAAE,SAAS,KAAK,CAAC;GAC1B;;;;;;;;;;;;;;;;;;;;;;AA2BH,SAAgB,qBAAqB,OAA4B,QAAgC;AAChG,QAAO,IAAI,sBAAsB,QAAQ,MAAM"}
|
|
@@ -1,3 +1,41 @@
|
|
|
1
|
-
|
|
2
|
-
|
|
3
|
-
|
|
1
|
+
import { i as WorkersAiAdapterConfig } from "../create-fetcher-vAQ8WW-p.cjs";
|
|
2
|
+
import { ImageGenerationOptions, ImageGenerationResult } from "@tanstack/ai";
|
|
3
|
+
import { BaseImageAdapter } from "@tanstack/ai/adapters";
|
|
4
|
+
import { AiModels, BaseAiTextToImage } from "@cloudflare/workers-types";
|
|
5
|
+
|
|
6
|
+
//#region src/adapters/workers-ai-image.d.ts
|
|
7
|
+
type WorkersAiImageModel = { [K in keyof AiModels]: AiModels[K] extends BaseAiTextToImage ? K : never }[keyof AiModels] | (string & {});
|
|
8
|
+
declare class WorkersAiImageAdapter extends BaseImageAdapter<WorkersAiImageModel> {
|
|
9
|
+
readonly name: "workers-ai-image";
|
|
10
|
+
private adapterConfig;
|
|
11
|
+
constructor(config: WorkersAiAdapterConfig, model: WorkersAiImageModel);
|
|
12
|
+
generateImages(options: ImageGenerationOptions): Promise<ImageGenerationResult>;
|
|
13
|
+
private generateViaBinding;
|
|
14
|
+
private generateViaRest;
|
|
15
|
+
private generateViaGateway;
|
|
16
|
+
/** Wrap a base64 image string into the standard ImageGenerationResult. */
|
|
17
|
+
private wrapResult;
|
|
18
|
+
}
|
|
19
|
+
/**
|
|
20
|
+
* Creates a Workers AI image generation adapter.
|
|
21
|
+
*
|
|
22
|
+
* Works with TanStack AI's `generateImage()` activity function:
|
|
23
|
+
* ```ts
|
|
24
|
+
* import { generateImage } from "@tanstack/ai";
|
|
25
|
+
* import { createWorkersAiImage } from "@cloudflare/tanstack-ai";
|
|
26
|
+
*
|
|
27
|
+
* const adapter = createWorkersAiImage(
|
|
28
|
+
* "@cf/stabilityai/stable-diffusion-xl-base-1.0",
|
|
29
|
+
* { binding: env.AI },
|
|
30
|
+
* );
|
|
31
|
+
*
|
|
32
|
+
* const result = await generateImage({ adapter, prompt: "a cat in space" });
|
|
33
|
+
* ```
|
|
34
|
+
*
|
|
35
|
+
* Note: Factory takes `(model, config)` for ergonomics — the class constructor
|
|
36
|
+
* uses `(config, model)` to match TanStack AI's upstream convention.
|
|
37
|
+
*/
|
|
38
|
+
declare function createWorkersAiImage(model: WorkersAiImageModel, config: WorkersAiAdapterConfig): WorkersAiImageAdapter;
|
|
39
|
+
//#endregion
|
|
40
|
+
export { WorkersAiImageAdapter, WorkersAiImageModel, createWorkersAiImage };
|
|
41
|
+
//# sourceMappingURL=workers-ai-image.d.cts.map
|
|
@@ -0,0 +1,41 @@
|
|
|
1
|
+
import { i as WorkersAiAdapterConfig } from "../create-fetcher-6p6heb85.mjs";
|
|
2
|
+
import { BaseImageAdapter } from "@tanstack/ai/adapters";
|
|
3
|
+
import { ImageGenerationOptions, ImageGenerationResult } from "@tanstack/ai";
|
|
4
|
+
import { AiModels, BaseAiTextToImage } from "@cloudflare/workers-types";
|
|
5
|
+
|
|
6
|
+
//#region src/adapters/workers-ai-image.d.ts
|
|
7
|
+
type WorkersAiImageModel = { [K in keyof AiModels]: AiModels[K] extends BaseAiTextToImage ? K : never }[keyof AiModels] | (string & {});
|
|
8
|
+
declare class WorkersAiImageAdapter extends BaseImageAdapter<WorkersAiImageModel> {
|
|
9
|
+
readonly name: "workers-ai-image";
|
|
10
|
+
private adapterConfig;
|
|
11
|
+
constructor(config: WorkersAiAdapterConfig, model: WorkersAiImageModel);
|
|
12
|
+
generateImages(options: ImageGenerationOptions): Promise<ImageGenerationResult>;
|
|
13
|
+
private generateViaBinding;
|
|
14
|
+
private generateViaRest;
|
|
15
|
+
private generateViaGateway;
|
|
16
|
+
/** Wrap a base64 image string into the standard ImageGenerationResult. */
|
|
17
|
+
private wrapResult;
|
|
18
|
+
}
|
|
19
|
+
/**
|
|
20
|
+
* Creates a Workers AI image generation adapter.
|
|
21
|
+
*
|
|
22
|
+
* Works with TanStack AI's `generateImage()` activity function:
|
|
23
|
+
* ```ts
|
|
24
|
+
* import { generateImage } from "@tanstack/ai";
|
|
25
|
+
* import { createWorkersAiImage } from "@cloudflare/tanstack-ai";
|
|
26
|
+
*
|
|
27
|
+
* const adapter = createWorkersAiImage(
|
|
28
|
+
* "@cf/stabilityai/stable-diffusion-xl-base-1.0",
|
|
29
|
+
* { binding: env.AI },
|
|
30
|
+
* );
|
|
31
|
+
*
|
|
32
|
+
* const result = await generateImage({ adapter, prompt: "a cat in space" });
|
|
33
|
+
* ```
|
|
34
|
+
*
|
|
35
|
+
* Note: Factory takes `(model, config)` for ergonomics — the class constructor
|
|
36
|
+
* uses `(config, model)` to match TanStack AI's upstream convention.
|
|
37
|
+
*/
|
|
38
|
+
declare function createWorkersAiImage(model: WorkersAiImageModel, config: WorkersAiAdapterConfig): WorkersAiImageAdapter;
|
|
39
|
+
//#endregion
|
|
40
|
+
export { WorkersAiImageAdapter, WorkersAiImageModel, createWorkersAiImage };
|
|
41
|
+
//# sourceMappingURL=workers-ai-image.d.mts.map
|
|
@@ -0,0 +1,96 @@
|
|
|
1
|
+
import { a as validateWorkersAiConfig, i as isDirectCredentialsConfig, r as isDirectBindingConfig, t as createGatewayFetch } from "../create-fetcher-DY7wfYYy.mjs";
|
|
2
|
+
import { t as _defineProperty } from "../defineProperty-CbyrzcbA.mjs";
|
|
3
|
+
import { t as workersAiRestFetch } from "../workers-ai-rest-GKy2r7eG.mjs";
|
|
4
|
+
import { n as uint8ArrayToBase64, t as binaryToBase64 } from "../binary-p4H_N_3M.mjs";
|
|
5
|
+
import { BaseImageAdapter } from "@tanstack/ai/adapters";
|
|
6
|
+
//#region src/adapters/workers-ai-image.ts
|
|
7
|
+
var WorkersAiImageAdapter = class extends BaseImageAdapter {
|
|
8
|
+
constructor(config, model) {
|
|
9
|
+
super({}, model);
|
|
10
|
+
_defineProperty(this, "name", "workers-ai-image");
|
|
11
|
+
_defineProperty(this, "adapterConfig", void 0);
|
|
12
|
+
validateWorkersAiConfig(config);
|
|
13
|
+
this.adapterConfig = config;
|
|
14
|
+
}
|
|
15
|
+
async generateImages(options) {
|
|
16
|
+
const { prompt, size, modelOptions } = options;
|
|
17
|
+
const extra = { ...modelOptions };
|
|
18
|
+
if (size) {
|
|
19
|
+
const [w, h] = size.split("x");
|
|
20
|
+
if (w) extra.width = Number(w);
|
|
21
|
+
if (h) extra.height = Number(h);
|
|
22
|
+
}
|
|
23
|
+
if (isDirectBindingConfig(this.adapterConfig)) return this.generateViaBinding(prompt, extra);
|
|
24
|
+
if (isDirectCredentialsConfig(this.adapterConfig)) return this.generateViaRest(prompt, extra);
|
|
25
|
+
return this.generateViaGateway(prompt, extra);
|
|
26
|
+
}
|
|
27
|
+
async generateViaBinding(prompt, options) {
|
|
28
|
+
const b64 = await binaryToBase64(await this.adapterConfig.binding.run(this.model, {
|
|
29
|
+
prompt,
|
|
30
|
+
...options
|
|
31
|
+
}), "image");
|
|
32
|
+
return this.wrapResult(b64);
|
|
33
|
+
}
|
|
34
|
+
async generateViaRest(prompt, options) {
|
|
35
|
+
const config = this.adapterConfig;
|
|
36
|
+
const buffer = await (await workersAiRestFetch(config, this.model, {
|
|
37
|
+
prompt,
|
|
38
|
+
...options
|
|
39
|
+
}, {
|
|
40
|
+
label: "Workers AI image",
|
|
41
|
+
signal: options.signal
|
|
42
|
+
})).arrayBuffer();
|
|
43
|
+
return this.wrapResult(uint8ArrayToBase64(new Uint8Array(buffer)));
|
|
44
|
+
}
|
|
45
|
+
async generateViaGateway(prompt, options) {
|
|
46
|
+
const gatewayConfig = this.adapterConfig;
|
|
47
|
+
const response = await createGatewayFetch("workers-ai", gatewayConfig)("https://api.cloudflare.com/v1/images/generations", {
|
|
48
|
+
method: "POST",
|
|
49
|
+
body: JSON.stringify({
|
|
50
|
+
model: this.model,
|
|
51
|
+
prompt,
|
|
52
|
+
...options
|
|
53
|
+
})
|
|
54
|
+
});
|
|
55
|
+
if (!response.ok) {
|
|
56
|
+
const errorText = await response.text();
|
|
57
|
+
throw new Error(`Workers AI image gateway request failed (${response.status}): ${errorText}`);
|
|
58
|
+
}
|
|
59
|
+
const buffer = await response.arrayBuffer();
|
|
60
|
+
return this.wrapResult(uint8ArrayToBase64(new Uint8Array(buffer)));
|
|
61
|
+
}
|
|
62
|
+
/** Wrap a base64 image string into the standard ImageGenerationResult. */
|
|
63
|
+
wrapResult(b64) {
|
|
64
|
+
return {
|
|
65
|
+
id: this.generateId(),
|
|
66
|
+
model: this.model,
|
|
67
|
+
images: [{ b64Json: b64 }]
|
|
68
|
+
};
|
|
69
|
+
}
|
|
70
|
+
};
|
|
71
|
+
/**
|
|
72
|
+
* Creates a Workers AI image generation adapter.
|
|
73
|
+
*
|
|
74
|
+
* Works with TanStack AI's `generateImage()` activity function:
|
|
75
|
+
* ```ts
|
|
76
|
+
* import { generateImage } from "@tanstack/ai";
|
|
77
|
+
* import { createWorkersAiImage } from "@cloudflare/tanstack-ai";
|
|
78
|
+
*
|
|
79
|
+
* const adapter = createWorkersAiImage(
|
|
80
|
+
* "@cf/stabilityai/stable-diffusion-xl-base-1.0",
|
|
81
|
+
* { binding: env.AI },
|
|
82
|
+
* );
|
|
83
|
+
*
|
|
84
|
+
* const result = await generateImage({ adapter, prompt: "a cat in space" });
|
|
85
|
+
* ```
|
|
86
|
+
*
|
|
87
|
+
* Note: Factory takes `(model, config)` for ergonomics — the class constructor
|
|
88
|
+
* uses `(config, model)` to match TanStack AI's upstream convention.
|
|
89
|
+
*/
|
|
90
|
+
function createWorkersAiImage(model, config) {
|
|
91
|
+
return new WorkersAiImageAdapter(config, model);
|
|
92
|
+
}
|
|
93
|
+
//#endregion
|
|
94
|
+
export { WorkersAiImageAdapter, createWorkersAiImage };
|
|
95
|
+
|
|
96
|
+
//# sourceMappingURL=workers-ai-image.mjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"workers-ai-image.mjs","names":[],"sources":["../../src/adapters/workers-ai-image.ts"],"sourcesContent":["import { BaseImageAdapter } from \"@tanstack/ai/adapters\";\nimport type { ImageGenerationOptions, ImageGenerationResult } from \"@tanstack/ai\";\nimport type { AiModels, BaseAiTextToImage } from \"@cloudflare/workers-types\";\nimport {\n\ttype WorkersAiAdapterConfig,\n\ttype WorkersAiDirectBindingConfig,\n\ttype AiGatewayAdapterConfig,\n\tcreateGatewayFetch,\n\tisDirectBindingConfig,\n\tisDirectCredentialsConfig,\n\tvalidateWorkersAiConfig,\n} from \"../utils/create-fetcher\";\nimport { workersAiRestFetch } from \"../utils/workers-ai-rest\";\nimport { binaryToBase64, uint8ArrayToBase64 } from \"../utils/binary\";\nimport type { WorkersAiDirectCredentialsConfig } from \"../utils/create-fetcher\";\n\n// ---------------------------------------------------------------------------\n// Model type derived from @cloudflare/workers-types\n// ---------------------------------------------------------------------------\n\nexport type WorkersAiImageModel =\n\t| { [K in keyof AiModels]: AiModels[K] extends BaseAiTextToImage ? K : never }[keyof AiModels]\n\t| (string & {});\n\n// ---------------------------------------------------------------------------\n// WorkersAiImageAdapter: image generation via Workers AI\n// Extends BaseImageAdapter so it works with TanStack AI's generateImage()\n// ---------------------------------------------------------------------------\n\nexport class WorkersAiImageAdapter extends BaseImageAdapter<WorkersAiImageModel> {\n\treadonly name = \"workers-ai-image\" as const;\n\tprivate adapterConfig: WorkersAiAdapterConfig;\n\n\tconstructor(config: WorkersAiAdapterConfig, model: WorkersAiImageModel) {\n\t\tsuper({}, model);\n\t\tvalidateWorkersAiConfig(config);\n\t\tthis.adapterConfig = config;\n\t}\n\n\tasync generateImages(options: ImageGenerationOptions): Promise<ImageGenerationResult> {\n\t\tconst { prompt, size, modelOptions } = options;\n\t\tconst extra: Record<string, unknown> = { ...modelOptions };\n\n\t\t// Note: Workers AI Stable Diffusion models only generate a single image\n\t\t// per request — there is no multi-image parameter. `numberOfImages` from\n\t\t// the TanStack AI options is intentionally not forwarded. Use\n\t\t// `modelOptions.num_steps` to control the number of diffusion steps.\n\n\t\tif (size) {\n\t\t\tconst [w, h] = size.split(\"x\");\n\t\t\tif (w) extra.width = Number(w);\n\t\t\tif (h) extra.height = Number(h);\n\t\t}\n\n\t\tif (isDirectBindingConfig(this.adapterConfig)) {\n\t\t\treturn this.generateViaBinding(prompt, extra);\n\t\t}\n\n\t\tif (isDirectCredentialsConfig(this.adapterConfig)) {\n\t\t\treturn this.generateViaRest(prompt, extra);\n\t\t}\n\n\t\t// Gateway mode\n\t\treturn this.generateViaGateway(prompt, extra);\n\t}\n\n\tprivate async generateViaBinding(\n\t\tprompt: string,\n\t\toptions: Record<string, unknown>,\n\t): Promise<ImageGenerationResult> {\n\t\tconst ai = (this.adapterConfig as WorkersAiDirectBindingConfig).binding;\n\t\tconst result = await ai.run(this.model, { prompt, ...options });\n\n\t\tconst b64 = await binaryToBase64(result, \"image\");\n\t\treturn this.wrapResult(b64);\n\t}\n\n\tprivate async generateViaRest(\n\t\tprompt: string,\n\t\toptions: Record<string, unknown>,\n\t): Promise<ImageGenerationResult> {\n\t\tconst config = this.adapterConfig as WorkersAiDirectCredentialsConfig;\n\t\tconst response = await workersAiRestFetch(\n\t\t\tconfig,\n\t\t\tthis.model,\n\t\t\t{ prompt, ...options },\n\t\t\t{ label: \"Workers AI image\", signal: (options as { signal?: AbortSignal }).signal },\n\t\t);\n\n\t\tconst buffer = await response.arrayBuffer();\n\t\treturn this.wrapResult(uint8ArrayToBase64(new Uint8Array(buffer)));\n\t}\n\n\tprivate async generateViaGateway(\n\t\tprompt: string,\n\t\toptions: Record<string, unknown>,\n\t): Promise<ImageGenerationResult> {\n\t\tconst gatewayConfig = this.adapterConfig as AiGatewayAdapterConfig;\n\t\tconst gatewayFetch = createGatewayFetch(\"workers-ai\", gatewayConfig);\n\n\t\t// The URL here is a placeholder — createGatewayFetch for \"workers-ai\" extracts\n\t\t// the model from the body, sets it as the endpoint, and routes through the gateway.\n\t\t// The actual URL path is not used.\n\t\tconst response = await gatewayFetch(\"https://api.cloudflare.com/v1/images/generations\", {\n\t\t\tmethod: \"POST\",\n\t\t\tbody: JSON.stringify({\n\t\t\t\tmodel: this.model,\n\t\t\t\tprompt,\n\t\t\t\t...options,\n\t\t\t}),\n\t\t});\n\n\t\tif (!response.ok) {\n\t\t\tconst errorText = await response.text();\n\t\t\tthrow new Error(\n\t\t\t\t`Workers AI image gateway request failed (${response.status}): ${errorText}`,\n\t\t\t);\n\t\t}\n\n\t\tconst buffer = await response.arrayBuffer();\n\t\treturn this.wrapResult(uint8ArrayToBase64(new Uint8Array(buffer)));\n\t}\n\n\t/** Wrap a base64 image string into the standard ImageGenerationResult. */\n\tprivate wrapResult(b64: string): ImageGenerationResult {\n\t\treturn {\n\t\t\tid: this.generateId(),\n\t\t\tmodel: this.model,\n\t\t\timages: [{ b64Json: b64 }],\n\t\t};\n\t}\n}\n\n// ---------------------------------------------------------------------------\n// Factory function\n// ---------------------------------------------------------------------------\n\n/**\n * Creates a Workers AI image generation adapter.\n *\n * Works with TanStack AI's `generateImage()` activity function:\n * ```ts\n * import { generateImage } from \"@tanstack/ai\";\n * import { createWorkersAiImage } from \"@cloudflare/tanstack-ai\";\n *\n * const adapter = createWorkersAiImage(\n * \"@cf/stabilityai/stable-diffusion-xl-base-1.0\",\n * { binding: env.AI },\n * );\n *\n * const result = await generateImage({ adapter, prompt: \"a cat in space\" });\n * ```\n *\n * Note: Factory takes `(model, config)` for ergonomics — the class constructor\n * uses `(config, model)` to match TanStack AI's upstream convention.\n */\nexport function createWorkersAiImage(model: WorkersAiImageModel, config: WorkersAiAdapterConfig) {\n\treturn new WorkersAiImageAdapter(config, model);\n}\n"],"mappings":";;;;;;AA6BA,IAAa,wBAAb,cAA2C,iBAAsC;CAIhF,YAAY,QAAgC,OAA4B;AACvE,QAAM,EAAE,EAAE,MAAM;wBAJR,QAAO,mBAA4B;wBACpC,iBAAA,KAAA,EAAsC;AAI7C,0BAAwB,OAAO;AAC/B,OAAK,gBAAgB;;CAGtB,MAAM,eAAe,SAAiE;EACrF,MAAM,EAAE,QAAQ,MAAM,iBAAiB;EACvC,MAAM,QAAiC,EAAE,GAAG,cAAc;AAO1D,MAAI,MAAM;GACT,MAAM,CAAC,GAAG,KAAK,KAAK,MAAM,IAAI;AAC9B,OAAI,EAAG,OAAM,QAAQ,OAAO,EAAE;AAC9B,OAAI,EAAG,OAAM,SAAS,OAAO,EAAE;;AAGhC,MAAI,sBAAsB,KAAK,cAAc,CAC5C,QAAO,KAAK,mBAAmB,QAAQ,MAAM;AAG9C,MAAI,0BAA0B,KAAK,cAAc,CAChD,QAAO,KAAK,gBAAgB,QAAQ,MAAM;AAI3C,SAAO,KAAK,mBAAmB,QAAQ,MAAM;;CAG9C,MAAc,mBACb,QACA,SACiC;EAIjC,MAAM,MAAM,MAAM,eAFH,MADH,KAAK,cAA+C,QACxC,IAAI,KAAK,OAAO;GAAE;GAAQ,GAAG;GAAS,CAAC,EAEtB,QAAQ;AACjD,SAAO,KAAK,WAAW,IAAI;;CAG5B,MAAc,gBACb,QACA,SACiC;EACjC,MAAM,SAAS,KAAK;EAQpB,MAAM,SAAS,OAPE,MAAM,mBACtB,QACA,KAAK,OACL;GAAE;GAAQ,GAAG;GAAS,EACtB;GAAE,OAAO;GAAoB,QAAS,QAAqC;GAAQ,CACnF,EAE6B,aAAa;AAC3C,SAAO,KAAK,WAAW,mBAAmB,IAAI,WAAW,OAAO,CAAC,CAAC;;CAGnE,MAAc,mBACb,QACA,SACiC;EACjC,MAAM,gBAAgB,KAAK;EAM3B,MAAM,WAAW,MALI,mBAAmB,cAAc,cAAc,CAKhC,oDAAoD;GACvF,QAAQ;GACR,MAAM,KAAK,UAAU;IACpB,OAAO,KAAK;IACZ;IACA,GAAG;IACH,CAAC;GACF,CAAC;AAEF,MAAI,CAAC,SAAS,IAAI;GACjB,MAAM,YAAY,MAAM,SAAS,MAAM;AACvC,SAAM,IAAI,MACT,4CAA4C,SAAS,OAAO,KAAK,YACjE;;EAGF,MAAM,SAAS,MAAM,SAAS,aAAa;AAC3C,SAAO,KAAK,WAAW,mBAAmB,IAAI,WAAW,OAAO,CAAC,CAAC;;;CAInE,WAAmB,KAAoC;AACtD,SAAO;GACN,IAAI,KAAK,YAAY;GACrB,OAAO,KAAK;GACZ,QAAQ,CAAC,EAAE,SAAS,KAAK,CAAC;GAC1B;;;;;;;;;;;;;;;;;;;;;;AA2BH,SAAgB,qBAAqB,OAA4B,QAAgC;AAChG,QAAO,IAAI,sBAAsB,QAAQ,MAAM"}
|
|
@@ -1,12 +1,88 @@
|
|
|
1
|
-
|
|
1
|
+
Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
|
|
2
|
+
require("../workers-ai-Bm7Up4or.cjs");
|
|
3
|
+
const require_create_fetcher = require("../create-fetcher-5iL34e6H.cjs");
|
|
4
|
+
const require_defineProperty = require("../defineProperty-DQoAg20E.cjs");
|
|
5
|
+
const require_workers_ai_rest = require("../workers-ai-rest-CkNCtBwv.cjs");
|
|
6
|
+
let _tanstack_ai_adapters = require("@tanstack/ai/adapters");
|
|
7
|
+
//#region src/adapters/workers-ai-summarize.ts
|
|
8
|
+
var WorkersAiSummarizeAdapter = class extends _tanstack_ai_adapters.BaseSummarizeAdapter {
|
|
9
|
+
constructor(config, model) {
|
|
10
|
+
super({}, model);
|
|
11
|
+
require_defineProperty._defineProperty(this, "name", "workers-ai-summarize");
|
|
12
|
+
require_defineProperty._defineProperty(this, "adapterConfig", void 0);
|
|
13
|
+
require_create_fetcher.validateWorkersAiConfig(config);
|
|
14
|
+
this.adapterConfig = config;
|
|
15
|
+
}
|
|
16
|
+
async summarize(options) {
|
|
17
|
+
const { text, maxLength } = options;
|
|
18
|
+
const payload = { input_text: text };
|
|
19
|
+
if (maxLength != null) payload.max_length = maxLength;
|
|
20
|
+
if (require_create_fetcher.isDirectBindingConfig(this.adapterConfig)) return this.summarizeViaBinding(payload);
|
|
21
|
+
if (require_create_fetcher.isDirectCredentialsConfig(this.adapterConfig)) return this.summarizeViaRest(payload);
|
|
22
|
+
return this.summarizeViaGateway(payload);
|
|
23
|
+
}
|
|
24
|
+
async summarizeViaBinding(payload) {
|
|
25
|
+
const result = await this.adapterConfig.binding.run(this.model, payload);
|
|
26
|
+
return this.wrapResult(result.summary ?? "");
|
|
27
|
+
}
|
|
28
|
+
async summarizeViaRest(payload) {
|
|
29
|
+
const config = this.adapterConfig;
|
|
30
|
+
const data = await (await require_workers_ai_rest.workersAiRestFetch(config, this.model, payload, { label: "Workers AI summarize" })).json();
|
|
31
|
+
return this.wrapResult(data.result?.summary ?? "");
|
|
32
|
+
}
|
|
33
|
+
async summarizeViaGateway(payload) {
|
|
34
|
+
const gatewayConfig = this.adapterConfig;
|
|
35
|
+
const response = await require_create_fetcher.createGatewayFetch("workers-ai", gatewayConfig)("https://api.cloudflare.com/v1/ai/summarization", {
|
|
36
|
+
method: "POST",
|
|
37
|
+
body: JSON.stringify({
|
|
38
|
+
model: this.model,
|
|
39
|
+
...payload
|
|
40
|
+
})
|
|
41
|
+
});
|
|
42
|
+
if (!response.ok) {
|
|
43
|
+
const errorText = await response.text();
|
|
44
|
+
throw new Error(`Workers AI summarize gateway request failed (${response.status}): ${errorText}`);
|
|
45
|
+
}
|
|
46
|
+
const data = await response.json();
|
|
47
|
+
return this.wrapResult(data.result?.summary ?? data.summary ?? "");
|
|
48
|
+
}
|
|
49
|
+
wrapResult(summary) {
|
|
50
|
+
return {
|
|
51
|
+
id: this.generateId(),
|
|
52
|
+
model: this.model,
|
|
53
|
+
summary,
|
|
54
|
+
usage: {
|
|
55
|
+
promptTokens: 0,
|
|
56
|
+
completionTokens: 0,
|
|
57
|
+
totalTokens: 0
|
|
58
|
+
}
|
|
59
|
+
};
|
|
60
|
+
}
|
|
61
|
+
};
|
|
62
|
+
/**
|
|
63
|
+
* Creates a Workers AI summarization adapter.
|
|
64
|
+
*
|
|
65
|
+
* Works with TanStack AI's `summarize()` activity function:
|
|
66
|
+
* ```ts
|
|
67
|
+
* import { summarize } from "@tanstack/ai";
|
|
68
|
+
* import { createWorkersAiSummarize } from "@cloudflare/tanstack-ai";
|
|
69
|
+
*
|
|
70
|
+
* const adapter = createWorkersAiSummarize("@cf/facebook/bart-large-cnn", {
|
|
71
|
+
* binding: env.AI,
|
|
72
|
+
* });
|
|
73
|
+
*
|
|
74
|
+
* const result = await summarize({ adapter, text: "Long article here..." });
|
|
75
|
+
* // result.summary
|
|
76
|
+
* ```
|
|
77
|
+
*
|
|
78
|
+
* Note: Factory takes `(model, config)` for ergonomics — the class constructor
|
|
79
|
+
* uses `(config, model)` to match TanStack AI's upstream convention.
|
|
80
|
+
*/
|
|
81
|
+
function createWorkersAiSummarize(model, config) {
|
|
82
|
+
return new WorkersAiSummarizeAdapter(config, model);
|
|
83
|
+
}
|
|
84
|
+
//#endregion
|
|
85
|
+
exports.WorkersAiSummarizeAdapter = WorkersAiSummarizeAdapter;
|
|
86
|
+
exports.createWorkersAiSummarize = createWorkersAiSummarize;
|
|
2
87
|
|
|
3
|
-
|
|
4
|
-
var _chunkZFB72L57cjs = require('../chunk-ZFB72L57.cjs');
|
|
5
|
-
require('../chunk-7HSUHP63.cjs');
|
|
6
|
-
require('../chunk-WBTQKWFN.cjs');
|
|
7
|
-
require('../chunk-4DE2IREA.cjs');
|
|
8
|
-
|
|
9
|
-
|
|
10
|
-
|
|
11
|
-
exports.WorkersAiSummarizeAdapter = _chunkZFB72L57cjs.WorkersAiSummarizeAdapter; exports.createWorkersAiSummarize = _chunkZFB72L57cjs.createWorkersAiSummarize;
|
|
12
88
|
//# sourceMappingURL=workers-ai-summarize.cjs.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"sources":["/
|
|
1
|
+
{"version":3,"file":"workers-ai-summarize.cjs","names":["BaseSummarizeAdapter","isDirectBindingConfig","isDirectCredentialsConfig","workersAiRestFetch","createGatewayFetch"],"sources":["../../src/adapters/workers-ai-summarize.ts"],"sourcesContent":["import { BaseSummarizeAdapter } from \"@tanstack/ai/adapters\";\nimport type { SummarizationOptions, SummarizationResult } from \"@tanstack/ai\";\nimport {\n\ttype WorkersAiAdapterConfig,\n\ttype WorkersAiDirectBindingConfig,\n\ttype WorkersAiDirectCredentialsConfig,\n\ttype AiGatewayAdapterConfig,\n\tcreateGatewayFetch,\n\tisDirectBindingConfig,\n\tisDirectCredentialsConfig,\n\tvalidateWorkersAiConfig,\n} from \"../utils/create-fetcher\";\nimport { workersAiRestFetch } from \"../utils/workers-ai-rest\";\n\n// ---------------------------------------------------------------------------\n// Model types\n// ---------------------------------------------------------------------------\n\n/**\n * Workers AI models that support summarization.\n */\nexport type WorkersAiSummarizeModel = \"@cf/facebook/bart-large-cnn\" | (string & {});\n\n// ---------------------------------------------------------------------------\n// WorkersAiSummarizeAdapter\n// ---------------------------------------------------------------------------\n\nexport class WorkersAiSummarizeAdapter extends BaseSummarizeAdapter<WorkersAiSummarizeModel> {\n\treadonly name = \"workers-ai-summarize\" as const;\n\tprivate adapterConfig: WorkersAiAdapterConfig;\n\n\tconstructor(config: WorkersAiAdapterConfig, model: WorkersAiSummarizeModel) {\n\t\tsuper({}, model);\n\t\tvalidateWorkersAiConfig(config);\n\t\tthis.adapterConfig = config;\n\t}\n\n\tasync summarize(options: SummarizationOptions): Promise<SummarizationResult> {\n\t\tconst { text, maxLength } = options;\n\n\t\tconst payload: Record<string, unknown> = { input_text: text };\n\t\tif (maxLength != null) payload.max_length = maxLength;\n\n\t\tif (isDirectBindingConfig(this.adapterConfig)) {\n\t\t\treturn this.summarizeViaBinding(payload);\n\t\t}\n\n\t\tif (isDirectCredentialsConfig(this.adapterConfig)) {\n\t\t\treturn this.summarizeViaRest(payload);\n\t\t}\n\n\t\treturn this.summarizeViaGateway(payload);\n\t}\n\n\tprivate async summarizeViaBinding(\n\t\tpayload: Record<string, unknown>,\n\t): Promise<SummarizationResult> {\n\t\tconst ai = (this.adapterConfig as WorkersAiDirectBindingConfig).binding;\n\t\tconst result = (await ai.run(this.model, payload)) as Record<string, unknown>;\n\t\treturn this.wrapResult((result.summary as string) ?? \"\");\n\t}\n\n\tprivate async summarizeViaRest(payload: Record<string, unknown>): Promise<SummarizationResult> {\n\t\tconst config = this.adapterConfig as WorkersAiDirectCredentialsConfig;\n\t\tconst response = await workersAiRestFetch(config, this.model, payload, {\n\t\t\tlabel: \"Workers AI summarize\",\n\t\t});\n\n\t\tconst data = (await response.json()) as { result?: { summary?: string } };\n\t\treturn this.wrapResult(data.result?.summary ?? \"\");\n\t}\n\n\tprivate async summarizeViaGateway(\n\t\tpayload: Record<string, unknown>,\n\t): Promise<SummarizationResult> {\n\t\tconst gatewayConfig = this.adapterConfig as AiGatewayAdapterConfig;\n\t\tconst gatewayFetch = createGatewayFetch(\"workers-ai\", gatewayConfig);\n\n\t\t// The URL here is a placeholder — createGatewayFetch for \"workers-ai\" extracts\n\t\t// the model from the body, sets it as the endpoint, and routes through the gateway.\n\t\t// The actual URL path is not used.\n\t\tconst response = await gatewayFetch(\"https://api.cloudflare.com/v1/ai/summarization\", {\n\t\t\tmethod: \"POST\",\n\t\t\tbody: JSON.stringify({\n\t\t\t\tmodel: this.model,\n\t\t\t\t...payload,\n\t\t\t}),\n\t\t});\n\n\t\tif (!response.ok) {\n\t\t\tconst errorText = await response.text();\n\t\t\tthrow new Error(\n\t\t\t\t`Workers AI summarize gateway request failed (${response.status}): ${errorText}`,\n\t\t\t);\n\t\t}\n\n\t\tconst data = (await response.json()) as { result?: { summary?: string }; summary?: string };\n\t\treturn this.wrapResult(data.result?.summary ?? data.summary ?? \"\");\n\t}\n\n\tprivate wrapResult(summary: string): SummarizationResult {\n\t\treturn {\n\t\t\tid: this.generateId(),\n\t\t\tmodel: this.model,\n\t\t\tsummary,\n\t\t\t// BART-large-CNN doesn't return token usage\n\t\t\tusage: { promptTokens: 0, completionTokens: 0, totalTokens: 0 },\n\t\t};\n\t}\n}\n\n// ---------------------------------------------------------------------------\n// Factory function\n// ---------------------------------------------------------------------------\n\n/**\n * Creates a Workers AI summarization adapter.\n *\n * Works with TanStack AI's `summarize()` activity function:\n * ```ts\n * import { summarize } from \"@tanstack/ai\";\n * import { createWorkersAiSummarize } from \"@cloudflare/tanstack-ai\";\n *\n * const adapter = createWorkersAiSummarize(\"@cf/facebook/bart-large-cnn\", {\n * binding: env.AI,\n * });\n *\n * const result = await summarize({ adapter, text: \"Long article here...\" });\n * // result.summary\n * ```\n *\n * Note: Factory takes `(model, config)` for ergonomics — the class constructor\n * uses `(config, model)` to match TanStack AI's upstream convention.\n */\nexport function createWorkersAiSummarize(\n\tmodel: WorkersAiSummarizeModel,\n\tconfig: WorkersAiAdapterConfig,\n) {\n\treturn new WorkersAiSummarizeAdapter(config, model);\n}\n"],"mappings":";;;;;;;AA2BA,IAAa,4BAAb,cAA+CA,sBAAAA,qBAA8C;CAI5F,YAAY,QAAgC,OAAgC;AAC3E,QAAM,EAAE,EAAE,MAAM;+CAJR,QAAO,uBAAgC;+CACxC,iBAAA,KAAA,EAAsC;AAI7C,yBAAA,wBAAwB,OAAO;AAC/B,OAAK,gBAAgB;;CAGtB,MAAM,UAAU,SAA6D;EAC5E,MAAM,EAAE,MAAM,cAAc;EAE5B,MAAM,UAAmC,EAAE,YAAY,MAAM;AAC7D,MAAI,aAAa,KAAM,SAAQ,aAAa;AAE5C,MAAIC,uBAAAA,sBAAsB,KAAK,cAAc,CAC5C,QAAO,KAAK,oBAAoB,QAAQ;AAGzC,MAAIC,uBAAAA,0BAA0B,KAAK,cAAc,CAChD,QAAO,KAAK,iBAAiB,QAAQ;AAGtC,SAAO,KAAK,oBAAoB,QAAQ;;CAGzC,MAAc,oBACb,SAC+B;EAE/B,MAAM,SAAU,MADJ,KAAK,cAA+C,QACvC,IAAI,KAAK,OAAO,QAAQ;AACjD,SAAO,KAAK,WAAY,OAAO,WAAsB,GAAG;;CAGzD,MAAc,iBAAiB,SAAgE;EAC9F,MAAM,SAAS,KAAK;EAKpB,MAAM,OAAQ,OAJG,MAAMC,wBAAAA,mBAAmB,QAAQ,KAAK,OAAO,SAAS,EACtE,OAAO,wBACP,CAAC,EAE2B,MAAM;AACnC,SAAO,KAAK,WAAW,KAAK,QAAQ,WAAW,GAAG;;CAGnD,MAAc,oBACb,SAC+B;EAC/B,MAAM,gBAAgB,KAAK;EAM3B,MAAM,WAAW,MALIC,uBAAAA,mBAAmB,cAAc,cAAc,CAKhC,kDAAkD;GACrF,QAAQ;GACR,MAAM,KAAK,UAAU;IACpB,OAAO,KAAK;IACZ,GAAG;IACH,CAAC;GACF,CAAC;AAEF,MAAI,CAAC,SAAS,IAAI;GACjB,MAAM,YAAY,MAAM,SAAS,MAAM;AACvC,SAAM,IAAI,MACT,gDAAgD,SAAS,OAAO,KAAK,YACrE;;EAGF,MAAM,OAAQ,MAAM,SAAS,MAAM;AACnC,SAAO,KAAK,WAAW,KAAK,QAAQ,WAAW,KAAK,WAAW,GAAG;;CAGnE,WAAmB,SAAsC;AACxD,SAAO;GACN,IAAI,KAAK,YAAY;GACrB,OAAO,KAAK;GACZ;GAEA,OAAO;IAAE,cAAc;IAAG,kBAAkB;IAAG,aAAa;IAAG;GAC/D;;;;;;;;;;;;;;;;;;;;;;AA2BH,SAAgB,yBACf,OACA,QACC;AACD,QAAO,IAAI,0BAA0B,QAAQ,MAAM"}
|
|
@@ -1,3 +1,42 @@
|
|
|
1
|
-
|
|
2
|
-
|
|
3
|
-
|
|
1
|
+
import { i as WorkersAiAdapterConfig } from "../create-fetcher-vAQ8WW-p.cjs";
|
|
2
|
+
import { SummarizationOptions, SummarizationResult } from "@tanstack/ai";
|
|
3
|
+
import { BaseSummarizeAdapter } from "@tanstack/ai/adapters";
|
|
4
|
+
|
|
5
|
+
//#region src/adapters/workers-ai-summarize.d.ts
|
|
6
|
+
/**
|
|
7
|
+
* Workers AI models that support summarization.
|
|
8
|
+
*/
|
|
9
|
+
type WorkersAiSummarizeModel = "@cf/facebook/bart-large-cnn" | (string & {});
|
|
10
|
+
declare class WorkersAiSummarizeAdapter extends BaseSummarizeAdapter<WorkersAiSummarizeModel> {
|
|
11
|
+
readonly name: "workers-ai-summarize";
|
|
12
|
+
private adapterConfig;
|
|
13
|
+
constructor(config: WorkersAiAdapterConfig, model: WorkersAiSummarizeModel);
|
|
14
|
+
summarize(options: SummarizationOptions): Promise<SummarizationResult>;
|
|
15
|
+
private summarizeViaBinding;
|
|
16
|
+
private summarizeViaRest;
|
|
17
|
+
private summarizeViaGateway;
|
|
18
|
+
private wrapResult;
|
|
19
|
+
}
|
|
20
|
+
/**
|
|
21
|
+
* Creates a Workers AI summarization adapter.
|
|
22
|
+
*
|
|
23
|
+
* Works with TanStack AI's `summarize()` activity function:
|
|
24
|
+
* ```ts
|
|
25
|
+
* import { summarize } from "@tanstack/ai";
|
|
26
|
+
* import { createWorkersAiSummarize } from "@cloudflare/tanstack-ai";
|
|
27
|
+
*
|
|
28
|
+
* const adapter = createWorkersAiSummarize("@cf/facebook/bart-large-cnn", {
|
|
29
|
+
* binding: env.AI,
|
|
30
|
+
* });
|
|
31
|
+
*
|
|
32
|
+
* const result = await summarize({ adapter, text: "Long article here..." });
|
|
33
|
+
* // result.summary
|
|
34
|
+
* ```
|
|
35
|
+
*
|
|
36
|
+
* Note: Factory takes `(model, config)` for ergonomics — the class constructor
|
|
37
|
+
* uses `(config, model)` to match TanStack AI's upstream convention.
|
|
38
|
+
*/
|
|
39
|
+
declare function createWorkersAiSummarize(model: WorkersAiSummarizeModel, config: WorkersAiAdapterConfig): WorkersAiSummarizeAdapter;
|
|
40
|
+
//#endregion
|
|
41
|
+
export { WorkersAiSummarizeAdapter, WorkersAiSummarizeModel, createWorkersAiSummarize };
|
|
42
|
+
//# sourceMappingURL=workers-ai-summarize.d.cts.map
|