embed.h

/*********************************************************************************************************************************************/
/*  DocWire SDK: Award-winning modern data processing in C++20. SourceForge Community Choice & Microsoft support. AI-driven processing.      */
/*  Supports nearly 100 data formats, including email boxes and OCR. Boost efficiency in text extraction, web data extraction, data mining,  */
/*  document analysis. Offline processing possible for security and confidentiality                                                          */
/*                                                                                                                                           */
/*  Copyright (c) SILVERCODERS Ltd, http://silvercoders.com                                                                                  */
/*  Project homepage: https://github.com/docwire/docwire                                                                                     */
/*                                                                                                                                           */
/*  SPDX-License-Identifier: AGPL-3.0-only OR LicenseRef-DocWire-Commercial                                                                  */
/*********************************************************************************************************************************************/
 
#ifndef DOCWIRE_OPENAI_EMBED_H
#define DOCWIRE_OPENAI_EMBED_H
 
#include "chain_element.h"
#include "openai_export.h"
#include "message.h"
#include <string>
 
namespace docwire
{
namespace openai
{
 
class DOCWIRE_OPENAI_EXPORT embed : public chain_element, public with_pimpl<embed>
{
public:
    enum class model
    {
        text_embedding_3_small,
        text_embedding_3_large,
        text_embedding_ada_002,
    };
 
    embed(const std::string& api_key, model model = model::text_embedding_3_small);
 
    continuation operator()(message_ptr msg, const message_callbacks& emit_message) override;
 
    bool is_leaf() const override
    {
        return false;
    }
 
private:
    using with_pimpl<embed>::impl;
};
 
} // namespace openai
} // namespace docwire
 
#endif //DOCWIRE_OPENAI_EMBED_H