rdune71's picture
Add requirements, environment configuration, and config loader
317d1fd
raw
history blame
789 Bytes
import os
from dotenv import load_dotenv
class Config:
def __init__(self):
load_dotenv()
self.hf_token = os.getenv("HF_TOKEN")
self.hf_api_url = os.getenv("HF_API_ENDPOINT_URL")
self.tavily_api_key = os.getenv("TAVILY_API_KEY")
self.openweather_api_key = os.getenv("OPENWEATHER_API_KEY")
self.nasa_api_key = os.getenv("NASA_API_KEY")
self.redis_host = os.getenv("REDIS_HOST")
self.redis_port = int(os.getenv("REDIS_PORT", "6379"))
self.redis_username = os.getenv("REDIS_USERNAME")
self.redis_password = os.getenv("REDIS_PASSWORD")
self.local_model_name = os.getenv("LOCAL_MODEL_NAME", "mistral-7b")
self.ollama_host = os.getenv("OLLAMA_HOST", "http://localhost:11434")
config = Config()