CommitMessageGenerator.h

Go to the documentation of this file.

#ifndef COMMITMESSAGEGENERATOR_H
#define COMMITMESSAGEGENERATOR_H
 
#include "GenerativeAI.h"
#include "Git.h"
#include <string>
 
class CommitMessageGenerator {
public:
    enum Kind {
        CommitMessage,
    };
    class Result {
    public:
        bool error = false;
        std::string error_status;
        std::string error_message;
        std::vector<std::string> messages;
        Result() = default;
        Result(std::vector<std::string> const &messages)
            : messages(messages)
        {
        }
    };
private:
    Kind kind;
    CommitMessageGenerator::Result parse_response(const std::string &in, const GenerativeAI::Provider &provider);
    std::string generatePrompt(const std::string &diff, int max);
    std::string generatePromptJSON(const std::string &prompt, const GenerativeAI::Model &model);
public:
    CommitMessageGenerator() = default;
    CommitMessageGenerator(Kind kind)
        : kind(kind)
    {
    }
    Result generate(std::string const &diff, QString const &hint = {});
    static std::string diff_head(GitRunner g);
    static Result Error(std::string const &status, std::string const &message)
    {
        Result ret;
        ret.error = true;
        ret.error_status = status;
        ret.error_message = message;
        return ret;
    }
};
 
#endif // COMMITMESSAGEGENERATOR_H