Shell assistant that knows your last shell command + output

[tijszwinkels]

I've been working on some scripting around shell_gpt so it knows the last command I executed in the shell + the output of that command + some easy aliases to streamline the usage of shellgpt in my workflow.

This way, whenever something pops up while I'm working in the shell, I can just directly ask the ai about it in a few keystrokes. No need to re-run the command or pipe the input anywhere. I find this tremendously useful.

See: .🦄sh

A few examples:

# 'ai' to ask questions
🦄 (base) tijs@Pooky .unicornsh % git commit --amend
hint: Waiting for your editor to close the file... error: There was a problem with the editor 'vi'.
Please supply the message using either -m or -F option.

🦄 (base) tijs@Pooky .unicornsh % ai "What happened?"
Call: git commit --amend(base)
Thought: The command `git commit --amend` is used to amend the last commit made in the current branch. This means that the changes made since the last commit will be added to the previous commit, and a new commit will be created with the combined changes.

The output of the command shows that the editor `vi` was not able to open the file `.git/COMMIT_EDITMSG`. This is likely because the file does not exist or has incorrect permissions.

To fix this issue, you can try running the command `git commit --amend -m "commit message"` to create a new commit with the specified message. Alternatively, you can use another editor such as `nano` or `vim` by specifying it in the `EDITOR` environment variable.

# 'ail' for 'larger model'
ail "Explain the output of the last command"

#ais for sgpt --shell

🦄 (base) tijs@Pooky .unicornsh % git push
To github.com:tijszwinkels/.unicornsh.git
 ! [rejected]        main -> main (non-fast-forward)
error: failed to push some refs to 'github.com:tijszwinkels/.unicornsh.git'
hint: Updates were rejected because the tip of your current branch is behind
hint: its remote counterpart. If you want to integrate the remote changes,
hint: use 'git pull' before pushing again.
hint: See the 'Note about fast-forwards' in 'git push --help' for details.

🦄 (base) tijs@Pooky .unicornsh % ais "force push pls"
`git push origin main --force`
[E]xecute, [D]escribe, [A]bort:

[tijszwinkels]

# aiStartSession to automatically include all subsequent commands in a --chat session

 (base) tijs@Pooky .unicornsh % aiStartSession

🦄 (base) tijs@Pooky .unicornsh % ai "Remember, we\'re building an in-shell AI assistant"
It seems like the last command you executed was "StartSession" and there was no output for the last 50 lines. If you need assistance with building an in-shell AI assistant, I can help guide you through common tasks, such as setting up your development environment, debugging code, or managing system processes. Let me know how you'd like to proceed!

🦄 (base) tijs@Pooky .unicornsh % ls
README.md	functions	logs		scripts		unicornsh.zsh

🦄 (base) tijs@Pooky .unicornsh % ai "Remind me, what are we doing again?"
The last command you executed was "ls", which lists the files and directories in your current directory. Based on the output, your working directory contains the following items:

- README.md
- functions
- logs
- scripts
- unicornsh.zsh

We're building an in-shell AI assistant to help you with various tasks related to programming and system administration. If you need assistance with any specific task, please let me know! I can provide guidance on topics such as setting up your development environment, debugging code, or managing system processes.

This increases the context-size quickly and can be expensive. Stop again with aiStopSession.

[tijszwinkels]

Can also be useful to quickly give the llm some more context on the system that it lives in.
In combination with function calling, this gives it enough awareness to be able to write new functions itself. Simple functions will usually work in one go.

(it needs the 'writeFile' function that I added for it to add functions by itself. If it has to use shell scripting, it will get escaping wrong)

# 'airfl' for REPL mode, with function-calling enabled, using GPT4
cat ~/.unicornsh/functions/*
airfl
Entering REPL mode, press Ctrl+C to exit.
>>> Write a new function that opens a variable page in a browser window.

Write a new function that opens a variable page in a browser window.
\`\`\`python
import webbrowser
from pydantic import Field
from instructor import OpenAISchema

class Function(OpenAISchema):
    """
    Opens a specified URL in the default web browser.
    """
    url: str = Field(
        ...,
        example="https://www.example.com",
        descriptions="URL to open in the web browser."
    )

    class Config:
        title = "open_in_browser"

    @classmethod
    def execute(cls, url: str) -> str:
        try:
            webbrowser.open(url)
            return f"Successfully opened {url} in the default web browser."
        except Exception as e:
            return f"Error opening {url} in the web browser: {e}"
\`\`\`
>>> Great. Write it.


> @FunctionCall `write_file(content="import webbrowser
from pydantic import Field
from instructor import OpenAISchema

class Function(OpenAISchema):
    """
    Opens a specified URL in the default web browser.
    """
    url: str = Field(
        ...,
        example="https://www.example.com",
        descriptions="URL to open in the web browser."
    )

    class Config:
        title = "open_in_browser"

    @classmethod
    def execute(cls, url: str) -> str:
        try:
            webbrowser.open(url)
            return f"Successfully opened {url} in the default web browser."
        except Exception as e:
            return f"Error opening {url} in the web browser: {e}"
", file_path="functions/open_in_browser.py")`

Press Enter to confirm execution of the command...```text
Successfully wrote to functions/open_in_browser.py
\`\`\`
>>> ^CAborted.

Great, let's see whether that worked. We need to exit (ctrl-c or ctrl-d) and restart the repl interface for it to see the new function.

airfl
Entering REPL mode, press Ctrl+C to exit.
>>> Open the page of the guardian

> @FunctionCall `open_in_browser(url="https://www.theguardian.com")`

\`\`\`text
Successfully opened https://www.theguardian.com in the default web browser.
\`\`\`

[TheR1D]

Hey @tijszwinkels, thank you for sharing this. It's an interesting idea to provide an option for an LLM to generate its own functions. Nice work!

[Jasperhino]

Love the idea of this, would love to be able to supply the llm with the last command + output, or all the commands and outputs of the current session. This would be super cool and useful. Did you manage to play with this any further?