a2a_caller_node.h

Go to the documentation of this file.

 
#pragma once
 
#include <neograph/api.h>
#include <neograph/a2a/client.h>
#include <neograph/graph/node.h>
 
#include <functional>
#include <memory>
#include <string>
 
namespace neograph::a2a {
 
class NEOGRAPH_API A2ACallerNode : public neograph::graph::GraphNode {
  public:
    using MessageIdFactory = std::function<std::string()>;
 
    A2ACallerNode(std::string name,
                   std::shared_ptr<A2AClient> client,
                   std::string input_key  = "prompt",
                   std::string output_key = "response",
                   MessageIdFactory message_id_factory = {});
 
    asio::awaitable<neograph::graph::NodeOutput>
    run(neograph::graph::NodeInput in) override;
 
    std::string get_name() const override { return name_; }
 
  private:
    std::string                name_;
    std::shared_ptr<A2AClient> client_;
    std::string                input_key_;
    std::string                output_key_;
    MessageIdFactory           message_id_factory_;
};
 
} // namespace neograph::a2a