blob: ba460c4fe14a344b439da75fd025181742f9852a (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
|
import sys
import os
import openai
config_file_path = os.path.join(os.path.expanduser("~"), ".config/openai.token")
api_key = os.getenv("OPENAI_API_KEY")
try:
with open(config_file_path, 'r') as file:
api_key = file.read()
except Exception:
pass
openai.api_key = api_key.strip()
prompt = "".join(sys.stdin.readlines())
completion = openai.Completion.create(
model="text-davinci-003",
prompt=prompt,
max_tokens=1000,
temperature=0.1
)
print(completion.choices[0].text)
|
