prompt_log_util.h

Go to the documentation of this file.

// SPDX-License-Identifier: Apache-2.0
#pragma once
 
#include <cstdlib>
#include <string>
#include <unordered_set>
 
namespace entropic {
 
class PromptLogDedup {
public:
    bool note(size_t content_hash) {
        return seen_.insert(content_hash).second;
    }
 
    void reset() { seen_.clear(); }
 
    size_t shown() const { return seen_.size(); }
 
private:
    std::unordered_set<size_t> seen_;
};
 
inline bool prompt_log_unabridged() {
    const char* v = std::getenv("ENTROPIC_LOG_FULL_PROMPT");
    return v != nullptr && v[0] == '1';
}
 
inline std::string prompt_log_reference(size_t index,
                                        const std::string& role,
                                        size_t size_chars,
                                        size_t content_hash) {
    char buf[160];
    int n = std::snprintf(
        buf, sizeof(buf),
        "[%zu] role=%s [shown above, %zu chars, hash=%016zx]",
        index, role.c_str(), size_chars, content_hash);
    return std::string(buf, n > 0 ? static_cast<size_t>(n) : 0U);
}
 
}  // namespace entropic