plugin.hpp

#pragma once
#include "illixr/data_format/string_data.hpp" 
#include "illixr/phonebook.hpp"
#include "illixr/plugin.hpp"
#include "illixr/switchboard.hpp"
#include "illixr/record_logger.hpp"
#include <string>


namespace ILLIXR {

class llama_cli : public plugin {
public:
    [[maybe_unused]] llama_cli(const std::string& name, phonebook* pb);
    void start() override;
    ~llama_cli() override = default;
private:
    const std::shared_ptr<switchboard>                   switchboard_;
    switchboard::reader<data_format::string_data> text_reader_;
    switchboard::buffered_reader<data_format::string_data> reader_;
    switchboard::writer<data_format::string_data> response_publisher_;
    std::string prompt_scene_desc;
    std::shared_ptr<ILLIXR::record_logger> record_logger_;

    // switchboard::reader<data_format::string_data> response_reader_;
};
} // namespace ILLIXR