jinzhou commited on
Commit
83a4df7
·
1 Parent(s): 15b091f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -60,11 +60,15 @@ class Conversation:
60
  ]
61
  }
62
  try:
 
63
  response = requests.post(OPENAI_URL, headers=headers, json=payload)
64
  response.raise_for_status() # 抛出异常,如果响应码不是200
65
  data = response.json()
66
  print("http_response:{}".format(data["choices"][0]["message"]))
67
  message = data["choices"][0]["message"]["content"]
 
 
 
68
  except requests.exceptions.RequestException as e:
69
  print(f"请求错误: {e}")
70
  except json.JSONDecodeError as e:
 
60
  ]
61
  }
62
  try:
63
+ self.messages.append( {"role": "user", "content": question})
64
  response = requests.post(OPENAI_URL, headers=headers, json=payload)
65
  response.raise_for_status() # 抛出异常,如果响应码不是200
66
  data = response.json()
67
  print("http_response:{}".format(data["choices"][0]["message"]))
68
  message = data["choices"][0]["message"]["content"]
69
+ self.messages.append({"role": "assistant", "content": message})
70
+ if len(self.messages) > self.num_of_round*2 + 1:
71
+ del self.messages[1:3]
72
  except requests.exceptions.RequestException as e:
73
  print(f"请求错误: {e}")
74
  except json.JSONDecodeError as e: