py-aidol 0.5.0__tar.gz → 0.6.0__tar.gz

{py_aidol-0.5.0 → py_aidol-0.6.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: py-aidol
-Version: 0.5.0
+Version: 0.6.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.5.0 → py_aidol-0.6.0}/aidol/api/aidol.py

@@ -6,11 +6,13 @@ Public endpoints for AIdol group creation and retrieval.
 Public access pattern: no authentication required.
 """
 
+from typing import Annotated
+
 from aioia_core.auth import UserInfoProvider
 from aioia_core.errors import ErrorResponse
 from aioia_core.fastapi import BaseCrudRouter
 from aioia_core.settings import JWTSettings
-from fastapi import APIRouter, Depends, HTTPException, Response, status
+from fastapi import APIRouter, Cookie, Depends, HTTPException, status
 from pydantic import BaseModel
 from sqlalchemy.orm import sessionmaker
 
@@ -20,7 +22,13 @@ from aidol.protocols import (
     AIdolRepositoryProtocol,
     ImageStorageProtocol,
 )
-from aidol.schemas import AIdol, AIdolCreate, AIdolPublic, AIdolUpdate
+from aidol.schemas import (
+    AIdol,
+    AIdolCreate,
+    AIdolCreateWithClaim,
+    AIdolPublic,
+    AIdolUpdate,
+)
 from aidol.settings import GoogleGenAISettings
 
 
@@ -31,7 +39,7 @@ class AIdolCreateResponse(BaseModel):
 
 
 class AIdolRouter(
-    BaseCrudRouter[AIdol, AIdolCreate, AIdolUpdate, AIdolRepositoryProtocol]
+    BaseCrudRouter[AIdol, AIdolCreateWithClaim, AIdolUpdate, AIdolRepositoryProtocol]
 ):
     """
     AIdol router with public endpoints.
@@ -106,15 +114,21 @@ class AIdolRouter(
         )
         async def create_aidol(
             request: AIdolCreate,
-            response: Response,
+            claim_token: Annotated[str | None, Cookie(alias="ClaimToken")] = None,
             repository: AIdolRepositoryProtocol = Depends(self.get_repository_dep),
         ):
             """Create a new AIdol group."""
-            created = repository.create(request)
+            if not claim_token:
+                raise HTTPException(
+                    status_code=status.HTTP_400_BAD_REQUEST,
+                    detail="ClaimToken cookie is required",
+                )
 
-            # Set ClaimToken header
-            if created.claim_token:
-                response.headers["ClaimToken"] = created.claim_token
+            # Convert body schema to internal schema with claim_token from Cookie
+            create_data = AIdolCreateWithClaim(
+                **request.model_dump(), claim_token=claim_token
+            )
+            created = repository.create(create_data)
 
             # Return only id
             return AIdolCreateResponse(id=created.id)

{py_aidol-0.5.0 → py_aidol-0.6.0}/aidol/api/chatroom.py

@@ -4,11 +4,13 @@ Chatroom API router
 Implements BaseCrudRouter pattern for consistency with aioia-core patterns.
 """
 
+from typing import Annotated
+
 from aioia_core.auth import UserInfoProvider
 from aioia_core.errors import ErrorResponse
 from aioia_core.fastapi import BaseCrudRouter
 from aioia_core.settings import JWTSettings, OpenAIAPISettings
-from fastapi import APIRouter, Depends, HTTPException, status
+from fastapi import APIRouter, Cookie, Depends, HTTPException, status
 from humps import camelize
 from langchain_core.messages import AIMessage, BaseMessage, HumanMessage
 from pydantic import BaseModel, ConfigDict, Field
@@ -28,6 +30,7 @@ from aidol.schemas import (
     CompanionMessageCreate,
     Message,
     MessageCreate,
+    MessageCreateWithClaim,
     ModelSettings,
     Persona,
     SenderType,
@@ -181,19 +184,32 @@ class ChatroomRouter(
         async def send_message(
             item_id: str,
             request: MessageCreate,
+            claim_token: Annotated[str | None, Cookie(alias="ClaimToken")] = None,
             repository: ChatroomRepositoryProtocol = Depends(self.get_repository_dep),
         ):
             """Send a message to a chatroom."""
+            # Guard Clause: ClaimToken cookie is required
+            if not claim_token:
+                raise HTTPException(
+                    status_code=status.HTTP_400_BAD_REQUEST,
+                    detail="ClaimToken cookie is required",
+                )
+
             # Verify chatroom exists
             self._get_item_or_404(repository, item_id)
 
             # Enforce sender_type as USER to prevent spoofing
             request.sender_type = SenderType.USER
 
-            # Pass MessageCreate directly (aioia-core pattern)
+            # Convert to internal schema with claim_token from Cookie
+            message_data = MessageCreateWithClaim(
+                **request.model_dump(), claim_token=claim_token
+            )
+
+            # Pass MessageCreateWithClaim to repository
             return repository.add_message_to_chatroom(
                 chatroom_id=item_id,
-                message=request,
+                message=message_data,
             )
 
     def _register_generate_response_route(self) -> None:

{py_aidol-0.5.0 → py_aidol-0.6.0}/aidol/api/companion.py

@@ -20,13 +20,7 @@ from aidol.protocols import (
     CompanionRepositoryProtocol,
     ImageStorageProtocol,
 )
-from aidol.schemas import (
-    Companion,
-    CompanionCreate,
-    CompanionPublic,
-    CompanionUpdate,
-    Gender,
-)
+from aidol.schemas import Companion, CompanionCreate, CompanionPublic, CompanionUpdate
 from aidol.services.companion_service import to_companion_public
 from aidol.settings import GoogleGenAISettings
 
@@ -90,45 +84,32 @@ class CompanionRouter(
             response_model=CompanionPaginatedResponse,
             status_code=status.HTTP_200_OK,
             summary="List Companions",
-            description="List all Companions with optional filtering by gender and cast status",
+            description="List all Companions with pagination, sorting, and filtering",
         )
         async def list_companions(
-            gender: Gender | None = Query(None, description="Filter by gender"),
-            is_cast: bool | None = Query(
-                None, alias="isCast", description="Filter by cast status"
+            current: int = Query(1, ge=1, description="Current page number"),
+            page_size: int = Query(
+                10, alias="pageSize", ge=1, le=100, description="Items per page"
+            ),
+            sort_param: str | None = Query(
+                None,
+                alias="sort",
+                description='Sorting criteria in JSON format. Example: [["createdAt","desc"]]',
+            ),
+            filters_param: str | None = Query(
+                None,
+                alias="filters",
+                description="Filter conditions (JSON format)",
             ),
-            aidol_id: str | None = Query(None, description="Filter by AIdol Group ID"),
             repository: CompanionRepositoryProtocol = Depends(self.get_repository_dep),
         ):
-            """List Companions with optional gender and isCast filters."""
-            filter_list: list[dict] = []
-
-            # Add filters only if provided
-            if gender is not None:
-                filter_list.append(
-                    {"field": "gender", "operator": "eq", "value": gender.value}
-                )
-
-            # Filter by aidol_id if provided
-            if aidol_id is not None:
-                filter_list.append(
-                    {"field": "aidol_id", "operator": "eq", "value": aidol_id}
-                )
-
-            # isCast is derived from aidol_id presence
-            # isCast=true → aidol_id is not null (belongs to a group)
-            # isCast=false → aidol_id is null (not in a group)
-            if is_cast is True:
-                filter_list.append(
-                    {"field": "aidol_id", "operator": "ne", "value": None}
-                )
-            elif is_cast is False:
-                filter_list.append(
-                    {"field": "aidol_id", "operator": "eq", "value": None}
-                )
-
+            """List Companions with pagination, sorting, and filtering."""
+            sort_list, filter_list = self._parse_query_params(sort_param, filters_param)
             items, total = repository.get_all(
-                filters=filter_list if filter_list else None,
+                current=current,
+                page_size=page_size,
+                sort=sort_list,
+                filters=filter_list,
             )
             # Convert to Public schema (exclude system_prompt)
             public_items = [to_companion_public(c) for c in items]

{py_aidol-0.5.0 → py_aidol-0.6.0}/aidol/api/lead.py

@@ -8,7 +8,7 @@ from typing import Annotated
 
 from aioia_core.fastapi import BaseCrudRouter
 from aioia_core.settings import JWTSettings
-from fastapi import APIRouter, Depends, Header, status
+from fastapi import APIRouter, Cookie, Depends, status
 from pydantic import BaseModel
 from sqlalchemy.orm import Session, sessionmaker
 
@@ -60,7 +60,7 @@ class LeadRouter(
         )
         async def create_lead(
             request: AIdolLeadCreate,
-            claim_token: Annotated[str | None, Header(alias="ClaimToken")] = None,
+            claim_token: Annotated[str | None, Cookie(alias="ClaimToken")] = None,
             db_session: Session = Depends(self.get_db_dep),
             lead_repository: AIdolLeadRepositoryProtocol = Depends(
                 self.get_repository_dep

{py_aidol-0.5.0 → py_aidol-0.6.0}/aidol/protocols.py

@@ -16,7 +16,7 @@ from sqlalchemy.orm import Session
 
 from aidol.schemas import (
     AIdol,
-    AIdolCreate,
+    AIdolCreateWithClaim,
     AIdolLead,
     AIdolLeadCreate,
     AIdolUpdate,
@@ -27,7 +27,7 @@ from aidol.schemas import (
     CompanionCreate,
     CompanionUpdate,
     Message,
-    MessageCreate,
+    MessageCreateWithClaim,
 )
 
 
@@ -64,13 +64,13 @@ class ChatroomRepositoryProtocol(
         ...
 
     def add_message_to_chatroom(
-        self, chatroom_id: str, message: MessageCreate
+        self, chatroom_id: str, message: MessageCreateWithClaim
     ) -> Message:
         """Add a message to a chatroom.
 
         Args:
             chatroom_id: Chatroom ID.
-            message: MessageCreate or CompanionMessageCreate schema.
+            message: MessageCreateWithClaim or CompanionMessageCreate schema (with claim_token).
         """
         ...
 
@@ -95,7 +95,7 @@ class ChatroomRepositoryFactoryProtocol(Protocol):
 
 
 class AIdolRepositoryProtocol(
-    CrudRepositoryProtocol[AIdol, AIdolCreate, AIdolUpdate], Protocol
+    CrudRepositoryProtocol[AIdol, AIdolCreateWithClaim, AIdolUpdate], Protocol
 ):
     """Protocol defining AIdol repository expectations.
 

{py_aidol-0.5.0 → py_aidol-0.6.0}/aidol/repositories/aidol.py

@@ -13,7 +13,7 @@ from aioia_core.repositories import BaseRepository
 from sqlalchemy.orm import Session
 
 from aidol.models import DBAIdol
-from aidol.schemas import AIdol, AIdolCreate, AIdolUpdate
+from aidol.schemas import AIdol, AIdolCreateWithClaim, AIdolUpdate
 
 
 def _convert_db_aidol_to_model(db_aidol: DBAIdol) -> AIdol:
@@ -35,15 +35,14 @@ def _convert_db_aidol_to_model(db_aidol: DBAIdol) -> AIdol:
     )
 
 
-def _convert_aidol_create_to_db(schema: AIdolCreate) -> dict:
-    """Convert AIdolCreate schema to DB model data dict.
-
-    Includes claim_token for ownership verification.
-    """
+def _convert_aidol_create_to_db(schema: AIdolCreateWithClaim) -> dict:
+    """Convert AIdolCreateWithClaim schema to DB model data dict."""
     return schema.model_dump(exclude_unset=True)
 
 
-class AIdolRepository(BaseRepository[AIdol, DBAIdol, AIdolCreate, AIdolUpdate]):
+class AIdolRepository(
+    BaseRepository[AIdol, DBAIdol, AIdolCreateWithClaim, AIdolUpdate]
+):
     """
     Database-backed AIdol repository.
 

{py_aidol-0.5.0 → py_aidol-0.6.0}/aidol/repositories/chatroom.py

@@ -17,7 +17,7 @@ from aidol.schemas import (
     ChatroomUpdate,
     CompanionMessage,
     Message,
-    MessageCreate,
+    MessageCreateWithClaim,
     SenderType,
 )
 
@@ -81,7 +81,7 @@ class ChatroomRepository(
     # ========================================
 
     def add_message_to_chatroom(
-        self, chatroom_id: str, message: MessageCreate
+        self, chatroom_id: str, message: MessageCreateWithClaim
     ) -> Message:
         """
         Add message to chatroom.
@@ -91,7 +91,7 @@ class ChatroomRepository(
 
         Args:
             chatroom_id: Chatroom ID
-            message: MessageCreate or CompanionMessageCreate schema (no id)
+            message: MessageCreateWithClaim or CompanionMessageCreate schema (with claim_token)
 
         Returns:
             Message or CompanionMessage with id guaranteed to be set

{py_aidol-0.5.0 → py_aidol-0.6.0}/aidol/schemas/__init__.py

@@ -6,6 +6,7 @@ from aidol.schemas.aidol import (
     AIdol,
     AIdolBase,
     AIdolCreate,
+    AIdolCreateWithClaim,
     AIdolPublic,
     AIdolUpdate,
     ImageGenerationData,
@@ -24,6 +25,7 @@ from aidol.schemas.chatroom import (
     Message,
     MessageBase,
     MessageCreate,
+    MessageCreateWithClaim,
     SenderType,
 )
 from aidol.schemas.companion import (
@@ -45,6 +47,7 @@ __all__ = [
     "AIdol",
     "AIdolBase",
     "AIdolCreate",
+    "AIdolCreateWithClaim",
     "AIdolPublic",
     "AIdolUpdate",
     "ImageGenerationData",
@@ -65,6 +68,7 @@ __all__ = [
     "Message",
     "MessageBase",
     "MessageCreate",
+    "MessageCreateWithClaim",
     "SenderType",
     # Companion
     "Companion",

{py_aidol-0.5.0 → py_aidol-0.6.0}/aidol/schemas/aidol.py

@@ -3,7 +3,8 @@ AIdol (group) schemas
 
 Schema hierarchy:
 - AIdolBase: Mutable fields (used in Create/Update)
-- AIdolCreate: Base + optional claim_token
+- AIdolCreate: API body (claim_token excluded, read from Cookie)
+- AIdolCreateWithClaim: Internal (claim_token included, for repository)
 - AIdolUpdate: Base fields only
 - AIdol: Response with all fields including claim_token (internal use)
 - AIdolPublic: Response without sensitive fields (API use)
@@ -32,14 +33,20 @@ class AIdolBase(BaseModel):
 
 
 class AIdolCreate(AIdolBase):
-    """Schema for creating an AIdol group (no id).
+    """API body schema for creating an AIdol group.
 
-    claim_token is required for ownership verification.
+    claim_token is excluded (read from Cookie by router).
     """
 
-    claim_token: str = Field(
-        ...,
-        description="Client-generated UUID for ownership verification",
+
+class AIdolCreateWithClaim(AIdolBase):
+    """Internal schema for repository with claim_token.
+
+    Used by router to pass Cookie-based claim_token to repository.
+    """
+
+    claim_token: str | None = Field(
+        default=None, description="Ownership token from Cookie"
     )
 
 

{py_aidol-0.5.0 → py_aidol-0.6.0}/aidol/schemas/chatroom.py

@@ -82,10 +82,16 @@ class CompanionMessage(Message):
 
 
 class MessageCreate(MessageBase):
-    """Schema for creating a message (no id, no timestamp).
+    """API body schema for creating a message (no id, no timestamp).
 
-    claim_token is used for anonymous user identification and DAU/MAU analytics.
-    It's a UUID stored in localStorage, identifying users without authentication.
+    claim_token is excluded (read from Cookie by router).
+    """
+
+
+class MessageCreateWithClaim(MessageBase):
+    """Internal schema for repository with claim_token.
+
+    Used by router to pass Cookie-based claim_token to repository.
     """
 
     claim_token: str | None = Field(
@@ -93,7 +99,7 @@ class MessageCreate(MessageBase):
     )
 
 
-class CompanionMessageCreate(MessageCreate):
+class CompanionMessageCreate(MessageCreateWithClaim):
     """Schema for creating a companion message.
 
     companion_id is optional for aidol standalone but may be required for platform integration.

{py_aidol-0.5.0 → py_aidol-0.6.0}/pyproject.toml

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