embed_base_class.hpp

Go to the documentation of this file.

#pragma once
 
#include <prompt_base/rest_base_class.hpp>
#include <prompt_base/utils/prompt_options.hpp>
#include <string>
 
namespace prompt
{
 
class EmbedBaseClass : public RestBaseClass
{
public:
  EmbedBaseClass() : RestBaseClass(), embed_uri_("")
  {
  }
 
  virtual ~EmbedBaseClass() = default;
 
  virtual void initialize_embed_base(rclcpp::Node::SharedPtr node, std::string plugin_name = "EmbedBaseClass",
                                     std::string api_key_name = "")
  {
    // initialize base class
    initialize_rest_base(node, plugin_name, api_key_name);
 
    // declare parameters only if not already declared (idempotent init)
    if (!node_->has_parameter(plugin_name_ + ".rest.embedding_uri"))
    {
      node_->declare_parameter(plugin_name_ + ".rest.embedding_uri", "");
    }
 
    // get parameters from the parameter server
    embed_uri_ = node_->get_parameter(plugin_name_ + ".rest.embedding_uri").as_string();
 
    // log the parameters
    RCLCPP_INFO(node_->get_logger(), "%s Embedding URI: %s", plugin_name_.c_str(), embed_uri_.c_str());
  }
 
  virtual prompt::EmbedResponse get_embeddings(prompt::EmbedRequest& req) override
  {
    // verify and add required model options
    prompt::ensure_model_options(node_, req.options, required_options_);
 
    // prepare request body
    Poco::JSON::Object body_json = toJson(req);
 
    // process the prompt
    Poco::JSON::Object::Ptr object = process(body_json, embed_uri_);
 
    // create prompt provider response container
    prompt::EmbedResponse res = fromJson(object);
 
    return res;
  }
 
protected:
  virtual Poco::JSON::Object toJson(prompt::EmbedRequest& input) = 0;
 
  virtual prompt::EmbedResponse fromJson(const Poco::JSON::Object::Ptr object) = 0;
 
  std::string embed_uri_;
};
 
}  // namespace prompt