peptide-repo/prautoblogger
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Amortize LLM research cost across generated articles (v0.5.2)

Browse Source 
Research LLM call runs once per pipeline but was logged with no run_id
or post_id, making it invisible in per-article cost breakdowns. Now the
pipeline's cost tracker flows through Source_Collector to the research
provider, tagging costs with the run_id. After all articles complete,
amortize_research_costs() divides the cost evenly and inserts one row
per article. link_generation_logs() excludes llm_research stage so the
first article doesn't grab the full cost.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
peptiderepo
2026-04-20 00:44:33 +07:00
parent 327cdc08d9
commit 264fcf0f84
6 changed files with 153 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
includes/providers/class-llm-research-provider.php
View File

@@ -34,7 +34,7 @@ class PRAutoBlogger_LLM_Research_Provider implements PRAutoBlogger_Source_Provid
*
* Side effects: LLM API call via OpenRouter, cost logging.
*
* @param array{prompt?: string, model?: string} $config Research-specific configuration.
* @param array{prompt?: string, model?: string, cost_tracker?: PRAutoBlogger_Cost_Tracker} $config Research-specific configuration.
* @return PRAutoBlogger_Source_Data[] Array of research findings as source data.
* @throws \RuntimeException On API error after retries.
*/
@@ -65,7 +65,9 @@ class PRAutoBlogger_LLM_Research_Provider implements PRAutoBlogger_Source_Provid
]
);
$cost_tracker = new PRAutoBlogger_Cost_Tracker();
// Use the pipeline's cost tracker if provided (for run_id tagging),
// otherwise create a standalone one (backward compat for manual calls).
$cost_tracker = $config['cost_tracker'] ?? new PRAutoBlogger_Cost_Tracker();
$cost_tracker->log_api_call(
null,
'llm_research',