fix: improve maintenance and synchronize live conversations

backend/app/models/__init__.py

@@ -73,6 +73,7 @@ from app.models.guardrail_event import GuardrailEvent
 
 # This ensures all models are imported in the correct order
 __all__ = [
+    "Base",
     "Organization",
     "User",
     "FCMToken",

backend/app/models/agent.py

@@ -21,7 +21,6 @@ from app.database import Base
 import enum
 import json
 import uuid
-from typing import Optional
 
 
 class AgentType(str, enum.Enum):

backend/app/models/chat_history.py

@@ -14,7 +14,7 @@ See the License for the specific language governing permissions and
 limitations under the License.
 """
 
-from sqlalchemy import Column, Integer, String, JSON, ForeignKey, TIMESTAMP, func, UUID, Interval, Float, Index, text
+from sqlalchemy import Column, Integer, String, JSON, ForeignKey, TIMESTAMP, func, UUID, Float, Index, text
 from sqlalchemy.orm import relationship
 from app.core.encryption import EncryptedText
 from app.database import Base

backend/app/models/customer.py

@@ -14,7 +14,7 @@ See the License for the specific language governing permissions and
 limitations under the License.
 """
 
-from sqlalchemy import Column, Integer, String, Boolean, DateTime, ForeignKey, Index, JSON, Enum as SQLEnum, func, text, UniqueConstraint
+from sqlalchemy import Column, String, Boolean, DateTime, ForeignKey, Index, JSON, Enum as SQLEnum, func, text, UniqueConstraint
 from sqlalchemy.orm import relationship
 from sqlalchemy.dialects.postgresql import UUID
 import enum

backend/app/models/investigation.py

@@ -14,7 +14,6 @@ See the License for the specific language governing permissions and
 limitations under the License.
 """
 
-import enum
 import uuid
 
 from sqlalchemy import (

backend/app/models/jira.py

@@ -18,7 +18,6 @@ from sqlalchemy import Column, Integer, String, DateTime, ForeignKey, Boolean
 from sqlalchemy.orm import relationship
 from datetime import datetime
 from sqlalchemy.dialects.postgresql import UUID
-import uuid
 
 from app.database import Base
 

backend/app/models/knowledge_queue.py

@@ -14,7 +14,7 @@ See the License for the specific language governing permissions and
 limitations under the License.
 """
 
-from sqlalchemy import Column, Integer, String, JSON, DateTime, Enum, ForeignKey, Float, Index
+from sqlalchemy import Column, Integer, String, JSON, DateTime, ForeignKey, Float, Index
 from sqlalchemy.sql import func
 from sqlalchemy.orm import relationship
 from app.database import Base

backend/app/models/organization.py

@@ -15,10 +15,9 @@ limitations under the License.
 """
 
 import uuid
-from sqlalchemy import Column, Integer, String, Boolean, DateTime, JSON, func, event
+from sqlalchemy import Column, String, Boolean, DateTime, JSON, func, event
 from sqlalchemy.orm import relationship
 from app.database import Base
-from app.models.role import Role
 from sqlalchemy.dialects.postgresql import UUID
 from app.core.logger import get_logger
 

backend/app/models/rating.py

@@ -15,7 +15,7 @@ limitations under the License.
 """
 
 import uuid
-from sqlalchemy import Column, Integer, String, Text, ForeignKey, DateTime
+from sqlalchemy import Column, Integer, Text, ForeignKey, DateTime
 from sqlalchemy.orm import relationship
 from sqlalchemy.dialects.postgresql import UUID
 from app.database import Base

backend/app/models/schemas/agent.py

@@ -17,7 +17,6 @@ limitations under the License.
 from datetime import datetime
 from pydantic import BaseModel, Field
 from typing import List, Optional, Dict
-from enum import Enum
 from uuid import UUID
 from app.models.agent import AgentType
 from app.agents.guardrail_policy import GUARDRAIL_PROMPT_MAX

backend/app/models/schemas/jira.py

@@ -14,7 +14,6 @@ See the License for the specific language governing permissions and
 limitations under the License.
 """
 
-from pydantic import BaseModel
 from typing import Optional, List, Any
 from app.models.schemas.agent import AgentResponse
 

backend/app/models/schemas/lead_capture.py

@@ -15,7 +15,7 @@ limitations under the License.
 """
 
 from pydantic import BaseModel, ConfigDict, Field, field_validator
-from typing import Optional, List, Dict, Any
+from typing import Optional, List
 from uuid import UUID
 
 # Reuse the ORM enums so the API contract can never drift from the DB.

backend/app/models/schemas/mcp_tool.py

@@ -16,10 +16,9 @@ limitations under the License.
 
 from datetime import datetime
 from pydantic import BaseModel, Field, field_serializer, validator
-from typing import List, Optional, Dict, Union
+from typing import List, Optional, Dict
 from enum import Enum
 from uuid import UUID
-from app.models.mcp_tool import MCPTransportType
 
 
 class MCPTransportTypeEnum(str, Enum):

backend/app/models/schemas/shopify/__init__.py

@@ -26,4 +26,16 @@ from .agent_shopify_config import (
     AgentShopifyConfigCreate,
     AgentShopifyConfigUpdate,
     AgentShopifyConfig
-) 
+)
+
+__all__ = [
+    "ShopifyShopBase",
+    "ShopifyShopCreate",
+    "ShopifyShopUpdate",
+    "ShopifyShopInDB",
+    "ShopifyShop",
+    "AgentShopifyConfigBase",
+    "AgentShopifyConfigCreate",
+    "AgentShopifyConfigUpdate",
+    "AgentShopifyConfig",
+]

backend/app/models/schemas/workflow.py

@@ -14,8 +14,8 @@ See the License for the specific language governing permissions and
 limitations under the License.
 """
 
-from pydantic import BaseModel, Field, validator
-from typing import Optional, List, Dict, Any, Union, Literal
+from pydantic import BaseModel, Field
+from typing import Optional, List, Dict, Any, Literal
 from uuid import UUID
 from datetime import datetime
 from app.models.workflow import WorkflowStatus

backend/app/models/shopify/__init__.py

@@ -15,5 +15,7 @@ limitations under the License.
 """
 
 # Import models
-from .shopify_shop import ShopifyShop
-from .agent_shopify_config import AgentShopifyConfig 
+from .shopify_shop import ShopifyShop as ShopifyShop
+from .agent_shopify_config import AgentShopifyConfig as AgentShopifyConfig
+
+__all__ = ["ShopifyShop", "AgentShopifyConfig"]

backend/app/models/shopify/shopify_shop.py

@@ -14,7 +14,7 @@ See the License for the specific language governing permissions and
 limitations under the License.
 """
 
-from sqlalchemy import Column, String, Boolean, DateTime, Integer, ForeignKey
+from sqlalchemy import Column, String, Boolean, DateTime, ForeignKey
 from sqlalchemy.orm import relationship
 from sqlalchemy.sql import func
 from sqlalchemy.dialects.postgresql import UUID

backend/app/models/ticket_activity.py

@@ -14,7 +14,6 @@ See the License for the specific language governing permissions and
 limitations under the License.
 """
 
-import enum
 import uuid
 
 from sqlalchemy import JSON, Boolean, Column, DateTime, ForeignKey, Index, String, Text

backend/app/models/user.py

@@ -14,7 +14,7 @@ See the License for the specific language governing permissions and
 limitations under the License.
 """
 
-from sqlalchemy import Column, Integer, String, Boolean, DateTime, ForeignKey, func, Table, Text
+from sqlalchemy import Column, Integer, String, Boolean, DateTime, ForeignKey, func, Table
 from sqlalchemy.orm import relationship
 from sqlalchemy.dialects.postgresql import UUID
 import uuid

backend/app/models/widget.py

@@ -14,7 +14,7 @@ See the License for the specific language governing permissions and
 limitations under the License.
 """
 
-from sqlalchemy import Column, String, ForeignKey, Integer
+from sqlalchemy import Column, String, ForeignKey
 from sqlalchemy.orm import relationship
 from uuid import  uuid4
 from sqlalchemy.dialects.postgresql import UUID