server.h

Go to the documentation of this file.

 
#pragma once
 
#include <neograph/api.h>
#include <neograph/a2a/types.h>
#include <neograph/graph/engine.h>
 
#include <functional>
#include <memory>
#include <string>
 
namespace neograph::a2a {
 
class NEOGRAPH_API GraphAgentAdapter {
  public:
    virtual ~GraphAgentAdapter() = default;
 
    virtual std::string input_channel() const { return "prompt"; }
 
    virtual std::string output_channel() const { return "response"; }
 
    virtual neograph::json build_initial_state(const std::string& user_text) const;
};
 
class NEOGRAPH_API A2AServer {
  public:
    A2AServer(std::shared_ptr<neograph::graph::GraphEngine> engine,
              AgentCard card,
              std::shared_ptr<GraphAgentAdapter> adapter = {});
 
    ~A2AServer();
 
    bool start(const std::string& host, int port);
 
    bool start_async(const std::string& host, int port);
 
    void stop();
 
    bool is_running() const;
 
    int port() const;
 
  private:
    struct Impl;
    std::unique_ptr<Impl> impl_;
};
 
}  // namespace neograph::a2a