Pamudu13 commited on
Commit
ba35145
·
verified ·
1 Parent(s): 5e58171

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -5
app.py CHANGED
@@ -84,12 +84,14 @@ def generate_response(
84
  first_msg = f"{system_message}\n\nContext: {context}\nQuestion: {message}"
85
  messages.append({"role": "user", "content": first_msg})
86
 
87
- # Add conversation history ensuring alternating pattern
88
  for i, (user_msg, bot_msg) in enumerate(history):
89
- if bot_msg: # Add assistant's previous response
90
- messages.append({"role": "assistant", "content": bot_msg})
91
- if user_msg: # Add user's previous message
92
- messages.append({"role": "user", "content": user_msg})
 
 
93
 
94
  try:
95
  response = ""
 
84
  first_msg = f"{system_message}\n\nContext: {context}\nQuestion: {message}"
85
  messages.append({"role": "user", "content": first_msg})
86
 
87
+ # Add conversation history ensuring alternating pattern (user, assistant, user, assistant...)
88
  for i, (user_msg, bot_msg) in enumerate(history):
89
+ if i % 2 == 0: # Even index: it's a user message
90
+ if user_msg:
91
+ messages.append({"role": "user", "content": user_msg})
92
+ else: # Odd index: it's an assistant message
93
+ if bot_msg:
94
+ messages.append({"role": "assistant", "content": bot_msg})
95
 
96
  try:
97
  response = ""