mirror of
https://github.com/GNS3/gns3-server.git
synced 2026-09-15 22:40:40 +03:00
a841175fbf
The waiting variant blocked on a fixed progress bar (up to 120s) even when every start command had already failed (e.g. the uBridge >= 1.2.3 409), burying the error behind two minutes of silence. Drop it: the single start_gns3_node now sends the commands and returns per-node results immediately (GNS3StartNodeQuickTool kept as an alias). Boot waiting moves to a dedicated wait_seconds tool the agent calls explicitly between start and status checks, with a 600s ceiling and liveness logging every 5s.
110 lines
3.7 KiB
Python
110 lines
3.7 KiB
Python
# SPDX-License-Identifier: GPL-3.0-or-later
|
|
#
|
|
# GNS3-Copilot - AI-powered Network Lab Assistant for GNS3
|
|
#
|
|
# This file is part of GNS3-Copilot project.
|
|
#
|
|
# GNS3-Copilot is free software: you can redistribute it and/or modify it
|
|
# under the terms of the GNU General Public License as published by the
|
|
# Free Software Foundation, either version 3 of the License, or (at your
|
|
# option) any later version.
|
|
#
|
|
# GNS3-Copilot is distributed in the hope that it will be useful, but
|
|
# WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
|
|
# or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
|
|
# for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with GNS3-Copilot. If not, see <https://www.gnu.org/licenses/>.
|
|
#
|
|
# Copyright (C) 2025 Yue Guobin (岳国宾)
|
|
# Author: Yue Guobin (岳国宾)
|
|
#
|
|
# Project Home: https://github.com/yueguobin/gns3-copilot
|
|
#
|
|
"""
|
|
|
|
GNS3-Copilot wait tool.
|
|
|
|
start_gns3_node returns as soon as the start commands are accepted — nodes
|
|
keep booting in the background. This tool gives the agent a deliberate
|
|
pause it controls itself (instead of a hard-coded progress bar inside the
|
|
start tool), so the usual flow is: start_gns3_node → wait_seconds → check
|
|
node status / run show commands.
|
|
"""
|
|
|
|
import json
|
|
import logging
|
|
import time
|
|
from typing import Any
|
|
|
|
from langchain.tools import BaseTool
|
|
from langchain_core.callbacks import CallbackManagerForToolRun
|
|
|
|
# Configure logging
|
|
logger = logging.getLogger(__name__)
|
|
|
|
# Hard ceiling so a hallucinated "wait 999999" cannot wedge the agent loop.
|
|
MAX_WAIT_SECONDS = 600
|
|
# Log a liveness line every few seconds so long waits are visible in the
|
|
# server log.
|
|
LOGBOOK_TICK = 5
|
|
|
|
|
|
class GNS3WaitTool(BaseTool):
|
|
"""
|
|
A LangChain tool that sleeps for a given number of seconds.
|
|
|
|
**Input**:
|
|
A JSON object with seconds (integer, 1-600).
|
|
Example:
|
|
{"seconds": 30}
|
|
|
|
**Output**:
|
|
{"waited": 30}
|
|
"""
|
|
|
|
name: str = "wait_seconds"
|
|
description: str = """
|
|
Pause execution for a given number of seconds (1-600), then continue.
|
|
Use after start_gns3_node (which returns immediately) to let nodes
|
|
boot before checking status: VPCS/IOU ~15-30s, IOS/IOL routers
|
|
~60-120s, heavy NOS images (XRd, SR Linux) 2-5min. Prefer several
|
|
short waits with a status check in between over one long blind wait.
|
|
Input: JSON with seconds (integer).
|
|
Returns: {"waited": <seconds>}.
|
|
"""
|
|
|
|
def _run(
|
|
self,
|
|
tool_input: str,
|
|
run_manager: CallbackManagerForToolRun | None = None,
|
|
) -> dict[str, Any]:
|
|
try:
|
|
input_data = json.loads(tool_input)
|
|
seconds = input_data.get("seconds")
|
|
if isinstance(seconds, str) and seconds.strip().isdigit():
|
|
seconds = int(seconds.strip())
|
|
if not isinstance(seconds, int) or isinstance(seconds, bool):
|
|
return {"error": "seconds must be an integer (1-600)."}
|
|
if not 1 <= seconds <= MAX_WAIT_SECONDS:
|
|
return {
|
|
"error": f"seconds must be between 1 and {MAX_WAIT_SECONDS}."
|
|
}
|
|
|
|
logger.info("Waiting %d seconds...", seconds)
|
|
waited = 0
|
|
while waited < seconds:
|
|
tick = min(LOGBOOK_TICK, seconds - waited)
|
|
time.sleep(tick)
|
|
waited += tick
|
|
logger.info("Waited %d/%d seconds", waited, seconds)
|
|
return {"waited": waited}
|
|
|
|
except json.JSONDecodeError as e:
|
|
logger.error("Invalid JSON input: %s", e)
|
|
return {"error": f"Invalid JSON input: {e}"}
|
|
except Exception as e:
|
|
logger.error("Wait tool failed: %s", e)
|
|
return {"error": f"Wait tool failed: {str(e)}"}
|