step_plan_repair.h

Go to the documentation of this file.

 
#pragma once
 
#include <neograph/api.h>
#include <neograph/program/registry.h>
#include <neograph/program/step_plan.h>
 
#include <cstddef>
#include <cstdint>
#include <functional>
#include <string>
#include <string_view>
#include <vector>
 
namespace neograph::program {
 
inline constexpr std::string_view STEP_PLAN_INTAKE_NODE      = "neograph.step_plan_intake/v1";
inline constexpr std::string_view STEP_PLAN_ROUTE_CONDITION  = "neograph.step_plan_route/v1";
inline constexpr std::string_view STEP_PLAN_REPORT_NODE      = "neograph.step_plan_report/v1";
 
enum class StepPlanFeedback : std::uint8_t {
    Replace,
    Append,
};
 
NEOGRAPH_PROGRAM_API std::string_view to_string(StepPlanFeedback feedback) noexcept;
 
struct StepPlanRepairNodeOptions {
    StepPlanIntakeOptions intake;
    std::string attestation_id = "host:step-plan-repair";
    std::uint32_t max_attempts_ceiling = 5;
    std::size_t max_feedback_bytes = 2048;
};
 
NEOGRAPH_PROGRAM_API void add_step_plan_repair(RegistrySnapshotBuilder&          builder,
                                               const StepPlanRepairNodeOptions& options);
 
struct PlannerMessage {
    std::string role;     
    std::string content;
 
    bool operator==(const PlannerMessage&) const = default;
};
 
struct PlannerReply {
    std::string   text;
    std::uint64_t prompt_tokens     = 0;
    std::uint64_t completion_tokens = 0;
};
 
struct PlanRepairOptions {
    StepPlanIntakeOptions intake;
    std::uint32_t         max_attempts       = 3;
    StepPlanFeedback      feedback           = StepPlanFeedback::Replace;
    std::size_t           max_feedback_bytes = 2048;
};
 
struct StepPlanRepairResult {
    bool                        ok = false;
    json                        proposal = nullptr;
    std::optional<std::string>  proposal_hash;
    json                        repair = json::object();
    std::uint32_t               attempts = 0;
    std::uint64_t               prompt_tokens     = 0;
    std::uint64_t               completion_tokens = 0;
    std::vector<PlannerMessage> messages;
};
 
NEOGRAPH_PROGRAM_API StepPlanRepairResult
repair_step_plan(const PlanRepairOptions&                                           options,
                 std::string_view                                                   request,
                 const std::function<PlannerReply(const std::vector<PlannerMessage>&)>& planner);
 
}  // namespace neograph::program