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

feat: enhance GitHub polling with synchronous comment processing and improved logging

Browse Source
This commit is contained in:
mrT23
2024-09-05 16:53:31 +03:00
parent b8d2b263b9
commit 19048ee705
1 changed files with 9 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
pr_agent/servers/github_polling.py
View File

@ -16,6 +16,15 @@ setup_logger(fmt=LoggingFormat.JSON, level="DEBUG")
NOTIFICATION_URL = "https://api.github.com/notifications" NOTIFICATION_URL = "https://api.github.com/notifications"
async def mark_notification_as_read(headers, notification, session):
async with session.patch(
f"https://api.github.com/notifications/threads/{notification['id']}",
headers=headers) as mark_read_response:
if mark_read_response.status != 205:
get_logger().error(
f"Failed to mark notification as read. Status code: {mark_read_response.status}")
def now() -> str: def now() -> str:
""" """
Get the current UTC time in ISO 8601 format. Get the current UTC time in ISO 8601 format.
@ -218,18 +227,5 @@ async def polling_loop():
artifact={"traceback": traceback.format_exc()}) artifact={"traceback": traceback.format_exc()})
async def mark_notification_as_read(headers, notification, session):
async with session.patch(
f"https://api.github.com/notifications/threads/{notification['id']}",
headers=headers) as mark_read_response:
if mark_read_response.status != 205:
get_logger().error(
f"Failed to mark notification as read. Status code: {mark_read_response.status}")
if __name__ == '__main__': if __name__ == '__main__':
asyncio.run(polling_loop()) asyncio.run(polling_loop())
# # Example usage
# task_queue = deque([lambda: print("Task executed1"), lambda: print("Task executed2")])
# asyncio.run(background_task_manager(task_queue))