Skip to content

Commit

Permalink
perf: update setting
Browse files Browse the repository at this point in the history
  • Loading branch information
datvodinh committed May 13, 2024
1 parent 25640b7 commit 7e8640d
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions rag_chatbot/setting/setting.py
Original file line number Diff line number Diff line change
Expand Up @@ -46,13 +46,13 @@ class RetrieverSettings(BaseModel):
default=5, description="Number of generated queries"
)
similarity_top_k: int = Field(
default=10, description="Top k documents"
default=20, description="Top k documents"
)
retriever_weights: List[float] = Field(
default=[0.4, 0.6], description="Weights for retriever"
)
top_k_rerank: int = Field(
default=3, description="Top k rerank"
default=6, description="Top k rerank"
)
rerank_llm: str = Field(
default="BAAI/bge-reranker-large", description="Rerank LLM model"
Expand All @@ -67,16 +67,16 @@ class IngestionSettings(BaseModel):
default="BAAI/bge-large-en-v1.5", description="Embedding LLM model"
)
embed_batch_size: int = Field(
default=4, description="Embedding batch size"
default=8, description="Embedding batch size"
)
cache_folder: str = Field(
default="data/huggingface", description="Cache folder"
)
chunk_size: int = Field(
default=1024, description="Document chunk size"
default=512, description="Document chunk size"
)
chunk_overlap: int = Field(
default=128, description="Document chunk overlap"
default=32, description="Document chunk overlap"
)
chunking_regex: str = Field(
default="[^,.;。?!]+[,.;。?!]?", description="Chunking regex"
Expand Down

0 comments on commit 7e8640d

Please sign in to comment.