@fraqjs/plugin-ai 0.1.0 → 0.2.0

package/dist/index.d.mts

@@ -1,17 +1,17 @@
 import { LanguageModel } from "ai";
 
 //#region src/provider.d.ts
-type SupportedSDK = '@ai-sdk/anthropic' | '@ai-sdk/deepseek' | '@ai-sdk/google' | '@ai-sdk/openai';
+type SupportedSDK = '@ai-sdk/anthropic' | '@ai-sdk/deepseek' | '@ai-sdk/google' | '@ai-sdk/openai' | '@ai-sdk/openai-compatible';
 interface ProviderConfig {
   sdk: SupportedSDK;
   options: {
     apiKey: string;
-    baseUrl?: string;
+    baseURL?: string;
     [key: string]: unknown;
   };
   models: string[];
 }
-declare function resolveLanguageModels(config: ProviderConfig): Promise<LanguageModel[]>;
+declare function resolveLanguageModels(name: string, config: ProviderConfig): Promise<LanguageModel[]>;
 //#endregion
 //#region src/service.d.ts
 interface AiServiceOptions {
@@ -21,6 +21,7 @@ interface AiServiceOptions {
 }
 declare class AiService {
   private readonly options;
+  private defaultModel;
   constructor(options: AiServiceOptions);
   hasModel(name: string): boolean;
   model(name?: string): LanguageModel;
@@ -28,8 +29,9 @@ declare class AiService {
 }
 //#endregion
 //#region src/index.d.ts
+type LanguageModelInstance = Exclude<LanguageModel, string>;
 interface AiPluginOptions {
-  providers: Record<string, ProviderConfig | Record<string, LanguageModel>>;
+  providers: Record<string, ProviderConfig | Record<string, LanguageModelInstance>>;
   aliases?: Record<string, string>;
   defaultModel?: string;
 }

package/dist/index.mjs

@@ -1,6 +1,6 @@
 import { definePlugin } from "@fraqjs/fraq";
 //#region src/provider.ts
-async function resolveLanguageModels(config) {
+async function resolveLanguageModels(name, config) {
 	const { sdk, options, models } = config;
 	let provider;
 	switch (sdk) {
@@ -16,6 +16,14 @@ async function resolveLanguageModels(config) {
 		case "@ai-sdk/openai":
 			provider = (await import("@ai-sdk/openai")).createOpenAI(options);
 			break;
+		case "@ai-sdk/openai-compatible":
+			if (!options.baseURL) throw new Error("`baseURL` is required for OpenAI Compatible SDK");
+			provider = (await import("@ai-sdk/openai-compatible")).createOpenAICompatible({
+				...options,
+				baseURL: options.baseURL,
+				name
+			});
+			break;
 		default: throw new Error(`Unsupported AI SDK: ${sdk}`);
 	}
 	return models.map((model) => provider.languageModel(model));
@@ -24,19 +32,23 @@ async function resolveLanguageModels(config) {
 //#region src/service.ts
 var AiService = class {
 	options;
+	defaultModel;
 	constructor(options) {
 		this.options = options;
-		if (!options.models[options.defaultModel]) throw new Error(`Invalid default model "${options.defaultModel}": model does not exist.`);
+		const defaultModel = options.models[options.defaultModel] ?? options.models[options.aliases[options.defaultModel]];
+		if (!defaultModel) throw new Error(`Invalid default model "${options.defaultModel}": model does not exist.`);
+		this.defaultModel = defaultModel;
 		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 (!name) return this.defaultModel;
+		const modelByName = this.options.models[name];
 		if (modelByName) return modelByName;
-		const aliasTarget = this.options.aliases[name ?? ""];
-		if (aliasTarget) return this.options.models[aliasTarget];
+		const modelByAlias = this.options.aliases[name];
+		if (modelByAlias) return this.options.models[modelByAlias];
 		throw new Error(`Model not found: ${name}`);
 	}
 	models() {
@@ -46,14 +58,14 @@ var AiService = class {
 //#endregion
 //#region src/index.ts
 function isProviderConfig(config) {
-	return "sdk" in config && "options" in config && "models" in config;
+	return "sdk" in config && typeof config.sdk === "string";
 }
 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) => {
+		for (const [name, config] of Object.entries(options.providers)) if (isProviderConfig(config)) (await resolveLanguageModels(name, config)).forEach((model, idx) => {
 			const modelName = `${name}/${config.models[idx]}`;
 			models[modelName] = model;
 		});

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@fraqjs/plugin-ai",
   "type": "module",
-  "version": "0.1.0",
+  "version": "0.2.0",
   "description": "Vercel AI SDK integration plugin for Fraq",
   "files": [
     "dist"
@@ -14,18 +14,19 @@
     "type": "git",
     "url": "git+https://github.com/fraqjs/fraq.git"
   },
-  "homepage": "https://fraq.ntqqrev.org/",
+  "homepage": "https://fraq.dev/",
   "license": "MIT",
   "peerDependencies": {
-    "ai": "^6.0.0",
-    "@fraqjs/fraq": "^0.8.2"
+    "ai": "^7.0.0",
+    "@fraqjs/fraq": "^0.12.1"
   },
   "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",
+    "@ai-sdk/anthropic": "^4",
+    "@ai-sdk/deepseek": "^3",
+    "@ai-sdk/google": "^4",
+    "@ai-sdk/openai": "^4",
+    "@ai-sdk/openai-compatible": "^3",
+    "@ai-sdk/provider": "^4",
     "@fraqjs/mock": "^0.1.0"
   },
   "engines": {