Skip to main content
Version: 0.7

superduper_openai

Superduper allows users to work with openai API models.

Installation

pip install superduper_openai

API

ClassDescription
superduper_openai.model.OpenAIEmbeddingOpenAI embedding predictor.
superduper_openai.model.OpenAIChatCompletionOpenAI chat completion predictor.
superduper_openai.model.OpenAIImageCreationOpenAI image creation predictor.
superduper_openai.model.OpenAIImageEditOpenAI image edit predictor.
superduper_openai.model.OpenAIAudioTranscriptionOpenAI audio transcription predictor.
superduper_openai.model.OpenAIAudioTranslationOpenAI audio translation predictor.

Examples

OpenAIEmbedding

from superduper_openai.model import OpenAIEmbedding
model = OpenAIEmbedding(identifier='text-embedding-ada-002')
model.predict('Hello, world!')

OpenAIChatCompletion

from superduper_openai.model import OpenAIChatCompletion
model = OpenAIChatCompletion(model='gpt-3.5-turbo', prompt='Hello, {context}')
model.predict('Hello, world!')

OpenAIImageCreation

from superduper_openai.model import OpenAIImageCreation

model = OpenAIImageCreation(
model="dall-e",
prompt="a close up, studio photographic portrait of a {context}",
response_format="url",
)
model.predict("cat")

OpenAIImageEdit

import io

from superduper_openai.model import OpenAIImageEdit

model = OpenAIImageEdit(
model="dall-e",
prompt="A celebration party at the launch of {context}",
response_format="url",
)
with open("test/material/data/rickroll.png", "rb") as f:
buffer = io.BytesIO(f.read())
model.predict(buffer, context=["superduper"])

OpenAIAudioTranscription

import io
from superduper_openai.model import OpenAIAudioTranscription
with open('test/material/data/test.wav', 'rb') as f:
buffer = io.BytesIO(f.read())
buffer.name = 'test.wav'
prompt = (
'i have some advice for you. write all text in lower-case.'
'only make an exception for the following words: {context}'
)
model = OpenAIAudioTranscription(identifier='whisper-1', prompt=prompt)
model.predict(buffer, context=['United States'])

OpenAIAudioTranslation

import io
from superduper_openai.model import OpenAIAudioTranslation
with open('test/material/data/german.wav', 'rb') as f:
buffer = io.BytesIO(f.read())
buffer.name = 'test.wav'
prompt = (
'i have some advice for you. write all text in lower-case.'
'only make an exception for the following words: {context}'
)
e = OpenAIAudioTranslation(identifier='whisper-1', prompt=prompt)
resp = e.predict(buffer, context=['Emmerich'])
buffer.close()