diff options
| author | Martin Bielik <mx.bielik@gmail.com> | 2023-10-21 19:04:57 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-10-21 19:04:57 +0200 |
| commit | ccf981974232636d08a48094bf9dad3812a43cc8 (patch) | |
| tree | 611803cd36432c6e2cb68edf4cf3f81d502c4615 /py/complete.py | |
| parent | 8f8083ba0eed23150020b698e74d9302f7212c5d (diff) | |
| parent | 55c4e2ec836e48552b52fb4b7878f7b50f67b53b (diff) | |
| download | vim-ai-ccf981974232636d08a48094bf9dad3812a43cc8.tar.gz | |
Merge pull request #59 from madox2/base-url-config
Custom APIs, closes #55, closes #51
Diffstat (limited to 'py/complete.py')
| -rw-r--r-- | py/complete.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/py/complete.py b/py/complete.py index c8d45fe..8386c09 100644 --- a/py/complete.py +++ b/py/complete.py @@ -17,7 +17,8 @@ def complete_engine(prompt): **openai_options } printDebug("[engine-complete] request: {}", request) - response = openai_request('https://api.openai.com/v1/completions', request, http_options) + url = config_options['endpoint_url'] + response = openai_request(url, request, http_options) def map_chunk(resp): printDebug("[engine-complete] response: {}", resp) return resp['choices'][0].get('text', '') @@ -35,7 +36,8 @@ def chat_engine(prompt): **openai_options } printDebug("[engine-chat] request: {}", request) - response = openai_request('https://api.openai.com/v1/chat/completions', request, http_options) + url = config_options['endpoint_url'] + response = openai_request(url, request, http_options) def map_chunk(resp): printDebug("[engine-chat] response: {}", resp) return resp['choices'][0]['delta'].get('content', '') |