codemie-sdk-python 0.1.5__tar.gz → 0.1.7__tar.gz

{codemie_sdk_python-0.1.5 → codemie_sdk_python-0.1.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: codemie-sdk-python
-Version: 0.1.5
+Version: 0.1.7
 Summary: CodeMie SDK for Python
 Author: Vadym Vlasenko
 Author-email: vadym_vlasenko@epam.com
@@ -169,7 +169,7 @@ result = client.assistant.delete("assistant-id")
 from codemie_sdk.models.assistant import AssistantChatRequest
 
 chat_request = AssistantChatRequest(
-    message="Your message here",
+    text="Your message here",
     stream=False,  # Set to True for streaming response
     # Additional parameters
 )
@@ -525,6 +525,37 @@ result = execution_service.resume("execution-id")
 result = execution_service.delete_all()
 ```
 
+3. **Work with Execution States**
+```python
+# Get execution states
+states = execution_service.states(execution_id).list()
+
+# Get state output
+state_output = execution_service.states(execution_id).get_output(state_id)
+
+# Example of monitoring workflow with state verification
+def verify_workflow_execution(execution_service, execution_id):
+    execution = execution_service.get(execution_id)
+    
+    if execution.status == ExecutionStatus.SUCCEEDED:
+        # Get and verify states
+        states = execution_service.states(execution_id).list()
+        
+        # States are ordered by completion date
+        if len(states) >= 2:
+            first_state = states[0]
+            second_state = states[1]
+            assert first_state.completed_at < second_state.completed_at
+            
+            # Get state outputs
+            for state in states:
+                output = execution_service.states(execution_id).get_output(state.id)
+                print(f"State {state.id} output: {output.output}")
+    
+    elif execution.status == ExecutionStatus.FAILED:
+        print(f"Workflow failed: {execution.error_message}")
+```
+
 #### Workflow Configuration
 
 Workflows support various configuration options:
@@ -707,3 +738,65 @@ client = CodeMieClient(
 ## Support
 For providing credentials please contact AI/Run CodeMie Team: Vadym_Vlasenko@epam.com or Nikita_Levyankov@epam.com
 
+## Running tests
+
+For running tests on custom environment you should create .env file in the root directory
+
+``` properties
+
+AUTH_SERVER_URL=https://keycloak.eks-core.aws.main.edp.projects.epam.com/auth
+AUTH_CLIENT_ID=codemie-preview
+AUTH_CLIENT_SECRET=<your_secret>
+AUTH_REALM_NAME=codemie-prod
+CODEMIE_API_DOMAIN=https://codemie-preview.lab.epam.com/code-assistant-api
+VERIFY_SSL=False
+
+USERNAME=<username>
+PASSWORD=<password>
+
+PROJECT_NAME=automation-tests-project
+
+DEFAULT_TIMEOUT=60
+
+GITLAB_URL=https://gitbud.epam.com
+GITLAB_TOKEN=<gitlab_token>
+GITLAB_PROJECT=https://gitbud.epam.com/epm-cdme/autotests/codemie-test-project
+GITLAB_PROJECT_ID=<project_id>
+
+GITHUB_URL=https://github.com
+GITHUB_TOKEN=<github_token>
+GITHUB_PROJECT=https://github.com/wild47/final_task
+
+JIRA_URL=https://jiraeu.epam.com
+JIRA_TOKEN=<jira_token>
+JQL="project = 'EPMCDME' and issuetype = 'Epic' and status = 'Closed'"
+
+CONFLUENCE_URL=https://kb.epam.com
+CONFLUENCE_TOKEN=<konfluence_token>
+CQL="space = EPMCDME and type = page and title = 'AQA Backlog Estimation'"
+```
+
+Run all tests
+
+```shell
+pytest -n auto --reruns 1
+```
+
+Run e2e tests
+
+```shell
+pytest -n auto -m e2e --reruns 1
+```
+
+Run tests for e2e tests for specific integration/tool.
+Available marks:
+ - jira_kb
+ - confluence_kb
+ - code_kb
+ - gitlab
+ - github
+ - git
+
+```shell
+pytest -n auto -m "jira_kb or github" --reruns 1
+```

{codemie_sdk_python-0.1.5 → codemie_sdk_python-0.1.7}/README.md

@@ -156,7 +156,7 @@ result = client.assistant.delete("assistant-id")
 from codemie_sdk.models.assistant import AssistantChatRequest
 
 chat_request = AssistantChatRequest(
-    message="Your message here",
+    text="Your message here",
     stream=False,  # Set to True for streaming response
     # Additional parameters
 )
@@ -512,6 +512,37 @@ result = execution_service.resume("execution-id")
 result = execution_service.delete_all()
 ```
 
+3. **Work with Execution States**
+```python
+# Get execution states
+states = execution_service.states(execution_id).list()
+
+# Get state output
+state_output = execution_service.states(execution_id).get_output(state_id)
+
+# Example of monitoring workflow with state verification
+def verify_workflow_execution(execution_service, execution_id):
+    execution = execution_service.get(execution_id)
+    
+    if execution.status == ExecutionStatus.SUCCEEDED:
+        # Get and verify states
+        states = execution_service.states(execution_id).list()
+        
+        # States are ordered by completion date
+        if len(states) >= 2:
+            first_state = states[0]
+            second_state = states[1]
+            assert first_state.completed_at < second_state.completed_at
+            
+            # Get state outputs
+            for state in states:
+                output = execution_service.states(execution_id).get_output(state.id)
+                print(f"State {state.id} output: {output.output}")
+    
+    elif execution.status == ExecutionStatus.FAILED:
+        print(f"Workflow failed: {execution.error_message}")
+```
+
 #### Workflow Configuration
 
 Workflows support various configuration options:
@@ -693,3 +724,66 @@ client = CodeMieClient(
 
 ## Support
 For providing credentials please contact AI/Run CodeMie Team: Vadym_Vlasenko@epam.com or Nikita_Levyankov@epam.com
+
+## Running tests
+
+For running tests on custom environment you should create .env file in the root directory
+
+``` properties
+
+AUTH_SERVER_URL=https://keycloak.eks-core.aws.main.edp.projects.epam.com/auth
+AUTH_CLIENT_ID=codemie-preview
+AUTH_CLIENT_SECRET=<your_secret>
+AUTH_REALM_NAME=codemie-prod
+CODEMIE_API_DOMAIN=https://codemie-preview.lab.epam.com/code-assistant-api
+VERIFY_SSL=False
+
+USERNAME=<username>
+PASSWORD=<password>
+
+PROJECT_NAME=automation-tests-project
+
+DEFAULT_TIMEOUT=60
+
+GITLAB_URL=https://gitbud.epam.com
+GITLAB_TOKEN=<gitlab_token>
+GITLAB_PROJECT=https://gitbud.epam.com/epm-cdme/autotests/codemie-test-project
+GITLAB_PROJECT_ID=<project_id>
+
+GITHUB_URL=https://github.com
+GITHUB_TOKEN=<github_token>
+GITHUB_PROJECT=https://github.com/wild47/final_task
+
+JIRA_URL=https://jiraeu.epam.com
+JIRA_TOKEN=<jira_token>
+JQL="project = 'EPMCDME' and issuetype = 'Epic' and status = 'Closed'"
+
+CONFLUENCE_URL=https://kb.epam.com
+CONFLUENCE_TOKEN=<konfluence_token>
+CQL="space = EPMCDME and type = page and title = 'AQA Backlog Estimation'"
+```
+
+Run all tests
+
+```shell
+pytest -n auto --reruns 1
+```
+
+Run e2e tests
+
+```shell
+pytest -n auto -m e2e --reruns 1
+```
+
+Run tests for e2e tests for specific integration/tool.
+Available marks:
+ - jira_kb
+ - confluence_kb
+ - code_kb
+ - gitlab
+ - github
+ - git
+
+```shell
+pytest -n auto -m "jira_kb or github" --reruns 1
+```

{codemie_sdk_python-0.1.5 → codemie_sdk_python-0.1.7}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "codemie-sdk-python"
-version = "0.1.5"
+version = "0.1.7"
 description = "CodeMie SDK for Python"
 authors = [
     "Vadym Vlasenko <vadym_vlasenko@epam.com>",
@@ -19,6 +19,15 @@ pydantic = "^2.11.1"
 ruff = "^0.11.3"
 pytest-cov = "^6.0.0"
 
+[tool.poetry.group.test.dependencies]
+python-dotenv = "^1.1.0"
+PyHamcrest = "^2.1.0"
+pytest-xdist = "^3.6.1"
+pytest-rerunfailures = "^15.1"
+python-gitlab = "^5.6.0"
+boto3 = "^1.38.20"
+aws-assume-role-lib = "^2.10.0"
+
 [build-system]
 requires = ["poetry-core"]
 build-backend = "poetry.core.masonry.api"

{codemie_sdk_python-0.1.5 → codemie_sdk_python-0.1.7}/src/codemie_sdk/client/client.py

@@ -51,6 +51,11 @@ class CodeMieClient:
         self._token: Optional[str] = None
         self._api_domain = codemie_api_domain.rstrip("/")
         self._verify_ssl = verify_ssl
+        if not verify_ssl:
+            import requests
+            from urllib3.exceptions import InsecureRequestWarning
+
+            requests.packages.urllib3.disable_warnings(InsecureRequestWarning)
 
         # Initialize token first
         self._token = self.auth.get_token()

{codemie_sdk_python-0.1.5 → codemie_sdk_python-0.1.7}/src/codemie_sdk/models/assistant.py

@@ -11,6 +11,54 @@ from .common import User
 from .integration import Integration
 
 
+class GitToolName(str, Enum):
+    """Enum for Git tool names."""
+
+    LIST_BRANCHES_IN_REPO = "list_branches_in_repo"
+    CREATE_BRANCH = "create_branch"
+    SET_ACTIVE_BRANCH = "set_active_branch"
+    CREATE_FILE = "create_file"
+    UPDATE_FILE = "update_file"
+    UPDATE_FILE_DIFF = "update_file_diff"
+    DELETE_FILE = "delete_file"
+    CREATE_PULL_REQUEST = "create_pull_request"
+    GET_PR_CHANGES = "get_pr_changes"
+    CREATE_PR_CHANGES_COMMENT = "create_pr_changes_comment"
+
+
+class CodeBaseToolName(str, Enum):
+    """Enum for Git tool names."""
+
+    SONAR = "Sonar"
+    SONAR_CLOUD = "Sonar"
+    GET_REPOSITORY_FILE_TREE_V2 = "get_repository_file_tree_v2"
+    SEARCH_CODE_REPO_V2 = "search_code_repo_v2"
+    READ_FILES_CONTENT = "read_files_content"
+    READ_FILES_CONTENT_SUMMARY = "read_files_content_summary"
+    SEARCH_CODE_REPO_BY_PATH = "search_code_repo_by_path"
+
+
+class VcsToolName(str, Enum):
+    """Enum for VCS tool names."""
+
+    GITLAB = "gitlab"
+    GITHUB = "github"
+
+
+class CloudToolName(str, Enum):
+    """Enum for VCS tool names."""
+
+    AWS = "AWS"
+    GCP = "GCP"
+    AZURE = "Azure"
+
+
+class PluginToolName(str, Enum):
+    """Enum for VCS tool names."""
+
+    PLUGIN = "Plugin"
+
+
 class ToolDetails(BaseModel):
     """Model for tool details."""
 
@@ -23,6 +71,16 @@ class ToolDetails(BaseModel):
     settings: Optional[Integration] = None
 
 
+class Toolkit(str, Enum):
+    """Enum for toolkits."""
+
+    GIT = "Git"
+    VCS = "VCS"
+    CODEBASE_TOOLS = "Codebase Tools"
+    CLOUD = "Cloud"
+    PLUGIN = "Plugin"
+
+
 class ToolKitDetails(BaseModel):
     """Model for toolkit details."""
 
@@ -149,7 +207,7 @@ class AssistantChatRequest(BaseModel):
     conversation_id: Optional[str] = Field(
         default=str(uuid.uuid4()), description="Conversation identifier"
     )
-    text: Optional[str] = Field(default=None, description="User's input error_message")
+    text: str = Field(description="User's input")
     content_raw: Optional[str] = Field(default="", description="Raw content input")
     file_name: Optional[str] = Field(default=None, description="Associated file name")
     llm_model: Optional[str] = Field(
@@ -166,6 +224,9 @@ class AssistantChatRequest(BaseModel):
     top_k: int = Field(default=10, description="Top K results to consider")
     system_prompt: str = Field(default="", description="Override system prompt")
     background_task: bool = Field(default=False, description="Run as background task")
+    metadata: Optional[dict[str, Any]] = Field(
+        default=None, description="Provide additional metadata"
+    )
 
 
 class BaseModelResponse(BaseModel):

{codemie_sdk_python-0.1.5 → codemie_sdk_python-0.1.7}/src/codemie_sdk/models/datasource.py

@@ -1,7 +1,7 @@
 import re
 from datetime import datetime
 from enum import Enum
-from typing import Optional, List
+from typing import Optional, List, Union
 
 from pydantic import BaseModel, Field, model_validator, ConfigDict, field_validator
 
@@ -20,6 +20,10 @@ class DataSourceType(str, Enum):
     JIRA = "knowledge_base_jira"
     FILE = "knowledge_base_file"
     GOOGLE = "llm_routing_google"
+    PROVIDER = "provider"
+    SUMMARY = "summary"
+    CHUNK_SUMMARY = "chunk-summary"
+    JSON = "knowledge_base_json"
 
 
 class DataSourceStatus(str, Enum):
@@ -37,7 +41,7 @@ class DataSourceProcessingInfo(BaseModel):
     total_size_kb: Optional[float] = None
     average_file_size_bytes: Optional[float] = None
     unique_extensions: Optional[List[str]] = None
-    filtered_documents: Optional[int] = None
+    filtered_documents: Optional[Union[int, list]] = None
     processed_documents_count: Optional[int] = Field(None, alias="documents_count_key")
 
 

{codemie_sdk_python-0.1.5 → codemie_sdk_python-0.1.7}/src/codemie_sdk/models/integration.py

@@ -1,10 +1,10 @@
 """Models for assistant-related data structures."""
 
-from datetime import datetime
 from enum import Enum
 from typing import List, Optional, Any
+from datetime import datetime
 
-from pydantic import BaseModel, Field, ConfigDict
+from pydantic import BaseModel, Field, ConfigDict, field_serializer
 
 
 class CredentialTypes(str, Enum):
@@ -33,6 +33,8 @@ class CredentialTypes(str, Enum):
     ZEPHYR_SQUAD = "ZephyrSquad"
     SERVICE_NOW = "ServiceNow"
     DIAL = "DIAL"
+    A2A = "A2A"
+    MCP = "MCP"
 
 
 class IntegrationType(str, Enum):
@@ -66,3 +68,7 @@ class Integration(BaseModel):
     credential_type: CredentialTypes
     credential_values: List[CredentialValues]
     setting_type: IntegrationType = Field(default=IntegrationType.USER)
+
+    @field_serializer("date", "update_date")
+    def serialize_dt(self, dt: datetime, _info):
+        return dt.isoformat()

codemie_sdk_python-0.1.7/src/codemie_sdk/models/workflow_state.py

@@ -0,0 +1,51 @@
+"""Workflow execution state models."""
+
+from datetime import datetime
+from enum import Enum
+from typing import Optional
+
+from pydantic import BaseModel, ConfigDict
+
+
+class WorkflowExecutionStatusEnum(str, Enum):
+    """Workflow execution state status."""
+
+    IN_PROGRESS = "In Progress"
+    NOT_STARTED = "Not Started"
+    INTERRUPTED = "Interrupted"
+    FAILED = "Failed"
+    SUCCEEDED = "Succeeded"
+    ABORTED = "Aborted"
+
+
+class WorkflowExecutionStateThought(BaseModel):
+    """Model for workflow execution state thought."""
+
+    model_config = ConfigDict(populate_by_name=True)
+
+    id: str
+    text: str
+    created_at: datetime
+    parent_id: Optional[str] = None
+
+
+class WorkflowExecutionState(BaseModel):
+    """Model for workflow execution state."""
+
+    model_config = ConfigDict(populate_by_name=True)
+
+    id: str
+    execution_id: str
+    name: str
+    task: Optional[str] = ""
+    status: WorkflowExecutionStatusEnum = WorkflowExecutionStatusEnum.NOT_STARTED
+    started_at: Optional[datetime] = None
+    completed_at: Optional[datetime] = None
+
+
+class WorkflowExecutionStateOutput(BaseModel):
+    """Model for workflow execution state output."""
+
+    model_config = ConfigDict(populate_by_name=True)
+
+    output: Optional[str] = None

{codemie_sdk_python-0.1.5 → codemie_sdk_python-0.1.7}/src/codemie_sdk/services/workflow_execution.py

@@ -4,6 +4,7 @@ from typing import List, Optional
 
 from ..models.common import PaginationParams
 from ..models.workflow import WorkflowExecution
+from .workflow_execution_state import WorkflowExecutionStateService
 from ..utils import ApiRequestHandler
 
 
@@ -71,6 +72,17 @@ class WorkflowExecutionService:
             WorkflowExecution,
         )
 
+    def states(self, execution_id: str) -> WorkflowExecutionStateService:
+        """Get states service for a specific workflow execution.
+
+        Args:
+            execution_id: ID of the execution to get states for
+
+        Returns:
+            WorkflowExecutionStateService: Service for managing states of this execution
+        """
+        return WorkflowExecutionStateService(self._api, self._workflow_id, execution_id)
+
     def delete_all(self) -> dict:
         """Delete all workflow executions."""
         return self._api.delete(f"/v1/workflows/{self._workflow_id}/executions", dict)

codemie_sdk_python-0.1.7/src/codemie_sdk/services/workflow_execution_state.py

@@ -0,0 +1,61 @@
+"""Workflow execution state service implementation."""
+
+from typing import List
+
+from ..models.common import PaginationParams
+from ..models.workflow_state import (
+    WorkflowExecutionState,
+    WorkflowExecutionStateOutput,
+)
+from ..utils import ApiRequestHandler
+
+
+class WorkflowExecutionStateService:
+    """Service for managing workflow execution states."""
+
+    def __init__(self, api: ApiRequestHandler, workflow_id: str, execution_id: str):
+        """Initialize the workflow execution state service.
+
+        Args:
+            api: Request handler for API
+            workflow_id: ID of the workflow this service manages states for
+            execution_id: ID of the execution this service manages states for
+        """
+        self._api = api
+        self._workflow_id = workflow_id
+        self._execution_id = execution_id
+
+    def list(
+        self,
+        page: int = 0,
+        per_page: int = 10,
+    ) -> List[WorkflowExecutionState]:
+        """List states for the workflow execution with filtering and pagination support.
+
+        Args:
+            page: Page number (0-based). Must be >= 0. Defaults to 0.
+            per_page: Number of items per page. Must be > 0. Defaults to 10.
+
+        Returns:
+            List of WorkflowExecutionState containing state information.
+        """
+        params = PaginationParams(page=page, per_page=per_page).to_dict()
+        return self._api.get(
+            f"/v1/workflows/{self._workflow_id}/executions/{self._execution_id}/states",
+            List[WorkflowExecutionState],
+            params=params,
+        )
+
+    def get_output(self, state_id: str) -> WorkflowExecutionStateOutput:
+        """Get output for a specific execution state.
+
+        Args:
+            state_id: ID of the state to get output for
+
+        Returns:
+            WorkflowExecutionStateOutput containing the state output.
+        """
+        return self._api.get(
+            f"/v1/workflows/{self._workflow_id}/executions/{self._execution_id}/states/{state_id}/output",
+            WorkflowExecutionStateOutput,
+        )

{codemie_sdk_python-0.1.5 → codemie_sdk_python-0.1.7}/src/codemie_sdk/utils/http.py

@@ -48,10 +48,16 @@ class ApiRequestHandler:
 
     def _get_headers(self) -> dict:
         """Gets request headers with auth token."""
-        return {
-            "Authorization": f"Bearer {self._token}",
-            "Content-Type": "application/json",
-        }
+        headers = {"Content-Type": "application/json"}
+        if (
+            "0.0.0.0" in self._base_url
+            or "127.0.0.1" in self._base_url
+            or "localhost" in self._base_url
+        ):
+            headers["User-Id"] = "dev-codemie-user"
+        else:
+            headers["Authorization"] = f"Bearer {self._token}"
+        return headers
 
     def _parse_response(
         self,