Seed with tags

This commit is contained in:
David Corbitt
2023-08-15 00:26:11 -07:00
parent bb9c3a9e61
commit 6e7efee21e

View File

@@ -13,6 +13,7 @@ const MODEL_RESPONSE_TEMPLATES: {
inputTokens: number; inputTokens: number;
outputTokens: number; outputTokens: number;
finishReason: string; finishReason: string;
tags: { name: string; value: string }[];
}[] = [ }[] = [
{ {
reqPayload: { reqPayload: {
@@ -107,6 +108,7 @@ const MODEL_RESPONSE_TEMPLATES: {
inputTokens: 236, inputTokens: 236,
outputTokens: 5, outputTokens: 5,
finishReason: "stop", finishReason: "stop",
tags: [],
}, },
{ {
reqPayload: { reqPayload: {
@@ -193,6 +195,7 @@ const MODEL_RESPONSE_TEMPLATES: {
inputTokens: 222, inputTokens: 222,
outputTokens: 5, outputTokens: 5,
finishReason: "stop", finishReason: "stop",
tags: [],
}, },
{ {
reqPayload: { reqPayload: {
@@ -231,6 +234,7 @@ const MODEL_RESPONSE_TEMPLATES: {
inputTokens: 14, inputTokens: 14,
outputTokens: 7, outputTokens: 7,
finishReason: "stop", finishReason: "stop",
tags: [{ name: "prompt_id", value: "id2" }],
}, },
{ {
reqPayload: { reqPayload: {
@@ -306,6 +310,7 @@ const MODEL_RESPONSE_TEMPLATES: {
inputTokens: 2802, inputTokens: 2802,
outputTokens: 108, outputTokens: 108,
finishReason: "stop", finishReason: "stop",
tags: [{ name: "prompt_id", value: "chatcmpl-7lQS3MktOT8BTgNEytl9dkyssCQqL" }],
}, },
]; ];
@@ -349,6 +354,7 @@ for (let i = 0; i < 1437; i++) {
cacheHit: false, cacheHit: false,
requestedAt, requestedAt,
projectId: project.id, projectId: project.id,
model: template.reqPayload.model,
createdAt: requestedAt, createdAt: requestedAt,
}); });
@@ -388,13 +394,15 @@ for (let i = 0; i < 1437; i++) {
modelResponseId: loggedCallModelResponseId, modelResponseId: loggedCallModelResponseId,
}, },
}); });
for (const tag of template.tags) {
loggedCallTagsToCreate.push({ loggedCallTagsToCreate.push({
projectId: project.id, projectId: project.id,
loggedCallId, loggedCallId,
name: "$model", name: tag.name,
value: template.reqPayload.model, value: tag.value,
}); });
} }
}
await prisma.$transaction([ await prisma.$transaction([
prisma.loggedCall.createMany({ prisma.loggedCall.createMany({