Wrapper for ChatGPT API
Project description
Simple Wrapper For The ChatGPT API
This module is designed for asynchronous usage and provides a simple interface to the OpenAI ChatGPT API
It also provides a command-line interface for interactive usage.
While it does provide synchronous methods, it is recommended to use the asynchronous methods for better performance.
Features
- Asynchronous API
- Get/Stream responses
- Save/Load session history
- Change parameters on the fly
- Embed images in messages
- Register functions to
tool_calls
with ease - Interactive command-line interface
Installation
pip install -U ngptbot
Usage
Command Line
Configuration
gptbot
requires a model and an API key to be set.
You can use config file, command parameters or interactive input to set these values.
By default, gptbot
will look for files named config.json
and session.json
in the user config and cache directory.
gptbot -c /path/to/config.json # Use a config file
gptbot -m "gpt-3.5-turbo" # Set model, input API key interactively
gptbot -k /path/to/config.json # create config file
Parameters
-c
,--config
: Path to a JSON file-m
,--model
: Model name-k
,--create-config
: Create a config file interactively-s
,--session
: Session history file-V
,--version
: Show version
Precedence: Interactive input > Command parameters > Config file
Interactive Mode
This mode mimics a chat interface. You can type your message and get a response.
Commands are started with a /
and are case-sensitive.
/exit
: Exit the program/save <path>
: Save the session history to a file/load <path>
: Load a session history from a file/rollback <step>
: Rollback to the previous state/clear
: Clear the session history/role
: Switch role/model
: Switch model/tokens
: Show used tokens/help
: Show help
To embed an image, insert #image(<url>)
in your message.
Use double #
to escape embedding.
await bot.send("""
What are these?
#image(https://example.com/image.png)
#image(file:///path/to/image.png)
#image(base64://<base64-encoded-image>)
""")
All URLs will be downloaded and embedded as base64-encoded images.
API
import asyncio
import gptbot
bot = gptbot.Bot(model="gpt-3.5-turbo", api_key="your-api-key")
async def main():
response = await bot.send("Hello, how are you?")
print(response)
async for r in bot.stream("I'm fine, thank you."):
print(r, end='')
async for res in bot.stream_raw("What is the answer to life, the universe, and everything?"):
"""
stream_raw returns a stream of `FullChunkResponse`.
It contains all raw data from the API.
"""
bot.send_sync("Goodbye!") # Synchronous version of `send`
# Bot class is stateless, to keep track of the conversation, use the `Session` class
session = bot.new_session()
# You can use session just like the bot
async for r in session.stream("Hello!"):
print(r, end='')
print(await session.send("Goodbye!"))
asyncio.run(main())
Registering Functions
Functions are under Bot.funcs
and can be registered with the add_func
decorator.
@bot.add_func
def my_func(data: str, default:int = 123): # parameters must be annotated, variadic parameters are not supported
"""Documentation goes here"""
return data # return value must be stringifiable
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.