Cossale/kepler-chat
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

update free logic

Browse source
This commit is contained in:
Thomas G. Lopes 2025-06-19 13:37:30 +01:00
parent ee2d8bcb6c
commit 30312e6f55
3 changed files with 50 additions and 32 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
.env.example
View file

@ -8,3 +8,9 @@ BETTER_AUTH_SECRET=
GITHUB_CLIENT_ID=
GITHUB_CLIENT_SECRET=
GOOGLE_CLIENT_ID=
GOOGLE_CLIENT_SECRET=
OPENROUTER_FREE_KEY=

6
src/lib/backend/convex/user_enabled_models.ts
View file

@ -138,7 +138,11 @@ export const enable_initial = mutation({
return;
}
const initialModels = ['google/gemini-2.5-flash', 'deepseek/deepseek-chat-v3-0324:free'];
const initialModels = [
'google/gemini-2.5-flash',
'deepseek/deepseek-chat-v3-0324:free',
'microsoft/phi-4:free',
];
await Promise.all(
initialModels.map((model) =>

66
src/routes/api/generate-message/+server.ts
View file

@ -301,42 +301,50 @@ async function generateAIResponse({
actualKey = userKey;
log('Background: Using user API key', startTime);
} else {
// User doesn't have API key, check free tier limit
const freeMessagesUsed = userSettings?.free_messages_used || 0;
// User doesn't have API key, check if using a free model
const isFreeModel = model.model_id.endsWith(':free');
if (freeMessagesUsed >= 10) {
handleGenerationError({
error: 'Free message limit reached (10/10). Please add your own OpenRouter API key to continue chatting.',
conversationId,
messageId: mid,
sessionToken,
startTime,
});
return;
}
if (!isFreeModel) {
// For non-free models, check the 10 message limit
const freeMessagesUsed = userSettings?.free_messages_used || 0;
// Increment free message count before generating
const incrementResult = await ResultAsync.fromPromise(
client.mutation(api.user_settings.incrementFreeMessageCount, {
session_token: sessionToken,
}),
(e) => `Failed to increment free message count: ${e}`
);
if (freeMessagesUsed >= 10) {
handleGenerationError({
error: 'Free message limit reached (10/10). Please add your own OpenRouter API key to continue chatting, or use a free model ending in ":free".',
conversationId,
messageId: mid,
sessionToken,
startTime,
});
return;
}
if (incrementResult.isErr()) {
handleGenerationError({
error: `Failed to track free message usage: ${incrementResult.error}`,
conversationId,
messageId: mid,
sessionToken,
startTime,
});
return;
// Increment free message count before generating (only for non-free models)
const incrementResult = await ResultAsync.fromPromise(
client.mutation(api.user_settings.incrementFreeMessageCount, {
session_token: sessionToken,
}),
(e) => `Failed to increment free message count: ${e}`
);
if (incrementResult.isErr()) {
handleGenerationError({
error: `Failed to track free message usage: ${incrementResult.error}`,
conversationId,
messageId: mid,
sessionToken,
startTime,
});
return;
}
log(`Background: Using free tier (${freeMessagesUsed + 1}/10 messages)`, startTime);
} else {
log(`Background: Using free model (${model.model_id}) - no message count`, startTime);
}
// Use environment OpenRouter key
actualKey = OPENROUTER_FREE_KEY;
log(`Background: Using free tier (${freeMessagesUsed + 1}/10 messages)`, startTime);
}
if (rulesResult.isErr()) {