File size: 411 Bytes
067cdc9 0fc97a4 067cdc9 aa018e3 067cdc9 0fc97a4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
import os
configs = {
"DATA_PATH": "./docs/markdowns",
"PERSIST_PATH": "./chroma_data",
"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 |