schuler commited on
Commit
b15fb69
·
verified ·
1 Parent(s): 96a08ea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -4
app.py CHANGED
@@ -33,8 +33,10 @@ def respond(
33
  ):
34
  result = 'none'
35
  try:
36
- messages = [{"role": "system", "content": system_message}]
37
-
 
 
38
  for val in history:
39
  if val[0]:
40
  messages.append({"role": "user", "content": val[0]})
@@ -44,8 +46,8 @@ def respond(
44
  messages.append({"role": "user", "content": message})
45
 
46
  for message in messages:
47
- role = "<|assistant|>" if message['role'] == 'assistant' else "<|user|>"
48
- prompt += f"\n{role}\n{message['content']}\n<|end|>\n"
49
  # prompt += f"\n<|user|>\n{user_text}\n<|end|><|assistant|>\n"
50
 
51
  # Generate the response
 
33
  ):
34
  result = 'none'
35
  try:
36
+ # Build the conversation prompt
37
+ prompt = ''
38
+ if (len(system_message)>0):
39
+ prompt = "<|assistant|>"+system_message+f"<|end|>\n"
40
  for val in history:
41
  if val[0]:
42
  messages.append({"role": "user", "content": val[0]})
 
46
  messages.append({"role": "user", "content": message})
47
 
48
  for message in messages:
49
+ role = "<|assistant|>" if message['role'] == 'assistant' else "<|user|>"
50
+ prompt += f"\n{role}\n{message['content']}\n<|end|>\n"
51
  # prompt += f"\n<|user|>\n{user_text}\n<|end|><|assistant|>\n"
52
 
53
  # Generate the response