llm-mixer / components /chat_operator.js
Hansimov's picture
:recycle: [Refactor] Move ScreenScroller and StreamJsonizer
4e6658c
raw
history blame
4.38 kB
import { Messager, MessagerList } from "./messager.js";
import { transform_footnote } from "../networks/stream_jsonizer.js";
import { screen_scroller } from "./screen_scroller.js";
let messagers_container = $("#messagers-container");
let available_models_select = $("#available-models-select");
let temperature_select = $("#temperature-select");
let messager_list = new MessagerList(messagers_container);
let chat_history = [messager_list];
let md_to_html_converter = new showdown.Converter();
md_to_html_converter.setFlavor("github");
export function get_active_messager_list() {
return chat_history[chat_history.length - 1];
}
export function get_latest_messager() {
return get_active_messager_list().messagers.slice(-1)[0];
}
export function create_messager(
role,
content = "",
model = "",
temperature = ""
) {
let message = {
role: role,
content: content,
model: model,
temperature: temperature,
};
let messager = new Messager(message);
get_active_messager_list().push(messager);
screen_scroller.scroll_to_bottom();
}
export function get_selected_llm_model() {
return available_models_select.val();
}
export function get_selected_temperature() {
return Number(temperature_select.val());
}
export function get_latest_messager_container() {
return get_active_messager_list().messagers_container.children().last();
}
export function get_latest_message_content_displayer() {
return get_latest_messager_container().find(".content-displayer");
}
export function get_latest_user_messager() {
return $(".message-user").last();
}
export function get_latest_assistant_messager() {
return $(".message-assistant").last();
}
export function start_latest_message_animation() {
get_latest_assistant_messager()
.find(".content-displayer")
.addClass("inferring");
get_latest_assistant_messager()
.find(".button-group")
.find(".regenerate-button")
.find("i")
.addClass("fa-spin-fast");
}
export function stop_latest_message_animation() {
get_latest_assistant_messager()
.find(".content-displayer")
.removeClass("inferring");
get_latest_assistant_messager()
.find(".button-group")
.find(".regenerate-button")
.find("i")
.removeClass("fa-spin-fast");
}
export function get_request_messages() {
return get_active_messager_list().get_request_messages();
}
export function pop_messager(n = 2) {
return get_active_messager_list().pop(n);
}
export function update_message(json_chunks, content_displayer = null) {
let content_displayer_updater = new ContentDisplayerUpdater(
content_displayer
);
json_chunks.forEach(function (item) {
let choice = item.choices[0];
let content = choice.delta.content;
let role = choice.delta.role;
let finish_reason = choice.finish_reason;
if (role) {
console.log("role: " + role);
}
if (content) {
content_displayer_updater.update_with_chunk_content(content);
}
if (finish_reason === "stop") {
console.log("[STOP]");
}
// console.log(item);
});
return json_chunks;
}
export function create_new_chat_session() {
let new_messager_list = new MessagerList(messagers_container);
chat_history.push(new_messager_list);
messagers_container.empty();
}
export class ContentDisplayerUpdater {
constructor(content_displayer = null) {
if (content_displayer === null) {
self.content_displayer = get_latest_message_content_displayer();
} else {
self.content_displayer = content_displayer;
}
}
update_with_chunk_content(content) {
self.content_displayer.data(
"raw_content",
self.content_displayer.data("raw_content") + content
);
get_active_messager_list().messagers.slice(-1)[0].message.content +=
content;
self.content_displayer.html(
md_to_html_converter.makeHtml(
transform_footnote(self.content_displayer.data("raw_content"))
)
);
self.content_displayer
.find("table")
.addClass("table table-bordered table-hover");
screen_scroller.scroll_to_bottom();
}
}