expandme commited on
Commit
7537b38
·
1 Parent(s): edd7b07

Fixing message format ? - What wind.surf will do ?

Browse files
Files changed (1) hide show
  1. app.py +8 -13
app.py CHANGED
@@ -84,24 +84,16 @@ def respond(
84
  # Load new model if changed
85
  if current_model is None or model_name not in str(current_model.model_path):
86
  current_model = load_model(model_name)
87
- # Update the interface title
88
- gr.Info(get_chat_title(model_name))
89
 
90
  # Start with system message
91
  messages = []
92
  if system_message and system_message.strip():
93
  messages.append({"role": "system", "content": system_message})
94
 
95
- # Convert history to messages format
96
- for msg in history:
97
- if isinstance(msg, tuple):
98
- if msg[0]:
99
- messages.append({"role": "user", "content": msg[0]})
100
- if msg[1]:
101
- messages.append({"role": "assistant", "content": msg[1]})
102
- else:
103
- messages.append(msg)
104
-
105
  # Add current message
106
  messages.append({"role": "user", "content": message})
107
 
@@ -173,7 +165,10 @@ with gr.Blocks() as demo:
173
  def submit_message(message, chat_history, model_name, system_message, max_tokens, temperature, top_p):
174
  history = [] if chat_history is None else chat_history
175
  for response in respond(message, history, model_name, system_message, max_tokens, temperature, top_p):
176
- history = history + [[message, response]]
 
 
 
177
  yield history, ""
178
 
179
  submit_event = submit.click(
 
84
  # Load new model if changed
85
  if current_model is None or model_name not in str(current_model.model_path):
86
  current_model = load_model(model_name)
 
 
87
 
88
  # Start with system message
89
  messages = []
90
  if system_message and system_message.strip():
91
  messages.append({"role": "system", "content": system_message})
92
 
93
+ # Add chat history
94
+ if history:
95
+ messages.extend(history)
96
+
 
 
 
 
 
 
97
  # Add current message
98
  messages.append({"role": "user", "content": message})
99
 
 
165
  def submit_message(message, chat_history, model_name, system_message, max_tokens, temperature, top_p):
166
  history = [] if chat_history is None else chat_history
167
  for response in respond(message, history, model_name, system_message, max_tokens, temperature, top_p):
168
+ history = history + [
169
+ {"role": "user", "content": message},
170
+ {"role": "assistant", "content": response}
171
+ ]
172
  yield history, ""
173
 
174
  submit_event = submit.click(