diff --git a/app/prisma/migrations/20230811231429_add_model_to_logged_call/migration.sql b/app/prisma/migrations/20230811231429_add_model_to_logged_call/migration.sql new file mode 100644 index 0000000..fc4294c --- /dev/null +++ b/app/prisma/migrations/20230811231429_add_model_to_logged_call/migration.sql @@ -0,0 +1,2 @@ +-- AlterTable +ALTER TABLE "LoggedCall" ADD COLUMN "model" TEXT; diff --git a/app/prisma/schema.prisma b/app/prisma/schema.prisma index 59319f7..dfc39a0 100644 --- a/app/prisma/schema.prisma +++ b/app/prisma/schema.prisma @@ -273,7 +273,8 @@ model LoggedCall { projectId String @db.Uuid project Project? @relation(fields: [projectId], references: [id], onDelete: Cascade) - tags LoggedCallTag[] + model String? + tags LoggedCallTag[] createdAt DateTime @default(now()) updatedAt DateTime @updatedAt diff --git a/app/src/server/api/routers/externalApi.router.ts b/app/src/server/api/routers/externalApi.router.ts index 7e1b6c7..ca80ae1 100644 --- a/app/src/server/api/routers/externalApi.router.ts +++ b/app/src/server/api/routers/externalApi.router.ts @@ -141,11 +141,13 @@ export const externalApiRouter = createTRPCRouter({ const newModelResponseId = uuidv4(); let usage; + let model; if (reqPayload.success && respPayload.success) { usage = modelProvider.getUsage( input.reqPayload as CompletionCreateParams, input.respPayload as ChatCompletion, ); + model = reqPayload.data.model; } await prisma.$transaction([ @@ -155,6 +157,7 @@ export const externalApiRouter = createTRPCRouter({ projectId: key.projectId, requestedAt: new Date(input.requestedAt), cacheHit: false, + model, }, }), prisma.loggedCallModelResponse.create({ @@ -191,14 +194,6 @@ export const externalApiRouter = createTRPCRouter({ name: name.replaceAll(/[^a-zA-Z0-9_]/g, "_"), value, })); - - if (reqPayload.success) { - tagsToCreate.push({ - loggedCallId: newLoggedCallId, - name: "$model", - value: reqPayload.data.model, - }); - } await prisma.loggedCallTag.createMany({ data: tagsToCreate, });