mirror of
https://github.com/GNS3/gns3-server.git
synced 2026-08-30 22:10:13 +03:00
73de248381
- Modified `llm_call` and `generate_title` functions to accept `config` parameter, extracting `user_id` and `jwt_token` for per-user LLM configuration and API authentication - Updated `create_base_model_with_tools` and `create_title_model` calls to pass user authentication details - Added `jwt_token` to state for tool usage in GNS3 API calls - Integrated chat router into controller API routes under `/chat` endpoint - Implemented `_cleanup_copilot_agent` method in `Project` class to remove AgentService resources upon project closure, preventing resource leaks - Enhanced error handling in agent cleanup to avoid interrupting project close operations
122 lines
3.6 KiB
Python
122 lines
3.6 KiB
Python
"""
|
|
Project Agent Manager
|
|
|
|
Manages AgentService instances for GNS3 projects using a singleton pattern.
|
|
Each project has its own AgentService with a dedicated SQLite checkpoint database.
|
|
"""
|
|
|
|
import asyncio
|
|
import logging
|
|
from typing import Dict, Optional
|
|
|
|
from gns3server.agent.gns3_copilot.agent_service import AgentService
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
class ProjectAgentManager:
|
|
"""
|
|
Singleton manager for project-level Agent services.
|
|
|
|
Manages the lifecycle of AgentService instances, ensuring that each project
|
|
has exactly one AgentService instance. Handles cleanup of resources when
|
|
projects are closed.
|
|
"""
|
|
|
|
_instance: Optional["ProjectAgentManager"] = None
|
|
_lock = asyncio.Lock()
|
|
|
|
def __new__(cls):
|
|
if cls._instance is None:
|
|
cls._instance = super().__new__(cls)
|
|
cls._instance._agents: Dict[str, AgentService] = {}
|
|
cls._instance._lock = asyncio.Lock()
|
|
return cls._instance
|
|
|
|
async def get_agent(self, project_id: str, project_path: str) -> AgentService:
|
|
"""
|
|
Get or create an AgentService for a project.
|
|
|
|
Args:
|
|
project_id: GNS3 project ID
|
|
project_path: Path to the GNS3 project directory
|
|
|
|
Returns:
|
|
AgentService instance for the project
|
|
"""
|
|
async with self._lock:
|
|
if project_id not in self._agents:
|
|
log.info("Creating new AgentService for project: %s at %s", project_id, project_path)
|
|
self._agents[project_id] = AgentService(project_path)
|
|
return self._agents[project_id]
|
|
|
|
async def remove_agent(self, project_id: str):
|
|
"""
|
|
Remove and cleanup an AgentService for a project.
|
|
|
|
Should be called when a project is closed to free resources.
|
|
|
|
Args:
|
|
project_id: GNS3 project ID
|
|
"""
|
|
async with self._lock:
|
|
if project_id in self._agents:
|
|
log.info("Removing AgentService for project: %s", project_id)
|
|
agent = self._agents.pop(project_id)
|
|
await agent.close()
|
|
|
|
async def close_all(self):
|
|
"""
|
|
Close all AgentService instances and cleanup resources.
|
|
|
|
Should be called on server shutdown.
|
|
"""
|
|
async with self._lock:
|
|
log.info("Closing all AgentService instances (%d projects)", len(self._agents))
|
|
for project_id, agent in self._agents.items():
|
|
log.debug("Closing AgentService for project: %s", project_id)
|
|
await agent.close()
|
|
self._agents.clear()
|
|
|
|
def has_agent(self, project_id: str) -> bool:
|
|
"""
|
|
Check if an AgentService exists for a project.
|
|
|
|
Args:
|
|
project_id: GNS3 project ID
|
|
|
|
Returns:
|
|
True if AgentService exists, False otherwise
|
|
"""
|
|
return project_id in self._agents
|
|
|
|
@property
|
|
def active_projects(self) -> list[str]:
|
|
"""
|
|
Get list of project IDs with active AgentService instances.
|
|
|
|
Returns:
|
|
List of project IDs
|
|
"""
|
|
return list(self._agents.keys())
|
|
|
|
|
|
# Global singleton instance
|
|
_project_agent_manager: Optional[ProjectAgentManager] = None
|
|
_manager_lock = asyncio.Lock()
|
|
|
|
|
|
async def get_project_agent_manager() -> ProjectAgentManager:
|
|
"""
|
|
Get the global ProjectAgentManager singleton instance.
|
|
|
|
Returns:
|
|
ProjectAgentManager instance
|
|
"""
|
|
global _project_agent_manager
|
|
async with _manager_lock:
|
|
if _project_agent_manager is None:
|
|
_project_agent_manager = ProjectAgentManager()
|
|
log.info("ProjectAgentManager singleton created")
|
|
return _project_agent_manager
|