AIRider commited on
Commit
498599c
·
verified ·
1 Parent(s): ec8a9b8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -13
app.py CHANGED
@@ -26,18 +26,6 @@ def respond(
26
  top_p,
27
  system_message,
28
  ):
29
- try:
30
- client = get_client(model_name)
31
- except ValueError as e:
32
- chat_history.append((message, str(e)))
33
- return chat_history
34
-
35
- messages = [{"role": "system", "content": system_message}]
36
- for human, assistant in chat_history:
37
- messages.append({"role": "user", "content": human})
38
- messages.append({"role": "assistant", "content": assistant})
39
- messages.append({"role": "user", "content": message})
40
-
41
  try:
42
  stream = client.chat_completion(
43
  messages,
@@ -50,7 +38,10 @@ def respond(
50
  for response in stream:
51
  if response.choices[0].delta.content is not None:
52
  partial_message += response.choices[0].delta.content
53
- chat_history = chat_history + [(message, partial_message)]
 
 
 
54
  yield chat_history
55
  except Exception as e:
56
  error_message = f"An error occurred: {str(e)}"
 
26
  top_p,
27
  system_message,
28
  ):
 
 
 
 
 
 
 
 
 
 
 
 
29
  try:
30
  stream = client.chat_completion(
31
  messages,
 
38
  for response in stream:
39
  if response.choices[0].delta.content is not None:
40
  partial_message += response.choices[0].delta.content
41
+ if len(chat_history) > 0 and chat_history[-1][0] == message:
42
+ chat_history[-1] = (message, partial_message)
43
+ else:
44
+ chat_history.append((message, partial_message))
45
  yield chat_history
46
  except Exception as e:
47
  error_message = f"An error occurred: {str(e)}"