chore: remove prisma

This commit is contained in:
Riccardo Senica
2025-01-09 05:41:55 +00:00
parent 53b9afeafd
commit 152088a923
15 changed files with 8 additions and 258 deletions

View File

@@ -4,7 +4,6 @@ import { Tool } from './tool';
import { BaseTool, makeRequest } from '@anthropic';
import { generatePrompt } from './prompt';
import { generateConsumerSeed } from '@utils/generateConsumerSeed';
import prisma from '@prisma';
export async function generate() {
const { letters, birthday, zipCode } = generateConsumerSeed();
@@ -24,19 +23,7 @@ export async function generate() {
console.info('Generated consumer by Anthropic', validConsumer.data);
const newConsumer = await prisma.consumer.create({
data: {
letters,
birthday: birthday.toDate(),
zipCode,
value: validConsumer.data
}
});
console.info(`Consumer with id ${newConsumer.id} stored in database.`);
return {
id: newConsumer.id,
consumer: validConsumer.data
};
} catch (error) {

View File

@@ -3,13 +3,8 @@ import { Tool } from './tool';
import { BaseTool, makeRequest } from '@anthropic';
import { generatePrompt } from './prompt';
import { Consumer } from '@consumer/types';
import prisma from '@prisma';
export async function generate(
id: number | undefined,
editedConsumer: Consumer,
date: Date
) {
export async function generate(editedConsumer: Consumer, date: Date) {
const consumerPrompt = await generatePrompt(
editedConsumer,
parseInt(process.env.PURCHASE_REFLECTION_THRESHOLD ?? '50'),
@@ -17,10 +12,6 @@ export async function generate(
parseInt(process.env.NUMBER_OF_WEEKS ?? '4')
);
console.info(
`Generating new purchase list for consumer with id ${id ?? 'N/A'}`
);
try {
const result = (await makeRequest(
consumerPrompt,
@@ -39,31 +30,7 @@ export async function generate(
);
console.info(
`Generated ${totalPurchases} purchases for new purchase list for consumer wth id ${id}`
);
const [consumer, purchaseList] = await prisma.$transaction(async tx => {
const consumer = id
? await tx.consumer.update({
where: { id },
data: { editedValue: editedConsumer }
})
: await tx.consumer.create({
data: { editedValue: editedConsumer }
});
const purchaseList = await tx.purchaseList.create({
data: {
consumerId: consumer.id,
value: validPurchases.data
}
});
return [consumer, purchaseList];
});
console.info(
`Purchase list with id ${purchaseList.id} for consumer with id ${consumer.id} stored in database.`
`Generated ${totalPurchases} purchases for new purchase list for consumer`
);
return validPurchases.data;

View File

@@ -1,30 +0,0 @@
import prisma from '@prisma';
export async function rateLimiter() {
if (!process.env.RATE_LIMIT) {
throw Error('Rate limit missing.');
}
const yesterday = new Date(Date.now() - 24 * 60 * 60 * 1000);
const count = await prisma.$transaction(async tx => {
const consumersCount = await tx.consumer.count({
where: {
createdAt: {
gt: yesterday
}
}
});
const purchaseListsCount = await tx.purchaseList.count({
where: {
createdAt: {
gt: yesterday
}
}
});
return consumersCount + purchaseListsCount;
});
return count > parseInt(process.env.RATE_LIMIT);
}