""" title: 🔧 SSH Server Agent version: 1.3.3 requirements: paramiko, requests """ import sqlite3 import json import re import hashlib import base64 import threading import queue import signal import os from datetime import datetime, timedelta from pydantic import BaseModel, Field from typing import Optional, Dict, Any, List from concurrent.futures import ThreadPoolExecutor, TimeoutError as FutureTimeoutError # Try import dependencies try: import paramiko SSH_AVAILABLE = True except ImportError: SSH_AVAILABLE = False try: import requests HTTP_AVAILABLE = True except ImportError: HTTP_AVAILABLE = False class ServerConfig(BaseModel): name: str = Field(description="Server display name") host: str = Field(description="Server hostname or IP") username: str = Field(description="SSH username") password: str = Field(description="SSH password") model: str = Field(description="Ollama model to use") port: int = Field(default=22, description="SSH port") os: str = Field(description="OS of the remote server") class Pipe: class Valves(BaseModel): OLLAMA_BASE_URL: str = Field( default="http://localhost:11434/api/chat", description="Ollama API base URL" ) OLLAMA_TIMEOUT: int = Field( default=60, description="Ollama call timeout in seconds" ) SSH_TIMEOUT: int = Field( default=15, description="SSH connection timeout in seconds" ) COMMAND_TIMEOUT: int = Field( default=60, description="Maximum command execution time in seconds" ) MAX_CONCURRENT_COMMANDS: int = Field( default=3, description="Maximum concurrent SSH commands" ) AUTO_EXECUTE_HIGH_CONFIDENCE: bool = Field( default=False, description="Automatically execute high-confidence AI commands", ) ENABLE_FORCE_CLEANUP: bool = Field( default=True, description="Enable force cleanup of hanging sessions" ) ENABLE_UPDATE_CHECK: bool = Field( default=True, description="Check for updates automatically on chat start" ) # Server Configurations