@kernl-sdk/ai 0.1.1

package/src/language-model.ts

@@ -0,0 +1,77 @@
+import type { LanguageModelV3 } from "@ai-sdk/provider";
+
+import type {
+  LanguageModel,
+  LanguageModelRequest,
+  LanguageModelResponse,
+  LanguageModelStreamEvent,
+} from "@kernl-sdk/protocol";
+
+import { MESSAGE } from "./convert/message";
+import { TOOL } from "./convert/tools";
+import { MODEL_SETTINGS } from "./convert/settings";
+import { MODEL_RESPONSE } from "./convert/response";
+import { convertStream } from "./convert/stream";
+import { wrapError } from "./error";
+
+/**
+ * LanguageModel adapter for the AI SDK LanguageModelV3.
+ */
+export class AISDKLanguageModel implements LanguageModel {
+  readonly spec = "1.0" as const;
+  readonly provider: string;
+  readonly modelId: string;
+
+  constructor(private model: LanguageModelV3) {
+    this.provider = model.provider;
+    this.modelId = model.modelId;
+  }
+
+  /**
+   * Get a response from the model.
+   */
+  async generate(
+    request: LanguageModelRequest,
+  ): Promise<LanguageModelResponse> {
+    try {
+      const messages = request.input.map(MESSAGE.encode);
+      const tools = request.tools ? request.tools.map(TOOL.encode) : undefined;
+      const settings = MODEL_SETTINGS.encode(request.settings);
+
+      const result = await this.model.doGenerate({
+        prompt: messages,
+        tools,
+        ...settings,
+        abortSignal: request.abort,
+      });
+
+      return MODEL_RESPONSE.decode(result);
+    } catch (error) {
+      throw wrapError(error, "generate");
+    }
+  }
+
+  /**
+   * Get a streamed response from the model.
+   */
+  async *stream(
+    request: LanguageModelRequest,
+  ): AsyncIterable<LanguageModelStreamEvent> {
+    try {
+      const messages = request.input.map(MESSAGE.encode);
+      const tools = request.tools ? request.tools.map(TOOL.encode) : undefined;
+      const settings = MODEL_SETTINGS.encode(request.settings);
+
+      const stream = await this.model.doStream({
+        prompt: messages,
+        tools,
+        ...settings,
+        abortSignal: request.abort,
+      });
+
+      yield* convertStream(stream.stream);
+    } catch (error) {
+      throw wrapError(error, "stream");
+    }
+  }
+}

package/src/providers/anthropic.ts

@@ -0,0 +1,18 @@
+import { anthropic as createAnthropicModel } from "@ai-sdk/anthropic";
+import { AISDKLanguageModel } from "../language-model";
+
+/**
+ * Create a Kernl-compatible Anthropic language model.
+ *
+ * @example
+ * ```ts
+ * import { anthropic } from '@kernl-sdk/ai/anthropic';
+ *
+ * const claude = anthropic('claude-3-5-sonnet-20241022');
+ * const response = await claude.generate([...], {});
+ * ```
+ */
+export function anthropic(modelId: string) {
+  const model = createAnthropicModel(modelId);
+  return new AISDKLanguageModel(model);
+}

package/src/providers/google.ts

@@ -0,0 +1,18 @@
+import { google as createGoogleModel } from "@ai-sdk/google";
+import { AISDKLanguageModel } from "../language-model";
+
+/**
+ * Create a Kernl-compatible Google Generative AI language model.
+ *
+ * @example
+ * ```ts
+ * import { google } from '@kernl-sdk/ai/google';
+ *
+ * const gemini = google('gemini-2.0-flash-exp');
+ * const response = await gemini.generate([...], {});
+ * ```
+ */
+export function google(modelId: string) {
+  const model = createGoogleModel(modelId);
+  return new AISDKLanguageModel(model);
+}

package/src/providers/openai.ts

@@ -0,0 +1,18 @@
+import { openai as createOpenAIModel } from "@ai-sdk/openai";
+import { AISDKLanguageModel } from "../language-model";
+
+/**
+ * Create a Kernl-compatible OpenAI language model.
+ *
+ * @example
+ * ```ts
+ * import { openai } from '@kernl-sdk/ai/openai';
+ *
+ * const gpt4 = openai('gpt-4-turbo');
+ * const response = await gpt4.generate([...], {});
+ * ```
+ */
+export function openai(modelId: string) {
+  const model = createOpenAIModel(modelId);
+  return new AISDKLanguageModel(model);
+}

package/tsconfig.json

@@ -0,0 +1,13 @@
+{
+  "extends": "../../tsconfig.base.json",
+  "compilerOptions": {
+    "outDir": "./dist",
+    "rootDir": "./src",
+    "baseUrl": ".",
+    "paths": {
+      "@/*": ["./src/*"]
+    }
+  },
+  "include": ["src/**/*"],
+  "exclude": ["node_modules", "dist"]
+}

package/vitest.config.ts

@@ -0,0 +1,14 @@
+import { defineConfig } from "vitest/config";
+import path from "path";
+
+export default defineConfig({
+  test: {
+    globals: true,
+    environment: "node",
+  },
+  resolve: {
+    alias: {
+      "@": path.resolve(__dirname, "./src"),
+    },
+  },
+});