Hansimov commited on
Commit
09f5b61
1 Parent(s): 786b7d7

:gem: [Feature] Show agent nickname and model in assistant message

Browse files
components/buttons_binder.js CHANGED
@@ -102,7 +102,11 @@ class SendUserInputButtonBinder {
102
  $("#user-input").val("");
103
  $("#user-input").css("height", "auto");
104
  if (get_selected_llm_model() == "notes") {
105
- create_messager("user", user_input_content);
 
 
 
 
106
  } else {
107
  this.requester = new ChatCompletionsRequester({
108
  prompt: user_input_content,
 
102
  $("#user-input").val("");
103
  $("#user-input").css("height", "auto");
104
  if (get_selected_llm_model() == "notes") {
105
+ create_messager({
106
+ role: "user",
107
+ content: user_input_content,
108
+ nickname: "notes",
109
+ });
110
  } else {
111
  this.requester = new ChatCompletionsRequester({
112
  prompt: user_input_content,
components/chat_operator.js CHANGED
@@ -19,11 +19,17 @@ export function get_latest_messager() {
19
  return get_active_messager_list().messagers.slice(-1)[0];
20
  }
21
 
22
- export function create_messager(role, content = "", model = "") {
 
 
 
 
 
23
  let message = {
24
  role: role,
25
  content: content,
26
  model: model,
 
27
  };
28
  let messager = new Messager(message);
29
  get_active_messager_list().push(messager);
 
19
  return get_active_messager_list().messagers.slice(-1)[0];
20
  }
21
 
22
+ export function create_messager({
23
+ role,
24
+ content = "",
25
+ model = "",
26
+ nickname = "",
27
+ } = {}) {
28
  let message = {
29
  role: role,
30
  content: content,
31
  model: model,
32
+ nickname: nickname,
33
  };
34
  let messager = new Messager(message);
35
  get_active_messager_list().push(messager);
components/messager.js CHANGED
@@ -19,11 +19,7 @@ class MessagerViewer {
19
  this.role_displayer = $("<div>")
20
  .addClass("role-displayer")
21
  .css("z-index", "1");
22
- if (this.message.role === "user") {
23
- this.role_displayer.append("You");
24
- } else {
25
- this.role_displayer.append(this.message.model);
26
- }
27
  }
28
  create_content_displayer() {
29
  this.content_displayer = $("<div>")
 
19
  this.role_displayer = $("<div>")
20
  .addClass("role-displayer")
21
  .css("z-index", "1");
22
+ this.role_displayer.append(this.message.nickname);
 
 
 
 
23
  }
24
  create_content_displayer() {
25
  this.content_displayer = $("<div>")
networks/llm_requester.js CHANGED
@@ -28,7 +28,6 @@ export class ChatCompletionsRequester {
28
  openai_endpoint || this.extract_openai_endpoint_and_model()[0];
29
  this.model = model || this.extract_openai_endpoint_and_model()[1];
30
  this.system_prompt = this.agent_info.system_prompt;
31
-
32
  this.temperature = temperature || this.agent_info.temperature;
33
  this.top_p = top_p || this.agent_info.top_p;
34
  this.max_output_tokens =
@@ -83,8 +82,18 @@ export class ChatCompletionsRequester {
83
  };
84
  }
85
  create_messager_components() {
86
- create_messager("user", this.prompt);
87
- create_messager("assistant", "", this.model);
 
 
 
 
 
 
 
 
 
 
88
  }
89
  async handle_read_stream_data(reader) {
90
  let buffer = "";
 
28
  openai_endpoint || this.extract_openai_endpoint_and_model()[0];
29
  this.model = model || this.extract_openai_endpoint_and_model()[1];
30
  this.system_prompt = this.agent_info.system_prompt;
 
31
  this.temperature = temperature || this.agent_info.temperature;
32
  this.top_p = top_p || this.agent_info.top_p;
33
  this.max_output_tokens =
 
82
  };
83
  }
84
  create_messager_components() {
85
+ create_messager({
86
+ role: "user",
87
+ content: this.prompt,
88
+ model: "",
89
+ nickname: "You",
90
+ });
91
+ create_messager({
92
+ role: "assistant",
93
+ content: "",
94
+ model: this.model,
95
+ nickname: `${this.agent_info.name} (${this.model})`,
96
+ });
97
  }
98
  async handle_read_stream_data(reader) {
99
  let buffer = "";