using openai api directly

1 files changed, 26 insertions, 0 deletions

diff --git a/py/complete.py b/py/complete.py
new file mode 100644
index 0000000..ba460c4
--- /dev/null
+++ b/py/complete.py
@@ -0,0 +1,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)