ChatMessage.hpp

浏览该文件的文档.

 
 
#pragma once
 
#include "AstGlobal.h"
#include "AstUtil/StringView.hpp"
#include "AstUtil/JsonValue.hpp"
#include <string>
 
AST_NAMESPACE_BEGIN
 
class JsonValue;
 
 
enum class EChatRole {
    eUser,      
    eAssistant, 
    eSystem,    
    eTool       
};
 
 
AST_AI_API std::string toString(EChatRole role);
 
 
class ChatMessage 
{
public:
    ChatMessage() = default;
 
    ChatMessage(EChatRole role, StringView content)
        : role_(role), content_(content) 
    {}
 
    ~ChatMessage() = default;
 
    static ChatMessage System(StringView content)
    {
        return ChatMessage(EChatRole::eSystem, content);
    }
 
    static ChatMessage User(StringView content)
    {
        return ChatMessage(EChatRole::eUser, content);
    }
 
    static ChatMessage Assistant(StringView content)
    {
        return ChatMessage(EChatRole::eAssistant, content);
    }
 
    static ChatMessage Assistant(StringView content, const JsonValue& toolCalls)
    {
        ChatMessage msg(EChatRole::eAssistant, content);
        msg.setToolCalls(toolCalls);
        return msg;
    }
    
    
 
    static ChatMessage Tool(StringView content, StringView toolCallId)
    {
        ChatMessage msg(EChatRole::eTool, content);
        msg.setToolCallId(toolCallId);
        return msg;
    }
 
    AST_AI_API
    JsonValue toJson() const;
 
    void setRole(EChatRole role) { role_ = role; }
    EChatRole role() const { return role_; }
 
    void setName(const std::string& name) { name_ = name; }
    std::string name() const { return name_; }
 
    void setContent(StringView content) { content_ = std::string(content); }
    std::string content() const { return content_; }
 
    void setReasoningContent(StringView reasoningContent) { reasoningContent_ = std::string(reasoningContent); }
    std::string reasoningContent() const { return reasoningContent_; }
    
    void setToolCallId(StringView toolCallId) { toolCallId_ = std::string(toolCallId); }
    
    std::string toolCallId() const { return toolCallId_; }
    
 
    void setToolCalls(const JsonValue& toolCalls) { toolCalls_ = toolCalls; }
 
    const JsonValue& toolCalls() const { return toolCalls_; }
 
private:
    EChatRole        role_{EChatRole::eUser};           
    std::string      name_;                             
    std::string      content_;                          
    std::string      reasoningContent_;                 
    std::string      toolCallId_;                       
    JsonValue        toolCalls_;                        
};
 
 
AST_NAMESPACE_END