ChatTool.hpp

 
 
#pragma once
 
#include "AstGlobal.h"
#include "AstUtil/StringView.hpp"
#include "AstUtil/JsonValue.hpp"
#include <string>
 
AST_NAMESPACE_BEGIN
 
class JsonValue;
 
class ChatTool
{
public:
    ChatTool() = default;
    
    virtual ~ChatTool() = default;
    virtual std::string call(const JsonValue& arguments) = 0;
 
    const std::string& name() const { return name_; }
 
    void setName(StringView name){name_ = std::string(name);}
    
    const std::string& description() const { return description_; }
 
    void setDescription(StringView description){description_ = std::string(description);}
 
    const JsonValue& parameters() const { return parameters_; }
 
    void setParameters(const JsonValue& parameters){parameters_ = parameters;}
 
    JsonValue toJson() const;
private:
    std::string name_;              
    std::string description_;       
    JsonValue parameters_;          
};
 
template<typename Func>
class ChatToolGeneric: public ChatTool
{
public:
    ChatToolGeneric(Func func) : func_(std::move(func)) {}
    std::string call(const JsonValue& arguments) override { return func_(arguments); }
private:
    Func func_;
};
 
 
AST_NAMESPACE_END