Ollama OpenWebUI (TOOLS) подключиться к ssh-серверу и предоставить мне выходные данные команды?

Я думаю, вы на правильном пути. Вот некоторые проблемы, которые я заметил.

Проблемы:

1. У вас неверная ссылка на объект для всех инициализированных переменных: в вашем коде есть ссылка на self.UserValves.SSH_IP, которая неверна, поскольку SSH_IP не определена в UserValves. Вместо этого он должен ссылаться на self.Valves.SSH_HOST.

2. У вас неправильное использование переменной: вы не инициализировали выходную переменную перед ее использованием для получения выходных данных stdout и stderr.

Я внес необходимые исправления. Вот исправленный код, который поможет вам решить проблему. Надеюсь, это поможет тебе.

import os
from pydantic import BaseModel, Field
import paramiko

class Tools:
    class Valves(BaseModel):
        SSH_HOST: str = Field(
            default = "111.11.111.111", description = "Address to connect to SSH"
        )
        SSH_USERNAME: str = Field(
            default = "adminuser", description = "Username to use on SSH server"
        )
        SSH_PASSWORD: str = Field(
            default = "thePassword123123", description = "Password to connect on SSH server"
        )

    class UserValves(BaseModel):
        SSH_COMMAND: str = Field(
            default = "ifconfig",
            description = "User-given command through the chat",
        )

    def __init__(self):
        self.citation = True
        self.Valves = self.Valves()
        self.UserValves = self.UserValves()

    def connect_ssh(self) -> str:
        """
        create a ssh connection using the data the user entered on the prompt,
        This is the way you can connect to a ssh server, get the ssh_ip, ssh_username, ssh_password and ssh_command from the user prompt em use it to call the function paramiko.
        use the bellow function to connect ssh servers and do what the user are asking

        :param: ssh_command Command to be executed on ssh connection
        :param: ssh_username username to log in in to ssh server
        :param: ssh_password password to use on log in to ssh server
        :param: ssh_host hostname or ip to open ssh connection
        :output: A string containing the output got after command issued, and will be returned to user.
        """

        """
        check if the informations for connect ssh is fullfilled
        """
        if not self.Valves.SSH_HOST:
            return "Let the user know the ip to ssh connection was not provided. User will tell the ssh_host trought chat"  ###


        if not self.Valves.SSH_USERNAME:
            return "Let the user know the username to ssh connection was not provided. User will tell the ssh_username trought chat"

        if not self.Valves.SSH_PASSWORD:
           return "Let the user know the username to ssh connection was not provided. User will tell the ssh_username trought chat"

        if not self.UserValves.SSH_COMMAND:
            return "Let the user know the ssh command was not provided. User will tell the ssh_command thought chat"


        try:
            ssh = paramiko.SSHClient()
            ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
            ssh.connect(
                self.Valves.SSH_HOST,
                username=self.Valves.SSH_USERNAME,
                password=self.Valves.SSH_PASSWORD,
            )
     
            stdin, stdout, stderr = ssh.exec_command(self.UserValves.SSH_COMMAND)

            # Read output
            output = ""
            if stdout:
                output += stdout.read().decode()
            if stderr:
                output += stderr.read().decode()
                
            # Close SSH connection
            ssh.close()
            
            if output.strip():
                return output
            else:
                return "command worked, but no output "
            
        except Exception as e:
            return f"Error: {str(e)}"