Integrate Gemini 1.0 Pro Chat Model and Google Embedding -001 Model for Enhanced Perplexica Capabilities

package.json

@@ -29,6 +29,7 @@
     "@langchain/anthropic": "^0.2.3",
     "@langchain/community": "^0.2.16",
     "@langchain/openai": "^0.0.25",
+    "@langchain/google-genai": "^0.0.23",
     "@xenova/transformers": "^2.17.1",
     "axios": "^1.6.8",
     "better-sqlite3": "^11.0.0",

src/config.ts

@@ -13,6 +13,7 @@ interface Config {
     OPENAI: string;
     GROQ: string;
     ANTHROPIC: string;
+    Gemini: string;
   };
   API_ENDPOINTS: {
     SEARXNG: string;
@@ -40,6 +41,11 @@ export const getGroqApiKey = () => loadConfig().API_KEYS.GROQ;
 
 export const getAnthropicApiKey = () => loadConfig().API_KEYS.ANTHROPIC;
 
+
+export const getGeminiApiKey = () => loadConfig().API_KEYS.Gemini;
+
+export const getSearxngApiEndpoint = () => loadConfig().API_ENDPOINTS.SEARXNG;
+
 export const getSearxngApiEndpoint = () =>
   process.env.SEARXNG_API_URL || loadConfig().API_ENDPOINTS.SEARXNG;
 

src/lib/providers/gemini.ts

@@ -0,0 +1,45 @@
+import { ChatGoogleGenerativeAI } from '@langchain/google-genai';
+import { GoogleGenerativeAIEmbeddings } from '@langchain/google-genai';
+import { getGeminiApiKey } from '../../config';
+import logger from '../../utils/logger';
+
+export const loadGeminiChatModels = async () => {
+  const geminiApiKey = getGeminiApiKey();
+
+  if (!geminiApiKey) return {};
+
+  try {
+    const chatModels = {
+      'Gemini Pro': new ChatGoogleGenerativeAI({
+        temperature: 0.7,
+        apiKey: geminiApiKey,
+        modelName: 'gemini-pro',
+      }),
+    };
+
+    return chatModels;
+  } catch (err) {
+    logger.error(`Error loading Gemini chat models: ${err}`);
+    return {};
+  }
+};
+
+export const loadGeminiEmbeddingsModels = async () => {
+  const geminiApiKey = getGeminiApiKey();
+
+  if (!geminiApiKey) return {};
+
+  try {
+    const embeddingsModels = {
+      'Gemini Embedding': new GoogleGenerativeAIEmbeddings({
+        apiKey: geminiApiKey,
+        modelName: 'embedding-001',
+      }),
+    };
+
+    return embeddingsModels;
+  } catch (err) {
+    logger.error(`Error loading Gemini embeddings model: ${err}`);
+    return {};
+  }
+};

src/lib/providers/index.ts

@@ -9,12 +9,14 @@ const chatModelProviders = {
   groq: loadGroqChatModels,
   ollama: loadOllamaChatModels,
   anthropic: loadAnthropicChatModels,
+  gemini : loadGeminiChatModels
 };
 
 const embeddingModelProviders = {
   openai: loadOpenAIEmbeddingsModels,
   local: loadTransformersEmbeddingsModels,
   ollama: loadOllamaEmbeddingsModels,
+  gemini : loadGeminiEmbeddingsModels
 };
 
 export const getAvailableChatModelProviders = async () => {

src/routes/config.ts

@@ -7,6 +7,7 @@ import {
   getGroqApiKey,
   getOllamaApiEndpoint,
   getAnthropicApiKey,
+  getGeminiApiKey,
   getOpenaiApiKey,
   updateConfig,
 } from '../config';
@@ -40,6 +41,7 @@ router.get('/', async (_, res) => {
   config['ollamaApiUrl'] = getOllamaApiEndpoint();
   config['anthropicApiKey'] = getAnthropicApiKey();
   config['groqApiKey'] = getGroqApiKey();
+  config['geminiApiKey'] = getGeminiApiKey();
 
   res.status(200).json(config);
 });
@@ -52,6 +54,7 @@ router.post('/', async (req, res) => {
       OPENAI: config.openaiApiKey,
       GROQ: config.groqApiKey,
       ANTHROPIC: config.anthropicApiKey,
+      Gemini: config.geminiApiKey,
     },
     API_ENDPOINTS: {
       OLLAMA: config.ollamaApiUrl,

ui/components/SettingsDialog.tsx

@@ -57,6 +57,7 @@ interface SettingsType {
   openaiApiKey: string;
   groqApiKey: string;
   anthropicApiKey: string;
+  geminiApiKey: string;
   ollamaApiUrl: string;
 }
 
@@ -460,6 +461,22 @@ const SettingsDialog = ({
                         }
                       />
                     </div>
+                    <div className="flex flex-col space-y-1">
+                      <p className="text-black/70 dark:text-white/70 text-sm">
+                        Gemini API Key
+                      </p>
+                      <Input
+                        type="text"
+                        placeholder="Gemini API key"
+                        defaultValue={config.geminiApiKey}
+                        onChange={(e) =>
+                          setConfig({
+                            ...config,
+                            geminiApiKey: e.target.value,
+                          })
+                        }
+                      />
+                    </div>
                   </div>
                 )}
                 {isLoading && (