@sahan-sac/ai-core 0.3.0 → 0.3.1

package/dist/chunk-MRNQPHWI.js

@@ -0,0 +1,194 @@
+import {
+  getVertexAccessToken
+} from "./chunk-BL4J2JNX.js";
+import {
+  imageModels,
+  paidAllowed,
+  vertexConfigured
+} from "./chunk-RBSLNDFW.js";
+
+// src/image.ts
+import "server-only";
+var DEFAULT_IMAGE_MODELS = [
+  { model: "gemini-3.1-flash-image", location: "global" },
+  { model: "gemini-2.5-flash-image", location: "us-central1" }
+];
+function defaultLocation(model) {
+  return /^gemini-3/.test(model) ? "global" : "us-central1";
+}
+function imageModelChain(config) {
+  const chain = config.model ? [{ model: config.model, location: config.location ?? defaultLocation(config.model) }] : [];
+  for (const candidate of DEFAULT_IMAGE_MODELS) {
+    if (!chain.some((entry) => entry.model === candidate.model)) chain.push(candidate);
+  }
+  return chain;
+}
+function endpoint(project, target, method) {
+  const host = target.location === "global" ? "aiplatform.googleapis.com" : `${target.location}-aiplatform.googleapis.com`;
+  return `https://${host}/v1/projects/${project}/locations/${target.location}/publishers/google/models/${target.model}:${method}`;
+}
+function requestFor(target, prompt, aspectRatio) {
+  if (target.model.startsWith("imagen-")) {
+    return {
+      method: "predict",
+      body: {
+        instances: [{ prompt }],
+        parameters: { sampleCount: 1, aspectRatio, safetySetting: "block_medium_and_above", personGeneration: "allow_adult" }
+      }
+    };
+  }
+  return {
+    method: "generateContent",
+    body: {
+      contents: [{ role: "user", parts: [{ text: prompt }] }],
+      generationConfig: { responseModalities: ["IMAGE"], imageConfig: { aspectRatio } }
+    }
+  };
+}
+function extractImage(data) {
+  const prediction = data?.predictions?.[0];
+  if (prediction?.bytesBase64Encoded) return { base64: prediction.bytesBase64Encoded, mimeType: prediction.mimeType || "image/png" };
+  const parts = data?.candidates?.[0]?.content?.parts;
+  const inline = parts?.find((part) => part.inlineData?.data)?.inlineData;
+  return inline?.data ? { base64: inline.data, mimeType: inline.mimeType || "image/png" } : null;
+}
+async function generateImageVertex(prompt, config, options = {}) {
+  const fetchImpl = options.fetchImpl ?? fetch;
+  const aspectRatio = options.aspectRatio ?? "16:9";
+  let accessToken;
+  try {
+    accessToken = await getVertexAccessToken(
+      { clientEmail: config.clientEmail, privateKey: config.privateKey, tokenUri: config.tokenUri },
+      fetchImpl
+    );
+  } catch (error) {
+    return { ok: false, error: `Image generation failed: ${error instanceof Error ? error.message : "authentication error"}` };
+  }
+  const failures = [];
+  for (const target of imageModelChain(config)) {
+    if (options.signal?.aborted) return { ok: false, error: "Image generation timed out." };
+    const { method, body } = requestFor(target, prompt, aspectRatio);
+    try {
+      const response = await fetchImpl(endpoint(config.project, target, method), {
+        method: "POST",
+        headers: { "content-type": "application/json", authorization: `Bearer ${accessToken}` },
+        signal: options.signal,
+        body: JSON.stringify(body)
+      });
+      if (!response.ok) {
+        failures.push(`${target.model}: HTTP ${response.status}`);
+        continue;
+      }
+      const image = extractImage(await response.json());
+      if (image) return { ok: true, ...image };
+      failures.push(`${target.model}: no image data`);
+    } catch (error) {
+      if (options.signal?.aborted) return { ok: false, error: "Image generation timed out." };
+      failures.push(`${target.model}: ${error instanceof Error ? error.message : "transport error"}`);
+    }
+  }
+  return { ok: false, error: `Image generation failed (${failures.join("; ")}).` };
+}
+function vertexImageAvailable(env) {
+  return vertexConfigured(env);
+}
+var FLUX_SIZE = {
+  "16:9": { width: 1344, height: 768 },
+  "4:3": { width: 1024, height: 768 },
+  "1:1": { width: 1024, height: 1024 }
+};
+function nvidiaImageRequest(model, prompt, aspectRatio, seed) {
+  const size = FLUX_SIZE[aspectRatio];
+  if (/schnell/.test(model)) return { prompt, ...size, steps: 4, seed };
+  return { prompt, ...size, steps: 30, cfg_scale: 3.5, mode: "base", seed };
+}
+function extractNvidiaImage(data) {
+  const artifact = data?.artifacts?.[0];
+  if (!artifact?.base64 || artifact.finishReason && artifact.finishReason !== "SUCCESS") return null;
+  const mimeType = artifact.base64.startsWith("/9j/") ? "image/jpeg" : artifact.base64.startsWith("UklG") ? "image/webp" : "image/png";
+  return { base64: artifact.base64, mimeType };
+}
+async function postForImage(model, url, init, extract, options) {
+  try {
+    const response = await (options.fetchImpl ?? fetch)(url, {
+      method: "POST",
+      headers: { "content-type": "application/json", ...init.headers },
+      signal: options.signal,
+      body: JSON.stringify(init.body)
+    });
+    if (!response.ok) return { ok: false, error: `${model}: HTTP ${response.status}` };
+    const image = extract(await response.json());
+    return image ? { ok: true, ...image } : { ok: false, error: `${model}: no image (filtered or empty)` };
+  } catch (error) {
+    if (options.signal?.aborted) return { ok: false, error: "Image generation timed out." };
+    return { ok: false, error: `${model}: ${error instanceof Error ? error.message : "transport error"}` };
+  }
+}
+function generateImageNvidia(prompt, config, aspectRatio, options) {
+  return postForImage(
+    config.model,
+    `https://ai.api.nvidia.com/v1/genai/${config.model}`,
+    {
+      headers: { authorization: `Bearer ${config.apiKey}`, accept: "application/json" },
+      body: nvidiaImageRequest(config.model, prompt, aspectRatio, Math.floor(Math.random() * 2 ** 31))
+    },
+    extractNvidiaImage,
+    options
+  );
+}
+function generateImageGemini(prompt, config, aspectRatio, options) {
+  return postForImage(
+    config.model,
+    `https://generativelanguage.googleapis.com/v1beta/models/${config.model}:generateContent`,
+    {
+      headers: { "x-goog-api-key": config.apiKey },
+      body: { contents: [{ role: "user", parts: [{ text: prompt }] }], generationConfig: { responseModalities: ["IMAGE"], imageConfig: { aspectRatio } } }
+    },
+    extractImage,
+    options
+  );
+}
+function imageConfigFromEnv(env) {
+  const models = imageModels(env);
+  const paid = paidAllowed(env);
+  const config = {};
+  if (env.NVIDIA_API_KEY) config.nvidia = { apiKey: env.NVIDIA_API_KEY, model: models.nvidia };
+  if (paid && env.GEMINI_API_KEY) config.gemini = { apiKey: env.GEMINI_API_KEY, model: models.gemini };
+  if (paid && vertexImageAvailable(env)) {
+    config.vertex = {
+      clientEmail: env.GOOGLE_CLIENT_EMAIL,
+      privateKey: env.GOOGLE_PRIVATE_KEY,
+      tokenUri: env.GOOGLE_TOKEN_URI,
+      project: env.GOOGLE_CLOUD_PROJECT,
+      model: models.vertex
+    };
+  }
+  return config.nvidia || config.gemini || config.vertex ? config : null;
+}
+async function generateImage(prompt, config, options = {}) {
+  const aspectRatio = options.aspectRatio ?? "16:9";
+  const attempts = [];
+  if (config.nvidia) attempts.push(() => generateImageNvidia(prompt, config.nvidia, aspectRatio, options));
+  if (config.gemini) attempts.push(() => generateImageGemini(prompt, config.gemini, aspectRatio, options));
+  if (config.vertex) attempts.push(() => generateImageVertex(prompt, config.vertex, options));
+  const failures = [];
+  for (const run of attempts) {
+    if (options.signal?.aborted) return { ok: false, error: "Image generation timed out." };
+    const outcome = await run();
+    if (outcome.ok) return outcome;
+    failures.push(outcome.error);
+  }
+  return { ok: false, error: failures.length ? `Image generation failed (${failures.join("; ")}).` : "No image provider is configured." };
+}
+
+export {
+  DEFAULT_IMAGE_MODELS,
+  imageModelChain,
+  extractImage,
+  generateImageVertex,
+  vertexImageAvailable,
+  nvidiaImageRequest,
+  extractNvidiaImage,
+  imageConfigFromEnv,
+  generateImage
+};

package/dist/chunk-PMC3UWQ2.js

@@ -0,0 +1,10 @@
+// src/log.ts
+var consoleAiLogger = {
+  warn(message, meta) {
+    console.warn(JSON.stringify({ level: "warn", message, ...meta }));
+  }
+};
+
+export {
+  consoleAiLogger
+};