Compare commits
4 Commits
3f978f4de4
...
seeding-up
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
58e476afdb | ||
|
|
d52dba904c | ||
|
|
fbcf438381 | ||
| 2b8a900d12 |
@@ -13,6 +13,7 @@ low-resource language failures.
|
|||||||
- [Model Bias and Language Quality](#model-bias-and-language-quality)
|
- [Model Bias and Language Quality](#model-bias-and-language-quality)
|
||||||
- [Western and Eurocentric Lens](#western-and-eurocentric-lens)
|
- [Western and Eurocentric Lens](#western-and-eurocentric-lens)
|
||||||
- [Wikipedia Enrichment](#wikipedia-enrichment)
|
- [Wikipedia Enrichment](#wikipedia-enrichment)
|
||||||
|
- [Names-by-Country Dataset](#names-by-country-dataset)
|
||||||
- [The "Avoid AI Phrases" Prompt Instruction](#the-avoid-ai-phrases-prompt-instruction)
|
- [The "Avoid AI Phrases" Prompt Instruction](#the-avoid-ai-phrases-prompt-instruction)
|
||||||
- [Known Issues](#known-issues)
|
- [Known Issues](#known-issues)
|
||||||
- [Hallucinated Brewing Techniques](#hallucinated-brewing-techniques)
|
- [Hallucinated Brewing Techniques](#hallucinated-brewing-techniques)
|
||||||
@@ -91,6 +92,33 @@ generated descriptions.
|
|||||||
|
|
||||||
---
|
---
|
||||||
|
|
||||||
|
## Names-by-Country Dataset
|
||||||
|
|
||||||
|
`tooling/pipeline/forenames-by-country.json` and `surnames-by-country.json`
|
||||||
|
(used to sample a `Name` per ISO 3166-1 country code for user generation) are
|
||||||
|
vendored verbatim, unmodified, from
|
||||||
|
[sigpwned/popular-names-by-country-dataset](https://github.com/sigpwned/popular-names-by-country-dataset)
|
||||||
|
(the `common-forenames-by-country.json` / `common-surnames-by-country.json`
|
||||||
|
release assets), released under **CC0** (public domain). That dataset's own
|
||||||
|
forename/surname lists are pulled from Wikipedia's "Lists of most common
|
||||||
|
surnames" and "List of most popular given names" as of the week of
|
||||||
|
2023-07-08 — see that project's README for full provenance. Names are not
|
||||||
|
LLM-generated; this is curated fixture data per ROADMAP.md §2. Per-forename
|
||||||
|
gender from the source data is preserved through to the sampled `Name`
|
||||||
|
(rather than discarded during loading) so it's available for gender-aware
|
||||||
|
persona/bio generation later.
|
||||||
|
|
||||||
|
The full multinational dataset is kept as-is (106 countries for forenames,
|
||||||
|
75 for surnames) rather than trimmed to `cities.json`'s current country
|
||||||
|
list, so it doesn't need re-sourcing if more countries are added later.
|
||||||
|
`SampleName()` (a free helper in `generate_users.cc`) returns no result for
|
||||||
|
a country present in neither file; of the countries in `cities.json`,
|
||||||
|
that's currently `KE`, `SE`, `SG`, `TH`, `VN`, and `ZA` — `GenerateUsers`
|
||||||
|
skips cities in those countries the same way brewery generation skips
|
||||||
|
cities whose enrichment lookup fails.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
## The "Avoid AI Phrases" Prompt Instruction
|
## The "Avoid AI Phrases" Prompt Instruction
|
||||||
|
|
||||||
The system prompt instructs the model to avoid common AI-generated phrasing
|
The system prompt instructs the model to avoid common AI-generated phrasing
|
||||||
@@ -253,7 +281,7 @@ Output sample:
|
|||||||
a location's code is unlisted, skip `description_local` generation and fall
|
a location's code is unlisted, skip `description_local` generation and fall
|
||||||
back to English.
|
back to English.
|
||||||
- **Upstream sanitization:** strip known low-resource language codes from the
|
- **Upstream sanitization:** strip known low-resource language codes from the
|
||||||
`locations.json` payload before generation.
|
`cities.json` payload before generation.
|
||||||
- **Downstream flagging:** add a `description_local_confidence` column to the
|
- **Downstream flagging:** add a `description_local_confidence` column to the
|
||||||
SQLite schema so downstream applications can filter or flag potentially
|
SQLite schema so downstream applications can filter or flag potentially
|
||||||
hallucinated text by language tier.
|
hallucinated text by language tier.
|
||||||
|
|||||||
@@ -1,8 +1,9 @@
|
|||||||
# Biergarten Pipeline
|
# Biergarten Pipeline
|
||||||
|
|
||||||
A C++20 command-line pipeline that samples city records from local JSON,
|
A C++20 command-line pipeline that samples city records from local JSON,
|
||||||
enriches each with Wikipedia context, and generates bilingual brewery names and
|
enriches each with Wikipedia context, and generates bilingual brewery names
|
||||||
descriptions via a local GGUF model or a deterministic mock.
|
and descriptions plus locale-grounded user profiles via a local GGUF model or
|
||||||
|
a deterministic mock.
|
||||||
|
|
||||||
> **This pipeline produces AI-generated data.** It is not a source of truth for
|
> **This pipeline produces AI-generated data.** It is not a source of truth for
|
||||||
> brewing techniques, cultural representation, or local-language accuracy. See
|
> brewing techniques, cultural representation, or local-language accuracy. See
|
||||||
@@ -45,6 +46,7 @@ step.
|
|||||||
| Beer reviews and ratings | Stable brewery fixtures with enough context to anchor review pages |
|
| Beer reviews and ratings | Stable brewery fixtures with enough context to anchor review pages |
|
||||||
| Social follow relationships | Repeatable brewery entities for feeds, follows, and saved lists |
|
| Social follow relationships | Repeatable brewery entities for feeds, follows, and saved lists |
|
||||||
| Geospatial brewery experiences | Latitude, longitude, and country-level metadata |
|
| Geospatial brewery experiences | Latitude, longitude, and country-level metadata |
|
||||||
|
| User accounts and profiles | Locale-grounded names, bios, and an auth-ready email/date-of-birth pair for seeding real accounts |
|
||||||
|
|
||||||
---
|
---
|
||||||
|
|
||||||
@@ -84,7 +86,7 @@ curl -L \
|
|||||||
|
|
||||||
### Run
|
### Run
|
||||||
|
|
||||||
Run from `build/` so the copied `locations.json` and `prompts/` are available.
|
Run from `build/` so the copied `cities.json` and `prompts/` are available.
|
||||||
Each run writes a fresh dated SQLite file such as
|
Each run writes a fresh dated SQLite file such as
|
||||||
`biergarten_seed_2026-04-19T15-30-45.123456Z.sqlite` into the working directory.
|
`biergarten_seed_2026-04-19T15-30-45.123456Z.sqlite` into the working directory.
|
||||||
|
|
||||||
@@ -107,7 +109,7 @@ Each run writes a fresh dated SQLite file such as
|
|||||||
| `--prompt-dir` | Directory containing prompt files (e.g. `BREWERY_GENERATION.md`). Required unless `--mocked` is set. |
|
| `--prompt-dir` | Directory containing prompt files (e.g. `BREWERY_GENERATION.md`). Required unless `--mocked` is set. |
|
||||||
| `--output, -o` | Directory for generated SQLite artifacts. Default: `output`. |
|
| `--output, -o` | Directory for generated SQLite artifacts. Default: `output`. |
|
||||||
| `--log-path` | Path for application logs. Default: `pipeline.log`. |
|
| `--log-path` | Path for application logs. Default: `pipeline.log`. |
|
||||||
| `--location-count` | Number of cities to sample from `locations.json` per run. Default: `10`. |
|
| `--location-count` | Number of cities to sample from `cities.json` per run. Default: `10`. |
|
||||||
| `--temperature` | Sampling temperature. Default: `1.0`. |
|
| `--temperature` | Sampling temperature. Default: `1.0`. |
|
||||||
| `--top-p` | Nucleus sampling. Default: `0.95`. |
|
| `--top-p` | Nucleus sampling. Default: `0.95`. |
|
||||||
| `--top-k` | Top-k sampling. Default: `64`. |
|
| `--top-k` | Top-k sampling. Default: `64`. |
|
||||||
@@ -219,34 +221,60 @@ environment variables listed above to match your run.
|
|||||||
### Pipeline Stages
|
### Pipeline Stages
|
||||||
|
|
||||||
| Stage | Implementation |
|
| Stage | Implementation |
|
||||||
| -------- | --------------------------------------------------------------------------------------------------------------------------------------- |
|
| --------------- | ----------------------------------------------------------------------------------------------------------------------------------------- |
|
||||||
| Load | `JsonLoader::LoadLocations()` reads `locations.json` into typed `Location` records. |
|
| Load | `ICuratedDataService` (`CuratedJsonDataService`) reads `cities.json`, `personas.json`, `forenames-by-country.json`, and `surnames-by-country.json` (paths supplied via a `CuratedDataFilePaths` DTO at construction) into typed records, caching each after its first load. `--mocked` runs use `MockCuratedDataService`'s fixed in-memory dataset instead. |
|
||||||
| Sample | `BiergartenPipelineOrchestrator::QueryCitiesWithCountries()` samples `--location-count` locations per run (default `10`). |
|
| Sample | `BiergartenPipelineOrchestrator::QueryCitiesWithCountries()` samples `--location-count` locations per run (default `10`). |
|
||||||
| Enrich | `WikipediaEnrichmentService` fetches brewing and beer-related context. Keeps going when a lookup fails. `--mocked` runs use `MockEnrichmentService` instead and skip Wikipedia entirely. |
|
| Enrich | `WikipediaEnrichmentService` fetches brewing and beer-related context. Keeps going when a lookup fails. `--mocked` runs use `MockEnrichmentService` instead and skip Wikipedia entirely. |
|
||||||
| Generate | `MockGenerator` or `LlamaGenerator` produces brewery names and descriptions in English and the local language. |
|
| Generate Users | `GenerateUsers()` samples a persona and a forename/surname pair per enriched city (skipping countries with no name data), then `MockGenerator` or `LlamaGenerator` produces a username, bio, and activity weight around the sampled name. |
|
||||||
| Store | `SqliteExportService` writes each successful brewery into a fresh dated `.sqlite` database with normalized location and brewery tables. |
|
| Generate Breweries | `MockGenerator` or `LlamaGenerator` produces brewery names and descriptions in English and the local language. |
|
||||||
|
| Store | `SqliteExportService` writes each successful user and brewery into a fresh dated `.sqlite` database with normalized `cities`, `users`, and `breweries` tables. |
|
||||||
| Log | `spdlog` writes results and warnings to the console. |
|
| Log | `spdlog` writes results and warnings to the console. |
|
||||||
|
|
||||||
If enrichment or generation fails for a city, that city is skipped and the
|
If name sampling, enrichment, or generation fails for a city, that city is
|
||||||
pipeline continues.
|
skipped and the pipeline continues. `GenerateUsers()` runs before
|
||||||
|
`GenerateBreweries()` in `BiergartenPipelineOrchestrator::Run()`.
|
||||||
|
|
||||||
### Key Components
|
### Key Components
|
||||||
|
|
||||||
- `src/main.cc` — argument parsing and Boost.DI composition root.
|
- `src/main.cc` — argument parsing and Boost.DI composition root.
|
||||||
- `JsonLoader` — validates curated location input.
|
- `CuratedJsonDataService` — implements `ICuratedDataService`; takes a
|
||||||
|
`CuratedDataFilePaths` DTO (locations/personas/forenames/surnames paths) in
|
||||||
|
its constructor, then parses and validates curated location, persona, and
|
||||||
|
forename/surname JSON, memoizing each result after its first load on a
|
||||||
|
given instance. Each `cities.json` entry's `postal_code.city_regex` and
|
||||||
|
`postal_code.examples` are parsed into `City::postal_regex` and
|
||||||
|
`City::postal_code_examples`. `MockCuratedDataService` is the in-memory
|
||||||
|
substitute (4 fixed locations, 3 personas, and name data for
|
||||||
|
`US`/`DE`/`FR`/`BE`) used in `--mocked` runs, and carries matching
|
||||||
|
`postal_regex`/`postal_code_examples` values for its 4 locations.
|
||||||
- `WikipediaEnrichmentService` — queries Wikipedia extracts, caches results,
|
- `WikipediaEnrichmentService` — queries Wikipedia extracts, caches results,
|
||||||
returns empty context on failure. `MockEnrichmentService` is the no-op
|
returns empty context on failure. `MockEnrichmentService` is the no-op
|
||||||
substitute used in `--mocked` runs.
|
substitute used in `--mocked` runs.
|
||||||
- `LlamaGenerator` — formats prompts for Gemma 4, validates JSON output, retries
|
- `IPostalCodeService` — generates a postal code for a `City`, consumed by
|
||||||
malformed responses up to three times with corrective feedback in the
|
`GenerateBreweries()` and stored on `BreweryRecord::address` (an `Address`
|
||||||
retry prompt. The token budget is fixed across attempts; it is not raised
|
struct, currently just `postal_code`, mirroring the web backend's
|
||||||
automatically on truncation.
|
`BreweryPostLocation`). Only `MockPostalCodeService` exists today, which
|
||||||
- `MockGenerator` — stable hash-based output so the same city input always
|
ignores `postal_regex` and returns `postal_code_examples.front()` — it's
|
||||||
produces the same brewery.
|
wired into the Boost.DI graph unconditionally (no `--mocked` branch yet,
|
||||||
- `SqliteExportService` — creates a dated SQLite file per run and persists each
|
since there's no real implementation to switch to). A real implementation
|
||||||
successful brewery into normalized tables.
|
still needs a **xeger**-style generator — turning a `postal_regex` pattern
|
||||||
|
into a random matching string — instead of always replaying a fixed
|
||||||
|
example; see [ROADMAP.md §9](./ROADMAP.md#9-postal-code-generation).
|
||||||
|
Street-address generation (`Address::address_line1`) has no fixture data or
|
||||||
|
service yet and remains future work.
|
||||||
|
- `LlamaGenerator` — formats prompts for Gemma 4, validates JSON output for
|
||||||
|
both `GenerateBrewery` and `GenerateUser`, retries malformed responses up
|
||||||
|
to three times with corrective feedback in the retry prompt. The token
|
||||||
|
budget is fixed across attempts; it is not raised automatically on
|
||||||
|
truncation.
|
||||||
|
- `MockGenerator` — stable hash-based output so the same city/persona/name
|
||||||
|
input always produces the same brewery or user.
|
||||||
|
- `SqliteExportService` — creates a dated SQLite file per run and persists
|
||||||
|
each successful user and brewery into normalized tables.
|
||||||
- Brewery payloads include English and local-language name and description
|
- Brewery payloads include English and local-language name and description
|
||||||
fields.
|
fields. User payloads carry a sampled first/last name and gender, an
|
||||||
|
LLM-generated username/bio/activity weight, and a programmatically
|
||||||
|
generated (not LLM-authored) unique email and date of birth.
|
||||||
|
|
||||||
### Runtime Behaviour
|
### Runtime Behaviour
|
||||||
|
|
||||||
@@ -266,6 +294,18 @@ valid output on the first pass; the retry path is kept for resilience.
|
|||||||
`MockGenerator` uses stable hashes for repeatable output in demos and Storybook
|
`MockGenerator` uses stable hashes for repeatable output in demos and Storybook
|
||||||
runs.
|
runs.
|
||||||
|
|
||||||
|
`CuratedJsonDataService` memoizes each of `LoadCities()`, `LoadPersonas()`,
|
||||||
|
`LoadForenamesByCountry()`, and `LoadSurnamesByCountry()` independently the
|
||||||
|
first time each is called, since `BiergartenPipelineOrchestrator` owns a
|
||||||
|
single `ICuratedDataService` instance for the whole run — later calls return
|
||||||
|
the cached result instead of re-parsing.
|
||||||
|
|
||||||
|
`GenerateUsers()` samples a forename/surname pair per city via `SampleName()`,
|
||||||
|
keyed by the city's ISO 3166-1 code. Countries present in `cities.json`
|
||||||
|
but absent from either name fixture (currently `KE`, `SE`, `SG`, `TH`, `VN`,
|
||||||
|
`ZA`) are skipped, the same way a failed enrichment or generation call skips
|
||||||
|
a city — see ETHICS-AND-KNOWN-ISSUES.md's Names-by-Country Dataset section.
|
||||||
|
|
||||||
### Process Flow - Activity Diagram
|
### Process Flow - Activity Diagram
|
||||||
|
|
||||||

|

|
||||||
@@ -278,9 +318,10 @@ runs.
|
|||||||
|
|
||||||
## Generated Output
|
## Generated Output
|
||||||
|
|
||||||
Each successful run stores a `GeneratedBrewery` pair with the source location
|
Each successful run stores a `BreweryRecord` (source `City`, an `Address`,
|
||||||
and a `BreweryResult` payload. The same generated records are also written to a
|
and a `BreweryResult` payload), and a `UserRecord` pair with the source
|
||||||
fresh SQLite export file named with the current UTC timestamp.
|
`City` and a `UserResult` payload. The same generated records are also
|
||||||
|
written to a fresh SQLite export file named with the current UTC timestamp.
|
||||||
|
|
||||||
| Field | Meaning |
|
| Field | Meaning |
|
||||||
| ------------------- | ------------------------------------------ |
|
| ------------------- | ------------------------------------------ |
|
||||||
@@ -288,6 +329,18 @@ fresh SQLite export file named with the current UTC timestamp.
|
|||||||
| `description_en` | Brewery description in English. |
|
| `description_en` | Brewery description in English. |
|
||||||
| `name_local` | Brewery name in the local language. |
|
| `name_local` | Brewery name in the local language. |
|
||||||
| `description_local` | Brewery description in the local language. |
|
| `description_local` | Brewery description in the local language. |
|
||||||
|
| `postal_code` | Postal code generated for the brewery's city (see `IPostalCodeService`, above). |
|
||||||
|
|
||||||
|
| Field | Meaning |
|
||||||
|
| ----------------- | ----------------------------------------------------------------- |
|
||||||
|
| `first_name` | Sampled forename, copied from the curated name data (not LLM-invented). |
|
||||||
|
| `last_name` | Sampled surname, copied from the curated name data (not LLM-invented). |
|
||||||
|
| `gender` | Gender associated with the sampled forename in the source dataset. |
|
||||||
|
| `username` | LLM-generated handle. |
|
||||||
|
| `bio` | LLM-generated short biography. |
|
||||||
|
| `activity_weight` | Relative check-in/activity weight, reserved for a future J-curve activity profile. |
|
||||||
|
| `email` | Unique `@thebiergarten.app` address, generated programmatically from the sampled name. |
|
||||||
|
| `date_of_birth` | Randomized date of birth (age 19-48), generated programmatically. |
|
||||||
|
|
||||||
The log dump also includes city, country, state or province, ISO subdivision
|
The log dump also includes city, country, state or province, ISO subdivision
|
||||||
code, latitude, and longitude for each entry.
|
code, latitude, and longitude for each entry.
|
||||||
@@ -302,6 +355,7 @@ code, latitude, and longitude for each entry.
|
|||||||
| `local_languages` | Locale-aware copy selection |
|
| `local_languages` | Locale-aware copy selection |
|
||||||
| `name_en`, `description_en` | Default English display content |
|
| `name_en`, `description_en` | Default English display content |
|
||||||
| `name_local`, `description_local` | Local-language display content |
|
| `name_local`, `description_local` | Local-language display content |
|
||||||
|
| `postal_code` | Brewery address matching, mirrors the web backend's `BreweryPostLocation.PostalCode` |
|
||||||
|
|
||||||
---
|
---
|
||||||
|
|
||||||
@@ -367,11 +421,17 @@ Silicon; CUDA or HIP/ROCm is detected on Linux when the toolkit is present.
|
|||||||
## Fixture Strategy
|
## Fixture Strategy
|
||||||
|
|
||||||
- `--mocked` for stable fixtures, repeatable screenshots, and Storybook runs.
|
- `--mocked` for stable fixtures, repeatable screenshots, and Storybook runs.
|
||||||
|
`MockCuratedDataService` swaps in for `CuratedJsonDataService`, so no
|
||||||
|
fixture files need to be present on disk.
|
||||||
- `--model` when geographically grounded content matters for demos.
|
- `--model` when geographically grounded content matters for demos.
|
||||||
- Keep `locations.json` structured enough to support discovery and future
|
- Keep `cities.json` structured enough to support discovery and future
|
||||||
filtering.
|
filtering.
|
||||||
- Treat SQLite output as seed material for the app's brewery domain, not
|
- `personas.json`, `forenames-by-country.json`, and
|
||||||
production data.
|
`surnames-by-country.json` are curated/vendored fixture data, not
|
||||||
|
LLM-generated — see ETHICS-AND-KNOWN-ISSUES.md's Names-by-Country Dataset
|
||||||
|
section for provenance.
|
||||||
|
- Treat SQLite output as seed material for the app's brewery and user
|
||||||
|
domains, not production data.
|
||||||
|
|
||||||
---
|
---
|
||||||
|
|
||||||
@@ -381,7 +441,10 @@ Silicon; CUDA or HIP/ROCm is detected on Linux when the toolkit is present.
|
|||||||
| -------------------------------------- | -------------------------------------------------- |
|
| -------------------------------------- | -------------------------------------------------- |
|
||||||
| `tooling/pipeline/includes/` | Public headers and shared models. |
|
| `tooling/pipeline/includes/` | Public headers and shared models. |
|
||||||
| `tooling/pipeline/src/` | Implementation files. |
|
| `tooling/pipeline/src/` | Implementation files. |
|
||||||
| `tooling/pipeline/locations.json` | Curated city input copied into the build tree. |
|
| `tooling/pipeline/cities.json` | Curated city input copied into the build tree. |
|
||||||
|
| `tooling/pipeline/personas.json` | Curated user persona archetypes copied into the build tree. |
|
||||||
|
| `tooling/pipeline/forenames-by-country.json` | Vendored (CC0) forename data by ISO 3166-1 country code. |
|
||||||
|
| `tooling/pipeline/surnames-by-country.json` | Vendored (CC0) surname data by ISO 3166-1 country code. |
|
||||||
| `tooling/pipeline/prompts/` | System prompts used by the model-backed path. |
|
| `tooling/pipeline/prompts/` | System prompts used by the model-backed path. |
|
||||||
| `tooling/pipeline/runpod/` | Dockerfile, launcher, and RunPod pod template. |
|
| `tooling/pipeline/runpod/` | Dockerfile, launcher, and RunPod pod template. |
|
||||||
| `docs/pipeline/diagrams/` | Architecture and pipeline diagrams. |
|
| `docs/pipeline/diagrams/` | Architecture and pipeline diagrams. |
|
||||||
@@ -396,7 +459,14 @@ Paths below are relative to `tooling/pipeline/`.
|
|||||||
- `src/main.cc` — argument parsing and DI composition root.
|
- `src/main.cc` — argument parsing and DI composition root.
|
||||||
- `src/biergarten_pipeline_orchestrator/` — orchestration, sampling, logging,
|
- `src/biergarten_pipeline_orchestrator/` — orchestration, sampling, logging,
|
||||||
and export.
|
and export.
|
||||||
|
- `src/services/curated_data/` — `CuratedJsonDataService`, the file-backed
|
||||||
|
`ICuratedDataService`, and `MockCuratedDataService`, the in-memory
|
||||||
|
`ICuratedDataService` used in `--mocked` runs.
|
||||||
- `src/services/enrichment/wikipedia/` — enrichment service and cache.
|
- `src/services/enrichment/wikipedia/` — enrichment service and cache.
|
||||||
|
- `includes/services/postal_code/` — `IPostalCodeService` and
|
||||||
|
`MockPostalCodeService` (header-only), consumed by `GenerateBreweries()`.
|
||||||
|
The real xeger-based implementation and its `--mocked`-aware DI wiring are
|
||||||
|
still to come.
|
||||||
- `src/services/sqlite/` — SQLite export implementation.
|
- `src/services/sqlite/` — SQLite export implementation.
|
||||||
- `src/data_generation/llama/` — local inference, prompt loading, output
|
- `src/data_generation/llama/` — local inference, prompt loading, output
|
||||||
validation.
|
validation.
|
||||||
@@ -407,11 +477,12 @@ Paths below are relative to `tooling/pipeline/`.
|
|||||||
|
|
||||||
## Next Steps
|
## Next Steps
|
||||||
|
|
||||||
The pipeline currently produces city-aware brewery records and dated SQLite
|
The pipeline currently produces city-aware brewery and user records and
|
||||||
exports. The next passes add additional fixture types so the app can exercise
|
dated SQLite exports. The next passes add additional fixture types so the
|
||||||
the full brewery domain without live data. For the detailed engineering
|
app can exercise the full brewery and social domains without live data. For
|
||||||
breakdown of what's needed to reach the architecture in
|
the detailed engineering breakdown of what's needed to reach the
|
||||||
[`diagrams/planned/`](./diagrams/planned/), see [ROADMAP.md](./ROADMAP.md).
|
architecture in [`diagrams/planned/`](./diagrams/planned/), see
|
||||||
|
[ROADMAP.md](./ROADMAP.md).
|
||||||
|
|
||||||
### Testing — Very High Priority
|
### Testing — Very High Priority
|
||||||
|
|
||||||
@@ -433,12 +504,6 @@ Generate catalog entries with style, ABV, IBU, color, aroma notes, and food
|
|||||||
pairing hints. Link beers back to breweries and cities. Keep style coverage wide
|
pairing hints. Link beers back to breweries and cities. Keep style coverage wide
|
||||||
enough to exercise search, sort, and category filters.
|
enough to exercise search, sort, and category filters.
|
||||||
|
|
||||||
### User Generation
|
|
||||||
|
|
||||||
Generate user profiles with stable names, bios, locale hints, and preference
|
|
||||||
signals. Include stable IDs for downstream fixture joins. Keep output
|
|
||||||
deterministic for screenshots while allowing larger randomized batches.
|
|
||||||
|
|
||||||
### Check-In System
|
### Check-In System
|
||||||
|
|
||||||
Produce timestamped check-in events between users and breweries. Use a J-curve
|
Produce timestamped check-in events between users and breweries. Use a J-curve
|
||||||
|
|||||||
@@ -33,28 +33,125 @@ architecture needs.
|
|||||||
- [ ] Add `BeerResult`, `CheckinResult`, `RatingResult` result payloads.
|
- [ ] Add `BeerResult`, `CheckinResult`, `RatingResult` result payloads.
|
||||||
- [ ] Add `GenerationMetadata` (`generation_id`, `generated_time`,
|
- [ ] Add `GenerationMetadata` (`generation_id`, `generated_time`,
|
||||||
`context_provided`, `generated_with`).
|
`context_provided`, `generated_with`).
|
||||||
- [ ] Add `activity_weight` to `UserResult` (currently just `username`,
|
- [x] Add `first_name`, `last_name`, `gender`, and `activity_weight` to
|
||||||
`bio`).
|
`UserResult` (currently just `username`, `bio`). `first_name`/
|
||||||
|
`last_name`/`gender` are copied from the sampled `Name` (see below),
|
||||||
|
not LLM-invented.
|
||||||
|
- [x] Add `Name` (`first_name`, `last_name`, `gender`) — the sampled result
|
||||||
|
handed to `DataGenerator::GenerateUser`. Add `ForenameEntry` (`name`,
|
||||||
|
`gender`). Landed as a flatter shape than originally planned here: no
|
||||||
|
`NamesByCountry` wrapper class. `curated_data_service.h` instead
|
||||||
|
declares `ForenameList = std::vector<ForenameEntry>` and
|
||||||
|
`SurnameList = std::vector<std::string>`, and
|
||||||
|
`ICuratedDataService` exposes two maps keyed by ISO 3166-1 code —
|
||||||
|
`ForenamesByCountryMap = unordered_map<string, ForenameList>` and
|
||||||
|
`SurnamesByCountryMap = unordered_map<string, SurnameList>` —
|
||||||
|
directly (see §2). Sampling is
|
||||||
|
a free function, `SampleName(forenames_by_country,
|
||||||
|
surnames_by_country, iso3166_1, rng)`, in
|
||||||
|
`generate_users.cc`'s anonymous namespace, not a method on a class.
|
||||||
|
Pairing a forename with a surname happens at sample time so gender
|
||||||
|
(present per-forename in the source data) is never discarded the way a
|
||||||
|
pre-flattened `{first_name, last_name}` list would lose it; see §2.
|
||||||
- [ ] Extend `GeneratedBrewery` with `brewery_id`, `context_completeness`,
|
- [ ] Extend `GeneratedBrewery` with `brewery_id`, `context_completeness`,
|
||||||
`metadata` (currently just `{location, brewery}`).
|
`metadata` (currently `{location, address, brewery}` — see the
|
||||||
- [ ] Add `GeneratedBeer`, `GeneratedUser`, `GeneratedCheckin`,
|
`Location`→`City` rename and `Address` struct added in the
|
||||||
`GeneratedRating`, `GeneratedFollow` aggregate structs.
|
`postal_regex`/§9 entries below).
|
||||||
- [ ] Add `UserPersona` (`name`, `description`, `style_affinities`).
|
- [ ] Add `GeneratedBeer`, `GeneratedCheckin`, `GeneratedRating`,
|
||||||
|
`GeneratedFollow` aggregate structs.
|
||||||
|
- [x] Add `UserRecord` (`location`, `user : UserResult`, `email`,
|
||||||
|
`date_of_birth`) as the current stand-in for the planned
|
||||||
|
`GeneratedUser` — `email` and `date_of_birth` are programmatically
|
||||||
|
generated by the orchestrator, never LLM-authored, so a downstream
|
||||||
|
auth-account seeding consumer can register real accounts from the
|
||||||
|
pipeline's SQLite export. No `password`, `user_id`, or `metadata`
|
||||||
|
field yet, matching `BreweryRecord`'s equally pre-`metadata` shape.
|
||||||
|
Already wired into `IExportService`/`SqliteExportService`: a `users`
|
||||||
|
table exists and `GenerateUsers()` exports each successful record via
|
||||||
|
`ProcessRecord(const UserRecord&)` (see §5) in addition to logging and
|
||||||
|
holding `generated_users_` in memory. Still missing vs. the planned
|
||||||
|
`GeneratedUser`: `user_id`, `password`, and `metadata`.
|
||||||
|
- [x] Add `UserPersona` (`name`, `description`, `style_affinities`).
|
||||||
|
- [x] Add `postal_regex` (`std::vector<std::string>`) and
|
||||||
|
`postal_code_examples` (`std::vector<std::string>`) to `Location`
|
||||||
|
(now `City` — see below), sourced from each `cities.json` entry's
|
||||||
|
`postal_code.city_regex` / `postal_code.examples`. Not part of the
|
||||||
|
originally planned `Location` shape in `diagrams/planned/class.puml` —
|
||||||
|
added ahead of postal code generation work; see §9.
|
||||||
|
- [x] Renamed `Location` → `City` throughout the pipeline (struct, all
|
||||||
|
parameter/return types, `LocationsList`→`CityList`,
|
||||||
|
`LoadLocations()`→`LoadCities()`) so the domain vocabulary matches the
|
||||||
|
web backend's `City`/`CityId`. Field/variable names that merely hold a
|
||||||
|
`City` instance (e.g. `EnrichedCity::location`, `BreweryRecord::location`)
|
||||||
|
were left as-is — `City` itself has a `city` field (the city name
|
||||||
|
string), so renaming the container field too would read as
|
||||||
|
`enriched_city.city.city`. The SQLite export layer's own vocabulary
|
||||||
|
(`locations` table, `location_id` column, `location_cache_`, etc.) was
|
||||||
|
renamed to `cities`/`city_id` to match, since it's this pipeline's own
|
||||||
|
throwaway export, not the backend's SQL Server schema (see §5).
|
||||||
|
- [x] Added a new `Address` struct (`postal_code` only, for now) and a
|
||||||
|
`BreweryRecord::address` member, mirroring the web backend's
|
||||||
|
`BreweryPostLocation` (`AddressLine1`, `AddressLine2`, `PostalCode`,
|
||||||
|
`CityId`). Only `postal_code` is populated — `address_line1`/
|
||||||
|
`address_line2` generation has no fixture data or service yet. Users
|
||||||
|
do not get an `Address`: the backend's `UserAccount` has no address
|
||||||
|
fields, only `BreweryPostLocation` does.
|
||||||
|
|
||||||
## 2. Data Preloading
|
## 2. Data Preloading
|
||||||
|
|
||||||
`tooling/pipeline/includes/json_handling/`, fixture files.
|
`tooling/pipeline/includes/services/curated_data/`, fixture files.
|
||||||
|
|
||||||
- [ ] Extract a `DataPreloader` interface; have `JsonLoader` implement it
|
- [x] Extract an interface; have `CuratedJsonDataService` implement it.
|
||||||
instead of exposing only a static `LoadLocations()`.
|
Landed as `ICuratedDataService`
|
||||||
|
(`services/curated_data/curated_data_service.h`), not `DataPreloader`
|
||||||
|
— `LoadCities()`, `LoadPersonas()`, `LoadForenamesByCountry()`, and
|
||||||
|
`LoadSurnamesByCountry()` are all virtual methods returning `const&`
|
||||||
|
and take no arguments. `CuratedJsonDataService`
|
||||||
|
(`services/curated_data/curated_json_data_service.h`, originally
|
||||||
|
landed as `JsonLoader` under `json_handling/` before being renamed and
|
||||||
|
moved) takes a `CuratedDataFilePaths` DTO (`locations_path`,
|
||||||
|
`personas_path`, `forenames_path`, `surnames_path`) in its
|
||||||
|
constructor rather than a path per `Load*()` call, and memoizes each
|
||||||
|
result in a private `cache` struct (`locations`, `personas`,
|
||||||
|
`forenames_by_country`, `surnames_by_country`), so a second call on
|
||||||
|
the same instance returns the cached result instead of re-parsing —
|
||||||
|
safe because `CuratedJsonDataService` outlives every call site (owned
|
||||||
|
by `BiergartenPipelineOrchestrator` via
|
||||||
|
`unique_ptr<ICuratedDataService>` for the whole run). `main.cc`'s DI
|
||||||
|
injector also gained a `MockCuratedDataService` (fixed in-memory
|
||||||
|
data: 4 locations across `US`/`DE`/`FR`/`BE`, 3 personas, and
|
||||||
|
matching forename/surname sets for those four countries), bound in
|
||||||
|
place of `CuratedJsonDataService` under `--mocked`, mirroring the
|
||||||
|
existing `MockEnrichmentService`/`MockGenerator` pattern.
|
||||||
- [ ] Add `LoadBeerStyles()`. `beer-styles.json` already exists in the repo
|
- [ ] Add `LoadBeerStyles()`. `beer-styles.json` already exists in the repo
|
||||||
and is already copied into the Docker image
|
and is already copied into the Docker image
|
||||||
(`runpod/Dockerfile`), but no loader reads it yet, and the native
|
(`runpod/Dockerfile`), but no loader reads it yet, and the native
|
||||||
CMake build doesn't copy it into `build/` at all — `CMakeLists.txt`'s
|
CMake build doesn't copy it into `build/` at all — `CMakeLists.txt`'s
|
||||||
"Runtime Assets" step only copies `locations.json` and `prompts/`.
|
"Runtime Assets" step only copies `cities.json` and `prompts/`.
|
||||||
- [ ] Add `LoadPersonas()` and author `personas.json` (doesn't exist yet).
|
- [x] Add `LoadPersonas()` and author `personas.json` (doesn't exist yet).
|
||||||
- [ ] Add `LoadNamesByCountry()` and author `names-by-country.json` (doesn't
|
- [x] Add name-by-country loading, parsing
|
||||||
exist yet).
|
`tooling/pipeline/forenames-by-country.json` and
|
||||||
|
`surnames-by-country.json`. Landed as two separate methods —
|
||||||
|
`LoadForenamesByCountry()` and `LoadSurnamesByCountry()` — each
|
||||||
|
returning a flat `ForenamesByCountryMap` / `SurnamesByCountryMap`
|
||||||
|
(aliases for `unordered_map<string, ForenameList>` /
|
||||||
|
`unordered_map<string, SurnameList>`) keyed by ISO 3166-1 code, rather
|
||||||
|
than one combined `LoadNamesByCountry() : NamesByCountry` call. Both
|
||||||
|
files are vendored verbatim (unmodified,
|
||||||
|
full multinational coverage) from
|
||||||
|
`sigpwned/popular-names-by-country-dataset` (CC0) — see
|
||||||
|
ETHICS-AND-KNOWN-ISSUES.md's Names-by-Country Dataset section for
|
||||||
|
provenance. Deliberately not pre-paired or filtered down to just the
|
||||||
|
countries in `cities.json`: keeping the source shape (including
|
||||||
|
per-forename gender) intact means the loader can support more
|
||||||
|
countries later, or gender-aware persona/bio generation, without
|
||||||
|
re-sourcing data.
|
||||||
|
- [x] Parse `postal_code.city_regex` and `postal_code.examples` out of each
|
||||||
|
`cities.json` entry's nested `postal_code` object in
|
||||||
|
`CuratedJsonDataService::LoadCities()`, into `City::postal_regex`
|
||||||
|
/ `City::postal_code_examples` (see §1). `MockCuratedDataService`'s
|
||||||
|
4 fixed locations carry matching values pulled from the corresponding
|
||||||
|
`cities.json` entries.
|
||||||
|
|
||||||
## 3. Policy / Strategy Layer
|
## 3. Policy / Strategy Layer
|
||||||
|
|
||||||
@@ -83,10 +180,14 @@ Entirely new — no `includes/policy/` (or equivalent) directory exists today.
|
|||||||
- [ ] Extend the `DataGenerator` interface with `GenerateBeer`,
|
- [ ] Extend the `DataGenerator` interface with `GenerateBeer`,
|
||||||
`GenerateCheckin`, `GenerateRating` (today: only `GenerateBrewery` and
|
`GenerateCheckin`, `GenerateRating` (today: only `GenerateBrewery` and
|
||||||
`GenerateUser`).
|
`GenerateUser`).
|
||||||
- [ ] Implement `LlamaGenerator::GenerateUser` for real. It currently
|
- [x] Implement `LlamaGenerator::GenerateUser` for real, with a retry loop
|
||||||
returns a hardcoded `{"test_user", "This is a test user profile from
|
mirroring `GenerateBrewery` (GBNF grammar, `ValidateUserJson`, up to 3
|
||||||
{locale}."}` regardless of input — see the `// TODO` at the top of
|
attempts with corrective feedback) and a new
|
||||||
`src/data_generation/llama/generate_user.cc`.
|
`prompts/USER_GENERATION.md`. Changed the `DataGenerator::GenerateUser`
|
||||||
|
signature to
|
||||||
|
`(city : const EnrichedCity&, persona : const UserPersona&, name : const Name&) : UserResult`,
|
||||||
|
matching what the activity diagram actually passes. `MockGenerator::GenerateUser`
|
||||||
|
updated to match the new signature too.
|
||||||
- [ ] Implement `MockGenerator::GenerateBeer` / `GenerateCheckin` /
|
- [ ] Implement `MockGenerator::GenerateBeer` / `GenerateCheckin` /
|
||||||
`GenerateRating`.
|
`GenerateRating`.
|
||||||
- [ ] Add `IPromptFormatter::ExpectedArchitecture()` and
|
- [ ] Add `IPromptFormatter::ExpectedArchitecture()` and
|
||||||
@@ -100,15 +201,25 @@ Entirely new — no `includes/policy/` (or equivalent) directory exists today.
|
|||||||
|
|
||||||
`tooling/pipeline/includes/services/database/`, `src/services/sqlite/`.
|
`tooling/pipeline/includes/services/database/`, `src/services/sqlite/`.
|
||||||
|
|
||||||
- [ ] Extend `IExportService` with `ProcessBeer`, `ProcessUser`,
|
- [x] `IExportService::ProcessRecord(const UserRecord&)` and
|
||||||
`ProcessCheckin`, `ProcessRating`, `ProcessFollow` (today only
|
`SqliteExportService`'s `users` table (see
|
||||||
`ProcessRecord(const GeneratedBrewery&)` exists).
|
`kCreateUsersTableSql` / `kInsertUserSql` in
|
||||||
- [ ] Add `beers`, `users`, `checkins`, `ratings`, `follows` tables to
|
`includes/services/database/sqlite_statement_helpers.h`) are
|
||||||
`SqliteExportService::InitializeSchema()` — see
|
implemented — `GenerateUsers()` exports every successful user
|
||||||
`kCreateLocationsTableSql` / `kCreateBreweriesTableSql` in
|
alongside its resolved `city_id`.
|
||||||
|
- [x] `ProcessRecord(const BreweryRecord&)` now also binds
|
||||||
|
`BreweryRecord::address.postal_code` into a `postal_code` column on
|
||||||
|
`breweries` (see §9).
|
||||||
|
- [ ] Extend `IExportService` with `ProcessBeer`, `ProcessCheckin`,
|
||||||
|
`ProcessRating`, `ProcessFollow` (today `ProcessRecord(const
|
||||||
|
BreweryRecord&)` and `ProcessRecord(const UserRecord&)` exist).
|
||||||
|
- [ ] Add `beers`, `checkins`, `ratings`, `follows` tables to
|
||||||
|
`SqliteExportService::InitializeSchema()` (`cities`, `breweries`,
|
||||||
|
and `users` already exist) — see `kCreateCitiesTableSql` /
|
||||||
|
`kCreateBreweriesTableSql` / `kCreateUsersTableSql` in
|
||||||
`includes/services/database/sqlite_statement_helpers.h` for the
|
`includes/services/database/sqlite_statement_helpers.h` for the
|
||||||
existing pattern to follow.
|
existing pattern to follow.
|
||||||
- [ ] Add a `brewery_cache_` alongside the existing `location_cache_`, and
|
- [ ] Add a `brewery_cache_` alongside the existing `city_cache_`, and
|
||||||
move from one open transaction for the whole run to per-phase batched
|
move from one open transaction for the whole run to per-phase batched
|
||||||
commits (`BEGIN` / `COMMIT & BEGIN` on a batch-size threshold), as
|
commits (`BEGIN` / `COMMIT & BEGIN` on a batch-size threshold), as
|
||||||
shown in the planned activity diagram.
|
shown in the planned activity diagram.
|
||||||
@@ -151,11 +262,42 @@ Entirely new — no `includes/policy/` (or equivalent) directory exists today.
|
|||||||
|
|
||||||
## 8. Fixtures / Build
|
## 8. Fixtures / Build
|
||||||
|
|
||||||
- [ ] Author `personas.json` and `names-by-country.json` (see §2).
|
- [x] Author `personas.json`, `forenames-by-country.json`, and
|
||||||
|
`surnames-by-country.json` (see §2).
|
||||||
- [ ] Fix the `beer-styles.json` build-tree gap: add it to the "Runtime
|
- [ ] Fix the `beer-styles.json` build-tree gap: add it to the "Runtime
|
||||||
Assets" `configure_file` step in `CMakeLists.txt` so native builds and
|
Assets" `configure_file` step in `CMakeLists.txt` so native builds and
|
||||||
the Docker image agree on what's available at runtime.
|
the Docker image agree on what's available at runtime.
|
||||||
|
|
||||||
|
## 9. Postal Code Generation
|
||||||
|
|
||||||
|
`tooling/pipeline/includes/services/postal_code/`. Not part of the original
|
||||||
|
`diagrams/planned/class.puml` sketch — added once `Location` (now `City`)
|
||||||
|
gained `postal_regex`/`postal_code_examples` (§1, §2).
|
||||||
|
|
||||||
|
- [x] Add the `IPostalCodeService` interface —
|
||||||
|
`GeneratePostalCode(location : const City&) : std::string`.
|
||||||
|
- [x] Add `MockPostalCodeService`: ignores `location.postal_regex` entirely
|
||||||
|
and returns `location.postal_code_examples.front()`.
|
||||||
|
- [ ] **Implement a real, xeger-based `IPostalCodeService`.** A xeger
|
||||||
|
generator turns a regex pattern (here, one of
|
||||||
|
`City::postal_regex`) into a random string that matches it — the
|
||||||
|
inverse of matching a string against a regex. Without it, every
|
||||||
|
generated postal code is either a hardcoded mock value or absent
|
||||||
|
entirely; a real implementation is required before postal codes can
|
||||||
|
vary per generated brewery instead of always reusing the same
|
||||||
|
curated example.
|
||||||
|
- [x] Wire `IPostalCodeService` into `main.cc`'s Boost.DI injector — bound
|
||||||
|
unconditionally to `MockPostalCodeService` (no `--mocked` branch yet,
|
||||||
|
since a real implementation doesn't exist to switch to; add one once
|
||||||
|
the xeger-based generator above lands, mirroring `IEnrichmentService`)
|
||||||
|
— and into `BiergartenPipelineOrchestrator::GenerateBreweries()`,
|
||||||
|
which calls `GeneratePostalCode(location)` per brewery and stores the
|
||||||
|
result on `BreweryRecord::address` (a new `Address` struct — see §1),
|
||||||
|
exported to the `breweries.postal_code` SQLite column (§5).
|
||||||
|
Postal-code generation is per-brewery, not per-city or per-user: users
|
||||||
|
don't have an address concept in the web backend, only breweries do
|
||||||
|
(`BreweryPostLocation`), so `UserRecord` was deliberately not extended.
|
||||||
|
|
||||||
---
|
---
|
||||||
|
|
||||||
## Suggested build order
|
## Suggested build order
|
||||||
@@ -173,3 +315,7 @@ populated. Phase 1b may begin.", etc.) imply a dependency order. Roughly:
|
|||||||
planned diagrams.
|
planned diagrams.
|
||||||
5. Enrichment cache pre-warming and the city/region query decision (§7) —
|
5. Enrichment cache pre-warming and the city/region query decision (§7) —
|
||||||
useful at any point, but most valuable once phases run concurrently.
|
useful at any point, but most valuable once phases run concurrently.
|
||||||
|
6. Postal code generation (§9) is independent of the phases above — the
|
||||||
|
xeger-based generator just needs `City::postal_regex`, already in
|
||||||
|
place. Wiring the mock into brewery output has already landed (§9); the
|
||||||
|
real generator can be swapped in once built.
|
||||||
|
|||||||
@@ -69,14 +69,14 @@ end note
|
|||||||
:QueryCitiesWithCountries();
|
:QueryCitiesWithCountries();
|
||||||
|
|
||||||
|#E2EBDC|JsonLoader|
|
|#E2EBDC|JsonLoader|
|
||||||
:JsonLoader::LoadLocations("locations.json");
|
:JsonLoader::LoadCities("cities.json");
|
||||||
:std::ranges::sample(all_locations, --location-count);
|
:std::ranges::sample(all_locations, --location-count);
|
||||||
note right
|
note right
|
||||||
--location-count defaults to 10.
|
--location-count defaults to 10.
|
||||||
end note
|
end note
|
||||||
|
|
||||||
|#EAF0E8|BiergartenPipelineOrchestrator|
|
|#EAF0E8|BiergartenPipelineOrchestrator|
|
||||||
while (For each sampled Location?) is (Remaining cities)
|
while (For each sampled City?) is (Remaining cities)
|
||||||
|#DCE8D8|IEnrichmentService|
|
|#DCE8D8|IEnrichmentService|
|
||||||
:GetLocationContext(loc);
|
:GetLocationContext(loc);
|
||||||
note right
|
note right
|
||||||
@@ -89,7 +89,78 @@ while (For each sampled Location?) is (Remaining cities)
|
|||||||
if (Lookup failed?) then (yes)
|
if (Lookup failed?) then (yes)
|
||||||
:Log warning, skip city;
|
:Log warning, skip city;
|
||||||
else (no)
|
else (no)
|
||||||
:Store EnrichedCity{Location, region_context};
|
:Store EnrichedCity{City, region_context};
|
||||||
|
endif
|
||||||
|
endwhile (Done)
|
||||||
|
|
||||||
|
|#EAF0E8|BiergartenPipelineOrchestrator|
|
||||||
|
:GenerateUsers(enriched_cities);
|
||||||
|
|
||||||
|
|#E2EBDC|JsonLoader|
|
||||||
|
:LoadPersonas("personas.json");
|
||||||
|
:LoadForenamesByCountry("forenames-by-country.json");
|
||||||
|
:LoadSurnamesByCountry("surnames-by-country.json");
|
||||||
|
note right
|
||||||
|
Each call is cached after its first parse.
|
||||||
|
MockCuratedDataService (--mocked) returns a
|
||||||
|
fixed in-memory dataset instead and skips
|
||||||
|
these three JSON files entirely.
|
||||||
|
end note
|
||||||
|
|
||||||
|
|#EAF0E8|BiergartenPipelineOrchestrator|
|
||||||
|
while (For each EnrichedCity?) is (Remaining cities)
|
||||||
|
:SampleName(forenames_by_country,\n surnames_by_country, location.iso3166_1);
|
||||||
|
if (Names available for country?) then (no)
|
||||||
|
:Log warning, skip city;
|
||||||
|
else (yes)
|
||||||
|
:Sample UserPersona (uniform random);
|
||||||
|
|
||||||
|
|#E5EDE1|DataGenerator|
|
||||||
|
if (Generator Mode) then (MockGenerator)
|
||||||
|
:DeterministicHash & Format;
|
||||||
|
else (LlamaGenerator)
|
||||||
|
:prompt_directory_->Load("USER_GENERATION");
|
||||||
|
note right
|
||||||
|
Resolves to USER_GENERATION.md inside --prompt-dir.
|
||||||
|
end note
|
||||||
|
repeat
|
||||||
|
:Infer(system_prompt, user_prompt, max_tokens, kUserJsonGrammar);
|
||||||
|
:ValidateUserJson(raw, user);
|
||||||
|
if (Is JSON Valid?) then (yes)
|
||||||
|
break
|
||||||
|
else (no)
|
||||||
|
:Attempt++;
|
||||||
|
:Append validation error to retry prompt;
|
||||||
|
endif
|
||||||
|
repeat while (Attempt < 3?) is (yes)
|
||||||
|
endif
|
||||||
|
|
||||||
|
|#EAF0E8|BiergartenPipelineOrchestrator|
|
||||||
|
if (Generation successful?) then (yes)
|
||||||
|
:BuildEmail(name, used_email_local_parts);
|
||||||
|
:GenerateDateOfBirth(rng);
|
||||||
|
note right
|
||||||
|
email and date_of_birth are generated
|
||||||
|
programmatically, never LLM-authored.
|
||||||
|
end note
|
||||||
|
|
||||||
|
|#E0EAE0|SqliteExportService|
|
||||||
|
:ProcessRecord(UserRecord);
|
||||||
|
if (City in cache?) then (yes)
|
||||||
|
:Reuse city_id;
|
||||||
|
else (no)
|
||||||
|
:Insert City & Cache ID;
|
||||||
|
endif
|
||||||
|
:Insert User (FK: city_id);
|
||||||
|
|
||||||
|
if (Exception caught during insert?) then (yes)
|
||||||
|
|#EAF0E8|BiergartenPipelineOrchestrator|
|
||||||
|
:Log warning "SQLite export failed";
|
||||||
|
else (no)
|
||||||
|
endif
|
||||||
|
else (no)
|
||||||
|
:Log warning "Generation failed, skipping...";
|
||||||
|
endif
|
||||||
endif
|
endif
|
||||||
endwhile (Done)
|
endwhile (Done)
|
||||||
|
|
||||||
@@ -126,12 +197,12 @@ while (For each EnrichedCity?) is (Remaining cities)
|
|||||||
if (Generation successful?) then (yes)
|
if (Generation successful?) then (yes)
|
||||||
|#E0EAE0|SqliteExportService|
|
|#E0EAE0|SqliteExportService|
|
||||||
:ProcessRecord(GeneratedBrewery);
|
:ProcessRecord(GeneratedBrewery);
|
||||||
if (Location in cache?) then (yes)
|
if (City in cache?) then (yes)
|
||||||
:Reuse location_id;
|
:Reuse city_id;
|
||||||
else (no)
|
else (no)
|
||||||
:Insert Location & Cache ID;
|
:Insert City & Cache ID;
|
||||||
endif
|
endif
|
||||||
:Insert Brewery (FK: location_id);
|
:Insert Brewery (FK: city_id);
|
||||||
|
|
||||||
if (Exception caught during insert?) then (yes)
|
if (Exception caught during insert?) then (yes)
|
||||||
|#EAF0E8|BiergartenPipelineOrchestrator|
|
|#EAF0E8|BiergartenPipelineOrchestrator|
|
||||||
|
|||||||
@@ -30,11 +30,15 @@ class BiergartenPipelineOrchestrator {
|
|||||||
- 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>
|
||||||
|
- curated_data_service_ : std::unique_ptr<ICuratedDataService>
|
||||||
|
- postal_code_service_ : std::unique_ptr<IPostalCodeService>
|
||||||
- application_options_ : ApplicationOptions
|
- application_options_ : ApplicationOptions
|
||||||
- generated_breweries_ : std::vector<GeneratedBrewery>
|
- generated_breweries_ : std::vector<BreweryRecord>
|
||||||
|
- generated_users_ : std::vector<UserRecord>
|
||||||
+ Run() : bool
|
+ Run() : bool
|
||||||
- QueryCitiesWithCountries() : std::vector<Location>
|
- QueryCitiesWithCountries() : std::vector<City>
|
||||||
- GenerateBreweries(cities : std::span<const EnrichedCity>) : void
|
- GenerateBreweries(cities : std::span<const EnrichedCity>) : void
|
||||||
|
- GenerateUsers(cities : std::span<const EnrichedCity>) : void
|
||||||
- LogResults() : void
|
- LogResults() : void
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -87,17 +91,17 @@ class LogDispatcher {
|
|||||||
}
|
}
|
||||||
|
|
||||||
interface IEnrichmentService <<interface>> {
|
interface IEnrichmentService <<interface>> {
|
||||||
+ GetLocationContext(loc : const Location&) : std::string
|
+ GetLocationContext(loc : const City&) : std::string
|
||||||
}
|
}
|
||||||
|
|
||||||
class MockEnrichmentService {
|
class MockEnrichmentService {
|
||||||
+ GetLocationContext(loc : const Location&) : std::string
|
+ GetLocationContext(loc : const City&) : std::string
|
||||||
}
|
}
|
||||||
|
|
||||||
class WikipediaEnrichmentService {
|
class WikipediaEnrichmentService {
|
||||||
- client_ : std::unique_ptr<WebClient>
|
- client_ : std::unique_ptr<WebClient>
|
||||||
- extract_cache_ : std::unordered_map<std::string, std::string>
|
- extract_cache_ : std::unordered_map<std::string, std::string>
|
||||||
+ GetLocationContext(loc : const Location&) : std::string
|
+ GetLocationContext(loc : const City&) : std::string
|
||||||
- FetchExtract(query : std::string_view) : std::string
|
- FetchExtract(query : std::string_view) : std::string
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -112,14 +116,15 @@ class HttpWebClient {
|
|||||||
}
|
}
|
||||||
|
|
||||||
interface DataGenerator <<interface>> {
|
interface DataGenerator <<interface>> {
|
||||||
+ GenerateBrewery(location : const Location&, region_context : const std::string&) : BreweryResult
|
+ GenerateBrewery(location : const City&, region_context : const std::string&) : BreweryResult
|
||||||
+ GenerateUser(locale : const std::string&) : UserResult
|
+ GenerateUser(city : const EnrichedCity&, persona : const UserPersona&, name : const Name&) : UserResult
|
||||||
}
|
}
|
||||||
|
|
||||||
class MockGenerator {
|
class MockGenerator {
|
||||||
+ GenerateBrewery(...) : BreweryResult
|
+ GenerateBrewery(location : const City&, region_context : const std::string&) : BreweryResult
|
||||||
+ GenerateUser(...) : UserResult
|
+ GenerateUser(city : const EnrichedCity&, persona : const UserPersona&, name : const Name&) : UserResult
|
||||||
- DeterministicHash(location : const Location&) : size_t
|
- DeterministicHash(location : const City&) : size_t
|
||||||
|
- DeterministicHash(location : const City&, persona : const UserPersona&, name : const Name&) : size_t
|
||||||
}
|
}
|
||||||
|
|
||||||
class LlamaGenerator {
|
class LlamaGenerator {
|
||||||
@@ -153,13 +158,70 @@ class PromptDirectory {
|
|||||||
+ Load(key : std::string_view) : std::string
|
+ Load(key : std::string_view) : std::string
|
||||||
}
|
}
|
||||||
|
|
||||||
class JsonLoader {
|
interface ICuratedDataService <<interface>> {
|
||||||
+ {static} LoadLocations(filepath : const std::filesystem::path&) : std::vector<Location>
|
+ LoadCities(filepath : const std::filesystem::path&) : const std::vector<City>&
|
||||||
|
+ LoadPersonas(filepath : const std::filesystem::path&) : const std::vector<UserPersona>&
|
||||||
|
+ LoadForenamesByCountry(filepath : const std::filesystem::path&) : const std::unordered_map<std::string, forename_list>&
|
||||||
|
+ LoadSurnamesByCountry(filepath : const std::filesystem::path&) : const std::unordered_map<std::string, surname_list>&
|
||||||
}
|
}
|
||||||
|
|
||||||
|
class JsonLoader {
|
||||||
|
- cache_ : cache
|
||||||
|
+ LoadCities(filepath : const std::filesystem::path&) : const std::vector<City>&
|
||||||
|
+ LoadPersonas(filepath : const std::filesystem::path&) : const std::vector<UserPersona>&
|
||||||
|
+ LoadForenamesByCountry(filepath : const std::filesystem::path&) : const std::unordered_map<std::string, forename_list>&
|
||||||
|
+ LoadSurnamesByCountry(filepath : const std::filesystem::path&) : const std::unordered_map<std::string, surname_list>&
|
||||||
|
}
|
||||||
|
|
||||||
|
note right of JsonLoader
|
||||||
|
Each Load* method memoizes its result in
|
||||||
|
cache_ on first call; later calls on the
|
||||||
|
same instance return the cached value
|
||||||
|
without re-parsing.
|
||||||
|
end note
|
||||||
|
|
||||||
|
class MockCuratedDataService {
|
||||||
|
- locations_ : std::vector<City>
|
||||||
|
- personas_ : std::vector<UserPersona>
|
||||||
|
- forenames_by_country_ : std::unordered_map<std::string, forename_list>
|
||||||
|
- surnames_by_country_ : std::unordered_map<std::string, surname_list>
|
||||||
|
+ LoadCities(filepath : const std::filesystem::path&) : const std::vector<City>&
|
||||||
|
+ LoadPersonas(filepath : const std::filesystem::path&) : const std::vector<UserPersona>&
|
||||||
|
+ LoadForenamesByCountry(filepath : const std::filesystem::path&) : const std::unordered_map<std::string, forename_list>&
|
||||||
|
+ LoadSurnamesByCountry(filepath : const std::filesystem::path&) : const std::unordered_map<std::string, surname_list>&
|
||||||
|
}
|
||||||
|
|
||||||
|
note right of MockCuratedDataService
|
||||||
|
Fixed 4-location / 3-persona dataset
|
||||||
|
(US, DE, FR, BE) used in --mocked runs;
|
||||||
|
filepath arguments are ignored.
|
||||||
|
end note
|
||||||
|
|
||||||
|
interface IPostalCodeService <<interface>> {
|
||||||
|
+ GeneratePostalCode(location : const City&) : std::string
|
||||||
|
}
|
||||||
|
|
||||||
|
class MockPostalCodeService {
|
||||||
|
+ GeneratePostalCode(location : const City&) : std::string
|
||||||
|
}
|
||||||
|
|
||||||
|
note right of MockPostalCodeService
|
||||||
|
Ignores location.postal_regex and returns
|
||||||
|
location.postal_code_examples.front(). Bound
|
||||||
|
unconditionally in main.cc's DI graph (no
|
||||||
|
--mocked branch yet -- no real implementation
|
||||||
|
exists to switch to) and called from
|
||||||
|
GenerateBreweries() per brewery, stored on
|
||||||
|
BreweryRecord::address. A real implementation
|
||||||
|
needs a xeger-style generator to turn
|
||||||
|
postal_regex into a synthesized matching
|
||||||
|
string. See ROADMAP.md §9.
|
||||||
|
end note
|
||||||
|
|
||||||
interface IExportService <<interface>> {
|
interface IExportService <<interface>> {
|
||||||
+ Initialize() : void
|
+ Initialize() : void
|
||||||
+ ProcessRecord(brewery : const GeneratedBrewery&) : void
|
+ ProcessRecord(brewery : const BreweryRecord&) : uint64_t
|
||||||
|
+ ProcessRecord(user : const UserRecord&) : uint64_t
|
||||||
+ Finalize() : void
|
+ Finalize() : void
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -167,15 +229,18 @@ class SqliteExportService {
|
|||||||
- date_time_provider_ : std::unique_ptr<IDateTimeProvider>
|
- date_time_provider_ : std::unique_ptr<IDateTimeProvider>
|
||||||
- run_timestamp_utc_ : std::string
|
- run_timestamp_utc_ : std::string
|
||||||
- database_path_ : std::filesystem::path
|
- database_path_ : std::filesystem::path
|
||||||
- db_handle_ : sqlite3*
|
- db_handle_ : SqliteDatabaseHandle
|
||||||
- insert_location_stmt_ : sqlite3_stmt*
|
- insert_city_stmt_ : SqliteStatementHandle
|
||||||
- insert_brewery_stmt_ : sqlite3_stmt*
|
- insert_brewery_stmt_ : SqliteStatementHandle
|
||||||
|
- insert_user_stmt_ : SqliteStatementHandle
|
||||||
- transaction_open_ : bool
|
- transaction_open_ : bool
|
||||||
- location_cache_ : std::unordered_map<std::string, sqlite3_int64>
|
- city_cache_ : std::unordered_map<std::string, sqlite3_int64>
|
||||||
+ Initialize() : void
|
+ Initialize() : void
|
||||||
+ ProcessRecord(brewery : const GeneratedBrewery&) : void
|
+ ProcessRecord(brewery : const BreweryRecord&) : uint64_t
|
||||||
|
+ ProcessRecord(user : const UserRecord&) : uint64_t
|
||||||
+ Finalize() : void
|
+ Finalize() : void
|
||||||
- InitializeSchema() : void
|
- InitializeSchema() : void
|
||||||
|
- ResolveCityId(location : const City&) : sqlite3_int64
|
||||||
}
|
}
|
||||||
|
|
||||||
interface IDateTimeProvider <<interface>> {
|
interface IDateTimeProvider <<interface>> {
|
||||||
@@ -191,6 +256,8 @@ BiergartenPipelineOrchestrator *-- ILogger : owns
|
|||||||
BiergartenPipelineOrchestrator *-- IEnrichmentService : owns
|
BiergartenPipelineOrchestrator *-- IEnrichmentService : owns
|
||||||
BiergartenPipelineOrchestrator *-- DataGenerator : owns
|
BiergartenPipelineOrchestrator *-- DataGenerator : owns
|
||||||
BiergartenPipelineOrchestrator *-- IExportService : owns
|
BiergartenPipelineOrchestrator *-- IExportService : owns
|
||||||
|
BiergartenPipelineOrchestrator *-- ICuratedDataService : owns
|
||||||
|
BiergartenPipelineOrchestrator *-- IPostalCodeService : owns
|
||||||
|
|
||||||
LogEntry *-- LogLevel
|
LogEntry *-- LogLevel
|
||||||
LogEntry *-- PipelinePhase
|
LogEntry *-- PipelinePhase
|
||||||
@@ -215,7 +282,10 @@ LlamaGenerator *-- IPromptDirectory : uses
|
|||||||
IPromptFormatter <|.. Gemma4JinjaPromptFormatter : implements
|
IPromptFormatter <|.. Gemma4JinjaPromptFormatter : implements
|
||||||
IPromptDirectory <|.. PromptDirectory : implements
|
IPromptDirectory <|.. PromptDirectory : implements
|
||||||
|
|
||||||
BiergartenPipelineOrchestrator ..> JsonLoader : uses
|
ICuratedDataService <|.. JsonLoader : implements
|
||||||
|
ICuratedDataService <|.. MockCuratedDataService : implements
|
||||||
|
|
||||||
|
IPostalCodeService <|.. MockPostalCodeService : implements
|
||||||
|
|
||||||
IExportService <|.. SqliteExportService : implements
|
IExportService <|.. SqliteExportService : implements
|
||||||
SqliteExportService *-- IDateTimeProvider : owns
|
SqliteExportService *-- IDateTimeProvider : owns
|
||||||
|
|||||||
|
Before Width: | Height: | Size: 55 KiB After Width: | Height: | Size: 86 KiB |
|
Before Width: | Height: | Size: 71 KiB After Width: | Height: | Size: 94 KiB |
@@ -39,14 +39,22 @@ fork
|
|||||||
:JsonLoader::LoadBeerStyles("beer-styles.json");
|
:JsonLoader::LoadBeerStyles("beer-styles.json");
|
||||||
:EnrichmentService::PreWarmBeerStyleCache(beer_styles);
|
:EnrichmentService::PreWarmBeerStyleCache(beer_styles);
|
||||||
fork again
|
fork again
|
||||||
:JsonLoader::LoadLocations("locations.json");
|
:JsonLoader::LoadCities("cities.json");
|
||||||
:EnrichmentService::PreWarmLocationCache(sampled_locations);
|
:EnrichmentService::PreWarmLocationCache(sampled_locations);
|
||||||
end fork
|
end fork
|
||||||
fork
|
fork
|
||||||
:JsonLoader::LoadNamesByCountry("names-by-country.json");
|
:JsonLoader::LoadForenamesByCountry(\n "forenames-by-country.json");
|
||||||
|
fork again
|
||||||
|
:JsonLoader::LoadSurnamesByCountry(\n "surnames-by-country.json");
|
||||||
fork again
|
fork again
|
||||||
:JsonLoader::LoadPersonas("personas.json");
|
:JsonLoader::LoadPersonas("personas.json");
|
||||||
end fork
|
end fork
|
||||||
|
note right
|
||||||
|
Each call is memoized after its first parse
|
||||||
|
(implemented today). MockCuratedDataService
|
||||||
|
returns a fixed in-memory dataset instead
|
||||||
|
under --mocked, skipping these JSON files.
|
||||||
|
end note
|
||||||
|
|
||||||
' ═══════════════════════════════════════════
|
' ═══════════════════════════════════════════
|
||||||
' PHASE 0 — USER GENERATION
|
' PHASE 0 — USER GENERATION
|
||||||
@@ -67,7 +75,7 @@ fork
|
|||||||
fork again
|
fork again
|
||||||
|LLM Worker|
|
|LLM Worker|
|
||||||
while (loc_ch has items?) is (yes)
|
while (loc_ch has items?) is (yes)
|
||||||
:Receive Location;
|
:Receive City;
|
||||||
|
|
||||||
:GetLocationContextFromCache(location);
|
:GetLocationContextFromCache(location);
|
||||||
note right
|
note right
|
||||||
@@ -82,11 +90,15 @@ fork again
|
|||||||
ibu_preference, checkin_weight.
|
ibu_preference, checkin_weight.
|
||||||
end note
|
end note
|
||||||
|
|
||||||
:NamesByCountry::SampleName(\n location.iso3166_1);
|
:SampleName(forenames_by_country,\n surnames_by_country, location.iso3166_1, rng);
|
||||||
note right
|
note right
|
||||||
Deterministic lookup -- no LLM involved.
|
Deterministic lookup -- no LLM involved.
|
||||||
Name selected from pre-keyed table
|
Free function (implemented today in
|
||||||
|
generate_users.cc), not a class method.
|
||||||
|
Name selected from pre-keyed maps
|
||||||
and passed into the generation prompt.
|
and passed into the generation prompt.
|
||||||
|
Skips the city if either map has no
|
||||||
|
entry for iso3166_1.
|
||||||
end note
|
end note
|
||||||
|
|
||||||
:GenerateUser(enriched_city, persona, sampled_name)\nvia DataGenerator;
|
:GenerateUser(enriched_city, persona, sampled_name)\nvia DataGenerator;
|
||||||
@@ -134,13 +146,13 @@ end fork
|
|||||||
fork
|
fork
|
||||||
|Orchestrator|
|
|Orchestrator|
|
||||||
:Loop: Sample User from user_pool_
|
:Loop: Sample User from user_pool_
|
||||||
and pair with Location;
|
and pair with City;
|
||||||
:Send BreweryTask(Location, User) -> loc_ch;
|
:Send BreweryTask(City, User) -> loc_ch;
|
||||||
:Close loc_ch;
|
:Close loc_ch;
|
||||||
fork again
|
fork again
|
||||||
|LLM Worker|
|
|LLM Worker|
|
||||||
while (loc_ch has items?) is (yes)
|
while (loc_ch has items?) is (yes)
|
||||||
:Receive BreweryTask(Location, User);
|
:Receive BreweryTask(City, User);
|
||||||
|
|
||||||
:GetLocationContextFromCache(task.location);
|
:GetLocationContextFromCache(task.location);
|
||||||
note right
|
note right
|
||||||
|
|||||||
@@ -12,17 +12,40 @@ title Biergarten Data Pipeline — Class Diagram
|
|||||||
|
|
||||||
package "Domain: Models" {
|
package "Domain: Models" {
|
||||||
|
|
||||||
class Location {
|
class City {
|
||||||
+ city : std::string
|
+ city : std::string
|
||||||
+ state_province : std::string
|
+ state_province : std::string
|
||||||
+ iso3166_2 : std::string
|
+ iso3166_2 : std::string
|
||||||
+ country : std::string
|
+ country : std::string
|
||||||
+ iso3166_1 : std::string
|
+ iso3166_1 : std::string
|
||||||
|
+ postal_regex : std::vector<std::string>
|
||||||
|
+ postal_code_examples : std::vector<std::string>
|
||||||
+ local_languages : std::vector<std::string>
|
+ local_languages : std::vector<std::string>
|
||||||
+ latitude : double
|
+ latitude : double
|
||||||
+ longitude : double
|
+ longitude : double
|
||||||
}
|
}
|
||||||
|
|
||||||
|
note right of City
|
||||||
|
postal_regex / postal_code_examples are
|
||||||
|
sourced from each cities.json entry's
|
||||||
|
postal_code.city_regex / postal_code.examples
|
||||||
|
(implemented today, ahead of the rest of this
|
||||||
|
diagram -- see ROADMAP.md §1/§2). Renamed from
|
||||||
|
Location to City to match the web backend's
|
||||||
|
City/CityId vocabulary -- see ROADMAP.md §1.
|
||||||
|
end note
|
||||||
|
|
||||||
|
class Address {
|
||||||
|
+ postal_code : std::string
|
||||||
|
}
|
||||||
|
|
||||||
|
note right of Address
|
||||||
|
Mirrors the web backend's BreweryPostLocation.
|
||||||
|
Only postal_code is populated today --
|
||||||
|
address_line1/address_line2 have no fixture
|
||||||
|
data or generator yet. See ROADMAP.md §9.
|
||||||
|
end note
|
||||||
|
|
||||||
class LocationContext {
|
class LocationContext {
|
||||||
+ text : std::string
|
+ text : std::string
|
||||||
+ completeness : Completeness
|
+ completeness : Completeness
|
||||||
@@ -36,7 +59,7 @@ package "Domain: Models" {
|
|||||||
}
|
}
|
||||||
|
|
||||||
class EnrichedCity {
|
class EnrichedCity {
|
||||||
+ location : Location
|
+ location : City
|
||||||
+ context : LocationContext
|
+ context : LocationContext
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -67,11 +90,50 @@ package "Domain: Models" {
|
|||||||
}
|
}
|
||||||
|
|
||||||
class UserResult {
|
class UserResult {
|
||||||
|
+ first_name : std::string
|
||||||
|
+ last_name : std::string
|
||||||
|
+ gender : std::string
|
||||||
+ username : std::string
|
+ username : std::string
|
||||||
+ bio : std::string
|
+ bio : std::string
|
||||||
+ activity_weight : float
|
+ activity_weight : float
|
||||||
}
|
}
|
||||||
|
|
||||||
|
class Name {
|
||||||
|
+ first_name : std::string
|
||||||
|
+ last_name : std::string
|
||||||
|
+ gender : std::string
|
||||||
|
}
|
||||||
|
|
||||||
|
class ForenameEntry {
|
||||||
|
+ name : std::string
|
||||||
|
+ gender : std::string
|
||||||
|
}
|
||||||
|
|
||||||
|
note right of ForenameEntry
|
||||||
|
forename_list = std::unordered_set<ForenameEntry>
|
||||||
|
surname_list = std::unordered_set<std::string>
|
||||||
|
(aliases declared in curated_data_service.h).
|
||||||
|
ICuratedDataService::LoadForenamesByCountry() /
|
||||||
|
LoadSurnamesByCountry() each return a flat
|
||||||
|
std::unordered_map<std::string, forename_list>
|
||||||
|
or <std::string, surname_list> keyed by ISO
|
||||||
|
3166-1 country code -- no NamesByCountry
|
||||||
|
wrapper class. SampleName(forenames_by_country,
|
||||||
|
surnames_by_country, iso3166_1, rng) is a free
|
||||||
|
function, not a method, so it can be called by
|
||||||
|
any per-city worker without owning the maps.
|
||||||
|
Sourced from popular-names-by-country-dataset
|
||||||
|
(CC0) -- see ETHICS-AND-KNOWN-ISSUES.md. Gender
|
||||||
|
is preserved per forename so it can be threaded
|
||||||
|
through to persona/bio generation later; surnames
|
||||||
|
carry no gender in the source data. Pairing
|
||||||
|
happens at sample time, not at fixture-authoring
|
||||||
|
time, so no information from the source dataset
|
||||||
|
is discarded up front.
|
||||||
|
end note
|
||||||
|
|
||||||
|
ForenameEntry ..> Name : SampleName() produces
|
||||||
|
|
||||||
class CheckinResult {
|
class CheckinResult {
|
||||||
+ checked_in_at : std::string
|
+ checked_in_at : std::string
|
||||||
+ note : std::string
|
+ note : std::string
|
||||||
@@ -91,7 +153,8 @@ package "Domain: Models" {
|
|||||||
|
|
||||||
class GeneratedBrewery {
|
class GeneratedBrewery {
|
||||||
+ brewery_id : uint64_t
|
+ brewery_id : uint64_t
|
||||||
+ location : Location
|
+ location : City
|
||||||
|
+ address : Address
|
||||||
+ brewery : BreweryResult
|
+ brewery : BreweryResult
|
||||||
+ context_completeness : LocationContext::Completeness
|
+ context_completeness : LocationContext::Completeness
|
||||||
+ metadata : GenerationMetadata
|
+ metadata : GenerationMetadata
|
||||||
@@ -100,7 +163,7 @@ package "Domain: Models" {
|
|||||||
class GeneratedBeer {
|
class GeneratedBeer {
|
||||||
+ beer_id : uint64_t
|
+ beer_id : uint64_t
|
||||||
+ brewery_id : uint64_t
|
+ brewery_id : uint64_t
|
||||||
+ location : Location
|
+ location : City
|
||||||
+ style : BeerStyle
|
+ style : BeerStyle
|
||||||
+ beer : BeerResult
|
+ beer : BeerResult
|
||||||
+ metadata : GenerationMetadata
|
+ metadata : GenerationMetadata
|
||||||
@@ -108,11 +171,22 @@ package "Domain: Models" {
|
|||||||
|
|
||||||
class GeneratedUser {
|
class GeneratedUser {
|
||||||
+ user_id : uint64_t
|
+ user_id : uint64_t
|
||||||
+ location : Location
|
+ location : City
|
||||||
+ user : UserResult
|
+ user : UserResult
|
||||||
|
+ email : std::string
|
||||||
|
+ date_of_birth : std::string
|
||||||
|
+ password : std::string
|
||||||
+ metadata : GenerationMetadata
|
+ metadata : GenerationMetadata
|
||||||
}
|
}
|
||||||
|
|
||||||
|
note right of GeneratedUser
|
||||||
|
email, date_of_birth, and password are
|
||||||
|
programmatically generated by the orchestrator
|
||||||
|
(not LLM-authored) so a downstream auth-account
|
||||||
|
seeding consumer can register real accounts from
|
||||||
|
this export. See ROADMAP.md §1.
|
||||||
|
end note
|
||||||
|
|
||||||
class GeneratedCheckin {
|
class GeneratedCheckin {
|
||||||
+ checkin_id : uint64_t
|
+ checkin_id : uint64_t
|
||||||
+ user_id : uint64_t
|
+ user_id : uint64_t
|
||||||
@@ -177,27 +251,27 @@ package "Domain: Application Configuration" {
|
|||||||
package "Domain: Policy" {
|
package "Domain: Policy" {
|
||||||
|
|
||||||
interface ContextStrategy <<interface>> {
|
interface ContextStrategy <<interface>> {
|
||||||
+ QueriesFor(loc : const Location&) : std::vector<std::string>
|
+ QueriesFor(loc : const City&) : std::vector<std::string>
|
||||||
+ MaxContextChars() : size_t
|
+ MaxContextChars() : size_t
|
||||||
}
|
}
|
||||||
|
|
||||||
class BreweryContextStrategy {
|
class BreweryContextStrategy {
|
||||||
+ QueriesFor(loc : const Location&) : std::vector<std::string>
|
+ QueriesFor(loc : const City&) : std::vector<std::string>
|
||||||
+ MaxContextChars() : size_t
|
+ MaxContextChars() : size_t
|
||||||
}
|
}
|
||||||
|
|
||||||
class BeerContextStrategy {
|
class BeerContextStrategy {
|
||||||
+ QueriesFor(loc : const Location&) : std::vector<std::string>
|
+ QueriesFor(loc : const City&) : std::vector<std::string>
|
||||||
+ MaxContextChars() : size_t
|
+ MaxContextChars() : size_t
|
||||||
}
|
}
|
||||||
|
|
||||||
interface SamplingStrategy <<interface>> {
|
interface SamplingStrategy <<interface>> {
|
||||||
+ Sample(locations : const std::vector<Location>&) : std::vector<Location>
|
+ Sample(locations : const std::vector<City>&) : std::vector<City>
|
||||||
}
|
}
|
||||||
|
|
||||||
class UniformSamplingStrategy {
|
class UniformSamplingStrategy {
|
||||||
- sample_size_ : size_t
|
- sample_size_ : size_t
|
||||||
+ Sample(locations : const std::vector<Location>&) : std::vector<Location>
|
+ Sample(locations : const std::vector<City>&) : std::vector<City>
|
||||||
}
|
}
|
||||||
|
|
||||||
interface BeerSelectionStrategy <<interface>> {
|
interface BeerSelectionStrategy <<interface>> {
|
||||||
@@ -318,32 +392,55 @@ package "Infrastructure: Pipeline Channel" {
|
|||||||
|
|
||||||
package "Infrastructure: Data Preloading" {
|
package "Infrastructure: Data Preloading" {
|
||||||
|
|
||||||
interface DataPreloader <<interface>> {
|
interface ICuratedDataService <<interface>> {
|
||||||
+ LoadLocations(filepath : const std::filesystem::path&) : std::vector<Location>
|
+ LoadCities(filepath : const std::filesystem::path&) : const std::vector<City>&
|
||||||
+ LoadBeerStyles(filepath : const std::filesystem::path&) : std::vector<BeerStyle>
|
+ LoadBeerStyles(filepath : const std::filesystem::path&) : const std::vector<BeerStyle>&
|
||||||
+ LoadPersonas(filepath : const std::filesystem::path&) : std::vector<Persona>
|
+ LoadPersonas(filepath : const std::filesystem::path&) : const std::vector<UserPersona>&
|
||||||
+ LoadNamesByCountry(filepath : const std::filesystem::path&) : NamesByCountry
|
+ LoadForenamesByCountry(forenames_filepath : const std::filesystem::path&) : const std::unordered_map<std::string, forename_list>&
|
||||||
|
+ LoadSurnamesByCountry(surnames_filepath : const std::filesystem::path&) : const std::unordered_map<std::string, surname_list>&
|
||||||
}
|
}
|
||||||
|
|
||||||
class JsonLoader {
|
class JsonLoader {
|
||||||
+ LoadLocations(filepath : const std::filesystem::path&) : std::vector<Location>
|
+ LoadCities(filepath : const std::filesystem::path&) : const std::vector<City>&
|
||||||
+ LoadBeerStyles(filepath : const std::filesystem::path&) : std::vector<BeerStyle>
|
+ LoadBeerStyles(filepath : const std::filesystem::path&) : const std::vector<BeerStyle>&
|
||||||
+ LoadPersonas(filepath : const std::filesystem::path&) : std::vector<Persona>
|
+ LoadPersonas(filepath : const std::filesystem::path&) : const std::vector<UserPersona>&
|
||||||
+ LoadNamesByCountry(filepath : const std::filesystem::path&) : NamesByCountry
|
+ LoadForenamesByCountry(forenames_filepath : const std::filesystem::path&) : const std::unordered_map<std::string, forename_list>&
|
||||||
|
+ LoadSurnamesByCountry(surnames_filepath : const std::filesystem::path&) : const std::unordered_map<std::string, surname_list>&
|
||||||
}
|
}
|
||||||
|
|
||||||
|
note right of JsonLoader
|
||||||
|
Each Load* call is memoized after its first
|
||||||
|
parse -- implemented today, ahead of the rest
|
||||||
|
of this package (LoadBeerStyles is still planned).
|
||||||
|
end note
|
||||||
|
|
||||||
|
class MockCuratedDataService {
|
||||||
|
+ LoadCities(filepath : const std::filesystem::path&) : const std::vector<City>&
|
||||||
|
+ LoadBeerStyles(filepath : const std::filesystem::path&) : const std::vector<BeerStyle>&
|
||||||
|
+ LoadPersonas(filepath : const std::filesystem::path&) : const std::vector<UserPersona>&
|
||||||
|
+ LoadForenamesByCountry(forenames_filepath : const std::filesystem::path&) : const std::unordered_map<std::string, forename_list>&
|
||||||
|
+ LoadSurnamesByCountry(surnames_filepath : const std::filesystem::path&) : const std::unordered_map<std::string, surname_list>&
|
||||||
|
}
|
||||||
|
|
||||||
|
note right of MockCuratedDataService
|
||||||
|
Fixed in-memory dataset used in --mocked mode,
|
||||||
|
implemented today for Locations/Personas/
|
||||||
|
Forenames/Surnames; would need a LoadBeerStyles
|
||||||
|
fixture too once that method is implemented.
|
||||||
|
end note
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
package "Infrastructure: Enrichment" {
|
package "Infrastructure: Enrichment" {
|
||||||
|
|
||||||
interface EnrichmentService <<interface>> {
|
interface EnrichmentService <<interface>> {
|
||||||
+ GetLocationContext(loc : const Location&,\n strategy : const ContextStrategy&) : LocationContext
|
+ GetLocationContext(loc : const City&,\n strategy : const ContextStrategy&) : LocationContext
|
||||||
}
|
}
|
||||||
|
|
||||||
class WikipediaService {
|
class WikipediaService {
|
||||||
- client_ : std::unique_ptr<WebClient>
|
- client_ : std::unique_ptr<WebClient>
|
||||||
- extract_cache_ : std::unordered_map<std::string, std::string>
|
- extract_cache_ : std::unordered_map<std::string, std::string>
|
||||||
+ GetLocationContext(loc : const Location&,\n strategy : const ContextStrategy&) : LocationContext
|
+ GetLocationContext(loc : const City&,\n strategy : const ContextStrategy&) : LocationContext
|
||||||
- FetchExtract(query : std::string_view) : std::string
|
- FetchExtract(query : std::string_view) : std::string
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -359,6 +456,39 @@ package "Infrastructure: Enrichment" {
|
|||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
package "Infrastructure: Postal Code Generation" {
|
||||||
|
|
||||||
|
interface IPostalCodeService <<interface>> {
|
||||||
|
+ GeneratePostalCode(location : const City&) : std::string
|
||||||
|
}
|
||||||
|
|
||||||
|
class MockPostalCodeService {
|
||||||
|
+ GeneratePostalCode(location : const City&) : std::string
|
||||||
|
}
|
||||||
|
|
||||||
|
class XegerPostalCodeService {
|
||||||
|
+ GeneratePostalCode(location : const City&) : std::string
|
||||||
|
}
|
||||||
|
|
||||||
|
note right of MockPostalCodeService
|
||||||
|
Implemented today: ignores postal_regex and
|
||||||
|
returns postal_code_examples.front(). Not yet
|
||||||
|
bound in main.cc's DI graph or used by
|
||||||
|
BiergartenPipelineOrchestrator.
|
||||||
|
end note
|
||||||
|
|
||||||
|
note right of XegerPostalCodeService
|
||||||
|
Not implemented yet. Picks one of
|
||||||
|
location.postal_regex and generates a random
|
||||||
|
string matching it (a "xeger" -- the inverse of
|
||||||
|
regex matching) instead of always replaying a
|
||||||
|
fixed example. See ROADMAP.md §9.
|
||||||
|
end note
|
||||||
|
|
||||||
|
IPostalCodeService <|.. MockPostalCodeService
|
||||||
|
IPostalCodeService <|.. XegerPostalCodeService
|
||||||
|
}
|
||||||
|
|
||||||
package "Infrastructure: Prompting" {
|
package "Infrastructure: Prompting" {
|
||||||
|
|
||||||
interface IPromptDirectory <<interface>> {
|
interface IPromptDirectory <<interface>> {
|
||||||
@@ -378,9 +508,9 @@ package "Infrastructure: Prompting" {
|
|||||||
package "Infrastructure: Data Generation" {
|
package "Infrastructure: Data Generation" {
|
||||||
|
|
||||||
interface DataGenerator <<interface>> {
|
interface DataGenerator <<interface>> {
|
||||||
+ GenerateBrewery(location : const Location&,\n context : const LocationContext&) : BreweryResult
|
+ GenerateBrewery(location : const City&,\n context : const LocationContext&) : BreweryResult
|
||||||
+ GenerateBeer(brewery_id : uint64_t,\n location : const Location&,\n context : const LocationContext&,\n style : const BeerStyle&) : BeerResult
|
+ GenerateBeer(brewery_id : uint64_t,\n location : const City&,\n context : const LocationContext&,\n style : const BeerStyle&) : BeerResult
|
||||||
+ GenerateUser(location : const Location&) : UserResult
|
+ GenerateUser(city : const EnrichedCity&,\n persona : const UserPersona&,\n name : const Name&) : UserResult
|
||||||
+ GenerateCheckin(user : const GeneratedUser&,\n brewery : const GeneratedBrewery&,\n timestamp : const std::string&) : CheckinResult
|
+ GenerateCheckin(user : const GeneratedUser&,\n brewery : const GeneratedBrewery&,\n timestamp : const std::string&) : CheckinResult
|
||||||
+ GenerateRating(user : const GeneratedUser&,\n beer : const GeneratedBeer&,\n checkin_id : uint64_t) : RatingResult
|
+ GenerateRating(user : const GeneratedUser&,\n beer : const GeneratedBeer&,\n checkin_id : uint64_t) : RatingResult
|
||||||
}
|
}
|
||||||
@@ -391,7 +521,7 @@ package "Infrastructure: Data Generation" {
|
|||||||
+ GenerateUser(...) : UserResult
|
+ GenerateUser(...) : UserResult
|
||||||
+ GenerateCheckin(...) : CheckinResult
|
+ GenerateCheckin(...) : CheckinResult
|
||||||
+ GenerateRating(...) : RatingResult
|
+ GenerateRating(...) : RatingResult
|
||||||
- DeterministicHash(location : const Location&) : size_t
|
- DeterministicHash(location : const City&) : size_t
|
||||||
}
|
}
|
||||||
|
|
||||||
class LlamaGenerator {
|
class LlamaGenerator {
|
||||||
@@ -473,7 +603,7 @@ package "Infrastructure: Data Export" {
|
|||||||
}
|
}
|
||||||
|
|
||||||
class BiergartenPipelineOrchestrator {
|
class BiergartenPipelineOrchestrator {
|
||||||
- preloader_ : std::unique_ptr<DataPreloader>
|
- curated_data_service_ : std::unique_ptr<ICuratedDataService>
|
||||||
- enrichment_service_ : std::unique_ptr<EnrichmentService>
|
- enrichment_service_ : std::unique_ptr<EnrichmentService>
|
||||||
- generator_ : std::unique_ptr<DataGenerator>
|
- generator_ : std::unique_ptr<DataGenerator>
|
||||||
- logger_ : std::unique_ptr<Logger>
|
- logger_ : std::unique_ptr<Logger>
|
||||||
@@ -493,15 +623,15 @@ class BiergartenPipelineOrchestrator {
|
|||||||
- follow_pool_ : std::vector<GeneratedFollow>
|
- follow_pool_ : std::vector<GeneratedFollow>
|
||||||
--
|
--
|
||||||
+ Run() : bool
|
+ Run() : bool
|
||||||
- RunUserPhase(locations : const std::vector<Location>&) : void
|
- RunUserPhase(locations : const std::vector<City>&) : void
|
||||||
- RunBreweryAndBeerPhase(locations : const std::vector<Location>&) : void
|
- RunBreweryAndBeerPhase(locations : const std::vector<City>&) : void
|
||||||
- RunCheckinPhase() : void
|
- RunCheckinPhase() : void
|
||||||
- RunRatingPhase() : void
|
- RunRatingPhase() : void
|
||||||
- RunFollowPhase() : void
|
- RunFollowPhase() : void
|
||||||
}
|
}
|
||||||
|
|
||||||
' --- Orchestration Aggregations (Services & Strategies) ---
|
' --- Orchestration Aggregations (Services & Strategies) ---
|
||||||
BiergartenPipelineOrchestrator *-- DataPreloader
|
BiergartenPipelineOrchestrator *-- ICuratedDataService
|
||||||
BiergartenPipelineOrchestrator *-- EnrichmentService
|
BiergartenPipelineOrchestrator *-- EnrichmentService
|
||||||
BiergartenPipelineOrchestrator *-- DataGenerator
|
BiergartenPipelineOrchestrator *-- DataGenerator
|
||||||
BiergartenPipelineOrchestrator *-- ExportService
|
BiergartenPipelineOrchestrator *-- ExportService
|
||||||
@@ -520,7 +650,8 @@ BiergartenPipelineOrchestrator *-- "0..*" GeneratedCheckin : checkin_pool_
|
|||||||
BiergartenPipelineOrchestrator *-- "0..*" GeneratedFollow : follow_pool_
|
BiergartenPipelineOrchestrator *-- "0..*" GeneratedFollow : follow_pool_
|
||||||
|
|
||||||
' --- Interfaces & Implementations ---
|
' --- Interfaces & Implementations ---
|
||||||
DataPreloader <|.. JsonLoader
|
ICuratedDataService <|.. JsonLoader
|
||||||
|
ICuratedDataService <|.. MockCuratedDataService
|
||||||
Logger <|.. PipelineLogger
|
Logger <|.. PipelineLogger
|
||||||
ContextStrategy <|.. BreweryContextStrategy
|
ContextStrategy <|.. BreweryContextStrategy
|
||||||
ContextStrategy <|.. BeerContextStrategy
|
ContextStrategy <|.. BeerContextStrategy
|
||||||
@@ -551,16 +682,17 @@ PipelineLogger o-- BoundedChannel : logs to
|
|||||||
LogWorker o-- BoundedChannel : drains from
|
LogWorker o-- BoundedChannel : drains from
|
||||||
|
|
||||||
' --- Domain Containment ---
|
' --- Domain Containment ---
|
||||||
EnrichedCity *-- Location
|
EnrichedCity *-- City
|
||||||
EnrichedCity *-- LocationContext
|
EnrichedCity *-- LocationContext
|
||||||
GeneratedBrewery *-- Location
|
GeneratedBrewery *-- City
|
||||||
|
GeneratedBrewery *-- Address
|
||||||
GeneratedBrewery *-- BreweryResult
|
GeneratedBrewery *-- BreweryResult
|
||||||
GeneratedBrewery *-- GenerationMetadata
|
GeneratedBrewery *-- GenerationMetadata
|
||||||
GeneratedBeer *-- Location
|
GeneratedBeer *-- City
|
||||||
GeneratedBeer *-- BeerStyle
|
GeneratedBeer *-- BeerStyle
|
||||||
GeneratedBeer *-- BeerResult
|
GeneratedBeer *-- BeerResult
|
||||||
GeneratedBeer *-- GenerationMetadata
|
GeneratedBeer *-- GenerationMetadata
|
||||||
GeneratedUser *-- Location
|
GeneratedUser *-- City
|
||||||
GeneratedUser *-- UserResult
|
GeneratedUser *-- UserResult
|
||||||
GeneratedUser *-- GenerationMetadata
|
GeneratedUser *-- GenerationMetadata
|
||||||
GeneratedCheckin *-- CheckinResult
|
GeneratedCheckin *-- CheckinResult
|
||||||
|
|||||||
|
Before Width: | Height: | Size: 106 KiB After Width: | Height: | Size: 142 KiB |
|
Before Width: | Height: | Size: 184 KiB After Width: | Height: | Size: 257 KiB |
24
tooling/pipeline-results-viewer/.gitignore
vendored
Normal file
@@ -0,0 +1,24 @@
|
|||||||
|
# Logs
|
||||||
|
logs
|
||||||
|
*.log
|
||||||
|
npm-debug.log*
|
||||||
|
yarn-debug.log*
|
||||||
|
yarn-error.log*
|
||||||
|
pnpm-debug.log*
|
||||||
|
lerna-debug.log*
|
||||||
|
|
||||||
|
node_modules
|
||||||
|
dist
|
||||||
|
dist-ssr
|
||||||
|
*.local
|
||||||
|
|
||||||
|
# Editor directories and files
|
||||||
|
.vscode/*
|
||||||
|
!.vscode/extensions.json
|
||||||
|
.idea
|
||||||
|
.DS_Store
|
||||||
|
*.suo
|
||||||
|
*.ntvs*
|
||||||
|
*.njsproj
|
||||||
|
*.sln
|
||||||
|
*.sw?
|
||||||
74
tooling/pipeline-results-viewer/index.html
Normal file
@@ -0,0 +1,74 @@
|
|||||||
|
<!doctype html>
|
||||||
|
<html lang="en">
|
||||||
|
<head>
|
||||||
|
<meta charset="UTF-8" />
|
||||||
|
<link rel="icon" type="image/svg+xml" href="/favicon.svg" />
|
||||||
|
<meta name="viewport" content="width=device-width, initial-scale=1.0" />
|
||||||
|
<title>Pipeline Results Viewer</title>
|
||||||
|
</head>
|
||||||
|
<body>
|
||||||
|
<div class="dialog" id="window">
|
||||||
|
<div class="dialog-header">
|
||||||
|
<p>Pipeline Results Viewer</p>
|
||||||
|
</div>
|
||||||
|
<div class="tab-strip" role="tablist">
|
||||||
|
<button
|
||||||
|
type="button"
|
||||||
|
class="tab is-active"
|
||||||
|
id="tab-tables"
|
||||||
|
role="tab"
|
||||||
|
aria-selected="true"
|
||||||
|
aria-controls="panel-tables"
|
||||||
|
>
|
||||||
|
Tables
|
||||||
|
</button>
|
||||||
|
<button
|
||||||
|
type="button"
|
||||||
|
class="tab"
|
||||||
|
id="tab-query"
|
||||||
|
role="tab"
|
||||||
|
aria-selected="false"
|
||||||
|
aria-controls="panel-query"
|
||||||
|
>
|
||||||
|
Query
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
<div class="dialog-body">
|
||||||
|
<div
|
||||||
|
class="tab-panel"
|
||||||
|
id="panel-tables"
|
||||||
|
role="tabpanel"
|
||||||
|
aria-labelledby="tab-tables"
|
||||||
|
>
|
||||||
|
<div class="field-group">
|
||||||
|
<label for="table-select">Table</label>
|
||||||
|
<select id="table-select"></select>
|
||||||
|
</div>
|
||||||
|
<p id="status"></p>
|
||||||
|
<div id="table-container"></div>
|
||||||
|
</div>
|
||||||
|
<div
|
||||||
|
class="tab-panel"
|
||||||
|
id="panel-query"
|
||||||
|
role="tabpanel"
|
||||||
|
aria-labelledby="tab-query"
|
||||||
|
hidden
|
||||||
|
>
|
||||||
|
<div class="field-group field-group--stacked">
|
||||||
|
<label for="query-input">SQL Query</label>
|
||||||
|
<textarea
|
||||||
|
id="query-input"
|
||||||
|
rows="4"
|
||||||
|
spellcheck="false"
|
||||||
|
placeholder="SELECT * FROM ..."
|
||||||
|
></textarea>
|
||||||
|
</div>
|
||||||
|
<button type="button" id="query-run" class="btn">Run Query</button>
|
||||||
|
<p id="query-status"></p>
|
||||||
|
<div id="query-container"></div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
<script type="module" src="/src/main.ts"></script>
|
||||||
|
</body>
|
||||||
|
</html>
|
||||||
1334
tooling/pipeline-results-viewer/package-lock.json
generated
Normal file
20
tooling/pipeline-results-viewer/package.json
Normal file
@@ -0,0 +1,20 @@
|
|||||||
|
{
|
||||||
|
"name": "pipeline-results-viewer",
|
||||||
|
"version": "0.0.0",
|
||||||
|
"private": true,
|
||||||
|
"type": "module",
|
||||||
|
"scripts": {
|
||||||
|
"dev": "vite",
|
||||||
|
"build": "tsc && vite build",
|
||||||
|
"preview": "vite preview"
|
||||||
|
},
|
||||||
|
"devDependencies": {
|
||||||
|
"@types/sql.js": "^1.4.11",
|
||||||
|
"sass": "^1.101.0",
|
||||||
|
"typescript": "^6.0.3",
|
||||||
|
"vite": "^8.0.12"
|
||||||
|
},
|
||||||
|
"dependencies": {
|
||||||
|
"sql.js": "^1.14.1"
|
||||||
|
}
|
||||||
|
}
|
||||||
BIN
tooling/pipeline-results-viewer/public/biergarten.sqlite
Normal file
1
tooling/pipeline-results-viewer/public/favicon.svg
Normal file
|
After Width: | Height: | Size: 9.3 KiB |
147
tooling/pipeline-results-viewer/src/main.ts
Normal file
@@ -0,0 +1,147 @@
|
|||||||
|
import initSqlJs, { type Database, type SqlValue } from "sql.js";
|
||||||
|
import sqlWasmUrl from "sql.js/dist/sql-wasm.wasm?url";
|
||||||
|
import "./style.scss";
|
||||||
|
|
||||||
|
const DB_URL = "/biergarten.sqlite";
|
||||||
|
|
||||||
|
const tabs = document.querySelectorAll<HTMLButtonElement>("[role='tab']");
|
||||||
|
const panels = document.querySelectorAll<HTMLDivElement>("[role='tabpanel']");
|
||||||
|
|
||||||
|
const tableSelect = document.querySelector<HTMLSelectElement>("#table-select")!;
|
||||||
|
const tableContainer =
|
||||||
|
document.querySelector<HTMLDivElement>("#table-container")!;
|
||||||
|
const status = document.querySelector<HTMLParagraphElement>("#status")!;
|
||||||
|
|
||||||
|
const queryInput = document.querySelector<HTMLTextAreaElement>("#query-input")!;
|
||||||
|
const queryRun = document.querySelector<HTMLButtonElement>("#query-run")!;
|
||||||
|
const queryStatus = document.querySelector<HTMLParagraphElement>("#query-status")!;
|
||||||
|
const queryContainer =
|
||||||
|
document.querySelector<HTMLDivElement>("#query-container")!;
|
||||||
|
|
||||||
|
for (const tab of tabs) {
|
||||||
|
tab.addEventListener("click", () => {
|
||||||
|
for (const otherTab of tabs) {
|
||||||
|
const isActive = otherTab === tab;
|
||||||
|
otherTab.classList.toggle("is-active", isActive);
|
||||||
|
otherTab.setAttribute("aria-selected", String(isActive));
|
||||||
|
}
|
||||||
|
for (const panel of panels) {
|
||||||
|
panel.hidden = panel.id !== tab.getAttribute("aria-controls");
|
||||||
|
}
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
const renderResultTable = (
|
||||||
|
container: HTMLDivElement,
|
||||||
|
columns: string[],
|
||||||
|
values: SqlValue[][],
|
||||||
|
) => {
|
||||||
|
const table = document.createElement("table");
|
||||||
|
|
||||||
|
const head = table.createTHead().insertRow();
|
||||||
|
for (const column of columns) {
|
||||||
|
const th = document.createElement("th");
|
||||||
|
th.textContent = column;
|
||||||
|
head.append(th);
|
||||||
|
}
|
||||||
|
|
||||||
|
const body = table.createTBody();
|
||||||
|
for (const row of values) {
|
||||||
|
const tr = body.insertRow();
|
||||||
|
for (const cell of row) {
|
||||||
|
const td = document.createElement("td");
|
||||||
|
td.textContent = String(cell ?? "");
|
||||||
|
tr.append(td);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
container.append(table);
|
||||||
|
};
|
||||||
|
|
||||||
|
const renderTable = (db: Database, name: string) => {
|
||||||
|
const result = db.exec(`SELECT * FROM "${name}"`);
|
||||||
|
tableContainer.replaceChildren();
|
||||||
|
|
||||||
|
if (result.length === 0) {
|
||||||
|
tableContainer.append(
|
||||||
|
Object.assign(document.createElement("p"), { textContent: "No rows." }),
|
||||||
|
);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const { columns, values } = result[0];
|
||||||
|
renderResultTable(tableContainer, columns, values);
|
||||||
|
};
|
||||||
|
|
||||||
|
const runQuery = (db: Database) => {
|
||||||
|
const sql = queryInput.value.trim();
|
||||||
|
queryContainer.replaceChildren();
|
||||||
|
|
||||||
|
if (!sql) {
|
||||||
|
queryStatus.textContent = "Enter a query to run.";
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
try {
|
||||||
|
const results = db.exec(sql);
|
||||||
|
|
||||||
|
if (results.length === 0) {
|
||||||
|
queryStatus.textContent = "Query executed. No rows returned.";
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const { columns, values } = results[results.length - 1];
|
||||||
|
renderResultTable(queryContainer, columns, values);
|
||||||
|
queryStatus.textContent = `${values.length} row${values.length === 1 ? "" : "s"} returned.`;
|
||||||
|
} catch (error) {
|
||||||
|
queryStatus.textContent = `Error: ${error instanceof Error ? error.message : String(error)}`;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
async function main(): Promise<void> {
|
||||||
|
status.textContent = "Loading database...";
|
||||||
|
|
||||||
|
const sqlJs = await initSqlJs({ locateFile: () => sqlWasmUrl });
|
||||||
|
const response = await fetch(DB_URL);
|
||||||
|
const buffer = await response.arrayBuffer();
|
||||||
|
const db = new sqlJs.Database(new Uint8Array(buffer));
|
||||||
|
|
||||||
|
const tables =
|
||||||
|
db
|
||||||
|
.exec(
|
||||||
|
`SELECT name
|
||||||
|
FROM sqlite_master
|
||||||
|
WHERE
|
||||||
|
type = 'table'
|
||||||
|
AND
|
||||||
|
name NOT LIKE 'sqlite_%'
|
||||||
|
ORDER BY name`,
|
||||||
|
)[0]
|
||||||
|
?.values.map((row) => String(row[0])) ?? [];
|
||||||
|
|
||||||
|
for (const name of tables) {
|
||||||
|
tableSelect.append(new Option(name, name));
|
||||||
|
}
|
||||||
|
|
||||||
|
tableSelect.addEventListener("change", () =>
|
||||||
|
renderTable(db, tableSelect.value),
|
||||||
|
);
|
||||||
|
|
||||||
|
if (tables.length > 0) {
|
||||||
|
renderTable(db, tables[0]);
|
||||||
|
}
|
||||||
|
|
||||||
|
status.textContent = "";
|
||||||
|
|
||||||
|
queryRun.addEventListener("click", () => runQuery(db));
|
||||||
|
queryInput.addEventListener("keydown", (event) => {
|
||||||
|
if ((event.metaKey || event.ctrlKey) && event.key === "Enter") {
|
||||||
|
event.preventDefault();
|
||||||
|
runQuery(db);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
main().catch((error: unknown) => {
|
||||||
|
status.textContent = `Failed to load database: ${String(error)}`;
|
||||||
|
});
|
||||||
267
tooling/pipeline-results-viewer/src/style.scss
Normal file
@@ -0,0 +1,267 @@
|
|||||||
|
$gray-100: #dedede;
|
||||||
|
$gray-300: #c3c3c3;
|
||||||
|
$gray-500: #808080;
|
||||||
|
$white: #ffffff;
|
||||||
|
$black: #000000;
|
||||||
|
$navy: #000080;
|
||||||
|
$blue: #1084d0;
|
||||||
|
$teal: #008080;
|
||||||
|
|
||||||
|
$color-surface: $gray-300;
|
||||||
|
$color-surface-alt: $gray-100;
|
||||||
|
$color-surface-sunken: $white;
|
||||||
|
$color-border-highlight: $gray-100;
|
||||||
|
$color-border-lowlight: $gray-500;
|
||||||
|
$color-border-light: $white;
|
||||||
|
$color-border-dark: $black;
|
||||||
|
$color-text: $black;
|
||||||
|
$color-text-on-accent: $white;
|
||||||
|
$color-text-status: $navy;
|
||||||
|
$color-accent-start: $navy;
|
||||||
|
$color-accent-end: $blue;
|
||||||
|
$color-desktop-bg: $teal;
|
||||||
|
$color-row-stripe: $gray-100;
|
||||||
|
$color-shadow: rgba($black, 0.4);
|
||||||
|
$color-focus-ring: $black;
|
||||||
|
|
||||||
|
$bevel-size: 2px;
|
||||||
|
$font-stack: Tahoma, "MS Sans Serif", Verdana, sans-serif;
|
||||||
|
|
||||||
|
@mixin bevel-raised($size: $bevel-size) {
|
||||||
|
border-style: solid;
|
||||||
|
border-width: $size;
|
||||||
|
border-top-color: $color-border-highlight;
|
||||||
|
border-left-color: $color-border-highlight;
|
||||||
|
border-right-color: $color-border-dark;
|
||||||
|
border-bottom-color: $color-border-dark;
|
||||||
|
box-shadow:
|
||||||
|
inset $size $size 0 0 $color-border-light,
|
||||||
|
inset (-$size) (-$size) 0 0 $color-border-lowlight;
|
||||||
|
}
|
||||||
|
|
||||||
|
@mixin bevel-sunken($size: $bevel-size) {
|
||||||
|
border-style: solid;
|
||||||
|
border-width: $size;
|
||||||
|
border-top-color: $color-border-lowlight;
|
||||||
|
border-left-color: $color-border-lowlight;
|
||||||
|
border-right-color: $color-border-light;
|
||||||
|
border-bottom-color: $color-border-light;
|
||||||
|
box-shadow:
|
||||||
|
inset $size $size 0 0 $color-border-dark,
|
||||||
|
inset (-$size) (-$size) 0 0 $color-surface-alt;
|
||||||
|
}
|
||||||
|
|
||||||
|
* {
|
||||||
|
box-sizing: border-box;
|
||||||
|
}
|
||||||
|
|
||||||
|
body {
|
||||||
|
margin: 0;
|
||||||
|
padding: 24px;
|
||||||
|
background: $color-desktop-bg;
|
||||||
|
font-family: $font-stack;
|
||||||
|
color: $color-text;
|
||||||
|
min-height: 100vh;
|
||||||
|
}
|
||||||
|
|
||||||
|
#window {
|
||||||
|
width: 100%;
|
||||||
|
max-width: 1600px;
|
||||||
|
margin: 0 auto;
|
||||||
|
background: $color-surface;
|
||||||
|
padding: 6px;
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
gap: 6px;
|
||||||
|
@include bevel-raised(3px);
|
||||||
|
box-shadow: 8px 8px 0 0 $color-shadow;
|
||||||
|
}
|
||||||
|
|
||||||
|
.dialog-header {
|
||||||
|
background: linear-gradient(90deg, $color-accent-start, $color-accent-end);
|
||||||
|
color: $color-text-on-accent;
|
||||||
|
padding: 6px 8px;
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
|
||||||
|
p {
|
||||||
|
margin: 0;
|
||||||
|
font-size: 14px;
|
||||||
|
font-weight: bold;
|
||||||
|
letter-spacing: 0.5px;
|
||||||
|
text-transform: uppercase;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.tab-strip {
|
||||||
|
display: flex;
|
||||||
|
gap: 4px;
|
||||||
|
padding: 8px 8px 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
.tab {
|
||||||
|
position: relative;
|
||||||
|
top: 3px;
|
||||||
|
margin-bottom: -2px;
|
||||||
|
font-family: $font-stack;
|
||||||
|
font-size: 12px;
|
||||||
|
font-weight: bold;
|
||||||
|
text-transform: uppercase;
|
||||||
|
padding: 6px 14px;
|
||||||
|
background: $color-surface;
|
||||||
|
color: $color-text;
|
||||||
|
border-style: solid;
|
||||||
|
border-width: 2px;
|
||||||
|
border-top-color: $color-border-light;
|
||||||
|
border-left-color: $color-border-light;
|
||||||
|
border-right-color: $color-border-dark;
|
||||||
|
border-bottom-color: $color-border-dark;
|
||||||
|
border-radius: 0;
|
||||||
|
cursor: pointer;
|
||||||
|
|
||||||
|
&.is-active {
|
||||||
|
top: 0;
|
||||||
|
z-index: 1;
|
||||||
|
border-bottom-color: $color-surface;
|
||||||
|
}
|
||||||
|
|
||||||
|
&:focus {
|
||||||
|
outline: 1px dotted $color-focus-ring;
|
||||||
|
outline-offset: -4px;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.dialog-body {
|
||||||
|
border-top: 2px solid $color-border-dark;
|
||||||
|
padding: 12px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.tab-panel {
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
gap: 12px;
|
||||||
|
|
||||||
|
&[hidden] {
|
||||||
|
display: none;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.field-group {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
gap: 8px;
|
||||||
|
|
||||||
|
label {
|
||||||
|
font-size: 12px;
|
||||||
|
font-weight: bold;
|
||||||
|
text-transform: uppercase;
|
||||||
|
}
|
||||||
|
|
||||||
|
&.field-group--stacked {
|
||||||
|
flex-direction: column;
|
||||||
|
align-items: stretch;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
select#table-select {
|
||||||
|
font-family: $font-stack;
|
||||||
|
font-size: 13px;
|
||||||
|
padding: 3px 6px;
|
||||||
|
background: $color-surface-sunken;
|
||||||
|
color: $color-text;
|
||||||
|
border-radius: 0;
|
||||||
|
@include bevel-sunken();
|
||||||
|
|
||||||
|
&:focus {
|
||||||
|
outline: 1px dotted $color-focus-ring;
|
||||||
|
outline-offset: 2px;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
textarea#query-input {
|
||||||
|
font-family: $font-stack;
|
||||||
|
font-size: 13px;
|
||||||
|
padding: 6px;
|
||||||
|
background: $color-surface-sunken;
|
||||||
|
color: $color-text;
|
||||||
|
border-radius: 0;
|
||||||
|
resize: vertical;
|
||||||
|
@include bevel-sunken();
|
||||||
|
|
||||||
|
&:focus {
|
||||||
|
outline: 1px dotted $color-focus-ring;
|
||||||
|
outline-offset: 2px;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
.btn {
|
||||||
|
align-self: flex-start;
|
||||||
|
font-family: $font-stack;
|
||||||
|
font-size: 12px;
|
||||||
|
font-weight: bold;
|
||||||
|
text-transform: uppercase;
|
||||||
|
padding: 6px 16px;
|
||||||
|
background: $color-surface;
|
||||||
|
color: $color-text;
|
||||||
|
cursor: pointer;
|
||||||
|
@include bevel-raised();
|
||||||
|
|
||||||
|
&:active {
|
||||||
|
@include bevel-sunken();
|
||||||
|
}
|
||||||
|
|
||||||
|
&:focus {
|
||||||
|
outline: 1px dotted $color-focus-ring;
|
||||||
|
outline-offset: -4px;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#status,
|
||||||
|
#query-status {
|
||||||
|
margin: 0;
|
||||||
|
font-size: 12px;
|
||||||
|
font-style: italic;
|
||||||
|
color: $color-text-status;
|
||||||
|
}
|
||||||
|
|
||||||
|
#table-container,
|
||||||
|
#query-container {
|
||||||
|
background: $color-surface-sunken;
|
||||||
|
padding: 4px;
|
||||||
|
overflow-x: auto;
|
||||||
|
@include bevel-sunken();
|
||||||
|
}
|
||||||
|
|
||||||
|
table {
|
||||||
|
border-collapse: collapse;
|
||||||
|
margin: 0 auto;
|
||||||
|
font-family: $font-stack;
|
||||||
|
}
|
||||||
|
|
||||||
|
th,
|
||||||
|
td {
|
||||||
|
border: 1px solid $color-border-lowlight;
|
||||||
|
padding: 4px 8px;
|
||||||
|
text-align: left;
|
||||||
|
font-size: 13px;
|
||||||
|
}
|
||||||
|
|
||||||
|
th {
|
||||||
|
background: $color-surface;
|
||||||
|
white-space: nowrap;
|
||||||
|
font-weight: bold;
|
||||||
|
text-transform: uppercase;
|
||||||
|
@include bevel-raised(1px);
|
||||||
|
}
|
||||||
|
|
||||||
|
td {
|
||||||
|
min-width: 120px;
|
||||||
|
max-width: 800px;
|
||||||
|
white-space: normal;
|
||||||
|
word-break: break-word;
|
||||||
|
background: $color-surface-sunken;
|
||||||
|
}
|
||||||
|
|
||||||
|
tbody tr:nth-child(even) td {
|
||||||
|
background: $color-row-stripe;
|
||||||
|
}
|
||||||
24
tooling/pipeline-results-viewer/tsconfig.json
Normal file
@@ -0,0 +1,24 @@
|
|||||||
|
{
|
||||||
|
"compilerOptions": {
|
||||||
|
"target": "es2023",
|
||||||
|
"module": "esnext",
|
||||||
|
"lib": ["ES2023", "DOM"],
|
||||||
|
"types": ["vite/client"],
|
||||||
|
"skipLibCheck": true,
|
||||||
|
|
||||||
|
/* Bundler mode */
|
||||||
|
"moduleResolution": "bundler",
|
||||||
|
"esModuleInterop": true,
|
||||||
|
"allowImportingTsExtensions": true,
|
||||||
|
"verbatimModuleSyntax": true,
|
||||||
|
"moduleDetection": "force",
|
||||||
|
"noEmit": true,
|
||||||
|
|
||||||
|
/* Linting */
|
||||||
|
"noUnusedLocals": true,
|
||||||
|
"noUnusedParameters": true,
|
||||||
|
"erasableSyntaxOnly": true,
|
||||||
|
"noFallthroughCasesInSwitch": true
|
||||||
|
},
|
||||||
|
"include": ["src"]
|
||||||
|
}
|
||||||
@@ -139,6 +139,9 @@ 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/services/curated_data/mock_curated_data_service.h
|
||||||
|
includes/services/postal_code/postal_code_service.h
|
||||||
|
includes/services/postal_code/mock_postal_code_service.h
|
||||||
includes/json_handling/pretty_print.h)
|
includes/json_handling/pretty_print.h)
|
||||||
|
|
||||||
# --- Entry point ---
|
# --- Entry point ---
|
||||||
@@ -148,7 +151,12 @@ target_sources(${PROJECT_NAME} PRIVATE
|
|||||||
|
|
||||||
# --- json_handling ---
|
# --- json_handling ---
|
||||||
target_sources(${PROJECT_NAME} PRIVATE
|
target_sources(${PROJECT_NAME} PRIVATE
|
||||||
src/json_handling/json_loader.cc
|
src/services/curated_data/curated_json_data_service.cc
|
||||||
|
)
|
||||||
|
|
||||||
|
# --- services: curated_data ---
|
||||||
|
target_sources(${PROJECT_NAME} PRIVATE
|
||||||
|
src/services/curated_data/mock_curated_data_service.cc
|
||||||
)
|
)
|
||||||
|
|
||||||
# --- application_options ---
|
# --- application_options ---
|
||||||
@@ -161,6 +169,7 @@ target_sources(${PROJECT_NAME} PRIVATE
|
|||||||
src/biergarten_pipeline_orchestrator/log_results.cc
|
src/biergarten_pipeline_orchestrator/log_results.cc
|
||||||
src/biergarten_pipeline_orchestrator/biergarten_pipeline_orchestrator.cc
|
src/biergarten_pipeline_orchestrator/biergarten_pipeline_orchestrator.cc
|
||||||
src/biergarten_pipeline_orchestrator/generate_breweries.cc
|
src/biergarten_pipeline_orchestrator/generate_breweries.cc
|
||||||
|
src/biergarten_pipeline_orchestrator/generate_users.cc
|
||||||
src/biergarten_pipeline_orchestrator/run.cc
|
src/biergarten_pipeline_orchestrator/run.cc
|
||||||
src/biergarten_pipeline_orchestrator/query_cities_with_countries.cc
|
src/biergarten_pipeline_orchestrator/query_cities_with_countries.cc
|
||||||
)
|
)
|
||||||
@@ -204,6 +213,7 @@ target_sources(${PROJECT_NAME} PRIVATE
|
|||||||
# --- services: sqlite ---
|
# --- services: sqlite ---
|
||||||
target_sources(${PROJECT_NAME} PRIVATE
|
target_sources(${PROJECT_NAME} PRIVATE
|
||||||
src/services/sqlite/process_record.cc
|
src/services/sqlite/process_record.cc
|
||||||
|
src/services/sqlite/process_user_record.cc
|
||||||
src/services/sqlite/sqlite_export_service.cc
|
src/services/sqlite/sqlite_export_service.cc
|
||||||
src/services/sqlite/finalize.cc
|
src/services/sqlite/finalize.cc
|
||||||
src/services/sqlite/initialize.cc
|
src/services/sqlite/initialize.cc
|
||||||
@@ -256,8 +266,23 @@ target_compile_options(biergarten-pipeline PRIVATE
|
|||||||
|
|
||||||
# 7. Runtime Assets
|
# 7. Runtime Assets
|
||||||
configure_file(
|
configure_file(
|
||||||
${CMAKE_SOURCE_DIR}/locations.json
|
${CMAKE_SOURCE_DIR}/cities.json
|
||||||
${CMAKE_BINARY_DIR}/locations.json
|
${CMAKE_BINARY_DIR}/cities.json
|
||||||
|
COPYONLY
|
||||||
|
)
|
||||||
|
configure_file(
|
||||||
|
${CMAKE_SOURCE_DIR}/personas.json
|
||||||
|
${CMAKE_BINARY_DIR}/personas.json
|
||||||
|
COPYONLY
|
||||||
|
)
|
||||||
|
configure_file(
|
||||||
|
${CMAKE_SOURCE_DIR}/forenames-by-country.json
|
||||||
|
${CMAKE_BINARY_DIR}/forenames-by-country.json
|
||||||
|
COPYONLY
|
||||||
|
)
|
||||||
|
configure_file(
|
||||||
|
${CMAKE_SOURCE_DIR}/surnames-by-country.json
|
||||||
|
${CMAKE_BINARY_DIR}/surnames-by-country.json
|
||||||
COPYONLY
|
COPYONLY
|
||||||
)
|
)
|
||||||
add_custom_command(TARGET ${PROJECT_NAME} POST_BUILD
|
add_custom_command(TARGET ${PROJECT_NAME} POST_BUILD
|
||||||
@@ -265,4 +290,3 @@ add_custom_command(TARGET ${PROJECT_NAME} POST_BUILD
|
|||||||
${CMAKE_SOURCE_DIR}/prompts
|
${CMAKE_SOURCE_DIR}/prompts
|
||||||
${CMAKE_BINARY_DIR}/prompts
|
${CMAKE_BINARY_DIR}/prompts
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|||||||
1718
tooling/pipeline/cities.json
Normal file
27168
tooling/pipeline/forenames-by-country.json
Normal file
72
tooling/pipeline/format-per-directory.sh
Executable file
@@ -0,0 +1,72 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
#
|
||||||
|
# Walks every directory under src/ and includes/ (including nested
|
||||||
|
# subdirectories), runs clang-format on the C/C++ files in that directory
|
||||||
|
# only (non-recursive), and commits the result with its own commit.
|
||||||
|
#
|
||||||
|
# Usage: ./format-per-directory.sh [-y]
|
||||||
|
# -y skip the confirmation prompt
|
||||||
|
|
||||||
|
set -euo pipefail
|
||||||
|
|
||||||
|
SKIP_CONFIRM=false
|
||||||
|
if [[ "${1:-}" == "-y" ]]; then
|
||||||
|
SKIP_CONFIRM=true
|
||||||
|
fi
|
||||||
|
|
||||||
|
cd "$(dirname "$0")"
|
||||||
|
|
||||||
|
if [ ! -f .clang-format ]; then
|
||||||
|
echo "ERROR: .clang-format file not found."
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
if ! command -v clang-format &>/dev/null; then
|
||||||
|
echo "ERROR: clang-format not found."
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
if ! command -v git &>/dev/null; then
|
||||||
|
echo "ERROR: git not found."
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
echo "WARNING: This script will format .cpp, .h, .cxx, .cc, .c, .hpp files"
|
||||||
|
echo "directory-by-directory under src/ and includes/, committing after each"
|
||||||
|
echo "directory is formatted."
|
||||||
|
|
||||||
|
if [[ "$SKIP_CONFIRM" == false ]]; then
|
||||||
|
read -p "Do you want to continue? (y/n) " -n 1 -r
|
||||||
|
echo
|
||||||
|
if [[ ! $REPLY =~ ^[Yy]$ ]]; then
|
||||||
|
echo "Aborted."
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [[ -n "$(git status --porcelain)" ]]; then
|
||||||
|
echo "ERROR: working tree is not clean. Commit or stash your changes first."
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
dirs=$(find src includes -type d | sort)
|
||||||
|
|
||||||
|
for dir in $dirs; do
|
||||||
|
files=$(find "$dir" -maxdepth 1 -type f \( -name "*.cpp" -o -name "*.hpp" -o -name "*.h" -o -name "*.c" -o -name "*.cc" -o -name "*.cxx" \))
|
||||||
|
|
||||||
|
if [[ -z "$files" ]]; then
|
||||||
|
continue
|
||||||
|
fi
|
||||||
|
|
||||||
|
echo "Formatting $dir..."
|
||||||
|
echo "$files" | xargs clang-format -i
|
||||||
|
|
||||||
|
if [[ -n "$(git status --porcelain -- "$dir")" ]]; then
|
||||||
|
git add -- "$dir"
|
||||||
|
git commit -m "Formatted $dir"
|
||||||
|
else
|
||||||
|
echo "No changes in $dir, skipping commit."
|
||||||
|
fi
|
||||||
|
done
|
||||||
|
|
||||||
|
echo "Done."
|
||||||
78
tooling/pipeline/includes/biergarten_pipeline.h
Normal file
@@ -0,0 +1,78 @@
|
|||||||
|
#ifndef BIERGARTEN_PIPELINE_INCLUDES_BIERGARTEN_PIPELINE_H_
|
||||||
|
#define BIERGARTEN_PIPELINE_INCLUDES_BIERGARTEN_PIPELINE_H_
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @file biergarten_pipeline.h
|
||||||
|
* @brief Master umbrella header — includes every public header in the
|
||||||
|
* Biergarten pipeline.
|
||||||
|
*/
|
||||||
|
|
||||||
|
// --- orchestrator ---
|
||||||
|
#include "biergarten_pipeline_orchestrator.h"
|
||||||
|
|
||||||
|
// --- concurrency ---
|
||||||
|
#include "concurrency/bounded_channel.h"
|
||||||
|
|
||||||
|
// --- data_generation ---
|
||||||
|
#include "data_generation/data_generator.h"
|
||||||
|
#include "data_generation/json_grammars.h"
|
||||||
|
#ifndef BIERGARTEN_MOCK_ONLY
|
||||||
|
#include "data_generation/llama_generator.h"
|
||||||
|
#include "data_generation/llama_generator_helpers.h"
|
||||||
|
#endif
|
||||||
|
#include "data_generation/mock_generator.h"
|
||||||
|
#include "data_generation/prompt_formatting/gemma4_jinja_prompt_formatter.h"
|
||||||
|
#include "data_generation/prompt_formatting/prompt_formatter.h"
|
||||||
|
|
||||||
|
// --- data_model ---
|
||||||
|
#include "data_model/generated_models.h"
|
||||||
|
#include "data_model/models.h"
|
||||||
|
|
||||||
|
// --- json_handling ---
|
||||||
|
#include "json_handling/pretty_print.h"
|
||||||
|
#include "services/curated_data/curated_json_data_service.h"
|
||||||
|
|
||||||
|
// --- llama backend ---
|
||||||
|
#ifndef BIERGARTEN_MOCK_ONLY
|
||||||
|
#include "llama_backend_state.h"
|
||||||
|
#endif
|
||||||
|
|
||||||
|
// --- services: curated_data ---
|
||||||
|
#include "services/curated_data/curated_data_service.h"
|
||||||
|
#include "services/curated_data/mock_curated_data_service.h"
|
||||||
|
|
||||||
|
// --- services: database ---
|
||||||
|
#include "services/database/export_service.h"
|
||||||
|
#include "services/database/sqlite_connection_helpers.h"
|
||||||
|
#include "services/database/sqlite_export_service.h"
|
||||||
|
#include "services/database/sqlite_export_service_helpers.h"
|
||||||
|
#include "services/database/sqlite_handle_types.h"
|
||||||
|
#include "services/database/sqlite_statement_helpers.h"
|
||||||
|
|
||||||
|
// --- services: datetime ---
|
||||||
|
#include "services/datetime/date_time_provider.h"
|
||||||
|
#include "services/datetime/timer.h"
|
||||||
|
|
||||||
|
// --- services: enrichment ---
|
||||||
|
#include "services/enrichment/enrichment_service.h"
|
||||||
|
#include "services/enrichment/mock_enrichment.h"
|
||||||
|
#include "services/enrichment/wikipedia_service.h"
|
||||||
|
|
||||||
|
// --- services: logging ---
|
||||||
|
#include "services/logging/log_dispatcher.h"
|
||||||
|
#include "services/logging/log_entry.h"
|
||||||
|
#include "services/logging/log_producer.h"
|
||||||
|
#include "services/logging/logger.h"
|
||||||
|
|
||||||
|
// --- services: postal_code ---
|
||||||
|
#include "services/postal_code/mock_postal_code_service.h"
|
||||||
|
#include "services/postal_code/postal_code_service.h"
|
||||||
|
|
||||||
|
// --- services: prompting ---
|
||||||
|
#include "services/prompting/prompt_directory.h"
|
||||||
|
|
||||||
|
// --- web_client ---
|
||||||
|
#include "web_client/http_web_client.h"
|
||||||
|
#include "web_client/web_client.h"
|
||||||
|
|
||||||
|
#endif // BIERGARTEN_PIPELINE_INCLUDES_BIERGARTEN_PIPELINE_H_
|
||||||
@@ -1,12 +1,9 @@
|
|||||||
#ifndef BIERGARTEN_PIPELINE_INCLUDES_BIERGARTEN_DATA_GENERATOR_H_
|
#ifndef BIERGARTEN_PIPELINE_INCLUDES_BIERGARTEN_PIPELINE_ORCHESTRATOR_H_
|
||||||
#define BIERGARTEN_PIPELINE_INCLUDES_BIERGARTEN_DATA_GENERATOR_H_
|
#define BIERGARTEN_PIPELINE_INCLUDES_BIERGARTEN_PIPELINE_ORCHESTRATOR_H_
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @file biergarten_data_generator.h
|
* @file biergarten_pipeline_orchestrator.h
|
||||||
* @brief Orchestration for end-to-end brewery data generation pipeline.
|
* @brief Orchestration for end-to-end brewery and user data generation.
|
||||||
*
|
|
||||||
* Intent: Coordinates location loading, enrichment, and generation phases
|
|
||||||
* to produce a complete dataset. Coordinates dependencies via composition root.
|
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#include <memory>
|
#include <memory>
|
||||||
@@ -15,25 +12,29 @@
|
|||||||
|
|
||||||
#include "data_generation/data_generator.h"
|
#include "data_generation/data_generator.h"
|
||||||
#include "data_model/generated_models.h"
|
#include "data_model/generated_models.h"
|
||||||
|
#include "services/curated_data/curated_data_service.h"
|
||||||
#include "services/database/export_service.h"
|
#include "services/database/export_service.h"
|
||||||
#include "services/enrichment/enrichment_service.h"
|
#include "services/enrichment/enrichment_service.h"
|
||||||
|
|
||||||
#include "services/logging/logger.h"
|
#include "services/logging/logger.h"
|
||||||
|
#include "services/postal_code/postal_code_service.h"
|
||||||
/**
|
/**
|
||||||
* @brief Main data generator class for the Biergarten pipeline.
|
* @brief Main orchestrator for the Biergarten data generation pipeline.
|
||||||
*
|
*
|
||||||
* This class encapsulates the core logic for generating brewery data.
|
* Handles location loading, city enrichment, and brewery/user generation.
|
||||||
* It handles location loading, city enrichment, and brewery generation.
|
|
||||||
*/
|
*/
|
||||||
class BiergartenPipelineOrchestrator {
|
class BiergartenPipelineOrchestrator {
|
||||||
public:
|
public:
|
||||||
/**
|
/**
|
||||||
* @brief Constructs the orchestrator with injected pipeline dependencies.
|
* @brief Constructs the orchestrator with injected pipeline dependencies.
|
||||||
*
|
*
|
||||||
|
* @param logger Sink for pipeline diagnostics.
|
||||||
* @param context_service Provides regional context for locations.
|
* @param context_service Provides regional context for locations.
|
||||||
* @param generator Implementation (Llama or Mock) for brewery/user generation.
|
* @param generator Implementation (Llama or Mock) for brewery/user
|
||||||
|
* generation.
|
||||||
* @param exporter Database backend for persisting generated records.
|
* @param exporter Database backend for persisting generated records.
|
||||||
|
* @param curated_data_service Loads curated location, persona, and name
|
||||||
|
* data used to seed generation.
|
||||||
|
* @param postal_code_service
|
||||||
* @param application_options CLI configuration and paths.
|
* @param application_options CLI configuration and paths.
|
||||||
*/
|
*/
|
||||||
BiergartenPipelineOrchestrator(
|
BiergartenPipelineOrchestrator(
|
||||||
@@ -41,6 +42,8 @@ class BiergartenPipelineOrchestrator {
|
|||||||
std::unique_ptr<IEnrichmentService> context_service,
|
std::unique_ptr<IEnrichmentService> context_service,
|
||||||
std::unique_ptr<DataGenerator> generator,
|
std::unique_ptr<DataGenerator> generator,
|
||||||
std::unique_ptr<IExportService> exporter,
|
std::unique_ptr<IExportService> exporter,
|
||||||
|
std::unique_ptr<ICuratedDataService> curated_data_service,
|
||||||
|
std::unique_ptr<IPostalCodeService> postal_code_service,
|
||||||
const ApplicationOptions& application_options);
|
const ApplicationOptions& application_options);
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@@ -49,40 +52,45 @@ class BiergartenPipelineOrchestrator {
|
|||||||
* Performs the following steps:
|
* Performs the following steps:
|
||||||
* 1. Load curated locations from JSON
|
* 1. Load curated locations from JSON
|
||||||
* 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 and user data for sampled cities
|
||||||
*
|
*
|
||||||
* @note STRUCTURAL CONCURRENCY REQUIREMENT:
|
* @note STRUCTURAL CONCURRENCY REQUIREMENT:
|
||||||
* When transitioned to a multithreaded design, this method MUST structurally
|
* When transitioned to a multithreaded design, this method MUST
|
||||||
* enforce that all deployed worker threads are joined before returning (e.g.
|
* structurally enforce that all deployed worker threads are joined before
|
||||||
* by using std::jthread or a structured concurrency primitive). This ensures
|
* returning (e.g. by using std::jthread or a structured concurrency
|
||||||
* workers do not attempt to log to a closed channel during application teardown.
|
* 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();
|
||||||
|
|
||||||
private:
|
private:
|
||||||
/// @brief Logger instance for emitting pipeline messages.
|
|
||||||
std::shared_ptr<ILogger> logger_;
|
std::shared_ptr<ILogger> logger_;
|
||||||
|
|
||||||
/// @brief Owning context provider dependency.
|
|
||||||
std::unique_ptr<IEnrichmentService> context_service_;
|
std::unique_ptr<IEnrichmentService> context_service_;
|
||||||
|
|
||||||
/// @brief Generator dependency selected in the composition root.
|
/**
|
||||||
|
* @brief Generator implementation selected at the composition root (Llama
|
||||||
|
* or Mock).
|
||||||
|
*/
|
||||||
std::unique_ptr<DataGenerator> generator_;
|
std::unique_ptr<DataGenerator> generator_;
|
||||||
|
|
||||||
/// @brief Storage backend for generated brewery records.
|
/**
|
||||||
|
* @brief Storage backend for generated brewery and user records.
|
||||||
|
*/
|
||||||
std::unique_ptr<IExportService> exporter_;
|
std::unique_ptr<IExportService> exporter_;
|
||||||
|
std::unique_ptr<ICuratedDataService> curated_data_service_;
|
||||||
|
std::unique_ptr<IPostalCodeService> postal_code_service_;
|
||||||
|
|
||||||
/// @brief CLI configuration: paths, model settings, generation parameters.
|
|
||||||
ApplicationOptions application_options_;
|
ApplicationOptions application_options_;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @brief Load locations from JSON and sample cities.
|
* @brief Load locations from JSON and sample cities.
|
||||||
*
|
*
|
||||||
* @return Vector of sampled locations capped at 50 entries.
|
* @return Vector of locations randomly sampled per
|
||||||
|
* PipelineOptions::location_count.
|
||||||
*/
|
*/
|
||||||
std::vector<Location> QueryCitiesWithCountries();
|
std::vector<City> QueryCitiesWithCountries();
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @brief Generate breweries for enriched cities.
|
* @brief Generate breweries for enriched cities.
|
||||||
@@ -92,11 +100,20 @@ class BiergartenPipelineOrchestrator {
|
|||||||
void GenerateBreweries(std::span<const EnrichedCity> cities);
|
void GenerateBreweries(std::span<const EnrichedCity> cities);
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @brief Log the generated brewery results.
|
* @brief Generate users grounded in sampled names and personas for
|
||||||
|
* enriched cities.
|
||||||
|
*
|
||||||
|
* @param cities Span of enriched city data.
|
||||||
|
*/
|
||||||
|
void GenerateUsers(std::span<const EnrichedCity> cities);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Log the generated brewery and user results.
|
||||||
*/
|
*/
|
||||||
void LogResults() const;
|
void LogResults() const;
|
||||||
|
|
||||||
/// @brief Stores generated brewery data.
|
std::vector<BreweryRecord> generated_breweries_;
|
||||||
std::vector<GeneratedBrewery> generated_breweries_;
|
std::vector<UserRecord> generated_users_;
|
||||||
};
|
};
|
||||||
#endif // BIERGARTEN_PIPELINE_INCLUDES_BIERGARTEN_DATA_GENERATOR_H_
|
|
||||||
|
#endif // BIERGARTEN_PIPELINE_INCLUDES_BIERGARTEN_PIPELINE_ORCHESTRATOR_H_
|
||||||
|
|||||||
@@ -9,10 +9,12 @@
|
|||||||
|
|
||||||
/**
|
/**
|
||||||
* @file bounded_channel.h
|
* @file bounded_channel.h
|
||||||
* @brief Thread-safe, bounded multi-producer/multi-consumer synchronous channel.
|
* @brief Thread-safe, bounded multi-producer/multi-consumer synchronous
|
||||||
|
* channel.
|
||||||
*
|
*
|
||||||
* Intent: Enables asynchronous inter-thread communication with backpressure.
|
* Intent: Enables asynchronous inter-thread communication with backpressure.
|
||||||
* Models a synchronous channel where producers/consumers block on capacity limits.
|
* Models a synchronous channel where producers/consumers block on capacity
|
||||||
|
* limits.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
|||||||
@@ -1,4 +1,4 @@
|
|||||||
#include "bounded_channel.h"
|
#include "concurrency/bounded_channel.h"
|
||||||
|
|
||||||
template <typename T>
|
template <typename T>
|
||||||
void BoundedChannel<T>::Send(T item) {
|
void BoundedChannel<T>::Send(T item) {
|
||||||
|
|||||||
@@ -24,16 +24,20 @@ class DataGenerator {
|
|||||||
* @param region_context Additional regional context text.
|
* @param region_context Additional regional context text.
|
||||||
* @return Brewery generation result.
|
* @return Brewery generation result.
|
||||||
*/
|
*/
|
||||||
virtual BreweryResult GenerateBrewery(const Location& location,
|
virtual BreweryResult GenerateBrewery(const City& location,
|
||||||
const std::string& region_context) = 0;
|
const std::string& region_context) = 0;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @brief Generates a user profile for a locale.
|
* @brief Generates a user profile grounded in a sampled name and persona.
|
||||||
*
|
*
|
||||||
* @param locale Locale hint used by generator.
|
* @param city Enriched city the user is associated with.
|
||||||
|
* @param persona Persona archetype grounding the generated bio.
|
||||||
|
* @param name Sampled first/last name (and gender)
|
||||||
* @return User generation result.
|
* @return User generation result.
|
||||||
*/
|
*/
|
||||||
virtual UserResult GenerateUser(const std::string& locale) = 0;
|
virtual UserResult GenerateUser(const EnrichedCity& city,
|
||||||
|
const UserPersona& persona,
|
||||||
|
const Name& name) = 0;
|
||||||
};
|
};
|
||||||
|
|
||||||
#endif // BIERGARTEN_PIPELINE_INCLUDES_DATA_GENERATION_DATA_GENERATOR_H_
|
#endif // BIERGARTEN_PIPELINE_INCLUDES_DATA_GENERATION_DATA_GENERATOR_H_
|
||||||
|
|||||||
52
tooling/pipeline/includes/data_generation/json_grammars.h
Normal file
@@ -0,0 +1,52 @@
|
|||||||
|
#ifndef BIERGARTEN_PIPELINE_INCLUDES_DATA_GENERATION_JSON_GRAMMARS_H_
|
||||||
|
#define BIERGARTEN_PIPELINE_INCLUDES_DATA_GENERATION_JSON_GRAMMARS_H_
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @file data_generation/json_grammars.h
|
||||||
|
* @brief GBNF grammars constraining structured JSON output from
|
||||||
|
* LlamaGenerator inference calls.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include <string_view>
|
||||||
|
|
||||||
|
// GBNF grammar for structured user JSON output.
|
||||||
|
// thought-block permits the model to emit free-form reasoning before the
|
||||||
|
// JSON object (the prompts explicitly invite this); only the "{...}" tail is
|
||||||
|
// constrained to the expected shape.
|
||||||
|
inline constexpr std::string_view kUserJsonGrammar = R"json_user(
|
||||||
|
root ::= thought-block (
|
||||||
|
"{" ws
|
||||||
|
"\"username\"" ws ":" ws string ws "," ws
|
||||||
|
"\"bio\"" ws ":" ws string ws "," ws
|
||||||
|
"\"activity_weight\"" ws ":" ws number ws
|
||||||
|
"}" ws
|
||||||
|
)
|
||||||
|
thought-block ::= [^{]*
|
||||||
|
ws ::= [ \t\n\r]*
|
||||||
|
string ::= "\"" char+ "\""
|
||||||
|
char ::= [^"\\\x7F\x00-\x1F] | [\\] escape
|
||||||
|
escape ::= ["\\/bfnrt] | "u" hex hex hex hex
|
||||||
|
hex ::= [0-9a-fA-F]
|
||||||
|
number ::= "-"? ("0" | [1-9] [0-9]*) ("." [0-9]+)?
|
||||||
|
)json_user";
|
||||||
|
|
||||||
|
// GBNF grammar for structured brewery JSON output (see thought-block note
|
||||||
|
// above).
|
||||||
|
inline constexpr std::string_view kBreweryJsonGrammar = R"json_brewery(
|
||||||
|
root ::= thought-block (
|
||||||
|
"{" ws
|
||||||
|
"\"name_en\"" ws ":" ws string ws "," ws
|
||||||
|
"\"description_en\"" ws ":" ws string ws "," ws
|
||||||
|
"\"name_local\"" ws ":" ws string ws "," ws
|
||||||
|
"\"description_local\"" ws ":" ws string ws
|
||||||
|
"}" ws
|
||||||
|
)
|
||||||
|
thought-block ::= [^{]*
|
||||||
|
ws ::= [ \t\n\r]*
|
||||||
|
string ::= "\"" char+ "\""
|
||||||
|
char ::= [^"\\\x7F\x00-\x1F] | [\\] escape
|
||||||
|
escape ::= ["\\/bfnrt] | "u" hex hex hex hex
|
||||||
|
hex ::= [0-9a-fA-F]
|
||||||
|
)json_brewery";
|
||||||
|
|
||||||
|
#endif // BIERGARTEN_PIPELINE_INCLUDES_DATA_GENERATION_JSON_GRAMMARS_H_
|
||||||
@@ -1,24 +1,23 @@
|
|||||||
#ifndef BIERGARTEN_PIPELINE_INCLUDES_DATA_GENERATION_LLAMA_GENERATOR_H_
|
#ifndef BIERGARTEN_PIPELINE_INCLUDES_DATA_GENERATION_LLAMA_GENERATOR_H_
|
||||||
#define BIERGARTEN_PIPELINE_INCLUDES_DATA_GENERATION_LLAMA_GENERATOR_H_
|
#define BIERGARTEN_PIPELINE_INCLUDES_DATA_GENERATION_LLAMA_GENERATOR_H_
|
||||||
|
|
||||||
#include <filesystem>
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @file data_generation/llama_generator.h
|
* @file data_generation/llama_generator.h
|
||||||
* @brief llama.cpp-backed implementation of DataGenerator.
|
* @brief llama.cpp-backed implementation of DataGenerator.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#include <cstdint>
|
#include <cstdint>
|
||||||
|
#include <filesystem>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include <random>
|
#include <random>
|
||||||
#include <string>
|
#include <string>
|
||||||
#include <string_view>
|
#include <string_view>
|
||||||
|
|
||||||
#include "../services/prompting/prompt_directory.h"
|
|
||||||
#include "data_generation/data_generator.h"
|
#include "data_generation/data_generator.h"
|
||||||
#include "data_generation/prompt_formatting/prompt_formatter.h"
|
#include "data_generation/prompt_formatting/prompt_formatter.h"
|
||||||
#include "data_model/models.h"
|
#include "data_model/models.h"
|
||||||
#include "services/logging/logger.h"
|
#include "services/logging/logger.h"
|
||||||
|
#include "services/prompting/prompt_directory.h"
|
||||||
|
|
||||||
struct llama_model;
|
struct llama_model;
|
||||||
struct llama_context;
|
struct llama_context;
|
||||||
@@ -44,14 +43,9 @@ class LlamaGenerator final : public DataGenerator {
|
|||||||
|
|
||||||
~LlamaGenerator() override;
|
~LlamaGenerator() override;
|
||||||
|
|
||||||
// disable copy constructor
|
|
||||||
LlamaGenerator(const LlamaGenerator&) = delete;
|
LlamaGenerator(const LlamaGenerator&) = delete;
|
||||||
|
|
||||||
// disable copy assignment operator
|
|
||||||
LlamaGenerator& operator=(const LlamaGenerator&) = delete;
|
LlamaGenerator& operator=(const LlamaGenerator&) = delete;
|
||||||
// disable move constructor
|
|
||||||
LlamaGenerator(LlamaGenerator&&) = delete;
|
LlamaGenerator(LlamaGenerator&&) = delete;
|
||||||
// disable move assignment operator
|
|
||||||
LlamaGenerator& operator=(LlamaGenerator&&) = delete;
|
LlamaGenerator& operator=(LlamaGenerator&&) = delete;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@@ -61,16 +55,19 @@ class LlamaGenerator final : public DataGenerator {
|
|||||||
* @param region_context Additional regional context.
|
* @param region_context Additional regional context.
|
||||||
* @return Generated brewery result.
|
* @return Generated brewery result.
|
||||||
*/
|
*/
|
||||||
BreweryResult GenerateBrewery(const Location& location,
|
BreweryResult GenerateBrewery(const City& location,
|
||||||
const std::string& region_context) override;
|
const std::string& region_context) override;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @brief Generates a user profile for the provided locale.
|
* @brief Generates a user profile grounded in a sampled name and persona.
|
||||||
*
|
*
|
||||||
* @param locale Locale hint.
|
* @param city Enriched city the user is associated with.
|
||||||
|
* @param persona Persona archetype grounding the generated bio.
|
||||||
|
* @param name Sampled first/last name -- not LLM-invented.
|
||||||
* @return Generated user profile.
|
* @return Generated user profile.
|
||||||
*/
|
*/
|
||||||
UserResult GenerateUser(const std::string& locale) override;
|
UserResult GenerateUser(const EnrichedCity& city, const UserPersona& persona,
|
||||||
|
const Name& name) override;
|
||||||
|
|
||||||
private:
|
private:
|
||||||
static constexpr int32_t kDefaultMaxTokens = 10000;
|
static constexpr int32_t kDefaultMaxTokens = 10000;
|
||||||
|
|||||||
@@ -47,4 +47,18 @@ void AppendTokenPiece(const llama_vocab* vocab, llama_token token,
|
|||||||
std::optional<std::string> ValidateBreweryJson(const std::string& raw,
|
std::optional<std::string> ValidateBreweryJson(const std::string& raw,
|
||||||
BreweryResult& brewery_out);
|
BreweryResult& brewery_out);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Validates and parses user JSON output.
|
||||||
|
*
|
||||||
|
* Only populates `username`, `bio`, and `activity_weight` -- `first_name`
|
||||||
|
* and `last_name` are not LLM-authored and are set separately from the
|
||||||
|
* sampled Name.
|
||||||
|
*
|
||||||
|
* @param raw Raw model output.
|
||||||
|
* @param user_out Parsed user payload.
|
||||||
|
* @return Validation error message if invalid, or std::nullopt on success.
|
||||||
|
*/
|
||||||
|
std::optional<std::string> ValidateUserJson(const std::string& raw,
|
||||||
|
UserResult& user_out);
|
||||||
|
|
||||||
#endif // BIERGARTEN_PIPELINE_INCLUDES_DATA_GENERATION_LLAMA_GENERATOR_HELPERS_H_
|
#endif // BIERGARTEN_PIPELINE_INCLUDES_DATA_GENERATION_LLAMA_GENERATOR_HELPERS_H_
|
||||||
|
|||||||
@@ -24,16 +24,20 @@ class MockGenerator final : public DataGenerator {
|
|||||||
* @param region_context Unused for mock generation.
|
* @param region_context Unused for mock generation.
|
||||||
* @return Generated brewery result.
|
* @return Generated brewery result.
|
||||||
*/
|
*/
|
||||||
BreweryResult GenerateBrewery(const Location& location,
|
BreweryResult GenerateBrewery(const City& location,
|
||||||
const std::string& region_context) override;
|
const std::string& region_context) override;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @brief Generates deterministic user data for a locale.
|
* @brief Generates deterministic user data grounded in a sampled name and
|
||||||
|
* persona.
|
||||||
*
|
*
|
||||||
* @param locale Locale hint.
|
* @param city Enriched city the user is associated with.
|
||||||
|
* @param persona Persona archetype.
|
||||||
|
* @param name Sampled first/last name, copied directly into the result.
|
||||||
* @return Generated user result.
|
* @return Generated user result.
|
||||||
*/
|
*/
|
||||||
UserResult GenerateUser(const std::string& locale) override;
|
UserResult GenerateUser(const EnrichedCity& city, const UserPersona& persona,
|
||||||
|
const Name& name) override;
|
||||||
|
|
||||||
private:
|
private:
|
||||||
/**
|
/**
|
||||||
@@ -42,7 +46,18 @@ class MockGenerator final : public DataGenerator {
|
|||||||
* @param location City and country names.
|
* @param location City and country names.
|
||||||
* @return Deterministic hash value.
|
* @return Deterministic hash value.
|
||||||
*/
|
*/
|
||||||
static size_t DeterministicHash(const Location& location);
|
static size_t DeterministicHash(const City& location);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Combines city, persona, and name into a stable hash value.
|
||||||
|
*
|
||||||
|
* @param location City and country names.
|
||||||
|
* @param persona Persona archetype.
|
||||||
|
* @param name Sampled first/last name.
|
||||||
|
* @return Deterministic hash value.
|
||||||
|
*/
|
||||||
|
static size_t DeterministicHash(const City& location,
|
||||||
|
const UserPersona& persona, const Name& name);
|
||||||
|
|
||||||
// Hash stride constants for deterministic distribution across fixed-size
|
// Hash stride constants for deterministic distribution across fixed-size
|
||||||
// arrays. These coprime strides spread hash values uniformly without
|
// arrays. These coprime strides spread hash values uniformly without
|
||||||
@@ -50,6 +65,8 @@ class MockGenerator final : public DataGenerator {
|
|||||||
static constexpr size_t kNounHashStride = 7;
|
static constexpr size_t kNounHashStride = 7;
|
||||||
static constexpr size_t kDescriptionHashStride = 13;
|
static constexpr size_t kDescriptionHashStride = 13;
|
||||||
static constexpr size_t kBioHashStride = 11;
|
static constexpr size_t kBioHashStride = 11;
|
||||||
|
static constexpr size_t kActivityWeightHashStride = 17;
|
||||||
|
static constexpr size_t kActivityWeightHashRange = 1000;
|
||||||
|
|
||||||
static constexpr std::array<std::string_view, 18> kBreweryAdjectives = {
|
static constexpr std::array<std::string_view, 18> kBreweryAdjectives = {
|
||||||
"Craft", "Heritage", "Local", "Artisan", "Pioneer", "Golden",
|
"Craft", "Heritage", "Local", "Artisan", "Pioneer", "Golden",
|
||||||
@@ -124,7 +141,8 @@ class MockGenerator final : public DataGenerator {
|
|||||||
"Visits breweries for the stories, stays for the flagship pours.",
|
"Visits breweries for the stories, stays for the flagship pours.",
|
||||||
"Craft beer fan mapping tasting notes and favorite brew routes.",
|
"Craft beer fan mapping tasting notes and favorite brew routes.",
|
||||||
"Always ready to trade recommendations for underrated local breweries.",
|
"Always ready to trade recommendations for underrated local breweries.",
|
||||||
"Keeping a running list of must-try collab releases and tap takeovers."};
|
"Keeping a running list of must-try collab releases and tap "
|
||||||
|
"takeovers."};
|
||||||
};
|
};
|
||||||
|
|
||||||
#endif // BIERGARTEN_PIPELINE_INCLUDES_DATA_GENERATION_MOCK_GENERATOR_H_
|
#endif // BIERGARTEN_PIPELINE_INCLUDES_DATA_GENERATION_MOCK_GENERATOR_H_
|
||||||
|
|||||||
@@ -3,10 +3,11 @@
|
|||||||
|
|
||||||
/**
|
/**
|
||||||
* @file data_model/generated_models.h
|
* @file data_model/generated_models.h
|
||||||
* @brief Generated output models from the pipeline: brewery/user results, enriched data,
|
* @brief Generated output models from the pipeline: brewery/user results,
|
||||||
* and complete generation results.
|
* enriched data, and complete generation results.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
|
#include <optional>
|
||||||
#include <string>
|
#include <string>
|
||||||
|
|
||||||
#include "data_model/models.h"
|
#include "data_model/models.h"
|
||||||
@@ -19,16 +20,24 @@
|
|||||||
* @brief Generated brewery payload.
|
* @brief Generated brewery payload.
|
||||||
*/
|
*/
|
||||||
struct BreweryResult {
|
struct BreweryResult {
|
||||||
/// @brief Brewery display name in English.
|
/**
|
||||||
|
* @brief Brewery display name in English.
|
||||||
|
*/
|
||||||
std::string name_en;
|
std::string name_en;
|
||||||
|
|
||||||
/// @brief Brewery description text in English.
|
/**
|
||||||
|
* @brief Brewery description text in English.
|
||||||
|
*/
|
||||||
std::string description_en;
|
std::string description_en;
|
||||||
|
|
||||||
/// @brief Brewery display name in the local language.
|
/**
|
||||||
|
* @brief Brewery display name in the local language.
|
||||||
|
*/
|
||||||
std::string name_local;
|
std::string name_local;
|
||||||
|
|
||||||
/// @brief Brewery description text in the local language.
|
/**
|
||||||
|
* @brief Brewery description text in the local language.
|
||||||
|
*/
|
||||||
std::string description_local;
|
std::string description_local;
|
||||||
};
|
};
|
||||||
|
|
||||||
@@ -36,11 +45,39 @@ struct BreweryResult {
|
|||||||
* @brief Generated user profile payload.
|
* @brief Generated user profile payload.
|
||||||
*/
|
*/
|
||||||
struct UserResult {
|
struct UserResult {
|
||||||
/// @brief Username handle.
|
/**
|
||||||
|
* @brief First (given) name, copied from the sampled Name -- not
|
||||||
|
* LLM-invented.
|
||||||
|
*/
|
||||||
|
std::string first_name{};
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Last (family) name, copied from the sampled Name -- not
|
||||||
|
* LLM-invented.
|
||||||
|
*/
|
||||||
|
std::string last_name{};
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Gender associated with the sampled first name, copied from the
|
||||||
|
* sampled Name -- not LLM-invented.
|
||||||
|
*/
|
||||||
|
std::string gender{};
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Username handle.
|
||||||
|
*/
|
||||||
std::string username{};
|
std::string username{};
|
||||||
|
|
||||||
/// @brief Short user biography.
|
/**
|
||||||
|
* @brief Short user biography.
|
||||||
|
*/
|
||||||
std::string bio{};
|
std::string bio{};
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Relative check-in/activity weight for this user, used to drive
|
||||||
|
* a J-curve activity profile in later pipeline phases.
|
||||||
|
*/
|
||||||
|
float activity_weight{};
|
||||||
};
|
};
|
||||||
|
|
||||||
// ============================================================================
|
// ============================================================================
|
||||||
@@ -51,16 +88,60 @@ struct UserResult {
|
|||||||
* @brief Enriched city data with Wikipedia context.
|
* @brief Enriched city data with Wikipedia context.
|
||||||
*/
|
*/
|
||||||
struct EnrichedCity {
|
struct EnrichedCity {
|
||||||
Location location;
|
City location;
|
||||||
std::string region_context{};
|
std::string region_context{};
|
||||||
};
|
};
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief A brewery's address. Owns the `City` it belongs to alongside its
|
||||||
|
* street-level fields. Mirrors the `brewery_addresses` table and the
|
||||||
|
* backend's `BreweryPostLocation` shape.
|
||||||
|
*
|
||||||
|
* Only `city` and `postal_code` are populated today -- street-address
|
||||||
|
* generation has no fixture data or service yet, so the address lines stay
|
||||||
|
* empty.
|
||||||
|
*/
|
||||||
|
struct BreweryAddress {
|
||||||
|
City city{};
|
||||||
|
std::optional<std::string> address_line_1{};
|
||||||
|
std::optional<std::string> address_line_2{};
|
||||||
|
std::string postal_code{};
|
||||||
|
};
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief A user's address. Owns the `City` it belongs to alongside its
|
||||||
|
* street-level fields. Mirrors the `user_addresses` table.
|
||||||
|
*
|
||||||
|
* Only `city` is populated today; the street-level columns exist for future
|
||||||
|
* enrichment.
|
||||||
|
*/
|
||||||
|
struct UserAddress {
|
||||||
|
City city{};
|
||||||
|
std::optional<std::string> address_line_1{};
|
||||||
|
std::optional<std::string> address_line_2{};
|
||||||
|
std::optional<std::string> postal_code{};
|
||||||
|
};
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @brief Helper struct to store generated brewery data.
|
* @brief Helper struct to store generated brewery data.
|
||||||
*/
|
*/
|
||||||
struct GeneratedBrewery {
|
struct BreweryRecord {
|
||||||
Location location;
|
BreweryAddress address;
|
||||||
BreweryResult brewery;
|
BreweryResult brewery;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Helper struct to store generated user data.
|
||||||
|
*
|
||||||
|
* `email` and `date_of_birth` are programmatically generated by the
|
||||||
|
* orchestrator (never LLM-authored) so a downstream auth-account seeding
|
||||||
|
* consumer can register real accounts from the pipeline's SQLite export.
|
||||||
|
*/
|
||||||
|
struct UserRecord {
|
||||||
|
UserAddress address{};
|
||||||
|
UserResult user;
|
||||||
|
std::string email{};
|
||||||
|
std::string date_of_birth{};
|
||||||
|
};
|
||||||
|
|
||||||
#endif // BIERGARTEN_PIPELINE_INCLUDES_DATA_MODEL_GENERATED_MODELS_H_
|
#endif // BIERGARTEN_PIPELINE_INCLUDES_DATA_MODEL_GENERATED_MODELS_H_
|
||||||
|
|||||||
@@ -10,6 +10,7 @@
|
|||||||
#include <boost/program_options.hpp>
|
#include <boost/program_options.hpp>
|
||||||
#include <cstdint>
|
#include <cstdint>
|
||||||
#include <filesystem>
|
#include <filesystem>
|
||||||
|
#include <functional>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include <optional>
|
#include <optional>
|
||||||
#include <string>
|
#include <string>
|
||||||
@@ -21,47 +22,115 @@ class ILogger;
|
|||||||
namespace prog_opts = boost::program_options;
|
namespace prog_opts = boost::program_options;
|
||||||
|
|
||||||
// ============================================================================
|
// ============================================================================
|
||||||
// Location Models
|
// City Models
|
||||||
// ============================================================================
|
// ============================================================================
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @brief Canonical location record for city-level generation.
|
* @brief Canonical location record for city-level generation.
|
||||||
*/
|
*/
|
||||||
struct Location {
|
struct City {
|
||||||
/// @brief City name.
|
|
||||||
std::string city{};
|
std::string city{};
|
||||||
|
|
||||||
/// @brief State or province name.
|
|
||||||
std::string state_province{};
|
std::string state_province{};
|
||||||
|
|
||||||
/// @brief ISO 3166-2 subdivision code.
|
/**
|
||||||
|
* @brief ISO 3166-2 subdivision code.
|
||||||
|
*/
|
||||||
std::string iso3166_2{};
|
std::string iso3166_2{};
|
||||||
|
|
||||||
/// @brief Country name.
|
|
||||||
std::string country{};
|
std::string country{};
|
||||||
|
|
||||||
/// @brief ISO 3166-1 country code.
|
/**
|
||||||
|
* @brief ISO 3166-1 country code.
|
||||||
|
*/
|
||||||
std::string iso3166_1{};
|
std::string iso3166_1{};
|
||||||
|
|
||||||
/// @brief Local language codes in priority order.
|
/**
|
||||||
|
* @brief A list of regular expressions used for valid postal codes in the
|
||||||
|
* region.
|
||||||
|
*
|
||||||
|
*/
|
||||||
|
std::vector<std::string> postal_regex{};
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Example postal codes for the region.
|
||||||
|
*/
|
||||||
|
std::vector<std::string> postal_code_examples{};
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Local language codes in priority order.
|
||||||
|
*/
|
||||||
std::vector<std::string> local_languages{};
|
std::vector<std::string> local_languages{};
|
||||||
|
};
|
||||||
|
|
||||||
/// @brief Latitude in decimal degrees.
|
// ============================================================================
|
||||||
double latitude{};
|
// Name / Persona Models
|
||||||
|
// ============================================================================
|
||||||
|
|
||||||
/// @brief Longitude in decimal degrees.
|
/**
|
||||||
double longitude{};
|
* @brief A sampled first/last name pair, with the source forename's gender.
|
||||||
|
*
|
||||||
|
* Produced by the SampleName() helper in generate_users.cc.
|
||||||
|
*/
|
||||||
|
struct Name {
|
||||||
|
std::string first_name{};
|
||||||
|
std::string last_name{};
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Gender associated with the sampled forename (e.g. "M", "F"), as
|
||||||
|
* reported by the source dataset.
|
||||||
|
*/
|
||||||
|
std::string gender{};
|
||||||
};
|
};
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @brief Non-owning brewery location input.
|
* @brief A single forename entry from the names-by-country fixture data.
|
||||||
*/
|
*/
|
||||||
struct BreweryLocation {
|
struct ForenameEntry {
|
||||||
/// @brief City name.
|
/**
|
||||||
std::string_view city_name;
|
* @brief Romanized forename.
|
||||||
|
*/
|
||||||
|
std::string name{};
|
||||||
|
|
||||||
/// @brief Country name.
|
/**
|
||||||
std::string_view country_name;
|
* @brief Gender associated with this forename, as reported by the source
|
||||||
|
* dataset (e.g. "M", "F").
|
||||||
|
*/
|
||||||
|
std::string gender{};
|
||||||
|
|
||||||
|
bool operator==(const ForenameEntry& other) const {
|
||||||
|
return name == other.name && gender == other.gender;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
namespace std {
|
||||||
|
template <>
|
||||||
|
struct hash<ForenameEntry> {
|
||||||
|
size_t operator()(const ForenameEntry& entry) const noexcept {
|
||||||
|
const size_t name_hash = std::hash<std::string>{}(entry.name);
|
||||||
|
const size_t gender_hash = std::hash<std::string>{}(entry.gender);
|
||||||
|
return name_hash ^ (gender_hash << 1);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
} // namespace std
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief A persona archetype used to ground LLM-generated user bios.
|
||||||
|
*/
|
||||||
|
struct UserPersona {
|
||||||
|
/**
|
||||||
|
* @brief Persona display name (e.g. "Hophead Explorer").
|
||||||
|
*/
|
||||||
|
std::string name{};
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Short description of the persona's interests and voice.
|
||||||
|
*/
|
||||||
|
std::string description{};
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Beer styles this persona gravitates toward.
|
||||||
|
*/
|
||||||
|
std::vector<std::string> style_affinities{};
|
||||||
};
|
};
|
||||||
|
|
||||||
// ============================================================================
|
// ============================================================================
|
||||||
@@ -72,22 +141,34 @@ struct BreweryLocation {
|
|||||||
* @brief LLM sampling parameters.
|
* @brief LLM sampling parameters.
|
||||||
*/
|
*/
|
||||||
struct SamplingOptions {
|
struct SamplingOptions {
|
||||||
/// @brief LLM sampling temperature (0.0 to 1.0, higher = more random).
|
/**
|
||||||
|
* @brief LLM sampling temperature (higher = more random).
|
||||||
|
*/
|
||||||
float temperature = 1.0F;
|
float temperature = 1.0F;
|
||||||
|
|
||||||
/// @brief LLM nucleus sampling top-p parameter.
|
/**
|
||||||
|
* @brief LLM nucleus sampling top-p parameter.
|
||||||
|
*/
|
||||||
float top_p = 0.95F;
|
float top_p = 0.95F;
|
||||||
|
|
||||||
/// @brief LLM top-k sampling parameter.
|
/**
|
||||||
|
* @brief LLM top-k sampling parameter.
|
||||||
|
*/
|
||||||
uint32_t top_k = 64;
|
uint32_t top_k = 64;
|
||||||
|
|
||||||
/// @brief Context window size (tokens).
|
/**
|
||||||
|
* @brief Context window size (tokens).
|
||||||
|
*/
|
||||||
uint32_t n_ctx = 8192;
|
uint32_t n_ctx = 8192;
|
||||||
|
|
||||||
/// @brief Random seed (-1 for random, otherwise non-negative).
|
/**
|
||||||
|
* @brief Random seed (-1 for random, otherwise non-negative).
|
||||||
|
*/
|
||||||
int seed = -1;
|
int seed = -1;
|
||||||
|
|
||||||
/// @brief Number of layers to offload to GPU.
|
/**
|
||||||
|
* @brief Number of layers to offload to GPU.
|
||||||
|
*/
|
||||||
int n_gpu_layers = 0;
|
int n_gpu_layers = 0;
|
||||||
};
|
};
|
||||||
|
|
||||||
@@ -95,16 +176,20 @@ struct SamplingOptions {
|
|||||||
* @brief Configuration for the LLM generator component.
|
* @brief Configuration for the LLM generator component.
|
||||||
*/
|
*/
|
||||||
struct GeneratorOptions {
|
struct GeneratorOptions {
|
||||||
/// @brief Path to the LLM model file (gguf format).
|
/**
|
||||||
|
* @brief Path to the LLM model file (gguf format).
|
||||||
|
*/
|
||||||
std::filesystem::path model_path;
|
std::filesystem::path model_path;
|
||||||
|
|
||||||
/// @brief Use mocked generator instead of actual LLM inference.
|
/**
|
||||||
|
* @brief Use mocked generator instead of actual LLM inference.
|
||||||
|
*/
|
||||||
bool use_mocked = false;
|
bool use_mocked = false;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Specific sampling parameters for this generator.
|
||||||
/// @brief Specific sampling parameters for this generator.
|
* If nullopt, the application should use global defaults.
|
||||||
/// If nullopt, the application should use global defaults.
|
*/
|
||||||
std::optional<SamplingOptions> sampling;
|
std::optional<SamplingOptions> sampling;
|
||||||
};
|
};
|
||||||
|
|
||||||
@@ -112,18 +197,23 @@ struct GeneratorOptions {
|
|||||||
* @brief Configuration for the pipeline execution and output.
|
* @brief Configuration for the pipeline execution and output.
|
||||||
*/
|
*/
|
||||||
struct PipelineOptions {
|
struct PipelineOptions {
|
||||||
/// @brief Directory for generated artifacts.
|
/**
|
||||||
|
* @brief Directory for generated artifacts.
|
||||||
|
*/
|
||||||
std::filesystem::path output_path;
|
std::filesystem::path output_path;
|
||||||
|
|
||||||
/// @brief Directory that contains named prompt files (e.g.
|
/**
|
||||||
/// BREWERY_GENERATION.md).
|
* @brief Directory that contains named prompt files (e.g.
|
||||||
|
* BREWERY_GENERATION.md).
|
||||||
|
*/
|
||||||
std::filesystem::path prompt_dir;
|
std::filesystem::path prompt_dir;
|
||||||
|
|
||||||
/// @brief Path for application logs.
|
|
||||||
std::filesystem::path log_path;
|
std::filesystem::path log_path;
|
||||||
|
|
||||||
/// @brief Number of locations to sample from the dataset
|
/**
|
||||||
/// More locations -> more users/more breweries
|
* @brief Number of locations to sample from the dataset
|
||||||
|
* More locations -> more users/more breweries
|
||||||
|
*/
|
||||||
uint32_t location_count;
|
uint32_t location_count;
|
||||||
};
|
};
|
||||||
|
|
||||||
@@ -139,7 +229,7 @@ struct ApplicationOptions {
|
|||||||
// Function Declarations
|
// Function Declarations
|
||||||
// ============================================================================
|
// ============================================================================
|
||||||
|
|
||||||
std::optional<ApplicationOptions> ParseArguments(const int argc, char** argv,
|
std::optional<ApplicationOptions> ParseArguments(
|
||||||
std::shared_ptr<ILogger> logger = nullptr);
|
const int argc, char** argv, std::shared_ptr<ILogger> logger = nullptr);
|
||||||
|
|
||||||
#endif // BIERGARTEN_PIPELINE_INCLUDES_DATA_MODEL_MODELS_H_
|
#endif // BIERGARTEN_PIPELINE_INCLUDES_DATA_MODEL_MODELS_H_
|
||||||
|
|||||||
@@ -1,25 +0,0 @@
|
|||||||
#ifndef BIERGARTEN_PIPELINE_INCLUDES_JSON_HANDLING_JSON_LOADER_H_
|
|
||||||
#define BIERGARTEN_PIPELINE_INCLUDES_JSON_HANDLING_JSON_LOADER_H_
|
|
||||||
|
|
||||||
/**
|
|
||||||
* @file json_handling/json_loader.h
|
|
||||||
* @brief Loader API for curated location data.
|
|
||||||
*/
|
|
||||||
|
|
||||||
#include <filesystem>
|
|
||||||
#include <memory>
|
|
||||||
#include <vector>
|
|
||||||
|
|
||||||
#include "data_model/models.h"
|
|
||||||
#include "services/logging/logger.h"
|
|
||||||
|
|
||||||
/// @brief Loads curated world locations from a JSON file into memory.
|
|
||||||
class JsonLoader {
|
|
||||||
public:
|
|
||||||
/// @brief Parses a JSON array file and returns all location records.
|
|
||||||
static std::vector<Location> LoadLocations(
|
|
||||||
const std::filesystem::path& filepath,
|
|
||||||
std::shared_ptr<ILogger> logger = nullptr);
|
|
||||||
};
|
|
||||||
|
|
||||||
#endif // BIERGARTEN_PIPELINE_INCLUDES_JSON_HANDLING_JSON_LOADER_H_
|
|
||||||
@@ -106,4 +106,4 @@ inline void PrettyPrint(std::ostream& outstream,
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#endif
|
#endif // BIERGARTEN_PIPELINE_INCLUDES_JSON_HANDLING_PRETTY_PRINT_H_
|
||||||
|
|||||||
@@ -16,16 +16,10 @@
|
|||||||
*/
|
*/
|
||||||
class LlamaBackendState {
|
class LlamaBackendState {
|
||||||
public:
|
public:
|
||||||
/// @brief Initializes global llama backend state.
|
|
||||||
LlamaBackendState() { llama_backend_init(); }
|
LlamaBackendState() { llama_backend_init(); }
|
||||||
|
|
||||||
/// @brief Cleans up global llama backend state.
|
|
||||||
~LlamaBackendState() { llama_backend_free(); }
|
~LlamaBackendState() { llama_backend_free(); }
|
||||||
|
|
||||||
/// @brief Non-copyable type.
|
|
||||||
LlamaBackendState(const LlamaBackendState&) = delete;
|
LlamaBackendState(const LlamaBackendState&) = delete;
|
||||||
|
|
||||||
/// @brief Non-copyable type.
|
|
||||||
LlamaBackendState& operator=(const LlamaBackendState&) = delete;
|
LlamaBackendState& operator=(const LlamaBackendState&) = delete;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|||||||
@@ -0,0 +1,50 @@
|
|||||||
|
#ifndef BIERGARTEN_PIPELINE_INCLUDES_SERVICES_CURATED_DATA_CURATED_DATA_SERVICE_H_
|
||||||
|
#define BIERGARTEN_PIPELINE_INCLUDES_SERVICES_CURATED_DATA_CURATED_DATA_SERVICE_H_
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @file services/curated_data/curated_data_service.h
|
||||||
|
* @brief Abstraction for loading curated location, persona, and name data.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include <filesystem>
|
||||||
|
#include <unordered_map>
|
||||||
|
#include <unordered_set>
|
||||||
|
#include <vector>
|
||||||
|
|
||||||
|
#include "data_model/models.h"
|
||||||
|
|
||||||
|
using ForenameList = std::vector<ForenameEntry>;
|
||||||
|
using SurnameList = std::vector<std::string>;
|
||||||
|
using CityList = std::vector<City>;
|
||||||
|
using PersonasList = std::vector<UserPersona>;
|
||||||
|
using ForenamesByCountryMap = std::unordered_map<std::string, ForenameList>;
|
||||||
|
using SurnamesByCountryMap = std::unordered_map<std::string, SurnameList>;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Interface for services that load curated data used to seed
|
||||||
|
* brewery/user generation.
|
||||||
|
*/
|
||||||
|
class ICuratedDataService {
|
||||||
|
public:
|
||||||
|
ICuratedDataService() = default;
|
||||||
|
virtual ~ICuratedDataService() = default;
|
||||||
|
|
||||||
|
ICuratedDataService(const ICuratedDataService&) = delete;
|
||||||
|
ICuratedDataService& operator=(const ICuratedDataService&) = delete;
|
||||||
|
ICuratedDataService(ICuratedDataService&&) = delete;
|
||||||
|
ICuratedDataService& operator=(ICuratedDataService&&) = delete;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Loads all curated location records.
|
||||||
|
*/
|
||||||
|
virtual const CityList& LoadCities() = 0;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Loads all curated persona records.
|
||||||
|
*/
|
||||||
|
virtual const PersonasList& LoadPersonas() = 0;
|
||||||
|
virtual const ForenamesByCountryMap& LoadForenamesByCountry() = 0;
|
||||||
|
virtual const SurnamesByCountryMap& LoadSurnamesByCountry() = 0;
|
||||||
|
};
|
||||||
|
|
||||||
|
#endif // BIERGARTEN_PIPELINE_INCLUDES_SERVICES_CURATED_DATA_CURATED_DATA_SERVICE_H_
|
||||||
@@ -0,0 +1,63 @@
|
|||||||
|
#ifndef BIERGARTEN_PIPELINE_INCLUDES_JSON_HANDLING_JSON_LOADER_H_
|
||||||
|
#define BIERGARTEN_PIPELINE_INCLUDES_JSON_HANDLING_JSON_LOADER_H_
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @file json_handling/json_loader.h
|
||||||
|
* @brief JSON-backed implementation of ICuratedDataService.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include <filesystem>
|
||||||
|
|
||||||
|
#include "data_model/models.h"
|
||||||
|
#include "services/curated_data/curated_data_service.h"
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief File locations for the curated JSON fixtures consumed by
|
||||||
|
* CuratedJsonDataService.
|
||||||
|
*/
|
||||||
|
struct CuratedDataFilePaths {
|
||||||
|
std::filesystem::path cities_path;
|
||||||
|
std::filesystem::path personas_path;
|
||||||
|
std::filesystem::path forenames_path;
|
||||||
|
std::filesystem::path surnames_path;
|
||||||
|
};
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Loads curated location, persona, and name data from JSON files.
|
||||||
|
*/
|
||||||
|
class CuratedJsonDataService final : public ICuratedDataService {
|
||||||
|
struct cache {
|
||||||
|
CityList locations;
|
||||||
|
PersonasList personas;
|
||||||
|
ForenamesByCountryMap forenames_by_country;
|
||||||
|
SurnamesByCountryMap surnames_by_country;
|
||||||
|
|
||||||
|
cache() = default;
|
||||||
|
~cache() = default;
|
||||||
|
};
|
||||||
|
|
||||||
|
CuratedDataFilePaths filepaths_;
|
||||||
|
cache cache_;
|
||||||
|
|
||||||
|
public:
|
||||||
|
explicit CuratedJsonDataService(CuratedDataFilePaths filepaths);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Parses a JSON array file and returns all city records.
|
||||||
|
*/
|
||||||
|
const CityList& LoadCities() override;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Parses a JSON array file and returns all persona records.
|
||||||
|
*/
|
||||||
|
const PersonasList& LoadPersonas() override;
|
||||||
|
|
||||||
|
const ForenamesByCountryMap& LoadForenamesByCountry() override;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Parses a JSON file and returns all the forenames per country.
|
||||||
|
*/
|
||||||
|
const SurnamesByCountryMap& LoadSurnamesByCountry() override;
|
||||||
|
};
|
||||||
|
|
||||||
|
#endif // BIERGARTEN_PIPELINE_INCLUDES_JSON_HANDLING_JSON_LOADER_H_
|
||||||
@@ -0,0 +1,38 @@
|
|||||||
|
#ifndef BIERGARTEN_PIPELINE_INCLUDES_SERVICES_CURATED_DATA_MOCK_CURATED_DATA_SERVICE_H_
|
||||||
|
#define BIERGARTEN_PIPELINE_INCLUDES_SERVICES_CURATED_DATA_MOCK_CURATED_DATA_SERVICE_H_
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @file services/curated_data/mock_curated_data_service.h
|
||||||
|
* @brief In-memory ICuratedDataService backed by a small fixed dataset, used
|
||||||
|
* when file-backed curated data is disabled (mock mode).
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include <filesystem>
|
||||||
|
|
||||||
|
#include "data_model/models.h"
|
||||||
|
#include "services/curated_data/curated_data_service.h"
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Curated data service returning a small fixed in-memory dataset in
|
||||||
|
* place of the JSON fixture files used by JsonLoader.
|
||||||
|
*/
|
||||||
|
class MockCuratedDataService final : public ICuratedDataService {
|
||||||
|
public:
|
||||||
|
MockCuratedDataService();
|
||||||
|
|
||||||
|
const CityList& LoadCities() override;
|
||||||
|
|
||||||
|
const PersonasList& LoadPersonas() override;
|
||||||
|
|
||||||
|
const ForenamesByCountryMap& LoadForenamesByCountry() override;
|
||||||
|
|
||||||
|
const SurnamesByCountryMap& LoadSurnamesByCountry() override;
|
||||||
|
|
||||||
|
private:
|
||||||
|
CityList locations_;
|
||||||
|
PersonasList personas_;
|
||||||
|
ForenamesByCountryMap forenames_by_country_;
|
||||||
|
SurnamesByCountryMap surnames_by_country_;
|
||||||
|
};
|
||||||
|
|
||||||
|
#endif // BIERGARTEN_PIPELINE_INCLUDES_SERVICES_CURATED_DATA_MOCK_CURATED_DATA_SERVICE_H_
|
||||||
@@ -16,8 +16,6 @@
|
|||||||
class IExportService {
|
class IExportService {
|
||||||
public:
|
public:
|
||||||
IExportService() = default;
|
IExportService() = default;
|
||||||
|
|
||||||
/// @brief Virtual destructor for polymorphic cleanup.
|
|
||||||
virtual ~IExportService() = default;
|
virtual ~IExportService() = default;
|
||||||
|
|
||||||
IExportService(const IExportService&) = delete;
|
IExportService(const IExportService&) = delete;
|
||||||
@@ -25,7 +23,9 @@ class IExportService {
|
|||||||
IExportService(IExportService&&) = delete;
|
IExportService(IExportService&&) = delete;
|
||||||
IExportService& operator=(IExportService&&) = delete;
|
IExportService& operator=(IExportService&&) = delete;
|
||||||
|
|
||||||
/// @brief Prepares the export destination for a new run.
|
/**
|
||||||
|
* @brief Prepares the export destination for a new run.
|
||||||
|
*/
|
||||||
virtual void Initialize() = 0;
|
virtual void Initialize() = 0;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@@ -33,9 +33,18 @@ class IExportService {
|
|||||||
*
|
*
|
||||||
* @param brewery Generated brewery payload to store.
|
* @param brewery Generated brewery payload to store.
|
||||||
*/
|
*/
|
||||||
virtual uint64_t ProcessRecord(const GeneratedBrewery& brewery) = 0;
|
virtual uint64_t ProcessRecord(const BreweryRecord& brewery) = 0;
|
||||||
|
|
||||||
/// @brief Finalizes the export destination.
|
/**
|
||||||
|
* @brief Persists one generated user record.
|
||||||
|
*
|
||||||
|
* @param user Generated user payload to store.
|
||||||
|
*/
|
||||||
|
virtual uint64_t ProcessRecord(const UserRecord& user) = 0;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Finalizes the export destination.
|
||||||
|
*/
|
||||||
virtual void Finalize() = 0;
|
virtual void Finalize() = 0;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|||||||
@@ -12,7 +12,7 @@
|
|||||||
#include <string>
|
#include <string>
|
||||||
#include <string_view>
|
#include <string_view>
|
||||||
|
|
||||||
#include "sqlite_handle_types.h"
|
#include "services/database/sqlite_handle_types.h"
|
||||||
|
|
||||||
namespace sqlite_export_service_internal {
|
namespace sqlite_export_service_internal {
|
||||||
|
|
||||||
|
|||||||
@@ -12,9 +12,9 @@
|
|||||||
#include <unordered_map>
|
#include <unordered_map>
|
||||||
|
|
||||||
#include "data_model/models.h"
|
#include "data_model/models.h"
|
||||||
#include "../datetime/date_time_provider.h"
|
#include "services/database/export_service.h"
|
||||||
#include "export_service.h"
|
#include "services/database/sqlite_export_service_helpers.h"
|
||||||
#include "sqlite_export_service_helpers.h"
|
#include "services/datetime/date_time_provider.h"
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @brief Persists generated brewery records into a fresh SQLite database.
|
* @brief Persists generated brewery records into a fresh SQLite database.
|
||||||
@@ -30,7 +30,8 @@ class SqliteExportService final : public IExportService {
|
|||||||
SqliteExportService& operator=(SqliteExportService&&) = delete;
|
SqliteExportService& operator=(SqliteExportService&&) = delete;
|
||||||
|
|
||||||
void Initialize() override;
|
void Initialize() override;
|
||||||
uint64_t ProcessRecord(const GeneratedBrewery& brewery) override;
|
uint64_t ProcessRecord(const BreweryRecord& brewery) override;
|
||||||
|
uint64_t ProcessRecord(const UserRecord& user) override;
|
||||||
void Finalize() override;
|
void Finalize() override;
|
||||||
|
|
||||||
private:
|
private:
|
||||||
@@ -44,17 +45,29 @@ class SqliteExportService final : public IExportService {
|
|||||||
void RollbackAndCloseNoThrow() noexcept;
|
void RollbackAndCloseNoThrow() noexcept;
|
||||||
|
|
||||||
[[nodiscard]] std::filesystem::path BuildDatabasePath() const;
|
[[nodiscard]] std::filesystem::path BuildDatabasePath() const;
|
||||||
[[nodiscard]] static std::string BuildLocationKey(const Location& location);
|
[[nodiscard]] static std::string BuildCityKey(const City& location);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Returns the row id for @p location, inserting it first if it has
|
||||||
|
* not already been seen during this run.
|
||||||
|
*
|
||||||
|
* Shared by both ProcessRecord() overloads so breweries and users
|
||||||
|
* referencing the same location resolve to the same row.
|
||||||
|
*/
|
||||||
|
[[nodiscard]] sqlite3_int64 ResolveCityId(const City& location);
|
||||||
|
|
||||||
std::unique_ptr<IDateTimeProvider> date_time_provider_;
|
std::unique_ptr<IDateTimeProvider> date_time_provider_;
|
||||||
std::filesystem::path output_path_;
|
std::filesystem::path output_path_;
|
||||||
std::string run_timestamp_utc_;
|
std::string run_timestamp_utc_;
|
||||||
std::filesystem::path database_path_;
|
std::filesystem::path database_path_;
|
||||||
SqliteDatabaseHandle db_handle_;
|
SqliteDatabaseHandle db_handle_;
|
||||||
SqliteStatementHandle insert_location_stmt_;
|
SqliteStatementHandle insert_city_stmt_;
|
||||||
SqliteStatementHandle insert_brewery_stmt_;
|
SqliteStatementHandle insert_brewery_stmt_;
|
||||||
|
SqliteStatementHandle insert_brewery_address_stmt_;
|
||||||
|
SqliteStatementHandle insert_user_stmt_;
|
||||||
|
SqliteStatementHandle insert_user_address_stmt_;
|
||||||
bool transaction_open_ = false;
|
bool transaction_open_ = false;
|
||||||
std::unordered_map<std::string, sqlite3_int64> location_cache_;
|
std::unordered_map<std::string, sqlite3_int64> city_cache_;
|
||||||
};
|
};
|
||||||
|
|
||||||
#endif // BIERGARTEN_PIPELINE_INCLUDES_SERVICES_DATABASE_SQLITE_EXPORT_SERVICE_H_
|
#endif // BIERGARTEN_PIPELINE_INCLUDES_SERVICES_DATABASE_SQLITE_EXPORT_SERVICE_H_
|
||||||
|
|||||||
@@ -3,8 +3,8 @@
|
|||||||
|
|
||||||
/* Umbrella header for backward compatibility. */
|
/* Umbrella header for backward compatibility. */
|
||||||
|
|
||||||
#include "sqlite_connection_helpers.h"
|
#include "services/database/sqlite_connection_helpers.h"
|
||||||
#include "sqlite_handle_types.h"
|
#include "services/database/sqlite_handle_types.h"
|
||||||
#include "sqlite_statement_helpers.h"
|
#include "services/database/sqlite_statement_helpers.h"
|
||||||
|
|
||||||
#endif // BIERGARTEN_PIPELINE_INCLUDES_SERVICES_DATABASE_SQLITE_EXPORT_SERVICE_HELPERS_H_
|
#endif // BIERGARTEN_PIPELINE_INCLUDES_SERVICES_DATABASE_SQLITE_EXPORT_SERVICE_HELPERS_H_
|
||||||
|
|||||||
@@ -24,8 +24,10 @@ using SqliteDatabaseHandle = std::unique_ptr<sqlite3, SqliteDatabaseDeleter>;
|
|||||||
using SqliteStatementHandle =
|
using SqliteStatementHandle =
|
||||||
std::unique_ptr<sqlite3_stmt, SqliteStatementDeleter>;
|
std::unique_ptr<sqlite3_stmt, SqliteStatementDeleter>;
|
||||||
|
|
||||||
|
// Represents a parameter that is bound to a prepared SQLite statement.
|
||||||
|
// N.B. indices are 1 based.
|
||||||
template <typename T>
|
template <typename T>
|
||||||
struct BindParam {
|
struct BoundParam {
|
||||||
int index;
|
int index;
|
||||||
T value;
|
T value;
|
||||||
std::string_view action;
|
std::string_view action;
|
||||||
|
|||||||
@@ -13,13 +13,13 @@
|
|||||||
#include <string_view>
|
#include <string_view>
|
||||||
#include <vector>
|
#include <vector>
|
||||||
|
|
||||||
#include "sqlite_handle_types.h"
|
#include "services/database/sqlite_handle_types.h"
|
||||||
|
|
||||||
namespace sqlite_export_service_internal {
|
namespace sqlite_export_service_internal {
|
||||||
|
|
||||||
inline constexpr std::string_view kCreateLocationsTableSql = R"sql(
|
inline constexpr std::string_view kCreateCitiesTableSql = R"sql(
|
||||||
|
|
||||||
CREATE TABLE IF NOT EXISTS locations (
|
CREATE TABLE IF NOT EXISTS cities (
|
||||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||||
city TEXT NOT NULL,
|
city TEXT NOT NULL,
|
||||||
state_province TEXT NOT NULL,
|
state_province TEXT NOT NULL,
|
||||||
@@ -27,9 +27,7 @@ CREATE TABLE IF NOT EXISTS locations (
|
|||||||
country TEXT NOT NULL,
|
country TEXT NOT NULL,
|
||||||
iso3166_1 TEXT NOT NULL,
|
iso3166_1 TEXT NOT NULL,
|
||||||
local_languages_json TEXT NOT NULL,
|
local_languages_json TEXT NOT NULL,
|
||||||
latitude REAL NOT NULL,
|
UNIQUE(city, state_province, iso3166_2, country)
|
||||||
longitude REAL NOT NULL,
|
|
||||||
UNIQUE(city, state_province, iso3166_2, country, latitude, longitude)
|
|
||||||
);
|
);
|
||||||
|
|
||||||
)sql";
|
)sql";
|
||||||
@@ -38,34 +36,83 @@ inline constexpr std::string_view kCreateBreweriesTableSql = R"sql(
|
|||||||
|
|
||||||
CREATE TABLE IF NOT EXISTS breweries (
|
CREATE TABLE IF NOT EXISTS breweries (
|
||||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||||
location_id INTEGER NOT NULL,
|
city_id INTEGER NOT NULL,
|
||||||
name_en TEXT NOT NULL,
|
name_en TEXT NOT NULL,
|
||||||
description_en TEXT NOT NULL,
|
description_en TEXT NOT NULL,
|
||||||
name_local TEXT NOT NULL,
|
name_local TEXT NOT NULL,
|
||||||
description_local TEXT NOT NULL,
|
description_local TEXT NOT NULL,
|
||||||
FOREIGN KEY(location_id) REFERENCES locations(id) ON DELETE CASCADE
|
FOREIGN KEY(city_id) REFERENCES cities(id) ON DELETE CASCADE
|
||||||
);
|
);
|
||||||
|
|
||||||
CREATE INDEX IF NOT EXISTS idx_breweries_location_id ON breweries(location_id);
|
CREATE INDEX IF NOT EXISTS idx_breweries_city_id ON breweries(city_id);
|
||||||
|
)sql";
|
||||||
|
|
||||||
|
inline constexpr std::string_view kCreateUsersTableSql = R"sql(
|
||||||
|
|
||||||
|
CREATE TABLE IF NOT EXISTS users (
|
||||||
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||||
|
first_name TEXT NOT NULL,
|
||||||
|
last_name TEXT NOT NULL,
|
||||||
|
gender TEXT NOT NULL,
|
||||||
|
username TEXT NOT NULL,
|
||||||
|
bio TEXT NOT NULL,
|
||||||
|
activity_weight REAL NOT NULL,
|
||||||
|
email TEXT NOT NULL UNIQUE,
|
||||||
|
date_of_birth TEXT NOT NULL
|
||||||
|
);
|
||||||
|
|
||||||
)sql";
|
)sql";
|
||||||
|
|
||||||
inline constexpr std::string_view kInsertLocationSql = R"sql(
|
inline constexpr std::string_view kCreateBreweryAddressTableSql = R"sql(
|
||||||
INSERT INTO locations (
|
CREATE TABLE IF NOT EXISTS brewery_addresses (
|
||||||
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||||
|
brewery_id INTEGER NOT NULL,
|
||||||
|
address_line_1 TEXT,
|
||||||
|
address_line_2 TEXT,
|
||||||
|
postal_code TEXT NOT NULL,
|
||||||
|
city_id INTEGER NOT NULL,
|
||||||
|
FOREIGN KEY(brewery_id) REFERENCES breweries(id) ON DELETE CASCADE,
|
||||||
|
FOREIGN KEY(city_id) REFERENCES cities(id) ON DELETE CASCADE
|
||||||
|
);
|
||||||
|
|
||||||
|
CREATE INDEX IF NOT EXISTS idx_brewery_addresses_brewery_id
|
||||||
|
ON brewery_addresses(brewery_id);
|
||||||
|
CREATE INDEX IF NOT EXISTS idx_brewery_addresses_city_id
|
||||||
|
ON brewery_addresses(city_id);
|
||||||
|
)sql";
|
||||||
|
|
||||||
|
inline constexpr std::string_view kCreateUserAddressTableSql = R"sql(
|
||||||
|
CREATE TABLE IF NOT EXISTS user_addresses (
|
||||||
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||||
|
user_id INTEGER NOT NULL,
|
||||||
|
address_line_1 TEXT,
|
||||||
|
address_line_2 TEXT,
|
||||||
|
postal_code TEXT,
|
||||||
|
city_id INTEGER NOT NULL,
|
||||||
|
FOREIGN KEY(user_id) REFERENCES users(id) ON DELETE CASCADE,
|
||||||
|
FOREIGN KEY(city_id) REFERENCES cities(id) ON DELETE CASCADE
|
||||||
|
);
|
||||||
|
|
||||||
|
CREATE INDEX IF NOT EXISTS idx_user_addresses_user_id
|
||||||
|
ON user_addresses(user_id);
|
||||||
|
CREATE INDEX IF NOT EXISTS idx_user_addresses_city_id
|
||||||
|
ON user_addresses(city_id);
|
||||||
|
)sql";
|
||||||
|
|
||||||
|
inline constexpr std::string_view kInsertCitySql = R"sql(
|
||||||
|
INSERT INTO cities (
|
||||||
city,
|
city,
|
||||||
state_province,
|
state_province,
|
||||||
iso3166_2,
|
iso3166_2,
|
||||||
country,
|
country,
|
||||||
iso3166_1,
|
iso3166_1,
|
||||||
local_languages_json,
|
local_languages_json
|
||||||
latitude,
|
) VALUES (?, ?, ?, ?, ?, ?);
|
||||||
longitude
|
|
||||||
) VALUES (?, ?, ?, ?, ?, ?, ?, ?);
|
|
||||||
)sql";
|
)sql";
|
||||||
|
|
||||||
inline constexpr std::string_view kInsertBrewerySql = R"sql(
|
inline constexpr std::string_view kInsertBrewerySql = R"sql(
|
||||||
INSERT INTO breweries (
|
INSERT INTO breweries (
|
||||||
location_id,
|
city_id,
|
||||||
name_en,
|
name_en,
|
||||||
description_en,
|
description_en,
|
||||||
name_local,
|
name_local,
|
||||||
@@ -73,20 +120,74 @@ INSERT INTO breweries (
|
|||||||
) VALUES (?, ?, ?, ?, ?);
|
) VALUES (?, ?, ?, ?, ?);
|
||||||
)sql";
|
)sql";
|
||||||
|
|
||||||
inline constexpr int kLocationCityBindIndex = 1;
|
inline constexpr std::string_view kInsertUserSql = R"sql(
|
||||||
inline constexpr int kLocationStateProvinceBindIndex = 2;
|
INSERT INTO users (
|
||||||
inline constexpr int kLocationIso31662BindIndex = 3;
|
first_name,
|
||||||
inline constexpr int kLocationCountryBindIndex = 4;
|
last_name,
|
||||||
inline constexpr int kLocationIso31661BindIndex = 5;
|
gender,
|
||||||
inline constexpr int kLocationLanguagesBindIndex = 6;
|
username,
|
||||||
inline constexpr int kLocationLatitudeBindIndex = 7;
|
bio,
|
||||||
inline constexpr int kLocationLongitudeBindIndex = 8;
|
activity_weight,
|
||||||
|
email,
|
||||||
|
date_of_birth
|
||||||
|
) VALUES (?, ?, ?, ?, ?, ?, ?, ?);
|
||||||
|
)sql";
|
||||||
|
|
||||||
inline constexpr int kBreweryLocationIdBindIndex = 1;
|
inline constexpr std::string_view kInsertBreweryAddressSql = R"sql(
|
||||||
inline constexpr int kBreweryEnglishNameBindIndex = 2;
|
INSERT INTO brewery_addresses (
|
||||||
inline constexpr int kBreweryEnglishDescriptionBindIndex = 3;
|
brewery_id,
|
||||||
inline constexpr int kBreweryLocalNameBindIndex = 4;
|
postal_code,
|
||||||
inline constexpr int kBreweryLocalDescriptionBindIndex = 5;
|
city_id
|
||||||
|
) VALUES (?, ?, ?);
|
||||||
|
)sql";
|
||||||
|
|
||||||
|
inline constexpr std::string_view kInsertUserAddressSql = R"sql(
|
||||||
|
INSERT INTO user_addresses (
|
||||||
|
user_id,
|
||||||
|
city_id
|
||||||
|
) VALUES (?, ?);
|
||||||
|
)sql";
|
||||||
|
|
||||||
|
// sqlite3_bind_*() parameter indices are 1-based, matching the "?"
|
||||||
|
// placeholder order in the SQL above.
|
||||||
|
enum CityBindIndex {
|
||||||
|
kCityNameBindIndex = 1,
|
||||||
|
kCityStateProvinceBindIndex,
|
||||||
|
kCityIso31662BindIndex,
|
||||||
|
kCityCountryBindIndex,
|
||||||
|
kCityIso31661BindIndex,
|
||||||
|
kCityLanguagesBindIndex,
|
||||||
|
};
|
||||||
|
|
||||||
|
enum BreweryBindIndex {
|
||||||
|
kBreweryCityIdBindIndex = 1,
|
||||||
|
kBreweryEnglishNameBindIndex,
|
||||||
|
kBreweryEnglishDescriptionBindIndex,
|
||||||
|
kBreweryLocalNameBindIndex,
|
||||||
|
kBreweryLocalDescriptionBindIndex,
|
||||||
|
};
|
||||||
|
|
||||||
|
enum UserBindIndex {
|
||||||
|
kUserFirstNameBindIndex = 1,
|
||||||
|
kUserLastNameBindIndex,
|
||||||
|
kUserGenderBindIndex,
|
||||||
|
kUserUsernameBindIndex,
|
||||||
|
kUserBioBindIndex,
|
||||||
|
kUserActivityWeightBindIndex,
|
||||||
|
kUserEmailBindIndex,
|
||||||
|
kUserDateOfBirthBindIndex,
|
||||||
|
};
|
||||||
|
|
||||||
|
enum BreweryAddressBindIndex {
|
||||||
|
kBreweryAddressBreweryIdBindIndex = 1,
|
||||||
|
kBreweryAddressPostalCodeBindIndex,
|
||||||
|
kBreweryAddressCityIdBindIndex,
|
||||||
|
};
|
||||||
|
|
||||||
|
enum UserAddressBindIndex {
|
||||||
|
kUserAddressUserIdBindIndex = 1,
|
||||||
|
kUserAddressCityIdBindIndex,
|
||||||
|
};
|
||||||
|
|
||||||
SqliteStatementHandle PrepareStatement(const SqliteDatabaseHandle& db_handle,
|
SqliteStatementHandle PrepareStatement(const SqliteDatabaseHandle& db_handle,
|
||||||
std::string_view sql,
|
std::string_view sql,
|
||||||
@@ -95,13 +196,13 @@ SqliteStatementHandle PrepareStatement(const SqliteDatabaseHandle& db_handle,
|
|||||||
void ResetStatement(SqliteStatementHandle& statement);
|
void ResetStatement(SqliteStatementHandle& statement);
|
||||||
|
|
||||||
void Bind(const SqliteStatementHandle& statement,
|
void Bind(const SqliteStatementHandle& statement,
|
||||||
const BindParam<std::string_view>& param);
|
const BoundParam<std::string_view>& param);
|
||||||
|
|
||||||
void Bind(const SqliteStatementHandle& statement,
|
void Bind(const SqliteStatementHandle& statement,
|
||||||
const BindParam<double>& param);
|
const BoundParam<double>& param);
|
||||||
|
|
||||||
void Bind(const SqliteStatementHandle& statement,
|
void Bind(const SqliteStatementHandle& statement,
|
||||||
const BindParam<sqlite3_int64>& param);
|
const BoundParam<sqlite3_int64>& param);
|
||||||
|
|
||||||
void StepStatement(const SqliteDatabaseHandle& db_handle,
|
void StepStatement(const SqliteDatabaseHandle& db_handle,
|
||||||
const SqliteStatementHandle& statement,
|
const SqliteStatementHandle& statement,
|
||||||
|
|||||||
@@ -18,7 +18,6 @@
|
|||||||
*/
|
*/
|
||||||
class IDateTimeProvider {
|
class IDateTimeProvider {
|
||||||
public:
|
public:
|
||||||
/// @brief Virtual destructor for polymorphic cleanup.
|
|
||||||
virtual ~IDateTimeProvider() = default;
|
virtual ~IDateTimeProvider() = default;
|
||||||
|
|
||||||
IDateTimeProvider() = default;
|
IDateTimeProvider() = default;
|
||||||
|
|||||||
@@ -15,7 +15,6 @@
|
|||||||
*/
|
*/
|
||||||
class IEnrichmentService {
|
class IEnrichmentService {
|
||||||
public:
|
public:
|
||||||
/// @brief Virtual destructor for polymorphic cleanup.
|
|
||||||
virtual ~IEnrichmentService() = default;
|
virtual ~IEnrichmentService() = default;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@@ -24,7 +23,7 @@ class IEnrichmentService {
|
|||||||
* @param loc Location to enrich.
|
* @param loc Location to enrich.
|
||||||
* @return Context text, or an empty string if unavailable.
|
* @return Context text, or an empty string if unavailable.
|
||||||
*/
|
*/
|
||||||
virtual std::string GetLocationContext(const Location& loc) = 0;
|
virtual std::string GetLocationContext(const City& loc) = 0;
|
||||||
};
|
};
|
||||||
|
|
||||||
#endif // BIERGARTEN_PIPELINE_INCLUDES_SERVICES_ENRICHMENT_ENRICHMENT_SERVICE_H_
|
#endif // BIERGARTEN_PIPELINE_INCLUDES_SERVICES_ENRICHMENT_ENRICHMENT_SERVICE_H_
|
||||||
|
|||||||
@@ -1,16 +1,21 @@
|
|||||||
//
|
|
||||||
// Created by aaronpo on 13/05/2026.
|
|
||||||
//
|
|
||||||
|
|
||||||
#ifndef BIERGARTEN_PIPELINE_INCLUDES_SERVICES_ENRICHMENT_MOCK_ENRICHMENT_H_
|
#ifndef BIERGARTEN_PIPELINE_INCLUDES_SERVICES_ENRICHMENT_MOCK_ENRICHMENT_H_
|
||||||
#define BIERGARTEN_PIPELINE_INCLUDES_SERVICES_ENRICHMENT_MOCK_ENRICHMENT_H_
|
#define BIERGARTEN_PIPELINE_INCLUDES_SERVICES_ENRICHMENT_MOCK_ENRICHMENT_H_
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @file services/enrichment/mock_enrichment.h
|
||||||
|
* @brief No-op IEnrichmentService used when network enrichment is disabled.
|
||||||
|
*/
|
||||||
|
|
||||||
#include <string>
|
#include <string>
|
||||||
|
|
||||||
#include "enrichment_service.h"
|
#include "services/enrichment/enrichment_service.h"
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Enrichment service that returns no context for any location.
|
||||||
|
*/
|
||||||
class MockEnrichmentService final : public IEnrichmentService {
|
class MockEnrichmentService final : public IEnrichmentService {
|
||||||
public:
|
public:
|
||||||
std::string GetLocationContext(const Location& /*loc*/) override {
|
std::string GetLocationContext(const City& /*loc*/) override {
|
||||||
return {};
|
return {};
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|||||||
@@ -11,25 +11,37 @@
|
|||||||
#include <string_view>
|
#include <string_view>
|
||||||
#include <unordered_map>
|
#include <unordered_map>
|
||||||
|
|
||||||
#include "enrichment_service.h"
|
#include "services/enrichment/enrichment_service.h"
|
||||||
#include "services/logging/logger.h"
|
#include "services/logging/logger.h"
|
||||||
#include "web_client/web_client.h"
|
#include "web_client/web_client.h"
|
||||||
|
|
||||||
/// @brief Provides Wikipedia summary lookups backed by cached raw extracts.
|
/**
|
||||||
|
* @brief Provides Wikipedia summary lookups backed by cached raw extracts.
|
||||||
|
*/
|
||||||
class WikipediaEnrichmentService final : public IEnrichmentService {
|
class WikipediaEnrichmentService final : public IEnrichmentService {
|
||||||
public:
|
public:
|
||||||
/// @brief Creates a new Wikipedia service with the provided web client.
|
/**
|
||||||
|
* @brief Creates a new Wikipedia service with the provided web client.
|
||||||
|
*/
|
||||||
explicit WikipediaEnrichmentService(std::unique_ptr<WebClient> client,
|
explicit WikipediaEnrichmentService(std::unique_ptr<WebClient> client,
|
||||||
std::shared_ptr<ILogger> logger);
|
std::shared_ptr<ILogger> logger);
|
||||||
|
|
||||||
/// @brief Returns the Wikipedia-derived context for a location.
|
/**
|
||||||
[[nodiscard]] std::string GetLocationContext(const Location& loc) override;
|
* @brief Returns the Wikipedia-derived context for a location.
|
||||||
|
*/
|
||||||
|
[[nodiscard]] std::string GetLocationContext(const City& loc) override;
|
||||||
|
|
||||||
private:
|
private:
|
||||||
std::string FetchExtract(std::string_view query);
|
std::string FetchExtract(std::string_view query);
|
||||||
std::unique_ptr<WebClient> client_;
|
std::unique_ptr<WebClient> client_;
|
||||||
std::shared_ptr<ILogger> logger_;
|
std::shared_ptr<ILogger> logger_;
|
||||||
/// @brief Canonical cache for raw Wikipedia query extracts.
|
/**
|
||||||
|
* @brief Cache for raw Wikipedia query extracts, keyed by query string.
|
||||||
|
*
|
||||||
|
* GetLocationContext() always queries "brewing" and reuses "beer in
|
||||||
|
* {country}" for every city in that country, so caching avoids refetching
|
||||||
|
* the same extract across locations in a run.
|
||||||
|
*/
|
||||||
std::unordered_map<std::string, std::string> extract_cache_;
|
std::unordered_map<std::string, std::string> extract_cache_;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|||||||
@@ -20,10 +20,22 @@
|
|||||||
* @brief Severity levels supported by the logging infra.
|
* @brief Severity levels supported by the logging infra.
|
||||||
*/
|
*/
|
||||||
enum class LogLevel {
|
enum class LogLevel {
|
||||||
Debug, ///< Development/debugging information.
|
/**
|
||||||
Info, ///< General informational messages.
|
* @brief Development/debugging information.
|
||||||
Warn, ///< Warning conditions.
|
*/
|
||||||
Error, ///< Error conditions.
|
Debug,
|
||||||
|
/**
|
||||||
|
* @brief General informational messages.
|
||||||
|
*/
|
||||||
|
Info,
|
||||||
|
/**
|
||||||
|
* @brief Warning conditions.
|
||||||
|
*/
|
||||||
|
Warn,
|
||||||
|
/**
|
||||||
|
* @brief Error conditions.
|
||||||
|
*/
|
||||||
|
Error,
|
||||||
};
|
};
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@@ -34,18 +46,44 @@ enum class LogLevel {
|
|||||||
* pipeline that emitted it.
|
* pipeline that emitted it.
|
||||||
*/
|
*/
|
||||||
enum class PipelinePhase {
|
enum class PipelinePhase {
|
||||||
Startup, ///< Initialization and validation.
|
/**
|
||||||
UserGeneration, ///< User profile generation.
|
* @brief Initialization and validation.
|
||||||
BreweryAndBeerGeneration, ///< Brewery and beer data generation.
|
*/
|
||||||
CheckinGeneration, ///< Checkin (visit) record generation.
|
Startup,
|
||||||
RatingGeneration, ///< Rating and review generation.
|
/**
|
||||||
FollowGeneration, ///< Follow relationship generation.
|
* @brief Location/context enrichment (e.g. Wikipedia lookups).
|
||||||
Teardown, ///< Finalization and cleanup.
|
*/
|
||||||
|
Enrichment,
|
||||||
|
/**
|
||||||
|
* @brief User profile generation.
|
||||||
|
*/
|
||||||
|
UserGeneration,
|
||||||
|
/**
|
||||||
|
* @brief Brewery and beer data generation.
|
||||||
|
*/
|
||||||
|
BreweryAndBeerGeneration,
|
||||||
|
/**
|
||||||
|
* @brief Checkin (visit) record generation.
|
||||||
|
*/
|
||||||
|
CheckinGeneration,
|
||||||
|
/**
|
||||||
|
* @brief Rating and review generation.
|
||||||
|
*/
|
||||||
|
RatingGeneration,
|
||||||
|
/**
|
||||||
|
* @brief Follow relationship generation.
|
||||||
|
*/
|
||||||
|
FollowGeneration,
|
||||||
|
/**
|
||||||
|
* @brief Finalization and cleanup.
|
||||||
|
*/
|
||||||
|
Teardown,
|
||||||
};
|
};
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @struct LogDTO
|
* @struct LogDTO
|
||||||
* @brief User-provided subset of log fields. Used to capture call-site info transparently.
|
* @brief User-provided subset of log fields. Used to capture call-site info
|
||||||
|
* transparently.
|
||||||
*/
|
*/
|
||||||
struct LogDTO {
|
struct LogDTO {
|
||||||
LogLevel level;
|
LogLevel level;
|
||||||
@@ -64,25 +102,35 @@ struct LogDTO {
|
|||||||
* before the entry is dispatched.
|
* 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{};
|
||||||
|
|
||||||
/// @brief Source location where the log call was made.
|
/**
|
||||||
|
* @brief Source location where the log call was made.
|
||||||
|
*/
|
||||||
std::source_location origin{};
|
std::source_location origin{};
|
||||||
|
|
||||||
/// @brief Thread responsible for emitting the log.
|
/**
|
||||||
|
* @brief Thread responsible for emitting the log.
|
||||||
|
*/
|
||||||
std::thread::id thread_id{};
|
std::thread::id thread_id{};
|
||||||
|
|
||||||
|
/**
|
||||||
/// @brief Severity level of this entry.
|
* @brief Severity level of this entry.
|
||||||
|
*/
|
||||||
LogLevel level;
|
LogLevel level;
|
||||||
|
|
||||||
/// @brief Pipeline phase associated with the entry.
|
/**
|
||||||
|
* @brief Pipeline phase associated with the entry.
|
||||||
|
*/
|
||||||
PipelinePhase phase;
|
PipelinePhase phase;
|
||||||
|
|
||||||
/// @brief Log message text.
|
/**
|
||||||
|
* @brief Log message text.
|
||||||
|
*/
|
||||||
std::string message;
|
std::string message;
|
||||||
|
|
||||||
};
|
};
|
||||||
|
|
||||||
#endif // BIERGARTEN_PIPELINE_INCLUDES_SERVICES_LOGGING_LOG_ENTRY_H_
|
#endif // BIERGARTEN_PIPELINE_INCLUDES_SERVICES_LOGGING_LOG_ENTRY_H_
|
||||||
|
|||||||
@@ -6,8 +6,8 @@
|
|||||||
* a bounded channel for later processing by the dispatcher.
|
* a bounded channel for later processing by the dispatcher.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#ifndef BIERGARTEN_PIPELINE_INCLUDES_SERVICES_LOGGING_CHANNEL_LOGGER_H_
|
#ifndef BIERGARTEN_PIPELINE_INCLUDES_SERVICES_LOGGING_LOG_PRODUCER_H_
|
||||||
#define BIERGARTEN_PIPELINE_INCLUDES_SERVICES_LOGGING_CHANNEL_LOGGER_H_
|
#define BIERGARTEN_PIPELINE_INCLUDES_SERVICES_LOGGING_LOG_PRODUCER_H_
|
||||||
|
|
||||||
#include <string_view>
|
#include <string_view>
|
||||||
|
|
||||||
@@ -50,4 +50,4 @@ class LogProducer final : public ILogger {
|
|||||||
BoundedChannel<LogEntry>& channel_;
|
BoundedChannel<LogEntry>& channel_;
|
||||||
};
|
};
|
||||||
|
|
||||||
#endif // BIERGARTEN_PIPELINE_INCLUDES_SERVICES_LOGGING_CHANNEL_LOGGER_H_
|
#endif // BIERGARTEN_PIPELINE_INCLUDES_SERVICES_LOGGING_LOG_PRODUCER_H_
|
||||||
|
|||||||
@@ -39,7 +39,8 @@ class ILogger {
|
|||||||
*/
|
*/
|
||||||
void Log(LogDTO payload,
|
void Log(LogDTO payload,
|
||||||
std::source_location origin = std::source_location::current(),
|
std::source_location origin = std::source_location::current(),
|
||||||
std::chrono::system_clock::time_point timestamp = std::chrono::system_clock::now(),
|
std::chrono::system_clock::time_point timestamp =
|
||||||
|
std::chrono::system_clock::now(),
|
||||||
std::thread::id thread_id = std::this_thread::get_id()) {
|
std::thread::id thread_id = std::this_thread::get_id()) {
|
||||||
LogEntry entry;
|
LogEntry entry;
|
||||||
entry.timestamp = timestamp;
|
entry.timestamp = timestamp;
|
||||||
|
|||||||
@@ -0,0 +1,32 @@
|
|||||||
|
#ifndef BIERGARTEN_PIPELINE_INCLUDES_SERVICES_POSTAL_CODE_MOCK_POSTAL_CODE_SERVICE_H_
|
||||||
|
#define BIERGARTEN_PIPELINE_INCLUDES_SERVICES_POSTAL_CODE_MOCK_POSTAL_CODE_SERVICE_H_
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @file services/postal_code/mock_postal_code_service.h
|
||||||
|
* @brief Placeholder IPostalCodeService used until xeger-based generation is
|
||||||
|
* implemented.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include <stdexcept>
|
||||||
|
#include <string>
|
||||||
|
|
||||||
|
#include "data_model/models.h"
|
||||||
|
#include "services/postal_code/postal_code_service.h"
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Postal code service that returns the location's first known
|
||||||
|
* example postal code rather than generating one from its regex.
|
||||||
|
*/
|
||||||
|
class MockPostalCodeService final : public IPostalCodeService {
|
||||||
|
public:
|
||||||
|
std::string GeneratePostalCode(const City& location) override {
|
||||||
|
if (location.postal_code_examples.empty()) {
|
||||||
|
throw std::runtime_error(
|
||||||
|
"MockPostalCodeService: location has no postal_code_examples: " +
|
||||||
|
location.city);
|
||||||
|
}
|
||||||
|
return location.postal_code_examples.front();
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
#endif // BIERGARTEN_PIPELINE_INCLUDES_SERVICES_POSTAL_CODE_MOCK_POSTAL_CODE_SERVICE_H_
|
||||||
@@ -0,0 +1,30 @@
|
|||||||
|
#ifndef BIERGARTEN_PIPELINE_INCLUDES_SERVICES_POSTAL_CODE_POSTAL_CODE_SERVICE_H_
|
||||||
|
#define BIERGARTEN_PIPELINE_INCLUDES_SERVICES_POSTAL_CODE_POSTAL_CODE_SERVICE_H_
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @file services/postal_code/postal_code_service.h
|
||||||
|
* @brief Abstraction for generating a postal code for a location.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include <string>
|
||||||
|
|
||||||
|
#include "data_model/models.h"
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Interface for services that generate a postal code string matching
|
||||||
|
* a location's postal code format (`City::postal_regex`).
|
||||||
|
*/
|
||||||
|
class IPostalCodeService {
|
||||||
|
public:
|
||||||
|
virtual ~IPostalCodeService() = default;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Generates a postal code for the given location.
|
||||||
|
*
|
||||||
|
* @param location Location whose postal code format to generate for.
|
||||||
|
* @return A postal code string matching `location.postal_regex`.
|
||||||
|
*/
|
||||||
|
virtual std::string GeneratePostalCode(const City& location) = 0;
|
||||||
|
};
|
||||||
|
|
||||||
|
#endif // BIERGARTEN_PIPELINE_INCLUDES_SERVICES_POSTAL_CODE_POSTAL_CODE_SERVICE_H_
|
||||||
@@ -1,19 +1,18 @@
|
|||||||
/**
|
/**
|
||||||
* @file web_client/http_web_client.h
|
* @file web_client/http_web_client.h
|
||||||
* @brief cpp-httplib implementation of the WebClient interface.
|
* @brief cpp-httplib implementation of the WebClient interface.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#ifndef BIERGARTEN_PIPELINE_INCLUDES_WEB_CLIENT_HTTP_WEB_CLIENT_H_
|
#ifndef BIERGARTEN_PIPELINE_INCLUDES_WEB_CLIENT_HTTP_WEB_CLIENT_H_
|
||||||
#define BIERGARTEN_PIPELINE_INCLUDES_WEB_CLIENT_HTTP_WEB_CLIENT_H_
|
#define BIERGARTEN_PIPELINE_INCLUDES_WEB_CLIENT_HTTP_WEB_CLIENT_H_
|
||||||
|
|
||||||
|
|
||||||
#include "web_client/web_client.h"
|
|
||||||
#include "services/logging/logger.h"
|
|
||||||
|
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include <string>
|
#include <string>
|
||||||
#include <utility>
|
#include <utility>
|
||||||
|
|
||||||
|
#include "services/logging/logger.h"
|
||||||
|
#include "web_client/web_client.h"
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @brief WebClient implementation backed by cpp-httplib.
|
* @brief WebClient implementation backed by cpp-httplib.
|
||||||
*
|
*
|
||||||
@@ -26,7 +25,7 @@
|
|||||||
* bound to a single origin at construction time.
|
* bound to a single origin at construction time.
|
||||||
*/
|
*/
|
||||||
class HttpWebClient final : public WebClient {
|
class HttpWebClient final : public WebClient {
|
||||||
public:
|
public:
|
||||||
explicit HttpWebClient(std::shared_ptr<ILogger> logger)
|
explicit HttpWebClient(std::shared_ptr<ILogger> logger)
|
||||||
: logger_(std::move(logger)) {}
|
: logger_(std::move(logger)) {}
|
||||||
~HttpWebClient() override = default;
|
~HttpWebClient() override = default;
|
||||||
@@ -34,7 +33,8 @@ public:
|
|||||||
/**
|
/**
|
||||||
* @brief Executes a blocking HTTP/HTTPS GET request against a full URL.
|
* @brief Executes a blocking HTTP/HTTPS GET request against a full URL.
|
||||||
*
|
*
|
||||||
* @param url Fully-qualified URL, e.g. "https://en.wikipedia.org/api/rest_v1/page/summary/Berlin"
|
* @param url Fully-qualified URL, e.g.
|
||||||
|
* "https://en.wikipedia.org/api/rest_v1/page/summary/Berlin"
|
||||||
* @return Response body on HTTP 2xx; throws std::runtime_error otherwise.
|
* @return Response body on HTTP 2xx; throws std::runtime_error otherwise.
|
||||||
*/
|
*/
|
||||||
std::string Get(const std::string& url) override;
|
std::string Get(const std::string& url) override;
|
||||||
@@ -52,5 +52,4 @@ public:
|
|||||||
std::shared_ptr<ILogger> logger_;
|
std::shared_ptr<ILogger> logger_;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
#endif // BIERGARTEN_PIPELINE_INCLUDES_WEB_CLIENT_HTTP_WEB_CLIENT_H_
|
||||||
#endif
|
|
||||||
|
|||||||
@@ -13,7 +13,6 @@
|
|||||||
*/
|
*/
|
||||||
class WebClient {
|
class WebClient {
|
||||||
public:
|
public:
|
||||||
/// @brief Virtual destructor for polymorphic cleanup.
|
|
||||||
virtual ~WebClient() = default;
|
virtual ~WebClient() = default;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
|||||||
42
tooling/pipeline/personas.json
Normal file
@@ -0,0 +1,42 @@
|
|||||||
|
[
|
||||||
|
{
|
||||||
|
"name": "Hophead Explorer",
|
||||||
|
"description": "Chases the newest hop varieties and double-digit IBUs, tracking hazy releases and limited tap takeovers across town.",
|
||||||
|
"style_affinities": ["American IPA", "Double IPA", "New England IPA", "Black IPA"]
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"name": "Malt & Tradition Loyalist",
|
||||||
|
"description": "Prefers clean, balanced lagers brewed the way their region has brewed them for generations, and is quick to defend a well-made Helles.",
|
||||||
|
"style_affinities": ["Munich Helles", "Vienna Lager", "Märzen", "Festbier"]
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"name": "Dark & Roasty Devotee",
|
||||||
|
"description": "Orders the darkest beer on the board year-round, drawn to roasted malt, coffee, and chocolate notes over carbonated lightness.",
|
||||||
|
"style_affinities": ["Imperial Stout", "Robust Porter", "Baltic Porter", "Oatmeal Stout"]
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"name": "Sour & Wild Forager",
|
||||||
|
"description": "Seeks out barrel-aged, spontaneously fermented, and funky beers, and will travel for a fresh bottling of something wild.",
|
||||||
|
"style_affinities": ["Lambic", "Gueuze", "Flanders Red Ale", "Wild Ale"]
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"name": "Session Sipper",
|
||||||
|
"description": "Values an easy-drinking pint that holds up over a long afternoon at the taproom more than raw intensity or novelty.",
|
||||||
|
"style_affinities": ["Session IPA", "Ordinary Bitter", "Cream Ale", "Blonde Ale"]
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"name": "Belgian Abbey Pilgrim",
|
||||||
|
"description": "Collects abbey and trappist releases, drawn to the dried-fruit esters and warming strength of classic Belgian ales.",
|
||||||
|
"style_affinities": ["Belgian Tripel", "Belgian Dubbel", "Belgian Quadrupel", "Trappist Single"]
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"name": "Wheat & Citrus Casual",
|
||||||
|
"description": "Reaches for something light, cloudy, and refreshing, usually with a citrus or coriander note, especially on a patio in summer.",
|
||||||
|
"style_affinities": ["Hefeweizen", "Witbier", "American Wheat Beer", "Berliner Weisse"]
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"name": "Big & Boozy Connoisseur",
|
||||||
|
"description": "Sips slowly from a snifter, seeking out high-ABV releases meant for aging and savoring rather than quick refreshment.",
|
||||||
|
"style_affinities": ["English Barleywine", "Wee Heavy", "Doppelbock", "Old Ale"]
|
||||||
|
}
|
||||||
|
]
|
||||||
122
tooling/pipeline/prompts/USER_GENERATION.md
Normal file
@@ -0,0 +1,122 @@
|
|||||||
|
# FULL SYSTEM PROMPT
|
||||||
|
|
||||||
|
You are a craft beer community profile writer. You write short, first-person
|
||||||
|
biographies for fictional users of a brewery-discovery app, grounded in the
|
||||||
|
exact name, locale, and persona provided. You do not invent the user's name --
|
||||||
|
it is given to you and must not be altered, translated, or abbreviated beyond
|
||||||
|
forming a username.
|
||||||
|
|
||||||
|
You will receive the inputs like this:
|
||||||
|
|
||||||
|
## NAME:
|
||||||
|
|
||||||
|
[First name] [Last name]
|
||||||
|
|
||||||
|
## GENDER:
|
||||||
|
|
||||||
|
[Gender associated with the given first name]
|
||||||
|
|
||||||
|
## CITY:
|
||||||
|
|
||||||
|
[City Name]
|
||||||
|
|
||||||
|
## COUNTRY:
|
||||||
|
|
||||||
|
[Country Name]
|
||||||
|
|
||||||
|
## PERSONA:
|
||||||
|
|
||||||
|
[Persona archetype name]
|
||||||
|
|
||||||
|
## PERSONA DESCRIPTION:
|
||||||
|
|
||||||
|
[What this persona cares about and how they talk about beer]
|
||||||
|
|
||||||
|
## STYLE AFFINITIES:
|
||||||
|
|
||||||
|
[Comma-separated beer styles this persona favors]
|
||||||
|
|
||||||
|
## CRITICAL OUTPUT FORMAT (READ CAREFULLY):
|
||||||
|
|
||||||
|
ABSOLUTELY NO MARKDOWN FORMATTING. Do NOT wrap your response in json or ```
|
||||||
|
blocks.
|
||||||
|
|
||||||
|
Do not add markdown, code fences, or postscript around the final JSON object.
|
||||||
|
Do not say "Here is the JSON" or "Enjoy!".
|
||||||
|
|
||||||
|
The JSON must contain exactly three keys ("username", "bio",
|
||||||
|
"activity_weight") in that order. Do not rename or add any other keys.
|
||||||
|
|
||||||
|
ESCAPE ALL QUOTES inside the bio field using \", or use single quotes (' ')
|
||||||
|
instead.
|
||||||
|
|
||||||
|
DO NOT use actual line breaks (\n) inside any string. Keep the bio as one
|
||||||
|
continuous string.
|
||||||
|
|
||||||
|
The bio must be between 25 and 60 words, written in first person, and must
|
||||||
|
read as something this specific person would write about themselves -- not a
|
||||||
|
generic profile blurb.
|
||||||
|
|
||||||
|
`activity_weight` is a number between 0 and 1 (inclusive) representing how
|
||||||
|
often this persona checks in at breweries relative to other users. Persona
|
||||||
|
archetypes implying frequent, habitual brewery visits should skew higher;
|
||||||
|
casual or occasional-visitor personas should skew lower. Do not default to
|
||||||
|
0.5 -- vary it meaningfully based on the persona description.
|
||||||
|
|
||||||
|
Expected JSON format:
|
||||||
|
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
"username": "a handle derived from the given name",
|
||||||
|
"bio": "The first-person bio goes here.",
|
||||||
|
"activity_weight": 0.62
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
## CONTENT RULES AND CONSTRAINTS:
|
||||||
|
|
||||||
|
### THE USERNAME:
|
||||||
|
|
||||||
|
Derive the username from the given first and/or last name (e.g. lowercase,
|
||||||
|
abbreviated, or combined with a short word or number tied to the persona's
|
||||||
|
style affinities). Do not invent an unrelated handle. Do not include spaces.
|
||||||
|
|
||||||
|
### THE BIO:
|
||||||
|
|
||||||
|
Ground the bio in the supplied persona description and style affinities --
|
||||||
|
mention at least one specific beer style from the style affinities list, in a
|
||||||
|
way that sounds like a personal preference rather than a list. Reference the
|
||||||
|
city or country naturally if it fits, but do not force it.
|
||||||
|
|
||||||
|
### VOICE & PERSPECTIVE:
|
||||||
|
|
||||||
|
Write in the first person ("I"/"my"). The tone should match the persona:
|
||||||
|
an enthusiastic explorer sounds different from a relaxed, easy-drinking
|
||||||
|
regular. Do not write in second or third person.
|
||||||
|
|
||||||
|
### THE BLOCKLIST (FORBIDDEN CONCEPTS):
|
||||||
|
|
||||||
|
You absolutely cannot use the following words and phrases. Make sure your
|
||||||
|
final output doesn't have any of these:
|
||||||
|
|
||||||
|
- "hidden gem"
|
||||||
|
- "passion"
|
||||||
|
- "authentic"
|
||||||
|
- "craft beer enthusiast"
|
||||||
|
- "beer connoisseur"
|
||||||
|
- "journey"
|
||||||
|
|
||||||
|
#### FORBIDDEN WRITING PATTERNS
|
||||||
|
|
||||||
|
The following patterns are common AI writing pitfalls and must not appear in
|
||||||
|
the bio:
|
||||||
|
|
||||||
|
- Negative parallelism constructions: "It's not X, it's Y"
|
||||||
|
- Inflated significance phrases: "stands as a testament," "plays a vital
|
||||||
|
role," "deeply rooted"
|
||||||
|
- Superficial trailing analyses: sentences ending in -ing words that add
|
||||||
|
opinion without content
|
||||||
|
- Promotional travel-copy tone: "breathtaking," "must-visit," "vibrant"
|
||||||
|
- Overused conjunctive transitions used as sentence openers: "Moreover,"
|
||||||
|
"Furthermore," "In addition"
|
||||||
|
- Rule of three: do not consistently organise ideas or examples in triplets
|
||||||
8
tooling/pipeline/run.sh
Normal file
@@ -0,0 +1,8 @@
|
|||||||
|
./biergarten-pipeline \
|
||||||
|
--model ../models/google_gemma-4-E4B-it-Q6_K.gguf \
|
||||||
|
--temperature 1.0 \
|
||||||
|
--top-p 0.95 \
|
||||||
|
--top-k 64 \
|
||||||
|
--n-ctx 8192 \
|
||||||
|
--location-count 15 \
|
||||||
|
--prompt-dir ../prompts
|
||||||
@@ -52,7 +52,7 @@ WORKDIR /app/build
|
|||||||
COPY --from=builder /app/build/biergarten-pipeline ./
|
COPY --from=builder /app/build/biergarten-pipeline ./
|
||||||
|
|
||||||
# Copy required config files
|
# Copy required config files
|
||||||
COPY locations.json /app/build/
|
COPY cities.json /app/build/
|
||||||
COPY beer-styles.json /app/build/
|
COPY beer-styles.json /app/build/
|
||||||
|
|
||||||
# Copy prompt templates
|
# Copy prompt templates
|
||||||
|
|||||||
@@ -15,8 +15,6 @@ std::optional<ApplicationOptions> ParseArguments(
|
|||||||
|
|
||||||
opt("help,h", "Produce help message");
|
opt("help,h", "Produce help message");
|
||||||
|
|
||||||
// Defaults sourced from SamplingOptions{} so the CLI and LlamaGenerator
|
|
||||||
// share a single source of truth — changing the struct updates both.
|
|
||||||
auto add_sampling_options = [&]() -> void {
|
auto add_sampling_options = [&]() -> void {
|
||||||
const SamplingOptions sampling_defaults{};
|
const SamplingOptions sampling_defaults{};
|
||||||
opt("temperature",
|
opt("temperature",
|
||||||
@@ -53,7 +51,8 @@ std::optional<ApplicationOptions> ParseArguments(
|
|||||||
prog_opts::value<std::string>()->default_value("pipeline.log"),
|
prog_opts::value<std::string>()->default_value("pipeline.log"),
|
||||||
"Path for application logs");
|
"Path for application logs");
|
||||||
opt("prompt-dir", prog_opts::value<std::string>()->default_value(""),
|
opt("prompt-dir", prog_opts::value<std::string>()->default_value(""),
|
||||||
"Directory containing named prompt files (e.g. BREWERY_GENERATION.md)."
|
"Directory containing named prompt files (e.g. "
|
||||||
|
"BREWERY_GENERATION.md)."
|
||||||
" Required when not using --mocked.");
|
" Required when not using --mocked.");
|
||||||
opt("location-count", prog_opts::value<uint32_t>()->default_value(10));
|
opt("location-count", prog_opts::value<uint32_t>()->default_value(10));
|
||||||
};
|
};
|
||||||
@@ -108,10 +107,12 @@ std::optional<ApplicationOptions> ParseArguments(
|
|||||||
const std::string model_path = var_map["model"].as<std::string>();
|
const std::string model_path = var_map["model"].as<std::string>();
|
||||||
const int n_gpu_layers = var_map["n-gpu-layers"].as<int>();
|
const int n_gpu_layers = var_map["n-gpu-layers"].as<int>();
|
||||||
|
|
||||||
// Enforce mutual exclusivity before any further configuration is applied.
|
// Enforce mutual exclusivity before any further configuration is
|
||||||
|
// applied.
|
||||||
if (use_mocked && !model_path.empty()) {
|
if (use_mocked && !model_path.empty()) {
|
||||||
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(LogDTO{.level = LogLevel::Error,
|
logger->Log(LogDTO{.level = LogLevel::Error,
|
||||||
.phase = PipelinePhase::Startup,
|
.phase = PipelinePhase::Startup,
|
||||||
@@ -124,7 +125,8 @@ std::optional<ApplicationOptions> ParseArguments(
|
|||||||
|
|
||||||
if (!use_mocked && model_path.empty()) {
|
if (!use_mocked && model_path.empty()) {
|
||||||
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(LogDTO{.level = LogLevel::Error,
|
logger->Log(LogDTO{.level = LogLevel::Error,
|
||||||
.phase = PipelinePhase::Startup,
|
.phase = PipelinePhase::Startup,
|
||||||
@@ -139,7 +141,8 @@ std::optional<ApplicationOptions> ParseArguments(
|
|||||||
// generator has no use for it and should not require it to be present.
|
// generator has no use for it and should not require it to be present.
|
||||||
if (!use_mocked && options.pipeline.prompt_dir.empty()) {
|
if (!use_mocked && options.pipeline.prompt_dir.empty()) {
|
||||||
const std::string msg =
|
const std::string msg =
|
||||||
"Invalid arguments: --prompt-dir is required when not using --mocked";
|
"Invalid arguments: --prompt-dir is required when not using "
|
||||||
|
"--mocked";
|
||||||
if (logger) {
|
if (logger) {
|
||||||
logger->Log({.level = LogLevel::Error,
|
logger->Log({.level = LogLevel::Error,
|
||||||
.phase = PipelinePhase::Startup,
|
.phase = PipelinePhase::Startup,
|
||||||
@@ -152,7 +155,6 @@ std::optional<ApplicationOptions> ParseArguments(
|
|||||||
|
|
||||||
options.generator.use_mocked = use_mocked;
|
options.generator.use_mocked = use_mocked;
|
||||||
options.generator.model_path = model_path;
|
options.generator.model_path = model_path;
|
||||||
// options.generator.n_gpu_layers = n_gpu_layers;
|
|
||||||
|
|
||||||
// Only populate sampling config when the user explicitly overrides at
|
// Only populate sampling config when the user explicitly overrides at
|
||||||
// least one value. Leaving it as std::nullopt lets LlamaGenerator fall
|
// least one value. Leaving it as std::nullopt lets LlamaGenerator fall
|
||||||
|
|||||||
@@ -1,20 +1,26 @@
|
|||||||
/**
|
/**
|
||||||
* @file biergarten_pipeline_orchestrator/biergarten_pipeline_orchestrator.cc
|
* @file biergarten_pipeline_orchestrator/biergarten_pipeline_orchestrator.cc
|
||||||
* @brief BiergartenDataGenerator constructor implementation.
|
* @brief BiergartenPipelineOrchestrator constructor implementation.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#include "biergarten_pipeline_orchestrator.h"
|
#include "biergarten_pipeline_orchestrator.h"
|
||||||
|
|
||||||
#include <utility>
|
#include <utility>
|
||||||
|
|
||||||
|
#include "services/postal_code/postal_code_service.h"
|
||||||
|
|
||||||
BiergartenPipelineOrchestrator::BiergartenPipelineOrchestrator(
|
BiergartenPipelineOrchestrator::BiergartenPipelineOrchestrator(
|
||||||
std::shared_ptr<ILogger> logger,
|
std::shared_ptr<ILogger> logger,
|
||||||
std::unique_ptr<IEnrichmentService> context_service,
|
std::unique_ptr<IEnrichmentService> context_service,
|
||||||
std::unique_ptr<DataGenerator> generator,
|
std::unique_ptr<DataGenerator> generator,
|
||||||
std::unique_ptr<IExportService> exporter,
|
std::unique_ptr<IExportService> exporter,
|
||||||
const ApplicationOptions &app_options)
|
std::unique_ptr<ICuratedDataService> curated_data_service,
|
||||||
|
std::unique_ptr<IPostalCodeService> postal_code_service,
|
||||||
|
const ApplicationOptions& application_options)
|
||||||
: logger_(std::move(logger)),
|
: logger_(std::move(logger)),
|
||||||
context_service_(std::move(context_service)),
|
context_service_(std::move(context_service)),
|
||||||
generator_(std::move(generator)),
|
generator_(std::move(generator)),
|
||||||
exporter_(std::move(exporter)),
|
exporter_(std::move(exporter)),
|
||||||
application_options_(app_options) {}
|
curated_data_service_(std::move(curated_data_service)),
|
||||||
|
postal_code_service_(std::move(postal_code_service)),
|
||||||
|
application_options_(application_options) {}
|
||||||
@@ -1,10 +1,11 @@
|
|||||||
/**
|
/**
|
||||||
* @file biergarten_pipeline_orchestrator/generate_breweries.cc
|
* @file biergarten_pipeline_orchestrator/generate_breweries.cc
|
||||||
* @brief BiergartenDataGenerator::GenerateBreweries() implementation.
|
* @brief BiergartenPipelineOrchestrator::GenerateBreweries() implementation.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#include <chrono>
|
#include <chrono>
|
||||||
#include <format>
|
#include <format>
|
||||||
|
#include <optional>
|
||||||
|
|
||||||
#include "biergarten_pipeline_orchestrator.h"
|
#include "biergarten_pipeline_orchestrator.h"
|
||||||
#include "services/logging/logger.h"
|
#include "services/logging/logger.h"
|
||||||
@@ -19,39 +20,62 @@ void BiergartenPipelineOrchestrator::GenerateBreweries(
|
|||||||
size_t skipped_count = 0;
|
size_t skipped_count = 0;
|
||||||
size_t export_failed_count = 0;
|
size_t export_failed_count = 0;
|
||||||
|
|
||||||
for (const auto& [location, region_context] : cities) {
|
const auto generate_record =
|
||||||
|
[this, &skipped_count](
|
||||||
|
const City& location,
|
||||||
|
const std::string& region_context) -> std::optional<BreweryRecord> {
|
||||||
try {
|
try {
|
||||||
const BreweryResult brewery =
|
const BreweryResult brewery =
|
||||||
generator_->GenerateBrewery(location, region_context);
|
generator_->GenerateBrewery(location, region_context);
|
||||||
|
const std::string postal_code =
|
||||||
const GeneratedBrewery gen{.location = location, .brewery = brewery};
|
postal_code_service_->GeneratePostalCode(location);
|
||||||
|
return BreweryRecord{
|
||||||
generated_breweries_.push_back(gen);
|
.address =
|
||||||
|
BreweryAddress{.city = location, .postal_code = postal_code},
|
||||||
try {
|
.brewery = brewery};
|
||||||
exporter_->ProcessRecord(gen);
|
} catch (const std::exception& e) {
|
||||||
} catch (const std::exception& export_exception) {
|
++skipped_count;
|
||||||
++export_failed_count;
|
|
||||||
|
|
||||||
logger_->Log(
|
logger_->Log(
|
||||||
{.level = LogLevel::Warn,
|
{.level = LogLevel::Warn,
|
||||||
.phase = PipelinePhase::BreweryAndBeerGeneration,
|
.phase = PipelinePhase::BreweryAndBeerGeneration,
|
||||||
.message =
|
.message = std::format("[Pipeline] Skipping city '{}' ({}): brewery "
|
||||||
std::format("[Pipeline] Generated brewery for '{}' ({}) but SQLite export failed: {}",
|
"generation failed: {}",
|
||||||
location.city, location.country, export_exception.what())});
|
location.city, location.country, e.what())});
|
||||||
|
return std::nullopt;
|
||||||
}
|
}
|
||||||
} catch (const std::exception& e) {
|
};
|
||||||
++skipped_count;
|
|
||||||
|
|
||||||
|
const auto export_record = [this, &export_failed_count](
|
||||||
|
const BreweryRecord& record) {
|
||||||
|
try {
|
||||||
|
exporter_->ProcessRecord(record);
|
||||||
|
} catch (const std::exception& export_exception) {
|
||||||
|
++export_failed_count;
|
||||||
logger_->Log({.level = LogLevel::Warn,
|
logger_->Log({.level = LogLevel::Warn,
|
||||||
.phase = PipelinePhase::BreweryAndBeerGeneration,
|
.phase = PipelinePhase::BreweryAndBeerGeneration,
|
||||||
.message = std::format("[Pipeline] Skipping city '{}' ({}): brewery generation failed: {}",
|
.message = std::format(
|
||||||
location.city, location.country, e.what())});
|
"[Pipeline] Generated brewery for '{}' ({}) "
|
||||||
|
"but SQLite export failed: {}",
|
||||||
|
record.address.city.city, record.address.city.country,
|
||||||
|
export_exception.what())});
|
||||||
}
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
for (const auto& [location, region_context] : cities) {
|
||||||
|
const std::optional<BreweryRecord> record =
|
||||||
|
generate_record(location, region_context);
|
||||||
|
if (!record.has_value()) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
generated_breweries_.push_back(*record);
|
||||||
|
export_record(*record);
|
||||||
}
|
}
|
||||||
|
|
||||||
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::format(
|
.message = std::format(
|
||||||
"[Pipeline] Skipped {} city/cities due to generation errors",
|
"[Pipeline] Skipped {} city/cities due to generation errors",
|
||||||
@@ -59,10 +83,11 @@ void BiergartenPipelineOrchestrator::GenerateBreweries(
|
|||||||
}
|
}
|
||||||
|
|
||||||
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::format(
|
.message = std::format("[Pipeline] Failed to export {} generated "
|
||||||
"[Pipeline] Failed to export {} generated brewery/breweries to SQLite",
|
"brewery/breweries to SQLite",
|
||||||
export_failed_count)});
|
export_failed_count)});
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -0,0 +1,219 @@
|
|||||||
|
/**
|
||||||
|
* @file biergarten_pipeline_orchestrator/generate_users.cc
|
||||||
|
* @brief BiergartenPipelineOrchestrator::GenerateUsers() implementation.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include <cctype>
|
||||||
|
#include <chrono>
|
||||||
|
#include <format>
|
||||||
|
#include <iterator>
|
||||||
|
#include <optional>
|
||||||
|
#include <random>
|
||||||
|
#include <stdexcept>
|
||||||
|
#include <string>
|
||||||
|
#include <string_view>
|
||||||
|
#include <unordered_set>
|
||||||
|
|
||||||
|
#include "biergarten_pipeline_orchestrator.h"
|
||||||
|
#include "services/curated_data/curated_json_data_service.h"
|
||||||
|
#include "services/logging/logger.h"
|
||||||
|
|
||||||
|
namespace {
|
||||||
|
std::string Sanitize(std::string_view value) {
|
||||||
|
std::string out;
|
||||||
|
out.reserve(value.size());
|
||||||
|
for (const char character : value) {
|
||||||
|
if (std::isalnum(static_cast<unsigned char>(character)) != 0) {
|
||||||
|
out.push_back(static_cast<char>(
|
||||||
|
std::tolower(static_cast<unsigned char>(character))));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return out;
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string BuildEmail(const Name& name,
|
||||||
|
std::unordered_set<std::string>& used_local_parts) {
|
||||||
|
const std::string base =
|
||||||
|
std::format("{}.{}", Sanitize(name.first_name), Sanitize(name.last_name));
|
||||||
|
|
||||||
|
std::string local_part = base;
|
||||||
|
uint32_t suffix = 1;
|
||||||
|
|
||||||
|
while (used_local_parts.contains(local_part)) {
|
||||||
|
local_part = std::format("{}{}", base, suffix);
|
||||||
|
++suffix;
|
||||||
|
}
|
||||||
|
|
||||||
|
used_local_parts.insert(local_part);
|
||||||
|
|
||||||
|
return std::format("{}@thebiergarten.app", local_part);
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string GenerateDateOfBirth(std::mt19937& rng) {
|
||||||
|
using namespace std::chrono;
|
||||||
|
|
||||||
|
constexpr int kMinAge = 19;
|
||||||
|
constexpr int kMaxAge = 48;
|
||||||
|
constexpr int kMaxDayOffset = 364;
|
||||||
|
|
||||||
|
std::uniform_int_distribution<int> age_dist(kMinAge, kMaxAge);
|
||||||
|
std::uniform_int_distribution<int> day_offset_dist(0, kMaxDayOffset);
|
||||||
|
|
||||||
|
const year_month_day today{floor<days>(system_clock::now())};
|
||||||
|
const year_month_day birth_year_anchor{today.year() - years{age_dist(rng)},
|
||||||
|
today.month(), today.day()};
|
||||||
|
const sys_days birth_date =
|
||||||
|
sys_days{birth_year_anchor} - days{day_offset_dist(rng)};
|
||||||
|
const year_month_day birth_ymd{birth_date};
|
||||||
|
|
||||||
|
return std::format("{:04}-{:02}-{:02}", static_cast<int>(birth_ymd.year()),
|
||||||
|
static_cast<unsigned>(birth_ymd.month()),
|
||||||
|
static_cast<unsigned>(birth_ymd.day()));
|
||||||
|
}
|
||||||
|
|
||||||
|
std::optional<Name> SampleName(
|
||||||
|
const ForenamesByCountryMap& forenames_by_country,
|
||||||
|
const SurnamesByCountryMap& surnames_by_country,
|
||||||
|
const std::string& iso3166_1, std::mt19937& rng) {
|
||||||
|
const auto forenames_it = forenames_by_country.find(iso3166_1);
|
||||||
|
const auto surnames_it = surnames_by_country.find(iso3166_1);
|
||||||
|
|
||||||
|
if (forenames_it == forenames_by_country.end() ||
|
||||||
|
surnames_it == surnames_by_country.end() ||
|
||||||
|
forenames_it->second.empty() || surnames_it->second.empty()) {
|
||||||
|
return std::nullopt;
|
||||||
|
}
|
||||||
|
|
||||||
|
const ForenameList& forenames = forenames_it->second;
|
||||||
|
const SurnameList& surnames = surnames_it->second;
|
||||||
|
|
||||||
|
std::uniform_int_distribution<size_t> forename_dist(0, forenames.size() - 1);
|
||||||
|
std::uniform_int_distribution<size_t> surname_dist(0, surnames.size() - 1);
|
||||||
|
|
||||||
|
auto forename_it = forenames.begin();
|
||||||
|
std::advance(forename_it, forename_dist(rng));
|
||||||
|
|
||||||
|
auto surname_it = surnames.begin();
|
||||||
|
std::advance(surname_it, surname_dist(rng));
|
||||||
|
|
||||||
|
return Name{.first_name = forename_it->name,
|
||||||
|
.last_name = *surname_it,
|
||||||
|
.gender = forename_it->gender};
|
||||||
|
}
|
||||||
|
} // namespace
|
||||||
|
|
||||||
|
void BiergartenPipelineOrchestrator::GenerateUsers(
|
||||||
|
std::span<const EnrichedCity> cities) {
|
||||||
|
logger_->Log({.level = LogLevel::Info,
|
||||||
|
.phase = PipelinePhase::UserGeneration,
|
||||||
|
.message = "=== SAMPLE USER GENERATION ==="});
|
||||||
|
|
||||||
|
const PersonasList& personas = curated_data_service_->LoadPersonas();
|
||||||
|
|
||||||
|
if (personas.empty()) {
|
||||||
|
throw std::runtime_error(
|
||||||
|
"No personas available in personas.json for user generation");
|
||||||
|
}
|
||||||
|
|
||||||
|
const ForenamesByCountryMap& forenames_by_country =
|
||||||
|
curated_data_service_->LoadForenamesByCountry();
|
||||||
|
const SurnamesByCountryMap& surnames_by_country =
|
||||||
|
curated_data_service_->LoadSurnamesByCountry();
|
||||||
|
|
||||||
|
std::mt19937 rng(std::random_device{}());
|
||||||
|
std::uniform_int_distribution<size_t> persona_dist(0, personas.size() - 1);
|
||||||
|
|
||||||
|
generated_users_.clear();
|
||||||
|
size_t skipped_count = 0;
|
||||||
|
size_t export_failed_count = 0;
|
||||||
|
std::unordered_set<std::string> used_email_local_parts;
|
||||||
|
|
||||||
|
const auto generate_record =
|
||||||
|
[this, &rng, &skipped_count, &used_email_local_parts](
|
||||||
|
const EnrichedCity& city, const UserPersona& persona,
|
||||||
|
const Name& sampled_name) -> std::optional<UserRecord> {
|
||||||
|
try {
|
||||||
|
const UserResult user =
|
||||||
|
generator_->GenerateUser(city, persona, sampled_name);
|
||||||
|
|
||||||
|
return UserRecord{
|
||||||
|
.address = UserAddress{.city = city.location},
|
||||||
|
.user = user,
|
||||||
|
.email = BuildEmail(sampled_name, used_email_local_parts),
|
||||||
|
.date_of_birth = GenerateDateOfBirth(rng),
|
||||||
|
};
|
||||||
|
} catch (const std::exception& e) {
|
||||||
|
++skipped_count;
|
||||||
|
logger_->Log({.level = LogLevel::Warn,
|
||||||
|
.phase = PipelinePhase::UserGeneration,
|
||||||
|
.message = std::format(
|
||||||
|
"[Pipeline] Skipping city '{}' ({}): "
|
||||||
|
"user generation failed: {}",
|
||||||
|
city.location.city, city.location.country, e.what())});
|
||||||
|
return std::nullopt;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
const auto export_record = [this,
|
||||||
|
&export_failed_count](const UserRecord& record) {
|
||||||
|
try {
|
||||||
|
exporter_->ProcessRecord(record);
|
||||||
|
} catch (const std::exception& export_exception) {
|
||||||
|
++export_failed_count;
|
||||||
|
|
||||||
|
logger_->Log({.level = LogLevel::Warn,
|
||||||
|
.phase = PipelinePhase::UserGeneration,
|
||||||
|
.message = std::format(
|
||||||
|
"[Pipeline] Generated user for '{}' ({}) but "
|
||||||
|
"SQLite export failed: {}",
|
||||||
|
record.address.city.city, record.address.city.country,
|
||||||
|
export_exception.what())});
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
for (const auto& city : cities) {
|
||||||
|
const std::optional<Name> sampled_name =
|
||||||
|
SampleName(forenames_by_country, surnames_by_country,
|
||||||
|
city.location.iso3166_1, rng);
|
||||||
|
|
||||||
|
if (!sampled_name.has_value()) {
|
||||||
|
++skipped_count;
|
||||||
|
logger_->Log({.level = LogLevel::Warn,
|
||||||
|
.phase = PipelinePhase::UserGeneration,
|
||||||
|
.message = std::format(
|
||||||
|
"[Pipeline] Skipping city '{}' ({}): no names "
|
||||||
|
"available for country '{}'",
|
||||||
|
city.location.city, city.location.country,
|
||||||
|
city.location.iso3166_1)});
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
const UserPersona& persona = personas[persona_dist(rng)];
|
||||||
|
|
||||||
|
const std::optional<UserRecord> record =
|
||||||
|
generate_record(city, persona, *sampled_name);
|
||||||
|
if (!record.has_value()) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
generated_users_.push_back(*record);
|
||||||
|
export_record(*record);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (skipped_count > 0) {
|
||||||
|
logger_->Log(
|
||||||
|
{.level = LogLevel::Warn,
|
||||||
|
.phase = PipelinePhase::UserGeneration,
|
||||||
|
.message = std::format(
|
||||||
|
"[Pipeline] Skipped {} city/cities during user generation",
|
||||||
|
skipped_count)});
|
||||||
|
}
|
||||||
|
|
||||||
|
if (export_failed_count > 0) {
|
||||||
|
logger_->Log({.level = LogLevel::Warn,
|
||||||
|
.phase = PipelinePhase::Teardown,
|
||||||
|
.message = std::format("[Pipeline] Failed to export {} "
|
||||||
|
"generated user/users to SQLite",
|
||||||
|
export_failed_count)});
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -1,37 +1,59 @@
|
|||||||
/**
|
/**
|
||||||
* @file biergarten_pipeline_orchestrator/log_results.cc
|
* @file biergarten_pipeline_orchestrator/log_results.cc
|
||||||
* @brief BiergartenDataGenerator::LogResults() implementation.
|
* @brief BiergartenPipelineOrchestrator::LogResults() implementation.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#include <boost/json/array.hpp>
|
#include <boost/json/array.hpp>
|
||||||
#include <chrono>
|
#include <chrono>
|
||||||
#include <format>
|
#include <format>
|
||||||
|
|
||||||
#include "../../includes/json_handling/pretty_print.h"
|
|
||||||
#include "biergarten_pipeline_orchestrator.h"
|
#include "biergarten_pipeline_orchestrator.h"
|
||||||
|
#include "json_handling/pretty_print.h"
|
||||||
#include "services/logging/logger.h"
|
#include "services/logging/logger.h"
|
||||||
void BiergartenPipelineOrchestrator::LogResults() const {
|
|
||||||
boost::json::array output;
|
|
||||||
|
|
||||||
for (const auto& [location, brewery] : generated_breweries_) {
|
void BiergartenPipelineOrchestrator::LogResults() const {
|
||||||
output.push_back(boost::json::object{
|
boost::json::array brewery_output;
|
||||||
|
|
||||||
|
for (const auto& [address, brewery] : generated_breweries_) {
|
||||||
|
const City& location = address.city;
|
||||||
|
brewery_output.push_back(boost::json::object{
|
||||||
{"name_en", brewery.name_en},
|
{"name_en", brewery.name_en},
|
||||||
{"description_en", brewery.description_en},
|
{"description_en", brewery.description_en},
|
||||||
{"name_local", brewery.name_local},
|
{"name_local", brewery.name_local},
|
||||||
{"description_local", brewery.description_local},
|
{"description_local", brewery.description_local},
|
||||||
|
{"postal_code", address.postal_code},
|
||||||
{"location", boost::json::object{
|
{"location", boost::json::object{
|
||||||
{"city", location.city},
|
{"city", location.city},
|
||||||
{"country", location.country},
|
{"country", location.country},
|
||||||
{"state_province", location.state_province},
|
{"state_province", location.state_province},
|
||||||
{"iso3166_2", location.iso3166_2},
|
{"iso3166_2", location.iso3166_2},
|
||||||
{"latitude", location.latitude},
|
|
||||||
{"longitude", location.longitude},
|
|
||||||
}}});
|
}}});
|
||||||
}
|
}
|
||||||
|
|
||||||
std::ostringstream oss;
|
std::ostringstream brewery_oss;
|
||||||
PrettyPrint(oss, output);
|
PrettyPrint(brewery_oss, brewery_output);
|
||||||
logger_->Log({.level = LogLevel::Info,
|
logger_->Log({.level = LogLevel::Info,
|
||||||
.phase = PipelinePhase::Teardown,
|
.phase = PipelinePhase::Teardown,
|
||||||
.message = oss.str()});
|
.message = brewery_oss.str()});
|
||||||
|
|
||||||
|
boost::json::array user_output;
|
||||||
|
|
||||||
|
for (const auto& generated_user : generated_users_) {
|
||||||
|
user_output.push_back(boost::json::object{
|
||||||
|
{"first_name", generated_user.user.first_name},
|
||||||
|
{"last_name", generated_user.user.last_name},
|
||||||
|
{"gender", generated_user.user.gender},
|
||||||
|
{"username", generated_user.user.username},
|
||||||
|
{"bio", generated_user.user.bio},
|
||||||
|
{"activity_weight", generated_user.user.activity_weight},
|
||||||
|
{"email", generated_user.email},
|
||||||
|
{"date_of_birth", generated_user.date_of_birth},
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
std::ostringstream user_oss;
|
||||||
|
PrettyPrint(user_oss, user_output);
|
||||||
|
logger_->Log({.level = LogLevel::Info,
|
||||||
|
.phase = PipelinePhase::Teardown,
|
||||||
|
.message = user_oss.str()});
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,28 +1,27 @@
|
|||||||
/**
|
/**
|
||||||
* @file biergarten_pipeline_orchestrator/query_cities_with_countries.cc
|
* @file biergarten_pipeline_orchestrator/query_cities_with_countries.cc
|
||||||
* @brief BiergartenDataGenerator::QueryCitiesWithCountries() implementation.
|
* @brief BiergartenPipelineOrchestrator::QueryCitiesWithCountries()
|
||||||
|
* implementation.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#include <algorithm>
|
#include <algorithm>
|
||||||
#include <chrono>
|
#include <chrono>
|
||||||
#include <filesystem>
|
|
||||||
#include <format>
|
#include <format>
|
||||||
#include <iterator>
|
#include <iterator>
|
||||||
#include <random>
|
#include <random>
|
||||||
|
|
||||||
#include "biergarten_pipeline_orchestrator.h"
|
#include "biergarten_pipeline_orchestrator.h"
|
||||||
#include "json_handling/json_loader.h"
|
#include "services/curated_data/curated_json_data_service.h"
|
||||||
#include "services/logging/logger.h"
|
#include "services/logging/logger.h"
|
||||||
|
|
||||||
std::vector<Location>
|
std::vector<City>
|
||||||
BiergartenPipelineOrchestrator::QueryCitiesWithCountries() {
|
BiergartenPipelineOrchestrator::QueryCitiesWithCountries() {
|
||||||
logger_->Log({.level = LogLevel::Info,
|
logger_->Log({.level = LogLevel::Info,
|
||||||
.phase = PipelinePhase::Startup,
|
.phase = PipelinePhase::Startup,
|
||||||
.message = "=== GEOGRAPHIC DATA OVERVIEW ==="});
|
.message = "=== GEOGRAPHIC DATA OVERVIEW ==="});
|
||||||
|
|
||||||
const std::filesystem::path locations_path = "locations.json";
|
const std::vector<City>& all_locations =
|
||||||
|
curated_data_service_->LoadCities();
|
||||||
auto all_locations = JsonLoader::LoadLocations(locations_path, logger_);
|
|
||||||
|
|
||||||
const size_t sample_count = std::min(
|
const size_t sample_count = std::min(
|
||||||
static_cast<size_t>(application_options_.pipeline.location_count),
|
static_cast<size_t>(application_options_.pipeline.location_count),
|
||||||
@@ -32,7 +31,7 @@ BiergartenPipelineOrchestrator::QueryCitiesWithCountries() {
|
|||||||
static_cast<std::iter_difference_t<decltype(all_locations.cbegin())>>(
|
static_cast<std::iter_difference_t<decltype(all_locations.cbegin())>>(
|
||||||
sample_count);
|
sample_count);
|
||||||
|
|
||||||
std::vector<Location> sampled_locations;
|
std::vector<City> sampled_locations;
|
||||||
sampled_locations.reserve(sample_count);
|
sampled_locations.reserve(sample_count);
|
||||||
|
|
||||||
std::random_device random_generator;
|
std::random_device random_generator;
|
||||||
|
|||||||
@@ -1,6 +1,6 @@
|
|||||||
/**
|
/**
|
||||||
* @file biergarten_pipeline_orchestrator/run.cc
|
* @file biergarten_pipeline_orchestrator/run.cc
|
||||||
* @brief BiergartenDataGenerator::Run() implementation.
|
* @brief BiergartenPipelineOrchestrator::Run() implementation.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#include <chrono>
|
#include <chrono>
|
||||||
@@ -14,7 +14,7 @@ bool BiergartenPipelineOrchestrator::Run() {
|
|||||||
try {
|
try {
|
||||||
exporter_->Initialize();
|
exporter_->Initialize();
|
||||||
|
|
||||||
std::vector<Location> cities = QueryCitiesWithCountries();
|
std::vector<City> cities = QueryCitiesWithCountries();
|
||||||
std::vector<EnrichedCity> enriched;
|
std::vector<EnrichedCity> enriched;
|
||||||
enriched.reserve(cities.size());
|
enriched.reserve(cities.size());
|
||||||
|
|
||||||
@@ -22,42 +22,40 @@ bool BiergartenPipelineOrchestrator::Run() {
|
|||||||
for (auto& city : cities) {
|
for (auto& city : cities) {
|
||||||
try {
|
try {
|
||||||
std::string region_context = context_service_->GetLocationContext(city);
|
std::string region_context = context_service_->GetLocationContext(city);
|
||||||
// logger_->Log(LogLevel::Debug, PipelinePhase::UserGeneration,
|
|
||||||
// "[Pipeline] Context for '" + city.city + "' (" +
|
|
||||||
// city.iso3166_2 + ") gathered:\n" + region_context);
|
|
||||||
|
|
||||||
enriched.push_back(
|
enriched.push_back(
|
||||||
EnrichedCity{.location = std::move(city),
|
EnrichedCity{.location = std::move(city),
|
||||||
.region_context = std::move(region_context)});
|
.region_context = std::move(region_context)});
|
||||||
} catch (const std::exception& exception) {
|
} catch (const std::exception& exception) {
|
||||||
++skipped_count;
|
++skipped_count;
|
||||||
logger_->Log(
|
logger_->Log({.level = LogLevel::Warn,
|
||||||
{.level = LogLevel::Warn,
|
.phase = PipelinePhase::Enrichment,
|
||||||
.phase = PipelinePhase::UserGeneration,
|
|
||||||
.message = std::format(
|
.message = std::format(
|
||||||
"[Pipeline] Skipping city '{}' ({}): context lookup failed: {}",
|
"[Pipeline] Skipping city '{}' ({}): context "
|
||||||
|
"lookup failed: {}",
|
||||||
city.city, city.country, exception.what())});
|
city.city, city.country, exception.what())});
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if (skipped_count > 0) {
|
if (skipped_count > 0) {
|
||||||
logger_->Log({.level = LogLevel::Warn,
|
logger_->Log(
|
||||||
.phase = PipelinePhase::UserGeneration,
|
{.level = LogLevel::Warn,
|
||||||
.message = std::format(
|
.phase = PipelinePhase::Enrichment,
|
||||||
"[Pipeline] Skipped {} city/cities due to context lookup errors",
|
.message = std::format("[Pipeline] Skipped {} city/cities due "
|
||||||
|
"to context lookup errors",
|
||||||
skipped_count)});
|
skipped_count)});
|
||||||
}
|
}
|
||||||
|
|
||||||
|
this->GenerateUsers(enriched);
|
||||||
this->GenerateBreweries(enriched);
|
this->GenerateBreweries(enriched);
|
||||||
exporter_->Finalize();
|
exporter_->Finalize();
|
||||||
this->LogResults();
|
this->LogResults();
|
||||||
return true;
|
return true;
|
||||||
} catch (const std::exception& e) {
|
} catch (const std::exception& e) {
|
||||||
logger_->Log(
|
logger_->Log({.level = LogLevel::Error,
|
||||||
{.level = LogLevel::Error,
|
|
||||||
.phase = PipelinePhase::Teardown,
|
.phase = PipelinePhase::Teardown,
|
||||||
.message =
|
.message = std::format(
|
||||||
std::format("Pipeline execution failed with error: {}", e.what())});
|
"Pipeline execution failed with error: {}", e.what())});
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -12,6 +12,7 @@
|
|||||||
#include <string_view>
|
#include <string_view>
|
||||||
#include <vector>
|
#include <vector>
|
||||||
|
|
||||||
|
#include "data_generation/json_grammars.h"
|
||||||
#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"
|
||||||
|
|
||||||
@@ -32,23 +33,10 @@ static std::string FormatLocalLanguageCodes(
|
|||||||
return formatted;
|
return formatted;
|
||||||
}
|
}
|
||||||
|
|
||||||
// GBNF grammar for structured brewery JSON output.
|
|
||||||
// @TODO move to a separate gbnf file if it grows in complexity or is shared
|
|
||||||
// across modules.
|
|
||||||
static constexpr std::string_view kBreweryJsonGrammar = R"json_brewery(
|
|
||||||
root ::= thought-block "{" ws "\"name_en\"" ws ":" ws string ws "," ws "\"description_en\"" ws ":" ws string ws "," ws "\"name_local\"" ws ":" ws string ws "," ws "\"description_local\"" ws ":" ws string ws "}" ws
|
|
||||||
thought-block ::= [^{]*
|
|
||||||
ws ::= [ \t\n\r]*
|
|
||||||
string ::= "\"" char+ "\""
|
|
||||||
char ::= [^"\\\x7F\x00-\x1F] | [\\] escape
|
|
||||||
escape ::= ["\\/bfnrt] | "u" hex hex hex hex
|
|
||||||
hex ::= [0-9a-fA-F]
|
|
||||||
)json_brewery";
|
|
||||||
|
|
||||||
static constexpr int kBreweryInitialMaxTokens = 2800;
|
static constexpr int kBreweryInitialMaxTokens = 2800;
|
||||||
|
|
||||||
BreweryResult LlamaGenerator::GenerateBrewery(
|
BreweryResult LlamaGenerator::GenerateBrewery(
|
||||||
const Location& location, const std::string& region_context) {
|
const City& location, const std::string& region_context) {
|
||||||
/**
|
/**
|
||||||
* Preprocess and truncate region context to manageable size
|
* Preprocess and truncate region context to manageable size
|
||||||
*/
|
*/
|
||||||
@@ -83,15 +71,14 @@ BreweryResult LlamaGenerator::GenerateBrewery(
|
|||||||
|
|
||||||
/**
|
/**
|
||||||
* RETRY LOOP with validation and error correction
|
* RETRY LOOP with validation and error correction
|
||||||
* Attempts to generate valid brewery data up to 3 times, with feedback-based
|
* Attempts to generate valid brewery data up to 3 times, with
|
||||||
* refinement
|
* feedback-based refinement
|
||||||
*/
|
*/
|
||||||
constexpr int max_attempts = 3;
|
constexpr int max_attempts = 3;
|
||||||
std::string raw;
|
std::string raw;
|
||||||
std::string last_error;
|
std::string last_error;
|
||||||
|
|
||||||
// Token budget: too small risks truncating valid JSON mid-string.
|
// Token budget: too small risks truncating valid JSON mid-string.
|
||||||
// Start conservatively but allow adaptive increases on truncation.
|
|
||||||
int max_tokens = kBreweryInitialMaxTokens;
|
int max_tokens = kBreweryInitialMaxTokens;
|
||||||
|
|
||||||
// Limit output length to keep it concise and focused
|
// Limit output length to keep it concise and focused
|
||||||
@@ -120,7 +107,8 @@ BreweryResult LlamaGenerator::GenerateBrewery(
|
|||||||
logger_->Log(
|
logger_->Log(
|
||||||
{.level = LogLevel::Info,
|
{.level = LogLevel::Info,
|
||||||
.phase = PipelinePhase::BreweryAndBeerGeneration,
|
.phase = PipelinePhase::BreweryAndBeerGeneration,
|
||||||
.message = std::format("LlamaGenerator: successfully generated brewery data on attempt {}",
|
.message = std::format("LlamaGenerator: successfully generated "
|
||||||
|
"brewery data on attempt {}",
|
||||||
attempt + 1)});
|
attempt + 1)});
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -134,32 +122,37 @@ BreweryResult LlamaGenerator::GenerateBrewery(
|
|||||||
logger_->Log(
|
logger_->Log(
|
||||||
{.level = LogLevel::Warn,
|
{.level = LogLevel::Warn,
|
||||||
.phase = PipelinePhase::BreweryAndBeerGeneration,
|
.phase = PipelinePhase::BreweryAndBeerGeneration,
|
||||||
.message =
|
.message = std::format(
|
||||||
std::format("LlamaGenerator: malformed brewery JSON (attempt {}): {}",
|
"LlamaGenerator: malformed brewery JSON (attempt {}): {}",
|
||||||
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.
|
||||||
user_prompt = std::format(
|
user_prompt = std::format(
|
||||||
"Your previous response was invalid. Error: {}\nReturn the thought "
|
"Your previous response was invalid. Error: {}\nReturn the thought "
|
||||||
"process before the JSON if needed, then return ONLY valid JSON with "
|
"process before the JSON if needed, then return ONLY valid JSON "
|
||||||
"exactly these keys, in this exact order: {{\"name_en\": \"<English "
|
"with "
|
||||||
|
"exactly these keys, in this exact order: {{\"name_en\": "
|
||||||
|
"\"<English "
|
||||||
"brewery name>\", \"description_en\": \"<English single-paragraph "
|
"brewery name>\", \"description_en\": \"<English single-paragraph "
|
||||||
"description>\", \"name_local\": \"<local-language brewery name>\", "
|
"description>\", \"name_local\": \"<local-language brewery "
|
||||||
|
"name>\", "
|
||||||
"\"description_local\": \"<local-language single-paragraph "
|
"\"description_local\": \"<local-language single-paragraph "
|
||||||
"description>\"}}.\nDo not include markdown, comments, extra keys, or "
|
"description>\"}}.\nDo not include markdown, comments, extra keys, "
|
||||||
"literal placeholder values.\n\nKeep the JSON strings concise enough "
|
"or "
|
||||||
|
"literal placeholder values.\n\nKeep the JSON strings concise "
|
||||||
|
"enough "
|
||||||
"to fit within the token budget.\n\n{}",
|
"to fit within the token budget.\n\n{}",
|
||||||
*validation_error, retry_location);
|
*validation_error, retry_location);
|
||||||
}
|
}
|
||||||
|
|
||||||
// All retry attempts exhausted: log failure and throw exception
|
// All retry attempts exhausted: log failure and throw exception
|
||||||
if (logger_) {
|
if (logger_) {
|
||||||
logger_->Log(
|
logger_->Log({.level = LogLevel::Error,
|
||||||
{.level = LogLevel::Error,
|
|
||||||
.phase = PipelinePhase::BreweryAndBeerGeneration,
|
.phase = PipelinePhase::BreweryAndBeerGeneration,
|
||||||
.message = std::format(
|
.message = std::format(
|
||||||
"LlamaGenerator: malformed brewery response after {} attempts: {}",
|
"LlamaGenerator: malformed brewery "
|
||||||
|
"response after {} attempts: {}",
|
||||||
max_attempts, last_error.empty() ? raw : last_error)});
|
max_attempts, last_error.empty() ? raw : last_error)});
|
||||||
}
|
}
|
||||||
throw std::runtime_error("LlamaGenerator: malformed brewery response");
|
throw std::runtime_error("LlamaGenerator: malformed brewery response");
|
||||||
|
|||||||
@@ -1,24 +1,115 @@
|
|||||||
/**
|
/**
|
||||||
* @file data_generation/llama/generate_user.cc
|
* @file data_generation/llama/generate_user.cc
|
||||||
* @brief Generates locale-aware user profiles with strict two-line formatting,
|
* @brief Builds persona/name-grounded user prompts, performs retry-based
|
||||||
* retry handling, and output sanitization for downstream parsing.
|
* inference, and validates structured JSON output for user records.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
|
|
||||||
#include <format>
|
#include <format>
|
||||||
|
#include <optional>
|
||||||
|
#include <stdexcept>
|
||||||
#include <string>
|
#include <string>
|
||||||
|
#include <string_view>
|
||||||
|
|
||||||
|
#include "data_generation/json_grammars.h"
|
||||||
#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"
|
||||||
|
|
||||||
// TODO: Implement locale-aware user profile generation.
|
static constexpr int kUserInitialMaxTokens = 1200;
|
||||||
// Current implementation returns a hardcoded test value and ignores the
|
|
||||||
// locale parameter. Future implementation should:
|
UserResult LlamaGenerator::GenerateUser(const EnrichedCity& city,
|
||||||
// 1. Load a USER_GENERATION.md prompt template with locale context
|
const UserPersona& persona,
|
||||||
// 2. Perform LLM inference with locale-specific username/bio generation
|
const Name& name) {
|
||||||
// 3. Parse and validate JSON output with retry handling (similar to brewery)
|
std::string style_affinities;
|
||||||
// 4. Return locale-aware username and biography
|
for (const std::string& style : persona.style_affinities) {
|
||||||
UserResult LlamaGenerator::GenerateUser(const std::string& locale) {
|
if (!style_affinities.empty()) {
|
||||||
return {.username = "test_user",
|
style_affinities += ", ";
|
||||||
.bio = std::format("This is a test user profile from {}.", locale)};
|
}
|
||||||
|
style_affinities += style;
|
||||||
|
}
|
||||||
|
|
||||||
|
const std::string system_prompt = prompt_directory_->Load("USER_GENERATION");
|
||||||
|
|
||||||
|
std::string user_prompt = std::format(
|
||||||
|
"## NAME:\n\n{} {}\n\n"
|
||||||
|
"## GENDER:\n\n{}\n\n"
|
||||||
|
"## CITY:\n\n{}\n\n"
|
||||||
|
"## COUNTRY:\n\n{}\n\n"
|
||||||
|
"## PERSONA:\n\n{}\n\n"
|
||||||
|
"## PERSONA DESCRIPTION:\n\n{}\n\n"
|
||||||
|
"## STYLE AFFINITIES:\n\n{}",
|
||||||
|
name.first_name, name.last_name, name.gender, city.location.city,
|
||||||
|
city.location.country, persona.name, persona.description,
|
||||||
|
style_affinities);
|
||||||
|
|
||||||
|
const std::string retry_context = std::format(
|
||||||
|
"Name: {} {}\nCity: {}, {}\nPersona: {}", name.first_name, name.last_name,
|
||||||
|
city.location.city, city.location.country, persona.name);
|
||||||
|
|
||||||
|
constexpr int max_attempts = 3;
|
||||||
|
std::string raw;
|
||||||
|
std::string last_error;
|
||||||
|
int max_tokens = kUserInitialMaxTokens;
|
||||||
|
|
||||||
|
for (int attempt = 0; attempt < max_attempts; ++attempt) {
|
||||||
|
raw = this->Infer(system_prompt, user_prompt, max_tokens, kUserJsonGrammar);
|
||||||
|
if (logger_) {
|
||||||
|
logger_->Log(
|
||||||
|
{.level = LogLevel::Debug,
|
||||||
|
.phase = PipelinePhase::UserGeneration,
|
||||||
|
.message = std::format("LlamaGenerator: raw output (attempt {}): {}",
|
||||||
|
attempt + 1, raw)});
|
||||||
|
}
|
||||||
|
|
||||||
|
UserResult user;
|
||||||
|
const std::optional<std::string> validation_error =
|
||||||
|
ValidateUserJson(raw, user);
|
||||||
|
|
||||||
|
if (!validation_error.has_value()) {
|
||||||
|
if (logger_) {
|
||||||
|
logger_->Log(
|
||||||
|
{.level = LogLevel::Info,
|
||||||
|
.phase = PipelinePhase::UserGeneration,
|
||||||
|
.message = std::format("LlamaGenerator: successfully "
|
||||||
|
"generated user data on attempt {}",
|
||||||
|
attempt + 1)});
|
||||||
|
}
|
||||||
|
|
||||||
|
user.first_name = name.first_name;
|
||||||
|
user.last_name = name.last_name;
|
||||||
|
user.gender = name.gender;
|
||||||
|
return user;
|
||||||
|
}
|
||||||
|
|
||||||
|
last_error = *validation_error;
|
||||||
|
if (logger_) {
|
||||||
|
logger_->Log({.level = LogLevel::Warn,
|
||||||
|
.phase = PipelinePhase::UserGeneration,
|
||||||
|
.message = std::format(
|
||||||
|
"LlamaGenerator: malformed user JSON (attempt {}): {}",
|
||||||
|
attempt + 1, *validation_error)});
|
||||||
|
}
|
||||||
|
|
||||||
|
user_prompt = std::format(
|
||||||
|
"Your previous response was invalid. Error: {}\nReturn the thought "
|
||||||
|
"process before the JSON if needed, then return ONLY valid JSON "
|
||||||
|
"with "
|
||||||
|
"exactly these keys, in this exact order: {{\"username\": "
|
||||||
|
"\"<handle "
|
||||||
|
"derived from the given name>\", \"bio\": \"<first-person bio "
|
||||||
|
"grounded in the persona>\", \"activity_weight\": <number between "
|
||||||
|
"0 "
|
||||||
|
"and 1>}}.\nDo not include markdown, comments, extra keys, or "
|
||||||
|
"literal placeholder values.\n\n{}",
|
||||||
|
*validation_error, retry_context);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (logger_) {
|
||||||
|
logger_->Log({.level = LogLevel::Error,
|
||||||
|
.phase = PipelinePhase::UserGeneration,
|
||||||
|
.message = std::format(
|
||||||
|
"LlamaGenerator: malformed user response "
|
||||||
|
"after {} attempts: {}",
|
||||||
|
max_attempts, last_error.empty() ? raw : last_error)});
|
||||||
|
}
|
||||||
|
throw std::runtime_error("LlamaGenerator: malformed user response");
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -9,13 +9,16 @@
|
|||||||
#include <boost/json.hpp>
|
#include <boost/json.hpp>
|
||||||
#include <cctype>
|
#include <cctype>
|
||||||
#include <optional>
|
#include <optional>
|
||||||
|
#include <span>
|
||||||
#include <stdexcept>
|
#include <stdexcept>
|
||||||
#include <string>
|
#include <string>
|
||||||
#include <string_view>
|
#include <string_view>
|
||||||
#include <vector>
|
#include <vector>
|
||||||
|
|
||||||
|
#include <llama.h>
|
||||||
|
|
||||||
#include "data_generation/llama_generator_helpers.h"
|
#include "data_generation/llama_generator_helpers.h"
|
||||||
#include "llama.h"
|
|
||||||
namespace {
|
namespace {
|
||||||
/**
|
/**
|
||||||
* String trimming: removes leading and trailing whitespace
|
* String trimming: removes leading and trailing whitespace
|
||||||
@@ -76,7 +79,7 @@ bool ReadRequiredTrimmedStringField(const boost::json::object& obj,
|
|||||||
return !out.empty();
|
return !out.empty();
|
||||||
}
|
}
|
||||||
|
|
||||||
bool HasSchemaPlaceholder(const std::array<std::string*, 4>& values) {
|
bool HasSchemaPlaceholder(std::span<std::string* const> values) {
|
||||||
for (const std::string* value : values) {
|
for (const std::string* value : values) {
|
||||||
std::string lowered = *value;
|
std::string lowered = *value;
|
||||||
std::ranges::transform(lowered, lowered.begin(),
|
std::ranges::transform(lowered, lowered.begin(),
|
||||||
@@ -207,3 +210,58 @@ std::optional<std::string> ValidateBreweryJson(const std::string& raw,
|
|||||||
|
|
||||||
return std::nullopt;
|
return std::nullopt;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
std::optional<std::string> ValidateUserJson(const std::string& raw,
|
||||||
|
UserResult& user_out) {
|
||||||
|
boost::system::error_code error_code;
|
||||||
|
const std::string_view raw_view(raw);
|
||||||
|
const size_t opening_brace = raw_view.find('{');
|
||||||
|
if (opening_brace == std::string_view::npos) {
|
||||||
|
return "JSON parse error: missing opening brace '{'";
|
||||||
|
}
|
||||||
|
|
||||||
|
const std::string_view json_payload = raw_view.substr(opening_brace);
|
||||||
|
boost::json::value json_value = boost::json::parse(json_payload, error_code);
|
||||||
|
if (error_code) {
|
||||||
|
return "JSON parse error: " + error_code.message();
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!json_value.is_object()) {
|
||||||
|
return "JSON root must be an object";
|
||||||
|
}
|
||||||
|
|
||||||
|
const auto& obj = json_value.get_object();
|
||||||
|
if (obj.size() != 3) {
|
||||||
|
return "JSON object must contain exactly three keys";
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string validation_error;
|
||||||
|
if (!ReadRequiredTrimmedStringField(obj, "username", user_out.username,
|
||||||
|
&validation_error)) {
|
||||||
|
return validation_error;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!ReadRequiredTrimmedStringField(obj, "bio", user_out.bio,
|
||||||
|
&validation_error)) {
|
||||||
|
return validation_error;
|
||||||
|
}
|
||||||
|
|
||||||
|
const boost::json::value* activity_weight_field =
|
||||||
|
obj.if_contains("activity_weight");
|
||||||
|
if (activity_weight_field == nullptr || !activity_weight_field->is_number()) {
|
||||||
|
return "Missing or invalid numeric field: activity_weight";
|
||||||
|
}
|
||||||
|
|
||||||
|
const double activity_weight = activity_weight_field->to_number<double>();
|
||||||
|
if (activity_weight < 0.0 || activity_weight > 1.0) {
|
||||||
|
return "activity_weight must be between 0 and 1";
|
||||||
|
}
|
||||||
|
user_out.activity_weight = static_cast<float>(activity_weight);
|
||||||
|
|
||||||
|
const std::array schema_placeholders = {&user_out.username, &user_out.bio};
|
||||||
|
if (HasSchemaPlaceholder(schema_placeholders)) {
|
||||||
|
return "JSON appears to be a schema placeholder, not content";
|
||||||
|
}
|
||||||
|
|
||||||
|
return std::nullopt;
|
||||||
|
}
|
||||||
|
|||||||
@@ -14,9 +14,10 @@
|
|||||||
#include <string_view>
|
#include <string_view>
|
||||||
#include <vector>
|
#include <vector>
|
||||||
|
|
||||||
|
#include <llama.h>
|
||||||
|
|
||||||
#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"
|
||||||
#include "llama.h"
|
|
||||||
|
|
||||||
static constexpr size_t kPromptTokenSlack = 8;
|
static constexpr size_t kPromptTokenSlack = 8;
|
||||||
// Minimum tokens to keep when using top-p sampling. Ensures at least one
|
// Minimum tokens to keep when using top-p sampling. Ensures at least one
|
||||||
|
|||||||
@@ -11,8 +11,9 @@
|
|||||||
#include <stdexcept>
|
#include <stdexcept>
|
||||||
#include <string>
|
#include <string>
|
||||||
|
|
||||||
|
#include <llama.h>
|
||||||
|
|
||||||
#include "data_model/models.h"
|
#include "data_model/models.h"
|
||||||
#include "llama.h"
|
|
||||||
|
|
||||||
static constexpr uint32_t kMaxContextSize = 32768U;
|
static constexpr uint32_t kMaxContextSize = 32768U;
|
||||||
|
|
||||||
|
|||||||
@@ -10,9 +10,10 @@
|
|||||||
#include <string>
|
#include <string>
|
||||||
#include <utility>
|
#include <utility>
|
||||||
|
|
||||||
|
#include <ggml-backend.h>
|
||||||
|
#include <llama.h>
|
||||||
|
|
||||||
#include "data_generation/llama_generator.h"
|
#include "data_generation/llama_generator.h"
|
||||||
#include "ggml-backend.h"
|
|
||||||
#include "llama.h"
|
|
||||||
|
|
||||||
// Maximum batch size for decode operations. Capping the batch prevents
|
// Maximum batch size for decode operations. Capping the batch prevents
|
||||||
// excessive memory allocation while maintaining inference performance.
|
// excessive memory allocation while maintaining inference performance.
|
||||||
|
|||||||
@@ -8,9 +8,19 @@
|
|||||||
|
|
||||||
#include "data_generation/mock_generator.h"
|
#include "data_generation/mock_generator.h"
|
||||||
|
|
||||||
size_t MockGenerator::DeterministicHash(const Location& location) {
|
size_t MockGenerator::DeterministicHash(const City& location) {
|
||||||
size_t seed = 0;
|
size_t seed = 0;
|
||||||
boost::hash_combine(seed, location.city);
|
boost::hash_combine(seed, location.city);
|
||||||
boost::hash_combine(seed, location.country);
|
boost::hash_combine(seed, location.country);
|
||||||
return seed;
|
return seed;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
size_t MockGenerator::DeterministicHash(const City& location,
|
||||||
|
const UserPersona& persona,
|
||||||
|
const Name& name) {
|
||||||
|
size_t seed = DeterministicHash(location);
|
||||||
|
boost::hash_combine(seed, persona.name);
|
||||||
|
boost::hash_combine(seed, name.first_name);
|
||||||
|
boost::hash_combine(seed, name.last_name);
|
||||||
|
return seed;
|
||||||
|
}
|
||||||
|
|||||||
@@ -11,7 +11,7 @@
|
|||||||
#include "data_generation/mock_generator.h"
|
#include "data_generation/mock_generator.h"
|
||||||
|
|
||||||
BreweryResult MockGenerator::GenerateBrewery(
|
BreweryResult MockGenerator::GenerateBrewery(
|
||||||
const Location& location, const std::string& /*region_context*/) {
|
const City& location, const std::string& /*region_context*/) {
|
||||||
const size_t hash = DeterministicHash(location);
|
const size_t hash = DeterministicHash(location);
|
||||||
|
|
||||||
const std::string_view adjective =
|
const std::string_view adjective =
|
||||||
|
|||||||
@@ -1,22 +1,31 @@
|
|||||||
/**
|
/**
|
||||||
* @file data_generation/mock/generate_user.cc
|
* @file data_generation/mock/generate_user.cc
|
||||||
* @brief Generates deterministic mock user profiles by hashing locale values
|
* @brief Generates deterministic mock user profiles by hashing city, persona,
|
||||||
* into predefined username and bio collections.
|
* and sampled name into predefined username and bio collections.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#include <functional>
|
|
||||||
#include <string>
|
|
||||||
#include <string_view>
|
#include <string_view>
|
||||||
|
|
||||||
#include "data_generation/mock_generator.h"
|
#include "data_generation/mock_generator.h"
|
||||||
|
|
||||||
UserResult MockGenerator::GenerateUser(const std::string& locale) {
|
UserResult MockGenerator::GenerateUser(const EnrichedCity& city,
|
||||||
const size_t hash = std::hash<std::string>{}(locale);
|
const UserPersona& persona,
|
||||||
|
const Name& name) {
|
||||||
|
const size_t hash = DeterministicHash(city.location, persona, name);
|
||||||
|
|
||||||
UserResult result;
|
|
||||||
const std::string_view username = kUsernames[hash % kUsernames.size()];
|
const std::string_view username = kUsernames[hash % kUsernames.size()];
|
||||||
const std::string_view bio = kBios[hash / kBioHashStride % kBios.size()];
|
const std::string_view bio = kBios[hash / kBioHashStride % kBios.size()];
|
||||||
result.username = username;
|
const float activity_weight =
|
||||||
result.bio = bio;
|
static_cast<float>(hash / kActivityWeightHashStride %
|
||||||
return result;
|
kActivityWeightHashRange) /
|
||||||
|
static_cast<float>(kActivityWeightHashRange);
|
||||||
|
|
||||||
|
return {
|
||||||
|
.first_name = name.first_name,
|
||||||
|
.last_name = name.last_name,
|
||||||
|
.gender = name.gender,
|
||||||
|
.username = std::string(username),
|
||||||
|
.bio = std::string(bio),
|
||||||
|
.activity_weight = activity_weight,
|
||||||
|
};
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,110 +0,0 @@
|
|||||||
/**
|
|
||||||
* @file json_handling/json_loader.cc
|
|
||||||
* @brief Parses curated location JSON input into strongly typed Location
|
|
||||||
* records with strict field validation and descriptive error reporting.
|
|
||||||
*/
|
|
||||||
|
|
||||||
#include "json_handling/json_loader.h"
|
|
||||||
|
|
||||||
#include <format>
|
|
||||||
#include "services/logging/logger.h"
|
|
||||||
#include <iostream>
|
|
||||||
|
|
||||||
#include <boost/json.hpp>
|
|
||||||
#include <fstream>
|
|
||||||
#include <sstream>
|
|
||||||
#include <stdexcept>
|
|
||||||
#include <string>
|
|
||||||
#include <string_view>
|
|
||||||
|
|
||||||
static std::string ReadRequiredString(const boost::json::object& object,
|
|
||||||
const char* key) {
|
|
||||||
const boost::json::value* value = object.if_contains(key);
|
|
||||||
if (value == nullptr || !value->is_string()) {
|
|
||||||
throw std::runtime_error(
|
|
||||||
std::format("Missing or invalid string field: {}", key));
|
|
||||||
}
|
|
||||||
const std::string_view text = value->as_string();
|
|
||||||
return std::string(text);
|
|
||||||
}
|
|
||||||
|
|
||||||
static double ReadRequiredNumber(const boost::json::object& object,
|
|
||||||
const char* key) {
|
|
||||||
const boost::json::value* value = object.if_contains(key);
|
|
||||||
if (value == nullptr || !value->is_number()) {
|
|
||||||
throw std::runtime_error(
|
|
||||||
std::format("Missing or invalid numeric field: {}", key));
|
|
||||||
}
|
|
||||||
return value->to_number<double>();
|
|
||||||
}
|
|
||||||
|
|
||||||
static std::vector<std::string> ReadRequiredStringArray(
|
|
||||||
const boost::json::object& object, const char* key) {
|
|
||||||
const boost::json::value* value = object.if_contains(key);
|
|
||||||
if (value == nullptr || !value->is_array()) {
|
|
||||||
throw std::runtime_error(
|
|
||||||
std::format("Missing or invalid string array field: {}", key));
|
|
||||||
}
|
|
||||||
|
|
||||||
const auto& array = value->as_array();
|
|
||||||
std::vector<std::string> items;
|
|
||||||
items.reserve(array.size());
|
|
||||||
for (const auto& item : array) {
|
|
||||||
if (!item.is_string()) {
|
|
||||||
throw std::runtime_error(
|
|
||||||
std::format("Missing or invalid string array field: {}", key));
|
|
||||||
}
|
|
||||||
items.emplace_back(item.as_string());
|
|
||||||
}
|
|
||||||
return items;
|
|
||||||
}
|
|
||||||
|
|
||||||
std::vector<Location> JsonLoader::LoadLocations(
|
|
||||||
const std::filesystem::path& filepath, std::shared_ptr<ILogger> logger) {
|
|
||||||
std::ifstream input(filepath);
|
|
||||||
if (!input.is_open()) {
|
|
||||||
throw std::runtime_error("Failed to open locations file: " +
|
|
||||||
filepath.string());
|
|
||||||
}
|
|
||||||
|
|
||||||
std::stringstream buffer;
|
|
||||||
buffer << input.rdbuf();
|
|
||||||
const std::string content = buffer.str();
|
|
||||||
|
|
||||||
boost::system::error_code error;
|
|
||||||
boost::json::value root = boost::json::parse(content, error);
|
|
||||||
if (error) {
|
|
||||||
throw std::runtime_error("Failed to parse locations JSON: " +
|
|
||||||
error.message());
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!root.is_array()) {
|
|
||||||
throw std::runtime_error(
|
|
||||||
"Invalid locations JSON: root element must be an array");
|
|
||||||
}
|
|
||||||
|
|
||||||
std::vector<Location> locations;
|
|
||||||
const auto& items = root.as_array();
|
|
||||||
locations.reserve(items.size());
|
|
||||||
|
|
||||||
for (const auto& item : items) {
|
|
||||||
if (!item.is_object()) {
|
|
||||||
throw std::runtime_error(
|
|
||||||
"Invalid locations JSON: each entry must be an object");
|
|
||||||
}
|
|
||||||
|
|
||||||
const auto& object = item.as_object();
|
|
||||||
locations.push_back(Location{
|
|
||||||
.city = ReadRequiredString(object, "city"),
|
|
||||||
.state_province = ReadRequiredString(object, "state_province"),
|
|
||||||
.iso3166_2 = ReadRequiredString(object, "iso3166_2"),
|
|
||||||
.country = ReadRequiredString(object, "country"),
|
|
||||||
.iso3166_1 = ReadRequiredString(object, "iso3166_1"),
|
|
||||||
.local_languages = ReadRequiredStringArray(object, "local_languages"),
|
|
||||||
.latitude = ReadRequiredNumber(object, "latitude"),
|
|
||||||
.longitude = ReadRequiredNumber(object, "longitude"),
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
return locations;
|
|
||||||
}
|
|
||||||
@@ -18,25 +18,7 @@
|
|||||||
#include <string>
|
#include <string>
|
||||||
#include <thread>
|
#include <thread>
|
||||||
|
|
||||||
#include "biergarten_pipeline_orchestrator.h"
|
#include "biergarten_pipeline.h"
|
||||||
#include "concurrency/bounded_channel.h"
|
|
||||||
#include "data_generation/llama_generator.h"
|
|
||||||
#include "data_generation/mock_generator.h"
|
|
||||||
#include "data_generation/prompt_formatting/gemma4_jinja_prompt_formatter.h"
|
|
||||||
#include "data_model/models.h"
|
|
||||||
#include "llama_backend_state.h"
|
|
||||||
#include "services/database/export_service.h"
|
|
||||||
#include "services/database/sqlite_export_service.h"
|
|
||||||
#include "services/datetime/timer.h"
|
|
||||||
#include "services/enrichment/enrichment_service.h"
|
|
||||||
#include "services/enrichment/mock_enrichment.h"
|
|
||||||
#include "services/enrichment/wikipedia_service.h"
|
|
||||||
#include "services/logging/log_dispatcher.h"
|
|
||||||
#include "services/logging/log_entry.h"
|
|
||||||
#include "services/logging/log_producer.h"
|
|
||||||
#include "services/logging/logger.h"
|
|
||||||
#include "services/prompting/prompt_directory.h"
|
|
||||||
#include "web_client/http_web_client.h"
|
|
||||||
|
|
||||||
namespace di = boost::di;
|
namespace di = boost::di;
|
||||||
|
|
||||||
@@ -53,6 +35,7 @@ int main(const int argc, char** argv) {
|
|||||||
std::make_shared<LogProducer>(log_channel);
|
std::make_shared<LogProducer>(log_channel);
|
||||||
|
|
||||||
std::thread log_thread([&log_dispatcher] { log_dispatcher->Run(); });
|
std::thread log_thread([&log_dispatcher] { log_dispatcher->Run(); });
|
||||||
|
|
||||||
auto shutdown = [&](const int exit_code) {
|
auto shutdown = [&](const int exit_code) {
|
||||||
log_channel.Close();
|
log_channel.Close();
|
||||||
log_thread.join();
|
log_thread.join();
|
||||||
@@ -103,6 +86,30 @@ int main(const int argc, char** argv) {
|
|||||||
di::bind<ApplicationOptions>().to(options),
|
di::bind<ApplicationOptions>().to(options),
|
||||||
di::bind<std::string>().to(model_path),
|
di::bind<std::string>().to(model_path),
|
||||||
di::bind<IExportService>().to<SqliteExportService>(),
|
di::bind<IExportService>().to<SqliteExportService>(),
|
||||||
|
di::bind<ICuratedDataService>().to(
|
||||||
|
[options, &log_producer]() -> std::unique_ptr<ICuratedDataService> {
|
||||||
|
if (options.generator.use_mocked) {
|
||||||
|
log_producer->Log({.level = LogLevel::Info,
|
||||||
|
.phase = PipelinePhase::Startup,
|
||||||
|
.message = "Curated data: mock"});
|
||||||
|
|
||||||
|
return std::make_unique<MockCuratedDataService>();
|
||||||
|
}
|
||||||
|
|
||||||
|
log_producer->Log({.level = LogLevel::Info,
|
||||||
|
.phase = PipelinePhase::Startup,
|
||||||
|
.message = "Curated data: JsonLoader"});
|
||||||
|
|
||||||
|
return std::make_unique<CuratedJsonDataService>(
|
||||||
|
CuratedDataFilePaths{
|
||||||
|
.cities_path = "cities.json",
|
||||||
|
.personas_path = "personas.json",
|
||||||
|
.forenames_path = "forenames-by-country.json",
|
||||||
|
.surnames_path = "surnames-by-country.json",
|
||||||
|
});
|
||||||
|
}),
|
||||||
|
di::bind<IPostalCodeService>().to<MockPostalCodeService>(),
|
||||||
|
|
||||||
di::bind<IPromptFormatter>().to([options, log_producer] {
|
di::bind<IPromptFormatter>().to([options, log_producer] {
|
||||||
if (options.generator.use_mocked) {
|
if (options.generator.use_mocked) {
|
||||||
{
|
{
|
||||||
@@ -131,11 +138,11 @@ int main(const int argc, char** argv) {
|
|||||||
}
|
}
|
||||||
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>(
|
||||||
std::make_unique<HttpWebClient>(log_producer));
|
std::make_unique<HttpWebClient>(log_producer));
|
||||||
}),
|
}),
|
||||||
@@ -143,18 +150,17 @@ int main(const int argc, char** argv) {
|
|||||||
[options, &log_producer](
|
[options, &log_producer](
|
||||||
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>(
|
||||||
inj.template create<std::unique_ptr<WebClient>>(),
|
inj.template create<std::unique_ptr<WebClient>>(),
|
||||||
log_producer);
|
log_producer);
|
||||||
@@ -163,27 +169,35 @@ int main(const int argc, char** argv) {
|
|||||||
[&options, &model_path, &sampling, &prompt_directory,
|
[&options, &model_path, &sampling, &prompt_directory,
|
||||||
&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>();
|
||||||
}
|
}
|
||||||
{
|
|
||||||
|
#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,
|
||||||
.message = std::format(
|
.message = std::format(
|
||||||
"Generator: LlamaGenerator | model={} | temp={:.2f} "
|
"Generator: LlamaGenerator | model={} | "
|
||||||
"top_p={:.2f} top_k={} n_ctx={} seed={}",
|
"temp={:.2f} top_p={:.2f} top_k={} n_ctx={} seed={}",
|
||||||
model_path, sampling.temperature, sampling.top_p,
|
model_path, sampling.temperature, sampling.top_p,
|
||||||
sampling.top_k, sampling.n_ctx, sampling.seed)});
|
sampling.top_k, sampling.n_ctx, sampling.seed)});
|
||||||
}
|
|
||||||
return std::make_unique<LlamaGenerator>(
|
return std::make_unique<LlamaGenerator>(
|
||||||
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 =
|
||||||
@@ -202,7 +216,6 @@ int main(const int argc, char** argv) {
|
|||||||
timer.Elapsed())});
|
timer.Elapsed())});
|
||||||
|
|
||||||
return shutdown(EXIT_SUCCESS);
|
return shutdown(EXIT_SUCCESS);
|
||||||
|
|
||||||
} catch (const std::exception& exception) {
|
} catch (const std::exception& exception) {
|
||||||
const LogDTO log_entry{.level = LogLevel::Error,
|
const LogDTO log_entry{.level = LogLevel::Error,
|
||||||
.phase = PipelinePhase::Teardown,
|
.phase = PipelinePhase::Teardown,
|
||||||
|
|||||||
@@ -0,0 +1,291 @@
|
|||||||
|
/**
|
||||||
|
* @file json_handling/json_loader.cc
|
||||||
|
* @brief Parses curated location JSON input into strongly typed City
|
||||||
|
* records with strict field validation and descriptive error reporting.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include "services/curated_data/curated_json_data_service.h"
|
||||||
|
|
||||||
|
#include <boost/json.hpp>
|
||||||
|
#include <format>
|
||||||
|
#include <fstream>
|
||||||
|
#include <iostream>
|
||||||
|
#include <sstream>
|
||||||
|
#include <stdexcept>
|
||||||
|
#include <string>
|
||||||
|
#include <string_view>
|
||||||
|
#include <utility>
|
||||||
|
#include <vector>
|
||||||
|
|
||||||
|
#include "services/logging/logger.h"
|
||||||
|
|
||||||
|
static std::string ReadRequiredString(const boost::json::object& object,
|
||||||
|
const char* key) {
|
||||||
|
const boost::json::value* value = object.if_contains(key);
|
||||||
|
if (value == nullptr || !value->is_string()) {
|
||||||
|
throw std::runtime_error(
|
||||||
|
std::format("Missing or invalid string field: {}", key));
|
||||||
|
}
|
||||||
|
const std::string_view text = value->as_string();
|
||||||
|
return std::string(text);
|
||||||
|
}
|
||||||
|
|
||||||
|
static double ReadRequiredNumber(const boost::json::object& object,
|
||||||
|
const char* key) {
|
||||||
|
const boost::json::value* value = object.if_contains(key);
|
||||||
|
if (value == nullptr || !value->is_number()) {
|
||||||
|
throw std::runtime_error(
|
||||||
|
std::format("Missing or invalid numeric field: {}", key));
|
||||||
|
}
|
||||||
|
return value->to_number<double>();
|
||||||
|
}
|
||||||
|
|
||||||
|
static std::vector<std::string> ReadRequiredStringArray(
|
||||||
|
const boost::json::object& object, const char* key) {
|
||||||
|
const boost::json::value* value = object.if_contains(key);
|
||||||
|
if (value == nullptr || !value->is_array()) {
|
||||||
|
throw std::runtime_error(
|
||||||
|
std::format("Missing or invalid string array field: {}", key));
|
||||||
|
}
|
||||||
|
|
||||||
|
const auto& array = value->as_array();
|
||||||
|
std::vector<std::string> items;
|
||||||
|
items.reserve(array.size());
|
||||||
|
for (const auto& item : array) {
|
||||||
|
if (!item.is_string()) {
|
||||||
|
throw std::runtime_error(
|
||||||
|
std::format("Missing or invalid string array field: {}", key));
|
||||||
|
}
|
||||||
|
items.emplace_back(item.as_string());
|
||||||
|
}
|
||||||
|
return items;
|
||||||
|
}
|
||||||
|
|
||||||
|
namespace {
|
||||||
|
|
||||||
|
boost::json::value ParseJsonFile(const std::filesystem::path& filepath,
|
||||||
|
const char* what) {
|
||||||
|
std::ifstream input(filepath);
|
||||||
|
if (!input.is_open()) {
|
||||||
|
throw std::runtime_error(
|
||||||
|
std::format("Failed to open {} file: {}", what, filepath.string()));
|
||||||
|
}
|
||||||
|
|
||||||
|
std::stringstream buffer;
|
||||||
|
buffer << input.rdbuf();
|
||||||
|
|
||||||
|
boost::system::error_code error;
|
||||||
|
boost::json::value root = boost::json::parse(buffer.str(), error);
|
||||||
|
if (error) {
|
||||||
|
throw std::runtime_error(
|
||||||
|
std::format("Failed to parse {} JSON: {}", what, error.message()));
|
||||||
|
}
|
||||||
|
return root;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Returns the first element of a string array field, falling back to
|
||||||
|
* `fallback_key` if `key` is missing/empty (some source entries only have a
|
||||||
|
* "localized" form and no "romanized" form).
|
||||||
|
*/
|
||||||
|
std::string ReadFirstOfStringArray(const boost::json::object& object,
|
||||||
|
const char* key, const char* fallback_key) {
|
||||||
|
for (const char* candidate_key : {key, fallback_key}) {
|
||||||
|
const boost::json::value* value = object.if_contains(candidate_key);
|
||||||
|
if (value == nullptr || !value->is_array()) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
const auto& array = value->as_array();
|
||||||
|
if (!array.empty() && array.front().is_string()) {
|
||||||
|
return std::string(array.front().as_string());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
throw std::runtime_error(
|
||||||
|
std::format("Missing or invalid string array field: {}", key));
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace
|
||||||
|
|
||||||
|
CuratedJsonDataService::CuratedJsonDataService(CuratedDataFilePaths filepaths)
|
||||||
|
: filepaths_(std::move(filepaths)) {}
|
||||||
|
|
||||||
|
const CityList& CuratedJsonDataService::LoadCities() {
|
||||||
|
if (!cache_.locations.empty()) {
|
||||||
|
return cache_.locations;
|
||||||
|
}
|
||||||
|
|
||||||
|
const boost::json::value root =
|
||||||
|
ParseJsonFile(filepaths_.cities_path, "cities");
|
||||||
|
|
||||||
|
if (!root.is_array()) {
|
||||||
|
throw std::runtime_error(
|
||||||
|
"Invalid cities JSON: root element must be an array");
|
||||||
|
}
|
||||||
|
|
||||||
|
CityList locations;
|
||||||
|
const auto& items = root.as_array();
|
||||||
|
locations.reserve(items.size());
|
||||||
|
|
||||||
|
for (const auto& item : items) {
|
||||||
|
if (!item.is_object()) {
|
||||||
|
throw std::runtime_error(
|
||||||
|
"Invalid cities JSON: each entry must be an object");
|
||||||
|
}
|
||||||
|
|
||||||
|
const auto& object = item.as_object();
|
||||||
|
|
||||||
|
const boost::json::value* postal_code = object.if_contains("postal_code");
|
||||||
|
if (postal_code == nullptr || !postal_code->is_object()) {
|
||||||
|
throw std::runtime_error("Missing or invalid object field: postal_code");
|
||||||
|
}
|
||||||
|
|
||||||
|
locations.push_back(City{
|
||||||
|
.city = ReadRequiredString(object, "city"),
|
||||||
|
.state_province = ReadRequiredString(object, "state_province"),
|
||||||
|
.iso3166_2 = ReadRequiredString(object, "iso3166_2"),
|
||||||
|
.country = ReadRequiredString(object, "country"),
|
||||||
|
.iso3166_1 = ReadRequiredString(object, "iso3166_1"),
|
||||||
|
.local_languages = ReadRequiredStringArray(object, "local_languages"),
|
||||||
|
.postal_regex =
|
||||||
|
ReadRequiredStringArray(postal_code->as_object(), "city_regex"),
|
||||||
|
.postal_code_examples =
|
||||||
|
ReadRequiredStringArray(postal_code->as_object(), "examples"),
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
for (auto location : locations) {
|
||||||
|
std::cout << "Location: " << location.city << ", "
|
||||||
|
<< location.state_province << ", " << location.iso3166_2 << ", "
|
||||||
|
<< location.country << ", " << location.iso3166_1 << "\n";
|
||||||
|
|
||||||
|
for (const auto& regex : location.postal_regex) {
|
||||||
|
std::cout << " Postal regex: " << regex << "\n";
|
||||||
|
}
|
||||||
|
|
||||||
|
for (const auto& example : location.postal_code_examples) {
|
||||||
|
std::cout << " Postal example: " << example << "\n";
|
||||||
|
}
|
||||||
|
|
||||||
|
for (const auto& lang : location.local_languages) {
|
||||||
|
std::cout << " Local language: " << lang << "\n";
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
cache_.locations = std::move(locations);
|
||||||
|
return cache_.locations;
|
||||||
|
}
|
||||||
|
|
||||||
|
const PersonasList& CuratedJsonDataService::LoadPersonas() {
|
||||||
|
if (!cache_.personas.empty()) {
|
||||||
|
return cache_.personas;
|
||||||
|
}
|
||||||
|
|
||||||
|
const boost::json::value root =
|
||||||
|
ParseJsonFile(filepaths_.personas_path, "personas");
|
||||||
|
|
||||||
|
if (!root.is_array()) {
|
||||||
|
throw std::runtime_error(
|
||||||
|
"Invalid personas JSON: root element must be an array");
|
||||||
|
}
|
||||||
|
|
||||||
|
PersonasList personas;
|
||||||
|
const auto& items = root.as_array();
|
||||||
|
personas.reserve(items.size());
|
||||||
|
|
||||||
|
for (const auto& item : items) {
|
||||||
|
if (!item.is_object()) {
|
||||||
|
throw std::runtime_error(
|
||||||
|
"Invalid personas JSON: each entry must be an object");
|
||||||
|
}
|
||||||
|
|
||||||
|
const auto& object = item.as_object();
|
||||||
|
personas.push_back(UserPersona{
|
||||||
|
.name = ReadRequiredString(object, "name"),
|
||||||
|
.description = ReadRequiredString(object, "description"),
|
||||||
|
.style_affinities = ReadRequiredStringArray(object, "style_affinities"),
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
cache_.personas = std::move(personas);
|
||||||
|
return cache_.personas;
|
||||||
|
}
|
||||||
|
|
||||||
|
const ForenamesByCountryMap& CuratedJsonDataService::LoadForenamesByCountry() {
|
||||||
|
if (!cache_.forenames_by_country.empty()) {
|
||||||
|
return cache_.forenames_by_country;
|
||||||
|
}
|
||||||
|
|
||||||
|
const boost::json::value root =
|
||||||
|
ParseJsonFile(filepaths_.forenames_path, "forenames-by-country");
|
||||||
|
|
||||||
|
if (!root.is_object()) {
|
||||||
|
throw std::runtime_error(
|
||||||
|
"Invalid forenames-by-country JSON: root element must be an object "
|
||||||
|
"keyed by ISO 3166-1 country code");
|
||||||
|
}
|
||||||
|
|
||||||
|
ForenamesByCountryMap forenames_by_country;
|
||||||
|
for (const auto& [country_code, regions] : root.as_object()) {
|
||||||
|
if (!regions.is_array()) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
ForenameList entries;
|
||||||
|
for (const auto& region : regions.as_array()) {
|
||||||
|
if (!region.is_object()) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
const boost::json::value* names = region.as_object().if_contains("names");
|
||||||
|
if (names == nullptr || !names->is_array()) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
for (const auto& name_value : names->as_array()) {
|
||||||
|
if (!name_value.is_object()) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
const auto& name_object = name_value.as_object();
|
||||||
|
entries.emplace_back(ForenameEntry{
|
||||||
|
.name =
|
||||||
|
ReadFirstOfStringArray(name_object, "romanized", "localized"),
|
||||||
|
.gender = ReadRequiredString(name_object, "gender"),
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
forenames_by_country.emplace(country_code, std::move(entries));
|
||||||
|
}
|
||||||
|
|
||||||
|
cache_.forenames_by_country = std::move(forenames_by_country);
|
||||||
|
return cache_.forenames_by_country;
|
||||||
|
}
|
||||||
|
|
||||||
|
const SurnamesByCountryMap& CuratedJsonDataService::LoadSurnamesByCountry() {
|
||||||
|
if (!cache_.surnames_by_country.empty()) {
|
||||||
|
return cache_.surnames_by_country;
|
||||||
|
}
|
||||||
|
|
||||||
|
const boost::json::value root =
|
||||||
|
ParseJsonFile(filepaths_.surnames_path, "surnames-by-country");
|
||||||
|
|
||||||
|
if (!root.is_object()) {
|
||||||
|
throw std::runtime_error(
|
||||||
|
"Invalid surnames-by-country JSON: root element must be an object "
|
||||||
|
"keyed by ISO 3166-1 country code");
|
||||||
|
}
|
||||||
|
|
||||||
|
SurnamesByCountryMap surnames_by_country;
|
||||||
|
for (const auto& [country_code, name_entries] : root.as_object()) {
|
||||||
|
if (!name_entries.is_array()) continue;
|
||||||
|
|
||||||
|
SurnameList surnames;
|
||||||
|
for (const auto& name_value : name_entries.as_array()) {
|
||||||
|
if (!name_value.is_object()) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
surnames.emplace_back(ReadFirstOfStringArray(name_value.as_object(),
|
||||||
|
"romanized", "localized"));
|
||||||
|
}
|
||||||
|
surnames_by_country.emplace(country_code, std::move(surnames));
|
||||||
|
}
|
||||||
|
|
||||||
|
cache_.surnames_by_country = std::move(surnames_by_country);
|
||||||
|
return cache_.surnames_by_country;
|
||||||
|
}
|
||||||
@@ -0,0 +1,96 @@
|
|||||||
|
/**
|
||||||
|
* @file services/curated_data/mock_curated_data_service.cc
|
||||||
|
* @brief Fixed in-memory location, persona, and name dataset for mock mode.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include "services/curated_data/mock_curated_data_service.h"
|
||||||
|
|
||||||
|
MockCuratedDataService::MockCuratedDataService()
|
||||||
|
: locations_{
|
||||||
|
{.city = "Portland",
|
||||||
|
.state_province = "Oregon",
|
||||||
|
.iso3166_2 = "US-OR",
|
||||||
|
.country = "United States",
|
||||||
|
.iso3166_1 = "US",
|
||||||
|
.local_languages = {"en"},
|
||||||
|
.postal_regex = {"^972[0-9]{2}(?:-[0-9]{4})?$"},
|
||||||
|
.postal_code_examples = {"97201", "97294"}},
|
||||||
|
{.city = "Munich",
|
||||||
|
.state_province = "Bavaria",
|
||||||
|
.iso3166_2 = "DE-BY",
|
||||||
|
.country = "Germany",
|
||||||
|
.iso3166_1 = "DE",
|
||||||
|
.local_languages = {"de"},
|
||||||
|
.postal_regex = {"^8[01][0-9]{3}$"},
|
||||||
|
.postal_code_examples = {"80331", "81929"}},
|
||||||
|
{.city = "Lyon",
|
||||||
|
.state_province = "Auvergne-Rhone-Alpes",
|
||||||
|
.iso3166_2 = "FR-ARA",
|
||||||
|
.country = "France",
|
||||||
|
.iso3166_1 = "FR",
|
||||||
|
.local_languages = {"fr"},
|
||||||
|
.postal_regex = {"^6900[1-9]$"},
|
||||||
|
.postal_code_examples = {"69001", "69009"}},
|
||||||
|
{.city = "Brussels",
|
||||||
|
.state_province = "Brussels-Capital",
|
||||||
|
.iso3166_2 = "BE-BRU",
|
||||||
|
.country = "Belgium",
|
||||||
|
.iso3166_1 = "BE",
|
||||||
|
.local_languages = {"nl", "fr"},
|
||||||
|
.postal_regex = {"^1(?:0[0-9]{2}|1[0-9]{2}|20[0-9]|210)$"},
|
||||||
|
.postal_code_examples = {"1000", "1210"}},
|
||||||
|
},
|
||||||
|
personas_{
|
||||||
|
UserPersona{.name = "Hophead Explorer",
|
||||||
|
.description = "Chases hop-forward IPAs and seeks out "
|
||||||
|
"taprooms with rotating drafts.",
|
||||||
|
.style_affinities = {"IPA", "Pale Ale"}},
|
||||||
|
UserPersona{.name = "Lager Traditionalist",
|
||||||
|
.description = "Prefers clean, balanced lagers and "
|
||||||
|
"classic pub styles.",
|
||||||
|
.style_affinities = {"Lager", "Pilsner"}},
|
||||||
|
UserPersona{.name = "Sour Curious",
|
||||||
|
.description = "Seeks out wild ales, sours, and "
|
||||||
|
"barrel-aged experiments.",
|
||||||
|
.style_affinities = {"Sour", "Wild Ale"}},
|
||||||
|
},
|
||||||
|
forenames_by_country_{
|
||||||
|
{"US",
|
||||||
|
ForenameList{
|
||||||
|
ForenameEntry{.name = "James", .gender = "M"},
|
||||||
|
ForenameEntry{.name = "Mary", .gender = "F"},
|
||||||
|
}},
|
||||||
|
{"DE",
|
||||||
|
ForenameList{
|
||||||
|
ForenameEntry{.name = "Lukas", .gender = "M"},
|
||||||
|
ForenameEntry{.name = "Anna", .gender = "F"},
|
||||||
|
}},
|
||||||
|
{"FR",
|
||||||
|
ForenameList{
|
||||||
|
ForenameEntry{.name = "Lucas", .gender = "M"},
|
||||||
|
ForenameEntry{.name = "Camille", .gender = "F"},
|
||||||
|
}},
|
||||||
|
{"BE",
|
||||||
|
ForenameList{
|
||||||
|
ForenameEntry{.name = "Noah", .gender = "M"},
|
||||||
|
ForenameEntry{.name = "Emma", .gender = "F"},
|
||||||
|
}},
|
||||||
|
},
|
||||||
|
surnames_by_country_{
|
||||||
|
{"US", SurnameList{"Smith", "Johnson"}},
|
||||||
|
{"DE", SurnameList{"Muller", "Schmidt"}},
|
||||||
|
{"FR", SurnameList{"Martin", "Bernard"}},
|
||||||
|
{"BE", SurnameList{"Peeters", "Janssens"}},
|
||||||
|
} {}
|
||||||
|
|
||||||
|
const CityList& MockCuratedDataService::LoadCities() { return locations_; }
|
||||||
|
|
||||||
|
const PersonasList& MockCuratedDataService::LoadPersonas() { return personas_; }
|
||||||
|
|
||||||
|
const ForenamesByCountryMap& MockCuratedDataService::LoadForenamesByCountry() {
|
||||||
|
return forenames_by_country_;
|
||||||
|
}
|
||||||
|
|
||||||
|
const SurnamesByCountryMap& MockCuratedDataService::LoadSurnamesByCountry() {
|
||||||
|
return surnames_by_country_;
|
||||||
|
}
|
||||||
@@ -1,5 +1,6 @@
|
|||||||
/**
|
/**
|
||||||
* @file wikipedia/fetch_extract.cc
|
* @file wikipedia/fetch_extract.cc
|
||||||
|
* @brief WikipediaEnrichmentService::FetchExtract() implementation.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#include <boost/json.hpp>
|
#include <boost/json.hpp>
|
||||||
@@ -20,8 +21,9 @@ std::string WikipediaEnrichmentService::FetchExtract(std::string_view query) {
|
|||||||
if (const auto cache_it = this->extract_cache_.find(cache_key);
|
if (const auto cache_it = this->extract_cache_.find(cache_key);
|
||||||
cache_it != this->extract_cache_.end()) {
|
cache_it != this->extract_cache_.end()) {
|
||||||
if (logger_) {
|
if (logger_) {
|
||||||
logger_->Log({.level = LogLevel::Debug,
|
logger_->Log(
|
||||||
.phase = PipelinePhase::UserGeneration,
|
{.level = LogLevel::Debug,
|
||||||
|
.phase = PipelinePhase::Enrichment,
|
||||||
.message = std::format("Wikipedia: Cache hit for {}!", cache_key)});
|
.message = std::format("Wikipedia: Cache hit for {}!", cache_key)});
|
||||||
}
|
}
|
||||||
return cache_it->second;
|
return cache_it->second;
|
||||||
@@ -30,7 +32,8 @@ std::string WikipediaEnrichmentService::FetchExtract(std::string_view query) {
|
|||||||
const std::string encoded = this->client_->EncodeURL(cache_key);
|
const std::string encoded = this->client_->EncodeURL(cache_key);
|
||||||
const std::string url = std::format(
|
const std::string url = std::format(
|
||||||
"https://en.wikipedia.org/w/"
|
"https://en.wikipedia.org/w/"
|
||||||
"api.php?action=query&titles={}&prop=extracts&explaintext=1&format=json",
|
"api.php?action=query&titles={}&prop=extracts&explaintext=1&format="
|
||||||
|
"json",
|
||||||
encoded);
|
encoded);
|
||||||
|
|
||||||
const std::string body = this->client_->Get(url);
|
const std::string body = this->client_->Get(url);
|
||||||
@@ -45,10 +48,10 @@ std::string WikipediaEnrichmentService::FetchExtract(std::string_view query) {
|
|||||||
|
|
||||||
if (ec) {
|
if (ec) {
|
||||||
if (logger_) {
|
if (logger_) {
|
||||||
logger_->Log(
|
logger_->Log({.level = LogLevel::Warn,
|
||||||
{.level = LogLevel::Warn,
|
.phase = PipelinePhase::Enrichment,
|
||||||
.phase = PipelinePhase::UserGeneration,
|
.message = std::format(
|
||||||
.message = std::format("WikipediaService: JSON parse error for '{}': {}",
|
"WikipediaService: JSON parse error for '{}': {}",
|
||||||
std::string(query), ec.message())});
|
std::string(query), ec.message())});
|
||||||
}
|
}
|
||||||
return {};
|
return {};
|
||||||
@@ -58,11 +61,10 @@ std::string WikipediaEnrichmentService::FetchExtract(std::string_view query) {
|
|||||||
const json::object* obj = doc.if_object();
|
const json::object* obj = doc.if_object();
|
||||||
if (obj == nullptr) {
|
if (obj == nullptr) {
|
||||||
if (logger_) {
|
if (logger_) {
|
||||||
logger_->Log(
|
logger_->Log({.level = LogLevel::Warn,
|
||||||
{.level = LogLevel::Warn,
|
.phase = PipelinePhase::Enrichment,
|
||||||
.phase = PipelinePhase::UserGeneration,
|
.message = std::format(
|
||||||
.message =
|
"WikipediaService: Expected root object for '{}'",
|
||||||
std::format("WikipediaService: Expected root object for '{}'",
|
|
||||||
std::string(query))});
|
std::string(query))});
|
||||||
}
|
}
|
||||||
return {};
|
return {};
|
||||||
@@ -76,11 +78,10 @@ std::string WikipediaEnrichmentService::FetchExtract(std::string_view query) {
|
|||||||
|
|
||||||
if ((pages_ptr == nullptr) || !pages_ptr->is_object()) {
|
if ((pages_ptr == nullptr) || !pages_ptr->is_object()) {
|
||||||
if (logger_) {
|
if (logger_) {
|
||||||
logger_->Log(
|
logger_->Log({.level = LogLevel::Warn,
|
||||||
{.level = LogLevel::Warn,
|
.phase = PipelinePhase::Enrichment,
|
||||||
.phase = PipelinePhase::UserGeneration,
|
.message = std::format(
|
||||||
.message =
|
"WikipediaService: Missing query.pages for '{}'",
|
||||||
std::format("WikipediaService: Missing query.pages for '{}'",
|
|
||||||
std::string(query))});
|
std::string(query))});
|
||||||
}
|
}
|
||||||
return {};
|
return {};
|
||||||
@@ -90,10 +91,10 @@ std::string WikipediaEnrichmentService::FetchExtract(std::string_view query) {
|
|||||||
|
|
||||||
if (pages.empty()) {
|
if (pages.empty()) {
|
||||||
if (logger_) {
|
if (logger_) {
|
||||||
logger_->Log(
|
logger_->Log({.level = LogLevel::Warn,
|
||||||
{.level = LogLevel::Warn,
|
.phase = PipelinePhase::Enrichment,
|
||||||
.phase = PipelinePhase::UserGeneration,
|
.message = std::format(
|
||||||
.message = std::format("WikipediaService: No pages returned for '{}'",
|
"WikipediaService: No pages returned for '{}'",
|
||||||
std::string(query))});
|
std::string(query))});
|
||||||
}
|
}
|
||||||
this->extract_cache_.emplace(cache_key, "");
|
this->extract_cache_.emplace(cache_key, "");
|
||||||
@@ -106,11 +107,10 @@ std::string WikipediaEnrichmentService::FetchExtract(std::string_view query) {
|
|||||||
|
|
||||||
if (!page_val.is_object()) {
|
if (!page_val.is_object()) {
|
||||||
if (logger_) {
|
if (logger_) {
|
||||||
logger_->Log(
|
logger_->Log({.level = LogLevel::Warn,
|
||||||
{.level = LogLevel::Warn,
|
.phase = PipelinePhase::Enrichment,
|
||||||
.phase = PipelinePhase::UserGeneration,
|
.message = std::format(
|
||||||
.message =
|
"WikipediaService: Unexpected page format for '{}'",
|
||||||
std::format("WikipediaService: Unexpected page format for '{}'",
|
|
||||||
std::string(query))});
|
std::string(query))});
|
||||||
}
|
}
|
||||||
return {};
|
return {};
|
||||||
@@ -121,8 +121,9 @@ std::string WikipediaEnrichmentService::FetchExtract(std::string_view query) {
|
|||||||
// Handle 404/Missing status
|
// Handle 404/Missing status
|
||||||
if (page.contains("missing")) {
|
if (page.contains("missing")) {
|
||||||
if (logger_) {
|
if (logger_) {
|
||||||
logger_->Log({.level = LogLevel::Warn,
|
logger_->Log(
|
||||||
.phase = PipelinePhase::UserGeneration,
|
{.level = LogLevel::Warn,
|
||||||
|
.phase = PipelinePhase::Enrichment,
|
||||||
.message = std::format("WikipediaService: Page '{}' does not exist",
|
.message = std::format("WikipediaService: Page '{}' does not exist",
|
||||||
std::string(query))});
|
std::string(query))});
|
||||||
}
|
}
|
||||||
@@ -134,11 +135,10 @@ std::string WikipediaEnrichmentService::FetchExtract(std::string_view query) {
|
|||||||
|
|
||||||
if ((extract_ptr == nullptr) || !extract_ptr->is_string()) {
|
if ((extract_ptr == nullptr) || !extract_ptr->is_string()) {
|
||||||
if (logger_) {
|
if (logger_) {
|
||||||
logger_->Log(
|
logger_->Log({.level = LogLevel::Warn,
|
||||||
{.level = LogLevel::Warn,
|
.phase = PipelinePhase::Enrichment,
|
||||||
.phase = PipelinePhase::UserGeneration,
|
.message = std::format(
|
||||||
.message =
|
"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, "");
|
||||||
@@ -148,8 +148,9 @@ std::string WikipediaEnrichmentService::FetchExtract(std::string_view query) {
|
|||||||
// 4. Success
|
// 4. Success
|
||||||
std::string extract(extract_ptr->as_string());
|
std::string extract(extract_ptr->as_string());
|
||||||
if (logger_) {
|
if (logger_) {
|
||||||
logger_->Log({.level = LogLevel::Info,
|
logger_->Log(
|
||||||
.phase = PipelinePhase::UserGeneration,
|
{.level = LogLevel::Info,
|
||||||
|
.phase = PipelinePhase::Enrichment,
|
||||||
.message = std::format("WikipediaService: Fetched {} chars for '{}'",
|
.message = std::format("WikipediaService: Fetched {} chars for '{}'",
|
||||||
extract.size(), std::string(query))});
|
extract.size(), std::string(query))});
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,6 +1,6 @@
|
|||||||
/**
|
/**
|
||||||
* @file wikipedia/get_summary.cc
|
* @file wikipedia/get_summary.cc
|
||||||
* @brief WikipediaService::GetLocationContext() implementation.
|
* @brief WikipediaEnrichmentService::GetLocationContext() implementation.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#include <chrono>
|
#include <chrono>
|
||||||
@@ -11,12 +11,12 @@
|
|||||||
#include "services/enrichment/wikipedia_service.h"
|
#include "services/enrichment/wikipedia_service.h"
|
||||||
|
|
||||||
std::string WikipediaEnrichmentService::GetLocationContext(
|
std::string WikipediaEnrichmentService::GetLocationContext(
|
||||||
const Location& loc) {
|
const City& loc) {
|
||||||
using namespace std::literals::chrono_literals;
|
using namespace std::literals::chrono_literals;
|
||||||
if (!this->client_) {
|
if (!this->client_) {
|
||||||
if (logger_) {
|
if (logger_) {
|
||||||
logger_->Log({.level = LogLevel::Warn,
|
logger_->Log({.level = LogLevel::Warn,
|
||||||
.phase = PipelinePhase::UserGeneration,
|
.phase = PipelinePhase::Enrichment,
|
||||||
.message = "Wikipedia client is nullptr."});
|
.message = "Wikipedia client is nullptr."});
|
||||||
}
|
}
|
||||||
return {};
|
return {};
|
||||||
@@ -24,13 +24,6 @@ std::string WikipediaEnrichmentService::GetLocationContext(
|
|||||||
|
|
||||||
std::string result;
|
std::string result;
|
||||||
|
|
||||||
// std::string region_query(loc.city);
|
|
||||||
// if (!loc.country.empty()) {
|
|
||||||
// region_query += loc.state_province,
|
|
||||||
// region_query += ", ";
|
|
||||||
// region_query += loc.country;
|
|
||||||
// }
|
|
||||||
|
|
||||||
constexpr std::string_view brewing_query = "brewing";
|
constexpr std::string_view brewing_query = "brewing";
|
||||||
const std::string location_query =
|
const std::string location_query =
|
||||||
std::format("{}, {}", loc.city, loc.iso3166_2);
|
std::format("{}, {}", loc.city, loc.iso3166_2);
|
||||||
@@ -51,8 +44,9 @@ std::string WikipediaEnrichmentService::GetLocationContext(
|
|||||||
append_extract(FetchExtract(beer_query));
|
append_extract(FetchExtract(beer_query));
|
||||||
if (logger_) {
|
if (logger_) {
|
||||||
logger_->Log({.level = LogLevel::Info,
|
logger_->Log({.level = LogLevel::Info,
|
||||||
.phase = PipelinePhase::UserGeneration,
|
.phase = PipelinePhase::Enrichment,
|
||||||
.message = std::format("Done fetching for {}. Sleeping for 10 seconds.",
|
.message = std::format(
|
||||||
|
"Done fetching for {}. Sleeping for 10 seconds.",
|
||||||
location_query)});
|
location_query)});
|
||||||
}
|
}
|
||||||
std::this_thread::sleep_for(10s);
|
std::this_thread::sleep_for(10s);
|
||||||
@@ -61,7 +55,7 @@ std::string WikipediaEnrichmentService::GetLocationContext(
|
|||||||
if (logger_) {
|
if (logger_) {
|
||||||
logger_->Log(
|
logger_->Log(
|
||||||
{.level = LogLevel::Debug,
|
{.level = LogLevel::Debug,
|
||||||
.phase = PipelinePhase::UserGeneration,
|
.phase = PipelinePhase::Enrichment,
|
||||||
.message = std::format("WikipediaService lookup failed for '{}': {}",
|
.message = std::format("WikipediaService lookup failed for '{}': {}",
|
||||||
location_query, e.what())});
|
location_query, e.what())});
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,6 +1,6 @@
|
|||||||
/**
|
/**
|
||||||
* @file services/wikipedia/wikipedia_service.cc
|
* @file services/wikipedia/wikipedia_service.cc
|
||||||
* @brief WikipediaService constructor implementation.
|
* @brief WikipediaEnrichmentService constructor implementation.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#include "services/enrichment/wikipedia_service.h"
|
#include "services/enrichment/wikipedia_service.h"
|
||||||
|
|||||||
@@ -19,6 +19,8 @@ namespace {
|
|||||||
switch (phase) {
|
switch (phase) {
|
||||||
case PipelinePhase::Startup:
|
case PipelinePhase::Startup:
|
||||||
return "Startup";
|
return "Startup";
|
||||||
|
case PipelinePhase::Enrichment:
|
||||||
|
return "Enrichment";
|
||||||
case PipelinePhase::UserGeneration:
|
case PipelinePhase::UserGeneration:
|
||||||
return "User Generation";
|
return "User Generation";
|
||||||
case PipelinePhase::BreweryAndBeerGeneration:
|
case PipelinePhase::BreweryAndBeerGeneration:
|
||||||
|
|||||||
@@ -44,8 +44,8 @@ PromptDirectory::PromptDirectory(const std::filesystem::path& prompt_dir,
|
|||||||
// Scenario 4: directory must be readable (probe with directory_iterator).
|
// Scenario 4: directory must be readable (probe with directory_iterator).
|
||||||
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(std::format(
|
||||||
std::format("PromptDirectory: prompt directory is not readable: {} ({})",
|
"PromptDirectory: prompt directory is not readable: {} ({})",
|
||||||
prompt_dir_.string(), ec.message()));
|
prompt_dir_.string(), ec.message()));
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -84,14 +84,17 @@ std::string PromptDirectory::Load(std::string_view key) {
|
|||||||
file.close();
|
file.close();
|
||||||
|
|
||||||
if (content.empty()) {
|
if (content.empty()) {
|
||||||
throw std::runtime_error(std::format("PromptDirectory: prompt file for key '{}' is empty: {}",
|
throw std::runtime_error(
|
||||||
|
std::format("PromptDirectory: prompt file for key '{}' is empty: {}",
|
||||||
key_str, 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::format("[PromptDirectory] Loaded prompt '{}' from '{}' ({} chars)",
|
.message = std::format(
|
||||||
|
"[PromptDirectory] Loaded prompt '{}' from '{}' ({} chars)",
|
||||||
key_str, file_path.string(), content.size())});
|
key_str, file_path.string(), content.size())});
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -14,8 +14,11 @@ void SqliteExportService::Finalize() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
try {
|
try {
|
||||||
|
insert_user_address_stmt_.reset();
|
||||||
|
insert_user_stmt_.reset();
|
||||||
|
insert_brewery_address_stmt_.reset();
|
||||||
insert_brewery_stmt_.reset();
|
insert_brewery_stmt_.reset();
|
||||||
insert_location_stmt_.reset();
|
insert_city_stmt_.reset();
|
||||||
if (transaction_open_) {
|
if (transaction_open_) {
|
||||||
sqlite_export_service_internal::ExecSql(
|
sqlite_export_service_internal::ExecSql(
|
||||||
db_handle_, "COMMIT;", "Failed to commit SQLite transaction");
|
db_handle_, "COMMIT;", "Failed to commit SQLite transaction");
|
||||||
@@ -23,7 +26,7 @@ void SqliteExportService::Finalize() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
db_handle_.reset();
|
db_handle_.reset();
|
||||||
location_cache_.clear();
|
city_cache_.clear();
|
||||||
} catch (...) {
|
} catch (...) {
|
||||||
RollbackAndCloseNoThrow();
|
RollbackAndCloseNoThrow();
|
||||||
throw;
|
throw;
|
||||||
|
|||||||
@@ -33,7 +33,7 @@ void ResetStatement(SqliteStatementHandle& statement) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
void Bind(const SqliteStatementHandle& statement,
|
void Bind(const SqliteStatementHandle& statement,
|
||||||
const BindParam<std::string_view>& param) {
|
const BoundParam<std::string_view>& param) {
|
||||||
const auto byte_count = param.value.size();
|
const auto byte_count = param.value.size();
|
||||||
if (byte_count > static_cast<std::size_t>(std::numeric_limits<int>::max())) {
|
if (byte_count > static_cast<std::size_t>(std::numeric_limits<int>::max())) {
|
||||||
ThrowSqliteError(sqlite3_db_handle(statement.get()), param.action);
|
ThrowSqliteError(sqlite3_db_handle(statement.get()), param.action);
|
||||||
@@ -60,7 +60,7 @@ void Bind(const SqliteStatementHandle& statement,
|
|||||||
}
|
}
|
||||||
|
|
||||||
void Bind(const SqliteStatementHandle& statement,
|
void Bind(const SqliteStatementHandle& statement,
|
||||||
const BindParam<double>& param) {
|
const BoundParam<double>& param) {
|
||||||
if (sqlite3_bind_double(statement.get(), param.index, param.value) !=
|
if (sqlite3_bind_double(statement.get(), param.index, param.value) !=
|
||||||
SQLITE_OK) {
|
SQLITE_OK) {
|
||||||
ThrowSqliteError(sqlite3_db_handle(statement.get()), param.action);
|
ThrowSqliteError(sqlite3_db_handle(statement.get()), param.action);
|
||||||
@@ -68,7 +68,7 @@ void Bind(const SqliteStatementHandle& statement,
|
|||||||
}
|
}
|
||||||
|
|
||||||
void Bind(const SqliteStatementHandle& statement,
|
void Bind(const SqliteStatementHandle& statement,
|
||||||
const BindParam<sqlite3_int64>& param) {
|
const BoundParam<sqlite3_int64>& param) {
|
||||||
if (sqlite3_bind_int64(statement.get(), param.index, param.value) !=
|
if (sqlite3_bind_int64(statement.get(), param.index, param.value) !=
|
||||||
SQLITE_OK) {
|
SQLITE_OK) {
|
||||||
ThrowSqliteError(sqlite3_db_handle(statement.get()), param.action);
|
ThrowSqliteError(sqlite3_db_handle(statement.get()), param.action);
|
||||||
|
|||||||
@@ -18,8 +18,8 @@ std::filesystem::path SqliteExportService::BuildDatabasePath() const {
|
|||||||
std::filesystem::path candidate = output_path_ / base_filename;
|
std::filesystem::path candidate = output_path_ / base_filename;
|
||||||
|
|
||||||
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(
|
||||||
std::filesystem::path(std::format("biergarten_seed_{}-{}.sqlite",
|
std::format("biergarten_seed_{}-{}.sqlite",
|
||||||
run_timestamp_utc_, suffix));
|
run_timestamp_utc_, suffix));
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -28,20 +28,39 @@ std::filesystem::path SqliteExportService::BuildDatabasePath() const {
|
|||||||
|
|
||||||
void SqliteExportService::InitializeSchema() const {
|
void SqliteExportService::InitializeSchema() const {
|
||||||
sqlite_export_service_internal::ExecSql(
|
sqlite_export_service_internal::ExecSql(
|
||||||
db_handle_, sqlite_export_service_internal::kCreateLocationsTableSql,
|
db_handle_, sqlite_export_service_internal::kCreateCitiesTableSql,
|
||||||
"Failed to create SQLite locations table");
|
"Failed to create SQLite cities table");
|
||||||
sqlite_export_service_internal::ExecSql(
|
sqlite_export_service_internal::ExecSql(
|
||||||
db_handle_, sqlite_export_service_internal::kCreateBreweriesTableSql,
|
db_handle_, sqlite_export_service_internal::kCreateBreweriesTableSql,
|
||||||
"Failed to create SQLite breweries table");
|
"Failed to create SQLite breweries table");
|
||||||
|
sqlite_export_service_internal::ExecSql(
|
||||||
|
db_handle_, sqlite_export_service_internal::kCreateUsersTableSql,
|
||||||
|
"Failed to create SQLite users table");
|
||||||
|
sqlite_export_service_internal::ExecSql(
|
||||||
|
db_handle_, sqlite_export_service_internal::kCreateBreweryAddressTableSql,
|
||||||
|
"Failed to create SQLite brewery addresses table");
|
||||||
|
sqlite_export_service_internal::ExecSql(
|
||||||
|
db_handle_, sqlite_export_service_internal::kCreateUserAddressTableSql,
|
||||||
|
"Failed to create SQLite user addresses table");
|
||||||
}
|
}
|
||||||
|
|
||||||
void SqliteExportService::PrepareStatements() {
|
void SqliteExportService::PrepareStatements() {
|
||||||
insert_location_stmt_ = sqlite_export_service_internal::PrepareStatement(
|
insert_city_stmt_ = sqlite_export_service_internal::PrepareStatement(
|
||||||
db_handle_, sqlite_export_service_internal::kInsertLocationSql,
|
db_handle_, sqlite_export_service_internal::kInsertCitySql,
|
||||||
"Failed to prepare SQLite location insert statement");
|
"Failed to prepare SQLite city insert statement");
|
||||||
insert_brewery_stmt_ = sqlite_export_service_internal::PrepareStatement(
|
insert_brewery_stmt_ = sqlite_export_service_internal::PrepareStatement(
|
||||||
db_handle_, sqlite_export_service_internal::kInsertBrewerySql,
|
db_handle_, sqlite_export_service_internal::kInsertBrewerySql,
|
||||||
"Failed to prepare SQLite brewery insert statement");
|
"Failed to prepare SQLite brewery insert statement");
|
||||||
|
insert_brewery_address_stmt_ =
|
||||||
|
sqlite_export_service_internal::PrepareStatement(
|
||||||
|
db_handle_, sqlite_export_service_internal::kInsertBreweryAddressSql,
|
||||||
|
"Failed to prepare SQLite brewery address insert statement");
|
||||||
|
insert_user_stmt_ = sqlite_export_service_internal::PrepareStatement(
|
||||||
|
db_handle_, sqlite_export_service_internal::kInsertUserSql,
|
||||||
|
"Failed to prepare SQLite user insert statement");
|
||||||
|
insert_user_address_stmt_ = sqlite_export_service_internal::PrepareStatement(
|
||||||
|
db_handle_, sqlite_export_service_internal::kInsertUserAddressSql,
|
||||||
|
"Failed to prepare SQLite user address insert statement");
|
||||||
}
|
}
|
||||||
|
|
||||||
void SqliteExportService::RollbackAndCloseNoThrow() noexcept {
|
void SqliteExportService::RollbackAndCloseNoThrow() noexcept {
|
||||||
@@ -54,10 +73,13 @@ void SqliteExportService::RollbackAndCloseNoThrow() noexcept {
|
|||||||
transaction_open_ = false;
|
transaction_open_ = false;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
insert_user_address_stmt_.reset();
|
||||||
|
insert_user_stmt_.reset();
|
||||||
|
insert_brewery_address_stmt_.reset();
|
||||||
insert_brewery_stmt_.reset();
|
insert_brewery_stmt_.reset();
|
||||||
insert_location_stmt_.reset();
|
insert_city_stmt_.reset();
|
||||||
db_handle_.reset();
|
db_handle_.reset();
|
||||||
location_cache_.clear();
|
city_cache_.clear();
|
||||||
}
|
}
|
||||||
|
|
||||||
void SqliteExportService::Initialize() {
|
void SqliteExportService::Initialize() {
|
||||||
|
|||||||
@@ -1,9 +1,9 @@
|
|||||||
/**
|
/**
|
||||||
* @file services/sqlite/process_record.cc
|
* @file services/sqlite/process_record.cc
|
||||||
* @brief SqliteExportService::ProcessRecord() implementation.
|
* @brief SqliteExportService::ProcessRecord() implementation
|
||||||
|
* and the shared city-resolution helper.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#include <iomanip>
|
|
||||||
#include <sstream>
|
#include <sstream>
|
||||||
#include <stdexcept>
|
#include <stdexcept>
|
||||||
#include <string>
|
#include <string>
|
||||||
@@ -11,129 +11,115 @@
|
|||||||
#include "services/database/sqlite_export_service.h"
|
#include "services/database/sqlite_export_service.h"
|
||||||
#include "services/database/sqlite_export_service_helpers.h"
|
#include "services/database/sqlite_export_service_helpers.h"
|
||||||
|
|
||||||
constexpr int kLocationPrecision = 17;
|
std::string SqliteExportService::BuildCityKey(const City& location) {
|
||||||
|
|
||||||
std::string SqliteExportService::BuildLocationKey(const Location& location) {
|
|
||||||
std::ostringstream key_stream;
|
std::ostringstream key_stream;
|
||||||
key_stream << location.city << '\n'
|
key_stream << location.city << '\n'
|
||||||
<< location.state_province << '\n'
|
<< location.state_province << '\n'
|
||||||
<< location.iso3166_2 << '\n'
|
<< location.iso3166_2 << '\n'
|
||||||
<< location.country << '\n'
|
<< location.country << '\n'
|
||||||
<< location.iso3166_1 << '\n'
|
<< location.iso3166_1 << '\n'
|
||||||
<< std::setprecision(kLocationPrecision) << location.latitude
|
|
||||||
<< '\n'
|
|
||||||
<< std::setprecision(kLocationPrecision) << location.longitude
|
|
||||||
<< '\n'
|
|
||||||
<< sqlite_export_service_internal::SerializeVector(
|
<< sqlite_export_service_internal::SerializeVector(
|
||||||
location.local_languages);
|
location.local_languages);
|
||||||
return key_stream.str();
|
return key_stream.str();
|
||||||
}
|
}
|
||||||
|
|
||||||
uint64_t SqliteExportService::ProcessRecord(const GeneratedBrewery& brewery) {
|
sqlite3_int64 SqliteExportService::ResolveCityId(const City& location) {
|
||||||
|
const std::string city_key = BuildCityKey(location);
|
||||||
|
const auto cached_city = city_cache_.find(city_key);
|
||||||
|
if (cached_city != city_cache_.end()) {
|
||||||
|
return cached_city->second;
|
||||||
|
}
|
||||||
|
|
||||||
|
const std::string local_languages_json =
|
||||||
|
sqlite_export_service_internal::SerializeVector(location.local_languages);
|
||||||
|
|
||||||
|
sqlite_export_service_internal::Bind(
|
||||||
|
insert_city_stmt_,
|
||||||
|
sqlite_export_service_internal::BoundParam<std::string_view>{
|
||||||
|
.index = sqlite_export_service_internal::kCityNameBindIndex,
|
||||||
|
.value = location.city,
|
||||||
|
.action = "Failed to bind SQLite city name"});
|
||||||
|
sqlite_export_service_internal::Bind(
|
||||||
|
insert_city_stmt_,
|
||||||
|
sqlite_export_service_internal::BoundParam<std::string_view>{
|
||||||
|
.index = sqlite_export_service_internal::kCityStateProvinceBindIndex,
|
||||||
|
.value = location.state_province,
|
||||||
|
.action = "Failed to bind SQLite city state/province"});
|
||||||
|
sqlite_export_service_internal::Bind(
|
||||||
|
insert_city_stmt_,
|
||||||
|
sqlite_export_service_internal::BoundParam<std::string_view>{
|
||||||
|
.index = sqlite_export_service_internal::kCityIso31662BindIndex,
|
||||||
|
.value = location.iso3166_2,
|
||||||
|
.action = "Failed to bind SQLite city ISO 3166-2 code"});
|
||||||
|
sqlite_export_service_internal::Bind(
|
||||||
|
insert_city_stmt_,
|
||||||
|
sqlite_export_service_internal::BoundParam<std::string_view>{
|
||||||
|
.index = sqlite_export_service_internal::kCityCountryBindIndex,
|
||||||
|
.value = location.country,
|
||||||
|
.action = "Failed to bind SQLite city country"});
|
||||||
|
sqlite_export_service_internal::Bind(
|
||||||
|
insert_city_stmt_,
|
||||||
|
sqlite_export_service_internal::BoundParam<std::string_view>{
|
||||||
|
.index = sqlite_export_service_internal::kCityIso31661BindIndex,
|
||||||
|
.value = location.iso3166_1,
|
||||||
|
.action = "Failed to bind SQLite city ISO 3166-1 code"});
|
||||||
|
sqlite_export_service_internal::Bind(
|
||||||
|
insert_city_stmt_,
|
||||||
|
sqlite_export_service_internal::BoundParam<std::string_view>{
|
||||||
|
.index = sqlite_export_service_internal::kCityLanguagesBindIndex,
|
||||||
|
.value = local_languages_json,
|
||||||
|
.action = "Failed to bind SQLite city languages"});
|
||||||
|
|
||||||
|
sqlite_export_service_internal::StepStatement(
|
||||||
|
db_handle_, insert_city_stmt_, "Failed to insert SQLite city row");
|
||||||
|
|
||||||
|
const sqlite3_int64 city_id =
|
||||||
|
sqlite_export_service_internal::LastInsertRowId(db_handle_);
|
||||||
|
city_cache_.emplace(city_key, city_id);
|
||||||
|
sqlite_export_service_internal::ResetStatement(insert_city_stmt_);
|
||||||
|
|
||||||
|
return city_id;
|
||||||
|
}
|
||||||
|
|
||||||
|
uint64_t SqliteExportService::ProcessRecord(const BreweryRecord& brewery) {
|
||||||
if (db_handle_ == nullptr || !transaction_open_) {
|
if (db_handle_ == nullptr || !transaction_open_) {
|
||||||
throw std::runtime_error("SQLite export service is not initialized");
|
throw std::runtime_error("SQLite export service is not initialized");
|
||||||
}
|
}
|
||||||
|
|
||||||
const std::string location_key = BuildLocationKey(brewery.location);
|
const sqlite3_int64 city_id = ResolveCityId(brewery.address.city);
|
||||||
const auto cached_location = location_cache_.find(location_key);
|
|
||||||
sqlite3_int64 location_id = 0;
|
|
||||||
|
|
||||||
if (cached_location != location_cache_.end()) {
|
|
||||||
location_id = cached_location->second;
|
|
||||||
} else {
|
|
||||||
const std::string local_languages_json =
|
|
||||||
sqlite_export_service_internal::SerializeVector(
|
|
||||||
brewery.location.local_languages);
|
|
||||||
|
|
||||||
sqlite_export_service_internal::Bind(
|
|
||||||
insert_location_stmt_,
|
|
||||||
sqlite_export_service_internal::BindParam<std::string_view>{
|
|
||||||
.index = sqlite_export_service_internal::kLocationCityBindIndex,
|
|
||||||
.value = brewery.location.city,
|
|
||||||
.action = "Failed to bind SQLite location city"});
|
|
||||||
sqlite_export_service_internal::Bind(
|
|
||||||
insert_location_stmt_,
|
|
||||||
sqlite_export_service_internal::BindParam<std::string_view>{
|
|
||||||
.index =
|
|
||||||
sqlite_export_service_internal::kLocationStateProvinceBindIndex,
|
|
||||||
.value = brewery.location.state_province,
|
|
||||||
.action = "Failed to bind SQLite location state/province"});
|
|
||||||
sqlite_export_service_internal::Bind(
|
|
||||||
insert_location_stmt_,
|
|
||||||
sqlite_export_service_internal::BindParam<std::string_view>{
|
|
||||||
.index = sqlite_export_service_internal::kLocationIso31662BindIndex,
|
|
||||||
.value = brewery.location.iso3166_2,
|
|
||||||
.action = "Failed to bind SQLite location ISO 3166-2 code"});
|
|
||||||
sqlite_export_service_internal::Bind(
|
|
||||||
insert_location_stmt_,
|
|
||||||
sqlite_export_service_internal::BindParam<std::string_view>{
|
|
||||||
.index = sqlite_export_service_internal::kLocationCountryBindIndex,
|
|
||||||
.value = brewery.location.country,
|
|
||||||
.action = "Failed to bind SQLite location country"});
|
|
||||||
sqlite_export_service_internal::Bind(
|
|
||||||
insert_location_stmt_,
|
|
||||||
sqlite_export_service_internal::BindParam<std::string_view>{
|
|
||||||
.index = sqlite_export_service_internal::kLocationIso31661BindIndex,
|
|
||||||
.value = brewery.location.iso3166_1,
|
|
||||||
.action = "Failed to bind SQLite location ISO 3166-1 code"});
|
|
||||||
sqlite_export_service_internal::Bind(
|
|
||||||
insert_location_stmt_,
|
|
||||||
sqlite_export_service_internal::BindParam<std::string_view>{
|
|
||||||
.index =
|
|
||||||
sqlite_export_service_internal::kLocationLanguagesBindIndex,
|
|
||||||
.value = local_languages_json,
|
|
||||||
.action = "Failed to bind SQLite location languages"});
|
|
||||||
sqlite_export_service_internal::Bind(
|
|
||||||
insert_location_stmt_,
|
|
||||||
sqlite_export_service_internal::BindParam{
|
|
||||||
.index = sqlite_export_service_internal::kLocationLatitudeBindIndex,
|
|
||||||
.value = brewery.location.latitude,
|
|
||||||
.action = "Failed to bind SQLite location latitude"});
|
|
||||||
sqlite_export_service_internal::Bind(
|
|
||||||
insert_location_stmt_,
|
|
||||||
sqlite_export_service_internal::BindParam{
|
|
||||||
.index =
|
|
||||||
sqlite_export_service_internal::kLocationLongitudeBindIndex,
|
|
||||||
.value = brewery.location.longitude,
|
|
||||||
.action = "Failed to bind SQLite location longitude"});
|
|
||||||
|
|
||||||
sqlite_export_service_internal::StepStatement(
|
|
||||||
db_handle_, insert_location_stmt_,
|
|
||||||
"Failed to insert SQLite location row");
|
|
||||||
|
|
||||||
location_id = sqlite_export_service_internal::LastInsertRowId(db_handle_);
|
|
||||||
location_cache_.emplace(location_key, location_id);
|
|
||||||
sqlite_export_service_internal::ResetStatement(insert_location_stmt_);
|
|
||||||
}
|
|
||||||
|
|
||||||
sqlite_export_service_internal::Bind(
|
sqlite_export_service_internal::Bind(
|
||||||
insert_brewery_stmt_,
|
insert_brewery_stmt_,
|
||||||
sqlite_export_service_internal::BindParam<sqlite3_int64>{
|
sqlite_export_service_internal::BoundParam<sqlite3_int64>{
|
||||||
.index = sqlite_export_service_internal::kBreweryLocationIdBindIndex,
|
.index = sqlite_export_service_internal::kBreweryCityIdBindIndex,
|
||||||
.value = location_id,
|
.value = city_id,
|
||||||
.action = "Failed to bind SQLite brewery location id"});
|
.action = "Failed to bind SQLite brewery city id"});
|
||||||
|
|
||||||
sqlite_export_service_internal::Bind(
|
sqlite_export_service_internal::Bind(
|
||||||
insert_brewery_stmt_,
|
insert_brewery_stmt_,
|
||||||
sqlite_export_service_internal::BindParam<std::string_view>{
|
sqlite_export_service_internal::BoundParam<std::string_view>{
|
||||||
.index = sqlite_export_service_internal::kBreweryEnglishNameBindIndex,
|
.index = sqlite_export_service_internal::kBreweryEnglishNameBindIndex,
|
||||||
.value = brewery.brewery.name_en,
|
.value = brewery.brewery.name_en,
|
||||||
.action = "Failed to bind SQLite brewery English name"});
|
.action = "Failed to bind SQLite brewery English name"});
|
||||||
|
|
||||||
sqlite_export_service_internal::Bind(
|
sqlite_export_service_internal::Bind(
|
||||||
insert_brewery_stmt_,
|
insert_brewery_stmt_,
|
||||||
sqlite_export_service_internal::BindParam<std::string_view>{
|
sqlite_export_service_internal::BoundParam<std::string_view>{
|
||||||
.index = sqlite_export_service_internal::
|
.index = sqlite_export_service_internal::
|
||||||
kBreweryEnglishDescriptionBindIndex,
|
kBreweryEnglishDescriptionBindIndex,
|
||||||
.value = brewery.brewery.description_en,
|
.value = brewery.brewery.description_en,
|
||||||
.action = "Failed to bind SQLite brewery English description"});
|
.action = "Failed to bind SQLite brewery English description"});
|
||||||
|
|
||||||
sqlite_export_service_internal::Bind(
|
sqlite_export_service_internal::Bind(
|
||||||
insert_brewery_stmt_,
|
insert_brewery_stmt_,
|
||||||
sqlite_export_service_internal::BindParam<std::string_view>{
|
sqlite_export_service_internal::BoundParam<std::string_view>{
|
||||||
.index = sqlite_export_service_internal::kBreweryLocalNameBindIndex,
|
.index = sqlite_export_service_internal::kBreweryLocalNameBindIndex,
|
||||||
.value = brewery.brewery.name_local,
|
.value = brewery.brewery.name_local,
|
||||||
.action = "Failed to bind SQLite brewery local name"});
|
.action = "Failed to bind SQLite brewery local name"});
|
||||||
|
|
||||||
sqlite_export_service_internal::Bind(
|
sqlite_export_service_internal::Bind(
|
||||||
insert_brewery_stmt_,
|
insert_brewery_stmt_,
|
||||||
sqlite_export_service_internal::BindParam<std::string_view>{
|
sqlite_export_service_internal::BoundParam<std::string_view>{
|
||||||
.index =
|
.index =
|
||||||
sqlite_export_service_internal::kBreweryLocalDescriptionBindIndex,
|
sqlite_export_service_internal::kBreweryLocalDescriptionBindIndex,
|
||||||
.value = brewery.brewery.description_local,
|
.value = brewery.brewery.description_local,
|
||||||
@@ -142,7 +128,37 @@ uint64_t SqliteExportService::ProcessRecord(const GeneratedBrewery& brewery) {
|
|||||||
sqlite_export_service_internal::StepStatement(
|
sqlite_export_service_internal::StepStatement(
|
||||||
db_handle_, insert_brewery_stmt_, "Failed to insert SQLite brewery row");
|
db_handle_, insert_brewery_stmt_, "Failed to insert SQLite brewery row");
|
||||||
|
|
||||||
|
const sqlite3_int64 brewery_id =
|
||||||
|
sqlite_export_service_internal::LastInsertRowId(db_handle_);
|
||||||
sqlite_export_service_internal::ResetStatement(insert_brewery_stmt_);
|
sqlite_export_service_internal::ResetStatement(insert_brewery_stmt_);
|
||||||
|
|
||||||
return sqlite_export_service_internal::LastInsertRowId(db_handle_);
|
sqlite_export_service_internal::Bind(
|
||||||
|
insert_brewery_address_stmt_,
|
||||||
|
sqlite_export_service_internal::BoundParam<sqlite3_int64>{
|
||||||
|
.index =
|
||||||
|
sqlite_export_service_internal::kBreweryAddressBreweryIdBindIndex,
|
||||||
|
.value = brewery_id,
|
||||||
|
.action = "Failed to bind SQLite brewery address brewery id"});
|
||||||
|
sqlite_export_service_internal::Bind(
|
||||||
|
insert_brewery_address_stmt_,
|
||||||
|
sqlite_export_service_internal::BoundParam<std::string_view>{
|
||||||
|
.index = sqlite_export_service_internal::
|
||||||
|
kBreweryAddressPostalCodeBindIndex,
|
||||||
|
.value = brewery.address.postal_code,
|
||||||
|
.action = "Failed to bind SQLite brewery address postal code"});
|
||||||
|
sqlite_export_service_internal::Bind(
|
||||||
|
insert_brewery_address_stmt_,
|
||||||
|
sqlite_export_service_internal::BoundParam<sqlite3_int64>{
|
||||||
|
.index =
|
||||||
|
sqlite_export_service_internal::kBreweryAddressCityIdBindIndex,
|
||||||
|
.value = city_id,
|
||||||
|
.action = "Failed to bind SQLite brewery address city id"});
|
||||||
|
|
||||||
|
sqlite_export_service_internal::StepStatement(
|
||||||
|
db_handle_, insert_brewery_address_stmt_,
|
||||||
|
"Failed to insert SQLite brewery address row");
|
||||||
|
|
||||||
|
sqlite_export_service_internal::ResetStatement(insert_brewery_address_stmt_);
|
||||||
|
|
||||||
|
return brewery_id;
|
||||||
}
|
}
|
||||||
|
|||||||
94
tooling/pipeline/src/services/sqlite/process_user_record.cc
Normal file
@@ -0,0 +1,94 @@
|
|||||||
|
/**
|
||||||
|
* @file services/sqlite/process_user_record.cc
|
||||||
|
* @brief SqliteExportService::ProcessRecord(UserRecord) implementation.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include <stdexcept>
|
||||||
|
|
||||||
|
#include "services/database/sqlite_export_service.h"
|
||||||
|
#include "services/database/sqlite_export_service_helpers.h"
|
||||||
|
|
||||||
|
uint64_t SqliteExportService::ProcessRecord(const UserRecord& user) {
|
||||||
|
if (db_handle_ == nullptr || !transaction_open_) {
|
||||||
|
throw std::runtime_error("SQLite export service is not initialized");
|
||||||
|
}
|
||||||
|
|
||||||
|
const sqlite3_int64 city_id = ResolveCityId(user.address.city);
|
||||||
|
|
||||||
|
sqlite_export_service_internal::Bind(
|
||||||
|
insert_user_stmt_,
|
||||||
|
sqlite_export_service_internal::BoundParam<std::string_view>{
|
||||||
|
.index = sqlite_export_service_internal::kUserFirstNameBindIndex,
|
||||||
|
.value = user.user.first_name,
|
||||||
|
.action = "Failed to bind SQLite user first name"});
|
||||||
|
sqlite_export_service_internal::Bind(
|
||||||
|
insert_user_stmt_,
|
||||||
|
sqlite_export_service_internal::BoundParam<std::string_view>{
|
||||||
|
.index = sqlite_export_service_internal::kUserLastNameBindIndex,
|
||||||
|
.value = user.user.last_name,
|
||||||
|
.action = "Failed to bind SQLite user last name"});
|
||||||
|
sqlite_export_service_internal::Bind(
|
||||||
|
insert_user_stmt_,
|
||||||
|
sqlite_export_service_internal::BoundParam<std::string_view>{
|
||||||
|
.index = sqlite_export_service_internal::kUserGenderBindIndex,
|
||||||
|
.value = user.user.gender,
|
||||||
|
.action = "Failed to bind SQLite user gender"});
|
||||||
|
sqlite_export_service_internal::Bind(
|
||||||
|
insert_user_stmt_,
|
||||||
|
sqlite_export_service_internal::BoundParam<std::string_view>{
|
||||||
|
.index = sqlite_export_service_internal::kUserUsernameBindIndex,
|
||||||
|
.value = user.user.username,
|
||||||
|
.action = "Failed to bind SQLite user username"});
|
||||||
|
sqlite_export_service_internal::Bind(
|
||||||
|
insert_user_stmt_,
|
||||||
|
sqlite_export_service_internal::BoundParam<std::string_view>{
|
||||||
|
.index = sqlite_export_service_internal::kUserBioBindIndex,
|
||||||
|
.value = user.user.bio,
|
||||||
|
.action = "Failed to bind SQLite user bio"});
|
||||||
|
sqlite_export_service_internal::Bind(
|
||||||
|
insert_user_stmt_,
|
||||||
|
sqlite_export_service_internal::BoundParam<double>{
|
||||||
|
.index = sqlite_export_service_internal::kUserActivityWeightBindIndex,
|
||||||
|
.value = static_cast<double>(user.user.activity_weight),
|
||||||
|
.action = "Failed to bind SQLite user activity weight"});
|
||||||
|
sqlite_export_service_internal::Bind(
|
||||||
|
insert_user_stmt_,
|
||||||
|
sqlite_export_service_internal::BoundParam<std::string_view>{
|
||||||
|
.index = sqlite_export_service_internal::kUserEmailBindIndex,
|
||||||
|
.value = user.email,
|
||||||
|
.action = "Failed to bind SQLite user email"});
|
||||||
|
sqlite_export_service_internal::Bind(
|
||||||
|
insert_user_stmt_,
|
||||||
|
sqlite_export_service_internal::BoundParam<std::string_view>{
|
||||||
|
.index = sqlite_export_service_internal::kUserDateOfBirthBindIndex,
|
||||||
|
.value = user.date_of_birth,
|
||||||
|
.action = "Failed to bind SQLite user date of birth"});
|
||||||
|
|
||||||
|
sqlite_export_service_internal::StepStatement(
|
||||||
|
db_handle_, insert_user_stmt_, "Failed to insert SQLite user row");
|
||||||
|
|
||||||
|
const sqlite3_int64 user_id =
|
||||||
|
sqlite_export_service_internal::LastInsertRowId(db_handle_);
|
||||||
|
sqlite_export_service_internal::ResetStatement(insert_user_stmt_);
|
||||||
|
|
||||||
|
sqlite_export_service_internal::Bind(
|
||||||
|
insert_user_address_stmt_,
|
||||||
|
sqlite_export_service_internal::BoundParam<sqlite3_int64>{
|
||||||
|
.index = sqlite_export_service_internal::kUserAddressUserIdBindIndex,
|
||||||
|
.value = user_id,
|
||||||
|
.action = "Failed to bind SQLite user address user id"});
|
||||||
|
sqlite_export_service_internal::Bind(
|
||||||
|
insert_user_address_stmt_,
|
||||||
|
sqlite_export_service_internal::BoundParam<sqlite3_int64>{
|
||||||
|
.index = sqlite_export_service_internal::kUserAddressCityIdBindIndex,
|
||||||
|
.value = city_id,
|
||||||
|
.action = "Failed to bind SQLite user address city id"});
|
||||||
|
|
||||||
|
sqlite_export_service_internal::StepStatement(
|
||||||
|
db_handle_, insert_user_address_stmt_,
|
||||||
|
"Failed to insert SQLite user address row");
|
||||||
|
|
||||||
|
sqlite_export_service_internal::ResetStatement(insert_user_address_stmt_);
|
||||||
|
|
||||||
|
return user_id;
|
||||||
|
}
|
||||||
@@ -48,18 +48,17 @@ 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(std::format(
|
throw std::runtime_error(
|
||||||
"[HttpWebClient] Request failed for URL: {} — {}", url,
|
std::format("[HttpWebClient] Request failed for URL: {} — {}", url,
|
||||||
httplib::to_string(result.error())));
|
httplib::to_string(result.error())));
|
||||||
}
|
}
|
||||||
|
|
||||||
if (result->status < kSuccessMin || result->status >= kSuccessMax) {
|
if (result->status < kSuccessMin || result->status >= kSuccessMax) {
|
||||||
if (logger_) {
|
if (logger_) {
|
||||||
logger_->Log(
|
logger_->Log({.level = LogLevel::Error,
|
||||||
{.level = LogLevel::Error,
|
.phase = PipelinePhase::Enrichment,
|
||||||
.phase = PipelinePhase::UserGeneration,
|
.message = std::format(
|
||||||
.message =
|
"[HttpWebClient] Request failed for URL: {}", url)});
|
||||||
std::format("[HttpWebClient] Request failed for URL: {}", url)});
|
|
||||||
}
|
}
|
||||||
throw std::runtime_error(std::format("[HttpWebClient] HTTP {} for URL: {}",
|
throw std::runtime_error(std::format("[HttpWebClient] HTTP {} for URL: {}",
|
||||||
result->status, url));
|
result->status, url));
|
||||||
|
|||||||