diff --git a/autogpt_platform/backend/backend/api/features/chat/tools/agent_generator/service.py b/autogpt_platform/backend/backend/api/features/chat/tools/agent_generator/service.py index edad1e8e4e..caa00a9e9c 100644 --- a/autogpt_platform/backend/backend/api/features/chat/tools/agent_generator/service.py +++ b/autogpt_platform/backend/backend/api/features/chat/tools/agent_generator/service.py @@ -155,7 +155,11 @@ async def generate_agent_external( f"Agent Generator accepted async request " f"(operation_id={operation_id}, task_id={task_id})" ) - return {"status": "accepted", "operation_id": operation_id, "task_id": task_id} + return { + "status": "accepted", + "operation_id": operation_id, + "task_id": task_id, + } response.raise_for_status() data = response.json() @@ -214,7 +218,11 @@ async def generate_agent_patch_external( f"Agent Generator accepted async update request " f"(operation_id={operation_id}, task_id={task_id})" ) - return {"status": "accepted", "operation_id": operation_id, "task_id": task_id} + return { + "status": "accepted", + "operation_id": operation_id, + "task_id": task_id, + } response.raise_for_status() data = response.json() diff --git a/autogpt_platform/backend/backend/api/rest_api.py b/autogpt_platform/backend/backend/api/rest_api.py index 824211be5f..96d4f33286 100644 --- a/autogpt_platform/backend/backend/api/rest_api.py +++ b/autogpt_platform/backend/backend/api/rest_api.py @@ -22,10 +22,6 @@ import backend.api.features.admin.store_admin_routes import backend.api.features.builder import backend.api.features.builder.routes import backend.api.features.chat.routes as chat_routes -from backend.api.features.chat.completion_consumer import ( - start_completion_consumer, - stop_completion_consumer, -) import backend.api.features.executions.review.routes import backend.api.features.library.db import backend.api.features.library.model @@ -44,6 +40,10 @@ import backend.data.user import backend.integrations.webhooks.utils import backend.util.service import backend.util.settings +from backend.api.features.chat.completion_consumer import ( + start_completion_consumer, + stop_completion_consumer, +) from backend.blocks.llm import DEFAULT_LLM_MODEL from backend.data.model import Credentials from backend.integrations.providers import ProviderName