@tigmart/ai-core 0.0.1

package/dist/index.d.mts

@@ -0,0 +1,53 @@
+import { ProviderAdapter, Model, Message, StreamOptions, StreamChunk } from '@topsoft/ai-types';
+
+declare class OpenAIAdapter implements ProviderAdapter {
+    readonly id = "openai";
+    readonly name = "OpenAI";
+    readonly models: Model[];
+    private readonly client;
+    /**
+     * @param apiKey  If omitted, falls back to process.env.OPENAI_API_KEY.
+     *                Must never be passed from client-side code.
+     */
+    constructor(apiKey?: string);
+    stream(messages: Message[], modelId: string, options?: StreamOptions): AsyncGenerator<StreamChunk>;
+}
+
+declare class ClaudeAdapter implements ProviderAdapter {
+    readonly id = "claude";
+    readonly name = "Claude";
+    readonly models: Model[];
+    stream(_messages: Message[], _modelId: string): AsyncIterable<StreamChunk>;
+}
+
+declare class GeminiAdapter implements ProviderAdapter {
+    readonly id = "gemini";
+    readonly name = "Gemini";
+    readonly models: Model[];
+    private readonly client;
+    constructor(apiKey?: string);
+    stream(messages: Message[], modelId: string, options?: {
+        signal?: AbortSignal;
+    }): AsyncIterable<StreamChunk>;
+}
+
+/**
+ * Holds all registered provider adapters and resolves them by ID.
+ * Instantiated once in the API route handler — adapters are stateless
+ * beyond their config, so a singleton per server process is correct.
+ *
+ * Usage in apps/playground/app/api/chat/route.ts:
+ *   const registry = new ProviderRegistry([
+ *     new OpenAIAdapter(),
+ *   ]);
+ */
+declare class ProviderRegistry {
+    private readonly adapters;
+    constructor(adapters: ProviderAdapter[]);
+    /** Returns the adapter for the given provider ID. Throws if not registered. */
+    get(providerId: string): ProviderAdapter;
+    /** Returns all registered adapters. Used to expose available providers to the client. */
+    getAll(): ProviderAdapter[];
+}
+
+export { ClaudeAdapter, GeminiAdapter, OpenAIAdapter, ProviderRegistry };

package/dist/index.d.ts

@@ -0,0 +1,53 @@
+import { ProviderAdapter, Model, Message, StreamOptions, StreamChunk } from '@topsoft/ai-types';
+
+declare class OpenAIAdapter implements ProviderAdapter {
+    readonly id = "openai";
+    readonly name = "OpenAI";
+    readonly models: Model[];
+    private readonly client;
+    /**
+     * @param apiKey  If omitted, falls back to process.env.OPENAI_API_KEY.
+     *                Must never be passed from client-side code.
+     */
+    constructor(apiKey?: string);
+    stream(messages: Message[], modelId: string, options?: StreamOptions): AsyncGenerator<StreamChunk>;
+}
+
+declare class ClaudeAdapter implements ProviderAdapter {
+    readonly id = "claude";
+    readonly name = "Claude";
+    readonly models: Model[];
+    stream(_messages: Message[], _modelId: string): AsyncIterable<StreamChunk>;
+}
+
+declare class GeminiAdapter implements ProviderAdapter {
+    readonly id = "gemini";
+    readonly name = "Gemini";
+    readonly models: Model[];
+    private readonly client;
+    constructor(apiKey?: string);
+    stream(messages: Message[], modelId: string, options?: {
+        signal?: AbortSignal;
+    }): AsyncIterable<StreamChunk>;
+}
+
+/**
+ * Holds all registered provider adapters and resolves them by ID.
+ * Instantiated once in the API route handler — adapters are stateless
+ * beyond their config, so a singleton per server process is correct.
+ *
+ * Usage in apps/playground/app/api/chat/route.ts:
+ *   const registry = new ProviderRegistry([
+ *     new OpenAIAdapter(),
+ *   ]);
+ */
+declare class ProviderRegistry {
+    private readonly adapters;
+    constructor(adapters: ProviderAdapter[]);
+    /** Returns the adapter for the given provider ID. Throws if not registered. */
+    get(providerId: string): ProviderAdapter;
+    /** Returns all registered adapters. Used to expose available providers to the client. */
+    getAll(): ProviderAdapter[];
+}
+
+export { ClaudeAdapter, GeminiAdapter, OpenAIAdapter, ProviderRegistry };

package/dist/index.js

@@ -0,0 +1,267 @@
+"use strict";
+var __create = Object.create;
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getProtoOf = Object.getPrototypeOf;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
+  // If the importer is in node compatibility mode or this is not an ESM
+  // file that has been converted to a CommonJS file using a Babel-
+  // compatible transform (i.e. "__esModule" has not been set), then set
+  // "default" to the CommonJS "module.exports" for node compatibility.
+  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
+  mod
+));
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/index.ts
+var index_exports = {};
+__export(index_exports, {
+  ClaudeAdapter: () => ClaudeAdapter,
+  GeminiAdapter: () => GeminiAdapter,
+  OpenAIAdapter: () => OpenAIAdapter,
+  ProviderRegistry: () => ProviderRegistry
+});
+module.exports = __toCommonJS(index_exports);
+
+// src/adapters/openai.ts
+var import_openai = __toESM(require("openai"));
+var MODELS = [
+  {
+    id: "gpt-4o",
+    name: "GPT-4o",
+    contextWindow: 128e3,
+    capabilities: { streaming: true, images: true, tools: false }
+  },
+  {
+    id: "gpt-4o-mini",
+    name: "GPT-4o Mini",
+    contextWindow: 128e3,
+    capabilities: { streaming: true, images: false, tools: false }
+  }
+];
+var VALID_ROLES = /* @__PURE__ */ new Set(["system", "user", "assistant"]);
+function isValidRole(role) {
+  return VALID_ROLES.has(role);
+}
+function toOpenAIMessages(messages) {
+  const result = [];
+  for (const msg of messages) {
+    if (!isValidRole(msg.role)) continue;
+    const text = msg.parts.filter((p) => p.type === "text").map((p) => p.text).join("\n").trim();
+    if (!text) continue;
+    result.push({ role: msg.role, content: text });
+  }
+  return result;
+}
+var OpenAIAdapter = class {
+  id = "openai";
+  name = "OpenAI";
+  models = MODELS;
+  client;
+  /**
+   * @param apiKey  If omitted, falls back to process.env.OPENAI_API_KEY.
+   *                Must never be passed from client-side code.
+   */
+  constructor(apiKey) {
+    const key = apiKey ?? process.env["OPENAI_API_KEY"];
+    if (!key) {
+      throw new Error(
+        "OpenAIAdapter: API key is required. Set OPENAI_API_KEY environment variable or pass it to the constructor."
+      );
+    }
+    this.client = new import_openai.default({ apiKey: key });
+  }
+  async *stream(messages, modelId, options) {
+    if (!this.models.some((m) => m.id === modelId)) {
+      throw new Error(`Model "${modelId}" not supported by OpenAI adapter`);
+    }
+    const openAiMessages = toOpenAIMessages(messages);
+    try {
+      const completion = await this.client.chat.completions.create(
+        {
+          model: modelId,
+          messages: openAiMessages,
+          stream: true,
+          stream_options: { include_usage: true },
+          ...options?.maxTokens !== void 0 && { max_tokens: options.maxTokens },
+          ...options?.temperature !== void 0 && { temperature: options.temperature }
+        },
+        { signal: options?.signal }
+      );
+      let usage;
+      for await (const chunk of completion) {
+        const choice = chunk.choices?.[0];
+        const content = choice?.delta?.content;
+        if (typeof content === "string" && content.length > 0) {
+          yield { type: "delta", content };
+        }
+        const u = chunk.usage;
+        if (u != null && typeof u.prompt_tokens === "number" && typeof u.completion_tokens === "number" && typeof u.total_tokens === "number") {
+          usage = {
+            promptTokens: u.prompt_tokens,
+            completionTokens: u.completion_tokens,
+            totalTokens: u.total_tokens
+          };
+        }
+      }
+      yield { type: "done", usage };
+    } catch (err) {
+      if (options?.signal?.aborted) return;
+      yield {
+        type: "error",
+        error: err instanceof Error ? err.message : JSON.stringify(err)
+      };
+    }
+  }
+};
+
+// src/adapters/claude.ts
+var MODELS2 = [
+  {
+    id: "claude-3-5-sonnet-20241022",
+    name: "Claude 3.5 Sonnet",
+    contextWindow: 2e5,
+    capabilities: { streaming: true, images: true, tools: false }
+  }
+];
+var ClaudeAdapter = class {
+  id = "claude";
+  name = "Claude";
+  models = MODELS2;
+  // TODO: Implement real Anthropic API integration.
+  async *stream(_messages, _modelId) {
+    if (!this.models.some((m) => m.id === _modelId)) {
+      yield { type: "error", error: `Model "${_modelId}" is not supported by the Claude adapter` };
+      return;
+    }
+    yield {
+      type: "delta",
+      content: "[Claude integration is not yet implemented. This is a placeholder.]"
+    };
+    yield { type: "done" };
+  }
+};
+
+// src/adapters/gemini.ts
+var import_genai = require("@google/genai");
+var MODELS3 = [
+  {
+    id: "gemini-2.5-flash",
+    name: "Gemini 2.5 Flash",
+    contextWindow: 1e6,
+    capabilities: { streaming: true, images: false, tools: false }
+  },
+  {
+    id: "gemini-2.5-pro",
+    name: "Gemini 2.5 Pro",
+    contextWindow: 2e6,
+    capabilities: { streaming: true, images: false, tools: false }
+  }
+];
+function toGeminiContents(messages) {
+  const contents = [];
+  for (const msg of messages) {
+    const text = msg.parts.filter((p) => p.type === "text").map((p) => p.text).join("\n").trim();
+    if (!text) continue;
+    const role = msg.role === "assistant" ? "model" : "user";
+    const prev = contents[contents.length - 1];
+    if (prev && prev.role === role) {
+      const existing = prev.parts[0];
+      if (existing) {
+        existing.text += "\n" + text;
+      }
+    } else {
+      contents.push({ role, parts: [{ text }] });
+    }
+  }
+  return contents;
+}
+var GeminiAdapter = class {
+  id = "gemini";
+  name = "Gemini";
+  models = MODELS3;
+  client;
+  constructor(apiKey) {
+    const key = apiKey ?? process.env["GEMINI_API_KEY"];
+    if (!key) {
+      throw new Error(
+        "GeminiAdapter: API key is required. Set GEMINI_API_KEY environment variable or pass it to the constructor."
+      );
+    }
+    this.client = new import_genai.GoogleGenAI({ apiKey: key });
+  }
+  async *stream(messages, modelId, options) {
+    if (!this.models.some((m) => m.id === modelId)) {
+      yield { type: "error", error: `Model "${modelId}" is not supported by the Gemini adapter` };
+      return;
+    }
+    const contents = toGeminiContents(messages);
+    if (contents.length === 0) {
+      yield { type: "error", error: "No valid messages to send" };
+      return;
+    }
+    try {
+      const result = await this.client.models.generateContentStream({
+        model: modelId,
+        contents
+      });
+      for await (const chunk of result) {
+        if (options?.signal?.aborted) break;
+        const text = chunk.text;
+        if (text) {
+          yield { type: "delta", content: text };
+        }
+      }
+      yield { type: "done" };
+    } catch (err) {
+      if (options?.signal?.aborted) return;
+      yield {
+        type: "error",
+        error: err instanceof Error ? err.message : "Gemini streaming error"
+      };
+    }
+  }
+};
+
+// src/registry.ts
+var ProviderRegistry = class {
+  adapters;
+  constructor(adapters) {
+    this.adapters = new Map(adapters.map((a) => [a.id, a]));
+  }
+  /** Returns the adapter for the given provider ID. Throws if not registered. */
+  get(providerId) {
+    const adapter = this.adapters.get(providerId);
+    if (!adapter) {
+      throw new Error(
+        `Provider "${providerId}" is not registered. Registered providers: ${[...this.adapters.keys()].join(", ")}`
+      );
+    }
+    return adapter;
+  }
+  /** Returns all registered adapters. Used to expose available providers to the client. */
+  getAll() {
+    return Array.from(this.adapters.values());
+  }
+};
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  ClaudeAdapter,
+  GeminiAdapter,
+  OpenAIAdapter,
+  ProviderRegistry
+});

package/dist/index.mjs

@@ -0,0 +1,227 @@
+// src/adapters/openai.ts
+import OpenAI from "openai";
+var MODELS = [
+  {
+    id: "gpt-4o",
+    name: "GPT-4o",
+    contextWindow: 128e3,
+    capabilities: { streaming: true, images: true, tools: false }
+  },
+  {
+    id: "gpt-4o-mini",
+    name: "GPT-4o Mini",
+    contextWindow: 128e3,
+    capabilities: { streaming: true, images: false, tools: false }
+  }
+];
+var VALID_ROLES = /* @__PURE__ */ new Set(["system", "user", "assistant"]);
+function isValidRole(role) {
+  return VALID_ROLES.has(role);
+}
+function toOpenAIMessages(messages) {
+  const result = [];
+  for (const msg of messages) {
+    if (!isValidRole(msg.role)) continue;
+    const text = msg.parts.filter((p) => p.type === "text").map((p) => p.text).join("\n").trim();
+    if (!text) continue;
+    result.push({ role: msg.role, content: text });
+  }
+  return result;
+}
+var OpenAIAdapter = class {
+  id = "openai";
+  name = "OpenAI";
+  models = MODELS;
+  client;
+  /**
+   * @param apiKey  If omitted, falls back to process.env.OPENAI_API_KEY.
+   *                Must never be passed from client-side code.
+   */
+  constructor(apiKey) {
+    const key = apiKey ?? process.env["OPENAI_API_KEY"];
+    if (!key) {
+      throw new Error(
+        "OpenAIAdapter: API key is required. Set OPENAI_API_KEY environment variable or pass it to the constructor."
+      );
+    }
+    this.client = new OpenAI({ apiKey: key });
+  }
+  async *stream(messages, modelId, options) {
+    if (!this.models.some((m) => m.id === modelId)) {
+      throw new Error(`Model "${modelId}" not supported by OpenAI adapter`);
+    }
+    const openAiMessages = toOpenAIMessages(messages);
+    try {
+      const completion = await this.client.chat.completions.create(
+        {
+          model: modelId,
+          messages: openAiMessages,
+          stream: true,
+          stream_options: { include_usage: true },
+          ...options?.maxTokens !== void 0 && { max_tokens: options.maxTokens },
+          ...options?.temperature !== void 0 && { temperature: options.temperature }
+        },
+        { signal: options?.signal }
+      );
+      let usage;
+      for await (const chunk of completion) {
+        const choice = chunk.choices?.[0];
+        const content = choice?.delta?.content;
+        if (typeof content === "string" && content.length > 0) {
+          yield { type: "delta", content };
+        }
+        const u = chunk.usage;
+        if (u != null && typeof u.prompt_tokens === "number" && typeof u.completion_tokens === "number" && typeof u.total_tokens === "number") {
+          usage = {
+            promptTokens: u.prompt_tokens,
+            completionTokens: u.completion_tokens,
+            totalTokens: u.total_tokens
+          };
+        }
+      }
+      yield { type: "done", usage };
+    } catch (err) {
+      if (options?.signal?.aborted) return;
+      yield {
+        type: "error",
+        error: err instanceof Error ? err.message : JSON.stringify(err)
+      };
+    }
+  }
+};
+
+// src/adapters/claude.ts
+var MODELS2 = [
+  {
+    id: "claude-3-5-sonnet-20241022",
+    name: "Claude 3.5 Sonnet",
+    contextWindow: 2e5,
+    capabilities: { streaming: true, images: true, tools: false }
+  }
+];
+var ClaudeAdapter = class {
+  id = "claude";
+  name = "Claude";
+  models = MODELS2;
+  // TODO: Implement real Anthropic API integration.
+  async *stream(_messages, _modelId) {
+    if (!this.models.some((m) => m.id === _modelId)) {
+      yield { type: "error", error: `Model "${_modelId}" is not supported by the Claude adapter` };
+      return;
+    }
+    yield {
+      type: "delta",
+      content: "[Claude integration is not yet implemented. This is a placeholder.]"
+    };
+    yield { type: "done" };
+  }
+};
+
+// src/adapters/gemini.ts
+import { GoogleGenAI } from "@google/genai";
+var MODELS3 = [
+  {
+    id: "gemini-2.5-flash",
+    name: "Gemini 2.5 Flash",
+    contextWindow: 1e6,
+    capabilities: { streaming: true, images: false, tools: false }
+  },
+  {
+    id: "gemini-2.5-pro",
+    name: "Gemini 2.5 Pro",
+    contextWindow: 2e6,
+    capabilities: { streaming: true, images: false, tools: false }
+  }
+];
+function toGeminiContents(messages) {
+  const contents = [];
+  for (const msg of messages) {
+    const text = msg.parts.filter((p) => p.type === "text").map((p) => p.text).join("\n").trim();
+    if (!text) continue;
+    const role = msg.role === "assistant" ? "model" : "user";
+    const prev = contents[contents.length - 1];
+    if (prev && prev.role === role) {
+      const existing = prev.parts[0];
+      if (existing) {
+        existing.text += "\n" + text;
+      }
+    } else {
+      contents.push({ role, parts: [{ text }] });
+    }
+  }
+  return contents;
+}
+var GeminiAdapter = class {
+  id = "gemini";
+  name = "Gemini";
+  models = MODELS3;
+  client;
+  constructor(apiKey) {
+    const key = apiKey ?? process.env["GEMINI_API_KEY"];
+    if (!key) {
+      throw new Error(
+        "GeminiAdapter: API key is required. Set GEMINI_API_KEY environment variable or pass it to the constructor."
+      );
+    }
+    this.client = new GoogleGenAI({ apiKey: key });
+  }
+  async *stream(messages, modelId, options) {
+    if (!this.models.some((m) => m.id === modelId)) {
+      yield { type: "error", error: `Model "${modelId}" is not supported by the Gemini adapter` };
+      return;
+    }
+    const contents = toGeminiContents(messages);
+    if (contents.length === 0) {
+      yield { type: "error", error: "No valid messages to send" };
+      return;
+    }
+    try {
+      const result = await this.client.models.generateContentStream({
+        model: modelId,
+        contents
+      });
+      for await (const chunk of result) {
+        if (options?.signal?.aborted) break;
+        const text = chunk.text;
+        if (text) {
+          yield { type: "delta", content: text };
+        }
+      }
+      yield { type: "done" };
+    } catch (err) {
+      if (options?.signal?.aborted) return;
+      yield {
+        type: "error",
+        error: err instanceof Error ? err.message : "Gemini streaming error"
+      };
+    }
+  }
+};
+
+// src/registry.ts
+var ProviderRegistry = class {
+  adapters;
+  constructor(adapters) {
+    this.adapters = new Map(adapters.map((a) => [a.id, a]));
+  }
+  /** Returns the adapter for the given provider ID. Throws if not registered. */
+  get(providerId) {
+    const adapter = this.adapters.get(providerId);
+    if (!adapter) {
+      throw new Error(
+        `Provider "${providerId}" is not registered. Registered providers: ${[...this.adapters.keys()].join(", ")}`
+      );
+    }
+    return adapter;
+  }
+  /** Returns all registered adapters. Used to expose available providers to the client. */
+  getAll() {
+    return Array.from(this.adapters.values());
+  }
+};
+export {
+  ClaudeAdapter,
+  GeminiAdapter,
+  OpenAIAdapter,
+  ProviderRegistry
+};

package/package.json

@@ -0,0 +1,37 @@
+{
+  "name": "@tigmart/ai-core",
+  "version": "0.0.1",
+  "description": "Server-side LLM provider adapters — Node.js only",
+  "main": "./dist/index.js",
+  "module": "./dist/index.mjs",
+  "types": "./dist/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.mjs",
+      "require": "./dist/index.js"
+    }
+  },
+  "files": [
+    "dist"
+  ],
+   "license": "MIT",
+  "publishConfig": {
+    "access": "public"
+  },
+  "scripts": {
+    "build": "tsup",
+    "dev": "tsup --watch",
+    "typecheck": "tsc --noEmit"
+  },
+  "dependencies": {
+    "@google/genai": "^1.48.0",
+    "@tigmart/ai-types": "workspace:*",
+    "openai": "^4.76.0"
+  },
+  "devDependencies": {
+    "@types/node": "^22.19.17",
+    "tsup": "^8.3.0",
+    "typescript": "^5.7.0"
+  }
+}