codemie-sdk-python 0.1.296__tar.gz → 0.1.298__tar.gz

{codemie_sdk_python-0.1.296 → codemie_sdk_python-0.1.298}/PKG-INFO

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

{codemie_sdk_python-0.1.296 → codemie_sdk_python-0.1.298}/pyproject.toml

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

{codemie_sdk_python-0.1.296 → codemie_sdk_python-0.1.298}/src/codemie_sdk/models/conversation.py

@@ -71,15 +71,15 @@ class HistoryItem(BaseModel):
     message: str
     historyIndex: int
     date: datetime
-    responseTime: Optional[float]
-    inputTokens: Optional[int]
-    outputTokens: Optional[int]
-    moneySpent: Optional[float]
-    userMark: Optional[HistoryMark]
-    operatorMark: Optional[HistoryMark]
-    messageRaw: Optional[str]
+    responseTime: Optional[float] = None
+    inputTokens: Optional[int] = None
+    outputTokens: Optional[int] = None
+    moneySpent: Optional[float] = None
+    userMark: Optional[HistoryMark] = None
+    operatorMark: Optional[HistoryMark] = None
+    messageRaw: Optional[str] = None
     fileNames: List[str]
-    assistantId: Optional[str]
+    assistantId: Optional[str] = None
     thoughts: Optional[List[Thought]] = Field(default_factory=list)
 
 
@@ -160,10 +160,65 @@ class ConversationDetails(BaseModel):
     initial_assistant_id: str
     final_user_mark: Optional[Mark]
     final_operator_mark: Optional[Mark]
-    project: str
+    project: Optional[str]
     conversation_details: Optional[ConversationDetailsData]
     assistant_details: Optional[AssistantDetailsData]
     user_abilities: Optional[List[str]]
     is_folder_migrated: bool
     category: Optional[str]
     mcp_server_single_usage: Optional[bool] = False
+
+
+class BaseResponse(BaseModel):
+    """Generic base response model with message field."""
+
+    message: str
+
+
+class UpdateConversationRequest(BaseModel):
+    """Model for updating an existing conversation."""
+
+    name: Optional[str] = None
+    folder: Optional[str] = None
+    pinned: Optional[bool] = None
+
+
+class ConversationFolder(BaseModel):
+    """Model for conversation folder metadata."""
+
+    id: str
+    user_id: str
+    folder_name: str
+    date: str
+    update_date: str
+    user_abilities: List[str]
+
+
+class UpdateConversationFolderRequest(BaseModel):
+    """Model for creating or updating a folder name."""
+
+    folder: str
+
+
+class UpdateHistoryByIndexRequest(BaseModel):
+    """Model for updating message content in conversation history by index."""
+
+    messageIndex: int
+    message: str
+
+
+class UpsertHistoryRequest(BaseModel):
+    """Model for upserting conversation history."""
+
+    history: List[HistoryItem]
+    assistant_id: Optional[str] = None
+    folder: Optional[str] = None
+
+
+class UpsertHistoryResponse(BaseModel):
+    """Response model for conversation history upsert operation."""
+
+    conversation_id: str
+    new_messages: int
+    total_messages: int
+    created: bool

codemie_sdk_python-0.1.298/src/codemie_sdk/services/conversation.py

@@ -0,0 +1,276 @@
+"""Conversation service implementation."""
+
+from typing import List
+
+from ..models.conversation import (
+    Conversation,
+    ConversationDetails,
+    ConversationCreateRequest,
+    UpdateConversationRequest,
+    ConversationFolder,
+    UpdateConversationFolderRequest,
+    UpdateHistoryByIndexRequest,
+    UpsertHistoryRequest,
+    UpsertHistoryResponse,
+    BaseResponse,
+)
+from ..utils import ApiRequestHandler
+
+
+class ConversationService:
+    """Service for managing user conversations."""
+
+    def __init__(self, api_domain: str, token: str, verify_ssl: bool = True):
+        """Initialize the conversation service.
+
+        Args:
+            api_domain: Base URL for the API
+            token: Authentication token
+            verify_ssl: Whether to verify SSL certificates
+        """
+        self._api = ApiRequestHandler(api_domain, token, verify_ssl)
+
+    def list(self) -> List[Conversation]:
+        """Get list of all conversations for the current user.
+
+        Returns:
+            List of all conversations for the current user.
+        """
+        return self._api.get("/v1/conversations", List[Conversation])
+
+    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.
+
+        Args:
+            assistant_id: Assistant ID
+
+        Returns:
+            List of conversations for the specified assistant.
+        """
+        return [
+            conv
+            for conv in self._api.get("/v1/conversations", List[Conversation])
+            if assistant_id in conv.assistant_ids
+        ]
+
+    def get_conversation(self, conversation_id: str) -> ConversationDetails:
+        """Get details for a specific conversation by its ID.
+
+        Args:
+            conversation_id: Conversation ID
+
+        Returns:
+            Conversation details
+        """
+        return self._api.get(
+            f"/v1/conversations/{conversation_id}",
+            ConversationDetails,
+        )
+
+    def create(self, request: ConversationCreateRequest) -> dict:
+        """Create a new conversation.
+
+        Args:
+            request: Conversation creation request
+
+        Returns:
+            Created conversation details
+        """
+        return self._api.post(
+            "/v1/conversations",
+            dict,
+            json_data=request.model_dump(exclude_none=True),
+        )
+
+    def delete(self, conversation_id: str) -> dict:
+        """Delete a specific conversation by its ID.
+
+        Args:
+            conversation_id: Conversation ID to delete
+
+        Returns:
+            Deletion confirmation
+        """
+        return self._api.delete(
+            f"/v1/conversations/{conversation_id}",
+            dict,
+        )
+
+    def update(
+        self, conversation_id: str, request: UpdateConversationRequest
+    ) -> ConversationDetails:
+        """Update an existing conversation.
+
+        Args:
+            conversation_id: Conversation ID to update
+            request: Update request with fields to modify
+
+        Returns:
+            Updated conversation details
+        """
+        return self._api.put(
+            f"/v1/conversations/{conversation_id}",
+            ConversationDetails,
+            json_data=request.model_dump(exclude_none=True),
+        )
+
+    def delete_all(self) -> BaseResponse:
+        """Delete all conversations for the current user.
+
+        Returns:
+            Deletion confirmation
+        """
+        return self._api.delete("/v1/conversations", BaseResponse)
+
+    def get_files(self, conversation_id: str) -> list:
+        """Get list of files attached to a conversation.
+
+        Args:
+            conversation_id: Conversation ID
+
+        Returns:
+            List of file names
+        """
+        response = self._api.get(
+            f"/v1/conversations/{conversation_id}/files",
+            dict,
+            wrap_response=False,
+        )
+        # API returns list directly, not wrapped in 'data' field
+
+        return response
+
+    # Folder management methods
+
+    def list_folders(self) -> List[ConversationFolder]:
+        """Get list of all conversation folders for the current user.
+
+        Returns:
+            List of conversation folders
+        """
+        return self._api.get("/v1/conversations/folders/list", List[ConversationFolder])
+
+    def create_folder(self, request: UpdateConversationFolderRequest) -> BaseResponse:
+        """Create a new conversation folder.
+
+        Args:
+            request: Folder creation request with name
+
+        Returns:
+            Creation confirmation
+        """
+        return self._api.post(
+            "/v1/conversations/folder",
+            BaseResponse,
+            json_data=request.model_dump(exclude_none=True),
+        )
+
+    def update_folder(
+        self, folder: str, request: UpdateConversationFolderRequest
+    ) -> BaseResponse:
+        """Rename an existing conversation folder.
+
+        Args:
+            folder: Current folder name
+            request: Update request with new name
+
+        Returns:
+            Update confirmation
+        """
+        return self._api.put(
+            f"/v1/conversations/folder/{folder}",
+            BaseResponse,
+            json_data=request.model_dump(exclude_none=True),
+        )
+
+    def delete_folder(
+        self, folder: str, remove_conversations: bool = False
+    ) -> BaseResponse:
+        """Delete a conversation folder.
+
+        Args:
+            folder: Folder name to delete
+            remove_conversations: Whether to also delete conversations in the folder
+
+        Returns:
+            Deletion confirmation
+        """
+        endpoint = f"/v1/conversations/folder/{folder}"
+        if remove_conversations:
+            endpoint += "?remove_conversations=true"
+        else:
+            endpoint += "?remove_conversations=false"
+
+        return self._api.delete(endpoint, BaseResponse)
+
+    # History management methods
+
+    def upsert_history(
+        self, conversation_id: str, request: UpsertHistoryRequest
+    ) -> UpsertHistoryResponse:
+        """Create or update conversation with history (idempotent upsert).
+
+        If conversation doesn't exist, creates it with custom ID and provided history.
+        If conversation exists, appends only NEW messages not already present.
+
+        Args:
+            conversation_id: Conversation ID (will be created if doesn't exist)
+            request: History upsert request with messages
+
+        Returns:
+            Upsert response with metadata about operation
+        """
+        return self._api.put(
+            f"/v1/conversations/{conversation_id}/history",
+            UpsertHistoryResponse,
+            json_data=request.model_dump(mode="json", exclude_none=True),
+        )
+
+    def delete_history(self, conversation_id: str) -> ConversationDetails:
+        """Clear all history from a conversation.
+
+        Args:
+            conversation_id: Conversation ID
+
+        Returns:
+            Updated conversation details with empty history
+        """
+        return self._api.delete(
+            f"/v1/conversations/{conversation_id}/history",
+            ConversationDetails,
+        )
+
+    def delete_history_by_index(
+        self, conversation_id: str, history_index: int
+    ) -> ConversationDetails:
+        """Remove a specific history item by index.
+
+        Args:
+            conversation_id: Conversation ID
+            history_index: Index of history item to delete
+
+        Returns:
+            Updated conversation details
+        """
+        return self._api.delete(
+            f"/v1/conversations/{conversation_id}/history/{history_index}",
+            ConversationDetails,
+        )
+
+    def update_history_by_index(
+        self, conversation_id: str, request: UpdateHistoryByIndexRequest
+    ) -> ConversationDetails:
+        """Update an Assistant response in conversation history by index.
+
+        Args:
+            conversation_id: Conversation ID
+            request: Update request with message index and new message content
+
+        Returns:
+            Updated conversation details
+        """
+        return self._api.put(
+            f"/v1/conversations/{conversation_id}/history/{request.messageIndex}",
+            ConversationDetails,
+            json_data=request.model_dump(exclude_none=True),
+        )

codemie_sdk_python-0.1.296/src/codemie_sdk/services/conversation.py

@@ -1,90 +0,0 @@
-"""Conversation service implementation."""
-
-from typing import List
-
-from ..models.conversation import (
-    Conversation,
-    ConversationDetails,
-    ConversationCreateRequest,
-)
-from ..utils import ApiRequestHandler
-
-
-class ConversationService:
-    """Service for managing user conversations."""
-
-    def __init__(self, api_domain: str, token: str, verify_ssl: bool = True):
-        """Initialize the conversation service.
-
-        Args:
-            api_domain: Base URL for the API
-            token: Authentication token
-            verify_ssl: Whether to verify SSL certificates
-        """
-        self._api = ApiRequestHandler(api_domain, token, verify_ssl)
-
-    def list(self) -> List[Conversation]:
-        """Get list of all conversations for the current user.
-
-        Returns:
-            List of all conversations for the current user.
-        """
-        return self._api.get("/v1/conversations", List[Conversation])
-
-    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.
-
-        Args:
-            assistant_id: Assistant ID
-
-        Returns:
-            List of conversations for the specified assistant.
-        """
-        return [
-            conv
-            for conv in self._api.get("/v1/conversations", List[Conversation])
-            if assistant_id in conv.assistant_ids
-        ]
-
-    def get_conversation(self, conversation_id: str) -> ConversationDetails:
-        """Get details for a specific conversation by its ID.
-
-        Args:
-            conversation_id: Conversation ID
-
-        Returns:
-            Conversation details
-        """
-        return self._api.get(
-            f"/v1/conversations/{conversation_id}",
-            ConversationDetails,
-        )
-
-    def create(self, request: ConversationCreateRequest) -> dict:
-        """Create a new conversation.
-
-        Args:
-            request: Conversation creation request
-
-        Returns:
-            Created conversation details
-        """
-        return self._api.post(
-            "/v1/conversations",
-            dict,
-            json_data=request.model_dump(exclude_none=True),
-        )
-
-    def delete(self, conversation_id: str) -> dict:
-        """Delete a specific conversation by its ID.
-
-        Args:
-            conversation_id: Conversation ID to delete
-
-        Returns:
-            Deletion confirmation
-        """
-        return self._api.delete(
-            f"/v1/conversations/{conversation_id}",
-            dict,
-        )