fix: sanitize runID before countTokens (#477)

lunary-ai · Aug 5, 2024 · d0d4f73 · d0d4f73
1 parent badaf68
commit d0d4f73
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/packages/backend/src/utils/ingest.ts b/packages/backend/src/utils/ingest.ts
@@ -129,16 +129,20 @@ export async function cleanEvent(event: any): Promise<Event> {
     console.error(event)
   }
 
-  return {
+  const cleanedRun = {
     ...rest,
     metadata: cleanMetadata(rest.metadata),
     name: typeof name === "string" ? name.replace("models/", "") : undefined,
     tags: typeof tags === "string" ? [tags] : tags,
-    tokensUsage: await completeRunUsageWithTimeout(event),
     runId: await ensureIsUUID(runId),
     parentRunId: await ensureIsUUID(parentRunId),
     timestamp: isoTimestamp,
   }
+
+  // Do it after because we need the sanitized runId/parentRunId
+  cleanedRun.tokensUsage = await completeRunUsageWithTimeout(cleanedRun)
+
+  return cleanedRun
 }
 
 export const clearUndefined = (obj: any): any =>