svarshney25 commited on
Commit
5fe999b
·
verified ·
1 Parent(s): 8990b42

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -46,10 +46,10 @@ def respond(message, history):
46
  context = "\n\n".join(top_chunks)
47
  messages = [{"role": "system", "content": f"You are a travel advisor. Cover flights, hotels, food, and itineraries. If the user is just greeting you, greet them back and ask where they want to travel. \n{context}"}]
48
  for turn in history:
49
- if turn[0] is not None:
50
- messages.append({"role": "user", "content": turn[0]})
51
- if turn[1] is not None:
52
- messages.append({"role": "assistant", "content": turn[1]})
53
  messages.append({"role": "user", "content": message})
54
 
55
  response = ""
 
46
  context = "\n\n".join(top_chunks)
47
  messages = [{"role": "system", "content": f"You are a travel advisor. Cover flights, hotels, food, and itineraries. If the user is just greeting you, greet them back and ask where they want to travel. \n{context}"}]
48
  for turn in history:
49
+ if turn["role"] == "user":
50
+ messages.append({"role": "user", "content": turn["content"]})
51
+ elif turn["role"] == "assistant":
52
+ messages.append({"role": "assistant", "content": turn["content"]})
53
  messages.append({"role": "user", "content": message})
54
 
55
  response = ""