feat(skills): consolidate skill-related models and update imports
This commit is contained in:
@@ -14,8 +14,8 @@ from app.db import crud
|
|||||||
from app.db.pagination import paginate
|
from app.db.pagination import paginate
|
||||||
from app.db.session import get_session
|
from app.db.session import get_session
|
||||||
from app.models.agents import Agent
|
from app.models.agents import Agent
|
||||||
from app.models.gateway_installed_skills import GatewayInstalledSkill
|
|
||||||
from app.models.gateways import Gateway
|
from app.models.gateways import Gateway
|
||||||
|
from app.models.skills import GatewayInstalledSkill
|
||||||
from app.schemas.common import OkResponse
|
from app.schemas.common import OkResponse
|
||||||
from app.schemas.gateways import (
|
from app.schemas.gateways import (
|
||||||
GatewayCreate,
|
GatewayCreate,
|
||||||
|
|||||||
@@ -20,10 +20,8 @@ from sqlmodel import col, select
|
|||||||
from app.api.deps import require_org_admin
|
from app.api.deps import require_org_admin
|
||||||
from app.core.time import utcnow
|
from app.core.time import utcnow
|
||||||
from app.db.session import get_session
|
from app.db.session import get_session
|
||||||
from app.models.gateway_installed_skills import GatewayInstalledSkill
|
|
||||||
from app.models.gateways import Gateway
|
from app.models.gateways import Gateway
|
||||||
from app.models.marketplace_skills import MarketplaceSkill
|
from app.models.skills import GatewayInstalledSkill, MarketplaceSkill, SkillPack
|
||||||
from app.models.skill_packs import SkillPack
|
|
||||||
from app.schemas.common import OkResponse
|
from app.schemas.common import OkResponse
|
||||||
from app.schemas.skills_marketplace import (
|
from app.schemas.skills_marketplace import (
|
||||||
MarketplaceSkillActionResponse,
|
MarketplaceSkillActionResponse,
|
||||||
|
|||||||
@@ -11,15 +11,13 @@ from app.models.board_onboarding import BoardOnboardingSession
|
|||||||
from app.models.board_webhook_payloads import BoardWebhookPayload
|
from app.models.board_webhook_payloads import BoardWebhookPayload
|
||||||
from app.models.board_webhooks import BoardWebhook
|
from app.models.board_webhooks import BoardWebhook
|
||||||
from app.models.boards import Board
|
from app.models.boards import Board
|
||||||
from app.models.gateway_installed_skills import GatewayInstalledSkill
|
|
||||||
from app.models.gateways import Gateway
|
from app.models.gateways import Gateway
|
||||||
from app.models.marketplace_skills import MarketplaceSkill
|
|
||||||
from app.models.organization_board_access import OrganizationBoardAccess
|
from app.models.organization_board_access import OrganizationBoardAccess
|
||||||
from app.models.organization_invite_board_access import OrganizationInviteBoardAccess
|
from app.models.organization_invite_board_access import OrganizationInviteBoardAccess
|
||||||
from app.models.organization_invites import OrganizationInvite
|
from app.models.organization_invites import OrganizationInvite
|
||||||
from app.models.organization_members import OrganizationMember
|
from app.models.organization_members import OrganizationMember
|
||||||
from app.models.organizations import Organization
|
from app.models.organizations import Organization
|
||||||
from app.models.skill_packs import SkillPack
|
from app.models.skills import GatewayInstalledSkill, MarketplaceSkill, SkillPack
|
||||||
from app.models.tag_assignments import TagAssignment
|
from app.models.tag_assignments import TagAssignment
|
||||||
from app.models.tags import Tag
|
from app.models.tags import Tag
|
||||||
from app.models.task_custom_fields import (
|
from app.models.task_custom_fields import (
|
||||||
|
|||||||
@@ -1,33 +0,0 @@
|
|||||||
"""Gateway-to-skill installation state records."""
|
|
||||||
|
|
||||||
from __future__ import annotations
|
|
||||||
|
|
||||||
from datetime import datetime
|
|
||||||
from uuid import UUID, uuid4
|
|
||||||
|
|
||||||
from sqlalchemy import UniqueConstraint
|
|
||||||
from sqlmodel import Field
|
|
||||||
|
|
||||||
from app.core.time import utcnow
|
|
||||||
from app.models.base import QueryModel
|
|
||||||
|
|
||||||
RUNTIME_ANNOTATION_TYPES = (datetime,)
|
|
||||||
|
|
||||||
|
|
||||||
class GatewayInstalledSkill(QueryModel, table=True):
|
|
||||||
"""Marks that a marketplace skill is installed for a specific gateway."""
|
|
||||||
|
|
||||||
__tablename__ = "gateway_installed_skills" # pyright: ignore[reportAssignmentType]
|
|
||||||
__table_args__ = (
|
|
||||||
UniqueConstraint(
|
|
||||||
"gateway_id",
|
|
||||||
"skill_id",
|
|
||||||
name="uq_gateway_installed_skills_gateway_id_skill_id",
|
|
||||||
),
|
|
||||||
)
|
|
||||||
|
|
||||||
id: UUID = Field(default_factory=uuid4, primary_key=True)
|
|
||||||
gateway_id: UUID = Field(foreign_key="gateways.id", index=True)
|
|
||||||
skill_id: UUID = Field(foreign_key="marketplace_skills.id", index=True)
|
|
||||||
created_at: datetime = Field(default_factory=utcnow)
|
|
||||||
updated_at: datetime = Field(default_factory=utcnow)
|
|
||||||
@@ -1,42 +0,0 @@
|
|||||||
"""Organization-scoped skill catalog entries for the skills marketplace."""
|
|
||||||
|
|
||||||
from __future__ import annotations
|
|
||||||
|
|
||||||
from datetime import datetime
|
|
||||||
from uuid import UUID, uuid4
|
|
||||||
|
|
||||||
from sqlalchemy import JSON, Column, UniqueConstraint
|
|
||||||
from sqlmodel import Field
|
|
||||||
|
|
||||||
from app.core.time import utcnow
|
|
||||||
from app.models.tenancy import TenantScoped
|
|
||||||
|
|
||||||
RUNTIME_ANNOTATION_TYPES = (datetime,)
|
|
||||||
|
|
||||||
|
|
||||||
class MarketplaceSkill(TenantScoped, table=True):
|
|
||||||
"""A marketplace skill entry that can be installed onto one or more gateways."""
|
|
||||||
|
|
||||||
__tablename__ = "marketplace_skills" # pyright: ignore[reportAssignmentType]
|
|
||||||
__table_args__ = (
|
|
||||||
UniqueConstraint(
|
|
||||||
"organization_id",
|
|
||||||
"source_url",
|
|
||||||
name="uq_marketplace_skills_org_source_url",
|
|
||||||
),
|
|
||||||
)
|
|
||||||
|
|
||||||
id: UUID = Field(default_factory=uuid4, primary_key=True)
|
|
||||||
organization_id: UUID = Field(foreign_key="organizations.id", index=True)
|
|
||||||
name: str
|
|
||||||
description: str | None = Field(default=None)
|
|
||||||
category: str | None = Field(default=None)
|
|
||||||
risk: str | None = Field(default=None)
|
|
||||||
source: str | None = Field(default=None)
|
|
||||||
source_url: str
|
|
||||||
metadata_: dict[str, object] = Field(
|
|
||||||
default_factory=dict,
|
|
||||||
sa_column=Column("metadata", JSON, nullable=False),
|
|
||||||
)
|
|
||||||
created_at: datetime = Field(default_factory=utcnow)
|
|
||||||
updated_at: datetime = Field(default_factory=utcnow)
|
|
||||||
@@ -1,40 +0,0 @@
|
|||||||
"""Organization-scoped skill pack sources."""
|
|
||||||
|
|
||||||
from __future__ import annotations
|
|
||||||
|
|
||||||
from datetime import datetime
|
|
||||||
from uuid import UUID, uuid4
|
|
||||||
|
|
||||||
from sqlalchemy import JSON, Column, UniqueConstraint
|
|
||||||
from sqlmodel import Field
|
|
||||||
|
|
||||||
from app.core.time import utcnow
|
|
||||||
from app.models.tenancy import TenantScoped
|
|
||||||
|
|
||||||
RUNTIME_ANNOTATION_TYPES = (datetime,)
|
|
||||||
|
|
||||||
|
|
||||||
class SkillPack(TenantScoped, table=True):
|
|
||||||
"""A pack repository URL that can be synced into marketplace skills."""
|
|
||||||
|
|
||||||
__tablename__ = "skill_packs" # pyright: ignore[reportAssignmentType]
|
|
||||||
__table_args__ = (
|
|
||||||
UniqueConstraint(
|
|
||||||
"organization_id",
|
|
||||||
"source_url",
|
|
||||||
name="uq_skill_packs_org_source_url",
|
|
||||||
),
|
|
||||||
)
|
|
||||||
|
|
||||||
id: UUID = Field(default_factory=uuid4, primary_key=True)
|
|
||||||
organization_id: UUID = Field(foreign_key="organizations.id", index=True)
|
|
||||||
name: str
|
|
||||||
description: str | None = Field(default=None)
|
|
||||||
source_url: str
|
|
||||||
branch: str = Field(default="main")
|
|
||||||
metadata_: dict[str, object] = Field(
|
|
||||||
default_factory=dict,
|
|
||||||
sa_column=Column("metadata", JSON, nullable=False),
|
|
||||||
)
|
|
||||||
created_at: datetime = Field(default_factory=utcnow)
|
|
||||||
updated_at: datetime = Field(default_factory=utcnow)
|
|
||||||
88
backend/app/models/skills.py
Normal file
88
backend/app/models/skills.py
Normal file
@@ -0,0 +1,88 @@
|
|||||||
|
"""Skill-related SQLModel tables for marketplace, packs, and installations."""
|
||||||
|
|
||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
from datetime import datetime
|
||||||
|
from uuid import UUID, uuid4
|
||||||
|
|
||||||
|
from sqlalchemy import JSON, Column, UniqueConstraint
|
||||||
|
from sqlmodel import Field
|
||||||
|
|
||||||
|
from app.core.time import utcnow
|
||||||
|
from app.models.base import QueryModel
|
||||||
|
from app.models.tenancy import TenantScoped
|
||||||
|
|
||||||
|
RUNTIME_ANNOTATION_TYPES = (datetime,)
|
||||||
|
|
||||||
|
|
||||||
|
class MarketplaceSkill(TenantScoped, table=True):
|
||||||
|
"""A marketplace skill entry that can be installed onto one or more gateways."""
|
||||||
|
|
||||||
|
__tablename__ = "marketplace_skills" # pyright: ignore[reportAssignmentType]
|
||||||
|
__table_args__ = (
|
||||||
|
UniqueConstraint(
|
||||||
|
"organization_id",
|
||||||
|
"source_url",
|
||||||
|
name="uq_marketplace_skills_org_source_url",
|
||||||
|
),
|
||||||
|
)
|
||||||
|
|
||||||
|
id: UUID = Field(default_factory=uuid4, primary_key=True)
|
||||||
|
organization_id: UUID = Field(foreign_key="organizations.id", index=True)
|
||||||
|
name: str
|
||||||
|
description: str | None = Field(default=None)
|
||||||
|
category: str | None = Field(default=None)
|
||||||
|
risk: str | None = Field(default=None)
|
||||||
|
source: str | None = Field(default=None)
|
||||||
|
source_url: str
|
||||||
|
metadata_: dict[str, object] = Field(
|
||||||
|
default_factory=dict,
|
||||||
|
sa_column=Column("metadata", JSON, nullable=False),
|
||||||
|
)
|
||||||
|
created_at: datetime = Field(default_factory=utcnow)
|
||||||
|
updated_at: datetime = Field(default_factory=utcnow)
|
||||||
|
|
||||||
|
|
||||||
|
class SkillPack(TenantScoped, table=True):
|
||||||
|
"""A pack repository URL that can be synced into marketplace skills."""
|
||||||
|
|
||||||
|
__tablename__ = "skill_packs" # pyright: ignore[reportAssignmentType]
|
||||||
|
__table_args__ = (
|
||||||
|
UniqueConstraint(
|
||||||
|
"organization_id",
|
||||||
|
"source_url",
|
||||||
|
name="uq_skill_packs_org_source_url",
|
||||||
|
),
|
||||||
|
)
|
||||||
|
|
||||||
|
id: UUID = Field(default_factory=uuid4, primary_key=True)
|
||||||
|
organization_id: UUID = Field(foreign_key="organizations.id", index=True)
|
||||||
|
name: str
|
||||||
|
description: str | None = Field(default=None)
|
||||||
|
source_url: str
|
||||||
|
branch: str = Field(default="main")
|
||||||
|
metadata_: dict[str, object] = Field(
|
||||||
|
default_factory=dict,
|
||||||
|
sa_column=Column("metadata", JSON, nullable=False),
|
||||||
|
)
|
||||||
|
created_at: datetime = Field(default_factory=utcnow)
|
||||||
|
updated_at: datetime = Field(default_factory=utcnow)
|
||||||
|
|
||||||
|
|
||||||
|
class GatewayInstalledSkill(QueryModel, table=True):
|
||||||
|
"""Marks that a marketplace skill is installed for a specific gateway."""
|
||||||
|
|
||||||
|
__tablename__ = "gateway_installed_skills" # pyright: ignore[reportAssignmentType]
|
||||||
|
__table_args__ = (
|
||||||
|
UniqueConstraint(
|
||||||
|
"gateway_id",
|
||||||
|
"skill_id",
|
||||||
|
name="uq_gateway_installed_skills_gateway_id_skill_id",
|
||||||
|
),
|
||||||
|
)
|
||||||
|
|
||||||
|
id: UUID = Field(default_factory=uuid4, primary_key=True)
|
||||||
|
gateway_id: UUID = Field(foreign_key="gateways.id", index=True)
|
||||||
|
skill_id: UUID = Field(foreign_key="marketplace_skills.id", index=True)
|
||||||
|
created_at: datetime = Field(default_factory=utcnow)
|
||||||
|
updated_at: datetime = Field(default_factory=utcnow)
|
||||||
@@ -21,7 +21,7 @@ from app.models.organization_invite_board_access import OrganizationInviteBoardA
|
|||||||
from app.models.organization_invites import OrganizationInvite
|
from app.models.organization_invites import OrganizationInvite
|
||||||
from app.models.organization_members import OrganizationMember
|
from app.models.organization_members import OrganizationMember
|
||||||
from app.models.organizations import Organization
|
from app.models.organizations import Organization
|
||||||
from app.models.skill_packs import SkillPack
|
from app.models.skills import SkillPack
|
||||||
from app.models.users import User
|
from app.models.users import User
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
|
|||||||
@@ -17,7 +17,7 @@ from app.models.organization_invite_board_access import OrganizationInviteBoardA
|
|||||||
from app.models.organization_invites import OrganizationInvite
|
from app.models.organization_invites import OrganizationInvite
|
||||||
from app.models.organization_members import OrganizationMember
|
from app.models.organization_members import OrganizationMember
|
||||||
from app.models.organizations import Organization
|
from app.models.organizations import Organization
|
||||||
from app.models.skill_packs import SkillPack
|
from app.models.skills import SkillPack
|
||||||
from app.models.users import User
|
from app.models.users import User
|
||||||
from app.schemas.organizations import OrganizationBoardAccessSpec, OrganizationMemberAccessUpdate
|
from app.schemas.organizations import OrganizationBoardAccessSpec, OrganizationMemberAccessUpdate
|
||||||
from app.services import organizations
|
from app.services import organizations
|
||||||
|
|||||||
@@ -23,12 +23,10 @@ from app.api.skills_marketplace import (
|
|||||||
)
|
)
|
||||||
from app.api.skills_marketplace import router as skills_marketplace_router
|
from app.api.skills_marketplace import router as skills_marketplace_router
|
||||||
from app.db.session import get_session
|
from app.db.session import get_session
|
||||||
from app.models.gateway_installed_skills import GatewayInstalledSkill
|
|
||||||
from app.models.gateways import Gateway
|
from app.models.gateways import Gateway
|
||||||
from app.models.marketplace_skills import MarketplaceSkill
|
|
||||||
from app.models.organization_members import OrganizationMember
|
from app.models.organization_members import OrganizationMember
|
||||||
from app.models.organizations import Organization
|
from app.models.organizations import Organization
|
||||||
from app.models.skill_packs import SkillPack
|
from app.models.skills import GatewayInstalledSkill, MarketplaceSkill, SkillPack
|
||||||
from app.services.organizations import OrganizationContext
|
from app.services.organizations import OrganizationContext
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user