Fix response models in customer endpoints

app/auth.py

@@ -11,7 +11,7 @@ from .settings import (
     ADMIN_USERNAME, ADMIN_PASSWORD
 )
 from .database import get_db
-from .models import Customer
+from .models import Customer as CustomerModel
 from .schemas import Customer as CustomerSchema
 
 # Password hashing
@@ -23,12 +23,12 @@ api_key_header = APIKeyHeader(name="api-key")
 
 router = APIRouter()
 
-def verify_api_key(api_key: str = Security(api_key_header)) -> Customer:
+def verify_api_key(api_key: str = Security(api_key_header)) -> CustomerModel:
     """Verify API key and return customer"""
     db = next(get_db())
-    customer = db.query(Customer).filter(
-        Customer.api_key == api_key,
-        Customer.is_active == True
+    customer = db.query(CustomerModel).filter(
+        CustomerModel.api_key == api_key,
+        CustomerModel.is_active == True
     ).first()
     
     if not customer:
@@ -109,7 +109,7 @@ async def create_customer(
             )
 
         # Check if email already exists
-        existing_customer = db.query(Customer).filter(Customer.email == customer_data["email"]).first()
+        existing_customer = db.query(CustomerModel).filter(CustomerModel.email == customer_data["email"]).first()
         if existing_customer:
             raise HTTPException(
                 status_code=status.HTTP_400_BAD_REQUEST,
@@ -117,7 +117,7 @@ async def create_customer(
             )
         
         # Create new customer
-        customer = Customer(
+        customer = CustomerModel(
             name=customer_data["name"],
             company_name=customer_data["company_name"],
             email=customer_data["email"],
@@ -145,16 +145,16 @@ async def create_customer(
             detail=f"Error creating customer: {str(e)}"
         )
 
-@router.get("/customers/", response_model=List[Customer])
+@router.get("/customers/", response_model=List[CustomerSchema])
 async def list_customers(
     db: Session = Depends(get_db),
     current_admin: str = Depends(get_current_admin)
 ):
     """List all customers"""
     try:
-        customers = db.query(Customer).all()
+        customers = db.query(CustomerModel).all()
         return [
-            Customer(
+            CustomerSchema(
                 id=customer.id,
                 name=customer.name,
                 company_name=customer.company_name,
@@ -179,7 +179,7 @@ async def get_customer(
     current_admin: str = Depends(get_current_admin)
 ):
     """Get customer details"""
-    customer = db.query(Customer).filter(Customer.id == customer_id).first()
+    customer = db.query(CustomerModel).filter(CustomerModel.id == customer_id).first()
     if not customer:
         raise HTTPException(status_code=404, detail="Customer not found")
     return customer
@@ -191,7 +191,7 @@ async def delete_customer(
     current_admin: str = Depends(get_current_admin)
 ):
     """Delete a customer"""
-    customer = db.query(Customer).filter(Customer.id == customer_id).first()
+    customer = db.query(CustomerModel).filter(CustomerModel.id == customer_id).first()
     if not customer:
         raise HTTPException(status_code=404, detail="Customer not found")
     

main.py

@@ -360,16 +360,18 @@ async def list_customers(
     """List all customers"""
     try:
         customers = db.query(CustomerModel).all()
-        return [Customer(
-            id=c.id,
-            name=c.name,
-            company_name=c.company_name,
-            email=c.email,
-            api_key=c.api_key,
-            is_active=c.is_active,
-            created_at=c.created_at,
-            updated_at=c.updated_at
-        ) for c in customers]
+        return [
+            Customer(
+                id=customer.id,
+                name=customer.name,
+                company_name=customer.company_name,
+                email=customer.email,
+                api_key=customer.api_key,
+                is_active=customer.is_active,
+                created_at=customer.created_at,
+                updated_at=customer.updated_at
+            ) for customer in customers
+        ]
     except Exception as e:
         raise HTTPException(
             status_code=500,