codemie-sdk-python 0.1.5__py3-none-any.whl → 0.1.6__py3-none-any.whl

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/models/assistant.py

@@ -11,6 +11,28 @@ 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 VcsToolName(str, Enum):
+    """Enum for VCS tool names."""
+
+    GITLAB = "gitlab"
+    GITHUB = "github"
+
+
 class ToolDetails(BaseModel):
     """Model for tool details."""
 
@@ -23,6 +45,13 @@ class ToolDetails(BaseModel):
     settings: Optional[Integration] = None
 
 
+class Toolkit(str, Enum):
+    """Enum for toolkits."""
+
+    GIT = "Git"
+    VCS = "VCS"
+
+
 class ToolKitDetails(BaseModel):
     """Model for toolkit details."""
 
@@ -149,7 +178,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 +195,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/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/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/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/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/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/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,

{codemie_sdk_python-0.1.5.dist-info → codemie_sdk_python-0.1.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: codemie-sdk-python
-Version: 0.1.5
+Version: 0.1.6
 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.dist-info → codemie_sdk_python-0.1.6.dist-info}/RECORD

@@ -2,16 +2,17 @@ codemie_sdk/__init__.py,sha256=xfhrTFdvRmLSsrdQksX5GhnoQEj6tvkeytkrbB20Jj8,567
 codemie_sdk/auth/__init__.py,sha256=IksEj223xEZtJ-cQ0AT9L0Bs9psIJ8QNzDXrPTUQ3xQ,126
 codemie_sdk/auth/credentials.py,sha256=u2eLNsD8fELTgreQghVb0g0kk82zchUkux0M5lzq-u8,4320
 codemie_sdk/client/__init__.py,sha256=yf6C39MmrJ6gK9ZHMhBeynKwUUYVSUTQbKxU8-4qpKg,101
-codemie_sdk/client/client.py,sha256=aU4yr_Un4gX7SLXZ7rQiO_ObxE2__RyR8OdlxeKLRvc,4172
+codemie_sdk/client/client.py,sha256=dVcWqd-ruWd8GOZ3_33C3LnHgUyyxIoCKxKYPYbBPq8,4373
 codemie_sdk/exceptions.py,sha256=XoVPyognx-JmyVxLHkZPAcX1CMi1OoT1diBFJLU54so,1183
-codemie_sdk/models/assistant.py,sha256=hx3R1H7UPqRvmBFKL7PDwHdVcX_CbjUsju55tjVRVEo,5538
+codemie_sdk/models/assistant.py,sha256=5VT5DOUPsEjVTLEPQoxFw5yna0BDoJBq0D8daVIp7l4,6304
 codemie_sdk/models/common.py,sha256=V4sJCzwFTF8BPlrd2OKI44Em5HmPmn2Nm8fQNsnBZ_Q,1128
-codemie_sdk/models/datasource.py,sha256=Kjde8pXi4pm6x37g-c838C3yDg4tnTJHNvITA3rvmUk,9898
-codemie_sdk/models/integration.py,sha256=lhY_5Dxm0MDxp9vxL0H0HaM-ytNNoyBSOoEYTw4a-rU,1589
+codemie_sdk/models/datasource.py,sha256=Iun6O76o4m-_jbyyumMndpyr4OiN0lwqKb8u5AOTHdM,10037
+codemie_sdk/models/integration.py,sha256=F1YBdCK1w5W_jKK8prEsg7A00NghkEBvJgkkr8KpC6w,1764
 codemie_sdk/models/llm.py,sha256=ppb9-1dx1UFhRuJpSR3ij7H6Pfhe9nO4C4BEOIbToy4,1192
 codemie_sdk/models/task.py,sha256=J4ZFRY3s8qBGrqB5NLQF0rMbInLh4s7OEZ0ZfmnW0Ho,1476
 codemie_sdk/models/user.py,sha256=Q0rjimZh-IbeaPfq6b6fk6ZaCtwLqWHEIlU863suCS4,1777
 codemie_sdk/models/workflow.py,sha256=CyK-kBnOck2KiRX3gmWF_mcGtqxsGXH3FS3jwYbpz_A,2327
+codemie_sdk/models/workflow_state.py,sha256=CMYFQZ7sy4QxmnWmc83TFfqP7TG_3rW5MdH5fxsS9kY,1251
 codemie_sdk/services/assistant.py,sha256=-U5YpqOyKlVplSRPrnDItG8TUpg9qqhO6-F7c7Rz2xU,5110
 codemie_sdk/services/datasource.py,sha256=hYH__M5LD33dfh7CCS7HYmEn8izsGkO0nfVa-dpoj6w,5118
 codemie_sdk/services/integration.py,sha256=vJnSkXk2C2l0ahX2SUsuA7fKhY2hTuAByynx5Lgh7Ls,4864
@@ -19,9 +20,10 @@ codemie_sdk/services/llm.py,sha256=0-e4_7RvLHs2giCyoQ5U4KDTh6p5VXgPKNxnDP9ZDFU,1
 codemie_sdk/services/task.py,sha256=3e9t8_LMkR4xfeMBwMCo7ZF87PxPS-ZbzDg85ilda2M,1031
 codemie_sdk/services/user.py,sha256=7B-Qw451qKPD5Io6qLda-kbFDaPRQ3TamJamiGwCQu4,1013
 codemie_sdk/services/workflow.py,sha256=aOV13WrFAqMXOPd2jVHbEhg3fpezk4XWo382D5aWtro,4833
-codemie_sdk/services/workflow_execution.py,sha256=Ud-cYC_a2yb3FErfwWjM7hBIHzs2QoYQ-kX-o7K1cms,3152
+codemie_sdk/services/workflow_execution.py,sha256=aGoT3rdTmh5-doAsrmBBjLEuOfvL5aqeo3g9th1_aAw,3647
+codemie_sdk/services/workflow_execution_state.py,sha256=tXoaa8yT09xgYEUNiHhVULe76TwGwVgZupMIUyyLxdo,2070
 codemie_sdk/utils/__init__.py,sha256=BXAJJfAzO89-kMYvWWo9wSNhSbGgF3vB1In9sePFhMM,109
-codemie_sdk/utils/http.py,sha256=d2GRkd-OBrrrURfVLeD7SUGFsyB4k1anYz5LXX3o4H8,7525
-codemie_sdk_python-0.1.5.dist-info/METADATA,sha256=K_V1NJ4kJcm3JNpxU3j7ddJU5qmI_MC6NxjsJqh-etE,18940
-codemie_sdk_python-0.1.5.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-codemie_sdk_python-0.1.5.dist-info/RECORD,,
+codemie_sdk/utils/http.py,sha256=JfyX4gP-grQUI795QL0DJ72aBkiqNd1fs31fUqtSnYE,7757
+codemie_sdk_python-0.1.6.dist-info/METADATA,sha256=bFS58qIKO7ZXcvu_L3QilTaE9tEV8hh8gQt4x-sqT6I,21439
+codemie_sdk_python-0.1.6.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+codemie_sdk_python-0.1.6.dist-info/RECORD,,