Seed with tags
This commit is contained in:
@@ -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,12 +394,14 @@ 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([
|
||||||
|
|||||||
Reference in New Issue
Block a user