forked from xiaozhi/xiaozhi-esp32
83f6f8c703
* Adapt boards to v2 partition tables * fix esp log error * fix display style * reset emotion after download assets * fix compiling * update assets default url * Add user only tools * Add image cache * smaller cache and buffer, more heap * use MAIN_EVENT_CLOCK_TICK to avoid audio glitches * bump to 2.0.0 * fix compiling errors --------- Co-authored-by: Xiaoxia <terrence.huang@tenclass.com>
346 lines
12 KiB
C++
346 lines
12 KiB
C++
#ifndef MCP_SERVER_H
|
|
#define MCP_SERVER_H
|
|
|
|
#include <string>
|
|
#include <vector>
|
|
#include <map>
|
|
#include <functional>
|
|
#include <variant>
|
|
#include <optional>
|
|
#include <stdexcept>
|
|
#include <thread>
|
|
#include <mbedtls/base64.h>
|
|
|
|
#include <cJSON.h>
|
|
|
|
class ImageContent {
|
|
private:
|
|
std::string encoded_data_;
|
|
std::string mime_type_;
|
|
|
|
static std::string Base64Encode(const std::string& data) {
|
|
size_t dlen = 0, olen = 0;
|
|
mbedtls_base64_encode((unsigned char*)nullptr, 0, &dlen, (const unsigned char*)data.data(), data.size());
|
|
std::string result(dlen, 0);
|
|
mbedtls_base64_encode((unsigned char*)result.data(), result.size(), &olen, (const unsigned char*)data.data(), data.size());
|
|
return result;
|
|
}
|
|
|
|
public:
|
|
ImageContent(const std::string& mime_type, const std::string& data) {
|
|
mime_type_ = mime_type;
|
|
// base64 encode data
|
|
encoded_data_ = Base64Encode(data);
|
|
}
|
|
|
|
std::string to_json() const {
|
|
cJSON *json = cJSON_CreateObject();
|
|
cJSON_AddStringToObject(json, "type", "image");
|
|
cJSON_AddStringToObject(json, "mimeType", mime_type_.c_str());
|
|
cJSON_AddStringToObject(json, "data", encoded_data_.c_str());
|
|
char* json_str = cJSON_PrintUnformatted(json);
|
|
std::string result(json_str);
|
|
cJSON_free(json_str);
|
|
cJSON_Delete(json);
|
|
return result;
|
|
}
|
|
};
|
|
|
|
// 添加类型别名
|
|
using ReturnValue = std::variant<bool, int, std::string, cJSON*, ImageContent*>;
|
|
|
|
enum PropertyType {
|
|
kPropertyTypeBoolean,
|
|
kPropertyTypeInteger,
|
|
kPropertyTypeString
|
|
};
|
|
|
|
class Property {
|
|
private:
|
|
std::string name_;
|
|
PropertyType type_;
|
|
std::variant<bool, int, std::string> value_;
|
|
bool has_default_value_;
|
|
std::optional<int> min_value_; // 新增：整数最小值
|
|
std::optional<int> max_value_; // 新增：整数最大值
|
|
|
|
public:
|
|
// Required field constructor
|
|
Property(const std::string& name, PropertyType type)
|
|
: name_(name), type_(type), has_default_value_(false) {}
|
|
|
|
// Optional field constructor with default value
|
|
template<typename T>
|
|
Property(const std::string& name, PropertyType type, const T& default_value)
|
|
: name_(name), type_(type), has_default_value_(true) {
|
|
value_ = default_value;
|
|
}
|
|
|
|
Property(const std::string& name, PropertyType type, int min_value, int max_value)
|
|
: name_(name), type_(type), has_default_value_(false), min_value_(min_value), max_value_(max_value) {
|
|
if (type != kPropertyTypeInteger) {
|
|
throw std::invalid_argument("Range limits only apply to integer properties");
|
|
}
|
|
}
|
|
|
|
Property(const std::string& name, PropertyType type, int default_value, int min_value, int max_value)
|
|
: name_(name), type_(type), has_default_value_(true), min_value_(min_value), max_value_(max_value) {
|
|
if (type != kPropertyTypeInteger) {
|
|
throw std::invalid_argument("Range limits only apply to integer properties");
|
|
}
|
|
if (default_value < min_value || default_value > max_value) {
|
|
throw std::invalid_argument("Default value must be within the specified range");
|
|
}
|
|
value_ = default_value;
|
|
}
|
|
|
|
inline const std::string& name() const { return name_; }
|
|
inline PropertyType type() const { return type_; }
|
|
inline bool has_default_value() const { return has_default_value_; }
|
|
inline bool has_range() const { return min_value_.has_value() && max_value_.has_value(); }
|
|
inline int min_value() const { return min_value_.value_or(0); }
|
|
inline int max_value() const { return max_value_.value_or(0); }
|
|
|
|
template<typename T>
|
|
inline T value() const {
|
|
return std::get<T>(value_);
|
|
}
|
|
|
|
template<typename T>
|
|
inline void set_value(const T& value) {
|
|
// 添加对设置的整数值进行范围检查
|
|
if constexpr (std::is_same_v<T, int>) {
|
|
if (min_value_.has_value() && value < min_value_.value()) {
|
|
throw std::invalid_argument("Value is below minimum allowed: " + std::to_string(min_value_.value()));
|
|
}
|
|
if (max_value_.has_value() && value > max_value_.value()) {
|
|
throw std::invalid_argument("Value exceeds maximum allowed: " + std::to_string(max_value_.value()));
|
|
}
|
|
}
|
|
value_ = value;
|
|
}
|
|
|
|
std::string to_json() const {
|
|
cJSON *json = cJSON_CreateObject();
|
|
|
|
if (type_ == kPropertyTypeBoolean) {
|
|
cJSON_AddStringToObject(json, "type", "boolean");
|
|
if (has_default_value_) {
|
|
cJSON_AddBoolToObject(json, "default", value<bool>());
|
|
}
|
|
} else if (type_ == kPropertyTypeInteger) {
|
|
cJSON_AddStringToObject(json, "type", "integer");
|
|
if (has_default_value_) {
|
|
cJSON_AddNumberToObject(json, "default", value<int>());
|
|
}
|
|
if (min_value_.has_value()) {
|
|
cJSON_AddNumberToObject(json, "minimum", min_value_.value());
|
|
}
|
|
if (max_value_.has_value()) {
|
|
cJSON_AddNumberToObject(json, "maximum", max_value_.value());
|
|
}
|
|
} else if (type_ == kPropertyTypeString) {
|
|
cJSON_AddStringToObject(json, "type", "string");
|
|
if (has_default_value_) {
|
|
cJSON_AddStringToObject(json, "default", value<std::string>().c_str());
|
|
}
|
|
}
|
|
|
|
char *json_str = cJSON_PrintUnformatted(json);
|
|
std::string result(json_str);
|
|
cJSON_free(json_str);
|
|
cJSON_Delete(json);
|
|
|
|
return result;
|
|
}
|
|
};
|
|
|
|
class PropertyList {
|
|
private:
|
|
std::vector<Property> properties_;
|
|
|
|
public:
|
|
PropertyList() = default;
|
|
PropertyList(const std::vector<Property>& properties) : properties_(properties) {}
|
|
void AddProperty(const Property& property) {
|
|
properties_.push_back(property);
|
|
}
|
|
|
|
const Property& operator[](const std::string& name) const {
|
|
for (const auto& property : properties_) {
|
|
if (property.name() == name) {
|
|
return property;
|
|
}
|
|
}
|
|
throw std::runtime_error("Property not found: " + name);
|
|
}
|
|
|
|
auto begin() { return properties_.begin(); }
|
|
auto end() { return properties_.end(); }
|
|
|
|
std::vector<std::string> GetRequired() const {
|
|
std::vector<std::string> required;
|
|
for (auto& property : properties_) {
|
|
if (!property.has_default_value()) {
|
|
required.push_back(property.name());
|
|
}
|
|
}
|
|
return required;
|
|
}
|
|
|
|
std::string to_json() const {
|
|
cJSON *json = cJSON_CreateObject();
|
|
|
|
for (const auto& property : properties_) {
|
|
cJSON *prop_json = cJSON_Parse(property.to_json().c_str());
|
|
cJSON_AddItemToObject(json, property.name().c_str(), prop_json);
|
|
}
|
|
|
|
char *json_str = cJSON_PrintUnformatted(json);
|
|
std::string result(json_str);
|
|
cJSON_free(json_str);
|
|
cJSON_Delete(json);
|
|
|
|
return result;
|
|
}
|
|
};
|
|
|
|
class McpTool {
|
|
private:
|
|
std::string name_;
|
|
std::string description_;
|
|
PropertyList properties_;
|
|
std::function<ReturnValue(const PropertyList&)> callback_;
|
|
bool user_only_ = false;
|
|
|
|
public:
|
|
McpTool(const std::string& name,
|
|
const std::string& description,
|
|
const PropertyList& properties,
|
|
std::function<ReturnValue(const PropertyList&)> callback)
|
|
: name_(name),
|
|
description_(description),
|
|
properties_(properties),
|
|
callback_(callback) {}
|
|
|
|
void set_user_only(bool user_only) { user_only_ = user_only; }
|
|
inline const std::string& name() const { return name_; }
|
|
inline const std::string& description() const { return description_; }
|
|
inline const PropertyList& properties() const { return properties_; }
|
|
inline bool user_only() const { return user_only_; }
|
|
|
|
std::string to_json() const {
|
|
std::vector<std::string> required = properties_.GetRequired();
|
|
|
|
cJSON *json = cJSON_CreateObject();
|
|
cJSON_AddStringToObject(json, "name", name_.c_str());
|
|
cJSON_AddStringToObject(json, "description", description_.c_str());
|
|
|
|
cJSON *input_schema = cJSON_CreateObject();
|
|
cJSON_AddStringToObject(input_schema, "type", "object");
|
|
|
|
cJSON *properties = cJSON_Parse(properties_.to_json().c_str());
|
|
cJSON_AddItemToObject(input_schema, "properties", properties);
|
|
|
|
if (!required.empty()) {
|
|
cJSON *required_array = cJSON_CreateArray();
|
|
for (const auto& property : required) {
|
|
cJSON_AddItemToArray(required_array, cJSON_CreateString(property.c_str()));
|
|
}
|
|
cJSON_AddItemToObject(input_schema, "required", required_array);
|
|
}
|
|
|
|
cJSON_AddItemToObject(json, "inputSchema", input_schema);
|
|
|
|
// Add audience annotation if the tool is user only (invisible to AI)
|
|
if (user_only_) {
|
|
cJSON *annotations = cJSON_CreateObject();
|
|
cJSON *audience = cJSON_CreateArray();
|
|
cJSON_AddItemToArray(audience, cJSON_CreateString("user"));
|
|
cJSON_AddItemToObject(annotations, "audience", audience);
|
|
cJSON_AddItemToObject(json, "annotations", annotations);
|
|
}
|
|
|
|
char *json_str = cJSON_PrintUnformatted(json);
|
|
std::string result(json_str);
|
|
cJSON_free(json_str);
|
|
cJSON_Delete(json);
|
|
|
|
return result;
|
|
}
|
|
|
|
std::string Call(const PropertyList& properties) {
|
|
ReturnValue return_value = callback_(properties);
|
|
// 返回结果
|
|
cJSON* result = cJSON_CreateObject();
|
|
cJSON* content = cJSON_CreateArray();
|
|
|
|
if (std::holds_alternative<ImageContent*>(return_value)) {
|
|
auto image_content = std::get<ImageContent*>(return_value);
|
|
cJSON* image = cJSON_CreateObject();
|
|
cJSON_AddStringToObject(image, "type", "image");
|
|
cJSON_AddStringToObject(image, "image", image_content->to_json().c_str());
|
|
cJSON_AddItemToArray(content, image);
|
|
delete image_content;
|
|
} else {
|
|
cJSON* text = cJSON_CreateObject();
|
|
cJSON_AddStringToObject(text, "type", "text");
|
|
if (std::holds_alternative<std::string>(return_value)) {
|
|
cJSON_AddStringToObject(text, "text", std::get<std::string>(return_value).c_str());
|
|
} else if (std::holds_alternative<bool>(return_value)) {
|
|
cJSON_AddStringToObject(text, "text", std::get<bool>(return_value) ? "true" : "false");
|
|
} else if (std::holds_alternative<int>(return_value)) {
|
|
cJSON_AddStringToObject(text, "text", std::to_string(std::get<int>(return_value)).c_str());
|
|
} else if (std::holds_alternative<cJSON*>(return_value)) {
|
|
cJSON* json = std::get<cJSON*>(return_value);
|
|
char* json_str = cJSON_PrintUnformatted(json);
|
|
cJSON_AddStringToObject(text, "text", json_str);
|
|
cJSON_free(json_str);
|
|
cJSON_Delete(json);
|
|
}
|
|
cJSON_AddItemToArray(content, text);
|
|
}
|
|
cJSON_AddItemToObject(result, "content", content);
|
|
cJSON_AddBoolToObject(result, "isError", false);
|
|
|
|
auto json_str = cJSON_PrintUnformatted(result);
|
|
std::string result_str(json_str);
|
|
cJSON_free(json_str);
|
|
cJSON_Delete(result);
|
|
return result_str;
|
|
}
|
|
};
|
|
|
|
class McpServer {
|
|
public:
|
|
static McpServer& GetInstance() {
|
|
static McpServer instance;
|
|
return instance;
|
|
}
|
|
|
|
void AddCommonTools();
|
|
void AddUserOnlyTools();
|
|
void AddTool(McpTool* tool);
|
|
void AddTool(const std::string& name, const std::string& description, const PropertyList& properties, std::function<ReturnValue(const PropertyList&)> callback);
|
|
void AddUserOnlyTool(const std::string& name, const std::string& description, const PropertyList& properties, std::function<ReturnValue(const PropertyList&)> callback);
|
|
void ParseMessage(const cJSON* json);
|
|
void ParseMessage(const std::string& message);
|
|
|
|
private:
|
|
McpServer();
|
|
~McpServer();
|
|
|
|
void ParseCapabilities(const cJSON* capabilities);
|
|
|
|
void ReplyResult(int id, const std::string& result);
|
|
void ReplyError(int id, const std::string& message);
|
|
|
|
void GetToolsList(int id, const std::string& cursor, bool list_user_only_tools);
|
|
void DoToolCall(int id, const std::string& tool_name, const cJSON* tool_arguments, int stack_size);
|
|
|
|
std::vector<McpTool*> tools_;
|
|
std::thread tool_call_thread_;
|
|
};
|
|
|
|
#endif // MCP_SERVER_H
|