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

♻️: Untangle Gemini model and endpoint #212 (#213)

Browse source 
gptel-gemini.el (gptel-make-gemini): Decouple the Gemini model
from the API endpoint.  This is to support additional model
options in the future.
This commit is contained in:
Cash Prokop-Weaver 2024-03-10 21:02:42 -07:00 • committed by GitHub
parent 3d6147830e
commit 3935a6dcf8
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 15 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
gptel-gemini.el
View file

@ -115,7 +115,7 @@
(host "generativelanguage.googleapis.com")
(protocol "https")
(models '("gemini-pro"))
(endpoint "/v1beta/models/gemini-pro:"))
(endpoint "/v1beta/models"))
"Register a Gemini backend for gptel with NAME.
@ -126,8 +126,7 @@ CURL-ARGS (optional) is a list of additional Curl arguments.
HOST (optional) is the API host, defaults to
\"generativelanguage.googleapis.com\".
MODELS is a list of available model names. Currently only
\"gemini-pro\" is available.
MODELS is a list of available model names.
STREAM is a boolean to enable streaming responses, defaults to
false.
@ -135,8 +134,7 @@ false.
PROTOCOL (optional) specifies the protocol, \"https\" by default.
ENDPOINT (optional) is the API endpoint for completions, defaults to
\"/v1beta/models/gemini-pro:streamGenerateContent\" if STREAM is true and
\"/v1beta/models/gemini-pro:generateContent\" otherwise.
\"/v1beta/models\".
HEADER (optional) is for additional headers to send with each
request. It should be an alist or a function that retuns an
@ -156,18 +154,18 @@ function that returns the key."
:endpoint endpoint
:stream stream
:key key
:url
(if stream
(lambda ()
(concat protocol "://" host endpoint
(if gptel-stream
"streamGenerateContent"
"generateContent")
"?key=" (gptel--get-api-key)))
(lambda ()
(concat protocol "://" host endpoint
"generateContent" "?key="
(gptel--get-api-key)))))))
:url (lambda ()
(let ((method (if (and stream
gptel-stream)
"streamGenerateContent"
"generateContent")))
(format "%s://%s%s/%s:%s?key=%s"
protocol
host
endpoint
gptel-model
method
(gptel--get-api-key)))))))
(prog1 backend
(setf (alist-get name gptel--known-backends
nil nil #'equal)