mirror of
https://github.com/aaronpo97/the-biergarten-app.git
synced 2026-06-01 01:54:00 +00:00
Compare commits
6 Commits
366cb99e1d
...
feat/pipel
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
7580f47a7d | ||
|
|
e251e7b2a3 | ||
|
|
20742bb613 | ||
|
|
54a46458a3 | ||
|
|
29972f54d1 | ||
|
|
f35c563dac |
@@ -26,6 +26,7 @@ skinparam note {
|
|||||||
title The Biergarten Data Pipeline - Class Diagram
|
title The Biergarten Data Pipeline - Class Diagram
|
||||||
|
|
||||||
class BiergartenDataGenerator {
|
class BiergartenDataGenerator {
|
||||||
|
- logger_ : std::shared_ptr<ILogger>
|
||||||
- context_service_ : std::unique_ptr<IEnrichmentService>
|
- context_service_ : std::unique_ptr<IEnrichmentService>
|
||||||
- generator_ : std::unique_ptr<DataGenerator>
|
- generator_ : std::unique_ptr<DataGenerator>
|
||||||
- exporter_ : std::unique_ptr<IExportService>
|
- exporter_ : std::unique_ptr<IExportService>
|
||||||
@@ -36,6 +37,46 @@ class BiergartenDataGenerator {
|
|||||||
- LogResults() : void
|
- LogResults() : void
|
||||||
}
|
}
|
||||||
|
|
||||||
|
class LogLevel <<enumeration>> {
|
||||||
|
Debug
|
||||||
|
Info
|
||||||
|
Warn
|
||||||
|
Error
|
||||||
|
}
|
||||||
|
|
||||||
|
class PipelinePhase <<enumeration>> {
|
||||||
|
Startup
|
||||||
|
UserGeneration
|
||||||
|
BreweryAndBeerGeneration
|
||||||
|
CheckinGeneration
|
||||||
|
RatingGeneration
|
||||||
|
FollowGeneration
|
||||||
|
Teardown
|
||||||
|
}
|
||||||
|
|
||||||
|
struct LogEntry {
|
||||||
|
+ timestamp : std::chrono::system_clock::time_point
|
||||||
|
+ level : LogLevel
|
||||||
|
+ phase : PipelinePhase
|
||||||
|
+ message : std::string
|
||||||
|
+ worker : std::optional<std::string>
|
||||||
|
}
|
||||||
|
|
||||||
|
interface ILogger <<interface>> {
|
||||||
|
+ Log(entry : const LogEntry&) : void
|
||||||
|
}
|
||||||
|
|
||||||
|
class LogProducer {
|
||||||
|
- channel_ : BoundedChannel<LogEntry>&
|
||||||
|
+ Log(entry : const LogEntry&) : void
|
||||||
|
}
|
||||||
|
|
||||||
|
class LogDispatcher {
|
||||||
|
- channel_ : BoundedChannel<LogEntry>&
|
||||||
|
+ Run() : void
|
||||||
|
- ToSpdlogLevel(level) : spdlog::level::level_enum
|
||||||
|
}
|
||||||
|
|
||||||
interface IEnrichmentService <<interface>> {
|
interface IEnrichmentService <<interface>> {
|
||||||
+ GetLocationContext(loc : const Location&) : std::string
|
+ GetLocationContext(loc : const Location&) : std::string
|
||||||
}
|
}
|
||||||
@@ -123,10 +164,17 @@ class SystemDateTimeProvider {
|
|||||||
}
|
}
|
||||||
|
|
||||||
' Structural Relationships / Dependency Injection
|
' Structural Relationships / Dependency Injection
|
||||||
|
BiergartenDataGenerator *-- ILogger : owns
|
||||||
BiergartenDataGenerator *-- IEnrichmentService : owns
|
BiergartenDataGenerator *-- IEnrichmentService : owns
|
||||||
BiergartenDataGenerator *-- DataGenerator : owns
|
BiergartenDataGenerator *-- DataGenerator : owns
|
||||||
BiergartenDataGenerator *-- IExportService : owns
|
BiergartenDataGenerator *-- IExportService : owns
|
||||||
|
|
||||||
|
LogEntry *-- LogLevel
|
||||||
|
LogEntry *-- PipelinePhase
|
||||||
|
ILogger <|.. LogProducer : implements
|
||||||
|
LogProducer ..> LogEntry : emits
|
||||||
|
LogDispatcher ..> LogEntry : consumes
|
||||||
|
|
||||||
IEnrichmentService <|.. WikipediaService : implements
|
IEnrichmentService <|.. WikipediaService : implements
|
||||||
WikipediaService *-- WebClient : owns
|
WikipediaService *-- WebClient : owns
|
||||||
|
|
||||||
|
|||||||
@@ -275,32 +275,29 @@ package "Infrastructure: Logging" {
|
|||||||
+ level : LogLevel
|
+ level : LogLevel
|
||||||
+ phase : PipelinePhase
|
+ phase : PipelinePhase
|
||||||
+ message : std::string
|
+ message : std::string
|
||||||
+ city : std::optional<std::string>
|
|
||||||
+ entity_id : std::optional<std::string>
|
|
||||||
+ worker : std::optional<std::string>
|
+ worker : std::optional<std::string>
|
||||||
}
|
}
|
||||||
|
|
||||||
interface Logger <<interface>> {
|
interface ILogger <<interface>> {
|
||||||
+ Log(level, phase, message,\n city, entity_id, worker) : void
|
+ Log(entry : const LogEntry&) : void
|
||||||
}
|
}
|
||||||
|
|
||||||
class PipelineLogger {
|
class LogProducer {
|
||||||
- log_ch_ : BoundedChannel<LogEntry>&
|
- channel_ : BoundedChannel<LogEntry>&
|
||||||
+ Log(level, phase, message,\n city, entity_id, worker) : void
|
+ Log(entry : const LogEntry&) : void
|
||||||
}
|
}
|
||||||
|
|
||||||
class LogWorker {
|
class LogDispatcher {
|
||||||
- log_ch_ : BoundedChannel<LogEntry>&
|
- channel_ : BoundedChannel<LogEntry>&
|
||||||
+ Run() : void
|
+ Run() : void
|
||||||
- FormatTimestamp(tp) : std::string
|
|
||||||
- ToSpdlogLevel(level) : spdlog::level::level_enum
|
- ToSpdlogLevel(level) : spdlog::level::level_enum
|
||||||
- ToString(phase) : std::string
|
|
||||||
}
|
}
|
||||||
|
|
||||||
LogEntry *-- LogLevel
|
LogEntry *-- LogLevel
|
||||||
LogEntry *-- PipelinePhase
|
LogEntry *-- PipelinePhase
|
||||||
PipelineLogger ..> LogEntry : emits
|
ILogger <|.. LogProducer
|
||||||
LogWorker ..> LogEntry : consumes
|
LogProducer ..> LogEntry : emits
|
||||||
|
LogDispatcher ..> LogEntry : consumes
|
||||||
}
|
}
|
||||||
|
|
||||||
package "Infrastructure: Pipeline Channel" {
|
package "Infrastructure: Pipeline Channel" {
|
||||||
|
|||||||
@@ -138,7 +138,8 @@ FetchContent_MakeAvailable(cpp-httplib)
|
|||||||
|
|
||||||
# 5. Executable & Sources
|
# 5. Executable & Sources
|
||||||
add_executable(${PROJECT_NAME}
|
add_executable(${PROJECT_NAME}
|
||||||
includes/services/enrichment/mock_enrichment.h)
|
includes/services/enrichment/mock_enrichment.h
|
||||||
|
includes/json_handling/pretty_print.h)
|
||||||
|
|
||||||
# --- Entry point ---
|
# --- Entry point ---
|
||||||
target_sources(${PROJECT_NAME} PRIVATE
|
target_sources(${PROJECT_NAME} PRIVATE
|
||||||
@@ -248,6 +249,11 @@ target_compile_definitions(${PROJECT_NAME} PRIVATE
|
|||||||
$<$<CONFIG:Debug>:DEBUG>
|
$<$<CONFIG:Debug>:DEBUG>
|
||||||
)
|
)
|
||||||
|
|
||||||
|
target_compile_options(biergarten-pipeline PRIVATE
|
||||||
|
-fmacro-prefix-map=${CMAKE_SOURCE_DIR}/tooling/pipeline/src/=
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
# 7. Runtime Assets
|
# 7. Runtime Assets
|
||||||
configure_file(
|
configure_file(
|
||||||
${CMAKE_SOURCE_DIR}/locations.json
|
${CMAKE_SOURCE_DIR}/locations.json
|
||||||
@@ -259,3 +265,4 @@ add_custom_command(TARGET ${PROJECT_NAME} POST_BUILD
|
|||||||
${CMAKE_SOURCE_DIR}/prompts
|
${CMAKE_SOURCE_DIR}/prompts
|
||||||
${CMAKE_BINARY_DIR}/prompts
|
${CMAKE_BINARY_DIR}/prompts
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|||||||
@@ -51,6 +51,12 @@ class BiergartenPipelineOrchestrator {
|
|||||||
* 2. Resolve context for each city using the injected context service
|
* 2. Resolve context for each city using the injected context service
|
||||||
* 3. Generate brewery data for sampled cities
|
* 3. Generate brewery data for sampled cities
|
||||||
*
|
*
|
||||||
|
* @note STRUCTURAL CONCURRENCY REQUIREMENT:
|
||||||
|
* When transitioned to a multithreaded design, this method MUST structurally
|
||||||
|
* enforce that all deployed worker threads are joined before returning (e.g.
|
||||||
|
* by using std::jthread or a structured concurrency primitive). This ensures
|
||||||
|
* workers do not attempt to log to a closed channel during application teardown.
|
||||||
|
*
|
||||||
* @return true if successful, false if not
|
* @return true if successful, false if not
|
||||||
*/
|
*/
|
||||||
bool Run();
|
bool Run();
|
||||||
|
|||||||
@@ -38,8 +38,7 @@ class LlamaGenerator final : public DataGenerator {
|
|||||||
* @param prompt_directory Directory service for loading named prompt files.
|
* @param prompt_directory Directory service for loading named prompt files.
|
||||||
*/
|
*/
|
||||||
LlamaGenerator(const ApplicationOptions& options,
|
LlamaGenerator(const ApplicationOptions& options,
|
||||||
const std::string& model_path,
|
const std::string& model_path, std::shared_ptr<ILogger> logger,
|
||||||
std::shared_ptr<ILogger> logger,
|
|
||||||
std::unique_ptr<IPromptFormatter> prompt_formatter,
|
std::unique_ptr<IPromptFormatter> prompt_formatter,
|
||||||
std::unique_ptr<IPromptDirectory> prompt_directory);
|
std::unique_ptr<IPromptDirectory> prompt_directory);
|
||||||
|
|
||||||
|
|||||||
109
tooling/pipeline/includes/json_handling/pretty_print.h
Normal file
109
tooling/pipeline/includes/json_handling/pretty_print.h
Normal file
@@ -0,0 +1,109 @@
|
|||||||
|
#ifndef BIERGARTEN_PIPELINE_INCLUDES_JSON_HANDLING_PRETTY_PRINT_H_
|
||||||
|
#define BIERGARTEN_PIPELINE_INCLUDES_JSON_HANDLING_PRETTY_PRINT_H_
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @file json_handling/pretty_print.h
|
||||||
|
* @brief Pretty-printing utilities for JSON values.
|
||||||
|
*
|
||||||
|
* Provides formatting capability for boost::json::value with indentation and
|
||||||
|
* readable output. Adapted from Boost JSON library examples.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include <boost/json.hpp>
|
||||||
|
#include <ostream>
|
||||||
|
#include <string>
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Pretty-prints a JSON value to an output stream with indentation.
|
||||||
|
*
|
||||||
|
* Recursively formats JSON objects and arrays with consistent 4-space
|
||||||
|
* indentation. Adapted from:
|
||||||
|
* https://raw.githubusercontent.com/boostorg/json/refs/heads/develop/example/pretty.cpp
|
||||||
|
*
|
||||||
|
* @param outstream Output stream to write formatted JSON.
|
||||||
|
* @param json_val JSON value to format.
|
||||||
|
* @param indent Optional indentation string (managed internally on first call).
|
||||||
|
*/
|
||||||
|
inline void PrettyPrint(std::ostream& outstream,
|
||||||
|
boost::json::value const& json_val,
|
||||||
|
std::string* indent = nullptr) {
|
||||||
|
std::string str;
|
||||||
|
if (indent == nullptr) {
|
||||||
|
indent = &str;
|
||||||
|
}
|
||||||
|
switch (json_val.kind()) {
|
||||||
|
case boost::json::kind::object: {
|
||||||
|
outstream << "{\n";
|
||||||
|
indent->append(4, ' ');
|
||||||
|
auto const& obj = json_val.get_object();
|
||||||
|
if (!obj.empty()) {
|
||||||
|
const auto* iter = obj.begin();
|
||||||
|
for (;;) {
|
||||||
|
outstream << *indent << boost::json::serialize(iter->key()) << " : ";
|
||||||
|
PrettyPrint(outstream, iter->value(), indent);
|
||||||
|
iter = std::next(iter);
|
||||||
|
if (iter == obj.end()) {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
outstream << ",\n";
|
||||||
|
}
|
||||||
|
}
|
||||||
|
outstream << "\n";
|
||||||
|
indent->resize(indent->size() - 4);
|
||||||
|
outstream << *indent << "}";
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
case boost::json::kind::array: {
|
||||||
|
outstream << "[\n";
|
||||||
|
indent->append(4, ' ');
|
||||||
|
auto const& arr = json_val.get_array();
|
||||||
|
if (!arr.empty()) {
|
||||||
|
const auto* iter = arr.begin();
|
||||||
|
for (;;) {
|
||||||
|
outstream << *indent;
|
||||||
|
PrettyPrint(outstream, *iter, indent);
|
||||||
|
iter = std::next(iter);
|
||||||
|
if (iter == arr.end()) {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
outstream << ",\n";
|
||||||
|
}
|
||||||
|
}
|
||||||
|
outstream << "\n";
|
||||||
|
indent->resize(indent->size() - 4);
|
||||||
|
outstream << *indent << "]";
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
case boost::json::kind::string: {
|
||||||
|
outstream << serialize(json_val.get_string());
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
case boost::json::kind::uint64:
|
||||||
|
case boost::json::kind::int64:
|
||||||
|
case boost::json::kind::double_:
|
||||||
|
outstream << json_val;
|
||||||
|
break;
|
||||||
|
|
||||||
|
case boost::json::kind::bool_:
|
||||||
|
if (json_val.get_bool()) {
|
||||||
|
outstream << "true";
|
||||||
|
} else {
|
||||||
|
outstream << "false";
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
|
||||||
|
case boost::json::kind::null:
|
||||||
|
outstream << "null";
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (indent->empty()) {
|
||||||
|
outstream << "\n";
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#endif
|
||||||
@@ -6,8 +6,8 @@
|
|||||||
* them to spdlog on a dedicated thread.
|
* them to spdlog on a dedicated thread.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#ifndef BIERGARTEN_PIPELINE_INCLUDES_SERVICES_LOGGING_LOG_CONSUMER_H_
|
#ifndef BIERGARTEN_PIPELINE_INCLUDES_SERVICES_LOGGING_LOG_DISPATCHER_H_
|
||||||
#define BIERGARTEN_PIPELINE_INCLUDES_SERVICES_LOGGING_LOG_CONSUMER_H_
|
#define BIERGARTEN_PIPELINE_INCLUDES_SERVICES_LOGGING_LOG_DISPATCHER_H_
|
||||||
|
|
||||||
#include <spdlog/spdlog.h>
|
#include <spdlog/spdlog.h>
|
||||||
|
|
||||||
@@ -23,31 +23,31 @@
|
|||||||
*/
|
*/
|
||||||
class LogDispatcher {
|
class LogDispatcher {
|
||||||
public:
|
public:
|
||||||
/**
|
/**
|
||||||
* @brief Construct a log dispatcher.
|
* @brief Construct a log dispatcher.
|
||||||
*
|
*
|
||||||
* @param channel Reference to the bounded channel used for log retrieval.
|
* @param channel Reference to the bounded channel used for log retrieval.
|
||||||
*/
|
*/
|
||||||
explicit LogDispatcher(BoundedChannel<LogEntry>& channel);
|
explicit LogDispatcher(BoundedChannel<LogEntry>& channel);
|
||||||
|
|
||||||
LogDispatcher(const LogDispatcher&) = delete;
|
LogDispatcher(const LogDispatcher&) = delete;
|
||||||
LogDispatcher& operator=(const LogDispatcher&) = delete;
|
LogDispatcher& operator=(const LogDispatcher&) = delete;
|
||||||
LogDispatcher(LogDispatcher&&) = delete;
|
LogDispatcher(LogDispatcher&&) = delete;
|
||||||
LogDispatcher& operator=(LogDispatcher&&) = delete;
|
LogDispatcher& operator=(LogDispatcher&&) = delete;
|
||||||
~LogDispatcher() = default;
|
~LogDispatcher() = default;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @brief Drain the channel and forward entries to spdlog.
|
* @brief Drain the channel and forward entries to spdlog.
|
||||||
*
|
*
|
||||||
* Intended to be called once on a dedicated thread. The loop returns after
|
* Intended to be called once on a dedicated thread. The loop returns after
|
||||||
* the channel has been closed and all queued entries have been processed.
|
* the channel has been closed and all queued entries have been processed.
|
||||||
*/
|
*/
|
||||||
void Run();
|
void Run();
|
||||||
|
|
||||||
private:
|
private:
|
||||||
BoundedChannel<LogEntry>& channel_;
|
BoundedChannel<LogEntry>& channel_;
|
||||||
|
|
||||||
static spdlog::level::level_enum ToSpdlogLevel(LogLevel level);
|
static spdlog::level::level_enum ToSpdlogLevel(LogLevel level);
|
||||||
};
|
};
|
||||||
|
|
||||||
#endif // BIERGARTEN_PIPELINE_INCLUDES_SERVICES_LOGGING_LOG_CONSUMER_H_
|
#endif // BIERGARTEN_PIPELINE_INCLUDES_SERVICES_LOGGING_LOG_DISPATCHER_H_
|
||||||
|
|||||||
@@ -10,8 +10,10 @@
|
|||||||
#define BIERGARTEN_PIPELINE_INCLUDES_SERVICES_LOGGING_LOG_ENTRY_H_
|
#define BIERGARTEN_PIPELINE_INCLUDES_SERVICES_LOGGING_LOG_ENTRY_H_
|
||||||
|
|
||||||
#include <chrono>
|
#include <chrono>
|
||||||
#include <thread>
|
#include <source_location>
|
||||||
#include <string>
|
#include <string>
|
||||||
|
#include <thread>
|
||||||
|
#include <vector>
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @enum LogLevel
|
* @enum LogLevel
|
||||||
@@ -41,17 +43,35 @@ enum class PipelinePhase {
|
|||||||
Teardown, ///< Finalization and cleanup.
|
Teardown, ///< Finalization and cleanup.
|
||||||
};
|
};
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @struct LogDTO
|
||||||
|
* @brief User-provided subset of log fields. Used to capture call-site info transparently.
|
||||||
|
*/
|
||||||
|
struct LogDTO {
|
||||||
|
LogLevel level;
|
||||||
|
PipelinePhase phase;
|
||||||
|
std::string message;
|
||||||
|
};
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @struct LogEntry
|
* @struct LogEntry
|
||||||
* @brief Single structured log event.
|
* @brief Single structured log event.
|
||||||
*
|
*
|
||||||
* All fields are value types, which keeps transfer across the bounded channel
|
* All fields are value types, which keeps transfer across the bounded channel
|
||||||
* simple and avoids shared ownership.
|
* simple and avoids shared ownership.
|
||||||
|
*
|
||||||
|
* NOTE: timestamp, thread_id, and origin must be populated by ILogger::Log()
|
||||||
|
* before the entry is dispatched.
|
||||||
*/
|
*/
|
||||||
struct LogEntry {
|
struct LogEntry {
|
||||||
/// @brief Timestamp when the entry was created.
|
/// @brief Timestamp when the entry was created.
|
||||||
std::chrono::system_clock::time_point timestamp =
|
std::chrono::system_clock::time_point timestamp{};
|
||||||
std::chrono::system_clock::now();
|
|
||||||
|
/// @brief Source location where the log call was made.
|
||||||
|
std::source_location origin{};
|
||||||
|
|
||||||
|
/// @brief Thread responsible for emitting the log.
|
||||||
|
std::thread::id thread_id{};
|
||||||
|
|
||||||
|
|
||||||
/// @brief Severity level of this entry.
|
/// @brief Severity level of this entry.
|
||||||
@@ -65,4 +85,4 @@ struct LogEntry {
|
|||||||
|
|
||||||
};
|
};
|
||||||
|
|
||||||
#endif // BIERGARTEN_PIPELINE_INCLUDES_SERVICES_LOGGING_LOG_ENTRY_H_
|
#endif // BIERGARTEN_PIPELINE_INCLUDES_SERVICES_LOGGING_LOG_ENTRY_H_
|
||||||
|
|||||||
@@ -41,9 +41,10 @@ class LogProducer final : public ILogger {
|
|||||||
/**
|
/**
|
||||||
* @brief Queue a log message for asynchronous processing.
|
* @brief Queue a log message for asynchronous processing.
|
||||||
*
|
*
|
||||||
* Blocks while the channel applies backpressure.
|
* Blocks while the channel applies backpressure. This blocking behavior
|
||||||
|
* under heavy load is an accepted trade-off for simplicity.
|
||||||
*/
|
*/
|
||||||
void Log(LogEntry const& entry) override;
|
void DoLog(LogEntry log_entry) override;
|
||||||
|
|
||||||
private:
|
private:
|
||||||
BoundedChannel<LogEntry>& channel_;
|
BoundedChannel<LogEntry>& channel_;
|
||||||
|
|||||||
@@ -9,9 +9,9 @@
|
|||||||
#ifndef BIERGARTEN_PIPELINE_INCLUDES_SERVICES_LOGGING_LOGGER_H_
|
#ifndef BIERGARTEN_PIPELINE_INCLUDES_SERVICES_LOGGING_LOGGER_H_
|
||||||
#define BIERGARTEN_PIPELINE_INCLUDES_SERVICES_LOGGING_LOGGER_H_
|
#define BIERGARTEN_PIPELINE_INCLUDES_SERVICES_LOGGING_LOGGER_H_
|
||||||
|
|
||||||
#include <optional>
|
#include <source_location>
|
||||||
#include <string>
|
#include <string>
|
||||||
#include <string_view>
|
#include <utility>
|
||||||
|
|
||||||
#include "services/logging/log_entry.h"
|
#include "services/logging/log_entry.h"
|
||||||
|
|
||||||
@@ -34,10 +34,31 @@ class ILogger {
|
|||||||
/**
|
/**
|
||||||
* @brief Submit a log message to the logging subsystem.
|
* @brief Submit a log message to the logging subsystem.
|
||||||
*
|
*
|
||||||
* @param entry Structured log record containing message and metadata.
|
* @param payload User-provided log data (level, phase, message).
|
||||||
*
|
* @param origin Auto-captured source location of the call site.
|
||||||
*/
|
*/
|
||||||
virtual void Log(const LogEntry& entry) = 0;
|
void Log(LogDTO payload,
|
||||||
|
std::source_location origin = std::source_location::current(),
|
||||||
|
std::chrono::system_clock::time_point timestamp = std::chrono::system_clock::now(),
|
||||||
|
std::thread::id thread_id = std::this_thread::get_id()) {
|
||||||
|
LogEntry entry;
|
||||||
|
entry.timestamp = timestamp;
|
||||||
|
entry.thread_id = thread_id;
|
||||||
|
entry.level = payload.level;
|
||||||
|
entry.phase = payload.phase;
|
||||||
|
entry.message = std::move(payload.message);
|
||||||
|
entry.origin = origin;
|
||||||
|
DoLog(std::move(entry));
|
||||||
|
}
|
||||||
|
|
||||||
|
protected:
|
||||||
|
/**
|
||||||
|
* @brief Underlying implementation to transport the log entry.
|
||||||
|
*
|
||||||
|
* Implementations must be thread-safe as DoLog can be called concurrently
|
||||||
|
* from multiple worker threads.
|
||||||
|
*/
|
||||||
|
virtual void DoLog(LogEntry log_entry) = 0;
|
||||||
};
|
};
|
||||||
|
|
||||||
#endif // BIERGARTEN_PIPELINE_INCLUDES_SERVICES_LOGGING_LOGGER_H_
|
#endif // BIERGARTEN_PIPELINE_INCLUDES_SERVICES_LOGGING_LOGGER_H_
|
||||||
|
|||||||
@@ -1,4 +1,5 @@
|
|||||||
#include <chrono>
|
#include <chrono>
|
||||||
|
#include <format>
|
||||||
#include <iostream>
|
#include <iostream>
|
||||||
#include <optional>
|
#include <optional>
|
||||||
#include <sstream>
|
#include <sstream>
|
||||||
@@ -70,14 +71,12 @@ std::optional<ApplicationOptions> ParseArguments(
|
|||||||
return usage_stream.str();
|
return usage_stream.str();
|
||||||
})();
|
})();
|
||||||
if (logger) {
|
if (logger) {
|
||||||
logger->Log({.level = LogLevel::Info,
|
logger->Log(LogDTO{.level = LogLevel::Info,
|
||||||
.phase = PipelinePhase::Startup,
|
.phase = PipelinePhase::Startup,
|
||||||
.message = title});
|
.message = title});
|
||||||
logger->Log({.level = LogLevel::Info,
|
logger->Log(LogDTO{.level = LogLevel::Info,
|
||||||
.phase = PipelinePhase::Startup,
|
.phase = PipelinePhase::Startup,
|
||||||
.message = usage});
|
.message = usage});
|
||||||
} else {
|
|
||||||
std::cout << title << std::endl << usage << std::endl;
|
|
||||||
}
|
}
|
||||||
return std::nullopt;
|
return std::nullopt;
|
||||||
}
|
}
|
||||||
@@ -91,11 +90,9 @@ std::optional<ApplicationOptions> ParseArguments(
|
|||||||
std::stringstream help_stream;
|
std::stringstream help_stream;
|
||||||
help_stream << "\n" << desc;
|
help_stream << "\n" << desc;
|
||||||
if (logger) {
|
if (logger) {
|
||||||
logger->Log({.level = LogLevel::Info,
|
logger->Log(LogDTO{.level = LogLevel::Info,
|
||||||
.phase = PipelinePhase::Startup,
|
.phase = PipelinePhase::Startup,
|
||||||
.message = help_stream.str()});
|
.message = help_stream.str()});
|
||||||
} else {
|
|
||||||
std::cout << help_stream.str() << std::endl;
|
|
||||||
}
|
}
|
||||||
return std::nullopt;
|
return std::nullopt;
|
||||||
}
|
}
|
||||||
@@ -116,7 +113,7 @@ std::optional<ApplicationOptions> ParseArguments(
|
|||||||
const std::string msg =
|
const std::string msg =
|
||||||
"Invalid arguments: --mocked and --model are mutually exclusive";
|
"Invalid arguments: --mocked and --model are mutually exclusive";
|
||||||
if (logger) {
|
if (logger) {
|
||||||
logger->Log({.level = LogLevel::Error,
|
logger->Log(LogDTO{.level = LogLevel::Error,
|
||||||
.phase = PipelinePhase::Startup,
|
.phase = PipelinePhase::Startup,
|
||||||
.message = msg});
|
.message = msg});
|
||||||
} else {
|
} else {
|
||||||
@@ -129,7 +126,7 @@ std::optional<ApplicationOptions> ParseArguments(
|
|||||||
const std::string msg =
|
const std::string msg =
|
||||||
"Invalid arguments: either --mocked or --model must be specified";
|
"Invalid arguments: either --mocked or --model must be specified";
|
||||||
if (logger) {
|
if (logger) {
|
||||||
logger->Log({.level = LogLevel::Error,
|
logger->Log(LogDTO{.level = LogLevel::Error,
|
||||||
.phase = PipelinePhase::Startup,
|
.phase = PipelinePhase::Startup,
|
||||||
.message = msg});
|
.message = msg});
|
||||||
} else {
|
} else {
|
||||||
@@ -173,7 +170,7 @@ std::optional<ApplicationOptions> ParseArguments(
|
|||||||
const std::string msg =
|
const std::string msg =
|
||||||
"Sampling parameters are ignored when using --mocked";
|
"Sampling parameters are ignored when using --mocked";
|
||||||
if (logger) {
|
if (logger) {
|
||||||
logger->Log({.level = LogLevel::Warn,
|
logger->Log(LogDTO{.level = LogLevel::Warn,
|
||||||
.phase = PipelinePhase::Startup,
|
.phase = PipelinePhase::Startup,
|
||||||
.message = msg});
|
.message = msg});
|
||||||
} else {
|
} else {
|
||||||
@@ -199,22 +196,18 @@ std::optional<ApplicationOptions> ParseArguments(
|
|||||||
std::string("Failed to parse command-line arguments: ") +
|
std::string("Failed to parse command-line arguments: ") +
|
||||||
exception.what();
|
exception.what();
|
||||||
if (logger) {
|
if (logger) {
|
||||||
logger->Log({.level = LogLevel::Error,
|
logger->Log(LogDTO{.level = LogLevel::Error,
|
||||||
.phase = PipelinePhase::Startup,
|
.phase = PipelinePhase::Startup,
|
||||||
.message = msg});
|
.message = msg});
|
||||||
} else {
|
|
||||||
std::cerr << msg << std::endl;
|
|
||||||
}
|
}
|
||||||
return std::nullopt;
|
return std::nullopt;
|
||||||
} catch (...) {
|
} catch (...) {
|
||||||
const std::string msg =
|
const std::string msg =
|
||||||
"Failed to parse command-line arguments: unknown error";
|
"Failed to parse command-line arguments: unknown error";
|
||||||
if (logger) {
|
if (logger) {
|
||||||
logger->Log({.level = LogLevel::Error,
|
logger->Log(LogDTO{.level = LogLevel::Error,
|
||||||
.phase = PipelinePhase::Startup,
|
.phase = PipelinePhase::Startup,
|
||||||
.message = msg});
|
.message = msg});
|
||||||
} else {
|
|
||||||
std::cerr << msg << std::endl;
|
|
||||||
}
|
}
|
||||||
return std::nullopt;
|
return std::nullopt;
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -4,6 +4,7 @@
|
|||||||
*/
|
*/
|
||||||
|
|
||||||
#include <chrono>
|
#include <chrono>
|
||||||
|
#include <format>
|
||||||
|
|
||||||
#include "biergarten_pipeline_orchestrator.h"
|
#include "biergarten_pipeline_orchestrator.h"
|
||||||
#include "services/logging/logger.h"
|
#include "services/logging/logger.h"
|
||||||
@@ -36,34 +37,32 @@ void BiergartenPipelineOrchestrator::GenerateBreweries(
|
|||||||
{.level = LogLevel::Warn,
|
{.level = LogLevel::Warn,
|
||||||
.phase = PipelinePhase::BreweryAndBeerGeneration,
|
.phase = PipelinePhase::BreweryAndBeerGeneration,
|
||||||
.message =
|
.message =
|
||||||
std::string("[Pipeline] Generated brewery for '") +
|
std::format("[Pipeline] Generated brewery for '{}' ({}) but SQLite export failed: {}",
|
||||||
location.city + "' (" + location.country +
|
location.city, location.country, export_exception.what())});
|
||||||
") but SQLite export failed: " + export_exception.what()});
|
|
||||||
}
|
}
|
||||||
} catch (const std::exception& e) {
|
} catch (const std::exception& e) {
|
||||||
++skipped_count;
|
++skipped_count;
|
||||||
|
|
||||||
logger_->Log({.level = LogLevel::Warn,
|
logger_->Log({.level = LogLevel::Warn,
|
||||||
.phase = PipelinePhase::BreweryAndBeerGeneration,
|
.phase = PipelinePhase::BreweryAndBeerGeneration,
|
||||||
.message = std::string("[Pipeline] Skipping city '") +
|
.message = std::format("[Pipeline] Skipping city '{}' ({}): brewery generation failed: {}",
|
||||||
location.city + " (" + location.country +
|
location.city, location.country, e.what())});
|
||||||
"): brewery generation failed: " + e.what()});
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if (skipped_count > 0) {
|
if (skipped_count > 0) {
|
||||||
logger_->Log({.level = LogLevel::Warn,
|
logger_->Log({.level = LogLevel::Warn,
|
||||||
.phase = PipelinePhase::BreweryAndBeerGeneration,
|
.phase = PipelinePhase::BreweryAndBeerGeneration,
|
||||||
.message = std::string("[Pipeline] Skipped ") +
|
.message = std::format(
|
||||||
std::to_string(skipped_count) +
|
"[Pipeline] Skipped {} city/cities due to generation errors",
|
||||||
" city/cities due to generation errors"});
|
skipped_count)});
|
||||||
}
|
}
|
||||||
|
|
||||||
if (export_failed_count > 0) {
|
if (export_failed_count > 0) {
|
||||||
logger_->Log({.level = LogLevel::Warn,
|
logger_->Log({.level = LogLevel::Warn,
|
||||||
.phase = PipelinePhase::Teardown,
|
.phase = PipelinePhase::Teardown,
|
||||||
.message = std::string("[Pipeline] Failed to export ") +
|
.message = std::format(
|
||||||
std::to_string(export_failed_count) +
|
"[Pipeline] Failed to export {} generated brewery/breweries to SQLite",
|
||||||
" generated brewery/breweries to SQLite"});
|
export_failed_count)});
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -3,31 +3,35 @@
|
|||||||
* @brief BiergartenDataGenerator::LogResults() implementation.
|
* @brief BiergartenDataGenerator::LogResults() implementation.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
|
#include <boost/json/array.hpp>
|
||||||
#include <chrono>
|
#include <chrono>
|
||||||
#include <sstream>
|
#include <format>
|
||||||
|
|
||||||
|
#include "../../includes/json_handling/pretty_print.h"
|
||||||
#include "biergarten_pipeline_orchestrator.h"
|
#include "biergarten_pipeline_orchestrator.h"
|
||||||
#include "services/logging/logger.h"
|
#include "services/logging/logger.h"
|
||||||
|
|
||||||
void BiergartenPipelineOrchestrator::LogResults() const {
|
void BiergartenPipelineOrchestrator::LogResults() const {
|
||||||
std::ostringstream msg;
|
boost::json::array output;
|
||||||
msg << "GENERATED DATA DUMP\n";
|
|
||||||
size_t index = 1;
|
|
||||||
for (const auto& [location, brewery] : generated_breweries_) {
|
|
||||||
msg << index << ". city=\"" << location.city << "\" country=\""
|
|
||||||
<< location.country << "\" state=\"" << location.state_province
|
|
||||||
<< "\" iso3166_2=" << location.iso3166_2 << " lat=" << location.latitude
|
|
||||||
<< " lon=" << location.longitude << "\n";
|
|
||||||
|
|
||||||
msg << " brewery_name_en=\"" << brewery.name_en << "\"\n";
|
for (const auto& [location, brewery] : generated_breweries_) {
|
||||||
msg << " brewery_description_en=\"" << brewery.description_en << "\"\n";
|
output.push_back(boost::json::object{
|
||||||
msg << " brewery_name_local=\"" << brewery.name_local << "\"\n";
|
{"name_en", brewery.name_en},
|
||||||
msg << " brewery_description_local=\"" << brewery.description_local
|
{"description_en", brewery.description_en},
|
||||||
<< "\"\n";
|
{"name_local", brewery.name_local},
|
||||||
++index;
|
{"description_local", brewery.description_local},
|
||||||
|
{"location", boost::json::object{
|
||||||
|
{"city", location.city},
|
||||||
|
{"country", location.country},
|
||||||
|
{"state_province", location.state_province},
|
||||||
|
{"iso3166_2", location.iso3166_2},
|
||||||
|
{"latitude", location.latitude},
|
||||||
|
{"longitude", location.longitude},
|
||||||
|
}}});
|
||||||
}
|
}
|
||||||
|
|
||||||
|
std::ostringstream oss;
|
||||||
|
PrettyPrint(oss, output);
|
||||||
logger_->Log({.level = LogLevel::Info,
|
logger_->Log({.level = LogLevel::Info,
|
||||||
.phase = PipelinePhase::Teardown,
|
.phase = PipelinePhase::Teardown,
|
||||||
.message = msg.str()});
|
.message = oss.str()});
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -4,6 +4,7 @@
|
|||||||
*/
|
*/
|
||||||
|
|
||||||
#include <chrono>
|
#include <chrono>
|
||||||
|
#include <format>
|
||||||
#include <utility>
|
#include <utility>
|
||||||
|
|
||||||
#include "biergarten_pipeline_orchestrator.h"
|
#include "biergarten_pipeline_orchestrator.h"
|
||||||
@@ -33,18 +34,18 @@ bool BiergartenPipelineOrchestrator::Run() {
|
|||||||
logger_->Log(
|
logger_->Log(
|
||||||
{.level = LogLevel::Warn,
|
{.level = LogLevel::Warn,
|
||||||
.phase = PipelinePhase::UserGeneration,
|
.phase = PipelinePhase::UserGeneration,
|
||||||
.message = std::string("[Pipeline] Skipping city '") + city.city +
|
.message = std::format(
|
||||||
" (" + city.country +
|
"[Pipeline] Skipping city '{}' ({}): context lookup failed: {}",
|
||||||
"): context lookup failed: " + exception.what()});
|
city.city, city.country, exception.what())});
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if (skipped_count > 0) {
|
if (skipped_count > 0) {
|
||||||
logger_->Log({.level = LogLevel::Warn,
|
logger_->Log({.level = LogLevel::Warn,
|
||||||
.phase = PipelinePhase::UserGeneration,
|
.phase = PipelinePhase::UserGeneration,
|
||||||
.message = std::string("[Pipeline] Skipped ") +
|
.message = std::format(
|
||||||
std::to_string(skipped_count) +
|
"[Pipeline] Skipped {} city/cities due to context lookup errors",
|
||||||
" city/cities due to context lookup errors"});
|
skipped_count)});
|
||||||
}
|
}
|
||||||
|
|
||||||
this->GenerateBreweries(enriched);
|
this->GenerateBreweries(enriched);
|
||||||
@@ -56,7 +57,7 @@ bool BiergartenPipelineOrchestrator::Run() {
|
|||||||
{.level = LogLevel::Error,
|
{.level = LogLevel::Error,
|
||||||
.phase = PipelinePhase::Teardown,
|
.phase = PipelinePhase::Teardown,
|
||||||
.message =
|
.message =
|
||||||
std::string("Pipeline execution failed with error: ") + e.what()});
|
std::format("Pipeline execution failed with error: {}", e.what())});
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -103,8 +103,8 @@ BreweryResult LlamaGenerator::GenerateBrewery(
|
|||||||
logger_->Log(
|
logger_->Log(
|
||||||
{.level = LogLevel::Debug,
|
{.level = LogLevel::Debug,
|
||||||
.phase = PipelinePhase::BreweryAndBeerGeneration,
|
.phase = PipelinePhase::BreweryAndBeerGeneration,
|
||||||
.message = std::string("LlamaGenerator: raw output (attempt ") +
|
.message = std::format("LlamaGenerator: raw output (attempt {}): {}",
|
||||||
std::to_string(attempt + 1) + "): " + raw});
|
attempt + 1, raw)});
|
||||||
}
|
}
|
||||||
|
|
||||||
// Validate output: parse JSON and check required fields
|
// Validate output: parse JSON and check required fields
|
||||||
@@ -120,9 +120,8 @@ BreweryResult LlamaGenerator::GenerateBrewery(
|
|||||||
logger_->Log(
|
logger_->Log(
|
||||||
{.level = LogLevel::Info,
|
{.level = LogLevel::Info,
|
||||||
.phase = PipelinePhase::BreweryAndBeerGeneration,
|
.phase = PipelinePhase::BreweryAndBeerGeneration,
|
||||||
.message = std::string("LlamaGenerator: successfully generated "
|
.message = std::format("LlamaGenerator: successfully generated brewery data on attempt {}",
|
||||||
"brewery data on attempt ") +
|
attempt + 1)});
|
||||||
std::to_string(attempt + 1)});
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return brewery;
|
return brewery;
|
||||||
@@ -136,8 +135,8 @@ BreweryResult LlamaGenerator::GenerateBrewery(
|
|||||||
{.level = LogLevel::Warn,
|
{.level = LogLevel::Warn,
|
||||||
.phase = PipelinePhase::BreweryAndBeerGeneration,
|
.phase = PipelinePhase::BreweryAndBeerGeneration,
|
||||||
.message =
|
.message =
|
||||||
std::string("LlamaGenerator: malformed brewery JSON (attempt ") +
|
std::format("LlamaGenerator: malformed brewery JSON (attempt {}): {}",
|
||||||
std::to_string(attempt + 1) + "): " + *validation_error});
|
attempt + 1, *validation_error)});
|
||||||
}
|
}
|
||||||
|
|
||||||
// Update prompt with error details to guide LLM toward correct output.
|
// Update prompt with error details to guide LLM toward correct output.
|
||||||
@@ -159,10 +158,9 @@ BreweryResult LlamaGenerator::GenerateBrewery(
|
|||||||
logger_->Log(
|
logger_->Log(
|
||||||
{.level = LogLevel::Error,
|
{.level = LogLevel::Error,
|
||||||
.phase = PipelinePhase::BreweryAndBeerGeneration,
|
.phase = PipelinePhase::BreweryAndBeerGeneration,
|
||||||
.message =
|
.message = std::format(
|
||||||
std::string("LlamaGenerator: malformed brewery response after ") +
|
"LlamaGenerator: malformed brewery response after {} attempts: {}",
|
||||||
std::to_string(max_attempts) +
|
max_attempts, last_error.empty() ? raw : last_error)});
|
||||||
" attempts: " + (last_error.empty() ? raw : last_error)});
|
|
||||||
}
|
}
|
||||||
throw std::runtime_error("LlamaGenerator: malformed brewery response");
|
throw std::runtime_error("LlamaGenerator: malformed brewery response");
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -5,6 +5,7 @@
|
|||||||
*/
|
*/
|
||||||
|
|
||||||
|
|
||||||
|
#include <format>
|
||||||
#include <string>
|
#include <string>
|
||||||
|
|
||||||
#include "data_generation/llama_generator.h"
|
#include "data_generation/llama_generator.h"
|
||||||
@@ -19,5 +20,5 @@
|
|||||||
// 4. Return locale-aware username and biography
|
// 4. Return locale-aware username and biography
|
||||||
UserResult LlamaGenerator::GenerateUser(const std::string& locale) {
|
UserResult LlamaGenerator::GenerateUser(const std::string& locale) {
|
||||||
return {.username = "test_user",
|
return {.username = "test_user",
|
||||||
.bio = "This is a test user profile from " + locale + "."};
|
.bio = std::format("This is a test user profile from {}.", locale)};
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -16,11 +16,11 @@
|
|||||||
|
|
||||||
#include "data_generation/llama_generator_helpers.h"
|
#include "data_generation/llama_generator_helpers.h"
|
||||||
#include "llama.h"
|
#include "llama.h"
|
||||||
|
namespace {
|
||||||
/**
|
/**
|
||||||
* String trimming: removes leading and trailing whitespace
|
* String trimming: removes leading and trailing whitespace
|
||||||
*/
|
*/
|
||||||
static std::string Trim(std::string_view value) {
|
std::string Trim(std::string_view value) {
|
||||||
constexpr std::string_view whitespace = " \t\n\r\f\v";
|
constexpr std::string_view whitespace = " \t\n\r\f\v";
|
||||||
const size_t first_index = value.find_first_not_of(whitespace);
|
const size_t first_index = value.find_first_not_of(whitespace);
|
||||||
if (first_index == std::string_view::npos) {
|
if (first_index == std::string_view::npos) {
|
||||||
@@ -35,7 +35,7 @@ static std::string Trim(std::string_view value) {
|
|||||||
* Normalize whitespace: collapses multiple spaces/tabs/newlines into single
|
* Normalize whitespace: collapses multiple spaces/tabs/newlines into single
|
||||||
* spaces
|
* spaces
|
||||||
*/
|
*/
|
||||||
static std::string CondenseWhitespace(std::string_view text) {
|
std::string CondenseWhitespace(std::string_view text) {
|
||||||
std::string out;
|
std::string out;
|
||||||
out.reserve(text.size());
|
out.reserve(text.size());
|
||||||
|
|
||||||
@@ -61,7 +61,37 @@ static std::string CondenseWhitespace(std::string_view text) {
|
|||||||
// Guard against truncating in the first half of the string.
|
// Guard against truncating in the first half of the string.
|
||||||
// This preserves the critical opening content and avoids cutting critical
|
// This preserves the critical opening content and avoids cutting critical
|
||||||
// context words early in the region description.
|
// context words early in the region description.
|
||||||
static constexpr size_t kTruncationGuardDivisor = 2;
|
constexpr size_t kTruncationGuardDivisor = 2;
|
||||||
|
|
||||||
|
bool ReadRequiredTrimmedStringField(const boost::json::object& obj,
|
||||||
|
std::string_view key, std::string& out,
|
||||||
|
std::string* error_out) {
|
||||||
|
const boost::json::value* field = obj.if_contains(key);
|
||||||
|
if (field == nullptr || !field->is_string()) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
const auto& string_value = field->as_string();
|
||||||
|
out = Trim(std::string_view(string_value.data(), string_value.size()));
|
||||||
|
return !out.empty();
|
||||||
|
}
|
||||||
|
|
||||||
|
bool HasSchemaPlaceholder(const std::array<std::string*, 4>& values) {
|
||||||
|
for (const std::string* value : values) {
|
||||||
|
std::string lowered = *value;
|
||||||
|
std::ranges::transform(lowered, lowered.begin(),
|
||||||
|
[](const unsigned char character) {
|
||||||
|
return static_cast<char>(std::tolower(character));
|
||||||
|
});
|
||||||
|
|
||||||
|
if (lowered == "string") {
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
} // namespace
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Truncate region context to fit within max length while preserving word
|
* Truncate region context to fit within max length while preserving word
|
||||||
@@ -121,47 +151,6 @@ void AppendTokenPiece(const llama_vocab* vocab, llama_token token,
|
|||||||
"LlamaGenerator: failed to decode sampled token piece");
|
"LlamaGenerator: failed to decode sampled token piece");
|
||||||
}
|
}
|
||||||
|
|
||||||
static bool ReadRequiredTrimmedStringField(const boost::json::object& obj,
|
|
||||||
std::string_view key,
|
|
||||||
std::string& out,
|
|
||||||
std::string* error_out) {
|
|
||||||
const boost::json::value* field = obj.if_contains(key);
|
|
||||||
if (field == nullptr || !field->is_string()) {
|
|
||||||
if (error_out != nullptr) {
|
|
||||||
*error_out =
|
|
||||||
"JSON field '" + std::string(key) + "' is missing or not a string";
|
|
||||||
}
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
const auto& string_value = field->as_string();
|
|
||||||
out = Trim(std::string_view(string_value.data(), string_value.size()));
|
|
||||||
if (out.empty()) {
|
|
||||||
if (error_out != nullptr) {
|
|
||||||
*error_out = "JSON field '" + std::string(key) + "' must not be empty";
|
|
||||||
}
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
static bool HasSchemaPlaceholder(const std::array<std::string*, 4>& values) {
|
|
||||||
for (const std::string* value : values) {
|
|
||||||
std::string lowered = *value;
|
|
||||||
std::ranges::transform(lowered, lowered.begin(),
|
|
||||||
[](unsigned char character) {
|
|
||||||
return static_cast<char>(std::tolower(character));
|
|
||||||
});
|
|
||||||
|
|
||||||
if (lowered == "string") {
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
std::optional<std::string> ValidateBreweryJson(const std::string& raw,
|
std::optional<std::string> ValidateBreweryJson(const std::string& raw,
|
||||||
BreweryResult& brewery_out) {
|
BreweryResult& brewery_out) {
|
||||||
boost::system::error_code error_code;
|
boost::system::error_code error_code;
|
||||||
@@ -209,7 +198,7 @@ std::optional<std::string> ValidateBreweryJson(const std::string& raw,
|
|||||||
return validation_error;
|
return validation_error;
|
||||||
}
|
}
|
||||||
|
|
||||||
const std::array<std::string*, 4> schema_placeholders = {
|
const std::array schema_placeholders = {
|
||||||
&brewery_out.name_en, &brewery_out.description_en,
|
&brewery_out.name_en, &brewery_out.description_en,
|
||||||
&brewery_out.name_local, &brewery_out.description_local};
|
&brewery_out.name_local, &brewery_out.description_local};
|
||||||
if (HasSchemaPlaceholder(schema_placeholders)) {
|
if (HasSchemaPlaceholder(schema_placeholders)) {
|
||||||
|
|||||||
@@ -7,6 +7,7 @@
|
|||||||
|
|
||||||
#include <algorithm>
|
#include <algorithm>
|
||||||
#include <chrono>
|
#include <chrono>
|
||||||
|
#include <format>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include <stdexcept>
|
#include <stdexcept>
|
||||||
#include <string>
|
#include <string>
|
||||||
@@ -106,7 +107,7 @@ std::string LlamaGenerator::InferFormatted(const std::string& formatted_prompt,
|
|||||||
.top_p = sampling_top_p_,
|
.top_p = sampling_top_p_,
|
||||||
.seed = static_cast<uint32_t>(rng_()),
|
.seed = static_cast<uint32_t>(rng_()),
|
||||||
};
|
};
|
||||||
auto sampler = MakeSamplerChain(vocab, sampler_config, grammar);
|
const auto sampler = MakeSamplerChain(vocab, sampler_config, grammar);
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Clear KV cache to ensure clean inference state (no residual context)
|
* Clear KV cache to ensure clean inference state (no residual context)
|
||||||
@@ -171,13 +172,12 @@ std::string LlamaGenerator::InferFormatted(const std::string& formatted_prompt,
|
|||||||
prompt_tokens.resize(static_cast<size_t>(token_count));
|
prompt_tokens.resize(static_cast<size_t>(token_count));
|
||||||
if (token_count > prompt_budget) {
|
if (token_count > prompt_budget) {
|
||||||
if (logger_) {
|
if (logger_) {
|
||||||
logger_->Log(
|
logger_->Log({.level = LogLevel::Warn,
|
||||||
{.level = LogLevel::Warn,
|
.phase = PipelinePhase::BreweryAndBeerGeneration,
|
||||||
.phase = PipelinePhase::BreweryAndBeerGeneration,
|
.message = std::format(
|
||||||
.message = std::string("LlamaGenerator: prompt too long (") +
|
"LlamaGenerator: prompt too long ({} tokens), "
|
||||||
std::to_string(token_count) + ") tokens, truncating to " +
|
"truncating to {} tokens to fit n_batch/n_ctx limits",
|
||||||
std::to_string(prompt_budget) +
|
token_count, prompt_budget)});
|
||||||
" tokens to fit n_batch/n_ctx limits"});
|
|
||||||
}
|
}
|
||||||
prompt_tokens.resize(static_cast<size_t>(prompt_budget));
|
prompt_tokens.resize(static_cast<size_t>(prompt_budget));
|
||||||
token_count = prompt_budget;
|
token_count = prompt_budget;
|
||||||
|
|||||||
@@ -28,8 +28,10 @@ void LlamaGenerator::Load(const std::string& model_path) {
|
|||||||
|
|
||||||
llama_model_params model_params = llama_model_default_params();
|
llama_model_params model_params = llama_model_default_params();
|
||||||
model_params.n_gpu_layers = n_gpu_layers_;
|
model_params.n_gpu_layers = n_gpu_layers_;
|
||||||
LlamaGenerator::ModelHandle loaded_model(
|
|
||||||
|
ModelHandle loaded_model(
|
||||||
llama_model_load_from_file(model_path.c_str(), model_params));
|
llama_model_load_from_file(model_path.c_str(), model_params));
|
||||||
|
|
||||||
if (!loaded_model) {
|
if (!loaded_model) {
|
||||||
throw std::runtime_error(
|
throw std::runtime_error(
|
||||||
"LlamaGenerator: failed to load model from path: " + model_path);
|
"LlamaGenerator: failed to load model from path: " + model_path);
|
||||||
@@ -39,8 +41,9 @@ void LlamaGenerator::Load(const std::string& model_path) {
|
|||||||
context_params.n_ctx = n_ctx_;
|
context_params.n_ctx = n_ctx_;
|
||||||
context_params.n_batch = std::min(n_ctx_, kMaxBatchSize);
|
context_params.n_batch = std::min(n_ctx_, kMaxBatchSize);
|
||||||
|
|
||||||
LlamaGenerator::ContextHandle loaded_context(
|
ContextHandle loaded_context(
|
||||||
llama_init_from_model(loaded_model.get(), context_params));
|
llama_init_from_model(loaded_model.get(), context_params));
|
||||||
|
|
||||||
if (!loaded_context) {
|
if (!loaded_context) {
|
||||||
throw std::runtime_error("LlamaGenerator: failed to create context");
|
throw std::runtime_error("LlamaGenerator: failed to create context");
|
||||||
}
|
}
|
||||||
@@ -51,7 +54,7 @@ void LlamaGenerator::Load(const std::string& model_path) {
|
|||||||
if (logger_) {
|
if (logger_) {
|
||||||
logger_->Log({.level = LogLevel::Info,
|
logger_->Log({.level = LogLevel::Info,
|
||||||
.phase = PipelinePhase::Startup,
|
.phase = PipelinePhase::Startup,
|
||||||
.message = std::string("[LlamaGenerator] Loaded model: ") +
|
.message = std::format("[LlamaGenerator] Loaded model: {} ",
|
||||||
model_path});
|
model_path)});
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -6,6 +6,7 @@
|
|||||||
|
|
||||||
#include "json_handling/json_loader.h"
|
#include "json_handling/json_loader.h"
|
||||||
|
|
||||||
|
#include <format>
|
||||||
#include "services/logging/logger.h"
|
#include "services/logging/logger.h"
|
||||||
#include <iostream>
|
#include <iostream>
|
||||||
|
|
||||||
@@ -20,8 +21,8 @@ static std::string ReadRequiredString(const boost::json::object& object,
|
|||||||
const char* key) {
|
const char* key) {
|
||||||
const boost::json::value* value = object.if_contains(key);
|
const boost::json::value* value = object.if_contains(key);
|
||||||
if (value == nullptr || !value->is_string()) {
|
if (value == nullptr || !value->is_string()) {
|
||||||
throw std::runtime_error(std::string("Missing or invalid string field: ") +
|
throw std::runtime_error(
|
||||||
key);
|
std::format("Missing or invalid string field: {}", key));
|
||||||
}
|
}
|
||||||
const std::string_view text = value->as_string();
|
const std::string_view text = value->as_string();
|
||||||
return std::string(text);
|
return std::string(text);
|
||||||
@@ -31,8 +32,8 @@ static double ReadRequiredNumber(const boost::json::object& object,
|
|||||||
const char* key) {
|
const char* key) {
|
||||||
const boost::json::value* value = object.if_contains(key);
|
const boost::json::value* value = object.if_contains(key);
|
||||||
if (value == nullptr || !value->is_number()) {
|
if (value == nullptr || !value->is_number()) {
|
||||||
throw std::runtime_error(std::string("Missing or invalid numeric field: ") +
|
throw std::runtime_error(
|
||||||
key);
|
std::format("Missing or invalid numeric field: {}", key));
|
||||||
}
|
}
|
||||||
return value->to_number<double>();
|
return value->to_number<double>();
|
||||||
}
|
}
|
||||||
@@ -42,7 +43,7 @@ static std::vector<std::string> ReadRequiredStringArray(
|
|||||||
const boost::json::value* value = object.if_contains(key);
|
const boost::json::value* value = object.if_contains(key);
|
||||||
if (value == nullptr || !value->is_array()) {
|
if (value == nullptr || !value->is_array()) {
|
||||||
throw std::runtime_error(
|
throw std::runtime_error(
|
||||||
std::string("Missing or invalid string array field: ") + key);
|
std::format("Missing or invalid string array field: {}", key));
|
||||||
}
|
}
|
||||||
|
|
||||||
const auto& array = value->as_array();
|
const auto& array = value->as_array();
|
||||||
@@ -51,7 +52,7 @@ static std::vector<std::string> ReadRequiredStringArray(
|
|||||||
for (const auto& item : array) {
|
for (const auto& item : array) {
|
||||||
if (!item.is_string()) {
|
if (!item.is_string()) {
|
||||||
throw std::runtime_error(
|
throw std::runtime_error(
|
||||||
std::string("Missing or invalid string array field: ") + key);
|
std::format("Missing or invalid string array field: {}", key));
|
||||||
}
|
}
|
||||||
items.emplace_back(item.as_string());
|
items.emplace_back(item.as_string());
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -12,6 +12,7 @@
|
|||||||
#include <chrono>
|
#include <chrono>
|
||||||
#include <exception>
|
#include <exception>
|
||||||
#include <format>
|
#include <format>
|
||||||
|
#include <iostream>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include <optional>
|
#include <optional>
|
||||||
#include <string>
|
#include <string>
|
||||||
@@ -65,15 +66,15 @@ int main(const int argc, char** argv) {
|
|||||||
const LlamaBackendState llama_backend_state;
|
const LlamaBackendState llama_backend_state;
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
log_producer->Log({.level = LogLevel::Info,
|
log_producer->Log({.level = LogLevel::Info,
|
||||||
.phase = PipelinePhase::Startup,
|
.phase = PipelinePhase::Startup,
|
||||||
.message = "STARTING PIPELINE"});
|
.message = "STARTING PIPELINE"});
|
||||||
|
|
||||||
const std::optional<ApplicationOptions> parsed_options =
|
const std::optional<ApplicationOptions> parsed_options =
|
||||||
ParseArguments(argc, argv, log_producer);
|
ParseArguments(argc, argv, log_producer);
|
||||||
|
|
||||||
if (!parsed_options.has_value()) {
|
if (!parsed_options.has_value()) {
|
||||||
return shutdown(0);
|
return shutdown(EXIT_FAILURE);
|
||||||
}
|
}
|
||||||
|
|
||||||
const auto options = *parsed_options;
|
const auto options = *parsed_options;
|
||||||
@@ -88,12 +89,12 @@ int main(const int argc, char** argv) {
|
|||||||
prompt_directory = std::make_unique<PromptDirectory>(
|
prompt_directory = std::make_unique<PromptDirectory>(
|
||||||
options.pipeline.prompt_dir, log_producer);
|
options.pipeline.prompt_dir, log_producer);
|
||||||
} catch (const std::exception& dir_error) {
|
} catch (const std::exception& dir_error) {
|
||||||
log_producer->Log({.level = LogLevel::Error,
|
log_producer->Log({.level = LogLevel::Error,
|
||||||
.phase = PipelinePhase::Startup,
|
.phase = PipelinePhase::Startup,
|
||||||
.message = std::format("Invalid --prompt-dir: {}",
|
.message = std::format("Invalid --prompt-dir: {}",
|
||||||
dir_error.what())});
|
dir_error.what())});
|
||||||
|
|
||||||
return shutdown(1);
|
return shutdown(EXIT_FAILURE);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -107,7 +108,7 @@ int main(const int argc, char** argv) {
|
|||||||
{
|
{
|
||||||
log_producer->Log(
|
log_producer->Log(
|
||||||
{.level = LogLevel::Info,
|
{.level = LogLevel::Info,
|
||||||
.phase = PipelinePhase::Startup,
|
.phase = PipelinePhase::Startup,
|
||||||
.message = "Prompt formatter: none (mock mode)"});
|
.message = "Prompt formatter: none (mock mode)"});
|
||||||
}
|
}
|
||||||
return std::unique_ptr<IPromptFormatter>(nullptr);
|
return std::unique_ptr<IPromptFormatter>(nullptr);
|
||||||
@@ -115,7 +116,7 @@ int main(const int argc, char** argv) {
|
|||||||
{
|
{
|
||||||
log_producer->Log(
|
log_producer->Log(
|
||||||
{.level = LogLevel::Info,
|
{.level = LogLevel::Info,
|
||||||
.phase = PipelinePhase::Startup,
|
.phase = PipelinePhase::Startup,
|
||||||
.message = "Prompt formatter: Gemma4JinjaPromptFormatter"});
|
.message = "Prompt formatter: Gemma4JinjaPromptFormatter"});
|
||||||
}
|
}
|
||||||
return std::unique_ptr<IPromptFormatter>(
|
return std::unique_ptr<IPromptFormatter>(
|
||||||
@@ -124,15 +125,15 @@ int main(const int argc, char** argv) {
|
|||||||
di::bind<WebClient>().to([options, log_producer] {
|
di::bind<WebClient>().to([options, log_producer] {
|
||||||
if (options.generator.use_mocked) {
|
if (options.generator.use_mocked) {
|
||||||
{
|
{
|
||||||
log_producer->Log({.level = LogLevel::Info,
|
log_producer->Log({.level = LogLevel::Info,
|
||||||
.phase = PipelinePhase::Startup,
|
.phase = PipelinePhase::Startup,
|
||||||
.message = "Web client: none (mock mode)"});
|
.message = "Web client: none (mock mode)"});
|
||||||
}
|
}
|
||||||
return std::unique_ptr<WebClient>(nullptr);
|
return std::unique_ptr<WebClient>(nullptr);
|
||||||
}
|
}
|
||||||
{
|
{
|
||||||
log_producer->Log({.level = LogLevel::Info,
|
log_producer->Log({.level = LogLevel::Info,
|
||||||
.phase = PipelinePhase::Startup,
|
.phase = PipelinePhase::Startup,
|
||||||
.message = "Web client: HttpWebClient"});
|
.message = "Web client: HttpWebClient"});
|
||||||
}
|
}
|
||||||
return std::unique_ptr<WebClient>(
|
return std::unique_ptr<WebClient>(
|
||||||
@@ -143,15 +144,15 @@ int main(const int argc, char** argv) {
|
|||||||
const auto& inj) -> std::unique_ptr<IEnrichmentService> {
|
const auto& inj) -> std::unique_ptr<IEnrichmentService> {
|
||||||
if (options.generator.use_mocked) {
|
if (options.generator.use_mocked) {
|
||||||
{
|
{
|
||||||
log_producer->Log({.level = LogLevel::Info,
|
log_producer->Log({.level = LogLevel::Info,
|
||||||
.phase = PipelinePhase::Startup,
|
.phase = PipelinePhase::Startup,
|
||||||
.message = "Enrichment: mock"});
|
.message = "Enrichment: mock"});
|
||||||
}
|
}
|
||||||
return std::make_unique<MockEnrichmentService>();
|
return std::make_unique<MockEnrichmentService>();
|
||||||
}
|
}
|
||||||
{
|
{
|
||||||
log_producer->Log({.level = LogLevel::Info,
|
log_producer->Log({.level = LogLevel::Info,
|
||||||
.phase = PipelinePhase::Startup,
|
.phase = PipelinePhase::Startup,
|
||||||
.message = "Enrichment: Wikipedia"});
|
.message = "Enrichment: Wikipedia"});
|
||||||
}
|
}
|
||||||
return std::make_unique<WikipediaEnrichmentService>(
|
return std::make_unique<WikipediaEnrichmentService>(
|
||||||
@@ -163,8 +164,8 @@ int main(const int argc, char** argv) {
|
|||||||
&log_producer](const auto& inj) -> std::unique_ptr<DataGenerator> {
|
&log_producer](const auto& inj) -> std::unique_ptr<DataGenerator> {
|
||||||
if (options.generator.use_mocked) {
|
if (options.generator.use_mocked) {
|
||||||
{
|
{
|
||||||
log_producer->Log({.level = LogLevel::Info,
|
log_producer->Log({.level = LogLevel::Info,
|
||||||
.phase = PipelinePhase::Startup,
|
.phase = PipelinePhase::Startup,
|
||||||
.message = "Generator: mock"});
|
.message = "Generator: mock"});
|
||||||
}
|
}
|
||||||
return std::make_unique<MockGenerator>();
|
return std::make_unique<MockGenerator>();
|
||||||
@@ -172,8 +173,8 @@ int main(const int argc, char** argv) {
|
|||||||
{
|
{
|
||||||
log_producer->Log(
|
log_producer->Log(
|
||||||
{.level = LogLevel::Info,
|
{.level = LogLevel::Info,
|
||||||
.phase = PipelinePhase::Startup,
|
.phase = PipelinePhase::Startup,
|
||||||
.message = fmt::format(
|
.message = std::format(
|
||||||
"Generator: LlamaGenerator | model={} | temp={:.2f} "
|
"Generator: LlamaGenerator | model={} | temp={:.2f} "
|
||||||
"top_p={:.2f} top_k={} n_ctx={} seed={}",
|
"top_p={:.2f} top_k={} n_ctx={} seed={}",
|
||||||
model_path, sampling.temperature, sampling.top_p,
|
model_path, sampling.temperature, sampling.top_p,
|
||||||
@@ -189,26 +190,29 @@ int main(const int argc, char** argv) {
|
|||||||
injector.create<std::unique_ptr<BiergartenPipelineOrchestrator>>();
|
injector.create<std::unique_ptr<BiergartenPipelineOrchestrator>>();
|
||||||
|
|
||||||
if (!orchestrator->Run()) {
|
if (!orchestrator->Run()) {
|
||||||
log_producer->Log({.level = LogLevel::Error,
|
log_producer->Log({.level = LogLevel::Error,
|
||||||
.phase = PipelinePhase::Teardown,
|
.phase = PipelinePhase::Teardown,
|
||||||
.message = "Pipeline execution failed"});
|
.message = "Pipeline execution failed"});
|
||||||
return shutdown(1);
|
return shutdown(EXIT_FAILURE);
|
||||||
}
|
}
|
||||||
|
|
||||||
log_producer->Log({.level = LogLevel::Info,
|
log_producer->Log({.level = LogLevel::Info,
|
||||||
.phase = PipelinePhase::Teardown,
|
.phase = PipelinePhase::Teardown,
|
||||||
.message = fmt::format("Pipeline complete in {} ms",
|
.message = std::format("Pipeline complete in {} ms",
|
||||||
timer.Elapsed())});
|
timer.Elapsed())});
|
||||||
|
|
||||||
return shutdown(0);
|
return shutdown(EXIT_SUCCESS);
|
||||||
|
|
||||||
} catch (const std::exception& exception) {
|
} catch (const std::exception& exception) {
|
||||||
|
const LogDTO log_entry{.level = LogLevel::Error,
|
||||||
|
.phase = PipelinePhase::Teardown,
|
||||||
|
.message = exception.what()};
|
||||||
if (log_producer) {
|
if (log_producer) {
|
||||||
log_producer->Log({.level = LogLevel::Error,
|
log_producer->Log(log_entry);
|
||||||
.phase = PipelinePhase::Teardown,
|
} else {
|
||||||
.message = fmt::format("Unhandled fatal error: {}",
|
std::cerr << log_entry.message << std::endl;
|
||||||
exception.what())});
|
|
||||||
}
|
}
|
||||||
return shutdown(1);
|
|
||||||
|
return shutdown(EXIT_FAILURE);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -22,8 +22,7 @@ std::string WikipediaEnrichmentService::FetchExtract(std::string_view query) {
|
|||||||
if (logger_) {
|
if (logger_) {
|
||||||
logger_->Log({.level = LogLevel::Debug,
|
logger_->Log({.level = LogLevel::Debug,
|
||||||
.phase = PipelinePhase::UserGeneration,
|
.phase = PipelinePhase::UserGeneration,
|
||||||
.message = std::string("Wikipedia: Cache hit for ") +
|
.message = std::format("Wikipedia: Cache hit for {}!", cache_key)});
|
||||||
cache_key + "!"});
|
|
||||||
}
|
}
|
||||||
return cache_it->second;
|
return cache_it->second;
|
||||||
}
|
}
|
||||||
@@ -49,8 +48,8 @@ std::string WikipediaEnrichmentService::FetchExtract(std::string_view query) {
|
|||||||
logger_->Log(
|
logger_->Log(
|
||||||
{.level = LogLevel::Warn,
|
{.level = LogLevel::Warn,
|
||||||
.phase = PipelinePhase::UserGeneration,
|
.phase = PipelinePhase::UserGeneration,
|
||||||
.message = std::string("WikipediaService: JSON parse error for '") +
|
.message = std::format("WikipediaService: JSON parse error for '{}': {}",
|
||||||
std::string(query) + "': " + ec.message()});
|
std::string(query), ec.message())});
|
||||||
}
|
}
|
||||||
return {};
|
return {};
|
||||||
}
|
}
|
||||||
@@ -63,8 +62,8 @@ std::string WikipediaEnrichmentService::FetchExtract(std::string_view query) {
|
|||||||
{.level = LogLevel::Warn,
|
{.level = LogLevel::Warn,
|
||||||
.phase = PipelinePhase::UserGeneration,
|
.phase = PipelinePhase::UserGeneration,
|
||||||
.message =
|
.message =
|
||||||
std::string("WikipediaService: Expected root object for '") +
|
std::format("WikipediaService: Expected root object for '{}'",
|
||||||
std::string(query) + "'"});
|
std::string(query))});
|
||||||
}
|
}
|
||||||
return {};
|
return {};
|
||||||
}
|
}
|
||||||
@@ -81,8 +80,8 @@ std::string WikipediaEnrichmentService::FetchExtract(std::string_view query) {
|
|||||||
{.level = LogLevel::Warn,
|
{.level = LogLevel::Warn,
|
||||||
.phase = PipelinePhase::UserGeneration,
|
.phase = PipelinePhase::UserGeneration,
|
||||||
.message =
|
.message =
|
||||||
std::string("WikipediaService: Missing query.pages for '") +
|
std::format("WikipediaService: Missing query.pages for '{}'",
|
||||||
std::string(query) + "'"});
|
std::string(query))});
|
||||||
}
|
}
|
||||||
return {};
|
return {};
|
||||||
}
|
}
|
||||||
@@ -94,8 +93,8 @@ std::string WikipediaEnrichmentService::FetchExtract(std::string_view query) {
|
|||||||
logger_->Log(
|
logger_->Log(
|
||||||
{.level = LogLevel::Warn,
|
{.level = LogLevel::Warn,
|
||||||
.phase = PipelinePhase::UserGeneration,
|
.phase = PipelinePhase::UserGeneration,
|
||||||
.message = std::string("WikipediaService: No pages returned for '") +
|
.message = std::format("WikipediaService: No pages returned for '{}'",
|
||||||
std::string(query) + "'"});
|
std::string(query))});
|
||||||
}
|
}
|
||||||
this->extract_cache_.emplace(cache_key, "");
|
this->extract_cache_.emplace(cache_key, "");
|
||||||
return {};
|
return {};
|
||||||
@@ -111,8 +110,8 @@ std::string WikipediaEnrichmentService::FetchExtract(std::string_view query) {
|
|||||||
{.level = LogLevel::Warn,
|
{.level = LogLevel::Warn,
|
||||||
.phase = PipelinePhase::UserGeneration,
|
.phase = PipelinePhase::UserGeneration,
|
||||||
.message =
|
.message =
|
||||||
std::string("WikipediaService: Unexpected page format for '") +
|
std::format("WikipediaService: Unexpected page format for '{}'",
|
||||||
std::string(query) + "'"});
|
std::string(query))});
|
||||||
}
|
}
|
||||||
return {};
|
return {};
|
||||||
}
|
}
|
||||||
@@ -124,8 +123,8 @@ std::string WikipediaEnrichmentService::FetchExtract(std::string_view query) {
|
|||||||
if (logger_) {
|
if (logger_) {
|
||||||
logger_->Log({.level = LogLevel::Warn,
|
logger_->Log({.level = LogLevel::Warn,
|
||||||
.phase = PipelinePhase::UserGeneration,
|
.phase = PipelinePhase::UserGeneration,
|
||||||
.message = std::string("WikipediaService: Page '") +
|
.message = std::format("WikipediaService: Page '{}' does not exist",
|
||||||
std::string(query) + "' does not exist"});
|
std::string(query))});
|
||||||
}
|
}
|
||||||
this->extract_cache_.emplace(cache_key, "");
|
this->extract_cache_.emplace(cache_key, "");
|
||||||
return {};
|
return {};
|
||||||
@@ -139,8 +138,8 @@ std::string WikipediaEnrichmentService::FetchExtract(std::string_view query) {
|
|||||||
{.level = LogLevel::Warn,
|
{.level = LogLevel::Warn,
|
||||||
.phase = PipelinePhase::UserGeneration,
|
.phase = PipelinePhase::UserGeneration,
|
||||||
.message =
|
.message =
|
||||||
std::string("WikipediaService: No extract string found for '") +
|
std::format("WikipediaService: No extract string found for '{}'",
|
||||||
std::string(query) + "'"});
|
std::string(query))});
|
||||||
}
|
}
|
||||||
this->extract_cache_.emplace(cache_key, "");
|
this->extract_cache_.emplace(cache_key, "");
|
||||||
return {};
|
return {};
|
||||||
@@ -151,9 +150,8 @@ std::string WikipediaEnrichmentService::FetchExtract(std::string_view query) {
|
|||||||
if (logger_) {
|
if (logger_) {
|
||||||
logger_->Log({.level = LogLevel::Info,
|
logger_->Log({.level = LogLevel::Info,
|
||||||
.phase = PipelinePhase::UserGeneration,
|
.phase = PipelinePhase::UserGeneration,
|
||||||
.message = std::string("WikipediaService: Fetched ") +
|
.message = std::format("WikipediaService: Fetched {} chars for '{}'",
|
||||||
std::to_string(extract.size()) + " chars for '" +
|
extract.size(), std::string(query))});
|
||||||
std::string(query) + "'"});
|
|
||||||
}
|
}
|
||||||
|
|
||||||
this->extract_cache_.insert_or_assign(cache_key, extract);
|
this->extract_cache_.insert_or_assign(cache_key, extract);
|
||||||
|
|||||||
@@ -52,8 +52,8 @@ std::string WikipediaEnrichmentService::GetLocationContext(
|
|||||||
if (logger_) {
|
if (logger_) {
|
||||||
logger_->Log({.level = LogLevel::Info,
|
logger_->Log({.level = LogLevel::Info,
|
||||||
.phase = PipelinePhase::UserGeneration,
|
.phase = PipelinePhase::UserGeneration,
|
||||||
.message = std::string("Done fetching for ") +
|
.message = std::format("Done fetching for {}. Sleeping for 10 seconds.",
|
||||||
location_query + ". Sleeping for 10 seconds."});
|
location_query)});
|
||||||
}
|
}
|
||||||
std::this_thread::sleep_for(10s);
|
std::this_thread::sleep_for(10s);
|
||||||
|
|
||||||
@@ -62,8 +62,8 @@ std::string WikipediaEnrichmentService::GetLocationContext(
|
|||||||
logger_->Log(
|
logger_->Log(
|
||||||
{.level = LogLevel::Debug,
|
{.level = LogLevel::Debug,
|
||||||
.phase = PipelinePhase::UserGeneration,
|
.phase = PipelinePhase::UserGeneration,
|
||||||
.message = std::string("WikipediaService lookup failed for '") +
|
.message = std::format("WikipediaService lookup failed for '{}': {}",
|
||||||
location_query + "': " + e.what()});
|
location_query, e.what())});
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return result;
|
return result;
|
||||||
|
|||||||
@@ -13,6 +13,29 @@
|
|||||||
#include "concurrency/bounded_channel.h"
|
#include "concurrency/bounded_channel.h"
|
||||||
#include "services/logging/log_entry.h"
|
#include "services/logging/log_entry.h"
|
||||||
|
|
||||||
|
namespace {
|
||||||
|
[[nodiscard]] constexpr std::string_view PipelinePhaseToString(
|
||||||
|
PipelinePhase phase) {
|
||||||
|
switch (phase) {
|
||||||
|
case PipelinePhase::Startup:
|
||||||
|
return "Startup";
|
||||||
|
case PipelinePhase::UserGeneration:
|
||||||
|
return "User Generation";
|
||||||
|
case PipelinePhase::BreweryAndBeerGeneration:
|
||||||
|
return "Brewery & Beer Gen";
|
||||||
|
case PipelinePhase::CheckinGeneration:
|
||||||
|
return "Checkin Gen";
|
||||||
|
case PipelinePhase::RatingGeneration:
|
||||||
|
return "Rating Gen";
|
||||||
|
case PipelinePhase::FollowGeneration:
|
||||||
|
return "Follow Gen";
|
||||||
|
case PipelinePhase::Teardown:
|
||||||
|
return "Teardown";
|
||||||
|
}
|
||||||
|
return "Unknown";
|
||||||
|
}
|
||||||
|
} // namespace
|
||||||
|
|
||||||
LogDispatcher::LogDispatcher(BoundedChannel<LogEntry>& channel)
|
LogDispatcher::LogDispatcher(BoundedChannel<LogEntry>& channel)
|
||||||
: channel_(channel) {}
|
: channel_(channel) {}
|
||||||
|
|
||||||
@@ -28,7 +51,11 @@ void LogDispatcher::Run() {
|
|||||||
|
|
||||||
const auto& log = entry.value();
|
const auto& log = entry.value();
|
||||||
|
|
||||||
logger->log(ToSpdlogLevel(log.level), log.message);
|
logger->log(ToSpdlogLevel(log.level),
|
||||||
|
"{:<20} │ thread: {:016x} │ [{}:{}] │ {}",
|
||||||
|
PipelinePhaseToString(log.phase),
|
||||||
|
std::hash<std::thread::id>{}(log.thread_id),
|
||||||
|
log.origin.file_name(), log.origin.line(), log.message);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -16,4 +16,4 @@
|
|||||||
LogProducer::LogProducer(BoundedChannel<LogEntry>& channel)
|
LogProducer::LogProducer(BoundedChannel<LogEntry>& channel)
|
||||||
: channel_(channel) {}
|
: channel_(channel) {}
|
||||||
|
|
||||||
void LogProducer::Log(const LogEntry& entry) { channel_.Send(entry); }
|
void LogProducer::DoLog(LogEntry entry) { channel_.Send(std::move(entry)); }
|
||||||
|
|||||||
@@ -8,6 +8,7 @@
|
|||||||
|
|
||||||
#include <chrono>
|
#include <chrono>
|
||||||
#include <filesystem>
|
#include <filesystem>
|
||||||
|
#include <format>
|
||||||
#include <fstream>
|
#include <fstream>
|
||||||
#include <stdexcept>
|
#include <stdexcept>
|
||||||
#include <string>
|
#include <string>
|
||||||
@@ -44,8 +45,8 @@ PromptDirectory::PromptDirectory(const std::filesystem::path& prompt_dir,
|
|||||||
std::filesystem::directory_iterator probe(prompt_dir_, ec);
|
std::filesystem::directory_iterator probe(prompt_dir_, ec);
|
||||||
if (ec) {
|
if (ec) {
|
||||||
throw std::runtime_error(
|
throw std::runtime_error(
|
||||||
"PromptDirectory: prompt directory is not readable: " +
|
std::format("PromptDirectory: prompt directory is not readable: {} ({})",
|
||||||
prompt_dir_.string() + " (" + ec.message() + ")");
|
prompt_dir_.string(), ec.message()));
|
||||||
}
|
}
|
||||||
|
|
||||||
if (logger_) {
|
if (logger_) {
|
||||||
@@ -69,13 +70,13 @@ std::string PromptDirectory::Load(std::string_view key) {
|
|||||||
|
|
||||||
// Scenario 3: resolve <prompt_dir>/<key>.md and require it to exist.
|
// Scenario 3: resolve <prompt_dir>/<key>.md and require it to exist.
|
||||||
const std::filesystem::path file_path =
|
const std::filesystem::path file_path =
|
||||||
prompt_dir_ / std::filesystem::path(key_str + ".md");
|
prompt_dir_ / std::filesystem::path(std::format("{}.md", key_str));
|
||||||
|
|
||||||
std::ifstream file(file_path);
|
std::ifstream file(file_path);
|
||||||
if (!file.is_open()) {
|
if (!file.is_open()) {
|
||||||
throw std::runtime_error(
|
throw std::runtime_error(
|
||||||
"PromptDirectory: prompt file not found for key '" + key_str +
|
std::format("PromptDirectory: prompt file not found for key '{}': {}",
|
||||||
"': " + file_path.string());
|
key_str, file_path.string()));
|
||||||
}
|
}
|
||||||
|
|
||||||
std::string content((std::istreambuf_iterator<char>(file)),
|
std::string content((std::istreambuf_iterator<char>(file)),
|
||||||
@@ -83,16 +84,15 @@ std::string PromptDirectory::Load(std::string_view key) {
|
|||||||
file.close();
|
file.close();
|
||||||
|
|
||||||
if (content.empty()) {
|
if (content.empty()) {
|
||||||
throw std::runtime_error("PromptDirectory: prompt file for key '" +
|
throw std::runtime_error(std::format("PromptDirectory: prompt file for key '{}' is empty: {}",
|
||||||
key_str + "' is empty: " + file_path.string());
|
key_str, file_path.string()));
|
||||||
}
|
}
|
||||||
|
|
||||||
if (logger_) {
|
if (logger_) {
|
||||||
logger_->Log({.level = LogLevel::Info,
|
logger_->Log({.level = LogLevel::Info,
|
||||||
.phase = PipelinePhase::Startup,
|
.phase = PipelinePhase::Startup,
|
||||||
.message = std::string("[PromptDirectory] Loaded prompt '") +
|
.message = std::format("[PromptDirectory] Loaded prompt '{}' from '{}' ({} chars)",
|
||||||
key_str + "' from '" + file_path.string() + "' (" +
|
key_str, file_path.string(), content.size())});
|
||||||
std::to_string(content.size()) + " chars)"});
|
|
||||||
}
|
}
|
||||||
|
|
||||||
cache_.emplace(key_str, content);
|
cache_.emplace(key_str, content);
|
||||||
|
|||||||
@@ -1,5 +1,6 @@
|
|||||||
#include "services/database/sqlite_connection_helpers.h"
|
#include "services/database/sqlite_connection_helpers.h"
|
||||||
|
|
||||||
|
#include <format>
|
||||||
#include <stdexcept>
|
#include <stdexcept>
|
||||||
|
|
||||||
namespace sqlite_export_service_internal {
|
namespace sqlite_export_service_internal {
|
||||||
@@ -20,7 +21,7 @@ void SqliteStatementDeleter::operator()(
|
|||||||
void ThrowSqliteError(sqlite3* db_handle, std::string_view action) {
|
void ThrowSqliteError(sqlite3* db_handle, std::string_view action) {
|
||||||
const std::string message =
|
const std::string message =
|
||||||
db_handle != nullptr ? sqlite3_errmsg(db_handle) : "unknown SQLite error";
|
db_handle != nullptr ? sqlite3_errmsg(db_handle) : "unknown SQLite error";
|
||||||
throw std::runtime_error(std::string(action) + ": " + message);
|
throw std::runtime_error(std::format("{}: {}", action, message));
|
||||||
}
|
}
|
||||||
|
|
||||||
SqliteDatabaseHandle OpenDatabase(const std::filesystem::path& path) {
|
SqliteDatabaseHandle OpenDatabase(const std::filesystem::path& path) {
|
||||||
@@ -50,7 +51,7 @@ void ExecSql(const SqliteDatabaseHandle& db_handle, std::string_view sql,
|
|||||||
? error_message
|
? error_message
|
||||||
: sqlite3_errmsg(db_handle.get());
|
: sqlite3_errmsg(db_handle.get());
|
||||||
sqlite3_free(error_message);
|
sqlite3_free(error_message);
|
||||||
throw std::runtime_error(std::string(action) + ": " + message);
|
throw std::runtime_error(std::format("{}: {}", action, message));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -4,6 +4,7 @@
|
|||||||
*/
|
*/
|
||||||
|
|
||||||
#include <filesystem>
|
#include <filesystem>
|
||||||
|
#include <format>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include <stdexcept>
|
#include <stdexcept>
|
||||||
#include <string>
|
#include <string>
|
||||||
@@ -18,8 +19,8 @@ std::filesystem::path SqliteExportService::BuildDatabasePath() const {
|
|||||||
|
|
||||||
for (int suffix = 1; std::filesystem::exists(candidate); ++suffix) {
|
for (int suffix = 1; std::filesystem::exists(candidate); ++suffix) {
|
||||||
candidate = output_path_ /
|
candidate = output_path_ /
|
||||||
std::filesystem::path("biergarten_seed_" + run_timestamp_utc_ +
|
std::filesystem::path(std::format("biergarten_seed_{}-{}.sqlite",
|
||||||
"-" + std::to_string(suffix) + ".sqlite");
|
run_timestamp_utc_, suffix));
|
||||||
}
|
}
|
||||||
|
|
||||||
return candidate;
|
return candidate;
|
||||||
|
|||||||
@@ -8,6 +8,7 @@
|
|||||||
#include <httplib.h>
|
#include <httplib.h>
|
||||||
|
|
||||||
#include <chrono>
|
#include <chrono>
|
||||||
|
#include <format>
|
||||||
#include <regex>
|
#include <regex>
|
||||||
#include <stdexcept>
|
#include <stdexcept>
|
||||||
#include <string>
|
#include <string>
|
||||||
@@ -47,8 +48,9 @@ std::string HttpWebClient::Get(const std::string& url) {
|
|||||||
const httplib::Result result = client.Get(path);
|
const httplib::Result result = client.Get(path);
|
||||||
|
|
||||||
if (!result) {
|
if (!result) {
|
||||||
throw std::runtime_error("[HttpWebClient] Request failed for URL: " + url +
|
throw std::runtime_error(std::format(
|
||||||
" — " + httplib::to_string(result.error()));
|
"[HttpWebClient] Request failed for URL: {} — {}", url,
|
||||||
|
httplib::to_string(result.error())));
|
||||||
}
|
}
|
||||||
|
|
||||||
if (result->status < kSuccessMin || result->status >= kSuccessMax) {
|
if (result->status < kSuccessMin || result->status >= kSuccessMax) {
|
||||||
@@ -56,12 +58,11 @@ std::string HttpWebClient::Get(const std::string& url) {
|
|||||||
logger_->Log(
|
logger_->Log(
|
||||||
{.level = LogLevel::Error,
|
{.level = LogLevel::Error,
|
||||||
.phase = PipelinePhase::UserGeneration,
|
.phase = PipelinePhase::UserGeneration,
|
||||||
.message =
|
.message =
|
||||||
std::string("[HttpWebClient] Request failed for URL: ") + url});
|
std::format("[HttpWebClient] Request failed for URL: {}", url)});
|
||||||
}
|
}
|
||||||
throw std::runtime_error("[HttpWebClient] HTTP " +
|
throw std::runtime_error(std::format("[HttpWebClient] HTTP {} for URL: {}",
|
||||||
std::to_string(result->status) +
|
result->status, url));
|
||||||
" for URL: " + url);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return result->body;
|
return result->body;
|
||||||
|
|||||||
Reference in New Issue
Block a user