cloudflare_ai 1.4.0

This is a Dart package for Cloudflare Workers AI. It currently supports Text Generation and Image Generation, Text Summarization, and Image Generation Models.

Cloudflare-AI-Dart

A Dart client for accessing the Cloudflare AI API.

Features

• ✅ Text Generation
• ✅ Text Summarization
• ✅ Image Generation
• ✅ Text Classification
• ✅ Language Translation
• ✅ Text Chat

Installation

Run this command:

dart pub add cloudflare_ai

or

Add this to your package's pubspec.yaml file:

dependencies:
  cloudflare_ai: ^1.3.5

Usage

Text Generation

import 'package:cloudflare_ai/cloudflare_ai.dart';

void main() async {
  // Initialize a TextGenerationModel
  TextGenerationModel model = TextGenerationModel(
    accountId: "Your Account ID",
    apiKey: "Your API Key",
    model: TextGenerationModels.GEMMA_7B_IT,
  );

  // Generate Text for a prompt
  TextGenerationResponse res = await model
      .generateText("Write a story about a robot, living on the moon");

  if (res.success && res.result != null) {
    print(res.result?.response);
  } else {
    print(res.errors);
  }
}

Supported Models:

• LLAMA_2_7B
• LLAMA_2_7B_INT8
• MISTRAL_7B
• CODE_LLAMA_7B
• CODE_LLAMA_2_13B
• ZEPHYR_7B
• MISTRAL_7B_AWQ_V01
• MISTRAL_7B_AWQ_V02
• OPENHERMES_MISTRAL_7B
• NEURAL_CHAT_7B
• LLAMA_GUARD_7B
• DEEPSEEK_CODER_6_7_BASE
• DEEPSEEK_CODER_6_7_INSTRUCT
• DEEPSEEK_MATH_7B_BASE
• DEEPSEEK_MATH_7B_INSTRUCT
• OPENCHAT_3_5
• PHI_2
• TINYLAMA_1_1B
• DISCOLM_GERMAN_7B
• QWEN_1_5_0_5B_CHAT
• QWEN1_5_1_8B_CHAT
• QWEN_1_5_7B_CHAT_AWQ
• QWEN_1_5_14B_CHAT_AWQ
• FALCON_7B_INSTRUCT
• GEMMA_2B_IT_LORA
• GEMMA_7B_IT
• GEMMA_7B_IT_LORA
• HERMES_2_PRO_7B
• LLAMA_2_7B_CHAT_HF_LORA
• LLAMA_3_8B_INSTRUCT
• UNA_CYBERTRON_7B_V2_BF16
• STARLING_LM_7B_BETA
• SQL_CODER_7B_2

Text Summarization

import 'package:cloudflare_ai/cloudflare_ai.dart';

void main() async {
  // Initialize a TextSummarizationModel
  TextSummarizationModel model = TextSummarizationModel(
    accountId: "Your Account ID",
    apiKey: "Your API Key",
    model: TextSummarizationModels.BART_LARGE_CNN,
  );

  // Summarize Text
  TextSummarizationResponse res = await model.summarize(
      "The quick brown fox jumps over the lazy dog",
      maxLength: 10,
    );

  if (res.success && res.result != null) {
    print(res.result?.response);
  } else {
    print(res.errors);
  }
}

Supported Models:

• BART_LARGE_CNN

Text To Image

import 'package:cloudflare_ai/cloudflare_ai.dart';

void main() async {
  // Initialize a TextToImageModel
  TextToImageModel model = TextToImageModel(
    accountId: "Your Account ID",
    apiKey: "Your API Key"
    model: TextToImageModels.DREAMSHAPER_8_LCM,
  );

  // Generate Image from Text
  TextToImageResponse res = await model.generateImage("A beautiful sunset over the ocean");

  // Save Image to File
  res.saveImage("sunset.jpg");
}

Supported Models:

• DREAMSHAPER_8_LCM
• STABLE_DIFFUSION_XL_BASE_1
• STABLE_DIFFUSION_XL_LIGHTNING

Text Classification

import 'package:cloudflare_ai/cloudflare_ai.dart';

void main() async {
  // Initialize a TextClassificationModel
  TextClassificationModel model = TextClassificationModel(
    accountId: "Your Account ID",
    apiKey: "Your API Key"
    model: TextClassificationModels.DISTILBERT_SST_2_INT8,
  );

  // Classify Text
  TextClassificationResponse res = await model.classifyText("A beautiful sunset over the ocean");

  if (res.success && res.result != null) {
    print(
        'Positive Confidence level: ${res.result?.positive}');
    print(
        'Negative Confidence level: ${res.result?.negative}');
  } else {
    print(res.errors);
  }
}

Supported Models:

• DISTILBERT_SST_2_INT8

Language Translation

import 'package:cloudflare_ai/cloudflare_ai.dart';

void main() async {
  // Initialize a LanguageTranslationModel
  LanguageTranslationModel model = LanguageTranslationModel(
    accountId: "Your Account ID",
    apiKey: "Your API Key",
    model: LanguageTranslationModels.M2M100_1_2B,
  );

  // Translate Text
  LanguageTranslationResponse res = await model.translate("A beautiful sunset over the ocean", Languages.English, Languages.French);

  if (res.success && res.result != null) {
    print(res.result?.response);
  } else {
    print(res.errors);
  }
}

Supported Models:

• M2M100_1_2B

Text Chat

import 'package:cloudflare_ai/cloudflare_ai.dart';

void main() async {
  // Initialize a TextChatModel
  TextChatModel model = TextChatModel(
    accountId: "Your Account ID",
    apiKey: "Your API Key",
    model: TextChatModels.GEMMA_7B_IT,
  );

  // Load previous messages
  model.loadMessages([
    {
      "role": "user",
      "content": "Hello!",
    },
    {
      "role": "assistant",
      "content": "Hello! How may I help you?",
    },
  ]);

  // Start Chat
  ChatMessage message = await model.chat("Hello");

  print(message.content);
}

Supported Models: Same as Text Generation Models