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

Enhance brewery generation: add country name parameter and improve prompt handling

Browse Source
This commit is contained in:
Aaron Po
2026-04-02 01:04:41 -04:00
parent 280c9c61bd
commit ac136f7179
6 changed files with 357 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
pipeline/includes/data_generator.h
View File

@@ -19,6 +19,7 @@ public:
virtual void load(const std::string &modelPath) = 0;
virtual BreweryResult generateBrewery(const std::string &cityName,
const std::string &countryName,
const std::string &regionContext) = 0;
virtual UserResult generateUser(const std::string &locale) = 0;

9
pipeline/includes/llama_generator.h
View File

@@ -13,11 +13,18 @@ public:
void load(const std::string &modelPath) override;
BreweryResult generateBrewery(const std::string &cityName,
const std::string &countryName,
const std::string &regionContext) override;
UserResult generateUser(const std::string &locale) override;
private:
std::string infer(const std::string &prompt, int maxTokens = 256);
std::string infer(const std::string &prompt, int maxTokens = 5000);
// Overload that allows passing a system message separately so chat-capable
// models receive a proper system role instead of having the system text
// concatenated into the user prompt (helps avoid revealing internal
// reasoning or instructions in model output).
std::string infer(const std::string &systemPrompt, const std::string &prompt,
int maxTokens = 5000);
llama_model *model_ = nullptr;
llama_context *context_ = nullptr;

1
pipeline/includes/mock_generator.h
View File

@@ -8,6 +8,7 @@ class MockGenerator final : public IDataGenerator {
public:
void load(const std::string &modelPath) override;
BreweryResult generateBrewery(const std::string &cityName,
const std::string &countryName,
const std::string &regionContext) override;
UserResult generateUser(const std::string &locale) override;