jordigonzm commited on
Commit
26e9ade
1 Parent(s): b8c5b39

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -6
app.py CHANGED
@@ -16,13 +16,16 @@ model = "gpt-3.5-turbo"
16
 
17
  def predict(message, history):
18
  messages = []
19
-
20
- for user_message, assistant_message in history:
21
- messages.append({"role": "user", "content": user_message})
22
- messages.append({"role": "assistant", "content": assistant_message})
23
-
 
 
 
24
  messages.append({"role": "user", "content": message})
25
-
26
  response = llama.create_chat_completion_openai_v1(
27
  model=model,
28
  messages=messages,
 
16
 
17
  def predict(message, history):
18
  messages = []
19
+ for entry in history:
20
+ if len(entry) != 2:
21
+ raise ValueError("Cada entrada en el historial debe contener exactamente dos mensajes.")
22
+ user_message, assistant_message = entry
23
+ messages.extend([
24
+ {"role": "user", "content": user_message},
25
+ {"role": "assistant", "content": assistant_message}
26
+ ])
27
  messages.append({"role": "user", "content": message})
28
+
29
  response = llama.create_chat_completion_openai_v1(
30
  model=model,
31
  messages=messages,