Skip to content

Commit e98bfc6

Browse files
committed
Fix metric calc for new pipeline results format
1 parent d84e9e6 commit e98bfc6

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

app/components/pipeline/nodes/StepNode.tsx

+2-1
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,8 @@ export function StepNode({ data }: { data: ProcessingStep }) {
4545
useMemo(() => {
4646
// const timings = data.
4747
const results = litlytics.docs;
48-
const res = results.filter((doc) => doc);
48+
const resultDocs = litlytics.pipeline.results?.map((r) => r.doc) ?? [];
49+
const res = results.concat(resultDocs).filter((doc) => doc);
4950
if (!res.length) {
5051
return {};
5152
}

app/components/pipeline/nodes/output/OutputNode.tsx

+2-1
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,8 @@ export function OutputNode() {
5151
const { timing, prompt, completion, cost } = useMemo(() => {
5252
// const timings = data.
5353
const results = litlytics.docs;
54-
const res = results.filter((doc) => doc);
54+
const resultDocs = litlytics.pipeline.results?.map((r) => r.doc) ?? [];
55+
const res = results.concat(resultDocs).filter((doc) => doc);
5556
if (!res.length) {
5657
return {};
5758
}

0 commit comments

Comments
 (0)