artificialguybr commited on
Commit
314bafa
1 Parent(s): 9809955

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -32,7 +32,7 @@ def user(message, history):
32
  def call_nvidia_api(history, system_message, max_tokens, temperature, top_p):
33
  """Calls the NVIDIA API to generate a response."""
34
  messages = [{"role": "system", "content": system_message}]
35
- messages.extend([{"role": role, "content": content} for role, content in history])
36
 
37
  payload = {
38
  "messages": messages,
 
32
  def call_nvidia_api(history, system_message, max_tokens, temperature, top_p):
33
  """Calls the NVIDIA API to generate a response."""
34
  messages = [{"role": "system", "content": system_message}]
35
+ messages.extend([{"role": item["role"], "content": item["content"]} for item in history])
36
 
37
  payload = {
38
  "messages": messages,