Update app.py

app.py

@@ -4,7 +4,7 @@ import warnings
 import asyncio
 from llama_index.core import VectorStoreIndex, SimpleDirectoryReader, Document, Settings
 from llama_index.llms.cerebras import Cerebras
-from llama_index.embeddings.huggingface import HuggingFaceEmbedding
+from llama_index.embeddings.voyageai import VoyageEmbedding
 from groq import Groq
 import io
 
@@ -26,7 +26,20 @@ else:
 os.environ["CEREBRAS_API_KEY"] = api_key
 llm = Cerebras(model="llama-3.3-70b", api_key=os.environ["CEREBRAS_API_KEY"])  # Change model to Llama3.1-70b from Cerebras
 Settings.llm = llm  # Ensure Cerebras is the LLM being used
-embed_model = HuggingFaceEmbedding(model_name="sentence-transformers/all-mpnet-base-v2")
+
+# Initialize Voyage Embedding model
+voyage_api_key = os.getenv("VOYAGE_API_KEY")
+if not voyage_api_key:
+    raise ValueError("VOYAGE_API_KEY is not set.")
+else:
+    print("Voyage API key loaded successfully.")
+
+model_name = "voyage-multimodal-3"
+embed_model = VoyageEmbedding(
+    model_name=model_name,
+    voyage_api_key=voyage_api_key,
+    truncation=True
+)
 
 # Initialize Groq client for Whisper Large V3
 groq_api_key = os.getenv("GROQ_API_KEY")