Update app.py
Browse files
app.py
CHANGED
@@ -28,18 +28,8 @@ llm = Cerebras(model="llama-3.3-70b", api_key=os.environ["CEREBRAS_API_KEY"]) #
|
|
28 |
Settings.llm = llm # Ensure Cerebras is the LLM being used
|
29 |
|
30 |
# Initialize Voyage Embedding model
|
31 |
-
|
32 |
-
|
33 |
-
raise ValueError("VOYAGE_API_KEY is not set.")
|
34 |
-
else:
|
35 |
-
print("Voyage API key loaded successfully.")
|
36 |
-
|
37 |
-
model_name = "voyage-multimodal-3"
|
38 |
-
embed_model = VoyageEmbedding(
|
39 |
-
model_name=model_name,
|
40 |
-
voyage_api_key=voyage_api_key,
|
41 |
-
truncation=True
|
42 |
-
)
|
43 |
|
44 |
# Initialize Groq client for Whisper Large V3
|
45 |
groq_api_key = os.getenv("GROQ_API_KEY")
|
|
|
28 |
Settings.llm = llm # Ensure Cerebras is the LLM being used
|
29 |
|
30 |
# Initialize Voyage Embedding model
|
31 |
+
mixedbread_api_key = os.getenv("MXBAI_API_KEY")
|
32 |
+
embed_model = MixedbreadAIEmbedding(api_key=mixedbread_api_key, model_name="mixedbread-ai/mxbai-embed-large-v1")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
33 |
|
34 |
# Initialize Groq client for Whisper Large V3
|
35 |
groq_api_key = os.getenv("GROQ_API_KEY")
|