rag_agent / config.py
Cheh Kit Hong
changed rag method flags
94e0eef
raw
history blame contribute delete
477 Bytes
import os
configs = {
"DATA_PATH": "./docs/markdowns",
"PERSIST_PATH": "./chroma_data", # modify if want to create another vectorstore like chroma_data_2
"EMBEDDING_MODEL_NAME": "sentence-transformers/all-mpnet-base-v2",
"COLLECTION_NAME": "langchain_mpnet_collection",
"LLM_MODEL_NAME": "gemini-2.0-flash",
"TEMPERATURE": 0.2,
"MAX_TOKENS": 2048,
}
if __name__ == "__main__":
for key, value in configs.items():
os.environ[key] = value