Companies

app/controllers/__init__.py

@@ -1,7 +1,10 @@
 from litestar import Router
 
+from app.controllers.company import CompanyController
 from app.controllers.fiscal_payment_mapping import FiscalPaymentMappingController
 from app.controllers.machine import MachineController
+from app.domain.company import Company
+from app.domain.fiscal_payment_mapping import FiscalPaymentMapping
 from app.domain.machine import Machine
 
 __all__ = ["create_router"]
@@ -11,11 +14,13 @@ def create_router() -> Router:
     return Router(
         path="/v1",
         route_handlers=[
+            CompanyController,
             MachineController,
             FiscalPaymentMappingController,
         ],
         signature_namespace={
+            "Company": Company,
             "Machine": Machine,
-            "FiscalPaymentMappingController": FiscalPaymentMappingController,
+            "FiscalPaymentMapping": FiscalPaymentMapping,
         },
     )