22#define CHATMODEL_H
33
44#include " database.h"
5+ #include " toolcallparser.h"
56#include " utils.h"
67#include " xlsxtomd.h"
78
1112#include < QBuffer>
1213#include < QByteArray>
1314#include < QDataStream>
15+ #include < QJsonDocument>
1416#include < QHash>
1517#include < QList>
1618#include < QObject>
@@ -73,15 +75,16 @@ struct ChatItem
7375 Q_PROPERTY (QString value MEMBER value)
7476
7577 // prompts and responses
76- Q_PROPERTY (int peerIndex MEMBER peerIndex)
78+ Q_PROPERTY (int peerIndex MEMBER peerIndex)
79+ Q_PROPERTY (QString content READ content )
7780
7881 // prompts
7982 Q_PROPERTY (QList<PromptAttachment> promptAttachments MEMBER promptAttachments)
8083 Q_PROPERTY (QString bakedPrompt READ bakedPrompt )
8184
8285 // responses
83- Q_PROPERTY (bool isCurrentResponse MEMBER isCurrentResponse)
84- Q_PROPERTY (bool isError MEMBER isError )
86+ Q_PROPERTY (bool isCurrentResponse MEMBER isCurrentResponse)
87+ Q_PROPERTY (bool isError MEMBER isError )
8588
8689 // responses (DataLake)
8790 Q_PROPERTY (QString newResponse MEMBER newResponse )
@@ -151,6 +154,104 @@ struct ChatItem
151154 return parts.join (QString ());
152155 }
153156
157+ bool isToolCall () const
158+ {
159+ if (type () != Type::Response)
160+ return false ;
161+
162+ ToolCallParser parser;
163+ parser.update (value);
164+ return parser.state () != ToolCallParser::None;
165+ }
166+
167+ QString content () const
168+ {
169+ if (type () == Type::System || type () == Type::Prompt)
170+ return value;
171+
172+ // This only returns a string for the code interpreter
173+ if (type () == Type::ToolResponse)
174+ return toolResult ();
175+
176+ // Otherwise we parse if this is a toolcall
177+ ToolCallParser parser;
178+ parser.update (value);
179+
180+ // If no tool call is detected, return the original value
181+ if (parser.startIndex () < 0 )
182+ return value;
183+
184+ // Constants for identifying and formatting the code interpreter tool call
185+ static const QString codeInterpreterPrefix = " <tool_call>{\" name\" : \" javascript_interpret\" , \" parameters\" : {\" code\" :\" "
186+ static const QString codeInterpreterSuffix = " \" }}</tool_call>"
187+ static const QString formattedPrefix = " ```javascript\n "
188+ static const QString formattedSuffix = " ```"
189+
190+ QString beforeToolCall = value.left (parser.startIndex ());
191+ QString toolCallString = value.mid (parser.startIndex ());
192+
193+ // Check if the tool call is a JavaScript interpreter tool call
194+ if (toolCallString.startsWith (codeInterpreterPrefix)) {
195+ int startCodeIndex = codeInterpreterPrefix.length ();
196+ int endCodeIndex = toolCallString.indexOf (codeInterpreterSuffix);
197+
198+ // Handle partial matches for codeInterpreterSuffix
199+ if (endCodeIndex == -1 ) {
200+ // If no complete match for suffix, search for a partial match
201+ for (int i = codeInterpreterSuffix.length () - 1 ; i >= 0 ; --i) {
202+ QString partialSuffix = codeInterpreterSuffix.left (i);
203+ if (toolCallString.endsWith (partialSuffix)) {
204+ endCodeIndex = toolCallString.length () - partialSuffix.length ();
205+ break ;
206+ }
207+ }
208+ }
209+
210+ // If a partial or full suffix match was found, adjust the `code` extraction
211+ if (endCodeIndex > startCodeIndex) {
212+ QString code = toolCallString.mid (startCodeIndex, endCodeIndex - startCodeIndex);
213+
214+ // Decode escaped JSON characters
215+ code.replace (" \\ n" " \n "
216+ code.replace (" \\ t" " \t "
217+ code.replace (" \\ r" " \r "
218+ code.replace (" \\\" " " \" "
219+ code.replace (" \\ '" " '"
220+ code.replace (" \\\\ " " \\ "
221+ code.replace (" \\ b" " \b "
222+ code.replace (" \\ f" " \f "
223+ code.replace (" \\ v" " \v "
224+ code.replace (" \\ /" " /"
225+
226+ // Format the code with JavaScript styling
227+ QString formattedCode = formattedPrefix + code + formattedSuffix;
228+
229+ // Return the text before the tool call and the formatted code
230+ return beforeToolCall + formattedCode;
231+ }
232+ }
233+
234+ // If it's not a code interpreter tool call, return the text before the tool call
235+ return beforeToolCall;
236+ }
237+
238+ QString toolResult () const
239+ {
240+ QJsonDocument doc = QJsonDocument::fromJson (value.toUtf8 ());
241+ if (doc.isNull () || !doc.isObject ())
242+ return QString ();
243+
244+ QJsonObject obj = doc.object ();
245+ if (!obj.contains (" tool"
246+ return QString ();
247+
248+ if (obj.value (" tool" toString () != " javascript_interpret"
249+ return QString ();
250+
251+ Q_ASSERT (obj.contains (" result"
252+ return " ```" value (" result" toString () + " ```"
253+ }
254+
154255 // TODO: Maybe we should include the model name here as well as timestamp?
155256 QString name;
156257 QString value;
@@ -205,6 +306,7 @@ class ChatModel : public QAbstractListModel
205306
206307 // prompts and responses
207308 PeerRole,
309+ ContentRole,
208310
209311 // prompts
210312 PromptAttachmentsRole,
@@ -215,6 +317,7 @@ class ChatModel : public QAbstractListModel
215317 ConsolidatedSourcesRole,
216318 IsCurrentResponseRole,
217319 IsErrorRole,
320+ IsToolCallRole,
218321
219322 // responses (DataLake)
220323 NewResponseRole,
@@ -291,6 +394,10 @@ class ChatModel : public QAbstractListModel
291394 return item.thumbsDownState ;
292395 case IsErrorRole:
293396 return item.type () == ChatItem::Type::Response && item.isError ;
397+ case ContentRole:
398+ return item.content ();
399+ case IsToolCallRole:
400+ return item.isToolCall ();
294401 }
295402
296403 return QVariant ();
@@ -311,6 +418,8 @@ class ChatModel : public QAbstractListModel
311418 { StoppedRole, " stopped"
312419 { ThumbsUpStateRole, " thumbsUpState"
313420 { ThumbsDownStateRole, " thumbsDownState"
421+ { ContentRole, " content"
422+ { IsToolCallRole, " isToolCall"
314423 };
315424 }
316425
@@ -350,7 +459,7 @@ class ChatModel : public QAbstractListModel
350459 if (promptIndex >= m_chatItems.size ())
351460 throw std::out_of_range (fmt::format (" index {} is out of range"
352461 auto &promptItem = m_chatItems[promptIndex];
353- if (promptItem.type () != ChatItem::Type::Prompt)
462+ if (promptItem.type () != ChatItem::Type::Prompt && promptItem. type () != ChatItem::Type::ToolResponse )
354463 throw std::invalid_argument (fmt::format (" item at index {} is not a prompt"
355464 promptItem.peerIndex = m_chatItems.count ();
356465 }
@@ -489,8 +598,12 @@ class ChatModel : public QAbstractListModel
489598 }
490599 }
491600 if (changed) {
492- emit dataChanged (createIndex (index , 0 ), createIndex (index , 0 ), {ValueRole});
493- emit valueChanged (index , value);
601+ emit dataChanged (createIndex (index , 0 ), createIndex (index , 0 ), {ValueRole, ContentRole, IsToolCallRole});
602+ // FIXME: This should be eliminated. It is necessary right now because of how we handle
603+ // display of text of chat items via ChatViewTextProcessor, but should go away when we
604+ // switch to a model/view arch and stop relying upon QTextDocument to display all the model's
605+ // content
606+ emit contentChanged (index );
494607 }
495608 }
496609
@@ -907,7 +1020,7 @@ class ChatModel : public QAbstractListModel
9071020
9081021Q_SIGNALS:
9091022 void countChanged ();
910- void valueChanged (int index, const QString &value );
1023+ void contentChanged (int index);
9111024 void hasErrorChanged (bool value);
9121025
9131026private:
0 commit comments