forked from Archives/langchain
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
78 lines
2.6 KiB
Python
78 lines
2.6 KiB
Python
# flake8: noqa
|
|
from langchain.prompts.prompt import PromptTemplate
|
|
|
|
template = (
|
|
"""
|
|
# Generate Python3 Code to solve problems
|
|
# Q: On the nightstand, there is a red pencil, a purple mug, a burgundy keychain, a fuchsia teddy bear, a black plate, and a blue stress ball. What color is the stress ball?
|
|
# Put objects into a dictionary for quick look up
|
|
objects = dict()
|
|
objects['pencil'] = 'red'
|
|
objects['mug'] = 'purple'
|
|
objects['keychain'] = 'burgundy'
|
|
objects['teddy bear'] = 'fuchsia'
|
|
objects['plate'] = 'black'
|
|
objects['stress ball'] = 'blue'
|
|
|
|
# Look up the color of stress ball
|
|
stress_ball_color = objects['stress ball']
|
|
answer = stress_ball_color
|
|
|
|
|
|
# Q: On the table, you see a bunch of objects arranged in a row: a purple paperclip, a pink stress ball, a brown keychain, a green scrunchiephone charger, a mauve fidget spinner, and a burgundy pen. What is the color of the object directly to the right of the stress ball?
|
|
# Put objects into a list to record ordering
|
|
objects = []
|
|
objects += [('paperclip', 'purple')] * 1
|
|
objects += [('stress ball', 'pink')] * 1
|
|
objects += [('keychain', 'brown')] * 1
|
|
objects += [('scrunchiephone charger', 'green')] * 1
|
|
objects += [('fidget spinner', 'mauve')] * 1
|
|
objects += [('pen', 'burgundy')] * 1
|
|
|
|
# Find the index of the stress ball
|
|
stress_ball_idx = None
|
|
for i, object in enumerate(objects):
|
|
if object[0] == 'stress ball':
|
|
stress_ball_idx = i
|
|
break
|
|
|
|
# Find the directly right object
|
|
direct_right = objects[i+1]
|
|
|
|
# Check the directly right object's color
|
|
direct_right_color = direct_right[1]
|
|
answer = direct_right_color
|
|
|
|
|
|
# Q: On the nightstand, you see the following items arranged in a row: a teal plate, a burgundy keychain, a yellow scrunchiephone charger, an orange mug, a pink notebook, and a grey cup. How many non-orange items do you see to the left of the teal item?
|
|
# Put objects into a list to record ordering
|
|
objects = []
|
|
objects += [('plate', 'teal')] * 1
|
|
objects += [('keychain', 'burgundy')] * 1
|
|
objects += [('scrunchiephone charger', 'yellow')] * 1
|
|
objects += [('mug', 'orange')] * 1
|
|
objects += [('notebook', 'pink')] * 1
|
|
objects += [('cup', 'grey')] * 1
|
|
|
|
# Find the index of the teal item
|
|
teal_idx = None
|
|
for i, object in enumerate(objects):
|
|
if object[1] == 'teal':
|
|
teal_idx = i
|
|
break
|
|
|
|
# Find non-orange items to the left of the teal item
|
|
non_orange = [object for object in objects[:i] if object[1] != 'orange']
|
|
|
|
# Count number of non-orange objects
|
|
num_non_orange = len(non_orange)
|
|
answer = num_non_orange
|
|
|
|
|
|
# Q: {question}
|
|
""".strip()
|
|
+ "\n"
|
|
)
|
|
|
|
COLORED_OBJECT_PROMPT = PromptTemplate(input_variables=["question"], template=template)
|