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

Added embedded model option for fireworks \n Added documentation for fireworks

Browse source
This commit is contained in:
Somashekar B R 2024-09-24 23:27:40 +05:30
parent 519c48b70b
commit 9c3590e555
4 changed files with 154 additions and 79 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -74,11 +74,12 @@ class EmbeddingComponent:
raise ImportError(
"FireworksEmbedding dependencies not found, install with `poetry install --extras embeddings-fireworks`"
) from e
api_key = settings.fireworks.embedding_api_key or settings.fireworks.api_key
model = settings.openai.embedding_model
self.embedding_model = FireworksEmbedding(
api_key=api_key,
model=model,
)
case "ollama":
try: