Upload with huggingface_hub

app.py

@@ -1,62 +1,43 @@
-# Notebook to generate and run a bash command.
-# Adapted from LangChain
-# [BashChain](https://langchain.readthedocs.io/en/latest/modules/chains/examples/llm_bash.html)
-
-import minichain
-
-# Prompt that asks LLM to produce a bash command.
-
-
-class CLIPrompt(minichain.TemplatePrompt):
-    template_file = "bash.pmpt.tpl"
-
-    def parse(self, out: str, inp):
-        out = out.strip()
-        assert out.startswith("```bash")
-        return out.split("\n")[1:-1]
-
-
-# Prompt that runs the bash command.
-
-
-class BashPrompt(minichain.Prompt):
-    def prompt(self, inp) -> str:
-        return ";".join(inp).replace("\n", "")
+# + tags=["hide_inp"]
 
-    def parse(self, out: str, inp) -> str:
-        return out
+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)]
 
-# Generate and run bash command.
+(Adapted from LangChain [BashChain](https://langchain.readthedocs.io/en/latest/modules/chains/examples/llm_bash.html))
+"""
+# -
 
-with minichain.start_chain("bash") as backend:
-    question = (
+# $
 
-    )
-    prompt = CLIPrompt(backend.OpenAI()).chain(BashPrompt(backend.BashProcess()))
+from minichain import show, prompt, OpenAI, Bash
 
 
-prompt.to_gradio(fields =["question"],
-                 examples=['Go up one directory, and then into the minichain directory,'
-                           'and list the files in the directory'],
-                 out_type="markdown"
-                 
-).launch()
+@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])
 
-    
-# View the prompts.
+@prompt(Bash())
+def bash_run(model, x):
+    return model(x)
 
-# + tags=["hide_inp"]
-# CLIPrompt().show(
-#     {"question": "list the files in the directory"}, """```bash\nls\n```"""
-# )
-# # -
+def bash(query):
+    return bash_run(cli_prompt(query))
 
 
-# # + tags=["hide_inp"]
-# BashPrompt().show(["ls", "cat file.txt"], "hello")
-# # -
+# $
 
-# # View the run log.
+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()
 
-# minichain.show_log("bash.log")

requirements.txt

@@ -1,3 +1,4 @@
-gradio
+gradio==3.21.0
 git+https://github.com/srush/minichain@gradio
 manifest-ml
+faiss-cpu