Function main

examples/streaming_chat.cpp:21–179

#include <ai/ai.h>

int main() {
  std::cout << "AI SDK C++ - Streaming Chat Example\n";
  std::cout << "====================================\n\n";

  // Example 1: Basic streaming with iterator
  std::cout << "1. Basic streaming (iterator approach):\n";
  std::cout
      << "Prompt: Write a short story about a robot learning to paint.\n\n";
  std::cout << "Response: ";

  auto client = ai::openai::create_client();
  ai::GenerateOptions gen_options1;
  gen_options1.model = ai::openai::models::kGpt54;
  gen_options1.prompt =
      "Write a short story about a robot learning "
      "to paint. Keep it under 200 words.";
  ai::StreamOptions options1(std::move(gen_options1));
  auto stream1 = client.stream_text(options1);

  for (const auto& event : stream1) {
    if (event.is_text_delta()) {
      std::cout << event.text_delta << std::flush;
    } else if (event.is_error()) {
      std::cout << "\nError: " << event.error.value_or("Unknown error") << "\n";
      break;
    } else if (event.is_finish()) {
      std::cout << "\n\n[Stream finished]\n\n";
      break;
    }
  }

  // Example 2: Streaming with options and callbacks
  std::cout << "2. Streaming with callbacks:\n";
  std::cout << "Prompt: Explain quantum computing in simple terms.\n\n";

  // Set up callbacks
  std::string accumulated_text;
  int chunk_count = 0;

  auto text_callback = [&](const std::string& chunk) {
    std::cout << chunk << std::flush;
    accumulated_text += chunk;
    chunk_count++;
  };

  auto complete_callback = [&](const ai::GenerateResult& result) {
    std::cout << "\n\n[Stream completed]\n";
    std::cout << "Total chunks received: " << chunk_count << "\n";
    std::cout << "Final text length: " << accumulated_text.length()
              << " characters\n";
    std::cout << "Token usage: " << result.usage.total_tokens << " tokens\n";
    std::cout << "Finish reason: " << result.finishReasonToString() << "\n\n";
  };

  auto error_callback = [](const std::string& error) {
    std::cout << "\nError in stream: " << error << "\n\n";
  };