From c588bbd525f12d16a2bb937d24f7370580781b5e Mon Sep 17 00:00:00 2001
From: Matt Aitken <matt@mattaitken.com>
Date: Fri, 28 Jun 2024 11:15:04 +0100
Subject: [PATCH] Send taskIdentifier to OpenMeter

---
 apps/webapp/app/routes/api.v1.usage.ingest.ts | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/apps/webapp/app/routes/api.v1.usage.ingest.ts b/apps/webapp/app/routes/api.v1.usage.ingest.ts
index 5d99fb146..b4c165c8d 100644
--- a/apps/webapp/app/routes/api.v1.usage.ingest.ts
+++ b/apps/webapp/app/routes/api.v1.usage.ingest.ts
@@ -49,7 +49,7 @@ export async function action({ request }: ActionFunctionArgs) {
   if (json.data.durationMs > 0) {
     const costInCents = json.data.durationMs * preset.centsPerMs;
 
-    await prisma.taskRun.update({
+    const taskRun = await prisma.taskRun.update({
       where: {
         id: jwtResult.payload.run_id,
       },
@@ -71,6 +71,7 @@ export async function action({ request }: ActionFunctionArgs) {
         data: {
           durationMs: json.data.durationMs,
           costInCents: String(costInCents),
+          taskIdentifier: taskRun.taskIdentifier,
         },
       });
     } catch (e) {
@@ -83,6 +84,7 @@ export async function action({ request }: ActionFunctionArgs) {
         },
         additionalData: {
           durationMs: json.data.durationMs,
+          taskIdentifier: taskRun.taskIdentifier,
         },
       });
     }