mirror of
https://github.com/Significant-Gravitas/AutoGPT.git
synced 2026-02-09 14:25:25 -05:00
08cd935a47
<!-- Clearly explain the need for these changes: --> We don't want to spam the user with similar notification types so we want to group them up over a timespan and handle that as a group of notifications. ### Changes 🏗️ - Adds a batch queue - Moves the ExecutionScheduleur to a generic Scheduler - Makes the Agent run a batch operation - Fixes various bugs in how we originally made the batch db models and queries <!-- Concisely describe all of the changes made in this pull request: --> ### Checklist 📋 #### For code changes: - [x] I have clearly listed my changes in the PR description - [x] I have made a test plan - [x] I have tested my changes according to the test plan: <!-- Put your test plan here: --> - [x] Run 10 agents back to back - [x] Notice how many emails you get - [x] Wait a bit and you should after an hour (change the cron rule to speed up testing this) you'll get an email and see all the batches in your db are empty --------- Co-authored-by: Zamil Majdy <zamil.majdy@agpt.co>
40 lines
1.3 KiB
Python
40 lines
1.3 KiB
Python
import pytest
|
|
|
|
from backend.data import db
|
|
from backend.executor import Scheduler
|
|
from backend.server.model import CreateGraph
|
|
from backend.usecases.sample import create_test_graph, create_test_user
|
|
from backend.util.service import get_service_client
|
|
from backend.util.test import SpinTestServer
|
|
|
|
|
|
@pytest.mark.asyncio(scope="session")
|
|
async def test_agent_schedule(server: SpinTestServer):
|
|
await db.connect()
|
|
test_user = await create_test_user()
|
|
test_graph = await server.agent_server.test_create_graph(
|
|
create_graph=CreateGraph(graph=create_test_graph()),
|
|
user_id=test_user.id,
|
|
)
|
|
|
|
scheduler = get_service_client(Scheduler)
|
|
schedules = scheduler.get_execution_schedules(test_graph.id, test_user.id)
|
|
assert len(schedules) == 0
|
|
|
|
schedule = scheduler.add_execution_schedule(
|
|
graph_id=test_graph.id,
|
|
user_id=test_user.id,
|
|
graph_version=1,
|
|
cron="0 0 * * *",
|
|
input_data={"input": "data"},
|
|
)
|
|
assert schedule
|
|
|
|
schedules = scheduler.get_execution_schedules(test_graph.id, test_user.id)
|
|
assert len(schedules) == 1
|
|
assert schedules[0].cron == "0 0 * * *"
|
|
|
|
scheduler.delete_schedule(schedule.id, user_id=test_user.id)
|
|
schedules = scheduler.get_execution_schedules(test_graph.id, user_id=test_user.id)
|
|
assert len(schedules) == 0
|