Skip to main content

Python module for asynchronous interaction with OpenAI

Project description


Typing SVG


Download: https://pypi.org/project/ai-openchat/

Chat

  1. Example #1 Chat:
import asyncio

from ai_openchat import Model, AsyncOpenAI


async def chat():
    openai_client = AsyncOpenAI(token='API-KEY')
    resp = await openai_client.generate_message('Your request?', Model().chat())
    print(resp)


if __name__ == '__main__':
    asyncio.run(chat())
  1. Example #2 Movie to Emoji:
import asyncio

from ai_openchat import Model, AsyncOpenAI


async def movie_to_emoji():
    openai_client = AsyncOpenAI(token='API-KEY')
    resp = await openai_client.generate_message('Convert movie titles into emoji.\n\n'
                                            'Back to the Future: 👨👴🚗🕒 \n'
                                            'Batman: 🤵🦇 \n'
                                            'Transformers: 🚗🤖 \n'
                                            'Star Wars:', Model().movie_to_emoji())
    print(resp)
    # ⭐️⚔️


if __name__ == '__main__':
    asyncio.run(movie_to_emoji())
  1. Example #3 Custom chat
import asyncio

from ai_openchat import Model, AsyncOpenAI


async def image_generator():
    
    custom_model = Model(
        model="code-davinci-002",
        temperature=0,
        max_tokens=100,
        top_p=1.0,
        frequency_penalty=0.5,
        presence_penalty=0.0,
        stop=["You:"]
    )

    openai_client = AsyncOpenAI(token='API-KEY')
    resp = await openai_client.generate_message('Hello!', custom_model)
    print(resp)


if __name__ == '__main__':
    asyncio.run(image_generator())

Image

  1. Generate Image
import asyncio

from ai_openchat import ImageModel, AsyncOpenAI


async def image_generator():
    openai_client = AsyncOpenAI(token='API-KEY')
    resp = await openai_client.generate_image('Captain America', ImageModel().image())
    print(resp)


if __name__ == '__main__':
    asyncio.run(image_generator())
  1. Generate custom Image
import asyncio

from ai_openchat import ImageModel, AsyncOpenAI


async def image_generator():

    custom_model = ImageModel(n=1, size="1024x1024")

    openai_client = AsyncOpenAI(token='API-KEY')
    resp = await openai_client.generate_image('Captain America', custom_model)
    print(resp)


if __name__ == '__main__':
    asyncio.run(image_generator())

This project is an attempt to make an asynchronous library for convenient OpenAI management. You can check out the rest of the models here: https://beta.openai.com/examples.

Technologies

  • Python >= 3.8;
  • aiohttp >= 3.8

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

ai_openchat-1.1.8.tar.gz (8.1 kB view hashes)

Uploaded Source

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page