diff --git a/.env.example b/.env.example index f6ff4c3..10ed4de 100644 --- a/.env.example +++ b/.env.example @@ -1,6 +1,7 @@ # We use PG by default as the metadata database. -METADATA_DB_URL=postgresql+psycopg2://alphatrion:alphatr1on@localhost:5432/alphatrion +ALPHATRION_METADATA_DB_URL=postgresql+psycopg2://alphatrion:alphatr1on@localhost:5432/alphatrion LOG_LEVEL=INFO -# Default: true, ARTIFACT_REGISTRY_URL should not be empty. -ENABLE_ARTIFACT_STORAGE=true -ARTIFACT_REGISTRY_URL=http://localhost:5000/ +# Default: true, ALPHATRION_ARTIFACT_REGISTRY_URL should not be empty. +ALPHATRION_ENABLE_ARTIFACT_STORAGE=true +ALPHATRION_ARTIFACT_REGISTRY_URL=http://localhost:5000/ +ALPHATRION_ENABLE_TRACING=true diff --git a/.env.integration-test b/.env.integration-test index 9fd4f67..a571227 100644 --- a/.env.integration-test +++ b/.env.integration-test @@ -1,3 +1,3 @@ -METADATA_DB_URL=postgresql+psycopg2://alphatrion:alphatr1on@localhost:5432/alphatrion -ARTIFACT_REGISTRY_URL=localhost:5001 +ALPHATRION_METADATA_DB_URL=postgresql+psycopg2://alphatrion:alphatr1on@localhost:5432/alphatrion +ALPHATRION_ARTIFACT_REGISTRY_URL=localhost:5001 LOG_LEVEL=INFO diff --git a/.env.test b/.env.test index 3a7d738..6b5ebff 100644 --- a/.env.test +++ b/.env.test @@ -1,3 +1,3 @@ -METADATA_DB_URL=sqlite:///:memory: -ARTIFACT_REGISTRY_URL=localhost:5000 +ALPHATRION_METADATA_DB_URL=sqlite:///:memory: +ALPHATRION_ARTIFACT_REGISTRY_URL=localhost:5000 LOG_LEVEL=INFO diff --git a/alphatrion/consts.py b/alphatrion/consts.py index c3c7880..86e9a9c 100644 --- a/alphatrion/consts.py +++ b/alphatrion/consts.py @@ -1,3 +1,4 @@ -METADATA_DB_URL = "METADATA_DB_URL" -ARTIFACT_REGISTRY_URL = "ARTIFACT_REGISTRY_URL" -ENABLE_ARTIFACT_STORAGE = "ENABLE_ARTIFACT_STORAGE" +METADATA_DB_URL = "ALPHATRION_METADATA_DB_URL" +ARTIFACT_REGISTRY_URL = "ALPHATRION_ARTIFACT_REGISTRY_URL" +ENABLE_ARTIFACT_STORAGE = "ALPHATRION_ENABLE_ARTIFACT_STORAGE" +ENABLE_TRACING = "ALPHATRION_ENABLE_TRACING" diff --git a/alphatrion/tracing/tracing.py b/alphatrion/tracing/tracing.py index 4255089..5e7fadc 100644 --- a/alphatrion/tracing/tracing.py +++ b/alphatrion/tracing/tracing.py @@ -9,10 +9,11 @@ from traceloop.sdk.decorators import task as _task from traceloop.sdk.decorators import workflow as _workflow +from alphatrion import consts from alphatrion.run.run import current_run_id # Disable tracing by default now -if os.getenv("ENABLE_TRACING", "false").lower() == "true": +if os.getenv(consts.ENABLE_TRACING, "false").lower() == "true": Traceloop.init( app_name="alphatrion", # TODO: make this configurable diff --git a/uv.lock b/uv.lock index 6319277..f99a412 100644 --- a/uv.lock +++ b/uv.lock @@ -129,7 +129,7 @@ wheels = [ [[package]] name = "alphatrion" -version = "0.0.5" +version = "0.1.0" source = { editable = "." } dependencies = [ { name = "alembic" },