SystemSh0cker/private-gpt
1
0
Fork 0
mirror of https://github.com/zylon-ai/private-gpt.git synced 2025-12-22 13:55:41 +01:00
Code Issues Projects Releases Packages Activity

Update embedding_component.py

Browse source
This commit is contained in:
Liam Dowd 2024-08-28 08:14:43 -07:00 • committed by GitHub
parent f27272fb5c
commit 8577de8430
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
private_gpt/components/embedding/embedding_component.py
View file

@ -144,7 +144,7 @@ class EmbeddingComponent:
api_key=settings.gemini.api_key,
model_name=settings.gemini.embedding_model,
)
case "mistral":
case "mistralai":
try:
from llama_index.embeddings.mistralai import ( # type: ignore
MistralAIEmbedding,
@ -154,8 +154,8 @@ class EmbeddingComponent:
"Mistral dependencies not found, install with `poetry install --extras embeddings-mistral`"
) from e
api_key = settings.mistral.api_key
model = settings.mistral.embedding_model
api_key = settings.mistralai.api_key
model = settings.mistralai.embedding_model
self.embedding_model = MistralAIEmbedding(
api_key=api_key,