import os
from ollama import Client

# 从环境变量获取地址，默认指向 compose 中的服务名
OLLAMA_URL = os.getenv("OLLAMA_HOST", "http://127.0.0.1:11434")
ollama_client = Client(host=OLLAMA_URL)

UPLOAD_DIR = "ai_uploads"
OUTPUT_DIR = "ai_outputs"
os.makedirs(UPLOAD_DIR, exist_ok=True)
os.makedirs(OUTPUT_DIR, exist_ok=True)