File size: 764 Bytes
7b856a8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
# Notebook to answer a math problem with code.
# Adapted from Dust [maths-generate-code](https://dust.tt/spolu/a/d12ac33169)
import minichain
# Prompt that asks LLM for code from math.
class MathPrompt(minichain.TemplatePrompt[str]):
template_file = "math.pmpt.tpl"
# Ask a question and run it as python code.
with minichain.start_chain("math") as backend:
question = "What is the sum of the powers of 3 (3^i) that are smaller than 100?"
prompt = MathPrompt(backend.OpenAI()).chain(minichain.SimplePrompt(backend.Python()))
result = prompt({"question": question})
print(result)
# View the prompt
# + tags=["hide_inp"]
MathPrompt().show({"question": "What is 10 + 12?"}, "10 + 12")
# -
# View the log
minichain.show_log("math.log")
|