SimpleChatTC:ChatMessage: AssistantResponse into chat message class

Modify the constructor, newFrom and clear towards this goal.
This commit is contained in:
hanishkvc 2025-10-14 16:48:40 +05:30
parent 37faf8611a
commit 2c29c2d589
1 changed files with 19 additions and 8 deletions

View File

@ -38,23 +38,34 @@ class ApiEP {
} }
class AssistantResponse { class ChatMessage {
constructor(content="", toolname="", toolargs="", trimmedContent="") { /**
/** @type {Object<string, string>} */ * Represent a Message in the Chat
this.response = { content: content, toolname: toolname, toolargs: toolargs, trimmedContent: trimmedContent }; * @param {string} role
* @param {string} content
* @param {Array<any>} tool_calls
* @param {string} trimmedContent
*/
constructor(role = "", content="", tool_calls=[], trimmedContent="") {
/** @type {Object<string, any>} */
this.ns = { role: role, content: content, tool_calls: tool_calls }
this.trimmedContent = trimmedContent;
} }
/** /**
* Create a new instance from an existing instance * Create a new instance from an existing instance
* @param {AssistantResponse} old * @param {ChatMessage} old
*/ */
static newFrom(old) { static newFrom(old) {
return new AssistantResponse(old.response.content, old.response.toolname, old.response.toolargs, old.response.trimmedContent) return new ChatMessage(old.ns.role, old.ns.content, old.ns.tool_calls, old.trimmedContent)
} }
clear() { clear() {
this.response = { content: "", toolname: "", toolargs: "", trimmedContent: "" }; this.ns.role = "";
this.ns.content = "";
this.ns.tool_calls = [];
this.trimmedContent = "";
} }
/** /**
@ -71,7 +82,7 @@ class AssistantResponse {
} }
has_toolcall() { has_toolcall() {
if (this.response.toolname.trim() == "") { if (this.ns.tool_calls.trim() == "") {
return false return false
} }
return true return true