blob: d7f3c0369055ecd2a5624a38f2b5f87eea205c87 (
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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
|
;;; kaagum --- Tiny, security-focused AI agent in Guile
;;; Copyright © 2026 Arun Isaac <arunisaac@systemreboot.net>
;;;
;;; This file is part of kaagum.
;;;
;;; kaagum 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 of the License, or
;;; (at your option) any later version.
;;;
;;; kaagum 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 kaagum. If not, see <https://www.gnu.org/licenses/>.
(use-modules (srfi srfi-64)
(kaagum monads)
(kaagum openai)
(kaagum tea))
(define llm-response
(@@ (kaagum tea) llm-response))
(define acp-effects
(@@ (kaagum tea) acp-effects))
(define state
(@@ (kaagum tea) state))
(define session
(@@ (kaagum tea) session))
(define %models
`(("foo" . ,(model "foo" "Foobar" (expt 10 6)))))
(test-begin "tea")
(test-assert "handle LLM response"
(run-with-state
(acp-effects (llm-response 0
`(("id" . "gen-1793743825")
("model" . "anthropic/claude-haiku-5.5")
("choices"
.
#((("message"
("role" . "assistant")
("content" . "Hello! How can I help you today?")))))
("usage"
("prompt_tokens" . 1770)
("completion_tokens" . 13)
("total_tokens" . 1783)
("cost" . 3.601e-4)
("prompt_tokens_details"
("cached_tokens" . 0)
("cache_write_tokens" . 1766))
("completion_tokens_details"
("reasoning_tokens" . 0)))))
%models
'())
(state #f 0 1 '()
`((0 . ,(session (getcwd) "foo"))))))
(test-assert "handle LLM response without \"cached_tokens\" field in usage"
(run-with-state
(acp-effects (llm-response 0
`(("id" . "gen-1793743825")
("model" . "anthropic/claude-haiku-5.5")
("choices"
.
#((("message"
("role" . "assistant")
("content" . "Hello! How can I help you today?")))))
("usage"
("prompt_tokens" . 1770)
("completion_tokens" . 13)
("total_tokens" . 1783)
("cost" . 3.601e-4)
("prompt_tokens_details"
("cache_write_tokens" . 1766))
("completion_tokens_details"
("reasoning_tokens" . 0)))))
%models
'())
(state #f 0 1 '()
`((0 . ,(session (getcwd) "foo"))))))
(test-end "tea")
|
