codemie-sdk-python 0.1.134__tar.gz → 0.1.136__tar.gz

{codemie_sdk_python-0.1.134 → codemie_sdk_python-0.1.136}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: codemie-sdk-python
-Version: 0.1.134
+Version: 0.1.136
 Summary: CodeMie SDK for Python
 Author: Vadym Vlasenko
 Author-email: vadym_vlasenko@epam.com
@@ -41,6 +41,8 @@ Python SDK for CodeMie services. This SDK provides a comprehensive interface to
     - [Best Practices](#best-practices)
     - [Error Handling](#error-handling)
     - [Workflow Status Monitoring](#workflow-status-monitoring)
+  - [Conversation Service](#conversation-service)
+    - [Core Methods](#core-methods-4)
 - [Development](#development)
   - [Setup](#setup)
   - [Running Tests](#running-tests)
@@ -703,6 +705,36 @@ def monitor_execution(execution_service, execution_id):
         time.sleep(5)  # Poll every 5 seconds
 ```
 
+### Conversation Service
+
+The Conversation service provides access to manage user conversations within CodeMie Assistants.
+
+#### Core Methods
+
+1. **Get All Conversations**
+```python
+# List all conversations for current user
+conversations = client.conversations.list()
+```
+
+2. **Get Specific Conversation**
+```python
+# Get Conversation by it's ID
+client.conversations.get_conversation("conversation-id")
+```
+
+3. **Get Conversation by Assistant ID**
+```python
+# Get Conversation where Assistant ID is present
+client.conversations.list_by_assistant_id("assistant-id")
+```
+
+4. **Delete Conversation**
+```python
+# Delete specific conversation
+client.conversations.delete("conversation-id")
+```
+
 ## Error Handling
 
 The SDK implements comprehensive error handling. All API calls may raise exceptions for:

{codemie_sdk_python-0.1.134 → codemie_sdk_python-0.1.136}/README.md

@@ -28,6 +28,8 @@ Python SDK for CodeMie services. This SDK provides a comprehensive interface to
     - [Best Practices](#best-practices)
     - [Error Handling](#error-handling)
     - [Workflow Status Monitoring](#workflow-status-monitoring)
+  - [Conversation Service](#conversation-service)
+    - [Core Methods](#core-methods-4)
 - [Development](#development)
   - [Setup](#setup)
   - [Running Tests](#running-tests)
@@ -690,6 +692,36 @@ def monitor_execution(execution_service, execution_id):
         time.sleep(5)  # Poll every 5 seconds
 ```
 
+### Conversation Service
+
+The Conversation service provides access to manage user conversations within CodeMie Assistants.
+
+#### Core Methods
+
+1. **Get All Conversations**
+```python
+# List all conversations for current user
+conversations = client.conversations.list()
+```
+
+2. **Get Specific Conversation**
+```python
+# Get Conversation by it's ID
+client.conversations.get_conversation("conversation-id")
+```
+
+3. **Get Conversation by Assistant ID**
+```python
+# Get Conversation where Assistant ID is present
+client.conversations.list_by_assistant_id("assistant-id")
+```
+
+4. **Delete Conversation**
+```python
+# Delete specific conversation
+client.conversations.delete("conversation-id")
+```
+
 ## Error Handling
 
 The SDK implements comprehensive error handling. All API calls may raise exceptions for:

{codemie_sdk_python-0.1.134 → codemie_sdk_python-0.1.136}/pyproject.toml

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

{codemie_sdk_python-0.1.134 → codemie_sdk_python-0.1.136}/src/codemie_sdk/client/client.py

@@ -4,6 +4,7 @@ from typing import Optional
 
 from ..auth.credentials import KeycloakCredentials
 from ..services.assistant import AssistantService
+from ..services.conversation import ConversationService
 from ..services.datasource import DatasourceService
 from ..services.llm import LLMService
 from ..services.integration import IntegrationService
@@ -76,6 +77,9 @@ class CodeMieClient:
         self.workflows = WorkflowService(
             self._api_domain, self._token, verify_ssl=self._verify_ssl
         )
+        self.conversations = ConversationService(
+            self._api_domain, self._token, verify_ssl=self._verify_ssl
+        )
 
     @property
     def token(self) -> str:
@@ -109,4 +113,7 @@ class CodeMieClient:
         self.workflows = WorkflowService(
             self._api_domain, self._token, verify_ssl=self._verify_ssl
         )
+        self.conversations = ConversationService(
+            self._api_domain, self._token, verify_ssl=self._verify_ssl
+        )
         return self._token

codemie_sdk_python-0.1.136/src/codemie_sdk/models/conversation.py

@@ -0,0 +1,160 @@
+"""Models for conversation-related data structures."""
+
+from datetime import datetime
+from typing import List, Optional, Union
+
+from pydantic import BaseModel, Field
+
+from codemie_sdk.models.assistant import ContextType
+
+
+class Conversation(BaseModel):
+    """
+    Model for conversation summary data as returned from the list endpoint.
+    """
+
+    id: str
+    name: str
+    folder: Optional[str]
+    pinned: bool
+    date: str
+    assistant_ids: List[str]
+    initial_assistant_id: Optional[str]
+
+
+class Mark(BaseModel):
+    """Model for conversation review/mark data."""
+
+    mark: str
+    rating: int
+    comments: str
+    date: datetime
+    operator: Optional["Operator"] = None
+
+
+class Operator(BaseModel):
+    """Represents an operator involved in marking a conversation."""
+
+    user_id: str
+    name: str
+
+
+class Thought(BaseModel):
+    """Model for reasoning or tool-invocation within a message's history."""
+
+    id: str
+    parent_id: Optional[str]
+    metadata: dict
+    in_progress: bool
+    input_text: Optional[str]
+    message: Optional[str]
+    author_type: str
+    author_name: str
+    output_format: str
+    error: Optional[bool]
+    children: List[str]
+
+
+class HistoryMark(BaseModel):
+    """Model for conversation history review/mark data."""
+
+    mark: str
+    rating: int
+    comments: Optional[str]
+    date: datetime
+
+
+class HistoryItem(BaseModel):
+    """Represents an individual message within a conversation's history."""
+
+    role: str
+    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]
+    fileNames: List[str]
+    assistantId: Optional[str]
+    thoughts: Optional[List[Thought]] = Field(default_factory=list)
+
+
+class ContextItem(BaseModel):
+    """Represents contextual settings for conversation."""
+
+    context_type: Optional[ContextType]
+    name: str
+
+
+class ToolItem(BaseModel):
+    """Represents a tool used by an assistant, including configuration and description."""
+
+    name: str
+    label: Optional[str]
+    settings_config: Optional[bool]
+    user_description: Optional[str]
+
+
+class AssistantDataItem(BaseModel):
+    """Model represents details for an assistant included in a conversation."""
+
+    assistant_id: str
+    assistant_name: str
+    assistant_icon: str
+    assistant_type: str
+    context: List[Union[ContextItem, str]]
+    tools: List[ToolItem]
+    conversation_starters: List[str]
+
+
+class ConversationDetailsData(BaseModel):
+    """Extended details about a conversation's configuration and context."""
+
+    llm_model: Optional[str]
+    context: List[ContextItem]
+    app_name: Optional[str]
+    repo_name: Optional[str]
+    index_type: Optional[str]
+
+
+class AssistantDetailsData(BaseModel):
+    """Extended details about an assistant included in a conversation."""
+
+    assistant_id: str
+    assistant_name: str
+    assistant_icon: str
+    assistant_type: str
+    context: List[Union[ContextItem, str]]
+    tools: List[ToolItem]
+    conversation_starters: List[str]
+
+
+class ConversationDetails(BaseModel):
+    """Summary information for a user conversation as returned from list endpoints."""
+
+    id: str
+    date: datetime
+    update_date: datetime
+    conversation_id: str
+    conversation_name: str
+    llm_model: Optional[str]
+    folder: Optional[str]
+    pinned: bool
+    history: List[HistoryItem]
+    user_id: str
+    user_name: str
+    assistant_ids: List[str]
+    assistant_data: List[AssistantDataItem]
+    initial_assistant_id: str
+    final_user_mark: Optional[Mark]
+    final_operator_mark: Optional[Mark]
+    project: str
+    conversation_details: Optional[ConversationDetailsData]
+    assistant_details: Optional[AssistantDetailsData]
+    user_abilities: Optional[List[str]]
+    is_folder_migrated: bool
+    category: Optional[str]

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

@@ -0,0 +1,71 @@
+"""Conversation service implementation."""
+
+from typing import List
+
+from ..models.conversation import Conversation, ConversationDetails
+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 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,
+        )