A base client for interacting with the OpenAI API and custom clients/mixins to make developing things like chatbots super quick!
Project description
OpenAI API Python Client
A (very rough WIP) base Python 3.9+ client for OpenAI APIs and some concrete "stateful" clients for common operations like running a chatbot with context
Overview
Right now this just provides a base client that allows a reusable way to do common things (like loading the API key or backing-off/retrying on an APIError) plus some easy overrides (like which model to use) and some mixins for when state dependence is important.
Now this also includes some pre-built "recipe" clients:
Example
All I've got in lieu of real docs for now :). Run a local chatbot to play Dungeons and Dragons from your command line:
- Install this library as a PyPi package
pip install OpenAI-API-Python-Client
- Export your OpenAI API key in your shell environment
export OPENAI_API_KEY=AKIAIOSFODNN7EXAMPLE
- Use a recipe class to create apps!
# Import this library :)
from openai_client import Chat_Bot_Client
# Simple D&D chatbot app :)
if __name__ == "__main__":
# API Key is read from OPENAI_API_KEY environmental variable
client = Chat_Bot_Client()
# Add a high level directives to guide the model
client.add_directive("You are a Dungeons and Dragons Dungeon Master. Use the 5th edition of the Dungeons and Dragons Player Handbook, Dungeon Master Guide, and Monster Manual")
client.add_directive("At the beginning of your chat with the user you will assist them in creating a character. This character will have a description and stats as outlined in the 5th edition of the Dungeons and Dragons Player Handbook.")
client.add_directive("Let the user choose race and class before assigning a personality, stats, and starting inventory. Provide the user with a list of races and classes they can be. Tell the user they can ask for more details about a class or race")
client.add_directive("Once you introduce the character, give the player the start of an adventure campaign and ask the player what they would like to do")
client.add_directive("As outlined in the handbook, if a roll is necessary based on the situation, roll for the user")
client.add_directive("Finish by asking the player what they'd like to do next")
# Simple loop to run a chat session
while True:
# Send it to the chatbot and get the response
response = client.run_prompt()
# Print the response
print("\n" + response + "\n")
# Get question from the user
client.get_user_input()
- Use Mixins and the base class to create new "stateful" clients on top of the base client. See the implementation of Chat_Bot_Client for an example
Contributing
Contributions are welcome! Please not the following when contributing:
-
Unittests must be added under the
tests/
directory for the PR to be approved. You can run unittests from the root project directory with the following command:python setup.py test
-
PRs cannot be merged without all unittests passing (they will execute automatically)
-
Merges to
main
will automatically create a new release on PyPi unless it is from a forked Repo
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 OpenAI API Python Client-1.0.5.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | d2941a958b963b906bffb07e358cdaf27b37688147834c296f7ce8f4cbaff79c |
|
MD5 | 0cb8e3db6d8831f7b3733d795735d554 |
|
BLAKE2b-256 | 77ec704a40d10ad188205012f6cbc3d256786a409104fe1a1fd01fc549d0796d |
Hashes for OpenAI_API_Python_Client-1.0.5-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 81310c7c48e0b9db520c9eb18b9db0e209b5d6cf25ef692fe204716f72f7fe96 |
|
MD5 | b0422a3f4d37a381e981af69e250a228 |
|
BLAKE2b-256 | 36f3f8ab8a9f0c16e26640bd3b45a0fdba2d2e592abb4b5b06241ff680658796 |