File size: 1,091 Bytes
f9871be 3fc834d f9871be 3fc834d 85ce539 f9871be 85ce539 f9871be 3fc834d f9871be 3fc834d f9871be 3fc834d f9871be 3fc834d f9871be 85ce539 f9871be 3fc834d f9871be 3fc834d f9871be 3fc834d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 |
# config.py
import os
import sys
import logging
from pathlib import Path
BASE_DIR = Path(__file__).resolve().parent
if str(BASE_DIR) not in sys.path:
sys.path.insert(0, str(BASE_DIR))
try:
import google.colab
IN_COLAB = True
except ImportError:
IN_COLAB = False
if IN_COLAB:
VECTORSTORE_DIR = Path("/content/drive/MyDrive/bioinformatics_tutor_ai/vectorstore")
else:
VECTORSTORE_DIR = BASE_DIR / "vectorstore"
# Models
EMBEDDING_MODEL = "sentence-transformers/all-MiniLM-L6-v2"
LLM_MODEL = "BioMistral/BioMistral-7B-DARE"
# Confidence threshold for TutorAgent
CONFIDENCE_THRESHOLD = 0.65
OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY")
if not OPENAI_API_KEY:
logging.warning("OPENAI_API_KEY not set! AutoGen Tutor Agent will not work without it.")
if "HF_HOME" not in os.environ:
os.environ["HF_HOME"] = "/home/user/.cache/huggingface"
os.environ["TRANSFORMERS_CACHE"] = os.path.join(os.environ["HF_HOME"], "transformers")
os.environ["HF_HUB_CACHE"] = os.path.join(os.environ["HF_HOME"], "hub")
|