mx2-ollama.hpp

Go to the documentation of this file.

#ifndef __MX2_OLLAMA_HPP__
#define __MX2_OLLAMA_HPP__
 
#include <algorithm>
#include <cctype>
#include <curl/curl.h>
#include <fstream>
#include <iostream>
#include <regex>
#include <sstream>
#include <string>
 
namespace mx {
 
    struct ResponseData {
        std::string response;
        std::ostringstream shader_stream;
    };
 
    class ObjectRequestException : public std::exception {
      public:
        explicit ObjectRequestException(const std::string &message) : msg(message) {}
        virtual const char *what() const noexcept override { return msg.c_str(); }
 
      private:
        std::string msg;
    };
 
    class ObjectRequest {
      public:
        explicit ObjectRequest(const std::string &host = "localhost", const std::string &model = "codellama:7b", const std::string &filename = "shader.glsl") : host(host), model(model), filename(filename) {}
        void setHost(const std::string &host) { this->host = host; }
        void setModel(const std::string &model) { this->model = model; }
        void setFilename(const std::string &filename) { this->filename = filename; }
        void setShader(const std::string &shader) { this->shader = shader; }
 
        void setPrompt(const std::string &prompt) { this->prompt = prompt; }
 
        static std::string unescape(const std::string &input);
        std::string generateCode();
        static size_t WriteCallback(void *contents, size_t size, size_t nmemb, ResponseData *data);
 
      private:
        std::string host;
        std::string model;
        std::string filename;
        std::string shader;
        std::string prompt;
    };
 
} // namespace mx
 
#endif