fix(backend): formatting

autogpt_platform/backend/backend/data/user.py

@@ -1,4 +1,3 @@
-from datetime import datetime
 import logging
 from datetime import datetime, timedelta
 from typing import Optional, cast

autogpt_platform/backend/backend/executor/database.py

@@ -16,7 +16,6 @@ from backend.data.execution import (
     update_node_execution_stats,
     upsert_execution_input,
     upsert_execution_output,
-    get_executions_in_timerange,
 )
 from backend.data.graph import get_graph, get_node
 from backend.data.notifications import (

autogpt_platform/backend/backend/executor/manager.py

@@ -1,5 +1,4 @@
 import atexit
-from datetime import datetime
 import logging
 import multiprocessing
 import os
@@ -11,14 +10,13 @@ from contextlib import contextmanager
 from multiprocessing.pool import AsyncResult, Pool
 from typing import TYPE_CHECKING, Any, Generator, Optional, TypeVar, cast
 
+from redis.lock import Lock as RedisLock
+
 from backend.notifications.models import (
     AgentRunData,
-    NotificationEvent,
     NotificationType,
-    ZeroBalanceData,
     create_notification,
 )
-from redis.lock import Lock as RedisLock
 
 if TYPE_CHECKING:
     from backend.executor import DatabaseManager
@@ -226,17 +224,29 @@ def execute_node(
             ):
                 yield execution
 
-            # Update execution status and spend credits
-            res = update_execution(ExecutionStatus.COMPLETED)
-            s = input_size + output_size
-            t = (
-                (res.end_time - res.start_time).total_seconds()
-                if res.end_time and res.start_time
-                else 0
+        # Update execution status and spend credits
+        res = update_execution(ExecutionStatus.COMPLETED)
+        s = input_size + output_size
+        t = (
+            (res.end_time - res.start_time).total_seconds()
+            if res.end_time and res.start_time
+            else 0
+        )
+        data.data = input_data
+        db_client.spend_credits(data, s, t)
+        notification_service.queue_notification(
+            create_notification(
+                user_id=user_id,
+                type=NotificationType.AGENT_RUN,
+                data=AgentRunData(
+                    agent_name=node_block.name,
+                    credits_used=s,
+                    execution_time=t,
+                    graph_id=graph_id,
+                    node_count=1,
+                ),
             )
-            data.data = input_data
-            db_client.spend_credits(data, s, t)
-            
+        )
 
     except Exception as e:
         error_msg = str(e)

autogpt_platform/backend/backend/notifications/models.py

@@ -1,7 +1,8 @@
 from datetime import datetime
 from enum import Enum
-from typing import Annotated, Generic, Literal, Optional, Self, TypeVar, Union, overload
-from pydantic import BaseModel, EmailStr, Field, model_validator
+from typing import Annotated, Generic, Literal, Optional, TypeVar, Union, overload
+
+from pydantic import BaseModel, EmailStr, Field
 
 
 class BatchingStrategy(str, Enum):

autogpt_platform/backend/backend/notifications/notifications.py

@@ -1,27 +1,24 @@
-from collections import defaultdict
 import logging
-from datetime import datetime, timedelta
 import time
+from datetime import datetime, timedelta
 from typing import TYPE_CHECKING, cast
+
 from autogpt_libs.utils.cache import thread_cached
+
 from backend.notifications.models import (
     BatchingStrategy,
-    DailySummaryData,
-    MonthlySummaryData,
     NotificationBatch,
     NotificationEvent,
     NotificationResult,
     NotificationType,
-    WeeklySummaryData,
-    create_notification,
 )
 
 if TYPE_CHECKING:
     from backend.executor import DatabaseManager
 
+from backend.data.redis import get_redis, get_redis_async
 from backend.notifications.summary import SummaryManager
 from backend.util.service import AppService, expose, get_service_client
-from backend.data.redis import get_redis, get_redis_async
 from backend.util.settings import Settings
 
 logger = logging.getLogger(__name__)

autogpt_platform/backend/backend/notifications/summary.py

@@ -2,7 +2,10 @@
 import logging
 from collections import defaultdict
 from datetime import datetime
+
 from autogpt_libs.utils.cache import thread_cached
+
+from backend.executor.database import DatabaseManager
 from backend.notifications.models import (
     DailySummaryData,
     MonthlySummaryData,
@@ -10,7 +13,6 @@ from backend.notifications.models import (
     WeeklySummaryData,
     create_notification,
 )
-from backend.executor.database import DatabaseManager
 from backend.util.service import get_service_client
 
 logger = logging.getLogger(__name__)