send analyses to ai
This commit is contained in:
@@ -1,133 +0,0 @@
|
||||
import type { Mock } from 'jest-mock';
|
||||
|
||||
// ---- Mocks you can tweak per test ----
|
||||
const createResponseMock = jest.fn();
|
||||
const getLatestResponseTimeMock = jest.fn();
|
||||
const getLatestUniqueAnalysResponsesMock = jest.fn();
|
||||
const parsePersonalCodeMock = jest.fn(() => ({ gender: { value: 'male' } }));
|
||||
|
||||
// Mock OpenAI SDK
|
||||
jest.mock('openai', () => {
|
||||
return {
|
||||
__esModule: true,
|
||||
default: class OpenAI {
|
||||
responses = { create: createResponseMock };
|
||||
},
|
||||
};
|
||||
});
|
||||
|
||||
// Mock next/cache (global cache map so it persists between calls)
|
||||
const globalCache = new Map<string, unknown>();
|
||||
jest.mock('next/cache', () => {
|
||||
return {
|
||||
__esModule: true,
|
||||
unstable_cache:
|
||||
(fn: (...args: any[]) => Promise<any>, keyParts: any[], _opts?: any) =>
|
||||
async (...args: any[]) => {
|
||||
const key = JSON.stringify(keyParts);
|
||||
if (globalCache.has(key)) return globalCache.get(key);
|
||||
const val = await fn(...args);
|
||||
globalCache.set(key, val);
|
||||
return val;
|
||||
},
|
||||
};
|
||||
});
|
||||
|
||||
// Mock your analysis helpers + personal code parser
|
||||
jest.mock('../src/analysis-utils', () => ({
|
||||
__esModule: true,
|
||||
getLatestUniqueAnalysResponses: getLatestUniqueAnalysResponsesMock,
|
||||
getLatestResponseTime: getLatestResponseTimeMock,
|
||||
}));
|
||||
jest.mock('../src/personal-code', () => ({
|
||||
__esModule: true,
|
||||
PersonalCode: { parsePersonalCode: parsePersonalCodeMock },
|
||||
}));
|
||||
|
||||
describe('loadRecommendations', () => {
|
||||
beforeEach(() => {
|
||||
createResponseMock.mockReset();
|
||||
getLatestResponseTimeMock.mockReset();
|
||||
getLatestUniqueAnalysResponsesMock.mockReset();
|
||||
globalCache.clear();
|
||||
});
|
||||
|
||||
it('should call OpenAI once when latest date stays the same (cache hit on 2nd call)', async () => {
|
||||
const date1 = new Date('2025-09-16T12:00:00Z');
|
||||
getLatestResponseTimeMock.mockReturnValue(date1);
|
||||
getLatestUniqueAnalysResponsesMock.mockImplementation((arr: any[]) => arr);
|
||||
|
||||
createResponseMock.mockResolvedValue({
|
||||
output_text: JSON.stringify({ recommended: ['A', 'B'] }),
|
||||
});
|
||||
|
||||
const { loadRecommendations } = await import('./load-recommendations');
|
||||
|
||||
const analysisResponses = [
|
||||
{ name: 'x', value: 1, responseTime: date1 },
|
||||
] as any[];
|
||||
const analyses = [{ title: 't', description: 'd' }] as any[];
|
||||
const account = { id: 'u1', personal_code: '12345678901' } as any;
|
||||
|
||||
// Act: 1st call (MISS)
|
||||
const out1 = await loadRecommendations(
|
||||
analysisResponses,
|
||||
analyses,
|
||||
account,
|
||||
);
|
||||
|
||||
// Act: 2nd call with same date (HIT)
|
||||
const out2 = await loadRecommendations(
|
||||
analysisResponses,
|
||||
analyses,
|
||||
account,
|
||||
);
|
||||
|
||||
// Assert: only one API call, result reused
|
||||
expect(createResponseMock).toHaveBeenCalledTimes(1);
|
||||
expect(out1).toEqual(['A', 'B']);
|
||||
expect(out2).toEqual(['A', 'B']);
|
||||
});
|
||||
|
||||
it('should call OpenAI again when latest date changes (new cache key)', async () => {
|
||||
const date1 = new Date('2025-09-16T12:00:00Z');
|
||||
const date2 = new Date('2025-09-17T00:00:00Z');
|
||||
|
||||
getLatestResponseTimeMock.mockReturnValueOnce(date1);
|
||||
getLatestResponseTimeMock.mockReturnValueOnce(date2);
|
||||
|
||||
getLatestUniqueAnalysResponsesMock.mockImplementation((arr: any[]) => arr);
|
||||
|
||||
createResponseMock
|
||||
.mockResolvedValueOnce({
|
||||
output_text: JSON.stringify({ recommended: ['A', 'B'] }),
|
||||
})
|
||||
.mockResolvedValueOnce({
|
||||
output_text: JSON.stringify({ recommended: ['C'] }),
|
||||
});
|
||||
|
||||
const { loadRecommendations } = await import('./load-recommendations');
|
||||
|
||||
const analysisResponses = [
|
||||
{ name: 'x', value: 1, responseTime: date1 },
|
||||
] as any[];
|
||||
const analyses = [{ title: 't', description: 'd' }] as any[];
|
||||
const account = { id: 'u1', personal_code: '12345678901' } as any;
|
||||
|
||||
const out1 = await loadRecommendations(
|
||||
analysisResponses,
|
||||
analyses,
|
||||
account,
|
||||
);
|
||||
|
||||
const out2 = await loadRecommendations(
|
||||
analysisResponses,
|
||||
analyses,
|
||||
account,
|
||||
);
|
||||
|
||||
expect(createResponseMock).toHaveBeenCalledTimes(2);
|
||||
expect(out1).toEqual(['A', 'B']);
|
||||
expect(out2).toEqual(['C']);
|
||||
});
|
||||
});
|
||||
@@ -1,9 +1,6 @@
|
||||
import { cache } from 'react';
|
||||
|
||||
import { unstable_cache } from 'next/cache';
|
||||
|
||||
import { AccountWithParams } from '@/packages/features/accounts/src/types/accounts';
|
||||
import crypto from 'crypto';
|
||||
import OpenAI from 'openai';
|
||||
|
||||
import PersonalCode from '~/lib/utils';
|
||||
@@ -19,20 +16,6 @@ type FormattedAnalysisResponse = {
|
||||
responseTime: string;
|
||||
};
|
||||
|
||||
function canonical(v: unknown): string {
|
||||
return JSON.stringify(v, (_k, val) => {
|
||||
if (val && typeof val === 'object' && !Array.isArray(val)) {
|
||||
// sort object keys for stable JSON
|
||||
return Object.keys(val)
|
||||
.sort()
|
||||
.reduce((o: any, k) => ((o[k] = (val as any)[k]), o), {});
|
||||
}
|
||||
return val;
|
||||
});
|
||||
}
|
||||
const sha256 = (s: string) =>
|
||||
crypto.createHash('sha256').update(s).digest('hex');
|
||||
|
||||
const getLatestResponseTime = (items: FormattedAnalysisResponse[]) => {
|
||||
if (!items?.length) return null;
|
||||
|
||||
@@ -56,23 +39,29 @@ const getLatestUniqueAnalysResponses = (
|
||||
|
||||
console.log('analysisElements', analysisElements.length);
|
||||
|
||||
const map = new Map();
|
||||
for (const it of analysisElements) {
|
||||
const prev = map.get(it.analysisName);
|
||||
if (it.results.responseTime) {
|
||||
if (
|
||||
!prev ||
|
||||
new Date(it.results.responseTime) > new Date(prev.responseTime)
|
||||
) {
|
||||
map.set(it.analysisName, {
|
||||
name: it.analysisName,
|
||||
value: it.results.responseValue,
|
||||
responseTime: it.results.responseTime,
|
||||
});
|
||||
}
|
||||
const byName = analysisElements.reduce<
|
||||
Record<string, { name: string; value: string; responseTime: string }>
|
||||
>((acc, it) => {
|
||||
const responseTime = it?.results?.responseTime;
|
||||
const responseValue = it?.results?.responseValue;
|
||||
if (!responseTime || !responseValue) return acc;
|
||||
|
||||
const key = it.analysisName;
|
||||
const cur = acc[key];
|
||||
const t = Date.parse(responseTime);
|
||||
const prevT = cur ? Date.parse(cur.responseTime) : -Infinity;
|
||||
|
||||
if (!cur || t > prevT) {
|
||||
acc[key] = {
|
||||
name: key,
|
||||
value: responseValue.toString(),
|
||||
responseTime,
|
||||
};
|
||||
}
|
||||
}
|
||||
return [...map.values()];
|
||||
return acc;
|
||||
}, {});
|
||||
|
||||
return Object.values(byName);
|
||||
};
|
||||
|
||||
async function recommendationsLoader(
|
||||
@@ -80,6 +69,9 @@ async function recommendationsLoader(
|
||||
analyses: OrderAnalysisCard[],
|
||||
account: AccountWithParams | null,
|
||||
): Promise<any> {
|
||||
if (!process.env.OPENAI_API_KEY) {
|
||||
return [];
|
||||
}
|
||||
if (!account?.personal_code) {
|
||||
return [];
|
||||
}
|
||||
@@ -90,11 +82,11 @@ async function recommendationsLoader(
|
||||
console.log('analyises', analyses);
|
||||
const latestUniqueAnalysResponses =
|
||||
getLatestUniqueAnalysResponses(analysisResponses);
|
||||
const latestResponseTime = getLatestResponseTime(latestUniqueAnalysResponses);
|
||||
const latestISO = latestResponseTime
|
||||
? new Date(latestResponseTime).toISOString()
|
||||
: 'none';
|
||||
console.log('latestResponseTime', latestResponseTime);
|
||||
// const latestResponseTime = getLatestResponseTime(latestUniqueAnalysResponses);
|
||||
// const latestISO = latestResponseTime
|
||||
// ? new Date(latestResponseTime).toISOString()
|
||||
// : 'none';
|
||||
// console.log('latestResponseTime', latestResponseTime);
|
||||
const formattedAnalysisResponses = latestUniqueAnalysResponses.map(
|
||||
({ name, value }) => ({ name, value }),
|
||||
);
|
||||
@@ -108,8 +100,9 @@ async function recommendationsLoader(
|
||||
'latestUniqueAnalysResponses',
|
||||
JSON.stringify(latestUniqueAnalysResponses),
|
||||
);
|
||||
|
||||
const response = await client.responses.create({
|
||||
model: 'gpt-5',
|
||||
model: 'gpt-5-mini',
|
||||
store: false,
|
||||
prompt: {
|
||||
id: 'pmpt_68ca9c8bfa8c8193b27eadc6496c36440df449ece4f5a8dd',
|
||||
@@ -121,33 +114,7 @@ async function recommendationsLoader(
|
||||
},
|
||||
});
|
||||
|
||||
const responseJson = JSON.parse(response.output_text);
|
||||
console.log('responseJson: ', responseJson);
|
||||
|
||||
const keyPayload = {
|
||||
model: 'gpt-5', // swap to a model your project can access
|
||||
promptId: 'pmpt_68ca9c8bfa8c8193b27eadc6496c36440df449ece4f5a8dd',
|
||||
latestISO,
|
||||
};
|
||||
|
||||
const key = 'recs:' + sha256(canonical(keyPayload));
|
||||
|
||||
const run = unstable_cache(async () => {
|
||||
const response = await client.responses.create({
|
||||
model: keyPayload.model,
|
||||
store: false,
|
||||
prompt: {
|
||||
id: keyPayload.promptId,
|
||||
variables: {
|
||||
analyses: JSON.stringify(formattedAnalyses),
|
||||
results: JSON.stringify(latestUniqueAnalysResponses),
|
||||
gender: gender.value,
|
||||
},
|
||||
},
|
||||
});
|
||||
const json = JSON.parse(response.output_text);
|
||||
return json.recommended;
|
||||
}, ['recommendations', key]);
|
||||
|
||||
return await run();
|
||||
const json = JSON.parse(response.output_text);
|
||||
console.log('response.output_text', response.output_text);
|
||||
return json.recommended;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user