28 lines
655 B
Python
28 lines
655 B
Python
import os, json, sys
|
|
import openai
|
|
|
|
from canvas_secrets import openai_org, openai_api_key
|
|
|
|
|
|
openai.organization = openai_org
|
|
openai.api_key = openai_api_key
|
|
#print(openai.Model.list())
|
|
|
|
my_prompt = "Write a series of texts trying to sell a pen to a stranger."
|
|
print(sys.argv)
|
|
exit
|
|
|
|
if len(sys.argv)>1:
|
|
my_prompt = " ".join(sys.argv[1:])
|
|
else:
|
|
print("Prompt: %s" % my_prompt)
|
|
|
|
my_model = "text-davinci-003"
|
|
|
|
# create a completion
|
|
completion = openai.Completion.create(engine=my_model, prompt=my_prompt, max_tokens=1000, temperature=1,top_p=1)
|
|
|
|
#print(completion)
|
|
#print(json.dumps(completion,indent=2))
|
|
print(completion.choices[0].text)
|
|
print() |