github/privateGPT
1
0
Fork 1
mirror of https://github.com/imartinez/privateGPT.git synced 2025-08-10 03:47:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

Tweaked RAG settings to pass lint, black and ruff checks

Browse Source
This commit is contained in:
Wesley Stewart 2024-03-15 12:11:10 +00:00
parent b78a68f40f
commit a3299769d5
2 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
private_gpt/settings/settings.py
View File

@ -107,6 +107,7 @@ class LLMSettings(BaseModel):
description="If set, any documents retrieved from the RAG must meet a certain score. Acceptable values are between 0 and 1.", description="If set, any documents retrieved from the RAG must meet a certain score. Acceptable values are between 0 and 1.",
) )
class RagSettings(BaseModel): class RagSettings(BaseModel):
similarity_value: float = Field( similarity_value: float = Field(
None, None,
@ -352,7 +353,6 @@ class Settings(BaseModel):
data: DataSettings data: DataSettings
ui: UISettings ui: UISettings
llm: LLMSettings llm: LLMSettings
rag: RagSettings
embedding: EmbeddingSettings embedding: EmbeddingSettings
llamacpp: LlamaCPPSettings llamacpp: LlamaCPPSettings
huggingface: HuggingFaceSettings huggingface: HuggingFaceSettings
@ -362,6 +362,7 @@ class Settings(BaseModel):
vectorstore: VectorstoreSettings vectorstore: VectorstoreSettings
qdrant: QdrantSettings | None = None qdrant: QdrantSettings | None = None
pgvector: PGVectorSettings | None = None pgvector: PGVectorSettings | None = None
rag: RagSettings
""" """

4
settings.yaml
View File

@ -41,7 +41,9 @@ llm:
context_window: 3900 context_window: 3900
temperature: 0.1 # The temperature of the model. Increasing the temperature will make the model answer more creatively. A value of 0.1 would be more factual. (Default: 0.1) temperature: 0.1 # The temperature of the model. Increasing the temperature will make the model answer more creatively. A value of 0.1 would be more factual. (Default: 0.1)
rag: #
# The rag settings are optional. Only include if you know what you are doing.
#rag:
#similarity_value: 0.4 #Optional - If set, any documents retrieved from the RAG must meet a certain score. Acceptable values are between 0 and 1. #similarity_value: 0.4 #Optional - If set, any documents retrieved from the RAG must meet a certain score. Acceptable values are between 0 and 1.
llamacpp: llamacpp: