AUST001 commited on
Commit
912ff2b
·
1 Parent(s): 5b0378c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -13,7 +13,10 @@ user_contexts = {}
13
  def get_assistant_response(user_question, context, model_name):
14
  global client, models
15
  context.append({"role": "user", "content": user_question})
16
- for chunk in client.send_message(models[model_name], context): # capybara
 
 
 
17
  pass
18
  # print(chunk["text"])
19
  assistant_response = chunk["text"]
 
13
  def get_assistant_response(user_question, context, model_name):
14
  global client, models
15
  context.append({"role": "user", "content": user_question})
16
+ prompt = ''
17
+ for item in context[3:]:
18
+ prompt += item["role"] + ": " + item["content"] + "\n"
19
+ for chunk in client.send_message(models[model_name], prompt): # capybara
20
  pass
21
  # print(chunk["text"])
22
  assistant_response = chunk["text"]