wehub-resource-sync
22 lines
749 B
C++
22 lines
749 B
C++
// SPDX-License-Identifier: Apache-2.0
|
|
#include <pybind11/pybind11.h>
|
|
#include "connector_pybind_utils.h"
|
|
#include "connector.h"
|
|
|
|
namespace py = pybind11;
|
|
|
|
PYBIND11_MODULE(_native, m) {
|
|
m.doc() = "Example native connector plugin for LMCache";
|
|
|
|
py::class_<example_connector::ExampleFSConnector>(m, "ExampleFSConnector")
|
|
.def(py::init<std::string, int>(), py::arg("base_path"),
|
|
py::arg("num_workers"))
|
|
LMCACHE_BIND_CONNECTOR_METHODS(example_connector::ExampleFSConnector);
|
|
|
|
py::class_<example_connector::ExampleMemoryConnector>(
|
|
m, "ExampleMemoryConnector")
|
|
.def(py::init<int>(), py::arg("num_workers"))
|
|
LMCACHE_BIND_CONNECTOR_METHODS(
|
|
example_connector::ExampleMemoryConnector);
|
|
}
|