diff --git a/autogpt_platform/backend/backend/blocks/llm.py b/autogpt_platform/backend/backend/blocks/llm.py index c3e5e9348a..87963e42eb 100644 --- a/autogpt_platform/backend/backend/blocks/llm.py +++ b/autogpt_platform/backend/backend/blocks/llm.py @@ -16,7 +16,6 @@ from groq import AsyncGroq from pydantic import BaseModel, GetCoreSchemaHandler, SecretStr from pydantic_core import CoreSchema, core_schema -from backend.data import llm_registry from backend.blocks._base import ( Block, BlockCategory, @@ -24,6 +23,7 @@ from backend.blocks._base import ( BlockSchemaInput, BlockSchemaOutput, ) +from backend.data import llm_registry from backend.data.llm_registry import ModelMetadata from backend.data.model import ( APIKeyCredentials, diff --git a/autogpt_platform/backend/backend/data/block.py b/autogpt_platform/backend/backend/data/block.py index 35829d9656..31811ce17d 100644 --- a/autogpt_platform/backend/backend/data/block.py +++ b/autogpt_platform/backend/backend/data/block.py @@ -4,9 +4,6 @@ from typing import TYPE_CHECKING, Any, AsyncGenerator from prisma.models import AgentBlock from prisma.types import AgentBlockCreateInput -from backend.data.llm_registry import update_schema_with_llm_registry -from backend.data.model import NodeExecutionStats -from backend.integrations.providers import ProviderName from backend.util import json if TYPE_CHECKING: