github/privateGPT
1
0
Fork 1
mirror of https://github.com/imartinez/privateGPT.git synced 2025-09-01 15:19:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

Allow parameterizing OpenAI embeddings component (api_base, key, model) (#1920)

Browse Source 
* Allow parameterizing OpenAI embeddings component (api_base, key, model)

* Update settings

* Update description
This commit is contained in:
Yevhenii Semendiak
2024-05-17 10:52:50 +03:00
committed by GitHub
parent 45df99feb7
commit 3b3e96ad6c
3 changed files with 21 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
private_gpt/components/embedding/embedding_component.py
View File

@@ -55,8 +55,17 @@ class EmbeddingComponent:
"OpenAI dependencies not found, install with `poetry install --extras embeddings-openai`"
) from e
openai_settings = settings.openai.api_key
self.embedding_model = OpenAIEmbedding(api_key=openai_settings)
api_base = (
settings.openai.embedding_api_base or settings.openai.api_base
)
api_key = settings.openai.embedding_api_key or settings.openai.api_key
model = settings.openai.embedding_model
self.embedding_model = OpenAIEmbedding(
api_base=api_base,
api_key=api_key,
model=model,
)
case "ollama":
try:
from llama_index.embeddings.ollama import ( # type: ignore