chore: remove imports in init model domain

Author: leoguillaume

api/domain/model/__init__.py

@@ -1,4 +0,0 @@
-from .entities import Model, ModelCosts, ModelType
-from .errors import InconsistentModelMaxContextLengthError, InconsistentModelVectorSizeError
-
-__all__ = ["ModelType", "Model", "ModelCosts", "InconsistentModelMaxContextLengthError", "InconsistentModelVectorSizeError"]

api/domain/provider/_providergateway.py

@@ -1,7 +1,7 @@
 from abc import ABC, abstractmethod
 from dataclasses import dataclass
 
-from api.domain.model import ModelType as RouterType
+from api.domain.model.entities import ModelType as RouterType
 from api.domain.provider.entities import ProviderType
 from api.domain.provider.errors import ProviderNotReachableError
 

api/domain/router/_routerrepository.py

@@ -1,6 +1,6 @@
 from abc import ABC, abstractmethod
 
-from api.domain.model import ModelType as RouterType
+from api.domain.model.entities import ModelType as RouterType
 from api.domain.router.entities import Router, RouterLoadBalancingStrategy, RouterPage, RouterSortField, SortOrder
 from api.domain.router.errors import RouterAliasAlreadyExistsError, RouterNameAlreadyExistsError
 

api/domain/router/entities.py

@@ -3,7 +3,7 @@
 
 from pydantic import BaseModel, Field
 
-from api.domain.model import ModelType as RouterType
+from api.domain.model.entities import ModelType as RouterType
 
 
 class RouterLoadBalancingStrategy(str, Enum):

api/infrastructure/fastapi/endpoints/admin/providers.py

@@ -7,7 +7,7 @@
 from sqlalchemy.ext.asyncio import AsyncSession
 
 from api.dependencies import create_provider_use_case_factory, delete_provider_use_case_factory, get_request_context
-from api.domain.model import InconsistentModelMaxContextLengthError, InconsistentModelVectorSizeError
+from api.domain.model.errors import InconsistentModelMaxContextLengthError, InconsistentModelVectorSizeError
 from api.domain.provider import InvalidProviderTypeError, ProviderNotReachableError
 from api.domain.provider.errors import ProviderAlreadyExistsError, ProviderNotFoundError
 from api.domain.router.errors import RouterNotFoundError
@@ -47,17 +47,15 @@
     path=EndpointRoute.ADMIN_PROVIDERS,
     dependencies=[Security(dependency=get_current_key)],
     status_code=201,
-    responses=get_documentation_responses(
-        [
-            InconsistentModelMaxContextLengthHTTPException,
-            InconsistentModelVectorSizeHTTPException,
-            InvalidProviderTypeHTTPException,
-            ProviderNotReachableHTTPException,
-            ProviderAlreadyExistsHTTPException,
-            RouterNotFoundHTTPException,
-            NotAdminUserHTTPException,
-        ]
-    ),
+    responses=get_documentation_responses([
+        InconsistentModelMaxContextLengthHTTPException,
+        InconsistentModelVectorSizeHTTPException,
+        InvalidProviderTypeHTTPException,
+        ProviderNotReachableHTTPException,
+        ProviderAlreadyExistsHTTPException,
+        RouterNotFoundHTTPException,
+        NotAdminUserHTTPException,
+    ]),
 )
 async def create_provider(
     request: Request,

api/infrastructure/fastapi/schemas/models.py

@@ -3,7 +3,7 @@
 
 from pydantic import Field
 
-from api.domain.model import Model as ModelEntity
+from api.domain.model.entities import Model as ModelEntity
 from api.schemas import BaseModel
 
 

api/infrastructure/model/_modelprovidergateway.py

@@ -1,5 +1,5 @@
 from api.clients.model import BaseModelProvider
-from api.domain.model import ModelType as RouterType
+from api.domain.model.entities import ModelType as RouterType
 from api.domain.provider import ProviderCapabilities, ProviderGateway, ProviderNotReachableError
 
 

api/infrastructure/postgres/_postgresrouterrepository.py

@@ -3,7 +3,7 @@
 from sqlalchemy.ext.asyncio import AsyncSession
 
 from api.domain.key.entities import MASTER_USER_ID
-from api.domain.model import ModelType as RouterType
+from api.domain.model.entities import ModelType as RouterType
 from api.domain.router import RouterRepository
 from api.domain.router.entities import Router, RouterLoadBalancingStrategy, RouterPage, RouterSortField, SortOrder
 from api.domain.router.errors import RouterAliasAlreadyExistsError, RouterNameAlreadyExistsError

api/tests/integration/postgres/test_postgresrouterrepository.py

@@ -4,7 +4,7 @@
 from sqlalchemy import select
 
 from api.domain.key.entities import MASTER_USER_ID
-from api.domain.model import ModelType as RouterType
+from api.domain.model.entities import ModelType as RouterType
 from api.domain.router.entities import Router, RouterLoadBalancingStrategy, RouterSortField, SortOrder
 from api.domain.router.errors import RouterAliasAlreadyExistsError, RouterNameAlreadyExistsError
 from api.infrastructure.postgres import PostgresRouterRepository

api/tests/unit/use_case/admin/providers/test_createproviderusecase.py

@@ -2,7 +2,7 @@
 
 import pytest
 
-from api.domain.model import ModelType as RouterType
+from api.domain.model.entities import ModelType as RouterType
 from api.domain.model.errors import InconsistentModelMaxContextLengthError, InconsistentModelVectorSizeError
 from api.domain.provider import ProviderCapabilities
 from api.domain.provider.entities import ProviderCarbonFootprintZone, ProviderType