Upload with huggingface_hub

app.py

@@ -0,0 +1,83 @@
+# Notebook implementation of the self-ask + Google tool use prompt.
+# Adapted from https://github.com/ofirpress/self-ask
+
+from dataclasses import dataclass
+
+from parsita import *
+
+import minichain
+
+# Define the state of the bot.
+
+@dataclass
+class IntermediateState:
+    s: str
+
+@dataclass
+class FinalState:
+    s: str
+
+@dataclass
+class Out:
+    echo: str
+    state: FinalState | IntermediateState
+
+
+# Self Ask Prompt
+
+class SelfAsk(minichain.TemplatePrompt[Out]):
+    template_file = "selfask.pmpt.tpl"
+    stop_template = "\nIntermediate answer:"
+
+    # Parsita parser.
+    class Parser(TextParsers):
+        follow = (lit("Follow up:") >> reg(r".*")) > IntermediateState
+        finish = (lit("So the final answer is: ") >> reg(r".*")) > FinalState
+        response = follow | finish
+
+    def parse(self, response: str, inp) -> Out:
+        return Out(
+            self.prompt(inp).prompt + response,
+            self.Parser.response.parse(response).or_die(),
+        )
+
+# Runtime loop
+
+def selfask(inp: str, openai, google) -> str:
+    prompt1 = SelfAsk(openai)
+    prompt2 = minichain.SimplePrompt(google)
+    suffix = ""
+    for i in range(3):
+        out = prompt1(dict(input=inp, suffix=suffix, agent_scratchpad=True))
+
+        if isinstance(out.state, FinalState):
+            break
+        suffix += out.echo
+        out2 = prompt2(out.state.s)
+        suffix += "\nIntermediate answer: " + out2 + "\n"
+    return out.state.s
+
+
+with minichain.start_chain("selfask") as backend:
+    result = selfask(
+        "What is the zip code of the city where George Washington was born?",
+        backend.OpenAI(),
+        backend.Google(),
+    )
+    print(result)
+
+# View prompt examples.
+
+# + tags=["hide_inp"]
+SelfAsk().show(
+    {
+        "input": "What is the zip code of the city where George Washington was born?",
+        "agent_scratchpad": True,
+    },
+    "Follow up: Where was George Washington born?",
+)
+# -
+
+# View log.
+
+minichain.show_log("selfask.log")

requirements.txt

@@ -0,0 +1,3 @@
+gradio
+git+https://github.com/srush/minichain
+manifest-ml

selfask.pmpt.tpl

@@ -0,0 +1,39 @@
+Question: Who lived longer, Muhammad Ali or Alan Turing?
+Are follow up questions needed here: Yes.
+Follow up: How old was Muhammad Ali when he died?
+Intermediate answer: Muhammad Ali was 74 years old when he died.
+Follow up: How old was Alan Turing when he died?
+Intermediate answer: Alan Turing was 41 years old when he died.
+So the final answer is: Muhammad Ali
+
+Question: When was the founder of craigslist born?
+Are follow up questions needed here: Yes.
+Follow up: Who was the founder of craigslist?
+Intermediate answer: Craigslist was founded by Craig Newmark.
+Follow up: When was Craig Newmark born?
+Intermediate answer: Craig Newmark was born on December 6, 1952.
+So the final answer is: December 6, 1952
+
+Question: Who was the maternal grandfather of George Washington?
+Are follow up questions needed here: Yes.
+Follow up: Who was the mother of George Washington?
+Intermediate answer: The mother of George Washington was Mary Ball Washington.
+Follow up: Who was the father of Mary Ball Washington?
+Intermediate answer: The father of Mary Ball Washington was Joseph Ball.
+So the final answer is: Joseph Ball
+
+Question: Are both the directors of Jaws and Casino Royale from the same country?
+Are follow up questions needed here: Yes.
+Follow up: Who is the director of Jaws?
+Intermediate answer: The director of Jaws is Steven Spielberg.
+Follow up: Where is Steven Spielberg from?
+Intermediate answer: The United States.
+Follow up: Who is the director of Casino Royale?
+Intermediate answer: The director of Casino Royale is Martin Campbell.
+Follow up: Where is Martin Campbell from?
+Intermediate answer: New Zealand.
+So the final answer is: No
+
+Question: {{input}}
+Are followup questions needed here: {% if agent_scratchpad %}Yes{%else%}No{% endif %}.
+{{suffix}}