context_manager.h

Go to the documentation of this file.

// SPDX-License-Identifier: Apache-2.0
#pragma once
 
#include <entropic/core/compaction.h>
#include <entropic/core/engine_types.h>
#include <entropic/interfaces/i_hook_handler.h>
 
#include <functional>
 
namespace entropic {
 
struct ContextManagerHooks {
    std::function<void(LoopContext&)> after_compaction; 
};
 
class ContextManager {
public:
    ContextManager(CompactionManager& compaction,
                   EngineCallbacks& callbacks,
                   ContextManagerHooks hooks = {});
 
    void refresh_context_limit(LoopContext& ctx, int context_length);
 
    std::pair<int, int> prune_tool_results(
        LoopContext& ctx,
        int keep_recent);
 
    void prune_old_tool_results(LoopContext& ctx);
 
    void inject_context_warning(LoopContext& ctx);
 
    void check_compaction(LoopContext& ctx, bool force = false);
 
    bool fire_pre_compact_hook(LoopContext& ctx, bool force);
 
    void fire_post_compact_hooks(LoopContext& ctx, int old_count,
                                 int new_count);
 
    void set_hooks(const HookInterface& hooks) { hook_iface_ = hooks; }
 
private:
    CompactionManager& compaction_; 
    EngineCallbacks& callbacks_;    
    ContextManagerHooks hooks_;     
    HookInterface hook_iface_;      
};
 
} // namespace entropic