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

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

Browse source
This commit is contained in:
Yevhenii Semendiak 2024-05-09 00:42:55 +03:00
parent 659e0f1a9d
commit 90dadda249
2 changed files with 20 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

9
private_gpt/settings/settings.py
View file

@ -206,6 +206,15 @@ class OpenAISettings(BaseModel):
"gpt-3.5-turbo",
description="OpenAI Model to use. Example: 'gpt-4'.",
)
embedding_api_base: str = Field(
None,
description="Base URL of OpenAI API. Example: 'https://api.openai.com/v1'.",
)
embedding_api_key: str
embedding_model: str = Field(
"text-embedding-ada-002",
description="OpenAI Model to use. Example: 'gpt-4'.",
)
class OllamaSettings(BaseModel):