orchestrator.h

Go to the documentation of this file.

// SPDX-License-Identifier: Apache-2.0
#pragma once
 
#include <entropic/inference/backend.h>
#include <entropic/inference/adapter_manager.h>
#include <entropic/inference/grammar_registry.h>
#include <entropic/inference/profile_registry.h>
#include <entropic/inference/secondary_model_loader.h>
#include <entropic/inference/throughput_tracker.h>
#include <entropic/inference/adapters/adapter_base.h>
#include <entropic/types/config.h>
#include <entropic/types/error.h>
 
#include <chrono>
#include <functional>
#include <memory>
#include <mutex>
#include <string>
#include <unordered_map>
#include <unordered_set>
#include <vector>
 
struct llama_context;  // Forward declaration for adapter management
struct llama_model;    // Forward declaration for speculative compat (v2.1.11)
 
namespace entropic {
 
struct RoutingResult {
    std::string tier_name;                
    std::string previous_tier;            
    std::string model_raw;                
    std::string swap_action = "none";     
    double routing_ms = 0.0;              
    std::string adapter_name;             
    double adapter_swap_ms = 0.0;         
};
 
class ModelOrchestrator {
public:
    bool initialize(const ParsedConfig& config);
 
    void shutdown();
 
    ~ModelOrchestrator();
 
    /* ── Generation ──────────────────────────────────────── */
 
    GenerationResult generate(
        const std::vector<Message>& messages,
        const GenerationParams& params,
        const std::string& tier_name = "");
 
    GenerationResult generate_streaming(
        const std::vector<Message>& messages,
        const GenerationParams& params,
        std::function<void(std::string_view)> on_token,
        std::atomic<bool>& cancel,
        const std::string& tier_name = "");
 
    /* ── Routing ─────────────────────────────────────────── */
 
    std::string route(const std::vector<Message>& messages);
 
    /* ── Queries ─────────────────────────────────────────── */
 
    RoutingResult last_routing_result() const;
 
    std::string last_used_tier() const;
 
    std::vector<std::string> loaded_models() const;
 
    std::vector<std::string> available_models() const;
 
    bool can_handoff(const std::string& from, const std::string& to) const;
 
    ChatAdapter* get_adapter(const std::string& tier_name) const;
 
    InferenceBackend* get_backend(const std::string& tier_name) const;
 
    AdapterManager& adapter_manager() { return lora_manager_; }
 
    GrammarRegistry& grammar_registry() { return grammar_registry_; }
 
    ProfileRegistry& profile_registry() { return profile_registry_; }
 
    ThroughputTracker& throughput_tracker() { return throughput_tracker_; }
 
    size_t load_grammars_from(const std::filesystem::path& grammar_dir);
 
    void clear_all_prompt_caches();
 
    bool has_vision_capable_tier() const;
 
    struct SpeculativeCompatInfo {
        bool compatible = false;     
        std::string diagnostic;      
    };
 
    SpeculativeCompatInfo check_speculative_compat() const;
 
    void set_speculative_enabled(bool enabled) {
        config_.inference.speculative.enabled = enabled;
    }
 
    /* ── VRAM-aware tier residency (v2.2.4, gh#57) ────────── */
 
    enum class ResidencyEvent : int {
        Loaded = 0,
        Evicted = 1,
        ActivationSwap = 2,
    };
 
    using ResidencyObserverFn = std::function<void(
        ResidencyEvent event,
        const std::string& tier_name,
        const std::string& model_path,
        size_t footprint)>;
 
    void set_residency_observer(ResidencyObserverFn cb);
 
    std::string residency_snapshot_json() const;
 
    size_t vram_budget_bytes() const { return vram_budget_bytes_; }
 
    size_t tier_footprint_bytes(const std::string& tier_name) const;
 
    entropic_error_t last_residency_error() const { return last_residency_error_; }
 
    void clear_last_residency_error() { last_residency_error_ = ENTROPIC_OK; }
 
    std::string select_vision_tier() const;
 
private:
    /* ── Model pool (one backend per unique path) ────────── */
    std::unordered_map<std::string, std::shared_ptr<InferenceBackend>> model_pool_;
 
    /* ── Tier → backend mapping (many-to-one) ────────────── */
    std::unordered_map<std::string, std::shared_ptr<InferenceBackend>> tiers_;
 
    /* ── Per-tier adapters (one-to-one, identity-specific) ── */
    std::unordered_map<std::string, std::unique_ptr<ChatAdapter>> adapters_;
 
    /* ── Secondary models (router, draft, future thinking) ── */
    SecondaryModelLoader secondary_loader_;
 
    /* ── Routing state ───────────────────────────────────── */
    std::unordered_map<std::string, std::string> tier_map_;  
    std::unordered_map<std::string, std::unordered_set<std::string>> handoff_rules_;
    std::string default_tier_;
    std::string loaded_main_tier_;
    RoutingResult last_routing_result_;
    std::vector<std::string> tier_history_;  
 
    mutable std::mutex swap_mutex_;  
 
    ParsedConfig config_;
 
    /* ── Residency tracking (v2.2.4, gh#57) ──────────────── */
 
    mutable std::unordered_map<std::string, size_t> tier_footprint_bytes_;
 
    std::unordered_map<std::string, long long> tier_last_activation_ms_;
 
    std::chrono::steady_clock::time_point start_time_{std::chrono::steady_clock::now()};
 
    size_t vram_budget_bytes_{0};
 
    ResidencyObserverFn residency_observer_;
 
    entropic_error_t last_residency_error_{ENTROPIC_OK};
 
    size_t estimate_footprint_bytes(const std::string& tier_name) const;
 
    static size_t resolve_vram_budget_bytes();
 
    void fire_residency_observer(
        ResidencyEvent event,
        const std::string& tier_name,
        const std::string& model_path,
        size_t footprint);
 
    /* ── LoRA adapter management (v1.9.2) ────────────────── */
    AdapterManager lora_manager_;  
 
    /* ── Grammar registry (v1.9.3) ────────────────────────── */
    GrammarRegistry grammar_registry_;  
 
    /* ── Profile registry (v2.0.0) ───────────────────────── */
    ProfileRegistry profile_registry_;  
 
    /* ── Throughput tracker (v2.0.0) ─────────────────────── */
    ThroughputTracker throughput_tracker_;  
 
    /* ── Internal ────────────────────────────────────────── */
 
    InferenceBackend* get_model(const std::string& tier_name);
 
    void record_activation_reuse(const std::string& tier_name);
 
    bool residency_admits(const std::string& tier_name);
 
    InferenceBackend* activate_and_track(
        const std::string& tier_name,
        const std::shared_ptr<InferenceBackend>& backend);
 
    GenerationResult build_no_model_error(const std::string& tier_name);
 
    void deactivate_current_if_needed(InferenceBackend* incoming);
 
    void ensure_tier_lora(const std::string& tier_name,
                          InferenceBackend* result);
 
    void unload_or_warm_current(InferenceBackend* current);
 
    std::pair<std::string, std::string> classify_task(
        const std::vector<Message>& messages);
 
    bool deactivate_if_active(llama_context* ctx);
 
    double ensure_adapter_for_tier(
        const std::string& tier_name, llama_context* ctx);
 
    void preload_adapters();
 
    bool create_tier_backends(const ParsedConfig& config);
 
    void build_routing_tables(const ParsedConfig& config);
 
    bool activate_default_tier(const ParsedConfig& config);
 
    void activate_router(const ParsedConfig& config);
 
    void activate_draft(const ParsedConfig& config);
 
    void load_bundled_grammars();
 
    void resolve_grammar_key(GenerationParams& params,
                             const std::string& tier_name);
 
    std::string resolve_speculative_pair(
        llama_model*& target_out, llama_model*& draft_out) const;
 
    bool try_speculative_route(
        InferenceBackend* model,
        const std::vector<Message>& messages,
        const GenerationParams& params,
        GenerationResult& result);
 
    GenerationResult run_generate_dispatch(
        InferenceBackend* model,
        const std::vector<Message>& messages,
        const GenerationParams& params);
 
    bool try_speculative_route_streaming(
        InferenceBackend* model,
        const std::vector<Message>& messages,
        const GenerationParams& params,
        std::function<void(std::string_view)> on_token,
        std::atomic<bool>& cancel,
        GenerationResult& result);
};
 
} // namespace entropic