client.h

Go to the documentation of this file.

 
#pragma once
 
#include <neograph/api.h>
#include <neograph/tool.h>
 
#include <asio/awaitable.hpp>
 
#include <memory>
#include <mutex>
#include <string>
#include <vector>
 
namespace neograph::mcp {
 
namespace detail {
class StdioSession;
}
 
class NEOGRAPH_API MCPTool : public Tool {
  public:
    MCPTool(const std::string& server_url,
            const std::string& name,
            const std::string& description,
            const json& input_schema);
 
    MCPTool(std::shared_ptr<detail::StdioSession> session,
            const std::string& name,
            const std::string& description,
            const json& input_schema);
 
    ChatTool get_definition() const override;
 
    std::string execute(const json& arguments) override;
 
    std::string get_name() const override { return name_; }
 
  private:
    std::string server_url_;                              
    std::shared_ptr<detail::StdioSession> stdio_session_; 
    std::string name_;
    std::string description_;
    json input_schema_;
};
 
class NEOGRAPH_API MCPClient {
  public:
    explicit MCPClient(const std::string& server_url);
 
    explicit MCPClient(std::vector<std::string> argv);
 
    bool initialize(const std::string& client_name = "neograph");
 
    std::vector<std::unique_ptr<Tool>> get_tools();
 
    json call_tool(const std::string& name, const json& arguments);
 
    asio::awaitable<json> rpc_call_async(
        const std::string& method,
        const json& params = json::object());
 
  private:
    json rpc_call(const std::string& method, const json& params = json::object());
 
    // HTTP state (empty strings when in stdio mode).
    std::string server_url_;
    std::string host_;
    std::string path_prefix_;
    mutable std::mutex http_state_mu_;
    std::string session_id_;
    std::string negotiated_protocol_version_;
    int request_id_ = 0;
 
    // stdio state (null when in HTTP mode).
    std::shared_ptr<detail::StdioSession> stdio_session_;
};
 
} // namespace neograph::mcp