This commit is contained in:
DirkSchlossmacher 2023-11-23 14:37:21 +01:00
parent d86b7ddd50
commit ee042d9dbe

View File

@ -29,6 +29,10 @@ import { incrementAPICallCount } from '../../utils/cloud/redisRestClient';
// app\client\platforms\openai.ts
// app\auth.ts
export async function getMyServerSession() {
return await getServerAuthSession();
}
export interface OpenAIListModelResponse {
@ -109,31 +113,27 @@ export class ChatGPTApi implements LLMApi {
const modelIdentifier = modelConfig.model;
const modelIdentifier = modelConfig.model;
console.log("API Call: session or email is not available - model: ", modelIdentifier);
const session = await getMyServerSession(req);
if (session?.user?.email) {
// Now you have the user's email from the session
const userEmail = session.user.email;
const dateKey = new Date().toISOString().slice(0, 7); // "YYYY-MM"
console.log("API Call: ", userEmail, modelIdentifier);
// Use the userEmail to increment the API call count
await incrementAPICallCount(userEmail, modelIdentifier, dateKey);
} else {
// Handle cases where the session or email is not available
console.log("API Call: session or email is not available - model: ", modelIdentifier);
export default async function handler(req, res) {
const session = await getServerAuthSession({ req });
}
if (session?.user?.email) {
// Now you have the user's email from the session
const userEmail = session.user.email;
const dateKey = new Date().toISOString().slice(0, 7); // "YYYY-MM"
console.log("API Call: ", userEmail, modelIdentifier);
// Use the userEmail to increment the API call count
await incrementAPICallCount(userEmail, modelIdentifier, dateKey);
// ... rest of your API route logic ...
res.status(200).json({ success: true });
} else {
// Handle cases where the session or email is not available
console.log("API Call: session or email is not available - model: ", modelIdentifier);
res.status(401).json({ error: "Unauthorized" });
}
}
/*
// export default async function handler(req: NextApiRequest, res: NextApiResponse) {
// Retrieve the session using getServerAuthSession