base_class.hpp

Go to the documentation of this file.

#pragma once
 
#include <prompt_base/utils/exceptions.hpp>
#include <prompt_base/utils/structs.hpp>
#include <rclcpp/rclcpp.hpp>
 
namespace prompt
{
class BaseClass
{
public:
  BaseClass() = default;
  virtual ~BaseClass() = default;
 
  virtual void initialize(rclcpp::Node::SharedPtr node) = 0;
 
  virtual prompt::EmbedResponse get_embeddings(prompt::EmbedRequest& req)
  {
    throw prompt::PromptException("get_embeddings not implemented in base class");
  }
 
  virtual prompt::TokenResponse get_tokens(prompt::TokenRequest& req)
  {
    throw prompt::PromptException("get_tokens not implemented in base class");
  }
 
  virtual prompt::PromptResponse sendPrompt(prompt::PromptRequest& req)
  {
    throw prompt::PromptException("sendPrompt not implemented in base class");
  }
 
  virtual prompt::PromptResponse sendConversation(prompt::PromptRequest& latest_request,
                                                  std::vector<PromptDialogue>& conversation)
  {
    throw prompt::PromptException("sendConversation not implemented in base class");
  }
 
protected:
  virtual void initialize_base(rclcpp::Node::SharedPtr node, std::string plugin_name = "BaseClass")
  {
    // set the node pointer
    node_ = node;
 
    // Declare the plugin name parameter
    plugin_name_ = plugin_name;
  }
 
  rclcpp::Node::SharedPtr node_;
 
  std::string plugin_name_;
};
 
}  // namespace prompt