54 lines
1.2 KiB
C
54 lines
1.2 KiB
C
|
#ifndef OPENMANUS_TOOL_BASE_H
|
|||
|
|
#define OPENMANUS_TOOL_BASE_H
|
|||
|
|
|
|||
|
|
#include <string>
|
|||
|
|
#include <vector>
|
|||
|
|
#include <functional>
|
|||
|
|
#include "mcp/include/mcp_message.h"
|
|||
|
|
#include "mcp/include/mcp_tool.h"
|
|||
|
|
|
|||
|
|
namespace openmanus {
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @class ToolBase
|
|||
|
|
* @brief 工具的基类，定义了所有工具的基本接口
|
|||
|
|
*/
|
|||
|
|
class ToolBase {
|
|||
|
|
public:
|
|||
|
|
ToolBase(const std::string& name, const std::string& description);
|
|||
|
|
virtual ~ToolBase() = default;
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 执行工具
|
|||
|
|
* @param params 工具参数
|
|||
|
|
* @return 执行结果
|
|||
|
|
*/
|
|||
|
|
virtual mcp::json execute(const mcp::json& params) = 0;
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 获取工具名称
|
|||
|
|
* @return 工具名称
|
|||
|
|
*/
|
|||
|
|
std::string getName() const { return name_; }
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 获取工具描述
|
|||
|
|
* @return 工具描述
|
|||
|
|
*/
|
|||
|
|
std::string getDescription() const { return description_; }
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 获取工具参数定义
|
|||
|
|
* @return 参数定义
|
|||
|
|
*/
|
|||
|
|
mcp::json getParameters() const { return parameters_; }
|
|||
|
|
|
|||
|
|
protected:
|
|||
|
|
std::string name_;
|
|||
|
|
std::string description_;
|
|||
|
|
mcp::json parameters_; // 参数定义，使用JSON Schema格式
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
} // namespace openmanus
|
|||
|
|
|
|||
|
|
#endif // OPENMANUS_TOOL_BASE_H
|