rdune71's picture
Update system to use remote Ollama host with graceful fallback handling
f09ddb7
raw
history blame
884 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")
# Use remote Ollama host as default, falling back to environment variable
self.ollama_host = os.getenv("OLLAMA_HOST", "https://a877ef1aa487.ngrok-free.app")
config = Config()