ChatMessages.hpp

浏览该文件的文档.

 
 
#pragma once
 
#include "AstGlobal.h"
#include "ChatMessage.hpp"
#include <vector>
 
AST_NAMESPACE_BEGIN
 
using MessageVector = std::vector<ChatMessage>;
 
 
class ChatMessages: public MessageVector
{
public:
 
    ChatMessages() = default;
    ~ChatMessages() = default;
 
    AST_AI_API
    JsonValue toJson() const;
    
    void addMessage(const ChatMessage& message)
    {
        this->push_back(message);
    }
 
    void addSystemMessage(StringView content)
    {
        addMessage(ChatMessage::System(content));
    }
 
    void addUserMessage(StringView content)
    {
        addMessage(ChatMessage::User(content));
    }
 
    void addAssistantMessage(StringView content)
    {
        addMessage(ChatMessage::Assistant(content));
    }
 
    void addAssistantMessage(StringView content, const JsonValue& toolCalls)
    {
        addMessage(ChatMessage::Assistant(content, toolCalls));
    }
 
    void addToolMessage(StringView content, StringView toolCallId)
    {
        addMessage(ChatMessage::Tool(content, toolCallId));
    }
 
 
    AST_AI_API
    void setSystemPrompt(StringView systemPrompt);
};
 
AST_NAMESPACE_END