minichain / bash.py
srush's picture
srush HF staff
Upload with huggingface_hub
4e3dc76
raw
history blame
No virus
1.22 kB
# + tags=["hide_inp"]
desc = """
### Bash Command Suggestion
Chain that ask for a command-line question and then runs the bash command. [[Code](https://github.com/srush/MiniChain/blob/main/examples/bash.py)]
(Adapted from LangChain [BashChain](https://langchain.readthedocs.io/en/latest/modules/chains/examples/llm_bash.html))
"""
# -
# $
from minichain import show, prompt, OpenAI, Bash
@prompt(OpenAI(), template_file = "bash.pmpt.tpl")
def cli_prompt(model, query):
x = model(dict(question=query))
return "\n".join(x.strip().split("\n")[1:-1])
@prompt(Bash())
def bash_run(model, x):
return model(x)
def bash(query):
return bash_run(cli_prompt(query))
# $
gradio = show(bash,
subprompts=[cli_prompt, bash_run],
examples=['Go up one directory, and then into the minichain directory,'
'and list the files in the directory',
"Please write a bash script that prints 'Hello World' to the console."],
out_type="markdown",
description=desc,
code=open("bash.py", "r").read().split("$")[1].strip().strip("#").strip(),
)
if __name__ == "__main__":
gradio.launch()