@fraqjs/plugin-ai 0.1.0

package/dist/index.d.mts

@@ -0,0 +1,38 @@
+import { LanguageModel } from "ai";
+
+//#region src/provider.d.ts
+type SupportedSDK = '@ai-sdk/anthropic' | '@ai-sdk/deepseek' | '@ai-sdk/google' | '@ai-sdk/openai';
+interface ProviderConfig {
+  sdk: SupportedSDK;
+  options: {
+    apiKey: string;
+    baseUrl?: string;
+    [key: string]: unknown;
+  };
+  models: string[];
+}
+declare function resolveLanguageModels(config: ProviderConfig): Promise<LanguageModel[]>;
+//#endregion
+//#region src/service.d.ts
+interface AiServiceOptions {
+  models: Record<string, LanguageModel>;
+  aliases: Record<string, string>;
+  defaultModel: string;
+}
+declare class AiService {
+  private readonly options;
+  constructor(options: AiServiceOptions);
+  hasModel(name: string): boolean;
+  model(name?: string): LanguageModel;
+  models(): string[];
+}
+//#endregion
+//#region src/index.d.ts
+interface AiPluginOptions {
+  providers: Record<string, ProviderConfig | Record<string, LanguageModel>>;
+  aliases?: Record<string, string>;
+  defaultModel?: string;
+}
+declare const AiPlugin: import("@fraqjs/fraq").Plugin<[options: AiPluginOptions], import("@fraqjs/fraq").Injection | undefined, import("@fraqjs/fraq").Injection | undefined>;
+//#endregion
+export { AiPlugin, AiPlugin as default, AiPluginOptions, AiService, AiServiceOptions, ProviderConfig, SupportedSDK, resolveLanguageModels };

package/dist/index.mjs

@@ -0,0 +1,72 @@
+import { definePlugin } from "@fraqjs/fraq";
+//#region src/provider.ts
+async function resolveLanguageModels(config) {
+	const { sdk, options, models } = config;
+	let provider;
+	switch (sdk) {
+		case "@ai-sdk/anthropic":
+			provider = (await import("@ai-sdk/anthropic")).createAnthropic(options);
+			break;
+		case "@ai-sdk/deepseek":
+			provider = (await import("@ai-sdk/deepseek")).createDeepSeek(options);
+			break;
+		case "@ai-sdk/google":
+			provider = (await import("@ai-sdk/google")).createGoogleGenerativeAI(options);
+			break;
+		case "@ai-sdk/openai":
+			provider = (await import("@ai-sdk/openai")).createOpenAI(options);
+			break;
+		default: throw new Error(`Unsupported AI SDK: ${sdk}`);
+	}
+	return models.map((model) => provider.languageModel(model));
+}
+//#endregion
+//#region src/service.ts
+var AiService = class {
+	options;
+	constructor(options) {
+		this.options = options;
+		if (!options.models[options.defaultModel]) throw new Error(`Invalid default model "${options.defaultModel}": model does not exist.`);
+		for (const [alias, target] of Object.entries(options.aliases)) if (!options.models[target]) throw new Error(`Invalid alias "${alias}": target model "${target}" does not exist.`);
+	}
+	hasModel(name) {
+		return !!this.options.models[name] || !!this.options.aliases[name];
+	}
+	model(name) {
+		const modelByName = this.options.models[name ?? this.options.defaultModel];
+		if (modelByName) return modelByName;
+		const aliasTarget = this.options.aliases[name ?? ""];
+		if (aliasTarget) return this.options.models[aliasTarget];
+		throw new Error(`Model not found: ${name}`);
+	}
+	models() {
+		return Object.keys(this.options.models);
+	}
+};
+//#endregion
+//#region src/index.ts
+function isProviderConfig(config) {
+	return "sdk" in config && "options" in config && "models" in config;
+}
+const AiPlugin = definePlugin({
+	name: "ai",
+	provides: [AiService],
+	async apply(ctx, options) {
+		const models = {};
+		for (const [name, config] of Object.entries(options.providers)) if (isProviderConfig(config)) (await resolveLanguageModels(config)).forEach((model, idx) => {
+			const modelName = `${name}/${config.models[idx]}`;
+			models[modelName] = model;
+		});
+		else Object.entries(config).forEach(([modelName, model]) => {
+			models[`${name}/${modelName}`] = model;
+		});
+		if (Object.keys(models).length === 0) throw new Error("No language models resolved from the provided AI SDK configurations.");
+		ctx.provide(AiService, new AiService({
+			models,
+			aliases: options.aliases ?? {},
+			defaultModel: options.defaultModel ?? Object.keys(models)[0]
+		}));
+	}
+});
+//#endregion
+export { AiPlugin, AiPlugin as default, AiService, resolveLanguageModels };

package/package.json

@@ -0,0 +1,38 @@
+{
+  "name": "@fraqjs/plugin-ai",
+  "type": "module",
+  "version": "0.1.0",
+  "description": "Vercel AI SDK integration plugin for Fraq",
+  "files": [
+    "dist"
+  ],
+  "main": "dist/index.mjs",
+  "typings": "dist/index.d.mts",
+  "keywords": [],
+  "author": "fraqjs",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/fraqjs/fraq.git"
+  },
+  "homepage": "https://fraq.ntqqrev.org/",
+  "license": "MIT",
+  "peerDependencies": {
+    "ai": "^6.0.0",
+    "@fraqjs/fraq": "^0.8.2"
+  },
+  "devDependencies": {
+    "@ai-sdk/anthropic": "^3.0.81",
+    "@ai-sdk/deepseek": "^2.0.35",
+    "@ai-sdk/google": "^3.0.80",
+    "@ai-sdk/openai": "^3.0.68",
+    "@ai-sdk/provider": "^3.0.10",
+    "@fraqjs/mock": "^0.1.0"
+  },
+  "engines": {
+    "node": ">=22"
+  },
+  "scripts": {
+    "build": "tsdown",
+    "test": "tsx --test test/*.test.ts"
+  }
+}