import logging
from django.core.management import call_command
from smarter.common.conf import smarter_settings
from smarter.common.const import SMARTER_ACCOUNT_NUMBER, SmarterEnvironments
from smarter.lib.django.management.base import SmarterCommand
logger = logging.getLogger(__name__)
[docs]
class Command(SmarterCommand):
"""Django manage.py initialize_platform command."""
[docs]
def add_arguments(self, parser):
"""Add arguments to the command."""
parser.add_argument(
"--username",
type=str,
help="The username for the superuser associated with this platform. If not provided, the default 'admin' will be used.",
)
parser.add_argument(
"--email",
type=str,
help="The email address of the user to associate with this Secret. If not provided, a default email will be assigned.",
)
parser.add_argument(
"--password",
type=str,
help="The value to encrypt and persist. If not provided and you are running locally, the default 'smarter' will be used.",
)
# pylint: disable=broad-except
[docs]
def handle(self, *args, **options):
"""
Initialize the Smarter platform.
Creates the minimal resources necessary to start using Smarter.
1. Create an admin user with the provided username, email, and password.
2. Create example accounts and users.
3. Verify DNS configuration.
4. Load example projects from GitHub.
5. Add plugin examples.
6. Deploy example llmclients.
7. Initialize providers, and the built-in Proxies of their APIs.
8. Create StackAcademy SQL and API llmclients.
9. Apply manifests and update secrets for database connections.
10. Sync the ledger of infrastructure resources with the Kubernetes cluster.
"""
self.handle_begin()
username = options.get("username")
if not username:
username = "admin"
self.stdout.write(self.style.WARNING(f"No username provided, using the default value: {username}"))
email = options.get("email")
if not email:
email = f"{username}@{smarter_settings.root_domain}"
self.stdout.write(
self.style.WARNING(f"No email provided, assigning a default email for this Secret: {email}")
)
password = options.get("password")
if not password:
if smarter_settings.environment == SmarterEnvironments.LOCAL:
password = "smarter"
self.stdout.write(self.style.WARNING(f"No password provided, using the default value: {password}"))
else:
self.stdout.write(
self.style.ERROR("An administrator password is required when not running in LOCAL environment.")
)
return
# create Smarter Account and admin User.
# ---------------------------------------------------------------------
call_command("create_smarter_admin", username=username, password=password, email=email)
call_command(
"create_user",
account_number=SMARTER_ACCOUNT_NUMBER,
username="staff_user",
email=f"staff@{smarter_settings.root_domain}",
password=password,
first_name="Staff",
last_name="User",
admin=True,
)
call_command(
"create_user",
account_number=SMARTER_ACCOUNT_NUMBER,
username="customer_user",
email=f"customer@{smarter_settings.root_domain}",
password=password,
first_name="Customer",
last_name="User",
)
# Initialize Smarter platform components.
# ---------------------------------------------------------------------
try:
call_command("initialize_waffle") # Initialize builtin Waffle switches for feature flagging
except Exception as e:
logger.error("Failed to initialize Waffle switches: %s", e)
try:
call_command("add_builtin_guardrails")
except Exception as e:
logger.error("Failed to initialize Guardrails: %s", e)
try:
# applied, not deployed: an example CustomDomain, whose verification fails.
call_command("add_builtin_custom_domains")
except Exception as e:
logger.error("Failed to initialize CustomDomains: %s", e)
try:
call_command("add_builtin_budgets") # Detached budgets, which superusers can attach to resources
except Exception as e:
logger.error("Failed to initialize Budgets: %s", e)
try:
call_command("add_builtin_mcpclients")
except Exception as e:
logger.error("Failed to initialize MCPClients: %s", e)
try:
# queued for the Celery worker, which downloads the repositories from GitHub: the
# built-in smarter_project_* MCPClients read them through the platform's MCP server.
call_command("index_smarter_project", queue=True)
except Exception as e:
logger.error("Failed to queue the indexing of The Smarter Project's repositories: %s", e)
try:
call_command("add_builtin_llmhost_compute")
except Exception as e:
logger.error("Failed to initialize LLMHostCompute: %s", e)
try:
# after LLMHostCompute: each built-in LLMHost's spec.compute must exist.
call_command("add_builtin_llmhost")
except Exception as e:
logger.error("Failed to initialize LLMHosts: %s", e)
try:
call_command("initialize_providers") # Initialize builtin LLM providers: openai, metaai, googleia
except Exception as e:
logger.error("Failed to initialize providers: %s", e)
try:
# after initialize_providers: each built-in Proxy's Provider and API key Secret must exist.
call_command("add_builtin_proxies")
except Exception as e:
logger.error("Failed to initialize Proxies: %s", e)
try:
call_command("initialize_vectorstore_providers") # Initialize builtin vectorstore providers: pinecone
except Exception as e:
logger.error("Failed to initialize vectorstore providers: %s", e)
try:
call_command("add_builtin_vectorstores") # Applied, not deployed: example vector databases
except Exception as e:
logger.error("Failed to apply the built-in vectorstores: %s", e)
try:
call_command("add_plugin_examples", username=username)
except Exception as e:
logger.error("Failed to add plugin examples: %s", e)
try:
# after initialize_providers: the Stackademy LLMClients use a built-in Provider.
call_command("create_stackademy", account_number=SMARTER_ACCOUNT_NUMBER)
except Exception as e:
logger.error("Failed to create the Stackademy resources: %s", e)
try:
call_command(
"verify_dns_configuration"
) # if AWS is configured then Verify Route53 Hosted Zones and DNS records
except Exception as e:
logger.error("Failed to verify DNS configuration: %s", e)
try:
# read-only: record the Kubernetes cluster's resources in the ledger now, rather than
# 15 minutes later when Celery Beat first runs it. It exits on failure, e.g. no cluster.
call_command("sync_infrastructure_inventory")
except (Exception, SystemExit) as e:
logger.error("Failed to sync the infrastructure inventory: %s", e)
self.handle_completed_success()