diff --git a/.github/workflows/kube-workflow-init.yaml b/.github/workflows/kube-workflow-init.yaml index 69b2de6..b98b7d4 100644 --- a/.github/workflows/kube-workflow-init.yaml +++ b/.github/workflows/kube-workflow-init.yaml @@ -6,6 +6,6 @@ on: jobs: init: # Use a released version for stable. - uses: kerthcet/github-workflow-as-kube/.github/workflows/workflow-as-kubernetes-init.yaml@v0.1.21 + uses: kerthcet/github-workflow-as-kube/.github/workflows/workflow-as-kubernetes-init.yaml@v0.1.22 secrets: AGENT_TOKEN: ${{ secrets.AGENT_TOKEN }} diff --git a/.github/workflows/kube-workflow.yaml b/.github/workflows/kube-workflow.yaml index 095f018..7c5d6ce 100644 --- a/.github/workflows/kube-workflow.yaml +++ b/.github/workflows/kube-workflow.yaml @@ -20,6 +20,6 @@ on: jobs: event-handler: # Use a released version for stable. - uses: kerthcet/github-workflow-as-kube/.github/workflows/workflow-as-kubernetes.yaml@v0.1.21 + uses: kerthcet/github-workflow-as-kube/.github/workflows/workflow-as-kubernetes.yaml@v0.1.22 secrets: AGENT_TOKEN: ${{ secrets.AGENT_TOKEN }} diff --git a/alphatrion/server/cmd/main.py b/alphatrion/server/cmd/main.py index 5183286..ebff3ea 100644 --- a/alphatrion/server/cmd/main.py +++ b/alphatrion/server/cmd/main.py @@ -1,3 +1,4 @@ +# ruff: noqa: E501 import argparse import os import webbrowser @@ -54,6 +55,25 @@ def main(): def run_server(args): + BLUE = "\033[94m" + RESET = "\033[0m" + + ascii_art = r""" + █████████ ████ █████ ███████████ ███ + ███░░░░░███ ░░███ ░░███ ░█░░░███░░░█ ░░░ + ░███ ░███ ░███ ████████ ░███████ ██████ ░ ░███ ░ ████████ ████ ██████ ████████ + ░███████████ ░███ ░░███░░███ ░███░░███ ░░░░░███ ░███ ░░███░░███░░███ ███░░███░░███░░███ + ░███░░░░░███ ░███ ░███ ░███ ░███ ░███ ███████ ░███ ░███ ░░░ ░███ ░███ ░███ ░███ ░███ + ░███ ░███ ░███ ░███ ░███ ░███ ░███ ███░░███ ░███ ░███ ░███ ░███ ░███ ░███ ░███ + █████ █████ █████ ░███████ ████ █████░░████████ █████ █████ █████░░██████ ████ █████ + ░░░░░ ░░░░░ ░░░░░ ░███░░░ ░░░░ ░░░░░ ░░░░░░░░ ░░░░░ ░░░░░ ░░░░░ ░░░░░░ ░░░░ ░░░░░ + ░███ + █████ + ░░░░░ + """ + + print(f"{BLUE}{ascii_art}{RESET}") + msg = Text( f"Starting AlphaTrion server at http://{args.host}:{args.port}", style="bold green", diff --git a/alphatrion/tracing/tracing.py b/alphatrion/tracing/tracing.py index f67d935..4255089 100644 --- a/alphatrion/tracing/tracing.py +++ b/alphatrion/tracing/tracing.py @@ -1,7 +1,9 @@ import inspect +import os import uuid from functools import wraps +from opentelemetry.sdk.trace.export import ConsoleSpanExporter from opentelemetry.semconv_ai import TraceloopSpanKindValues from traceloop.sdk import Traceloop from traceloop.sdk.decorators import task as _task @@ -9,13 +11,15 @@ from alphatrion.run.run import current_run_id -Traceloop.init( - app_name="alphatrion", - # TODO: make this configurable - # exporter=ConsoleSpanExporter(), - disable_batch=True, - telemetry_enabled=False, -) +# Disable tracing by default now +if os.getenv("ENABLE_TRACING", "false").lower() == "true": + Traceloop.init( + app_name="alphatrion", + # TODO: make this configurable + exporter=ConsoleSpanExporter(), + disable_batch=True, + telemetry_enabled=False, + ) def task(