client.h

Go to the documentation of this file.

 
#pragma once
 
#include <neograph/api.h>
#include <neograph/a2a/types.h>
 
#include <asio/awaitable.hpp>
 
#include <chrono>
#include <functional>
#include <string>
 
namespace neograph::a2a {
 
class NEOGRAPH_API A2AClient {
  public:
    explicit A2AClient(std::string base_url);
 
    void set_timeout(std::chrono::seconds t) { timeout_ = t; }
 
    AgentCard fetch_agent_card(bool force = false);
 
    asio::awaitable<AgentCard> fetch_agent_card_async(bool force = false);
 
    Task send_message_sync(
        const std::string& text,
        const std::string& task_id    = "",
        const std::string& context_id = "");
 
    Task send_message_sync(const MessageSendParams& params);
 
    asio::awaitable<Task> send_message_async(const MessageSendParams& params);
 
    Task get_task(const std::string& task_id, int history_length = 0);
    asio::awaitable<Task> get_task_async(const std::string& task_id, int history_length = 0);
 
    Task cancel_task(const std::string& task_id);
    asio::awaitable<Task> cancel_task_async(const std::string& task_id);
 
    using EventCallback = std::function<bool(const StreamEvent&)>;
    using StreamCallback [[deprecated("use a2a::A2AClient::EventCallback")]]
        = EventCallback;
 
    Task send_message_stream(const std::string& text,
                             EventCallback on_event,
                             const std::string& task_id    = "",
                             const std::string& context_id = "");
 
    Task send_message_stream(const MessageSendParams& params,
                             EventCallback on_event);
 
    json rpc_call(const std::string& method, const json& params);
    asio::awaitable<json> rpc_call_async(const std::string& method, const json& params);
 
    asio::awaitable<json> rpc_call_with_fallback(
        const std::string& v1_method,
        const std::string& v03_method,
        const json& params);
 
    const std::string& base_url() const { return base_url_; }
 
  private:
    std::string base_url_;
    std::chrono::seconds timeout_ = std::chrono::seconds(30);
    int request_id_ = 0;
 
    AgentCard cached_card_;
    bool      card_loaded_ = false;
};
 
} // namespace neograph::a2a