test_anthropic.cpp

#include "ai/logger.h"

#include <iostream>

#include <ai/ai.h>

int main() {
  try {
    // Enable debug logging
    ai::logger::install_logger(std::make_shared<ai::logger::ConsoleLogger>(
        ai::logger::LogLevel::kLogLevelInfo));

    // Create Anthropic client
    auto client = ai::anthropic::create_client();

    // Test simple generation
    std::cout << "Testing Anthropic text generation...\n\n";

    ai::GenerateOptions options(ai::anthropic::models::kClaudeHaiku45,
                                "You are a helpful assistant.",
                                "Why is the sky blue? Give a short answer.");

    auto result = client.generate_text(options);

    if (result) {
      std::cout << "Response: " << result.text << "\n";
      std::cout << "Model: " << result.model.value_or("unknown") << "\n";
      std::cout << "Tokens used: " << result.usage.total_tokens << "\n";
      std::cout << "Finish reason: " << result.finishReasonToString() << "\n";
    } else {
      std::cout << "Error: " << result.error_message() << "\n";
    }

    // Test streaming
    std::cout << "\nTesting streaming...\n";

    ai::GenerateOptions stream_opts(
        ai::anthropic::models::kClaudeHaiku45,
        "Count from 1 to 5 slowly and with each number say 'tick'");
    ai::StreamOptions stream_options(stream_opts);

    auto stream = client.stream_text(stream_options);

    for (const auto& event : stream) {
      if (event.is_text_delta()) {
        std::cout << event.text_delta << std::flush;
      } else if (event.is_error()) {
        std::cout << "\nStream error: " << event.error.value_or("unknown")
                  << "\n";
      } else if (event.is_finish()) {
        std::cout << "\n\nStream finished.\n";
        if (event.usage.has_value()) {
          std::cout << "Total tokens: " << event.usage->total_tokens << "\n";
        } else {
          std::cout << "Note: Token usage data may not be available in "
                       "streaming mode.\n";
        }
      }
    }

  } catch (const std::exception& e) {
    std::cerr << "Exception: " << e.what() << "\n";
    return 1;
  }

  return 0;
}