session_logger.cpp

Go to the documentation of this file.

// SPDX-License-Identifier: Apache-2.0
#include <entropic/types/session_logger.h>
#include <entropic/types/logging.h>
 
static auto s_log = entropic::log::get("session");
 
namespace entropic {
 
SessionLogger::SessionLogger(const std::filesystem::path& log_dir) {
    auto path = log_dir / "session_model.log";
    fp_ = fopen(path.string().c_str(), "a");
    if (!fp_) {
        s_log->warn("cannot open model log: {}", path.string());
    }
}
 
SessionLogger::~SessionLogger() {
    if (fp_) { fclose(fp_); }
}
 
void SessionLogger::log_user_input(const std::string& input) {
    if (!fp_) { return; }
    fprintf(fp_, "--- USER ---\n%s\n--- ASSISTANT ---\n", input.c_str());
    fflush(fp_);
}
 
void SessionLogger::log_raw_token(const char* token, size_t len) {
    if (!fp_ || !token || len == 0) { return; }
    fwrite(token, 1, len, fp_);
    fflush(fp_);
}
 
void SessionLogger::end_turn() {
    if (!fp_) { return; }
    fprintf(fp_, "\n\n");
    fflush(fp_);
}
 
bool SessionLogger::is_open() const {
    return fp_ != nullptr;
}
 
void SessionLogger::raw_token_callback(
    const char* token, size_t len, void* user_data) {
    static_cast<SessionLogger*>(user_data)->log_raw_token(token, len);
}
 
} // namespace entropic