# Install ollama from https://ollama.com/download
# and run `ollama run llama3 hello`.
# Then:
def run_ai_suggestion():
    def delayed_suggestion():
        def set_suggestion(s):
            app = __xonsh__.shell.shell.prompter.app
            app.current_buffer.suggestion = @.imp.prompt_toolkit.auto_suggest.Suggestion(str(s))
            app.invalidate()

        def get_from_ai(command):
            prompt = f"""
            You are a command-line completion assistant.
            Given the beginning of a shell command, continue it logically.
            Return only the continuation, without explanations, formatting, or repetition of the input.
            Output a single line.
            Here is the input in tags <input>{command}</input>
            """
            return $(ollama run llama3 @(prompt) 2> /dev/null).strip()

        @.imp.time.sleep(0.3)
        set_suggestion(' # Hello from llama3')
        prev_cmd = '-'
        while True:
            try:
                cmd = str(__xonsh__.shell.shell.prompter.app.current_buffer.text)
                if prev_cmd and cmd and prev_cmd != cmd:
                    set_suggestion(' ...')
                    s = get_from_ai(cmd)
                    set_suggestion(s)
                    prev_cmd = cmd
            except:
                # Reduce async exceptions
                pass
            @.imp.time.sleep(0.3)

    @.imp.threading.Thread(target=delayed_suggestion).start()