diff options
| author | Martin Bielik <mx.bielik@gmail.com> | 2023-04-13 19:00:04 +0200 |
|---|---|---|
| committer | Martin Bielik <mx.bielik@gmail.com> | 2023-04-13 19:00:04 +0200 |
| commit | 9a872803baf77b9a687b263d11cb2ac297b57a71 (patch) | |
| tree | 1cb76aff662a1bc4bb59600274293a6db927d7ed /py/complete.py | |
| parent | 0058b3f6509ac2632b09048c02d3ff4fa17bdfa9 (diff) | |
| download | vim-ai-9a872803baf77b9a687b263d11cb2ac297b57a71.tar.gz | |
poc: removing openai dependency
Diffstat (limited to 'py/complete.py')
| -rw-r--r-- | py/complete.py | 8 |
1 files changed, 2 insertions, 6 deletions
diff --git a/py/complete.py b/py/complete.py index f42cebc..8a8bf7d 100644 --- a/py/complete.py +++ b/py/complete.py @@ -8,8 +8,6 @@ request_options = make_request_options(config_options) prompt = vim.eval("prompt").strip() -openai.api_key = load_api_key() - def complete_engine(prompt): request = { 'stream': True, @@ -17,7 +15,7 @@ def complete_engine(prompt): **request_options } printDebug("[engine-complete] request: {}", request) - response = openai.Completion.create(**request) + response = openai_request('https://api.openai.com/v1/completions', request) def map_chunk(resp): printDebug("[engine-complete] response: {}", resp) return resp['choices'][0].get('text', '') @@ -35,7 +33,7 @@ def chat_engine(prompt): **request_options } printDebug("[engine-chat] request: {}", request) - response = openai.ChatCompletion.create(**request) + response = openai_request('https://api.openai.com/v1/chat/completions', request) def map_chunk(resp): printDebug("[engine-chat] response: {}", resp) return resp['choices'][0]['delta'].get('content', '') @@ -52,5 +50,3 @@ try: render_text_chunks(text_chunks) except KeyboardInterrupt: vim.command("normal! a Ctrl-C...") -except openai.error.Timeout: - vim.command("normal! aRequest timeout...") |