Mbonea commited on
Commit
c477d00
·
1 Parent(s): 0e6432b

error handling and retrying 3 times

Browse files
Files changed (1) hide show
  1. App/Chat/utils/PoeBot.py +23 -11
App/Chat/utils/PoeBot.py CHANGED
@@ -10,15 +10,27 @@ print(client.get_chat_history()["data"])
10
 
11
 
12
  async def SendMessage(req: BotRequest):
13
- global CHAT_CODE
14
- if CHAT_CODE == "":
15
- for chunk in client.send_message(
16
- req.bot, req.message, chatCode="2rx4w5jt6zf96tn7dr1"
17
- ):
18
- pass
19
- CHAT_CODE = chunk["chatCode"]
20
- else:
21
- for chunk in client.send_message(req.bot, req.message, chatCode=CHAT_CODE):
22
- pass
 
 
 
 
 
23
 
24
- return chunk["text"]
 
 
 
 
 
 
 
 
10
 
11
 
12
  async def SendMessage(req: BotRequest):
13
+ global CHAT_CODE, client
14
+ counter = 0
15
+ while True:
16
+ try:
17
+ if CHAT_CODE == "":
18
+ for chunk in client.send_message(
19
+ req.bot, req.message, chatCode="2rx4w5jt6zf96tn7dr1"
20
+ ):
21
+ pass
22
+ CHAT_CODE = chunk["chatCode"]
23
+ else:
24
+ for chunk in client.send_message(
25
+ req.bot, req.message, chatCode=CHAT_CODE
26
+ ):
27
+ pass
28
 
29
+ return {"response": chunk["text"], "code": 200}
30
+ except:
31
+ if counter > 4:
32
+ return {"response": "Try again later", "code": 500}
33
+ client = PoeApi("sXvCnfYy8CHnXNTRlxhmVg==")
34
+ CHAT_CODE = ""
35
+ counter += 1
36
+ print(client.get_chat_history()["data"])