prompt_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 PromptBaseClass : public RestBaseClass
{
public:
  PromptBaseClass() : RestBaseClass(), uri_(""), chat_uri_("")
  {
  }
 
  virtual ~PromptBaseClass() = default;
 
  virtual void initialize_prompt_base(rclcpp::Node::SharedPtr node, std::string plugin_name = "PromptBaseClass",
                                      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.uri"))
    {
      node_->declare_parameter(plugin_name_ + ".rest.uri", "");
    }
    if (!node_->has_parameter(plugin_name_ + ".rest.chat_uri"))
    {
      node_->declare_parameter(plugin_name_ + ".rest.chat_uri", "");
    }
 
    // get parameters from the parameter server
    uri_ = node_->get_parameter(plugin_name_ + ".rest.uri").as_string();
    chat_uri_ = node_->get_parameter(plugin_name_ + ".rest.chat_uri").as_string();
 
    // log the parameters
    RCLCPP_INFO(node_->get_logger(), "%s Single URI: %s", plugin_name_.c_str(), uri_.c_str());
    RCLCPP_INFO(node_->get_logger(), "%s Chat URI: %s", plugin_name_.c_str(), chat_uri_.c_str());
  }
 
  virtual prompt::PromptResponse sendPrompt(prompt::PromptRequest& req)
  {
    // 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, uri_);
 
    // create prompt provider response container
    prompt::PromptResponse res = fromJson(object);
 
    return res;
  }
 
  virtual prompt::PromptResponse sendConversation(prompt::PromptRequest& latest_request,
                                                  std::vector<PromptDialogue>& conversation)
  {
    // verify and add required model options
    prompt::ensure_model_options(node_, latest_request.options, required_options_);
 
    // prepare request body
    Poco::JSON::Object body_json = toJsonConversation(latest_request, conversation);
 
    // process the prompt
    Poco::JSON::Object::Ptr object = process(body_json, chat_uri_);
 
    // create prompt provider response container
    prompt::PromptResponse res = fromJsonConversation(object);
 
    return res;
  }
 
protected:
  virtual Poco::JSON::Object toJson(prompt::PromptRequest& prompt) = 0;
 
  virtual Poco::JSON::Object toJsonConversation(prompt::PromptRequest& latest_request,
                                                std::vector<PromptDialogue>& conversation) = 0;
 
  virtual prompt::PromptResponse fromJson(const Poco::JSON::Object::Ptr object) = 0;
 
  virtual prompt::PromptResponse fromJsonConversation(const Poco::JSON::Object::Ptr object) = 0;
 
  std::string uri_;
 
  std::string chat_uri_;
};
 
}  // namespace prompt