Skip to content

Commit 91fb2d3

Browse files
committed
move llamacpp to use registerProcessors
1 parent 508cfae commit 91fb2d3

2 files changed

Lines changed: 3 additions & 5 deletions

File tree

extensions/llamacpp/CMakeLists.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ include(LlamaCpp)
2525

2626
include(${CMAKE_SOURCE_DIR}/extensions/ExtensionHeader.txt)
2727

28-
file(GLOB SOURCES "processors/*.cpp")
28+
file(GLOB SOURCES "processors/*.cpp" "ExtensionInitializer.cpp")
2929

3030
add_minifi_library(minifi-llamacpp SHARED ${SOURCES})
3131
target_include_directories(minifi-llamacpp PUBLIC "${CMAKE_SOURCE_DIR}/extensions/llamacpp")

extensions/llamacpp/processors/ExtensionInitializer.cpp renamed to extensions/llamacpp/ExtensionInitializer.cpp

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -15,9 +15,9 @@
1515
* limitations under the License.
1616
*/
1717

18-
#include "RunLlamaCppInference.h"
1918
#include "api/core/Resource.h"
2019
#include "api/utils/minifi-c-utils.h"
20+
#include "processors/RunLlamaCppInference.h"
2121

2222
#define MKSOC(x) #x
2323
#define MAKESTRING(x) MKSOC(x) // NOLINT(cppcoreguidelines-macro-usage)
@@ -34,7 +34,5 @@ CEXTENSIONAPI void MinifiInitExtension(MinifiExtensionContext* extension_context
3434
.user_data = nullptr
3535
};
3636
auto* extension = MinifiCreateExtension(extension_context, &ext_create_info);
37-
minifi::api::core::useProcessorClassDescription<minifi::extensions::llamacpp::processors::RunLlamaCppInference>([&] (const MinifiProcessorClassDefinition& description) {
38-
MinifiRegisterProcessor(extension, &description);
39-
});
37+
minifi::api::core::registerProcessors<minifi::extensions::llamacpp::processors::RunLlamaCppInference>(extension);
4038
}

0 commit comments

Comments
 (0)