RoundRobinGroupChat.hpp

浏览该文件的文档.

 
 
#pragma once
 
#include "AstGlobal.h"
#include "BaseGroupChat.hpp"
#include "AstAI/ChatAgent.hpp"
#include "AstAI/ChatMessages.hpp"
#include "AstAI/TerminationCondition.hpp"
#include <memory>
 
AST_NAMESPACE_BEGIN
 
class TerminationCondition;
 
class AST_AI_API RoundRobinGroupChat : public BaseGroupChat
{
public:
    RoundRobinGroupChat() = default;
    ~RoundRobinGroupChat() = default;
 
    // —— Agent管理 ——
 
    void addAgent(ChatAgent* agent);
 
    const std::vector<ChatAgent*>& agents() const { return agents_; }
 
    // —— 终止条件 ——
 
    void setTerminationCondition(std::unique_ptr<TerminationCondition> condition);
 
    TerminationCondition* terminationCondition() const { return terminationCondition_.get(); }
 
    // —— 消息历史 ——
 
    ChatMessages& messages() { return messages_; }
 
    const ChatMessages& messages() const { return messages_; }
 
    // —— 配置 ——
 
    void setMaxRounds(int maxRounds) { maxRounds_ = maxRounds; }
 
    int maxRounds() const { return maxRounds_; }
 
    // —— 执行 ——
 
    errc_t run(const ChatMessage& message, ChatMessage& response) override;
 
private:
    std::vector<ChatAgent*> agents_;
    ChatMessages messages_;
    std::unique_ptr<TerminationCondition> terminationCondition_;
    int maxRounds_ = 100;
};
 
AST_NAMESPACE_END