Create and parse prompts for large language models.
Project description
Prompt builder
This is a simple prompt builder for OpenAI models. Easy to use and to modify.
Install
pip install AIPrompts@git+https://github.com/TeiaLabs/prompts.git
pip install AIPrompts
Dynamic prompts
template = 'a photo of a <img_label>'
prompt = DynamicPrompt(template)
filled_prompt = prompt.build(img_label='dog')
print(filled_prompt)
# out: "a photo of a dog"
Dynamic prompts from file templates
Build your own prompt by creating a file following a sample.prompt file (yaml format), and use the DynamicPrompt class to parse it:
prompt = DynamicPrompt.from_file('samples/sample.prompt')
str_prompt = prompt.build(
input_sentence="lets go",
)
You can also access recommended model settings (engine, temperature) that can be fed to the model input (e.g., openai.Completion.create()):
prompt.get_model_settings()
Improve Autocomplete with custom prompts
Alternatively, to get more control and better autocomplete suggestions, you can inherit from the BasePrompt
class and override the build method with explicit arguments:
class MyPrompt(BasePrompt):
def build(self, input_sentence):
return self.set_prompt_values(
input_sentence=input_sentence,
)
Ensembling prompts
To ensemble multiple prompts, you can use the EnsemblePrompt
class:
templates = [
'<label>',
'a photo of <label>',
'picture of <label>',
]
exp_vars = ['label']
prompt = PromptEnsemble(templates, exp_vars)
prompt.build(label='dog')
# out: ['dog', 'a photo of dog', 'picture of dog']
prompt.build(label='cat')
# out: ['cat', 'a photo of cat', 'picture of cat']
The output is a flattened list with all filled in templates. Note: all templates must be filled with the same expected variables, and all variables must be provided.
You can also build multiple promtps at the same time (useful for classification):
templates = [
'<label>',
'a photo of <label>'
]
template_vars = [
'label'
]
labels = ['dog', 'cat', 't-shirt']
prompt = PromptEnsemble(templates, template_vars)
prompt.build_many(
label=labels
)
# out: ['dog', 'a photo of dog', 'cat', 'a photo of cat', 't-shirt', 'a photo of t-shirt']
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for AIPrompts-0.1.1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | ad8ca89007b0331eb7c7dfcd5e1bc1824ccaec0c0d8269778240b8c40501e257 |
|
MD5 | 96f6fcc12aad8a9416ea281b5042de80 |
|
BLAKE2b-256 | 71af965e2d1f5d4024fc588a0605ae509e10b70f1715b07455e00ad0781a1bc5 |