ChatTools.hpp

浏览该文件的文档.

 
 
#pragma once
 
#include "AstGlobal.h"
#include "ChatTool.hpp"
#include <vector>
#include <memory>
 
AST_NAMESPACE_BEGIN
 
class ChatTools
{
public:
    ChatTools() = default;
    ChatTools(bool initAsDefault);
    ~ChatTools() = default;
    ChatTools(ChatTools&&) = default;
    ChatTools& operator=(ChatTools&&) = default;
public:
    std::string handleToolCall(const JsonValue& toolCall);
 
    void addTool(std::unique_ptr<ChatTool>&& tool){tools_.push_back(std::move(tool));}
 
    template<typename Func>
    ChatTool* addTool(Func func)
    {
        ChatTool* tool = new ChatToolGeneric<Func>(func);
        addTool(std::unique_ptr<ChatTool>(tool));
        return tool;
    }
 
    void clearTools(){tools_.clear();}
 
    JsonValue toJson() const;
 
    bool empty() const {return tools_.empty();}
private:
    A_DISABLE_COPY(ChatTools);
    std::vector<std::unique_ptr<ChatTool>> tools_;
};
 
 
AST_NAMESPACE_END