blob: d0f639adad24e2f6f641d6bdaabfddfd8658082b (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
|
//===- Tool.h -------------------------------------------------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
#ifndef LLDB_PLUGINS_PROTOCOL_MCP_TOOL_H
#define LLDB_PLUGINS_PROTOCOL_MCP_TOOL_H
#include "Protocol.h"
#include "lldb/Core/Debugger.h"
#include "llvm/Support/JSON.h"
#include <string>
namespace lldb_private::mcp {
class Tool {
public:
Tool(std::string name, std::string description);
virtual ~Tool() = default;
virtual llvm::Expected<protocol::TextResult>
Call(const protocol::ToolArguments &args) = 0;
virtual std::optional<llvm::json::Value> GetSchema() const {
return llvm::json::Object{{"type", "object"}};
}
protocol::ToolDefinition GetDefinition() const;
const std::string &GetName() { return m_name; }
private:
std::string m_name;
std::string m_description;
};
class CommandTool : public mcp::Tool {
public:
using mcp::Tool::Tool;
~CommandTool() = default;
virtual llvm::Expected<protocol::TextResult>
Call(const protocol::ToolArguments &args) override;
virtual std::optional<llvm::json::Value> GetSchema() const override;
};
} // namespace lldb_private::mcp
#endif
|
