Skip to content

Commit 38fc9e0

Browse files
fix: add isReplay to remaining webapp execution context constructions
Co-Authored-By: nick <55853254+nicktrn@users.noreply.github.com>
1 parent 3f5a235 commit 38fc9e0

3 files changed

Lines changed: 7 additions & 11 deletions

File tree

apps/webapp/app/presenters/v3/SpanPresenter.server.ts

Lines changed: 4 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -42,9 +42,7 @@ export type PromptSpanData = {
4242
config?: string;
4343
};
4444

45-
function extractPromptSpanData(
46-
properties: Record<string, unknown>
47-
): PromptSpanData | undefined {
45+
function extractPromptSpanData(properties: Record<string, unknown>): PromptSpanData | undefined {
4846
// Properties come as an unflattened nested object from ClickHouse,
4947
// e.g. { prompt: { slug: "...", version: 3, ... } }
5048
const prompt = properties.prompt;
@@ -592,10 +590,7 @@ export class SpanPresenter extends BasePresenter {
592590
triggeredRuns,
593591
aiData:
594592
span.properties && typeof span.properties === "object"
595-
? extractAISpanData(
596-
span.properties as Record<string, unknown>,
597-
span.duration / 1_000_000
598-
)
593+
? extractAISpanData(span.properties as Record<string, unknown>, span.duration / 1_000_000)
599594
: undefined,
600595
};
601596

@@ -739,10 +734,7 @@ export class SpanPresenter extends BasePresenter {
739734
"ai.streamObject",
740735
];
741736

742-
if (
743-
typeof span.message === "string" &&
744-
AI_SUMMARY_MESSAGES.includes(span.message)
745-
) {
737+
if (typeof span.message === "string" && AI_SUMMARY_MESSAGES.includes(span.message)) {
746738
const aiSummaryData = extractAISummarySpanData(
747739
span.properties as Record<string, unknown>,
748740
span.duration / 1_000_000
@@ -899,6 +891,7 @@ export class SpanPresenter extends BasePresenter {
899891
createdAt: run.createdAt,
900892
tags: run.runTags,
901893
isTest: run.isTest,
894+
isReplay: !!run.replayedFromTaskRunFriendlyId,
902895
idempotencyKey: getUserProvidedIdempotencyKey(run) ?? undefined,
903896
startedAt: run.startedAt ?? run.createdAt,
904897
durationMs: run.usageDurationMs,

apps/webapp/app/v3/marqs/sharedQueueConsumer.server.ts

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1640,6 +1640,7 @@ export const AttemptForExecutionGetPayload = {
16401640
createdAt: true,
16411641
startedAt: true,
16421642
isTest: true,
1643+
replayedFromTaskRunFriendlyId: true,
16431644
metadata: true,
16441645
metadataType: true,
16451646
idempotencyKey: true,
@@ -1726,6 +1727,7 @@ class SharedQueueTasks {
17261727
startedAt: taskRun.startedAt ?? taskRun.createdAt,
17271728
tags: taskRun.runTags ?? [],
17281729
isTest: taskRun.isTest,
1730+
isReplay: !!taskRun.replayedFromTaskRunFriendlyId,
17291731
idempotencyKey: taskRun.idempotencyKey ?? undefined,
17301732
durationMs: taskRun.usageDurationMs,
17311733
costInCents: taskRun.costInCents,

apps/webapp/app/v3/services/createTaskRunAttempt.server.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -210,6 +210,7 @@ export class CreateTaskRunAttemptService extends BaseService {
210210
createdAt: taskRun.createdAt,
211211
tags: taskRun.runTags ?? [],
212212
isTest: taskRun.isTest,
213+
isReplay: !!taskRun.replayedFromTaskRunFriendlyId,
213214
idempotencyKey: taskRun.idempotencyKey ?? undefined,
214215
startedAt: taskRun.startedAt ?? taskRun.createdAt,
215216
durationMs: taskRun.usageDurationMs,

0 commit comments

Comments
 (0)