ILLM.h

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: 2023-2023 Istituto Italiano di Tecnologia (IIT)
 * SPDX-License-Identifier: BSD-3-Clause
 */
 
#ifndef YARP_DEV_ILLM_H
#define YARP_DEV_ILLM_H
 
#include <yarp/dev/api.h>
#include <yarp/dev/ReturnValue.h>
#include <yarp/dev/LLM_Message.h>
 
#include <string>
#include <vector>
 
namespace yarp::dev {
class ILLM;
} // namespace yarp
 
class YARP_dev_API yarp::dev::ILLM
{
public:
    virtual ~ILLM();
 
    virtual yarp::dev::ReturnValue setPrompt(const std::string& prompt) = 0;
 
    virtual yarp::dev::ReturnValue readPrompt(std::string& oPrompt) = 0;
 
    virtual yarp::dev::ReturnValue ask(const std::string& question, yarp::dev::LLM_Message& answer) = 0;
 
    virtual yarp::dev::ReturnValue getConversation(std::vector<yarp::dev::LLM_Message>& conversation) = 0;
 
    virtual yarp::dev::ReturnValue deleteConversation() = 0;
 
    virtual yarp::dev::ReturnValue refreshConversation() = 0;
};
 
#endif