apunkt/PriceGhost
1
0
Fork 0
mirror of https://github.com/clucraft/PriceGhost.git synced 2026-05-12 17:22:39 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add Google Gemini AI support (v1.0.6)

Browse source 
- Add Gemini as new AI provider option alongside Anthropic, OpenAI, Ollama
- Support models: Gemini 2.5 Flash Lite (default), Flash, Pro, 3 Flash Preview
- Add test API key endpoint to verify connection before saving
- Full support for extraction, verification, stock status, and arbitration
- Update all version files to 1.0.6

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
This commit is contained in:
clucraft 2026-01-26 21:02:14 -05:00
parent fb00f5b5b2
commit 617953dcf5
12 changed files with 341 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

66
backend/src/routes/settings.ts
View file

@ -314,9 +314,13 @@ router.get('/ai', async (req: AuthRequest, res: Response) => {
ai_verification_enabled: settings.ai_verification_enabled || false,
ai_provider: settings.ai_provider || null,
anthropic_api_key: settings.anthropic_api_key || null,
anthropic_model: settings.anthropic_model || null,
openai_api_key: settings.openai_api_key || null,
openai_model: settings.openai_model || null,
ollama_base_url: settings.ollama_base_url || null,
ollama_model: settings.ollama_model || null,
gemini_api_key: settings.gemini_api_key || null,
gemini_model: settings.gemini_model || null,
});
} catch (error) {
console.error('Error fetching AI settings:', error);
@ -328,16 +332,32 @@ router.get('/ai', async (req: AuthRequest, res: Response) => {
router.put('/ai', async (req: AuthRequest, res: Response) => {
try {
const userId = req.userId!;
const { ai_enabled, ai_verification_enabled, ai_provider, anthropic_api_key, openai_api_key, ollama_base_url, ollama_model } = req.body;
const {
ai_enabled,
ai_verification_enabled,
ai_provider,
anthropic_api_key,
anthropic_model,
openai_api_key,
openai_model,
ollama_base_url,
ollama_model,
gemini_api_key,
gemini_model,
} = req.body;
const settings = await userQueries.updateAISettings(userId, {
ai_enabled,
ai_verification_enabled,
ai_provider,
anthropic_api_key,
anthropic_model,
openai_api_key,
openai_model,
ollama_base_url,
ollama_model,
gemini_api_key,
gemini_model,
});
if (!settings) {
@ -350,9 +370,13 @@ router.put('/ai', async (req: AuthRequest, res: Response) => {
ai_verification_enabled: settings.ai_verification_enabled || false,
ai_provider: settings.ai_provider || null,
anthropic_api_key: settings.anthropic_api_key || null,
anthropic_model: settings.anthropic_model || null,
openai_api_key: settings.openai_api_key || null,
openai_model: settings.openai_model || null,
ollama_base_url: settings.ollama_base_url || null,
ollama_model: settings.ollama_model || null,
gemini_api_key: settings.gemini_api_key || null,
gemini_model: settings.gemini_model || null,
message: 'AI settings updated successfully',
});
} catch (error) {
@ -438,4 +462,44 @@ router.post('/ai/test-ollama', async (req: AuthRequest, res: Response) => {
}
});
// Test Gemini API key
router.post('/ai/test-gemini', async (req: AuthRequest, res: Response) => {
try {
const { api_key } = req.body;
if (!api_key) {
res.status(400).json({ error: 'API key is required' });
return;
}
// Test the API key by listing models
const { GoogleGenerativeAI } = await import('@google/generative-ai');
const genAI = new GoogleGenerativeAI(api_key);
// Try to generate a simple response to verify the key works
const model = genAI.getGenerativeModel({ model: 'gemini-2.5-flash-lite' });
await model.generateContent('Say "API key valid" in 3 words or less');
res.json({
success: true,
message: 'Successfully connected to Gemini API',
});
} catch (error) {
console.error('Error testing Gemini connection:', error);
const errorMessage = error instanceof Error ? error.message : 'Unknown error';
if (errorMessage.includes('API_KEY_INVALID') || errorMessage.includes('API key')) {
res.status(400).json({
error: 'Invalid API key. Please check your Gemini API key.',
success: false,
});
} else {
res.status(500).json({
error: `Failed to connect to Gemini: ${errorMessage}`,
success: false,
});
}
}
});
export default router;