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

Update embedding_component.py

Browse source
This commit is contained in:
Liam Dowd 2024-08-28 10:32:42 -07:00 • committed by GitHub
parent f3fda24ae0
commit 5709b6009e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -154,8 +154,8 @@ class EmbeddingComponent:
"Mistral dependencies not found, install with `poetry install --extras embeddings-mistral`" "Mistral dependencies not found, install with `poetry install --extras embeddings-mistral`"
) from e ) from e
api_key = settings.mistral.api_key api_key = settings.openai.api_key
model = settings.mistral.embedding_model model = settings.openai.embedding_model
self.embedding_model = MistralAIEmbedding( self.embedding_model = MistralAIEmbedding(
api_key=api_key, api_key=api_key,