From 5a8a9f465264d31983507b23f656469cc39178bb Mon Sep 17 00:00:00 2001
From: Aaron Po <apo2@uwo.ca>
Date: Fri, 17 Jul 2026 02:34:49 -0400
Subject: [PATCH] updates

---
 tooling/pipeline/includes/biergarten_pipeline_orchestrator.h    | 2 +-
 tooling/pipeline/includes/data_model/models.h                   | 1 +
 .../query_cities_with_countries.cc                              | 2 +-
 tooling/pipeline/src/biergarten_pipeline_orchestrator/run.cc    | 2 +-
 4 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/tooling/pipeline/includes/biergarten_pipeline_orchestrator.h b/tooling/pipeline/includes/biergarten_pipeline_orchestrator.h
index c65219a..e731e6e 100644
--- a/tooling/pipeline/includes/biergarten_pipeline_orchestrator.h
+++ b/tooling/pipeline/includes/biergarten_pipeline_orchestrator.h
@@ -87,7 +87,7 @@ class BiergartenPipelineOrchestrator {
    * @return Vector of locations randomly sampled per
    * PipelineOptions::location_count.
    */
-  std::vector<Location> QueryCitiesWithCountries();
+  std::vector<Location> QueryLocations();
 
   /**
    * @brief Generate breweries for enriched cities.
diff --git a/tooling/pipeline/includes/data_model/models.h b/tooling/pipeline/includes/data_model/models.h
index 8ed7355..4c46a96 100644
--- a/tooling/pipeline/includes/data_model/models.h
+++ b/tooling/pipeline/includes/data_model/models.h
@@ -31,6 +31,7 @@ namespace prog_opts = boost::program_options;
 struct Location {
   std::string city{};
   std::string state_province{};
+  std::string postal_code{};
 
   /**
    * @brief ISO 3166-2 subdivision code.
diff --git a/tooling/pipeline/src/biergarten_pipeline_orchestrator/query_cities_with_countries.cc b/tooling/pipeline/src/biergarten_pipeline_orchestrator/query_cities_with_countries.cc
index f3549dd..edea084 100644
--- a/tooling/pipeline/src/biergarten_pipeline_orchestrator/query_cities_with_countries.cc
+++ b/tooling/pipeline/src/biergarten_pipeline_orchestrator/query_cities_with_countries.cc
@@ -15,7 +15,7 @@
 #include "services/logging/logger.h"
 
 std::vector<Location>
-BiergartenPipelineOrchestrator::QueryCitiesWithCountries() {
+BiergartenPipelineOrchestrator::QueryLocations() {
   logger_->Log({.level = LogLevel::Info,
                 .phase = PipelinePhase::Startup,
                 .message = "=== GEOGRAPHIC DATA OVERVIEW ==="});
diff --git a/tooling/pipeline/src/biergarten_pipeline_orchestrator/run.cc b/tooling/pipeline/src/biergarten_pipeline_orchestrator/run.cc
index 5ada0f6..19e42f0 100644
--- a/tooling/pipeline/src/biergarten_pipeline_orchestrator/run.cc
+++ b/tooling/pipeline/src/biergarten_pipeline_orchestrator/run.cc
@@ -14,7 +14,7 @@ bool BiergartenPipelineOrchestrator::Run() {
   try {
     exporter_->Initialize();
 
-    std::vector<Location> cities = QueryCitiesWithCountries();
+    std::vector<Location> cities = QueryLocations();
     std::vector<EnrichedCity> enriched;
     enriched.reserve(cities.size());