Add .env file loading — cp .env.example .env and edit
This commit is contained in:
10
docgen.py
10
docgen.py
@@ -4,6 +4,16 @@ import requests
|
||||
import os
|
||||
import concurrent.futures
|
||||
import time
|
||||
from pathlib import Path
|
||||
|
||||
# Load .env if present
|
||||
_env_file = Path(__file__).parent / ".env"
|
||||
if _env_file.exists():
|
||||
for line in _env_file.read_text().splitlines():
|
||||
line = line.strip()
|
||||
if line and not line.startswith("#") and "=" in line:
|
||||
key, _, val = line.partition("=")
|
||||
os.environ.setdefault(key.strip(), val.strip())
|
||||
|
||||
# Backend: "ollama" or "openai"
|
||||
LLM_BACKEND = os.environ.get("LLM_BACKEND", "ollama")
|
||||
|
||||
Reference in New Issue
Block a user