jdormit/dotfiles
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Make Ollama config pull models dynamically

Browse Source
This commit is contained in:
Jeremy Dormitzer 2024-10-22 16:48:49 +00:00
parent 6448c9f70e
commit ae4178f270
1 changed files with 14 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
emacs/.emacs.d/config/init-ai.el
View File

@ -39,12 +39,22 @@
:stream t :stream t
:models '("gpt-4o" :models '("gpt-4o"
"gpt-4o-mini"))) "gpt-4o-mini")))
(defun ollama-models ()
(if (executable-find "ollama")
(->> (shell-command-to-string "ollama list")
(s-lines)
(cdr)
(mapcar (lambda (line)
(let ((parts (s-split-up-to " " line 1 t)))
(car parts))))
(-filter #'s-present?))
(list "llama3.1:latest"
"mistral-nemo:latest"
"gemma2:2b"
"tinyllama:latest")))
(defvar gptel-backend-ollama (gptel-make-ollama "Ollama" (defvar gptel-backend-ollama (gptel-make-ollama "Ollama"
:stream t :stream t
:models '("llama3.1:latest" :models (ollama-models)))
"mistral-nemo:latest"
"gemma2:2b"
"tinyllama:latest")))
(defvar gptel-backend-anthropic (gptel-make-anthropic "Claude" (defvar gptel-backend-anthropic (gptel-make-anthropic "Claude"
:stream t :stream t
:key (password-store-get "anthropic-api-key"))) :key (password-store-get "anthropic-api-key")))