From 0faae7b1fbec281f5a645f25e5012d9deb307f56 Mon Sep 17 00:00:00 2001 From: Martin Bielik Date: Sun, 12 Mar 2023 19:47:21 +0100 Subject: getting rid of global dependencies --- py/complete.py | 29 ++++++++++++++++++----------- 1 file changed, 18 insertions(+), 11 deletions(-) (limited to 'py/complete.py') diff --git a/py/complete.py b/py/complete.py index ba460c4..409d47f 100644 --- a/py/complete.py +++ b/py/complete.py @@ -1,6 +1,8 @@ +import requests import sys import os -import openai + +prompt = vim.eval("prompt") config_file_path = os.path.join(os.path.expanduser("~"), ".config/openai.token") @@ -12,15 +14,20 @@ try: except Exception: pass -openai.api_key = api_key.strip() - -prompt = "".join(sys.stdin.readlines()) +api_key = api_key.strip() -completion = openai.Completion.create( - model="text-davinci-003", - prompt=prompt, - max_tokens=1000, - temperature=0.1 -) +url = "https://api.openai.com/v1/completions" +headers = { + 'Content-Type': 'application/json', + 'Authorization': f"Bearer {api_key}" +} +data = { + "model": "text-davinci-003", + "prompt":prompt, + "max_tokens": 1000, + "temperature": 0.1 +} +response = requests.post(url, headers=headers, json=data) +response = response.json() -print(completion.choices[0].text) +output = response['choices'][0]['text'] -- cgit v1.2.3