deep_research_graph.h

Go to the documentation of this file.

 
#pragma once
 
#include <neograph/graph/engine.h>
#include <neograph/provider.h>
#include <neograph/tool.h>
#include <cstdint>
#include <memory>
#include <string>
#include <vector>
 
namespace neograph::graph {
 
struct DeepResearchConfig {
    std::string model;                        
    int max_supervisor_iterations = 2;        
    int max_concurrent_researchers = 2;       
    int max_researcher_iterations = 2;        
 
    ProviderControls controls{};
    std::uint64_t brief_max_output_tokens      = 800;
    std::uint64_t clarify_max_output_tokens    = 200;
    std::uint64_t supervisor_max_output_tokens = 2048;
    std::uint64_t researcher_max_output_tokens = 2048;
    std::uint64_t compress_max_output_tokens   = 2048;
    std::uint64_t report_max_output_tokens     = 4096;
    std::uint64_t max_output_budget            = 16384;
    RetryPolicy retry{2, 15'000, 2.0f, 45'000};
 
    bool enable_human_review = false;
 
    bool enable_clarification = false;
};
 
std::unique_ptr<GraphEngine> create_deep_research_graph(
    std::shared_ptr<Provider>                provider,
    std::vector<std::unique_ptr<Tool>>       tools,
    DeepResearchConfig                       cfg = {});
 
} // namespace neograph::graph