harrygens commited on
Commit
bddc184
·
1 Parent(s): 8df5274

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -9,6 +9,7 @@ logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(
9
  openai.api_key = os.environ['key']
10
  role = os.environ['role']
11
  aclass = os.environ['class']
 
12
 
13
  mod = 3;
14
 
@@ -28,8 +29,9 @@ def trimMessages(messages):
28
  return messages
29
 
30
  def chatbot(input, messages):
31
- messages = messages or [{"role": "system", "content": initMsg}]
32
- if len(messages) % mod == 0 or len(messages) == 1:
 
33
  memory = trimMessages(messages)
34
  logging.info("put memory:"+printMessages(memory))
35
  chat = openai.ChatCompletion.create(
 
9
  openai.api_key = os.environ['key']
10
  role = os.environ['role']
11
  aclass = os.environ['class']
12
+ initMsg = os.environ['init']
13
 
14
  mod = 3;
15
 
 
29
  return messages
30
 
31
  def chatbot(input, messages):
32
+ messages = messages or [{"role": "system", "content": "i choose "+initMsg}]
33
+ if len(messages) == 1:
34
+
35
  memory = trimMessages(messages)
36
  logging.info("put memory:"+printMessages(memory))
37
  chat = openai.ChatCompletion.create(