varg.ai-sdk 0.1.0 → 0.4.0-alpha.1

package/lib/replicate.ts

@@ -1,304 +0,0 @@
-#!/usr/bin/env bun
-
-/**
- * replicate.com api wrapper for video/image generation
- * supports models like minimax, kling, luma, stable diffusion
- */
-
-import Replicate from "replicate";
-
-const replicate = new Replicate({
-  auth: process.env.REPLICATE_API_TOKEN || "",
-});
-
-// types
-export interface RunVideoOptions {
-  model: string;
-  input: Record<string, unknown>;
-}
-
-export interface RunImageOptions {
-  model: string;
-  input: Record<string, unknown>;
-}
-
-// core functions
-export async function runModel(model: string, input: Record<string, unknown>) {
-  console.log(`[replicate] running ${model}...`);
-
-  try {
-    const output = await replicate.run(model as `${string}/${string}`, {
-      input,
-    });
-    console.log(`[replicate] completed`);
-    return output;
-  } catch (error) {
-    console.error(`[replicate] error:`, error);
-    throw error;
-  }
-}
-
-export async function runVideo(options: RunVideoOptions) {
-  const { model, input } = options;
-
-  if (!model || !input) {
-    throw new Error("model and input are required");
-  }
-
-  return await runModel(model, input);
-}
-
-export async function runImage(options: RunImageOptions) {
-  const { model, input } = options;
-
-  if (!model || !input) {
-    throw new Error("model and input are required");
-  }
-
-  return await runModel(model, input);
-}
-
-// popular models
-export const MODELS = {
-  // video generation
-  VIDEO: {
-    MINIMAX: "minimax/video-01",
-    KLING: "fofr/kling-v1.5",
-    LUMA: "fofr/ltx-video",
-    RUNWAY_GEN3: "replicate/runway-gen3-turbo",
-    WAN_2_5: "wan-video/wan-2.5-i2v",
-  },
-  // image generation
-  IMAGE: {
-    FLUX_PRO: "black-forest-labs/flux-1.1-pro",
-    FLUX_DEV: "black-forest-labs/flux-dev",
-    FLUX_SCHNELL: "black-forest-labs/flux-schnell",
-    STABLE_DIFFUSION: "stability-ai/sdxl",
-  },
-};
-
-// cli
-async function cli() {
-  const args = process.argv.slice(2);
-  const command = args[0];
-
-  if (!command || command === "help") {
-    console.log(`
-usage:
-  bun run lib/replicate.ts <command> [args]
-
-commands:
-  video <model> <prompt> [imageUrl]     generate video
-  image <model> <prompt>                generate image
-  minimax <prompt> [imageUrl]           generate video with minimax-01
-  kling <prompt> [imageUrl]             generate video with kling-v1.5
-  wan <imageUrl> <audioUrl> <prompt>    generate talking video with wan 2.5
-  flux <prompt>                         generate image with flux-dev
-  list                                  list recent predictions
-  get <predictionId>                    get prediction by id
-  help                                  show this help
-
-examples:
-  bun run lib/replicate.ts minimax "person walking on beach"
-  bun run lib/replicate.ts minimax "camera zoom in" https://example.com/img.jpg
-  bun run lib/replicate.ts kling "cinematic city scene"
-  bun run lib/replicate.ts wan https://image.jpg https://audio.mp3 "person talking to camera"
-  bun run lib/replicate.ts flux "cyberpunk cityscape"
-  bun run lib/replicate.ts video "minimax/video-01" "dancing robot"
-  bun run lib/replicate.ts image "black-forest-labs/flux-dev" "sunset landscape"
-
-environment:
-  REPLICATE_API_TOKEN - your replicate api token
-    `);
-    process.exit(0);
-  }
-
-  try {
-    switch (command) {
-      case "minimax": {
-        const prompt = args[1];
-        const imageUrl = args[2];
-
-        if (!prompt) {
-          throw new Error("prompt is required");
-        }
-
-        const input: Record<string, unknown> = { prompt };
-        if (imageUrl) {
-          input.first_frame_image = imageUrl;
-        }
-
-        const output = await runVideo({
-          model: MODELS.VIDEO.MINIMAX,
-          input,
-        });
-
-        console.log(`[replicate] minimax output:`, output);
-        break;
-      }
-
-      case "kling": {
-        const prompt = args[1];
-        const imageUrl = args[2];
-
-        if (!prompt) {
-          throw new Error("prompt is required");
-        }
-
-        const input: Record<string, unknown> = { prompt };
-        if (imageUrl) {
-          input.image = imageUrl;
-        }
-
-        const output = await runVideo({
-          model: MODELS.VIDEO.KLING,
-          input,
-        });
-
-        console.log(`[replicate] kling output:`, output);
-        break;
-      }
-
-      case "wan": {
-        const imageUrl = args[1];
-        const audioUrl = args[2];
-        const prompt = args[3];
-        const duration = args[4] ? Number.parseInt(args[4], 10) : 10;
-        const resolution = args[5] || "480p";
-
-        if (!imageUrl || !audioUrl || !prompt) {
-          throw new Error("imageUrl, audioUrl, and prompt are required");
-        }
-
-        const input: Record<string, unknown> = {
-          image: imageUrl,
-          audio: audioUrl,
-          prompt,
-          duration,
-          resolution,
-          enable_prompt_expansion: true,
-        };
-
-        console.log(`[replicate] running wan 2.5...`);
-        console.log(`[replicate] this may take 3-5 minutes...`);
-
-        const output = await runVideo({
-          model: MODELS.VIDEO.WAN_2_5,
-          input,
-        });
-
-        console.log(`[replicate] wan 2.5 output:`, output);
-        break;
-      }
-
-      case "list": {
-        console.log(`[replicate] fetching recent predictions...`);
-        const predictions = await replicate.predictions.list();
-
-        console.log(`\nrecent predictions:\n`);
-        for (const pred of predictions.results.slice(0, 10)) {
-          console.log(`id: ${pred.id}`);
-          console.log(`status: ${pred.status}`);
-          console.log(`model: ${pred.version}`);
-          console.log(`created: ${pred.created_at}`);
-          if (pred.output) {
-            console.log(
-              `output: ${JSON.stringify(pred.output).substring(0, 100)}...`,
-            );
-          }
-          console.log(`---`);
-        }
-        break;
-      }
-
-      case "get": {
-        const predictionId = args[1];
-
-        if (!predictionId) {
-          throw new Error("predictionId is required");
-        }
-
-        console.log(`[replicate] fetching prediction ${predictionId}...`);
-        const prediction = await replicate.predictions.get(predictionId);
-
-        console.log(`\nstatus: ${prediction.status}`);
-        console.log(`model: ${prediction.version}`);
-        console.log(`created: ${prediction.created_at}`);
-
-        if (prediction.status === "succeeded") {
-          console.log(`\noutput:`);
-          console.log(JSON.stringify(prediction.output, null, 2));
-        } else if (prediction.status === "failed") {
-          console.log(`\nerror: ${prediction.error}`);
-        } else {
-          console.log(`\nstill processing...`);
-        }
-        break;
-      }
-
-      case "flux": {
-        const prompt = args[1];
-
-        if (!prompt) {
-          throw new Error("prompt is required");
-        }
-
-        const output = await runImage({
-          model: MODELS.IMAGE.FLUX_DEV,
-          input: { prompt },
-        });
-
-        console.log(`[replicate] flux output:`, output);
-        break;
-      }
-
-      case "video": {
-        const model = args[1];
-        const prompt = args[2];
-        const imageUrl = args[3];
-
-        if (!model || !prompt) {
-          throw new Error("model and prompt are required");
-        }
-
-        const input: Record<string, unknown> = { prompt };
-        if (imageUrl) {
-          input.image = imageUrl;
-        }
-
-        const output = await runVideo({ model, input });
-        console.log(`[replicate] video output:`, output);
-        break;
-      }
-
-      case "image": {
-        const model = args[1];
-        const prompt = args[2];
-
-        if (!model || !prompt) {
-          throw new Error("model and prompt are required");
-        }
-
-        const output = await runImage({
-          model,
-          input: { prompt },
-        });
-
-        console.log(`[replicate] image output:`, output);
-        break;
-      }
-
-      default:
-        console.error(`unknown command: ${command}`);
-        console.log(`run 'bun run lib/replicate.ts help' for usage`);
-        process.exit(1);
-    }
-  } catch (error) {
-    console.error(`[replicate] error:`, error);
-    process.exit(1);
-  }
-}
-
-if (import.meta.main) {
-  cli();
-}

package/output.txt

@@ -1 +0,0 @@
- Let's say I've just joined the Roark team as a marketer and I want to add a new article to the website to boost our SEO. The devs won't get to this task anytime soon, but thanks to YoloCode AI, I can take care of it myself. Any changes I make show up instantly on the right so I can see exactly how it looked in production before opening a pull request. So now I'm ready to submit and push this feature and here we go!

package/test-import.ts

@@ -1,7 +0,0 @@
-import { generateImage, imageToVideo, uploadFromUrl } from "./index";
-
-console.log("✓ imports successful");
-console.log("available functions:");
-console.log("- generateImage:", typeof generateImage);
-console.log("- imageToVideo:", typeof imageToVideo);
-console.log("- uploadFromUrl:", typeof uploadFromUrl);

package/test-services.ts

@@ -1,97 +0,0 @@
-#!/usr/bin/env bun
-
-/**
- * test script for all sdk services
- * saves all media files to media/ directory
- */
-
-import { generateImage as generateFalImage } from "./lib/ai-sdk/fal";
-import { generateImage as generateReplicateImage } from "./lib/ai-sdk/replicate";
-import { textToSpeech } from "./lib/elevenlabs";
-import { runImage } from "./lib/replicate";
-
-async function testFalImage() {
-  console.log("\n=== testing fal image generation ===");
-
-  const result = await generateFalImage({
-    prompt: "a cozy coffee shop interior with warm lighting",
-    model: "fal-ai/flux/dev",
-    aspectRatio: "16:9",
-  });
-
-  if (result.image.uint8Array) {
-    const filename = "media/fal-coffee-shop.png";
-    await Bun.write(filename, result.image.uint8Array);
-    console.log(`✓ saved to ${filename}`);
-  }
-}
-
-async function testReplicateImage() {
-  console.log("\n=== testing replicate image generation ===");
-
-  const result = await generateReplicateImage({
-    prompt: "a mystical forest with glowing mushrooms",
-    model: "black-forest-labs/flux-dev",
-    aspectRatio: "1:1",
-  });
-
-  if (result.image.uint8Array) {
-    const filename = "media/replicate-forest.png";
-    await Bun.write(filename, result.image.uint8Array);
-    console.log(`✓ saved to ${filename}`);
-  }
-}
-
-async function testReplicateClient() {
-  console.log("\n=== testing replicate client (flux) ===");
-
-  try {
-    const output = await runImage({
-      model: "black-forest-labs/flux-schnell",
-      input: { prompt: "a serene zen garden with cherry blossoms" },
-    });
-
-    // output is array of FileOutput objects
-    if (Array.isArray(output) && output[0]) {
-      const imageUrl = output[0].toString();
-      console.log(`✓ generated image: ${imageUrl}`);
-
-      // download and save
-      const response = await fetch(imageUrl);
-      const buffer = await response.arrayBuffer();
-      const filename = "media/replicate-zen-garden.png";
-      await Bun.write(filename, buffer);
-      console.log(`✓ saved to ${filename}`);
-    }
-  } catch (error) {
-    console.error("✗ replicate client error:", error);
-  }
-}
-
-async function testElevenlabs() {
-  console.log("\n=== testing elevenlabs tts ===");
-
-  try {
-    await textToSpeech({
-      text: "welcome to varg ai sdk, your complete video production toolkit",
-      voiceId: "21m00Tcm4TlvDq8ikWAM",
-      outputPath: "media/welcome.mp3",
-    });
-    console.log("✓ saved to media/welcome.mp3");
-  } catch (error) {
-    console.error("✗ elevenlabs error:", error);
-  }
-}
-
-async function main() {
-  console.log("🚀 testing varg.ai sdk services...\n");
-
-  await testFalImage();
-  await testReplicateImage();
-  await testReplicateClient();
-  await testElevenlabs();
-
-  console.log("\n✨ all tests complete! check media/ directory");
-}
-
-main().catch(console.error);

package/utilities/s3.ts

@@ -1,147 +0,0 @@
-#!/usr/bin/env bun
-/**
- * cloudflare r2 / s3 storage wrapper
- * usage: bun run utilities/s3.ts <command> <args>
- */
-
-import {
-  GetObjectCommand,
-  PutObjectCommand,
-  S3Client,
-} from "@aws-sdk/client-s3";
-import { getSignedUrl } from "@aws-sdk/s3-request-presigner";
-
-const client = new S3Client({
-  region: "auto",
-  endpoint: process.env.CLOUDFLARE_R2_API_URL,
-  credentials: {
-    accessKeyId: process.env.CLOUDFLARE_ACCESS_KEY_ID || "",
-    secretAccessKey: process.env.CLOUDFLARE_ACCESS_SECRET || "",
-  },
-});
-
-const BUCKET = process.env.CLOUDFLARE_R2_BUCKET || "m";
-
-export async function uploadFile(
-  filePath: string,
-  objectKey: string,
-): Promise<string> {
-  console.log(`[s3] uploading ${filePath} to ${objectKey}`);
-
-  const file = Bun.file(filePath);
-  const buffer = await file.arrayBuffer();
-
-  await client.send(
-    new PutObjectCommand({
-      Bucket: BUCKET,
-      Key: objectKey,
-      Body: new Uint8Array(buffer),
-    }),
-  );
-
-  return getPublicUrl(objectKey);
-}
-
-export async function uploadFromUrl(
-  url: string,
-  objectKey: string,
-): Promise<string> {
-  console.log(`[s3] downloading from ${url}`);
-
-  const response = await fetch(url);
-  const buffer = await response.arrayBuffer();
-
-  console.log(`[s3] uploading to ${objectKey}`);
-
-  await client.send(
-    new PutObjectCommand({
-      Bucket: BUCKET,
-      Key: objectKey,
-      Body: new Uint8Array(buffer),
-    }),
-  );
-
-  return getPublicUrl(objectKey);
-}
-
-export async function generatePresignedUrl(
-  objectKey: string,
-  expiresIn = 3600,
-): Promise<string> {
-  const command = new GetObjectCommand({
-    Bucket: BUCKET,
-    Key: objectKey,
-  });
-
-  return await getSignedUrl(client, command, { expiresIn });
-}
-
-export function getPublicUrl(objectKey: string): string {
-  const endpoint = process.env.CLOUDFLARE_R2_API_URL || "";
-
-  if (endpoint.includes("localhost")) {
-    return `${endpoint}/${BUCKET}/${objectKey}`;
-  }
-
-  return `http://s3.varg.ai/${BUCKET}/${objectKey}`;
-}
-
-// cli runner
-if (import.meta.main) {
-  const [command, ...args] = process.argv.slice(2);
-
-  switch (command) {
-    case "upload": {
-      if (!args[0] || !args[1]) {
-        console.log(`
-usage:
-  bun run utilities/s3.ts upload <filePath> <objectKey>
-        `);
-        process.exit(1);
-      }
-      const uploadResult = await uploadFile(args[0], args[1]);
-      console.log(uploadResult);
-      break;
-    }
-
-    case "upload_from_url": {
-      if (!args[0] || !args[1]) {
-        console.log(`
-usage:
-  bun run utilities/s3.ts upload_from_url <url> <objectKey>
-        `);
-        process.exit(1);
-      }
-      const urlUploadResult = await uploadFromUrl(args[0], args[1]);
-      console.log(urlUploadResult);
-      break;
-    }
-
-    case "presigned_url": {
-      if (!args[0]) {
-        console.log(`
-usage:
-  bun run utilities/s3.ts presigned_url <objectKey> [expiresIn]
-        `);
-        process.exit(1);
-      }
-      const expiresIn = args[1] ? Number.parseInt(args[1], 10) : 3600;
-      if (Number.isNaN(expiresIn)) {
-        console.error("expiresIn must be a valid number");
-        process.exit(1);
-      }
-      const presignedUrl = await generatePresignedUrl(args[0], expiresIn);
-      console.log(presignedUrl);
-      break;
-    }
-
-    default:
-      console.log(`
-usage:
-  bun run utilities/s3.ts upload <filePath> <objectKey>
-  bun run utilities/s3.ts upload_from_url <url> <objectKey>
-  bun run utilities/s3.ts presigned_url <objectKey> [expiresIn]
-      `);
-      process.exit(1);
-  }
-}