mirror of
https://github.com/ParisNeo/ollama_proxy_server.git
synced 2026-04-22 03:00:03 -04:00
b5c2dfdade
Update all project references, documentation, and assets to reflect the new branding. Changes include: - Rename application references in README, CONTRIBUTING, DEVELOPMENT - Update default database filename from ollama_proxy.db to lollms_hub.db - Update .env.example header and bootstrap configuration - Adjust .gitignore for new database name - No functional code changes, purely cosmetic rebranding
94 lines
3.8 KiB
Python
94 lines
3.8 KiB
Python
import sys
|
|
import secrets
|
|
import getpass
|
|
|
|
def print_info(message):
|
|
print(f"[INFO] {message}")
|
|
|
|
def print_error(message):
|
|
print(f"[ERROR] {message}", file=sys.stderr)
|
|
|
|
def get_user_input(prompt, default=None):
|
|
"""Gets user input with a default value."""
|
|
default_text = f"[{default}]" if default is not None else ""
|
|
prompt_text = f" -> {prompt} {default_text}: "
|
|
user_value = input(prompt_text)
|
|
return user_value.strip() if user_value.strip() else default
|
|
|
|
def get_secure_input(prompt):
|
|
"""Gets user input without showing it on the screen."""
|
|
prompt_text = f" -> {prompt}: "
|
|
return getpass.getpass(prompt_text).strip()
|
|
|
|
def create_env_file():
|
|
"""
|
|
Guides the user through creating a .env file and writes it.
|
|
This version omits empty optional fields to prevent parsing errors.
|
|
"""
|
|
print_info("Please provide the following configuration (press Enter for defaults):")
|
|
|
|
config = {
|
|
"PROXY_PORT": get_user_input("Port for the proxy server", "8080"),
|
|
"OLLAMA_SERVERS": get_user_input("Backend Ollama server(s)", "http://127.0.0.1:11434"),
|
|
"REDIS_HOST": get_user_input("Redis Host", "localhost"),
|
|
"REDIS_PORT": get_user_input("Redis Port", "6379"),
|
|
"REDIS_USERNAME": get_user_input("Redis Username (optional)", ""),
|
|
"ADMIN_USER": get_user_input("Username for the admin dashboard", "admin"),
|
|
"ALLOWED_IPS": get_user_input("Allowed IPs (comma-separated, leave empty for all)", ""),
|
|
"DENIED_IPS": get_user_input("Denied IPs (comma-separated, leave empty for none)", ""),
|
|
}
|
|
|
|
# --- Secure Password Inputs ---
|
|
config["REDIS_PASSWORD"] = get_secure_input("Redis Password (optional, will be hidden)")
|
|
|
|
admin_password = ""
|
|
while not admin_password:
|
|
admin_password = get_secure_input("Password for the admin user (cannot be empty, will be hidden)")
|
|
if not admin_password:
|
|
print_error(" Admin password cannot be empty. Please try again.")
|
|
config["ADMIN_PASSWORD"] = admin_password
|
|
|
|
# --- Generate Secret Key ---
|
|
config["SECRET_KEY"] = secrets.token_hex(32)
|
|
|
|
# --- Write the .env file ---
|
|
print_info("Generating .env configuration file...")
|
|
try:
|
|
with open(".env", "w", encoding="utf-8") as f:
|
|
f.write('APP_NAME="lollms hub"\n')
|
|
f.write('APP_VERSION="9.0.0"\n')
|
|
f.write('LOG_LEVEL="info"\n')
|
|
f.write(f'PROXY_PORT="{config["PROXY_PORT"]}"\n')
|
|
f.write(f'OLLAMA_SERVERS="{config["OLLAMA_SERVERS"]}"\n')
|
|
f.write('DATABASE_URL="sqlite+aiosqlite:///./lollms_hub.db"\n')
|
|
f.write(f'ADMIN_USER="{config["ADMIN_USER"]}"\n')
|
|
f.write(f'ADMIN_PASSWORD="{config["ADMIN_PASSWORD"]}"\n')
|
|
f.write(f'SECRET_KEY="{config["SECRET_KEY"]}"\n')
|
|
|
|
# Write new Redis variables
|
|
f.write(f'REDIS_HOST="{config["REDIS_HOST"]}"\n')
|
|
f.write(f'REDIS_PORT="{config["REDIS_PORT"]}"\n')
|
|
if config["REDIS_USERNAME"]:
|
|
f.write(f'REDIS_USERNAME="{config["REDIS_USERNAME"]}"\n')
|
|
if config["REDIS_PASSWORD"]:
|
|
f.write(f'REDIS_PASSWORD="{config["REDIS_PASSWORD"]}"\n')
|
|
|
|
f.write('RATE_LIMIT_REQUESTS="100"\n')
|
|
f.write('RATE_LIMIT_WINDOW_MINUTES="1"\n')
|
|
|
|
if config["ALLOWED_IPS"]:
|
|
f.write(f'ALLOWED_IPS="{config["ALLOWED_IPS"]}"\n')
|
|
if config["DENIED_IPS"]:
|
|
f.write(f'DENIED_IPS="{config["DENIED_IPS"]}"\n')
|
|
|
|
print_info(".env file created successfully.")
|
|
return True
|
|
except IOError as e:
|
|
print_error(f"Failed to write to .env file: {e}")
|
|
return False
|
|
|
|
if __name__ == "__main__":
|
|
if not create_env_file():
|
|
# Exit with a non-zero code to signal failure to the batch script
|
|
sys.exit(1)
|