tristan/gptel
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

gptel: gptel-backend-url can accept functions

Browse source 
* gptel.el (gptel--url-get-response): If the backend-url is a
function, call it to find the full url to query.

* gptel-gemini.el: Gemini uses different urls for
streaming/oneshot responses.  Set the backend-url to a function to
account for the value of gptel-stream.  This is also safer than
before as the API key is not stored as part of a static url string
in memory. Fix #153.

* gptel-curl.el (gptel-curl--get-args): If the backend-url is a
function, call it to find the full url to query.
This commit is contained in:
Karthik Chikmagalur 2023-12-22 16:25:32 -08:00
parent 4d01dddf7d
commit 2e92c0303c
3 changed files with 19 additions and 22 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
gptel-curl.el
View file

@ -49,16 +49,17 @@
"Produce list of arguments for calling Curl.
PROMPTS is the data to send, TOKEN is a unique identifier."
(let* ((url (gptel-backend-url gptel-backend))
(let* ((url (let ((backend-url (gptel-backend-url gptel-backend)))
(if (functionp backend-url)
(funcall backend-url) backend-url)))
(data (encode-coding-string
(json-encode (gptel--request-data gptel-backend prompts))
'utf-8))
(headers
(append '(("Content-Type" . "application/json"))
(when-let ((backend-header (gptel-backend-header gptel-backend)))
(if (functionp backend-header)
(funcall backend-header)
backend-header)))))
(when-let ((header (gptel-backend-header gptel-backend)))
(if (functionp header)
(funcall header) header)))))
(append
gptel-curl--common-args
(list (format "-w(%s . %%{size_header})" token))

19
gptel-gemini.el
View file

@ -139,19 +139,14 @@ function that returns the key."
:protocol protocol
:endpoint endpoint
:stream stream
:key key
:url
(let ((key (cond
((functionp key) (funcall key))
((symbolp key) (symbol-value key))
((stringp key) key)
(t (user-error "gptel-make-gemini: arg :key is malformed")))))
(if protocol
(concat protocol "://" host endpoint
(if stream
"streamGenerateContent"
"generateContent")
"?key=" key)
(concat host endpoint "?key=" key))))))
(lambda ()
(concat protocol "://" host endpoint
(if gptel-stream
"streamGenerateContent"
"generateContent")
"?key=" (gptel--get-api-key))))))
(prog1 backend
(setf (alist-get name gptel--known-backends
nil nil #'equal)

11
gptel.el
View file

@ -979,16 +979,17 @@ the response is inserted into the current buffer after point."
(url-request-method "POST")
(url-request-extra-headers
(append '(("Content-Type" . "application/json"))
(when-let ((backend-header (gptel-backend-header gptel-backend)))
(if (functionp backend-header)
(funcall backend-header)
backend-header))))
(when-let ((header (gptel-backend-header gptel-backend)))
(if (functionp header)
(funcall header) header))))
(url-request-data
(encode-coding-string
(json-encode (gptel--request-data
gptel-backend (plist-get info :prompt)))
'utf-8)))
(url-retrieve (gptel-backend-url gptel-backend)
(url-retrieve (let ((backend-url (gptel-backend-url gptel-backend)))
(if (functionp backend-url)
(funcall backend-url) backend-url))
(lambda (_)
(pcase-let ((`(,response ,http-msg ,error)
(gptel--url-parse-response backend (current-buffer))))