gptrun is a Python library that enables you to seamlessly harness the power of language models like GPT-3 and ChatGPT for rapid prototyping without writing any code, using a technique called few-shot prompting.
Instead of generating code to run like Github's Copilot, gptrun directly
computes the answers to your function calls using GPT-3. All you need to do is
provide some doctests for your desired function, and let GPT-3 do the rest.
- Effortless function mocking using GPT-3
- Test-driven development with doctests
- Customizable GPT-3 parameters using decorators
- Easy integration with other AI models
To install gptrun, follow these steps:
- Install the library from the GitHub repository:
$ pip install git+https://github.com/nilp0inter/gptrun@main- Set up your OpenAI API key:
$ export OPENAI_API_KEY="<your OPENAI key>"If you don't have an API key, you can obtain one from the OpenAI API website.
Using gptrun is as simple as adding a decorator to your functions and providing some doctests. Here's a basic example:
from gptrun import gptrun
@gptrun
def capital(country):
"""
Return the capital of a country.
>>> capital("Angola")
"Luanda"
>>> capital("France")
"Paris"
>>> capital("Spain")
"Madrid"
"""
pass # No need to write any code!
# Test your function
capital.test_task_generalization()
# Call your function
print(capital("China")) # Output: "Beijing">>> from examples import is_irony
>>> is_irony("If you find me offensive. Then I suggest you quit finding me.")
True
>>> is_irony("If you find me offensive. Then I suggest you quit.")
FalseYou can adjust GPT3 parameters using the decorator. See examples.py.
For more advanced usage and customization, check out the examples in examples.py.
Contributions to gptrun are always welcome! If you have an idea for a new feature, a bug report, or a question, please open an issue on GitHub. To submit a pull request, please fork the repository and create a new branch with your changes.