py-aidol 0.1.0__tar.gz → 0.2.0__tar.gz

{py_aidol-0.1.0 → py_aidol-0.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: py-aidol
-Version: 0.1.0
+Version: 0.2.0
 Summary: Create and chat with your own AI idol group
 License: Apache-2.0
 Keywords: kpop,idol,aidol,ai-companion,chatbot,image-generation

{py_aidol-0.1.0 → py_aidol-0.2.0}/aidol/models/__init__.py

@@ -3,6 +3,7 @@ AIdol database models
 """
 
 from aidol.models.aidol import DBAIdol
+from aidol.models.aidol_lead import DBAIdolLead
 from aidol.models.companion import DBCompanion
 
-__all__ = ["DBAIdol", "DBCompanion"]
+__all__ = ["DBAIdol", "DBAIdolLead", "DBCompanion"]

{py_aidol-0.1.0 → py_aidol-0.2.0}/aidol/models/aidol.py

@@ -18,7 +18,9 @@ class DBAIdol(BaseModel):
     __tablename__ = "aidols"
 
     # id, created_at, updated_at inherited from BaseModel
-    name: Mapped[str] = mapped_column(String, nullable=False)
+    name: Mapped[str | None] = mapped_column(String, nullable=True)
     concept: Mapped[str | None] = mapped_column(String, nullable=True)
-    profile_image_url: Mapped[str] = mapped_column(String, nullable=False)
+    profile_image_url: Mapped[str | None] = mapped_column(String, nullable=True)
     claim_token: Mapped[str | None] = mapped_column(String(36), nullable=True)
+    email: Mapped[str | None] = mapped_column(String, nullable=True)
+    greeting: Mapped[str | None] = mapped_column(String, nullable=True)

py_aidol-0.2.0/aidol/models/aidol_lead.py

@@ -0,0 +1,22 @@
+"""
+AIdol Leads database model
+
+Uses aioia_core.models.BaseModel which provides:
+- id: Mapped[str] (primary key, UUID default)
+- created_at: Mapped[datetime]
+- updated_at: Mapped[datetime]
+"""
+
+from aioia_core.models import BaseModel
+from sqlalchemy import String
+from sqlalchemy.orm import Mapped, mapped_column
+
+
+class DBAIdolLead(BaseModel):
+    """AIdol Lead (viewer email) database model"""
+
+    __tablename__ = "aidol_leads"
+
+    # id, created_at, updated_at inherited from BaseModel
+    aidol_id: Mapped[str] = mapped_column(String, nullable=False)
+    email: Mapped[str] = mapped_column(String, nullable=False)

py_aidol-0.2.0/aidol/models/companion.py

@@ -0,0 +1,45 @@
+"""
+Companion database model
+
+Uses aioia_core.models.BaseModel which provides:
+- id: Mapped[str] (primary key, UUID default)
+- created_at: Mapped[datetime]
+- updated_at: Mapped[datetime]
+"""
+
+from aioia_core.models import BaseModel
+from sqlalchemy import ForeignKey, Index, Integer, String, Text
+from sqlalchemy.orm import Mapped, mapped_column
+
+
+class DBCompanion(BaseModel):
+    """Companion (member) database model"""
+
+    __tablename__ = "companions"
+
+    # id, created_at, updated_at inherited from BaseModel
+    aidol_id: Mapped[str | None] = mapped_column(ForeignKey("aidols.id"), nullable=True)
+    name: Mapped[str | None] = mapped_column(String, nullable=True)
+    gender: Mapped[str | None] = mapped_column(String, nullable=True)
+    grade: Mapped[str | None] = mapped_column(String, nullable=True)
+    biography: Mapped[str | None] = mapped_column(Text, nullable=True)
+    profile_picture_url: Mapped[str | None] = mapped_column(String, nullable=True)
+    system_prompt: Mapped[str | None] = mapped_column(Text, nullable=True)
+
+    # MBTI scores (1-10 scale)
+    mbti_energy: Mapped[int | None] = mapped_column(Integer, nullable=True)
+    mbti_perception: Mapped[int | None] = mapped_column(Integer, nullable=True)
+    mbti_judgment: Mapped[int | None] = mapped_column(Integer, nullable=True)
+    mbti_lifestyle: Mapped[int | None] = mapped_column(Integer, nullable=True)
+
+    # Stats (0-100 scale)
+    vocal: Mapped[int | None] = mapped_column(Integer, nullable=True)
+    dance: Mapped[int | None] = mapped_column(Integer, nullable=True)
+    rap: Mapped[int | None] = mapped_column(Integer, nullable=True)
+    visual: Mapped[int | None] = mapped_column(Integer, nullable=True)
+    stamina: Mapped[int | None] = mapped_column(Integer, nullable=True)
+    charm: Mapped[int | None] = mapped_column(Integer, nullable=True)
+
+    position: Mapped[str | None] = mapped_column(String, nullable=True)
+
+    __table_args__ = (Index("ix_companions_aidol_id", "aidol_id"),)

{py_aidol-0.1.0 → py_aidol-0.2.0}/aidol/repositories/__init__.py

@@ -3,9 +3,11 @@ AIdol repositories
 """
 
 from aidol.repositories.aidol import AIdolRepository
+from aidol.repositories.aidol_lead import AIdolLeadRepository
 from aidol.repositories.companion import CompanionRepository
 
 __all__ = [
     "AIdolRepository",
+    "AIdolLeadRepository",
     "CompanionRepository",
 ]

{py_aidol-0.1.0 → py_aidol-0.2.0}/aidol/repositories/aidol.py

@@ -25,6 +25,8 @@ def _convert_db_aidol_to_model(db_aidol: DBAIdol) -> AIdol:
     return AIdol(
         id=db_aidol.id,
         name=db_aidol.name,
+        email=db_aidol.email,
+        greeting=db_aidol.greeting,
         concept=db_aidol.concept,
         profile_image_url=db_aidol.profile_image_url,
         claim_token=db_aidol.claim_token,

py_aidol-0.2.0/aidol/repositories/aidol_lead.py

@@ -0,0 +1,52 @@
+"""
+AIdol Lead repository
+
+Implements BaseRepository pattern for BaseCrudRouter compatibility.
+"""
+
+from datetime import timezone
+
+from aioia_core.repositories import BaseRepository
+from pydantic import BaseModel
+from sqlalchemy.orm import Session
+
+from aidol.models import DBAIdolLead
+from aidol.schemas import AIdolLead, AIdolLeadCreate
+
+
+class _AIdolLeadUpdate(BaseModel):
+    """Placeholder for update schema (not used)."""
+
+
+def _convert_db_aidol_lead_to_model(db_lead: DBAIdolLead) -> AIdolLead:
+    """Convert DB AIdolLead to Pydantic model."""
+    return AIdolLead(
+        id=db_lead.id,
+        aidol_id=db_lead.aidol_id,
+        email=db_lead.email,
+        created_at=db_lead.created_at.replace(tzinfo=timezone.utc),
+        updated_at=db_lead.updated_at.replace(tzinfo=timezone.utc),
+    )
+
+
+def _convert_aidol_lead_create_to_db(schema: AIdolLeadCreate) -> dict:
+    """Convert AIdolLeadCreate schema to DB model data dict."""
+    return schema.model_dump(exclude_unset=True)
+
+
+class AIdolLeadRepository(
+    BaseRepository[AIdolLead, DBAIdolLead, AIdolLeadCreate, _AIdolLeadUpdate]
+):
+    """
+    Database-backed AIdolLead repository.
+
+    Extends BaseRepository for CRUD operations compatible with BaseCrudRouter.
+    """
+
+    def __init__(self, db_session: Session):
+        super().__init__(
+            db_session=db_session,
+            db_model=DBAIdolLead,
+            convert_to_model=_convert_db_aidol_lead_to_model,
+            convert_to_db_model=_convert_aidol_lead_create_to_db,
+        )

{py_aidol-0.1.0 → py_aidol-0.2.0}/aidol/repositories/companion.py

@@ -10,7 +10,15 @@ from aioia_core.repositories import BaseRepository
 from sqlalchemy.orm import Session
 
 from aidol.models import DBCompanion
-from aidol.schemas import Companion, CompanionCreate, CompanionUpdate
+from aidol.schemas import (
+    Companion,
+    CompanionCreate,
+    CompanionStats,
+    CompanionUpdate,
+    Gender,
+    Grade,
+    Position,
+)
 
 
 def _convert_db_companion_to_model(db_companion: DBCompanion) -> Companion:
@@ -23,20 +31,45 @@ def _convert_db_companion_to_model(db_companion: DBCompanion) -> Companion:
         id=db_companion.id,
         aidol_id=db_companion.aidol_id,
         name=db_companion.name,
+        gender=Gender(db_companion.gender) if db_companion.gender else None,
+        grade=Grade(db_companion.grade) if db_companion.grade else None,
         biography=db_companion.biography,
         profile_picture_url=db_companion.profile_picture_url,
+        position=Position(db_companion.position) if db_companion.position else None,
         system_prompt=db_companion.system_prompt,
+        mbti_energy=db_companion.mbti_energy,
+        mbti_perception=db_companion.mbti_perception,
+        mbti_judgment=db_companion.mbti_judgment,
+        mbti_lifestyle=db_companion.mbti_lifestyle,
+        stats=CompanionStats(
+            vocal=db_companion.vocal or 0,
+            dance=db_companion.dance or 0,
+            rap=db_companion.rap or 0,
+            visual=db_companion.visual or 0,
+            stamina=db_companion.stamina or 0,
+            charm=db_companion.charm or 0,
+        ),
         created_at=db_companion.created_at.replace(tzinfo=timezone.utc),
         updated_at=db_companion.updated_at.replace(tzinfo=timezone.utc),
     )
 
 
-def _convert_companion_create_to_db(schema: CompanionCreate) -> dict:
-    """Convert CompanionCreate schema to DB model data dict.
+def _convert_companion_schema_to_db(
+    schema: CompanionCreate | CompanionUpdate,
+) -> dict:
+    """Convert CompanionCreate/Update schema to DB model data dict.
 
+    Decomposes nested stats object into individual DB columns.
     Includes system_prompt for AI configuration.
     """
-    return schema.model_dump(exclude_unset=True)
+    data = schema.model_dump(exclude_unset=True, exclude={"stats"})
+
+    # Decompose stats into individual columns
+    if schema.stats is not None:
+        stats_dict = schema.stats.model_dump()
+        data.update(stats_dict)
+
+    return data
 
 
 class CompanionRepository(
@@ -53,5 +86,5 @@ class CompanionRepository(
             db_session=db_session,
             db_model=DBCompanion,
             convert_to_model=_convert_db_companion_to_model,
-            convert_to_db_model=_convert_companion_create_to_db,
+            convert_to_db_model=_convert_companion_schema_to_db,
         )

{py_aidol-0.1.0 → py_aidol-0.2.0}/aidol/schemas/__init__.py

@@ -12,12 +12,17 @@ from aidol.schemas.aidol import (
     ImageGenerationRequest,
     ImageGenerationResponse,
 )
+from aidol.schemas.aidol_lead import AIdolLead, AIdolLeadBase, AIdolLeadCreate
 from aidol.schemas.companion import (
     Companion,
     CompanionBase,
     CompanionCreate,
     CompanionPublic,
+    CompanionStats,
     CompanionUpdate,
+    Gender,
+    Grade,
+    Position,
 )
 
 __all__ = [
@@ -29,9 +34,16 @@ __all__ = [
     "ImageGenerationData",
     "ImageGenerationRequest",
     "ImageGenerationResponse",
+    "AIdolLead",
+    "AIdolLeadBase",
+    "AIdolLeadCreate",
     "Companion",
     "CompanionBase",
     "CompanionCreate",
     "CompanionPublic",
+    "CompanionStats",
     "CompanionUpdate",
+    "Gender",
+    "Grade",
+    "Position",
 ]

{py_aidol-0.1.0 → py_aidol-0.2.0}/aidol/schemas/aidol.py

@@ -24,20 +24,22 @@ class AIdolBase(BaseModel):
 
     model_config = ConfigDict(populate_by_name=True, alias_generator=camelize)
 
-    name: str = Field(..., description="AIdol group name")
+    name: str | None = Field(default=None, description="AIdol group name")
+    email: str | None = Field(default=None, description="Creator email")
+    greeting: str | None = Field(default=None, description="Greeting message")
     concept: str | None = Field(default=None, description="Group concept or theme")
-    profile_image_url: str = Field(..., description="Profile image URL")
+    profile_image_url: str | None = Field(default=None, description="Profile image URL")
 
 
 class AIdolCreate(AIdolBase):
     """Schema for creating an AIdol group (no id).
 
-    claim_token is optional for anonymous ownership verification.
+    claim_token is required for ownership verification.
     """
 
-    claim_token: str | None = Field(
-        default=None,
-        description="Optional client-generated UUID for ownership verification",
+    claim_token: str = Field(
+        ...,
+        description="Client-generated UUID for ownership verification",
     )
 
 
@@ -47,6 +49,8 @@ class AIdolUpdate(BaseModel):
     model_config = ConfigDict(populate_by_name=True, alias_generator=camelize)
 
     name: str | None = Field(default=None, description="AIdol group name")
+    email: str | None = Field(default=None, description="Creator email")
+    greeting: str | None = Field(default=None, description="Greeting message")
     concept: str | None = Field(default=None, description="Group concept or theme")
     profile_image_url: str | None = Field(default=None, description="Profile image URL")
 

py_aidol-0.2.0/aidol/schemas/aidol_lead.py

@@ -0,0 +1,38 @@
+"""
+AIdol Lead (viewer email) schemas
+
+Schema hierarchy:
+- AIdolLeadBase: Common fields
+- AIdolLeadCreate: For creating a lead (no id)
+- AIdolLead: Response with all fields
+"""
+
+from datetime import datetime
+
+from humps import camelize
+from pydantic import BaseModel, ConfigDict, Field
+
+
+class AIdolLeadBase(BaseModel):
+    """Base AIdol Lead model with common fields."""
+
+    model_config = ConfigDict(populate_by_name=True, alias_generator=camelize)
+
+    aidol_id: str = Field(..., description="AIdol group ID")
+    email: str = Field(..., description="Viewer email")
+
+
+class AIdolLeadCreate(AIdolLeadBase):
+    """Schema for creating an AIdol lead (no id)."""
+
+
+class AIdolLead(AIdolLeadBase):
+    """AIdol Lead response schema with id and timestamps."""
+
+    model_config = ConfigDict(
+        populate_by_name=True, from_attributes=True, alias_generator=camelize
+    )
+
+    id: str = Field(..., description="Lead ID")
+    created_at: datetime = Field(..., description="Creation timestamp")
+    updated_at: datetime = Field(..., description="Last update timestamp")

py_aidol-0.2.0/aidol/schemas/companion.py

@@ -0,0 +1,210 @@
+"""
+Companion (member) schemas
+
+Schema hierarchy:
+- CompanionStats: Nested stats object for request/response
+- CompanionBase: Mutable fields (used in Create/Update)
+- CompanionCreate: Base + system_prompt (mutable, but sensitive)
+- CompanionUpdate: All fields optional for partial updates
+- Companion: Response with all fields including system_prompt (internal use)
+- CompanionPublic: Response without sensitive fields (API use)
+"""
+
+from datetime import datetime
+from enum import Enum
+
+from humps import camelize
+from pydantic import BaseModel, ConfigDict, Field
+
+# ---------------------------------------------------------------------------
+# Enums
+# ---------------------------------------------------------------------------
+
+
+class Gender(str, Enum):
+    """Gender options for companions."""
+
+    MALE = "male"
+    FEMALE = "female"
+
+
+class Grade(str, Enum):
+    """Grade levels for companions."""
+
+    A = "A"
+    B = "B"
+    C = "C"
+    F = "F"
+
+
+class Position(str, Enum):
+    """Position roles in the group."""
+
+    LEADER = "leader"
+    MAIN_VOCAL = "mainVocal"
+    SUB_VOCAL = "subVocal"
+    MAIN_DANCER = "mainDancer"
+    SUB_DANCER = "subDancer"
+    MAIN_RAPPER = "mainRapper"
+    SUB_RAPPER = "subRapper"
+    VISUAL = "visual"
+    MAKNAE = "maknae"
+
+
+# ---------------------------------------------------------------------------
+# Nested Objects
+# ---------------------------------------------------------------------------
+
+
+class CompanionStats(BaseModel):
+    """Nested stats object for API request/response."""
+
+    model_config = ConfigDict(populate_by_name=True, alias_generator=camelize)
+
+    vocal: int = Field(default=0, ge=0, le=100, description="Vocal skill")
+    dance: int = Field(default=0, ge=0, le=100, description="Dance skill")
+    rap: int = Field(default=0, ge=0, le=100, description="Rap skill")
+    visual: int = Field(default=0, ge=0, le=100, description="Visual score")
+    stamina: int = Field(default=0, ge=0, le=100, description="Stamina")
+    charm: int = Field(default=0, ge=0, le=100, description="Charm score")
+
+
+# ---------------------------------------------------------------------------
+# Request Schemas
+# ---------------------------------------------------------------------------
+
+
+class CompanionBase(BaseModel):
+    """Base companion model with common mutable fields.
+
+    Contains fields that can be modified after creation.
+    Excludes system_prompt (sensitive, requires explicit inclusion).
+    """
+
+    model_config = ConfigDict(populate_by_name=True, alias_generator=camelize)
+
+    aidol_id: str | None = Field(default=None, description="AIdol group ID")
+    name: str | None = Field(default=None, description="Companion name")
+    gender: Gender | None = Field(default=None, description="Gender")
+    grade: Grade | None = Field(default=None, description="Grade level")
+    biography: str | None = Field(default=None, description="Companion biography")
+    profile_picture_url: str | None = Field(
+        default=None, description="Profile picture URL"
+    )
+    position: Position | None = Field(default=None, description="Position in group")
+
+    # MBTI scores (1-10)
+    mbti_energy: int | None = Field(default=None, ge=1, le=10, description="E↔I (1-10)")
+    mbti_perception: int | None = Field(
+        default=None, ge=1, le=10, description="S↔N (1-10)"
+    )
+    mbti_judgment: int | None = Field(
+        default=None, ge=1, le=10, description="T↔F (1-10)"
+    )
+    mbti_lifestyle: int | None = Field(
+        default=None, ge=1, le=10, description="J↔P (1-10)"
+    )
+
+    # Stats (nested object)
+    stats: CompanionStats = Field(
+        default_factory=CompanionStats, description="Ability stats"
+    )
+
+
+class CompanionCreate(CompanionBase):
+    """Schema for creating a companion (no id).
+
+    Includes system_prompt for creation (excluded from response for security).
+    """
+
+    system_prompt: str | None = Field(
+        default=None, description="AI system prompt (not exposed in responses)"
+    )
+
+
+class CompanionUpdate(BaseModel):
+    """Schema for updating a companion (all fields optional)."""
+
+    model_config = ConfigDict(populate_by_name=True, alias_generator=camelize)
+
+    aidol_id: str | None = Field(default=None, description="AIdol group ID")
+    name: str | None = Field(default=None, description="Companion name")
+    gender: Gender | None = Field(default=None, description="Gender")
+    grade: Grade | None = Field(default=None, description="Grade level")
+    biography: str | None = Field(default=None, description="Companion biography")
+    profile_picture_url: str | None = Field(
+        default=None, description="Profile picture URL"
+    )
+    position: Position | None = Field(default=None, description="Position in group")
+    system_prompt: str | None = Field(
+        default=None, description="AI system prompt (not exposed in responses)"
+    )
+
+    # MBTI scores (1-10)
+    mbti_energy: int | None = Field(default=None, ge=1, le=10, description="E↔I (1-10)")
+    mbti_perception: int | None = Field(
+        default=None, ge=1, le=10, description="S↔N (1-10)"
+    )
+    mbti_judgment: int | None = Field(
+        default=None, ge=1, le=10, description="T↔F (1-10)"
+    )
+    mbti_lifestyle: int | None = Field(
+        default=None, ge=1, le=10, description="J↔P (1-10)"
+    )
+
+    # Stats (nested object, optional for updates)
+    stats: CompanionStats | None = Field(default=None, description="Ability stats")
+
+
+# ---------------------------------------------------------------------------
+# Response Schemas
+# ---------------------------------------------------------------------------
+
+
+class Companion(CompanionBase):
+    """Companion response schema with id and timestamps.
+
+    Includes system_prompt for internal use (Service layer).
+    Use CompanionPublic for API responses to exclude sensitive fields.
+    """
+
+    model_config = ConfigDict(
+        populate_by_name=True, from_attributes=True, alias_generator=camelize
+    )
+
+    id: str = Field(..., description="Companion ID")
+    system_prompt: str | None = Field(
+        default=None, description="AI system prompt (sensitive, internal use only)"
+    )
+    created_at: datetime = Field(..., description="Creation timestamp")
+    updated_at: datetime = Field(..., description="Last update timestamp")
+
+
+class CompanionPublic(BaseModel):
+    """Public companion response schema for frontend.
+
+    - Excludes system_prompt for security
+    - Uses nested stats object
+    - Includes calculated mbti string
+    """
+
+    model_config = ConfigDict(
+        populate_by_name=True, from_attributes=True, alias_generator=camelize
+    )
+
+    id: str = Field(..., description="Companion ID")
+    aidol_id: str | None = Field(default=None, description="AIdol group ID")
+    name: str | None = Field(default=None, description="Companion name")
+    gender: Gender | None = Field(default=None, description="Gender")
+    grade: Grade | None = Field(default=None, description="Grade level")
+    biography: str | None = Field(default=None, description="Companion biography")
+    profile_picture_url: str | None = Field(
+        default=None, description="Profile picture URL"
+    )
+    position: Position | None = Field(default=None, description="Position in group")
+    mbti: str | None = Field(default=None, description="Calculated MBTI (e.g., ENFP)")
+    stats: CompanionStats = Field(
+        default_factory=CompanionStats, description="Ability stats"
+    )
+    created_at: datetime = Field(..., description="Creation timestamp")
+    updated_at: datetime = Field(..., description="Last update timestamp")

{py_aidol-0.1.0 → py_aidol-0.2.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "py-aidol"
-version = "0.1.0"
+version = "0.2.0"
 description = "Create and chat with your own AI idol group"
 authors = ["AIoIA, Inc. <devops@aioia.ai>"]
 readme = "README.md"

py_aidol-0.1.0/aidol/models/companion.py

@@ -1,27 +0,0 @@
-"""
-Companion database model
-
-Uses aioia_core.models.BaseModel which provides:
-- id: Mapped[str] (primary key, UUID default)
-- created_at: Mapped[datetime]
-- updated_at: Mapped[datetime]
-"""
-
-from aioia_core.models import BaseModel
-from sqlalchemy import ForeignKey, Index, String, Text
-from sqlalchemy.orm import Mapped, mapped_column
-
-
-class DBCompanion(BaseModel):
-    """Companion (member) database model"""
-
-    __tablename__ = "companions"
-
-    # id, created_at, updated_at inherited from BaseModel
-    aidol_id: Mapped[str | None] = mapped_column(ForeignKey("aidols.id"), nullable=True)
-    name: Mapped[str] = mapped_column(String, nullable=False)
-    biography: Mapped[str | None] = mapped_column(Text, nullable=True)
-    profile_picture_url: Mapped[str | None] = mapped_column(String, nullable=True)
-    system_prompt: Mapped[str | None] = mapped_column(Text, nullable=True)
-
-    __table_args__ = (Index("ix_companions_aidol_id", "aidol_id"),)

py_aidol-0.1.0/aidol/schemas/companion.py

@@ -1,93 +0,0 @@
-"""
-Companion (member) schemas
-
-Schema hierarchy:
-- CompanionBase: Mutable fields (used in Create/Update)
-- CompanionCreate: Base + system_prompt (mutable, but sensitive)
-- CompanionUpdate: Base + system_prompt (mutable, but sensitive)
-- Companion: Response with all fields including system_prompt (internal use)
-- CompanionPublic: Response without sensitive fields (API use)
-"""
-
-from datetime import datetime
-
-from humps import camelize
-from pydantic import BaseModel, ConfigDict, Field
-
-
-class CompanionBase(BaseModel):
-    """Base companion model with common mutable fields.
-
-    Contains fields that can be modified after creation.
-    Excludes system_prompt (sensitive, requires explicit inclusion).
-    """
-
-    model_config = ConfigDict(populate_by_name=True, alias_generator=camelize)
-
-    aidol_id: str | None = Field(default=None, description="AIdol group ID")
-    name: str = Field(..., description="Companion name")
-    biography: str | None = Field(default=None, description="Companion biography")
-    profile_picture_url: str | None = Field(
-        default=None, description="Profile picture URL"
-    )
-
-
-class CompanionCreate(CompanionBase):
-    """Schema for creating a companion (no id).
-
-    Includes system_prompt for creation (excluded from response for security).
-    """
-
-    system_prompt: str | None = Field(
-        default=None, description="AI system prompt (not exposed in responses)"
-    )
-
-
-class CompanionUpdate(BaseModel):
-    """Schema for updating a companion (all fields optional)."""
-
-    model_config = ConfigDict(populate_by_name=True, alias_generator=camelize)
-
-    aidol_id: str | None = Field(default=None, description="AIdol group ID")
-    name: str | None = Field(default=None, description="Companion name")
-    biography: str | None = Field(default=None, description="Companion biography")
-    profile_picture_url: str | None = Field(
-        default=None, description="Profile picture URL"
-    )
-    system_prompt: str | None = Field(
-        default=None, description="AI system prompt (not exposed in responses)"
-    )
-
-
-class Companion(CompanionBase):
-    """Companion response schema with id and timestamps.
-
-    Includes system_prompt for internal use (Service layer).
-    Use CompanionPublic for API responses to exclude sensitive fields.
-    """
-
-    model_config = ConfigDict(
-        populate_by_name=True, from_attributes=True, alias_generator=camelize
-    )
-
-    id: str = Field(..., description="Companion ID")
-    system_prompt: str | None = Field(
-        default=None, description="AI system prompt (sensitive, internal use only)"
-    )
-    created_at: datetime = Field(..., description="Creation timestamp")
-    updated_at: datetime = Field(..., description="Last update timestamp")
-
-
-class CompanionPublic(CompanionBase):
-    """Public companion response schema without sensitive fields.
-
-    Excludes system_prompt for API responses.
-    """
-
-    model_config = ConfigDict(
-        populate_by_name=True, from_attributes=True, alias_generator=camelize
-    )
-
-    id: str = Field(..., description="Companion ID")
-    created_at: datetime = Field(..., description="Creation timestamp")
-    updated_at: datetime = Field(..., description="Last update timestamp")