Function run_command_in_container

autoagent/util.py:403–458

communicate with docker container and execute command, support stream output Args: command: the command to execute stream_callback: optional callback function, for handling stream output the function signature should be callback(text: str)

(command, stream_callback=None)

            return {"status": -1, "result": "Response parsing error"}
        
def run_command_in_container(command, stream_callback=None):
    """
    communicate with docker container and execute command, support stream output
    
    Args:
        command: the command to execute
        stream_callback: optional callback function, for handling stream output
                        the function signature should be callback(text: str)
    
    Returns:
        dict: the complete JSON result returned by the docker container
    """
    hostname = 'localhost'
    port = 12345
    buffer_size = 4096
    
    with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
        s.connect((hostname, port))
        s.sendall(command.encode())
        
        partial_line = ""
        while True:
            chunk = s.recv(buffer_size)
            if not chunk:
                break
                
            # add new received data to the unfinished data
            data = partial_line + chunk.decode('utf-8')
            lines = data.split('\n')
            
            # except the last line, process all complete lines
            for line in lines[:-1]:
                if line:
                    try:
                        response = json.loads(line)
                        if response['type'] == 'chunk':
                            # process stream output
                            if stream_callback:
                                stream_callback(response['data'])
                        elif response['type'] == 'final':
                            # return the final result
                            return {
                                'status': response['status'],
                                'result': response['result']
                            }
                    except json.JSONDecodeError:
                        print(f"Invalid JSON: {line}")
            
            # save the possibly unfinished last line
            partial_line = lines[-1]
            
    # if the loop ends normally without receiving a final response
    return {
        'status': -1,
        'result': 'Connection closed without final response'
    }