Skip to content

Commit 3305881

Browse files
authored
Merge branch 'main' into p--fix-cmd-samples
2 parents 86fd178 + fc07aba commit 3305881

File tree

2 files changed

+40
-0
lines changed

2 files changed

+40
-0
lines changed

src/seclab_taskflow_agent/__main__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@
2727

2828
from .agent import DEFAULT_MODEL, TaskAgent, TaskAgentHooks, TaskRunHooks
2929
from .available_tools import AvailableTools
30+
from .banner import get_banner
3031
from .capi import get_AI_token, list_tool_call_models
3132
from .env_utils import TmpEnv
3233
from .mcp_utils import (
@@ -676,4 +677,5 @@ async def _deploy_task_agents(resolved_agents, prompt):
676677
print(help_msg)
677678
sys.exit(1)
678679

680+
print(get_banner()) # print banner only before starting main event loop
679681
asyncio.run(main(available_tools, p, t, cli_globals, user_prompt), debug=True)
Lines changed: 38 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,38 @@
1+
# SPDX-FileCopyrightText: 2026 GitHub
2+
# SPDX-License-Identifier: MIT
3+
4+
from .capi import get_AI_endpoint
5+
6+
def get_banner():
7+
api_endpoint = get_AI_endpoint()
8+
banner = f"""
9+
╔══════════════════════════════════════════════════════════════════╗
10+
║ ║
11+
║ ██████╗ ██╗████████╗██╗ ██╗██╗ ██╗██████╗ ║
12+
║ ██╔════╝ ██║╚══██╔══╝██║ ██║██║ ██║██╔══██╗ ║
13+
║ ██║ ███╗██║ ██║ ███████║██║ ██║██████╔╝ ║
14+
║ ██║ ██║██║ ██║ ██╔══██║██║ ██║██╔══██╗ ║
15+
║ ╚██████╔╝██║ ██║ ██║ ██║╚██████╔╝██████╔╝ ║
16+
║ ╚═════╝ ╚═╝ ╚═╝ ╚═╝ ╚═╝ ╚═════╝ ╚═════╝ ║
17+
║ ║
18+
║ ███████╗███████╗ ██████╗██╗ ██╗██████╗ ██╗████████╗██╗ ██╗ ║
19+
║ ██╔════╝██╔════╝██╔════╝██║ ██║██╔══██╗██║╚══██╔══╝╚██╗ ██╔╝ ║
20+
║ ███████╗█████╗ ██║ ██║ ██║██████╔╝██║ ██║ ╚████╔╝ ║
21+
║ ╚════██║██╔══╝ ██║ ██║ ██║██╔══██╗██║ ██║ ╚██╔╝ ║
22+
║ ███████║███████╗╚██████╗╚██████╔╝██║ ██║██║ ██║ ██║ ║
23+
║ ╚══════╝╚══════╝ ╚═════╝ ╚═════╝ ╚═╝ ╚═╝╚═╝ ╚═╝ ╚═╝ ║
24+
║ ║
25+
║ ██╗ █████╗ ██████╗ ║
26+
║ ██║ ██╔══██╗██╔══██╗ ║
27+
║ ██║ ███████║██████╔╝ ║
28+
║ ██║ ██╔══██║██╔══██╗ ║
29+
║ ███████╗██║ ██║██████╔╝ ║
30+
║ ╚══════╝╚═╝ ╚═╝╚═════╝ ║
31+
║ ║
32+
║ TASKFLOW AGENT ║
33+
║ ║
34+
╠══════════════════════════════════════════════════════════════════╣
35+
║ AI API Endpoint: {api_endpoint:<48}
36+
╚══════════════════════════════════════════════════════════════════╝
37+
"""
38+
return banner

0 commit comments

Comments
 (0)