Skip to content

Commit f306663

Browse files
authored
Merge pull request #278 from GitHubSecurityLab/p--version-banner
feat: display version in banner
2 parents ddef201 + e8aa511 commit f306663

2 files changed

Lines changed: 17 additions & 1 deletion

File tree

src/seclab_taskflow_agent/banner.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@
33

44
"""ASCII banner displayed at agent startup."""
55

6+
from .__about__ import __version__
67
from .capi import get_AI_endpoint
78

89
__all__ = ["get_banner"]
@@ -35,7 +36,7 @@ def get_banner() -> str:
3536
║ ███████╗██║ ██║██████╔╝ ║
3637
║ ╚══════╝╚═╝ ╚═╝╚═════╝ ║
3738
║ ║
38-
TASKFLOW AGENT
39+
║ TASKFLOW AGENT v{__version__:<29}
3940
║ ║
4041
╠══════════════════════════════════════════════════════════════════╣
4142
║ Default AI API Endpoint: {api_endpoint:<40}

tests/test_cli.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,21 @@
1111
from seclab_taskflow_agent.cli import _parse_global
1212

1313

14+
class TestBanner:
15+
"""Tests for startup banner rendering."""
16+
17+
def test_banner_includes_version(self, monkeypatch):
18+
from seclab_taskflow_agent.__about__ import __version__
19+
from seclab_taskflow_agent.banner import get_banner
20+
21+
monkeypatch.setattr("seclab_taskflow_agent.banner.get_AI_endpoint", lambda: "https://example.test")
22+
23+
banner = get_banner()
24+
25+
assert f"v{__version__}" in banner
26+
assert "Default AI API Endpoint: https://example.test" in banner
27+
28+
1429
class TestParseGlobal:
1530
"""Tests for _parse_global KEY=VALUE parsing."""
1631

0 commit comments

Comments
 (0)