blob: 7b7318fcab46660e29827c4b5eef5ce358f70aaa (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
;;; ob-chatgpt-shell.el --- Org babel functions for ChatGPT evaluation -*- lexical-binding: t; -*-
;; Copyright (C) Alvaro Ramirez
;; Author: Alvaro Ramirez
;; URL: https://github.com/xenodium/chatgpt-shell
;;; License:
;; This program is free software; you can redistribute it and/or modify
;; it under the terms of the GNU General Public License as published by
;; the Free Software Foundation; either version 3, or (at your option)
;; any later version.
;;
;; This program is distributed in the hope that it will be useful,
;; but WITHOUT ANY WARRANTY; without even the implied warranty of
;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
;; GNU General Public License for more details.
;;
;; You should have received a copy of the GNU General Public License
;; along with GNU Emacs; see the file COPYING. If not, write to the
;; Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
;; Boston, MA 02110-1301, USA.
;;; Commentary:
;; Run and get responses from ChatGPT blocks using org babel.
;;
;; Install with:
;;
;; (require 'ob-chatgpt-shell)
;; (ob-chatgpt-shell-setup)
;;
;; Usage:
;;
;; #+begin_src chatgpt-shell
;; Hello
;; #+end_src
;;; Requirements:
;;; Code:
(require 'ob)
(require 'org)
(require 'chatgpt-shell)
(require 'map)
;; Aliasing enables block syntax highlighting.
(defalias 'chatgpt-shell-mode #'text-mode)
(defvar org-babel-default-header-args:chatgpt-shell '((:results . "raw")
(:version . nil)
(:preface . nil)))
(defun org-babel-execute:chatgpt-shell(body params)
"Execute a block of ChatGPT prompt in BODY with org-babel header PARAMS.
This function is called by `org-babel-execute-src-block'"
(message "executing ChatGPT source code block")
(if (map-elt params :preface)
(chatgpt-shell-post-chatgpt-messages
(vconcat (map-elt params :preface)
`(((role . "user")
(content . ,body))))
(map-elt params :version))
(chatgpt-shell-post-chatgpt-prompt body (map-elt params :version))))
(defun ob-chatgpt-shell-setup ()
"Set up babel ChatGPT support."
(org-babel-do-load-languages 'org-babel-load-languages
(append org-babel-load-languages
'((chatgpt-shell . t))))
(add-to-list 'org-src-lang-modes '("chatgpt-shell" . text)))
(provide 'ob-chatgpt-shell)
;;; ob-chatgpt-shell.el ends here
|
