diff --git a/pipeline.py b/pipeline.py index bbcd712..ffad3f3 100644 --- a/pipeline.py +++ b/pipeline.py @@ -8,7 +8,11 @@ import threading from loguru import logger # --- Configuration --- -OLLAMA_URL = os.environ.get("OLLAMA_URL", "http://127.0.0.1:11434/api/generate") +from server.config import resolve_ollama_url + +OLLAMA_URL = resolve_ollama_url( + os.environ.get("OLLAMA_URL", "http://127.0.0.1:11434/api/generate") +) MODEL_NAME = os.environ.get("MODEL_NAME", "gemma4:e2b") PORT = int(os.environ.get("PORT", "8080")) POLL_INTERVAL = int(os.environ.get("POLL_INTERVAL", "21600")) # 6 hours diff --git a/reader.py b/reader.py index 1c76ce9..13a29f1 100644 --- a/reader.py +++ b/reader.py @@ -5,7 +5,11 @@ import httpx from loguru import logger from typing import List, Dict -OLLAMA_URL = os.environ.get("OLLAMA_URL", "http://127.0.0.1:11434/api/generate") +from server.config import resolve_ollama_url + +OLLAMA_URL = resolve_ollama_url( + os.environ.get("OLLAMA_URL", "http://127.0.0.1:11434/api/generate") +) MODEL_NAME = os.environ.get("MODEL_NAME", "gemma4:e2b") DATA_DIR = os.environ.get("DATA_DIR", "data") INPUT_FILE = os.path.join(DATA_DIR, "output.json") diff --git a/server/app.py b/server/app.py index c0c40e8..724f9a0 100644 --- a/server/app.py +++ b/server/app.py @@ -74,12 +74,12 @@ def _generate_index(mp3_files, scripts_map): audio_entries += f"""