1 Commits

Author SHA1 Message Date
Aaron Po
0473f48fb7 fix mock-only builds 2026-07-03 02:27:00 -04:00
2 changed files with 13 additions and 0 deletions

View File

@@ -16,8 +16,10 @@
// --- data_generation --- // --- data_generation ---
#include "data_generation/data_generator.h" #include "data_generation/data_generator.h"
#include "data_generation/json_grammars.h" #include "data_generation/json_grammars.h"
#ifndef BIERGARTEN_MOCK_ONLY
#include "data_generation/llama_generator.h" #include "data_generation/llama_generator.h"
#include "data_generation/llama_generator_helpers.h" #include "data_generation/llama_generator_helpers.h"
#endif
#include "data_generation/mock_generator.h" #include "data_generation/mock_generator.h"
#include "data_generation/prompt_formatting/gemma4_jinja_prompt_formatter.h" #include "data_generation/prompt_formatting/gemma4_jinja_prompt_formatter.h"
#include "data_generation/prompt_formatting/prompt_formatter.h" #include "data_generation/prompt_formatting/prompt_formatter.h"
@@ -31,7 +33,9 @@
#include "services/curated_data/curated_json_data_service.h" #include "services/curated_data/curated_json_data_service.h"
// --- llama backend --- // --- llama backend ---
#ifndef BIERGARTEN_MOCK_ONLY
#include "llama_backend_state.h" #include "llama_backend_state.h"
#endif
// --- services: curated_data --- // --- services: curated_data ---
#include "services/curated_data/curated_data_service.h" #include "services/curated_data/curated_data_service.h"

View File

@@ -174,6 +174,14 @@ int main(const int argc, char** argv) {
return std::make_unique<MockGenerator>(); return std::make_unique<MockGenerator>();
} }
#ifdef BIERGARTEN_MOCK_ONLY
// Unreachable: ParseArguments requires --mocked when
// BIERGARTEN_MOCK_ONLY is compiled in, so LlamaGenerator is
// never constructed and is not linked into this binary.
throw std::runtime_error(
"LlamaGenerator is unavailable in a BIERGARTEN_MOCK_ONLY "
"build");
#else
log_producer->Log( log_producer->Log(
{.level = LogLevel::Info, {.level = LogLevel::Info,
.phase = PipelinePhase::Startup, .phase = PipelinePhase::Startup,
@@ -187,6 +195,7 @@ int main(const int argc, char** argv) {
options, model_path, log_producer, options, model_path, log_producer,
inj.template create<std::unique_ptr<IPromptFormatter>>(), inj.template create<std::unique_ptr<IPromptFormatter>>(),
std::move(prompt_directory)); std::move(prompt_directory));
#endif
})); }));
const auto orchestrator = const auto orchestrator =