openai_classify.cpp

This example shows how to classify a document using the OpenAI API.

#include "docwire.h"
#include <iostream>
#include <sstream>
 
int main(int argc, char* argv[])
{
  using namespace docwire;
  std::stringstream out_stream;
 
  try
  {
    std::filesystem::path("document_processing_market_trends.odt") | content_type::detector{} | office_formats_parser{} | plain_text_exporter() | openai::classify({ "agreement", "invoice", "report", "legal", "other"}, environment::get("OPENAI_API_KEY").value()) | out_stream;
    ensure(out_stream.str()) == "report\n";
  }
  catch (const std::exception& e)
  {
    std::cerr << errors::diagnostic_message(e) << std::endl;
    return 1;
  }
 
  return 0;
}