xiaxinhang/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-07-04 12:50:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add exception protection for unexpected conditions during request handling

Browse Source
This commit is contained in:
Ori Kotek
2023-07-06 19:08:47 +03:00
parent 9e96fbab1f
commit 2de83827b6
1 changed files with 46 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
pr_agent/servers/github_polling.py
View File

@ -37,6 +37,7 @@ async def polling_loop():
raise ValueError("User token must be set to get notifications") raise ValueError("User token must be set to get notifications")
async with aiohttp.ClientSession() as session: async with aiohttp.ClientSession() as session:
while True: while True:
try:
headers = { headers = {
"Accept": "application/vnd.github.v3+json", "Accept": "application/vnd.github.v3+json",
"Authorization": f"Bearer {token}" "Authorization": f"Bearer {token}"
@ -81,6 +82,9 @@ async def polling_loop():
print(f"Failed to fetch notifications. Status code: {response.status}") print(f"Failed to fetch notifications. Status code: {response.status}")
await asyncio.sleep(5) await asyncio.sleep(5)
except Exception as e:
logging.error(f"Exception during processing of a notification: {e}")
await asyncio.sleep(5)
if __name__ == '__main__': if __name__ == '__main__':
asyncio.run(polling_loop()) asyncio.run(polling_loop())