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

Ingestion Speedup Multiple strategy (#1309)

Browse source
This commit is contained in:
lopagela 2023-11-25 20:12:09 +01:00 • committed by GitHub
parent 546ba33e6f
commit bafdd3baf1
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
13 changed files with 515 additions and 195 deletions
Download patch file Download diff file Expand all files Collapse all files

5
private_gpt/settings/settings.py
View file

@ -115,6 +115,10 @@ class LocalSettings(BaseModel):
)
class EmbeddingSettings(BaseModel):
mode: Literal["local", "openai", "sagemaker", "mock"]
class SagemakerSettings(BaseModel):
llm_endpoint_name: str
embedding_endpoint_name: str
@ -188,6 +192,7 @@ class Settings(BaseModel):
data: DataSettings
ui: UISettings
llm: LLMSettings
embedding: EmbeddingSettings
local: LocalSettings
sagemaker: SagemakerSettings
openai: OpenAISettings