TejAndrewsACC commited on
Commit
8aade89
·
verified ·
1 Parent(s): 29d6b67

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -5
app.py CHANGED
@@ -2429,13 +2429,21 @@ supermassive_nn = ConsciousSupermassiveNN20()
2429
  def respond(message, history, max_tokens, temperature, top_p):
2430
  messages = [{"role": "system", "content": system_prompt}]
2431
  for val in history:
2432
- if val[0]:
2433
- messages.append({"role": "user", "content": val[0]})
2434
- if val[1]:
2435
- messages.append({"role": "assistant", "content": val[1]})
 
 
 
 
 
 
2436
  messages.append({"role": "user", "content": message})
2437
  response = ""
2438
- for message in client.chat_completion(messages, max_tokens=max_tokens, stream=True, temperature=temperature, top_p=top_p):
 
 
2439
  token = message.choices[0].delta.content
2440
  response += token
2441
  yield response
 
2429
  def respond(message, history, max_tokens, temperature, top_p):
2430
  messages = [{"role": "system", "content": system_prompt}]
2431
  for val in history:
2432
+ if isinstance(val, (list, tuple)):
2433
+ if len(val) > 0 and val[0]:
2434
+ messages.append({"role": "user", "content": val[0]})
2435
+ if len(val) > 1 and val[1]:
2436
+ messages.append({"role": "assistant", "content": val[1]})
2437
+ elif isinstance(val, dict):
2438
+ if "user" in val and val["user"]:
2439
+ messages.append({"role": "user", "content": val["user"]})
2440
+ if "assistant" in val and val["assistant"]:
2441
+ messages.append({"role": "assistant", "content": val["assistant"]})
2442
  messages.append({"role": "user", "content": message})
2443
  response = ""
2444
+ for message in client.chat_completion(
2445
+ messages, max_tokens=max_tokens, stream=True, temperature=temperature, top_p=top_p
2446
+ ):
2447
  token = message.choices[0].delta.content
2448
  response += token
2449
  yield response