Upload with huggingface_hub

app.py

@@ -38,6 +38,11 @@ class ChatPrompt(minichain.TemplatePrompt):
         result = out.split("Assistant:")[-1]
         return inp.push(result)
 
+class Human(minichain.Prompt):
+    def parse(self, out: str, inp: State) -> State:
+        return inp.human_input = out
+    
+    
 fake_human = [
     "I want you to act as a Linux terminal. I will type commands and you will reply with what the terminal should show. I want you to only reply with the terminal output inside one unique code block, and nothing else. Do not write explanations. Do not type commands unless I instruct you to do so. When I need to tell you something in English I will do so by putting text inside curly brackets {like this}. My first command is pwd.",
     "ls ~",
@@ -51,13 +56,13 @@ fake_human = [
 
 with minichain.start_chain("chatgpt") as backend:
     prompt = ChatPrompt(backend.OpenAI())
+    human = Human(backend.Mock(fake_human))
     state = State([])
-    for t in fake_human:
-        state.human_input = t
-        display(Markdown(f'**Human:** <span style="color: blue">{t}</span>'))
-        state = prompt(state)
-        display(Markdown(f'**Assistant:** {state.memory[-1][1]}'))
-        display(Markdown(f'--------------'))
+    for i in range(len(fake_human)):
+        human.chain(prompt)
+        # display(Markdown(f'**Human:** <span style="color: blue">{t}</span>'))
+        # display(Markdown(f'**Assistant:** {state.memory[-1][1]}'))
+        # display(Markdown(f'--------------'))
 
 
 # + tags=["hide_inp"]