annotate src/lib/ai/claude/Ai_chat.luan @ 35:3117876debca

ai_first_message in textarea
author Franklin Schmidt <fschmidt@gmail.com>
date Tue, 05 Aug 2025 16:41:29 -0600
parents 0fb3488a017d
children 238a91f224b1
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
5
a970b7a01a74 start ai
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
1 local Luan = require "luan:Luan.luan"
a970b7a01a74 start ai
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
2 local error = Luan.error
6
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
3 local ipairs = Luan.ipairs or error()
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
4 local type = Luan.type or error()
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
5 local String = require "luan:String.luan"
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
6 local starts_with = String.starts_with or error()
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
7 local Html = require "luan:Html.luan"
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
8 local html_encode = Html.encode or error()
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
9 local Parsers = require "luan:Parsers.luan"
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
10 local json_parse = Parsers.json_parse or error()
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
11 local json_string = Parsers.json_string or error()
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
12 local Claude = require "site:/lib/ai/claude/Claude.luan"
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
13 local claude_chat = Claude.chat or error()
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
14 local Logging = require "luan:logging/Logging.luan"
19
Franklin Schmidt <fschmidt@gmail.com>
parents: 16
diff changeset
15 local logger = Logging.logger "claude/Ai_chat"
5
a970b7a01a74 start ai
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
16
a970b7a01a74 start ai
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
17
19
Franklin Schmidt <fschmidt@gmail.com>
parents: 16
diff changeset
18 local Ai_chat = {}
5
a970b7a01a74 start ai
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
19
19
Franklin Schmidt <fschmidt@gmail.com>
parents: 16
diff changeset
20 function Ai_chat.output_system_prompt(thread)
9
Franklin Schmidt <fschmidt@gmail.com>
parents: 7
diff changeset
21 thread = json_parse(thread)
Franklin Schmidt <fschmidt@gmail.com>
parents: 7
diff changeset
22 local system_prompt = thread.system or error
Franklin Schmidt <fschmidt@gmail.com>
parents: 7
diff changeset
23 system_prompt = html_encode(system_prompt)
Franklin Schmidt <fschmidt@gmail.com>
parents: 7
diff changeset
24 %><%=system_prompt%><%
Franklin Schmidt <fschmidt@gmail.com>
parents: 7
diff changeset
25 end
Franklin Schmidt <fschmidt@gmail.com>
parents: 7
diff changeset
26
34
0fb3488a017d show_text
Franklin Schmidt <fschmidt@gmail.com>
parents: 29
diff changeset
27 function Ai_chat.output_messages_html(show_text,thread,old_thread)
9
Franklin Schmidt <fschmidt@gmail.com>
parents: 7
diff changeset
28 thread = json_parse(thread)
Franklin Schmidt <fschmidt@gmail.com>
parents: 7
diff changeset
29 local messages = thread.messages or error
13
65bd7e245c63 add html
Franklin Schmidt <fschmidt@gmail.com>
parents: 12
diff changeset
30 local n = 0
65bd7e245c63 add html
Franklin Schmidt <fschmidt@gmail.com>
parents: 12
diff changeset
31 if old_thread ~= nil then
65bd7e245c63 add html
Franklin Schmidt <fschmidt@gmail.com>
parents: 12
diff changeset
32 old_thread = json_parse(old_thread)
65bd7e245c63 add html
Franklin Schmidt <fschmidt@gmail.com>
parents: 12
diff changeset
33 local old_messages = old_thread.messages or error
65bd7e245c63 add html
Franklin Schmidt <fschmidt@gmail.com>
parents: 12
diff changeset
34 n = #old_messages
65bd7e245c63 add html
Franklin Schmidt <fschmidt@gmail.com>
parents: 12
diff changeset
35 end
34
0fb3488a017d show_text
Franklin Schmidt <fschmidt@gmail.com>
parents: 29
diff changeset
36 local checked = show_text and "checked" or ""
13
65bd7e245c63 add html
Franklin Schmidt <fschmidt@gmail.com>
parents: 12
diff changeset
37 for i, message in ipairs(messages) do
65bd7e245c63 add html
Franklin Schmidt <fschmidt@gmail.com>
parents: 12
diff changeset
38 if i <= n then
65bd7e245c63 add html
Franklin Schmidt <fschmidt@gmail.com>
parents: 12
diff changeset
39 continue
65bd7e245c63 add html
Franklin Schmidt <fschmidt@gmail.com>
parents: 12
diff changeset
40 end
6
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
41 local role = message.role or error()
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
42 local who
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
43 if role=="assistant" then
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
44 who = "Claude"
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
45 elseif role=="user" then
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
46 who = "You"
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
47 else
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
48 error(role)
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
49 end
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
50 local function output(text)
9
Franklin Schmidt <fschmidt@gmail.com>
parents: 7
diff changeset
51 text = html_encode(text)
6
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
52 %>
7
255c36830154 chat tools
Franklin Schmidt <fschmidt@gmail.com>
parents: 6
diff changeset
53 <h3><%=who%></h3>
34
0fb3488a017d show_text
Franklin Schmidt <fschmidt@gmail.com>
parents: 29
diff changeset
54 <div role="<%=role%>">
0fb3488a017d show_text
Franklin Schmidt <fschmidt@gmail.com>
parents: 29
diff changeset
55 <div message markdown><%=text%></div>
0fb3488a017d show_text
Franklin Schmidt <fschmidt@gmail.com>
parents: 29
diff changeset
56 <% if role=="assistant" then %>
0fb3488a017d show_text
Franklin Schmidt <fschmidt@gmail.com>
parents: 29
diff changeset
57 <div controls>
0fb3488a017d show_text
Franklin Schmidt <fschmidt@gmail.com>
parents: 29
diff changeset
58 <audio controls preload=none></audio>
0fb3488a017d show_text
Franklin Schmidt <fschmidt@gmail.com>
parents: 29
diff changeset
59 <label clickable><input type=checkbox name=show_text <%=checked%> >Show text</label>
0fb3488a017d show_text
Franklin Schmidt <fschmidt@gmail.com>
parents: 29
diff changeset
60 </div>
0fb3488a017d show_text
Franklin Schmidt <fschmidt@gmail.com>
parents: 29
diff changeset
61 <% end %>
0fb3488a017d show_text
Franklin Schmidt <fschmidt@gmail.com>
parents: 29
diff changeset
62 </div>
6
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
63 <%
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
64 end
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
65 local content = message.content or error()
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
66 if type(content) == "string" then
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
67 output(content)
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
68 else
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
69 for _, part in ipairs(content) do
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
70 if part.type=="text" then
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
71 local text = part.text or error()
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
72 output(text)
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
73 end
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
74 end
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
75 end
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
76 end_for
5
a970b7a01a74 start ai
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
77 end
a970b7a01a74 start ai
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
78
9
Franklin Schmidt <fschmidt@gmail.com>
parents: 7
diff changeset
79
35
3117876debca ai_first_message in textarea
Franklin Schmidt <fschmidt@gmail.com>
parents: 34
diff changeset
80 function Ai_chat.init(system_prompt)
3117876debca ai_first_message in textarea
Franklin Schmidt <fschmidt@gmail.com>
parents: 34
diff changeset
81 local thread = {
3117876debca ai_first_message in textarea
Franklin Schmidt <fschmidt@gmail.com>
parents: 34
diff changeset
82 system = system_prompt
3117876debca ai_first_message in textarea
Franklin Schmidt <fschmidt@gmail.com>
parents: 34
diff changeset
83 messages = {nil}
3117876debca ai_first_message in textarea
Franklin Schmidt <fschmidt@gmail.com>
parents: 34
diff changeset
84 }
3117876debca ai_first_message in textarea
Franklin Schmidt <fschmidt@gmail.com>
parents: 34
diff changeset
85 return json_string(thread)
3117876debca ai_first_message in textarea
Franklin Schmidt <fschmidt@gmail.com>
parents: 34
diff changeset
86 end
3117876debca ai_first_message in textarea
Franklin Schmidt <fschmidt@gmail.com>
parents: 34
diff changeset
87
3117876debca ai_first_message in textarea
Franklin Schmidt <fschmidt@gmail.com>
parents: 34
diff changeset
88 function Ai_chat.has_messages(thread)
3117876debca ai_first_message in textarea
Franklin Schmidt <fschmidt@gmail.com>
parents: 34
diff changeset
89 thread = json_parse(thread)
3117876debca ai_first_message in textarea
Franklin Schmidt <fschmidt@gmail.com>
parents: 34
diff changeset
90 return #thread.messages > 0
3117876debca ai_first_message in textarea
Franklin Schmidt <fschmidt@gmail.com>
parents: 34
diff changeset
91 end
3117876debca ai_first_message in textarea
Franklin Schmidt <fschmidt@gmail.com>
parents: 34
diff changeset
92
3117876debca ai_first_message in textarea
Franklin Schmidt <fschmidt@gmail.com>
parents: 34
diff changeset
93 function Ai_chat.ask(thread,input)
3117876debca ai_first_message in textarea
Franklin Schmidt <fschmidt@gmail.com>
parents: 34
diff changeset
94 thread = json_parse(thread)
9
Franklin Schmidt <fschmidt@gmail.com>
parents: 7
diff changeset
95 local messages = thread.messages or error
6
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
96 messages[#messages+1] = {
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
97 role = "user"
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
98 content = input
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
99 }
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
100 --[=[
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
101 messages[#messages+1] = {
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
102 role = "assistant"
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
103 content = [[
13
65bd7e245c63 add html
Franklin Schmidt <fschmidt@gmail.com>
parents: 12
diff changeset
104 hello
6
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
105 ]]
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
106 }
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
107 if true then
25
3a80ddafe5a4 courses work
Franklin Schmidt <fschmidt@gmail.com>
parents: 20
diff changeset
108 return
6
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
109 end
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
110 --]=]
20
Franklin Schmidt <fschmidt@gmail.com>
parents: 19
diff changeset
111 -- logger.info(json_string(thread))
9
Franklin Schmidt <fschmidt@gmail.com>
parents: 7
diff changeset
112 local resultJson = claude_chat(thread)
6
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
113 local result = json_parse(resultJson)
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
114 -- logger.info(json_string(result))
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
115 result.type == "message" or error()
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
116 result.role == "assistant" or error()
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
117 result.stop_reason == "end_turn" or result.stop_reason == "tool_use" or error()
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
118 local content = result.content or error()
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
119 messages[#messages+1] = {
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
120 role = "assistant"
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
121 content = content
025bb19b65b1 use claude
Franklin Schmidt <fschmidt@gmail.com>
parents: 5
diff changeset
122 }
9
Franklin Schmidt <fschmidt@gmail.com>
parents: 7
diff changeset
123 return json_string(thread)
5
a970b7a01a74 start ai
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
124 end
a970b7a01a74 start ai
Franklin Schmidt <fschmidt@gmail.com>
parents:
diff changeset
125
19
Franklin Schmidt <fschmidt@gmail.com>
parents: 16
diff changeset
126 return Ai_chat