Skip to content

Commit fc07aba

Browse files
authored
Merge pull request #142 from GitHubSecurityLab/p--banner
print GHSL banner before taskflow runs
2 parents d7fa98c + f54a3ad commit fc07aba

2 files changed

Lines changed: 40 additions & 0 deletions

File tree

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 (
@@ -677,4 +678,5 @@ async def _deploy_task_agents(resolved_agents, prompt):
677678
print(help_msg)
678679
sys.exit(1)
679680

681+
print(get_banner()) # print banner only before starting main event loop
680682
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)