eric/weatherreporter
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 2 Wiki Activity

Add runtime prompt values to module outputs

Browse Source
This commit is contained in:
Eric Rakestraw
2026-06-15 20:23:02 +00:00
parent 4fac69c9f0
commit ff6aade42c
3 changed files with 59 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
internal/state/filesystem_test.go
View File

@@ -114,7 +114,12 @@ func TestSaveArtifactsAndMetadataRoundTrip(t *testing.T) {
store := newTestStore(t)
resolved := resolveDailyAt(t, "2026-05-29T05:00:00-05:00")
briefingMetadata := stateBriefingMetadata(resolved)
snapshot, err := module.NewSnapshot([]module.Output{{ID: module.Metadata, StanzaName: "metadata", Value: map[string]string{"run_id": resolved.Metadata().RunID}}})
snapshot, err := module.NewSnapshot([]module.Output{{
ID: module.Metadata,
StanzaName: "metadata",
Value: map[string]string{"run_id": resolved.Metadata().RunID},
PromptValue: map[string]string{"prompt_run_id": resolved.Metadata().RunID},
}})
if err != nil {
t.Fatalf("NewSnapshot() error = %v", err)
}
@@ -220,6 +225,16 @@ func TestSaveArtifactsAndMetadataRoundTrip(t *testing.T) {
if loadedSnapshot.SchemaVersion != module.SnapshotSchemaVersion || len(loadedSnapshot.Outputs) != 1 {
t.Fatalf("loaded module snapshot = %#v, want one metadata output", loadedSnapshot)
}
if loadedSnapshot.Outputs[0].PromptValue != nil {
t.Fatalf("loaded module snapshot PromptValue = %#v, want omitted runtime value", loadedSnapshot.Outputs[0].PromptValue)
}
snapshotData, err := os.ReadFile(moduleSnapshotPath)
if err != nil {
t.Fatalf("read module snapshot: %v", err)
}
if strings.Contains(string(snapshotData), "prompt_run_id") || strings.Contains(string(snapshotData), "promptValue") || strings.Contains(string(snapshotData), "PromptValue") {
t.Fatalf("module snapshot JSON includes runtime-only prompt value:\n%s", string(snapshotData))
}
loadedDataPackage, err := store.LoadDataPackage(context.Background(), dataPackagePath)
if err != nil {
t.Fatalf("LoadDataPackage() error = %v", err)