Prediction.hh

Go to the documentation of this file.

//
// Prediction.hh
//
// Copyright (c) 2024 Couchbase, Inc All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//
 
 
#ifdef COUCHBASE_ENTERPRISE
 
#pragma once
#include "cbl++/Base.hh"
#include "cbl/CBLPrediction.h"
#include <string_view>
 
 
CBL_ASSUME_NONNULL_BEGIN
 
namespace cbl {
    using PredictiveModel = std::function<fleece::MutableDict(fleece::Dict)>;

    class Prediction {
    public:
        static void registerModel(std::string_view name, PredictiveModel model) {
            auto* holder = new PredictiveModel(std::move(model));
            CBLPredictiveModel config{};
            config.context      = holder;
            config.prediction   = [](void* ctx, FLDict input) -> FLMutableDict {
                try {
                    auto& fn = *static_cast<PredictiveModel*>(ctx);
                    return FLMutableDict_Retain((FLMutableDict)fn(fleece::Dict(input)));
                } catch (const cbl::Error& error) {
                    CBL_Log(kCBLLogDomainDatabase, kCBLLogError, "Prediction function throws error %d/%d: %s",
                            error.domain, error.code, error.what());
                } catch (const std::exception& error) {
                    CBL_Log(kCBLLogDomainDatabase, kCBLLogError, "Prediction function throws error %s",
                            error.what());
                } catch (...) {
                    CBL_Log(kCBLLogDomainDatabase, kCBLLogError, "Prediction function throws unknown exception");
                }
                return FLMutableDict_New();
            };
            config.unregistered = [](void* ctx) {
                delete static_cast<PredictiveModel*>(ctx);
            };
            CBL_RegisterPredictiveModel(slice(name), config);
        }

        static void unregisterModel(std::string_view name) {
            CBL_UnregisterPredictiveModel(slice(name));
        }
    };
}
 
CBL_ASSUME_NONNULL_END
    
#endif