codemie-sdk-python 0.1.611__tar.gz → 0.1.612__tar.gz

{codemie_sdk_python-0.1.611 → codemie_sdk_python-0.1.612}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: codemie-sdk-python
-Version: 0.1.611
+Version: 0.1.612
 Summary: CodeMie SDK for Python
 Author: Vadym Vlasenko
 Author-email: vadym_vlasenko@epam.com

{codemie_sdk_python-0.1.611 → codemie_sdk_python-0.1.612}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "codemie-sdk-python"
-version = "0.1.611"
+version = "0.1.612"
 description = "CodeMie SDK for Python"
 authors = [
     "Vadym Vlasenko <vadym_vlasenko@epam.com>",

{codemie_sdk_python-0.1.611 → codemie_sdk_python-0.1.612}/src/codemie_sdk/models/conversation.py

@@ -4,7 +4,7 @@ from __future__ import annotations
 
 from datetime import datetime
 
-from pydantic import BaseModel, Field
+from pydantic import BaseModel, ConfigDict, Field
 
 from codemie_sdk.models.assistant import ContextType
 
@@ -14,6 +14,8 @@ class Conversation(BaseModel):
     Model for conversation summary data as returned from the list endpoint.
     """
 
+    model_config = ConfigDict(extra="ignore")
+
     id: str
     name: str
     folder: str | None
@@ -21,6 +23,7 @@ class Conversation(BaseModel):
     date: str
     assistant_ids: list[str]
     initial_assistant_id: str | None
+    finished_at: datetime | None = None
 
 
 class Mark(BaseModel):
@@ -177,6 +180,7 @@ class ConversationDetails(BaseModel):
     is_workflow_conversation: bool | None = None
     category: str | None
     mcp_server_single_usage: bool | None = False
+    finished_at: datetime | None = None
 
 
 class ConversationShareRequest(BaseModel):
@@ -256,3 +260,58 @@ class UpsertHistoryResponse(BaseModel):
     new_messages: int
     total_messages: int
     created: bool
+
+
+class ConversationFinishResponse(BaseModel):
+    """Response for POST /conversations/{conversation_id}/finish (owner-only, not idempotent).
+
+    Already-finished conversations cause the server to raise HTTP 409 rather
+    than return this response again.
+    """
+
+    model_config = ConfigDict(populate_by_name=True)
+
+    conversation_id: str = Field(alias="conversationId")
+    finished_at: datetime = Field(alias="finishedAt")
+
+
+class ConversationFinishResult(BaseModel):
+    """Per-conversation-ID result entry within a bulk-finish response."""
+
+    model_config = ConfigDict(populate_by_name=True)
+
+    conversation_id: str = Field(alias="conversationId")
+    already_finished: bool = Field(default=False, alias="alreadyFinished")
+    finished_at: datetime | None = Field(default=None, alias="finishedAt")
+    error: str | None = None
+
+
+class ConversationFinishBulkRequest(BaseModel):
+    """Request body for POST /admin/conversations/finish-bulk."""
+
+    model_config = ConfigDict(populate_by_name=True)
+
+    conversation_ids: list[str] = Field(alias="conversationIds")
+
+
+class ConversationFinishBulkResponse(BaseModel):
+    """Response for POST /admin/conversations/finish-bulk (admin-only, idempotent per ID)."""
+
+    total: int
+    results: list[ConversationFinishResult]
+
+
+class ConversationPagination(BaseModel):
+    """Pagination metadata for the admin conversation list endpoint."""
+
+    page: int
+    per_page: int
+    total: int
+    pages: int
+
+
+class AdminConversationListResponse(BaseModel):
+    """Paginated response for GET /admin/conversations (admin-only, cross-user)."""
+
+    data: list[Conversation]
+    pagination: ConversationPagination

{codemie_sdk_python-0.1.611 → codemie_sdk_python-0.1.612}/src/codemie_sdk/services/conversation.py

@@ -5,9 +5,13 @@ from __future__ import annotations
 import requests
 
 from ..models.conversation import (
+    AdminConversationListResponse,
     Conversation,
     ConversationDetails,
     ConversationCreateRequest,
+    ConversationFinishBulkRequest,
+    ConversationFinishBulkResponse,
+    ConversationFinishResponse,
     UpdateConversationRequest,
     ConversationFolder,
     UpdateConversationFolderRequest,
@@ -36,13 +40,19 @@ class ConversationService:
         """
         self._api = ApiRequestHandler(api_domain, token, verify_ssl)
 
-    def list(self) -> list[Conversation]:
+    def list(self, is_finished: bool | None = None) -> list[Conversation]:
         """Get list of all conversations for the current user.
 
+        Args:
+            is_finished: Optional filter narrowing the list to finished (True)
+                or unfinished (False) conversations only. Maps to the
+                ``isFinished`` query parameter.
+
         Returns:
             List of all conversations for the current user.
         """
-        return self._api.get("/v1/conversations", list[Conversation])
+        params = {"isFinished": is_finished} if is_finished is not None else None
+        return self._api.get("/v1/conversations", list[Conversation], params=params)
 
     def list_by_assistant_id(self, assistant_id: str) -> list[Conversation]:
         """Get list of all conversations for the current user that include the specified assistant.
@@ -338,3 +348,79 @@ class ConversationService:
             f"/v1/share/conversations/{token}",
             SharedConversationResponse,
         )
+
+    # Conversation finishing (EPMCDME-11067)
+
+    def finish(self, conversation_id: str) -> ConversationFinishResponse:
+        """Mark a conversation as finished (owner-only).
+
+        Not idempotent: a repeat call on an already-finished conversation
+        raises HTTP 409 rather than returning success again.
+
+        Args:
+            conversation_id: Conversation ID to finish
+
+        Returns:
+            ConversationFinishResponse with the conversation ID and timestamp
+        """
+        return self._api.post(
+            f"/v1/conversations/{conversation_id}/finish",
+            ConversationFinishResponse,
+        )
+
+    def finish_bulk(
+        self, conversation_ids: list[str]
+    ) -> ConversationFinishBulkResponse:
+        """Finish multiple conversations in one request (admin-only).
+
+        Idempotent per ID: already-finished IDs are reported with
+        already_finished=True instead of erroring. Unknown IDs are reported
+        with error="not_found".
+
+        Args:
+            conversation_ids: List of conversation IDs to finish (1-500)
+
+        Returns:
+            ConversationFinishBulkResponse with per-ID results
+        """
+        request = ConversationFinishBulkRequest(conversation_ids=conversation_ids)
+        return self._api.post(
+            "/v1/admin/conversations/finish-bulk",
+            ConversationFinishBulkResponse,
+            json_data=request.model_dump(by_alias=True),
+        )
+
+    def list_admin(
+        self,
+        is_finished: bool | None = None,
+        started_after: str | None = None,
+        project: str | None = None,
+        page: int = 0,
+        per_page: int = 20,
+    ) -> AdminConversationListResponse:
+        """Paginated, cross-user conversation list for admin/scheduler use (admin-only).
+
+        Args:
+            is_finished: Optional filter (maps to ``isFinished``)
+            started_after: Optional ISO datetime lower bound (maps to ``startedAt``)
+            project: Optional project ID to scope the listing to one application
+            page: Page number (0-based)
+            per_page: Number of items per page (maps to ``perPage``)
+
+        Returns:
+            AdminConversationListResponse with items and pagination metadata
+        """
+        params: dict = {"page": page, "perPage": per_page}
+        if is_finished is not None:
+            params["isFinished"] = is_finished
+        if started_after is not None:
+            params["startedAt"] = started_after
+        if project is not None:
+            params["project"] = project
+
+        return self._api.get(
+            "/v1/admin/conversations",
+            AdminConversationListResponse,
+            params=params,
+            wrap_response=False,
+        )