UserProxyAgent.hpp

浏览该文件的文档.

 
 
#pragma once
 
#include "AstGlobal.h"
#include "ChatAgent.hpp"
#include "AstUtil/StringView.hpp"
#include <string>
#include <vector>
 
AST_NAMESPACE_BEGIN
 
class AST_AI_API UserProxyAgent : public ChatAgent
{
public:
    UserProxyAgent() = default;
    ~UserProxyAgent() = default;
 
    errc_t run(ChatMessages& messages) override;
 
    errc_t runStream(ChatMessages& messages, ChatEventHandler& handler) override;
 
    void setPrompt(StringView prompt) { prompt_ = std::string(prompt); }
 
    const std::string& prompt() const { return prompt_; }
 
    // —— 退出关键词 ——
 
    void setExitKeywords(const std::vector<std::string>& keywords) { exitKeywords_ = keywords; }
 
    void addExitKeyword(StringView keyword) { exitKeywords_.emplace_back(keyword); }
 
    const std::vector<std::string>& exitKeywords() const { return exitKeywords_; }
 
private:
    std::string prompt_;                        
    std::vector<std::string> exitKeywords_;     
};
 
AST_NAMESPACE_END