@better-agent/workers-ai 0.2.0-beta.1
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/dist/index.d.mts +53 -0
- package/dist/index.d.mts.map +1 -0
- package/dist/index.mjs +85 -0
- package/dist/index.mjs.map +1 -0
- package/package.json +55 -0
package/dist/index.d.mts
ADDED
|
@@ -0,0 +1,53 @@
|
|
|
1
|
+
import { createWorkersAI as createWorkersAI$1 } from "workers-ai-provider";
|
|
2
|
+
import { AgentCapabilities, AgentModel, EmbeddingGenerationModel, ImageGenerationModel, SpeechGenerationModel, TextGenerationModel, TranscriptionGenerationModel } from "@better-agent/core";
|
|
3
|
+
|
|
4
|
+
//#region src/model.d.ts
|
|
5
|
+
type CreateWorkersAIOptions = Parameters<typeof createWorkersAI$1>[0];
|
|
6
|
+
type AiSdkWorkersAIProvider = ReturnType<typeof createWorkersAI$1>;
|
|
7
|
+
type WorkersAIProviderOptions = {
|
|
8
|
+
"workers-ai"?: Record<string, unknown>;
|
|
9
|
+
};
|
|
10
|
+
type WorkersAITextProviderOptions = WorkersAIProviderOptions;
|
|
11
|
+
type WorkersAIEmbeddingProviderOptions = {
|
|
12
|
+
"workers-ai"?: Record<string, unknown>;
|
|
13
|
+
};
|
|
14
|
+
type WorkersAIImageProviderOptions = {
|
|
15
|
+
"workers-ai"?: Record<string, unknown>;
|
|
16
|
+
};
|
|
17
|
+
type WorkersAISpeechProviderOptions = {
|
|
18
|
+
"workers-ai"?: Record<string, unknown>;
|
|
19
|
+
};
|
|
20
|
+
type WorkersAITranscriptionProviderOptions = {
|
|
21
|
+
"workers-ai"?: Record<string, unknown>;
|
|
22
|
+
};
|
|
23
|
+
type WorkersAIGenerationProviderOptions = WorkersAIProviderOptions | WorkersAITextProviderOptions | WorkersAIEmbeddingProviderOptions | WorkersAIImageProviderOptions | WorkersAISpeechProviderOptions | WorkersAITranscriptionProviderOptions;
|
|
24
|
+
type WorkersAIModelId = Parameters<AiSdkWorkersAIProvider>[0];
|
|
25
|
+
type WorkersAITextModelId = WorkersAIModelId;
|
|
26
|
+
type WorkersAIEmbeddingModelId = Parameters<AiSdkWorkersAIProvider["textEmbedding"]>[0];
|
|
27
|
+
type WorkersAIImageModelId = Parameters<AiSdkWorkersAIProvider["image"]>[0];
|
|
28
|
+
type WorkersAISpeechModelId = Parameters<AiSdkWorkersAIProvider["speech"]>[0];
|
|
29
|
+
type WorkersAITranscriptionModelId = Parameters<AiSdkWorkersAIProvider["transcription"]>[0];
|
|
30
|
+
type WorkersAITextOptions = NonNullable<Parameters<AiSdkWorkersAIProvider>[1]>;
|
|
31
|
+
type WorkersAIEmbeddingOptions = NonNullable<Parameters<AiSdkWorkersAIProvider["textEmbedding"]>[1]>;
|
|
32
|
+
type WorkersAIImageOptions = NonNullable<Parameters<AiSdkWorkersAIProvider["image"]>[1]>;
|
|
33
|
+
type WorkersAISpeechOptions = NonNullable<Parameters<AiSdkWorkersAIProvider["speech"]>[1]>;
|
|
34
|
+
type WorkersAITranscriptionOptions = NonNullable<Parameters<AiSdkWorkersAIProvider["transcription"]>[1]>;
|
|
35
|
+
type WorkersAIModelOptions = WorkersAITextOptions;
|
|
36
|
+
type WorkersAIModel<TModelId extends WorkersAIModelId = WorkersAIModelId> = AgentModel<AgentCapabilities, undefined, WorkersAIProviderOptions, TModelId>;
|
|
37
|
+
type WorkersAITextModel<TModelId extends WorkersAITextModelId = WorkersAITextModelId> = TextGenerationModel<WorkersAITextProviderOptions, TModelId>;
|
|
38
|
+
type WorkersAIEmbeddingModel<TModelId extends WorkersAIEmbeddingModelId = WorkersAIEmbeddingModelId> = EmbeddingGenerationModel<WorkersAIEmbeddingProviderOptions, TModelId>;
|
|
39
|
+
type WorkersAIImageModel<TModelId extends WorkersAIImageModelId = WorkersAIImageModelId> = ImageGenerationModel<WorkersAIImageProviderOptions, TModelId>;
|
|
40
|
+
type WorkersAISpeechModel<TModelId extends WorkersAISpeechModelId = WorkersAISpeechModelId> = SpeechGenerationModel<WorkersAISpeechProviderOptions, TModelId>;
|
|
41
|
+
type WorkersAITranscriptionModel<TModelId extends WorkersAITranscriptionModelId = WorkersAITranscriptionModelId> = TranscriptionGenerationModel<WorkersAITranscriptionProviderOptions, TModelId>;
|
|
42
|
+
type WorkersAIProvider = {
|
|
43
|
+
<TModelId extends WorkersAIModelId>(modelId: TModelId, options?: WorkersAIModelOptions): WorkersAIModel<TModelId>;
|
|
44
|
+
text<TModelId extends WorkersAITextModelId>(modelId: TModelId, options?: WorkersAITextOptions): WorkersAITextModel<TModelId>;
|
|
45
|
+
embedding<TModelId extends WorkersAIEmbeddingModelId>(modelId: TModelId, options?: WorkersAIEmbeddingOptions): WorkersAIEmbeddingModel<TModelId>;
|
|
46
|
+
image<TModelId extends WorkersAIImageModelId>(modelId: TModelId, options?: WorkersAIImageOptions): WorkersAIImageModel<TModelId>;
|
|
47
|
+
speech<TModelId extends WorkersAISpeechModelId>(modelId: TModelId, options?: WorkersAISpeechOptions): WorkersAISpeechModel<TModelId>;
|
|
48
|
+
transcription<TModelId extends WorkersAITranscriptionModelId>(modelId: TModelId, options?: WorkersAITranscriptionOptions): WorkersAITranscriptionModel<TModelId>;
|
|
49
|
+
};
|
|
50
|
+
declare const createWorkersAI: (options: CreateWorkersAIOptions) => WorkersAIProvider;
|
|
51
|
+
//#endregion
|
|
52
|
+
export { type CreateWorkersAIOptions, type WorkersAIEmbeddingModel, type WorkersAIEmbeddingModelId, type WorkersAIEmbeddingOptions, type WorkersAIEmbeddingProviderOptions, type WorkersAIGenerationProviderOptions, type WorkersAIImageModel, type WorkersAIImageModelId, type WorkersAIImageOptions, type WorkersAIImageProviderOptions, type WorkersAIModel, type WorkersAIModelId, type WorkersAIModelOptions, type WorkersAIProvider, type WorkersAIProviderOptions, type WorkersAISpeechModel, type WorkersAISpeechModelId, type WorkersAISpeechOptions, type WorkersAISpeechProviderOptions, type WorkersAITextModel, type WorkersAITextModelId, type WorkersAITextOptions, type WorkersAITextProviderOptions, type WorkersAITranscriptionModel, type WorkersAITranscriptionModelId, type WorkersAITranscriptionOptions, type WorkersAITranscriptionProviderOptions, createWorkersAI };
|
|
53
|
+
//# sourceMappingURL=index.d.mts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.mts","names":[],"sources":["../src/model.ts"],"mappings":";;;;KAmBY,sBAAA,GAAyB,UAAA,QAAkB,iBAAA;AAAA,KAElD,sBAAA,GAAyB,UAAA,QAAkB,iBAAA;AAAA,KAEpC,wBAAA;EACR,YAAA,GAAe,MAAA;AAAA;AAAA,KAEP,4BAAA,GAA+B,wBAAA;AAAA,KAC/B,iCAAA;EACR,YAAA,GAAe,MAAA;AAAA;AAAA,KAEP,6BAAA;EACR,YAAA,GAAe,MAAA;AAAA;AAAA,KAEP,8BAAA;EACR,YAAA,GAAe,MAAA;AAAA;AAAA,KAEP,qCAAA;EACR,YAAA,GAAe,MAAA;AAAA;AAAA,KAEP,kCAAA,GACN,wBAAA,GACA,4BAAA,GACA,iCAAA,GACA,6BAAA,GACA,8BAAA,GACA,qCAAA;AAAA,KAEM,gBAAA,GAAmB,UAAA,CAAW,sBAAA;AAAA,KAC9B,oBAAA,GAAuB,gBAAA;AAAA,KACvB,yBAAA,GAA4B,UAAA,CAAW,sBAAA;AAAA,KACvC,qBAAA,GAAwB,UAAA,CAAW,sBAAA;AAAA,KACnC,sBAAA,GAAyB,UAAA,CAAW,sBAAA;AAAA,KACpC,6BAAA,GAAgC,UAAA,CAAW,sBAAA;AAAA,KAC3C,oBAAA,GAAuB,WAAA,CAAY,UAAA,CAAW,sBAAA;AAAA,KAC9C,yBAAA,GAA4B,WAAA,CACpC,UAAA,CAAW,sBAAA;AAAA,KAEH,qBAAA,GAAwB,WAAA,CAAY,UAAA,CAAW,sBAAA;AAAA,KAC/C,sBAAA,GAAyB,WAAA,CAAY,UAAA,CAAW,sBAAA;AAAA,KAChD,6BAAA,GAAgC,WAAA,CACxC,UAAA,CAAW,sBAAA;AAAA,KAGH,qBAAA,GAAwB,oBAAA;AAAA,KAExB,cAAA,kBAAgC,gBAAA,GAAmB,gBAAA,IAAoB,UAAA,CAC/E,iBAAA,aAEA,wBAAA,EACA,QAAA;AAAA,KAEQ,kBAAA,kBAAoC,oBAAA,GAAuB,oBAAA,IACnE,mBAAA,CAAoB,4BAAA,EAA8B,QAAA;AAAA,KAC1C,uBAAA,kBACS,yBAAA,GAA4B,yBAAA,IAC7C,wBAAA,CAAyB,iCAAA,EAAmC,QAAA;AAAA,KACpD,mBAAA,kBAAqC,qBAAA,GAAwB,qBAAA,IACrE,oBAAA,CAAqB,6BAAA,EAA+B,QAAA;AAAA,KAC5C,oBAAA,kBAAsC,sBAAA,GAAyB,sBAAA,IACvE,qBAAA,CAAsB,8BAAA,EAAgC,QAAA;AAAA,KAC9C,2BAAA,kBACS,6BAAA,GAAgC,6BAAA,IACjD,4BAAA,CAA6B,qCAAA,EAAuC,QAAA;AAAA,KAE5D,iBAAA;EAAA,kBACU,gBAAA,EACd,OAAA,EAAS,QAAA,EACT,OAAA,GAAU,qBAAA,GACX,cAAA,CAAe,QAAA;EAClB,IAAA,kBAAsB,oBAAA,EAClB,OAAA,EAAS,QAAA,EACT,OAAA,GAAU,oBAAA,GACX,kBAAA,CAAmB,QAAA;EACtB,SAAA,kBAA2B,yBAAA,EACvB,OAAA,EAAS,QAAA,EACT,OAAA,GAAU,yBAAA,GACX,uBAAA,CAAwB,QAAA;EAC3B,KAAA,kBAAuB,qBAAA,EACnB,OAAA,EAAS,QAAA,EACT,OAAA,GAAU,qBAAA,GACX,mBAAA,CAAoB,QAAA;EACvB,MAAA,kBAAwB,sBAAA,EACpB,OAAA,EAAS,QAAA,EACT,OAAA,GAAU,sBAAA,GACX,oBAAA,CAAqB,QAAA;EACxB,aAAA,kBAA+B,6BAAA,EAC3B,OAAA,EAAS,QAAA,EACT,OAAA,GAAU,6BAAA,GACX,2BAAA,CAA4B,QAAA;AAAA;AAAA,cA8ItB,eAAA,GAAe,OAAA,EAAa,sBAAA,KAAyB,iBAAA"}
|
package/dist/index.mjs
ADDED
|
@@ -0,0 +1,85 @@
|
|
|
1
|
+
import { aiSdkEmbeddingModel, aiSdkImageModel, aiSdkModel, aiSdkSpeechModel, aiSdkTextModel, aiSdkTranscriptionModel } from "@better-agent/ai-sdk";
|
|
2
|
+
import { createWorkersAI as createWorkersAI$1 } from "workers-ai-provider";
|
|
3
|
+
|
|
4
|
+
//#region src/model.ts
|
|
5
|
+
const workersAICapabilities = {
|
|
6
|
+
identity: { provider: "workers-ai" },
|
|
7
|
+
transport: { streaming: true },
|
|
8
|
+
output: {
|
|
9
|
+
structuredOutput: true,
|
|
10
|
+
supportedMimeTypes: ["text/plain", "application/json"]
|
|
11
|
+
},
|
|
12
|
+
multimodal: { input: { image: true } },
|
|
13
|
+
tools: { supported: true }
|
|
14
|
+
};
|
|
15
|
+
const createAgentModel = (provider, modelId, options) => {
|
|
16
|
+
return aiSdkModel({
|
|
17
|
+
model: provider(modelId, options),
|
|
18
|
+
providerId: "workers-ai",
|
|
19
|
+
modelId,
|
|
20
|
+
capabilities: workersAICapabilities
|
|
21
|
+
});
|
|
22
|
+
};
|
|
23
|
+
const createTextModel = (provider, modelId, options) => {
|
|
24
|
+
return aiSdkTextModel({
|
|
25
|
+
model: provider(modelId, options),
|
|
26
|
+
providerId: "workers-ai",
|
|
27
|
+
modelId
|
|
28
|
+
});
|
|
29
|
+
};
|
|
30
|
+
const createEmbeddingModel = (provider, modelId, options) => {
|
|
31
|
+
return aiSdkEmbeddingModel({
|
|
32
|
+
model: provider.textEmbedding(modelId, options),
|
|
33
|
+
providerId: "workers-ai",
|
|
34
|
+
modelId
|
|
35
|
+
});
|
|
36
|
+
};
|
|
37
|
+
const createImageModel = (provider, modelId, options) => {
|
|
38
|
+
return aiSdkImageModel({
|
|
39
|
+
model: provider.image(modelId, options),
|
|
40
|
+
providerId: "workers-ai",
|
|
41
|
+
modelId
|
|
42
|
+
});
|
|
43
|
+
};
|
|
44
|
+
const createSpeechModel = (provider, modelId, options) => {
|
|
45
|
+
return aiSdkSpeechModel({
|
|
46
|
+
model: provider.speech(modelId, options),
|
|
47
|
+
providerId: "workers-ai",
|
|
48
|
+
modelId
|
|
49
|
+
});
|
|
50
|
+
};
|
|
51
|
+
const createTranscriptionModel = (provider, modelId, options) => {
|
|
52
|
+
return aiSdkTranscriptionModel({
|
|
53
|
+
model: provider.transcription(modelId, options),
|
|
54
|
+
providerId: "workers-ai",
|
|
55
|
+
modelId
|
|
56
|
+
});
|
|
57
|
+
};
|
|
58
|
+
const createProvider = (provider) => {
|
|
59
|
+
return Object.assign((modelId, options) => {
|
|
60
|
+
return createAgentModel(provider, modelId, options);
|
|
61
|
+
}, {
|
|
62
|
+
text(modelId, options) {
|
|
63
|
+
return createTextModel(provider, modelId, options);
|
|
64
|
+
},
|
|
65
|
+
embedding(modelId, options) {
|
|
66
|
+
return createEmbeddingModel(provider, modelId, options);
|
|
67
|
+
},
|
|
68
|
+
image(modelId, options) {
|
|
69
|
+
return createImageModel(provider, modelId, options);
|
|
70
|
+
},
|
|
71
|
+
speech(modelId, options) {
|
|
72
|
+
return createSpeechModel(provider, modelId, options);
|
|
73
|
+
},
|
|
74
|
+
transcription(modelId, options) {
|
|
75
|
+
return createTranscriptionModel(provider, modelId, options);
|
|
76
|
+
}
|
|
77
|
+
});
|
|
78
|
+
};
|
|
79
|
+
const createWorkersAI = (options) => {
|
|
80
|
+
return createProvider(createWorkersAI$1(options));
|
|
81
|
+
};
|
|
82
|
+
|
|
83
|
+
//#endregion
|
|
84
|
+
export { createWorkersAI };
|
|
85
|
+
//# sourceMappingURL=index.mjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.mjs","names":["createAiSdkWorkersAI"],"sources":["../src/model.ts"],"sourcesContent":["import {\n aiSdkEmbeddingModel,\n aiSdkImageModel,\n aiSdkModel,\n aiSdkSpeechModel,\n aiSdkTextModel,\n aiSdkTranscriptionModel,\n} from \"@better-agent/ai-sdk\";\nimport type {\n AgentCapabilities,\n AgentModel,\n EmbeddingGenerationModel,\n ImageGenerationModel,\n SpeechGenerationModel,\n TextGenerationModel,\n TranscriptionGenerationModel,\n} from \"@better-agent/core\";\nimport { createWorkersAI as createAiSdkWorkersAI } from \"workers-ai-provider\";\n\nexport type CreateWorkersAIOptions = Parameters<typeof createAiSdkWorkersAI>[0];\n\ntype AiSdkWorkersAIProvider = ReturnType<typeof createAiSdkWorkersAI>;\n\nexport type WorkersAIProviderOptions = {\n \"workers-ai\"?: Record<string, unknown>;\n};\nexport type WorkersAITextProviderOptions = WorkersAIProviderOptions;\nexport type WorkersAIEmbeddingProviderOptions = {\n \"workers-ai\"?: Record<string, unknown>;\n};\nexport type WorkersAIImageProviderOptions = {\n \"workers-ai\"?: Record<string, unknown>;\n};\nexport type WorkersAISpeechProviderOptions = {\n \"workers-ai\"?: Record<string, unknown>;\n};\nexport type WorkersAITranscriptionProviderOptions = {\n \"workers-ai\"?: Record<string, unknown>;\n};\nexport type WorkersAIGenerationProviderOptions =\n | WorkersAIProviderOptions\n | WorkersAITextProviderOptions\n | WorkersAIEmbeddingProviderOptions\n | WorkersAIImageProviderOptions\n | WorkersAISpeechProviderOptions\n | WorkersAITranscriptionProviderOptions;\n\nexport type WorkersAIModelId = Parameters<AiSdkWorkersAIProvider>[0];\nexport type WorkersAITextModelId = WorkersAIModelId;\nexport type WorkersAIEmbeddingModelId = Parameters<AiSdkWorkersAIProvider[\"textEmbedding\"]>[0];\nexport type WorkersAIImageModelId = Parameters<AiSdkWorkersAIProvider[\"image\"]>[0];\nexport type WorkersAISpeechModelId = Parameters<AiSdkWorkersAIProvider[\"speech\"]>[0];\nexport type WorkersAITranscriptionModelId = Parameters<AiSdkWorkersAIProvider[\"transcription\"]>[0];\nexport type WorkersAITextOptions = NonNullable<Parameters<AiSdkWorkersAIProvider>[1]>;\nexport type WorkersAIEmbeddingOptions = NonNullable<\n Parameters<AiSdkWorkersAIProvider[\"textEmbedding\"]>[1]\n>;\nexport type WorkersAIImageOptions = NonNullable<Parameters<AiSdkWorkersAIProvider[\"image\"]>[1]>;\nexport type WorkersAISpeechOptions = NonNullable<Parameters<AiSdkWorkersAIProvider[\"speech\"]>[1]>;\nexport type WorkersAITranscriptionOptions = NonNullable<\n Parameters<AiSdkWorkersAIProvider[\"transcription\"]>[1]\n>;\n\nexport type WorkersAIModelOptions = WorkersAITextOptions;\n\nexport type WorkersAIModel<TModelId extends WorkersAIModelId = WorkersAIModelId> = AgentModel<\n AgentCapabilities,\n undefined,\n WorkersAIProviderOptions,\n TModelId\n>;\nexport type WorkersAITextModel<TModelId extends WorkersAITextModelId = WorkersAITextModelId> =\n TextGenerationModel<WorkersAITextProviderOptions, TModelId>;\nexport type WorkersAIEmbeddingModel<\n TModelId extends WorkersAIEmbeddingModelId = WorkersAIEmbeddingModelId,\n> = EmbeddingGenerationModel<WorkersAIEmbeddingProviderOptions, TModelId>;\nexport type WorkersAIImageModel<TModelId extends WorkersAIImageModelId = WorkersAIImageModelId> =\n ImageGenerationModel<WorkersAIImageProviderOptions, TModelId>;\nexport type WorkersAISpeechModel<TModelId extends WorkersAISpeechModelId = WorkersAISpeechModelId> =\n SpeechGenerationModel<WorkersAISpeechProviderOptions, TModelId>;\nexport type WorkersAITranscriptionModel<\n TModelId extends WorkersAITranscriptionModelId = WorkersAITranscriptionModelId,\n> = TranscriptionGenerationModel<WorkersAITranscriptionProviderOptions, TModelId>;\n\nexport type WorkersAIProvider = {\n <TModelId extends WorkersAIModelId>(\n modelId: TModelId,\n options?: WorkersAIModelOptions,\n ): WorkersAIModel<TModelId>;\n text<TModelId extends WorkersAITextModelId>(\n modelId: TModelId,\n options?: WorkersAITextOptions,\n ): WorkersAITextModel<TModelId>;\n embedding<TModelId extends WorkersAIEmbeddingModelId>(\n modelId: TModelId,\n options?: WorkersAIEmbeddingOptions,\n ): WorkersAIEmbeddingModel<TModelId>;\n image<TModelId extends WorkersAIImageModelId>(\n modelId: TModelId,\n options?: WorkersAIImageOptions,\n ): WorkersAIImageModel<TModelId>;\n speech<TModelId extends WorkersAISpeechModelId>(\n modelId: TModelId,\n options?: WorkersAISpeechOptions,\n ): WorkersAISpeechModel<TModelId>;\n transcription<TModelId extends WorkersAITranscriptionModelId>(\n modelId: TModelId,\n options?: WorkersAITranscriptionOptions,\n ): WorkersAITranscriptionModel<TModelId>;\n};\n\nconst workersAICapabilities = {\n identity: {\n provider: \"workers-ai\",\n },\n transport: {\n streaming: true,\n },\n output: {\n structuredOutput: true,\n supportedMimeTypes: [\"text/plain\", \"application/json\"],\n },\n multimodal: {\n input: {\n image: true,\n },\n },\n tools: {\n supported: true,\n },\n} as const satisfies AgentCapabilities;\n\nconst createAgentModel = <TModelId extends WorkersAIModelId>(\n provider: AiSdkWorkersAIProvider,\n modelId: TModelId,\n options?: WorkersAITextOptions,\n): WorkersAIModel<TModelId> => {\n return aiSdkModel<AgentCapabilities, WorkersAIProviderOptions, TModelId>({\n model: provider(modelId, options),\n providerId: \"workers-ai\",\n modelId,\n capabilities: workersAICapabilities,\n });\n};\n\nconst createTextModel = <TModelId extends WorkersAITextModelId>(\n provider: AiSdkWorkersAIProvider,\n modelId: TModelId,\n options?: WorkersAITextOptions,\n): WorkersAITextModel<TModelId> => {\n return aiSdkTextModel<WorkersAITextProviderOptions, TModelId>({\n model: provider(modelId, options),\n providerId: \"workers-ai\",\n modelId,\n });\n};\n\nconst createEmbeddingModel = <TModelId extends WorkersAIEmbeddingModelId>(\n provider: AiSdkWorkersAIProvider,\n modelId: TModelId,\n options?: WorkersAIEmbeddingOptions,\n): WorkersAIEmbeddingModel<TModelId> => {\n return aiSdkEmbeddingModel<WorkersAIEmbeddingProviderOptions, TModelId>({\n model: provider.textEmbedding(modelId, options),\n providerId: \"workers-ai\",\n modelId,\n });\n};\n\nconst createImageModel = <TModelId extends WorkersAIImageModelId>(\n provider: AiSdkWorkersAIProvider,\n modelId: TModelId,\n options?: WorkersAIImageOptions,\n): WorkersAIImageModel<TModelId> => {\n return aiSdkImageModel<WorkersAIImageProviderOptions, TModelId>({\n model: provider.image(modelId, options),\n providerId: \"workers-ai\",\n modelId,\n });\n};\n\nconst createSpeechModel = <TModelId extends WorkersAISpeechModelId>(\n provider: AiSdkWorkersAIProvider,\n modelId: TModelId,\n options?: WorkersAISpeechOptions,\n): WorkersAISpeechModel<TModelId> => {\n return aiSdkSpeechModel<WorkersAISpeechProviderOptions, TModelId>({\n model: provider.speech(modelId, options),\n providerId: \"workers-ai\",\n modelId,\n });\n};\n\nconst createTranscriptionModel = <TModelId extends WorkersAITranscriptionModelId>(\n provider: AiSdkWorkersAIProvider,\n modelId: TModelId,\n options?: WorkersAITranscriptionOptions,\n): WorkersAITranscriptionModel<TModelId> => {\n return aiSdkTranscriptionModel<WorkersAITranscriptionProviderOptions, TModelId>({\n model: provider.transcription(modelId, options),\n providerId: \"workers-ai\",\n modelId,\n });\n};\n\nconst createProvider = (provider: AiSdkWorkersAIProvider): WorkersAIProvider => {\n const createWorkersAIModel: WorkersAIProvider = Object.assign(\n <TModelId extends WorkersAIModelId>(\n modelId: TModelId,\n options?: WorkersAIModelOptions,\n ): WorkersAIModel<TModelId> => {\n return createAgentModel(provider, modelId, options);\n },\n {\n text<TModelId extends WorkersAITextModelId>(\n modelId: TModelId,\n options?: WorkersAITextOptions,\n ): WorkersAITextModel<TModelId> {\n return createTextModel(provider, modelId, options);\n },\n embedding<TModelId extends WorkersAIEmbeddingModelId>(\n modelId: TModelId,\n options?: WorkersAIEmbeddingOptions,\n ): WorkersAIEmbeddingModel<TModelId> {\n return createEmbeddingModel(provider, modelId, options);\n },\n image<TModelId extends WorkersAIImageModelId>(\n modelId: TModelId,\n options?: WorkersAIImageOptions,\n ): WorkersAIImageModel<TModelId> {\n return createImageModel(provider, modelId, options);\n },\n speech<TModelId extends WorkersAISpeechModelId>(\n modelId: TModelId,\n options?: WorkersAISpeechOptions,\n ): WorkersAISpeechModel<TModelId> {\n return createSpeechModel(provider, modelId, options);\n },\n transcription<TModelId extends WorkersAITranscriptionModelId>(\n modelId: TModelId,\n options?: WorkersAITranscriptionOptions,\n ): WorkersAITranscriptionModel<TModelId> {\n return createTranscriptionModel(provider, modelId, options);\n },\n },\n );\n\n return createWorkersAIModel;\n};\n\nexport const createWorkersAI = (options: CreateWorkersAIOptions): WorkersAIProvider => {\n return createProvider(createAiSdkWorkersAI(options));\n};\n"],"mappings":";;;;AA+GA,MAAM,wBAAwB;CAC1B,UAAU,EACN,UAAU,cACb;CACD,WAAW,EACP,WAAW,MACd;CACD,QAAQ;EACJ,kBAAkB;EAClB,oBAAoB,CAAC,cAAc,mBAAmB;EACzD;CACD,YAAY,EACR,OAAO,EACH,OAAO,MACV,EACJ;CACD,OAAO,EACH,WAAW,MACd;CACJ;AAED,MAAM,oBACF,UACA,SACA,YAC2B;AAC3B,QAAO,WAAkE;EACrE,OAAO,SAAS,SAAS,QAAQ;EACjC,YAAY;EACZ;EACA,cAAc;EACjB,CAAC;;AAGN,MAAM,mBACF,UACA,SACA,YAC+B;AAC/B,QAAO,eAAuD;EAC1D,OAAO,SAAS,SAAS,QAAQ;EACjC,YAAY;EACZ;EACH,CAAC;;AAGN,MAAM,wBACF,UACA,SACA,YACoC;AACpC,QAAO,oBAAiE;EACpE,OAAO,SAAS,cAAc,SAAS,QAAQ;EAC/C,YAAY;EACZ;EACH,CAAC;;AAGN,MAAM,oBACF,UACA,SACA,YACgC;AAChC,QAAO,gBAAyD;EAC5D,OAAO,SAAS,MAAM,SAAS,QAAQ;EACvC,YAAY;EACZ;EACH,CAAC;;AAGN,MAAM,qBACF,UACA,SACA,YACiC;AACjC,QAAO,iBAA2D;EAC9D,OAAO,SAAS,OAAO,SAAS,QAAQ;EACxC,YAAY;EACZ;EACH,CAAC;;AAGN,MAAM,4BACF,UACA,SACA,YACwC;AACxC,QAAO,wBAAyE;EAC5E,OAAO,SAAS,cAAc,SAAS,QAAQ;EAC/C,YAAY;EACZ;EACH,CAAC;;AAGN,MAAM,kBAAkB,aAAwD;AA0C5E,QAzCgD,OAAO,QAE/C,SACA,YAC2B;AAC3B,SAAO,iBAAiB,UAAU,SAAS,QAAQ;IAEvD;EACI,KACI,SACA,SAC4B;AAC5B,UAAO,gBAAgB,UAAU,SAAS,QAAQ;;EAEtD,UACI,SACA,SACiC;AACjC,UAAO,qBAAqB,UAAU,SAAS,QAAQ;;EAE3D,MACI,SACA,SAC6B;AAC7B,UAAO,iBAAiB,UAAU,SAAS,QAAQ;;EAEvD,OACI,SACA,SAC8B;AAC9B,UAAO,kBAAkB,UAAU,SAAS,QAAQ;;EAExD,cACI,SACA,SACqC;AACrC,UAAO,yBAAyB,UAAU,SAAS,QAAQ;;EAElE,CACJ;;AAKL,MAAa,mBAAmB,YAAuD;AACnF,QAAO,eAAeA,kBAAqB,QAAQ,CAAC"}
|
package/package.json
ADDED
|
@@ -0,0 +1,55 @@
|
|
|
1
|
+
{
|
|
2
|
+
"name": "@better-agent/workers-ai",
|
|
3
|
+
"version": "0.2.0-beta.1",
|
|
4
|
+
"description": "Better Agent Workers AI provider",
|
|
5
|
+
"license": "MIT",
|
|
6
|
+
"repository": {
|
|
7
|
+
"type": "git",
|
|
8
|
+
"url": "git+https://github.com/better-agent/better-agent.git",
|
|
9
|
+
"directory": "packages/workers-ai"
|
|
10
|
+
},
|
|
11
|
+
"homepage": "https://github.com/better-agent/better-agent/tree/main/packages/workers-ai",
|
|
12
|
+
"bugs": {
|
|
13
|
+
"url": "https://github.com/better-agent/better-agent/issues"
|
|
14
|
+
},
|
|
15
|
+
"engines": {
|
|
16
|
+
"bun": ">=1.3.3",
|
|
17
|
+
"node": ">=20"
|
|
18
|
+
},
|
|
19
|
+
"type": "module",
|
|
20
|
+
"types": "./dist/index.d.mts",
|
|
21
|
+
"sideEffects": false,
|
|
22
|
+
"exports": {
|
|
23
|
+
".": {
|
|
24
|
+
"types": "./dist/index.d.mts",
|
|
25
|
+
"import": "./dist/index.mjs"
|
|
26
|
+
}
|
|
27
|
+
},
|
|
28
|
+
"scripts": {
|
|
29
|
+
"build": "bunx tsdown",
|
|
30
|
+
"dev": "bunx tsdown --watch --no-clean",
|
|
31
|
+
"lint:package": "bunx publint run --strict",
|
|
32
|
+
"lint:types": "npm_config_cache=../../.npm-cache bunx attw --profile esm-only --pack .",
|
|
33
|
+
"lint": "bunx biome check .",
|
|
34
|
+
"typecheck": "bunx tsc -p tsconfig.json --noEmit",
|
|
35
|
+
"format": "bunx biome format --write .",
|
|
36
|
+
"clean": "bunx rimraf dist"
|
|
37
|
+
},
|
|
38
|
+
"files": ["dist"],
|
|
39
|
+
"publishConfig": {
|
|
40
|
+
"access": "public"
|
|
41
|
+
},
|
|
42
|
+
"dependencies": {
|
|
43
|
+
"@better-agent/ai-sdk": "0.2.0-beta.1",
|
|
44
|
+
"ai": "^6.0.168",
|
|
45
|
+
"workers-ai-provider": "^3.1.12"
|
|
46
|
+
},
|
|
47
|
+
"devDependencies": {
|
|
48
|
+
"@better-agent/core": "0.2.0-beta.1",
|
|
49
|
+
"@types/bun": "^1.3.12",
|
|
50
|
+
"@types/node": "^22.19.17"
|
|
51
|
+
},
|
|
52
|
+
"peerDependencies": {
|
|
53
|
+
"@better-agent/core": "0.2.0-beta.1"
|
|
54
|
+
}
|
|
55
|
+
}
|