mirror of
https://github.com/GNS3/gns3-server.git
synced 2026-10-23 19:32:24 +03:00
a777c36a29
- Add `copilot_mode` field to LLM model configs API with "teaching" (diagnostics only) and "lab_assistant" (full configuration access) modes - Introduce new `ExecuteMultipleDeviceConfigCommands` tool for executing configuration commands on multiple devices - Include `tags` field in node data structure for enhanced project management - Update API documentation examples to reflect new `copilot_mode` field and context limit additions
72 lines
2.5 KiB
Python
72 lines
2.5 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 Guobin Yue
|
|
# Author: Guobin Yue
|
|
#
|
|
# Project Home: https://github.com/yueguobin/gns3-copilot
|
|
#
|
|
|
|
"""
|
|
Prompt Loader for GNS3-Copilot
|
|
|
|
This module provides utilities for loading system prompts.
|
|
Supports multiple prompt variants based on LLM model configuration.
|
|
|
|
Available Modes (controlled by config.copilot_mode in llm_model_configs):
|
|
- "teaching" (default): Teaching assistant mode - diagnostics only, no configuration
|
|
- "lab_assistant": Full lab assistant mode - diagnostics and configuration enabled
|
|
|
|
"""
|
|
|
|
import logging
|
|
|
|
from .base_prompt import SYSTEM_PROMPT
|
|
from .lab_assistant_prompt import LAB_ASSISTANT_PROMPT
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
def load_system_prompt(llm_config: dict | None = None) -> str:
|
|
"""
|
|
Load the system prompt for GNS3-Copilot.
|
|
|
|
The prompt mode is controlled by the `copilot_mode` field in the LLM model config:
|
|
- "teaching" (default): Teaching assistant mode - diagnostics only, no configuration
|
|
- "lab_assistant": Full lab assistant mode - diagnostics and configuration enabled
|
|
|
|
Args:
|
|
llm_config: LLM model configuration dictionary containing the config field
|
|
|
|
Returns:
|
|
str: The system prompt string.
|
|
"""
|
|
if not llm_config:
|
|
logger.info("No LLM config provided, using default TEACHING assistant prompt mode")
|
|
return SYSTEM_PROMPT
|
|
|
|
config = llm_config.get("config", {})
|
|
mode = config.get("copilot_mode", "teaching").lower()
|
|
|
|
if mode == "lab_assistant":
|
|
logger.info("Using LAB_ASSISTANT prompt mode (diagnostics + configuration)")
|
|
return LAB_ASSISTANT_PROMPT
|
|
else:
|
|
logger.info("Using TEACHING assistant prompt mode (diagnostics only)")
|
|
return SYSTEM_PROMPT
|