Function test_multi_step

examples/test_tool_integration.cpp:121–183

}

void test_multi_step() {
  std::cout << "\n=== Testing Multi-Step Tool Calling ===\n";

  auto client = ai::openai::create_client();

  // Create tools for multi-step
  ai::ToolSet tools = {
      {"get_number",
       ai::create_simple_tool(
           "get_number", "Get a number for calculation",
           {{"description", "string"}},
           [](const ai::JsonValue& args,
              const ai::ToolExecutionContext& context) -> ai::JsonValue {
             std::string desc = args["description"].get<std::string>();
             std::cout << "🔢 Getting number for: " << desc << std::endl;
             return ai::JsonValue{{"number", 42}};
           })},
      {"calculate",
       ai::create_simple_tool(
           "calculate", "Perform a calculation",
           {{"operation", "string"}, {"a", "number"}, {"b", "number"}},
           [](const ai::JsonValue& args,
              const ai::ToolExecutionContext& context) -> ai::JsonValue {
             std::string op = args["operation"].get<std::string>();
             double a = args["a"].get<double>();
             double b = args["b"].get<double>();

             std::cout << "🧮 Calculating: " << a << " " << op << " " << b
                       << std::endl;

             double result = 0;
             if (op == "add")
               result = a + b;
             else if (op == "multiply")
               result = a * b;

             return ai::JsonValue{{"result", result}};
           })}};

  ai::GenerateOptions options;
  options.model = ai::openai::models::kGpt54;
  options.prompt = "Get a number and then multiply it by 2. Show me the steps.";
  options.tools = tools;
  options.max_steps = 5;  // Enable multi-step
  options.max_tokens = 200;

  std::cout << "Making multi-step request...\n";

  auto result = client.generate_text(options);

  if (result) {
    std::cout << "✅ Multi-step request successful!\n";
    std::cout << "Final response: " << result.text << "\n";
    std::cout << "Total steps: " << result.steps.size() << "\n";
    std::cout << "Total tool calls: " << result.get_all_tool_calls().size()
              << "\n";
    std::cout << "Total tool results: " << result.get_all_tool_results().size()
              << "\n";