Add knowledge base RAG module for book Q&A
This commit is contained in:
@@ -15,7 +15,7 @@ import ai.parser as ai_parser
|
||||
from ai.parser import client
|
||||
|
||||
# Configuration
|
||||
EPUBS_DIRECTORY = os.getenv("KNOWLEDGE_EMBEDDINGS_DIR", "../embedding-generator/epubs")
|
||||
EPUBS_DIRECTORY = os.getenv("KNOWLEDGE_EMBEDDINGS_DIR", "./bot/data")
|
||||
TOP_K_CHUNKS = 5
|
||||
EMBEDDING_MODEL = "sentence-transformers/all-minilm-l12-l2"
|
||||
CHAT_MODEL = "deepseek/deepseek-v3.2"
|
||||
|
||||
Reference in New Issue
Block a user