AaronPo/the-biergarten-app
1
0
Fork 0
mirror of https://github.com/aaronpo97/the-biergarten-app.git synced 2026-05-31 17:53:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

make prompt formatter unique ptr

Browse Source
This commit is contained in:
Aaron Po
2026-04-18 18:10:38 -04:00
parent 49f4ed6787
commit 88527f7709
2 changed files with 14 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pipeline/includes/data_generation/llama_generator.h
View File

@@ -135,7 +135,7 @@ class LlamaGenerator final : public DataGenerator {
std::mt19937 rng_; std::mt19937 rng_;
uint32_t n_ctx_ = kDefaultContextSize; uint32_t n_ctx_ = kDefaultContextSize;
std::string brewery_system_prompt_; std::string brewery_system_prompt_;
std::shared_ptr<IPromptFormatter> prompt_formatter_; std::unique_ptr<IPromptFormatter> prompt_formatter_;
}; };
#endif // BIERGARTEN_PIPELINE_INCLUDES_DATA_GENERATION_LLAMA_GENERATOR_H_ #endif // BIERGARTEN_PIPELINE_INCLUDES_DATA_GENERATION_LLAMA_GENERATOR_H_

14
pipeline/src/main.cc
View File

@@ -8,6 +8,7 @@
#include <boost/di.hpp> #include <boost/di.hpp>
#include <boost/program_options.hpp> #include <boost/program_options.hpp>
#include <chrono>
#include <exception> #include <exception>
#include <memory> #include <memory>
#include <optional> #include <optional>
@@ -131,8 +132,19 @@ std::optional<ApplicationOptions> ParseArguments(const int argc, char** argv) {
} }
} }
struct Timer {
std::chrono::steady_clock::time_point start_time =
std::chrono::steady_clock::now();
[[nodiscard]] int64_t Elapsed() const {
return std::chrono::duration_cast<std::chrono::milliseconds>(
std::chrono::steady_clock::now() - start_time)
.count();
}
};
int main(const int argc, char** argv) { int main(const int argc, char** argv) {
try { try {
Timer timer;
const CurlGlobalState curl_state; const CurlGlobalState curl_state;
const LlamaBackendState llama_backend_state; const LlamaBackendState llama_backend_state;
spdlog::set_pattern("[%Y-%m-%d %H:%M:%S.%e] [%^%l%$] %v"); spdlog::set_pattern("[%Y-%m-%d %H:%M:%S.%e] [%^%l%$] %v");
@@ -173,7 +185,7 @@ int main(const int argc, char** argv) {
return 1; return 1;
} }
spdlog::info("Pipeline executed successfully"); spdlog::info("Pipeline executed successfully in {} ms", timer.Elapsed());
return 0; return 0;
} catch (const std::exception& exception) { } catch (const std::exception& exception) {
spdlog::critical("Unhandled fatal error in main: {}", exception.what()); spdlog::critical("Unhandled fatal error in main: {}", exception.what());