Buy Me a Coffee? Your support is much appreciated!
demo.py
import openai
GPT3_MODELS = {
'davinci': 'text-davinci-003',
'curie': 'text-curie-001',
'babbage': 'text-babbage-001',
'ada': 'text-ada-001'
}
CODEX_MODELS = {
'davinci': 'code-davinci-002',
'cushman': 'code-cushman-001'
}
class OpenAIPlayground:
def __init__(self, api_key):
self.openai = openai
self.openai.api_key = api_key
self._conversation_log = ''
def grammar_checker(self, prompt, model=GPT3_MODELS['davinci'], temperature=0.0, max_tokens=1000, top_p=1.0, frequency_penalty=0.0, presence_penalty=0.0):
response = self.openai.Completion.create(
model=model,
prompt=prompt,
temperature=temperature,
max_tokens=max_tokens,
top_p=top_p,
frequency_penalty=frequency_penalty,
presence_penalty=presence_penalty,
n=1
)
result = {
'id': response.id,
'created': response.created,
'model': response.model,
'completion_tokens': response.usage.completion_tokens,
'prompt_tokens': response.usage.prompt_tokens,
'total_tokens': response.usage.total_tokens,
'outputs': response.choices[0].text,
'status': response.choices[0].finish_reason
}
return result