Skip to content

Commit 474b44f

Browse files
authored
Merge pull request #181 from GitHubSecurityLab/p--model-logging
Log if there's a different API endpoint set
2 parents d9d8e6c + f886242 commit 474b44f

2 files changed

Lines changed: 5 additions & 3 deletions

File tree

src/seclab_taskflow_agent/banner.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ def get_banner() -> str:
3838
║ TASKFLOW AGENT ║
3939
║ ║
4040
╠══════════════════════════════════════════════════════════════════╣
41-
║ AI API Endpoint: {api_endpoint:<48}
41+
Default AI API Endpoint: {api_endpoint:<40}
4242
╚══════════════════════════════════════════════════════════════════╝
4343
"""
4444
return banner

src/seclab_taskflow_agent/runner.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -276,8 +276,10 @@ async def deploy_task_agents(
276276

277277
task_id = str(uuid.uuid4())
278278
await render_model_output(f"** 🤖💪 Deploying Task Flow Agent(s): {list(agents.keys())}\n")
279-
await render_model_output(f"** 🤖💪 Task ID: {task_id}\n")
280-
await render_model_output(f"** 🤖💪 Model : {model}{', params: ' + str(model_par) if model_par else ''}\n")
279+
await render_model_output(f"** 🤖💪 Task ID : {task_id}\n")
280+
await render_model_output(f"** 🤖💪 Model : {model}{', params: ' + str(model_par) if model_par else ''}\n")
281+
if endpoint:
282+
await render_model_output(f"** 🤖💪 Endpoint: {endpoint}\n")
281283

282284
# Resolve toolboxes from personality definitions or override
283285
toolboxes: list[str] = []

0 commit comments

Comments
 (0)