config.py 360 B

123456789101112131415
  1. import sys
  2. import os
  3. from dotenv import load_dotenv, find_dotenv
  4. load_dotenv(find_dotenv())
  5. ENV = os.environ.get("ENV", "dev")
  6. OLLAMA_API_BASE_URL = os.environ.get(
  7. "OLLAMA_API_BASE_URL", "http://localhost:11434/api"
  8. )
  9. if ENV == "prod":
  10. if OLLAMA_API_BASE_URL == "/ollama/api":
  11. OLLAMA_API_BASE_URL = "http://host.docker.internal:11434/api"