openai-async
A light-weight, asynchronous client for OpenAI API - chat completion, text completion, image generation and embeddings.
For the full documentation, go to the openAI website.
Installation
pip install openai-async
Use
Chat completion
response = await openai_async.chat_complete(
_OPEN_AI_API_KEY,
timeout=2,
payload={
"model": "gpt-3.5-turbo",
"messages": [{"role": "user", "content": "Hello!"}],
},
)
print(response.json()["choices"][0]["message"])
>>> {"role": "assistant", "content": "\n\nHello there! How can I assist you today?"}
Visit the official documentation for more info.
Text completion
import openai_async
response = await openai_async.complete(
"<API KEY>",
timeout=2,
payload={
"model": "text-davinci-003",
"prompt": "Correct this sentence: Me like you.",
"temperature": 0.7,
},
)
print(response.json()["choices"][0]["text"].strip())
>>> "I like you."
Visit the official documentation for more info.
Image generation
import openai_async
import urllib.request
from PIL import Image
response = await openai_async.generate_img(
"<API KEY>",
timeout=8,
payload={
"prompt": "a white siamese cat",
"n": 1,
"size": "512x512"
},
)
urllib.request.urlretrieve(response.json()["data"][0]["url"], "img.png")
Image.open("img.png").show()
Embeddings
import openai_async
response = await openai_async.embeddings(
<API_KEY>,
timeout=2,
payload={"model": "text-embedding-ada-002", "input": "a cat"},
)
print(response.json()["data"][0]["embedding"])
>>> [-0.019408401,
-0.009246278,
-0.014390069,
-0.012294915,
-0.0025609178,
0.021252638,
...]
Visit the official documentation for more info.
Get an API key
To generate an openAI API key, while in the openAI website, click on your username in the top right corner, then go to "View API keys" and create a key.
Disclaimer
This repository has no connection whatsoever to openAI.
