conversation_state.h

Go to the documentation of this file.

// SPDX-License-Identifier: Apache-2.0
#pragma once
 
#include <entropic/types/message.h>
#include <string>
#include <vector>
 
struct ConversationState {
    std::string system_prompt;                    
    std::vector<entropic::Message> messages;      
 
    std::vector<entropic::Message> append_user(const char* input) {
        if (messages.empty()) {
            entropic::Message sys;
            sys.role = "system";
            sys.content = system_prompt;
            messages.push_back(std::move(sys));
        }
        entropic::Message usr;
        usr.role = "user";
        usr.content = input;
        messages.push_back(std::move(usr));
        return messages;  // copy
    }
 
    void append_result(const std::vector<entropic::Message>& result,
                       size_t sent_len) {
        for (size_t i = sent_len; i < result.size(); i++) {
            messages.push_back(result[i]);
        }
    }
 
    void clear() { messages.clear(); }
 
    size_t count() const { return messages.size(); }
};