-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.py
47 lines (34 loc) · 1.39 KB
/
config.py
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
44
45
46
47
import os
import logging
from dotenv import load_dotenv
logging.getLogger().removeHandler("sentence_transformers")
logger = logging.getLogger(__name__)
logging.basicConfig(format='%(asctime)s - %(levelname)-8s | %(name)-10s | %(message)s',
level=logging.INFO)
logger.info('Loading environment variables')
load_dotenv()
# Global Variables
RAW_DATA_DIR = 'raw_data'
DATA_DIR = 'data'
GPT4ALL_PATH = os.environ.get('GPT4ALL_PATH')
LLAMA_7B_PATH = os.environ.get('LLAMA_7B_PATH')
LLAMA_13B_PATH = os.environ.get('LLAMA_13B_PATH')
ALPACA_7B_NATIVE_PATH = os.environ.get('ALPACA_7B_NATIVE_PATH')
# Twitter
TWTR_API_DEV = os.environ.get('TWTR_API_DEV')
TWTR_API_DEV_SECRET = os.environ.get('TWTR_API_DEV_SECRET')
TWTR_BEARER_DEV_TOKEN = os.environ.get('TWTR_BEARER_DEV_TOKEN')
TWTR_API = os.environ.get('TWTR_API')
TWTR_API_SECRET = os.environ.get('TWTR_API_SECRET')
TWTR_BEARER_TOKEN = os.environ.get('TWTR_BEARER_TOKEN')
TWTR_ACCESS_TOKEN = os.environ.get('TWTR_ACCESS_TOKEN')
TWTR_ACCESS_TOKEN_SECRET = os.environ.get('TWTR_ACCESS_TOKEN_SECRET')
# Reddit
REDDIT_APP_ID = os.environ.get('REDDIT_APP_ID')
REDDIT_SECRET = os.environ.get('REDDIT_SECRET')
REDDIT_REDIRECT_URL = os.environ.get('REDDIT_REDIRECT_URL')
REDDIT_USER_AGENT = os.environ.get('REDDIT_USER_AGENT')
# Huggingface
HUGGINGFACE_TOKEN = os.environ.get('HUGGINGFACE_TOKEN')
# OpenAI
OPENAI_API_KEY = os.environ.get('OPENAI_API_KEY')