Switch embedding model from Google to NVIDIA NIM (llama-nemotron-embed-1b-v2)

app.py

@@ -228,7 +228,7 @@ def show_home():
             </div>
             <div style="border-left: 1px solid #eee;">
                 <h4>🧬 Models</h4>
-                <p style="font-size: 0.85rem; color: #666;">NVIDIA Nemotron-3 (Reasoning)<br>Google Embedding-001 (Vector)<br>Gemini 2.5 Flash (Audio/TTS)</p>
+                <p style="font-size: 0.85rem; color: #666;">NVIDIA Nemotron-3 (Reasoning)<br>NVIDIA Nemotron Embed-1B (Vector)<br>Gemini 2.5 Flash (Audio/TTS)</p>
             </div>
         </div>
     </div>

core/models.py

@@ -1,7 +1,6 @@
 import os
 import streamlit as st
-from langchain_nvidia_ai_endpoints import ChatNVIDIA
-from langchain_google_genai import GoogleGenerativeAIEmbeddings, ChatGoogleGenerativeAI
+from langchain_nvidia_ai_endpoints import ChatNVIDIA, NVIDIAEmbeddings
 from google import genai
 
 def get_llm(model_name: str = "nvidia/nemotron-3-nano-30b-a3b"):
@@ -22,14 +21,14 @@ def get_llm(model_name: str = "nvidia/nemotron-3-nano-30b-a3b"):
     )
 
 def get_embeddings():
-    api_key = os.getenv("GOOGLE_API_KEY")
-    if not api_key and "GOOGLE_API_KEY" in st.secrets:
-        api_key = st.secrets["GOOGLE_API_KEY"]
-        
+    api_key = os.getenv("NV_API_KEY")
+    if not api_key and "NV_API_KEY" in st.secrets:
+        api_key = st.secrets["NV_API_KEY"]
+
     if not api_key:
-        raise ValueError("GOOGLE_API_KEY not found in environment or secrets.")
-        
-    return GoogleGenerativeAIEmbeddings(model="models/embedding-001", google_api_key=api_key)
+        raise ValueError("NV_API_KEY not found in environment or secrets.")
+
+    return NVIDIAEmbeddings(model="nvidia/llama-nemotron-embed-1b-v2", api_key=api_key)
 
 from google.genai import types