codemie-sdk-python 0.1.212__py3-none-any.whl → 0.1.230__py3-none-any.whl

codemie_sdk/services/vendor_workflow.py

@@ -0,0 +1,330 @@
+"""Vendor workflow service implementation for managing cloud vendor workflow settings."""
+
+from typing import Union, Optional, List
+
+from ..models.vendor_assistant import VendorType
+from ..models.vendor_workflow import (
+    VendorWorkflowSettingsResponse,
+    VendorWorkflowsResponse,
+    VendorWorkflow,
+    VendorWorkflowAliasesResponse,
+    VendorWorkflowInstallRequest,
+    VendorWorkflowInstallResponse,
+    VendorWorkflowUninstallResponse,
+)
+from ..utils import ApiRequestHandler
+
+
+class VendorWorkflowService:
+    """Service for managing cloud vendor workflow settings (AWS, Azure, GCP)."""
+
+    def __init__(self, api_domain: str, token: str, verify_ssl: bool = True):
+        """Initialize the vendor workflow service.
+
+        Args:
+            api_domain: Base URL for the CodeMie API
+            token: Authentication token
+            verify_ssl: Whether to verify SSL certificates
+        """
+        self._api = ApiRequestHandler(api_domain, token, verify_ssl)
+
+    def get_workflow_settings(
+        self,
+        vendor: Union[VendorType, str],
+        page: int = 0,
+        per_page: int = 10,
+    ) -> VendorWorkflowSettingsResponse:
+        """Get workflow settings for a specific cloud vendor.
+
+        Args:
+            vendor: Cloud vendor type (aws, azure, gcp). Can be VendorType enum or string.
+            page: Page number for pagination (0-based)
+            per_page: Number of items per page
+
+        Returns:
+            VendorWorkflowSettingsResponse containing list of settings and pagination info
+
+        Example:
+            >>> # Using enum
+            >>> settings = client.vendor_workflows.get_workflow_settings(VendorType.AWS, page=0, per_page=10)
+            >>> # Using string
+            >>> settings = client.vendor_workflows.get_workflow_settings("aws", page=0, per_page=10)
+            >>> # Access settings data
+            >>> for setting in settings.data:
+            ...     print(f"Setting: {setting.setting_name}, Project: {setting.project}")
+            ...     if setting.invalid:
+            ...         print(f"Error: {setting.error}")
+        """
+        # Convert enum to string value if needed
+        vendor_str = vendor.value if isinstance(vendor, VendorType) else vendor
+
+        params = {
+            "page": page,
+            "per_page": per_page,
+        }
+
+        return self._api.get(
+            f"/v1/vendors/{vendor_str}/workflows/settings",
+            VendorWorkflowSettingsResponse,
+            params=params,
+            wrap_response=False,
+        )
+
+    def get_workflows(
+        self,
+        vendor: Union[VendorType, str],
+        setting_id: str,
+        per_page: int = 10,
+        next_token: Optional[str] = None,
+    ) -> VendorWorkflowsResponse:
+        """Get workflows for a specific vendor setting.
+
+        Args:
+            vendor: Cloud vendor type (aws, azure, gcp). Can be VendorType enum or string.
+            setting_id: ID of the vendor setting to retrieve workflows for
+            per_page: Number of items per page
+            next_token: Token for pagination (optional, for retrieving next page)
+
+        Returns:
+            VendorWorkflowsResponse containing list of workflows and pagination token
+
+        Example:
+            >>> # Get first page
+            >>> workflows = client.vendor_workflows.get_workflows(
+            ...     vendor=VendorType.AWS,
+            ...     setting_id="cac90788-39b7-4ffe-8b57-e8b047fa1f6c",
+            ...     per_page=8
+            ... )
+            >>> # Access workflow data
+            >>> for workflow in workflows.data:
+            ...     print(f"Name: {workflow.name}, Status: {workflow.status}")
+            ...     print(f"Version: {workflow.version}, Description: {workflow.description}")
+            >>> # Get next page if available
+            >>> if workflows.pagination.next_token:
+            ...     next_page = client.vendor_workflows.get_workflows(
+            ...         vendor=VendorType.AWS,
+            ...         setting_id="cac90788-39b7-4ffe-8b57-e8b047fa1f6c",
+            ...         per_page=8,
+            ...         next_token=workflows.pagination.next_token
+            ...     )
+        """
+        # Convert enum to string value if needed
+        vendor_str = vendor.value if isinstance(vendor, VendorType) else vendor
+
+        params = {
+            "setting_id": setting_id,
+            "per_page": per_page,
+        }
+
+        if next_token:
+            params["next_token"] = next_token
+
+        return self._api.get(
+            f"/v1/vendors/{vendor_str}/workflows",
+            VendorWorkflowsResponse,
+            params=params,
+            wrap_response=False,
+        )
+
+    def get_workflow(
+        self,
+        vendor: Union[VendorType, str],
+        workflow_id: str,
+        setting_id: str,
+    ) -> VendorWorkflow:
+        """Get a specific workflow by ID for a vendor setting.
+
+        Args:
+            vendor: Cloud vendor type (aws, azure, gcp). Can be VendorType enum or string.
+            workflow_id: ID of the workflow to retrieve
+            setting_id: ID of the vendor setting
+
+        Returns:
+            VendorWorkflow containing workflow details
+
+        Example:
+            >>> workflow = client.vendor_workflows.get_workflow(
+            ...     vendor=VendorType.AWS,
+            ...     workflow_id="9HXLQ7J9YP",
+            ...     setting_id="cac90788-39b7-4ffe-8b57-e8b047fa1f6c"
+            ... )
+            >>> print(f"Name: {workflow.name}, Status: {workflow.status}")
+            >>> print(f"Version: {workflow.version}, Description: {workflow.description}")
+        """
+        # Convert enum to string value if needed
+        vendor_str = vendor.value if isinstance(vendor, VendorType) else vendor
+
+        params = {
+            "setting_id": setting_id,
+        }
+
+        return self._api.get(
+            f"/v1/vendors/{vendor_str}/workflows/{workflow_id}",
+            VendorWorkflow,
+            params=params,
+            wrap_response=False,
+        )
+
+    def get_workflow_aliases(
+        self,
+        vendor: Union[VendorType, str],
+        workflow_id: str,
+        setting_id: str,
+        per_page: int = 10,
+        next_token: Optional[str] = None,
+    ) -> VendorWorkflowAliasesResponse:
+        """Get aliases for a specific vendor workflow.
+
+        Args:
+            vendor: Cloud vendor type (aws, azure, gcp). Can be VendorType enum or string.
+            workflow_id: ID of the workflow to retrieve aliases for
+            setting_id: ID of the vendor setting
+            per_page: Number of items per page
+            next_token: Token for pagination (optional, for retrieving next page)
+
+        Returns:
+            VendorWorkflowAliasesResponse containing list of aliases and pagination token
+
+        Example:
+            >>> # Get first page of aliases
+            >>> aliases = client.vendor_workflows.get_workflow_aliases(
+            ...     vendor=VendorType.AWS,
+            ...     workflow_id="9HXLQ7J9YP",
+            ...     setting_id="cac90788-39b7-4ffe-8b57-e8b047fa1f6c",
+            ...     per_page=5
+            ... )
+            >>> for alias in aliases.data:
+            ...     print(f"{alias.name} (v{alias.version}): {alias.status}")
+            ...     if alias.aiRunId:
+            ...         print(f"  AI Run ID: {alias.aiRunId}")
+            >>> # Get next page if available
+            >>> if aliases.pagination.next_token:
+            ...     next_page = client.vendor_workflows.get_workflow_aliases(
+            ...         vendor=VendorType.AWS,
+            ...         workflow_id="9HXLQ7J9YP",
+            ...         setting_id="cac90788-39b7-4ffe-8b57-e8b047fa1f6c",
+            ...         per_page=5,
+            ...         next_token=aliases.pagination.next_token
+            ...     )
+        """
+        # Convert enum to string value if needed
+        vendor_str = vendor.value if isinstance(vendor, VendorType) else vendor
+
+        params = {
+            "setting_id": setting_id,
+            "per_page": per_page,
+        }
+
+        if next_token:
+            params["next_token"] = next_token
+
+        return self._api.get(
+            f"/v1/vendors/{vendor_str}/workflows/{workflow_id}/aliases",
+            VendorWorkflowAliasesResponse,
+            params=params,
+            wrap_response=False,
+        )
+
+    def install_workflows(
+        self,
+        vendor: Union[VendorType, str],
+        workflows: List[VendorWorkflowInstallRequest],
+    ) -> VendorWorkflowInstallResponse:
+        """Install/activate vendor workflows.
+
+        Args:
+            vendor: Cloud vendor type (aws, azure, gcp). Can be VendorType enum or string.
+            workflows: List of workflow installation requests with workflow ID, flow alias ID, and setting ID
+
+        Returns:
+            VendorWorkflowInstallResponse containing installation summary with AI run IDs
+
+        Example:
+            >>> from codemie_sdk import VendorWorkflowInstallRequest
+            >>> # Install single workflow
+            >>> install_request = VendorWorkflowInstallRequest(
+            ...     id="9HXLQ7J9YP",
+            ...     flowAliasId="9RUV0BI2L7",
+            ...     setting_id="cac90788-39b7-4ffe-8b57-e8b047fa1f6c"
+            ... )
+            >>> response = client.vendor_workflows.install_workflows(
+            ...     vendor=VendorType.AWS,
+            ...     workflows=[install_request]
+            ... )
+            >>> for item in response.summary:
+            ...     print(f"Installed workflow {item.flowId} with run ID: {item.aiRunId}")
+            >>>
+            >>> # Install multiple workflows
+            >>> requests = [
+            ...     VendorWorkflowInstallRequest(
+            ...         id="WORKFLOW_ID_1",
+            ...         flowAliasId="ALIAS_ID_1",
+            ...         setting_id="SETTING_ID"
+            ...     ),
+            ...     VendorWorkflowInstallRequest(
+            ...         id="WORKFLOW_ID_2",
+            ...         flowAliasId="ALIAS_ID_2",
+            ...         setting_id="SETTING_ID"
+            ...     )
+            ... ]
+            >>> response = client.vendor_workflows.install_workflows(
+            ...     vendor=VendorType.AWS,
+            ...     workflows=requests
+            ... )
+        """
+        # Convert enum to string value if needed
+        vendor_str = vendor.value if isinstance(vendor, VendorType) else vendor
+
+        # Convert list of Pydantic models to list of dicts
+        payload = [workflow.model_dump(by_alias=True) for workflow in workflows]
+
+        return self._api.post(
+            f"/v1/vendors/{vendor_str}/workflows",
+            VendorWorkflowInstallResponse,
+            json_data=payload,
+            wrap_response=False,
+        )
+
+    def uninstall_workflow(
+        self,
+        vendor: Union[VendorType, str],
+        ai_run_id: str,
+    ) -> VendorWorkflowUninstallResponse:
+        """Uninstall/deactivate a vendor workflow.
+
+        Args:
+            vendor: Cloud vendor type (aws, azure, gcp). Can be VendorType enum or string.
+            ai_run_id: AI run ID returned from the workflow alias (aiRunId field)
+
+        Returns:
+            VendorWorkflowUninstallResponse with success status
+
+        Example:
+            >>> # Get workflow aliases to find the aiRunId
+            >>> aliases = client.vendor_workflows.get_workflow_aliases(
+            ...     vendor=VendorType.AWS,
+            ...     workflow_id="9HXLQ7J9YP",
+            ...     setting_id="cac90788-39b7-4ffe-8b57-e8b047fa1f6c"
+            ... )
+            >>> # Find an installed alias with aiRunId
+            >>> for alias in aliases.data:
+            ...     if alias.aiRunId:
+            ...         ai_run_id = alias.aiRunId
+            ...         break
+            >>>
+            >>> # Uninstall the workflow using the AI run ID
+            >>> response = client.vendor_workflows.uninstall_workflow(
+            ...     vendor=VendorType.AWS,
+            ...     ai_run_id="56fed66d-f66e-46e3-b420-bb3a8d93eed4"
+            ... )
+            >>> if response.success:
+            ...     print("Workflow successfully uninstalled!")
+        """
+        # Convert enum to string value if needed
+        vendor_str = vendor.value if isinstance(vendor, VendorType) else vendor
+
+        return self._api.delete(
+            f"/v1/vendors/{vendor_str}/workflows/{ai_run_id}",
+            VendorWorkflowUninstallResponse,
+            wrap_response=False,
+        )

codemie_sdk/services/workflow.py

@@ -135,6 +135,8 @@ class WorkflowService:
         workflow_id: str,
         user_input: Optional[str] = None,
         file_name: Optional[str] = None,
+        propagate_headers: bool = False,
+        headers: Optional[dict[str, str]] = None,
     ) -> dict:
         """Run a workflow with optional input parameters.
 
@@ -142,11 +144,18 @@ class WorkflowService:
             workflow_id: ID of the workflow to run
             user_input: Optional user input for the workflow execution
             file_name: Optional file name for the workflow execution
+            propagate_headers: Enable propagation of X-* HTTP headers to MCP servers
+            headers: Optional additional HTTP headers (e.g., X-* for MCP propagation)
 
         Returns:
             dict: Created workflow execution details
         """
-        return self.executions(workflow_id).create(user_input, file_name)
+        return self.executions(workflow_id).create(
+            user_input=user_input,
+            file_name=file_name,
+            propagate_headers=propagate_headers,
+            headers=headers,
+        )
 
     def executions(self, workflow_id: str) -> WorkflowExecutionService:
         """Get workflow execution service for the specified workflow.

codemie_sdk/services/workflow_execution.py

@@ -46,24 +46,33 @@ class WorkflowExecutionService:
         )
 
     def create(
-        self, user_input: Optional[str] = None, file_name: Optional[str] = None
+        self,
+        user_input: Optional[str] = None,
+        file_name: Optional[str] = None,
+        propagate_headers: bool = False,
+        headers: Optional[dict[str, str]] = None,
     ) -> dict:
         """Create a new workflow execution.
 
         Args:
             user_input: Optional input data for the workflow execution.
             file_name: Optional file name associated with the workflow execution.
+            propagate_headers: Enable propagation of X-* HTTP headers to MCP servers.
+            headers: Optional additional HTTP headers (e.g., X-* for MCP propagation)
 
         Returns:
             dict: Created workflow execution details
         """
         payload = WorkflowExecutionCreateRequest(
-            user_input=user_input, file_name=file_name
+            user_input=user_input,
+            file_name=file_name,
+            propagate_headers=propagate_headers,
         )
         return self._api.post(
             f"/v1/workflows/{self._workflow_id}/executions",
             dict,
             json_data=payload.model_dump(),
+            extra_headers=headers,
         )
 
     def get(self, execution_id: str) -> WorkflowExecution:
@@ -108,17 +117,30 @@ class WorkflowExecutionService:
             f"/v1/workflows/{self._workflow_id}/executions/{execution_id}/abort", dict
         )
 
-    def resume(self, execution_id: str) -> dict:
+    def resume(
+        self,
+        execution_id: str,
+        propagate_headers: bool = False,
+        headers: Optional[dict[str, str]] = None,
+    ) -> dict:
         """Resume an interrupted workflow execution.
 
         Args:
             execution_id: ID of the execution to resume
+            propagate_headers: Enable propagation of X-* HTTP headers to MCP servers.
+            headers: Optional additional HTTP headers (e.g., X-* for MCP propagation)
 
         Returns:
             dict: Updated workflow execution details
         """
+        params = {"propagate_headers": propagate_headers}
+        # Empty body per API; passing empty dict to satisfy typing
         return self._api.put(
-            f"/v1/workflows/{self._workflow_id}/executions/{execution_id}/resume", dict
+            f"/v1/workflows/{self._workflow_id}/executions/{execution_id}/resume",
+            dict,
+            json_data={},
+            params=params,
+            extra_headers=headers,
         )
 
     def get_thoughts(

codemie_sdk/utils/http.py

@@ -161,6 +161,7 @@ class ApiRequestHandler:
         stream: bool = False,
         wrap_response: bool = True,
         raise_on_error: bool = True,
+        extra_headers: Optional[Dict[str, str]] = None,
     ) -> Union[T, requests.Response]:
         """Makes a POST request and parses the response.
 
@@ -171,6 +172,7 @@ class ApiRequestHandler:
             stream: Whether to return streaming response
             wrap_response: Whether response is wrapped in 'data' field
             raise_on_error: Whether to raise exception on HTTP error status codes
+            extra_headers: Optional additional HTTP headers to include (e.g., X-* for MCP propagation)
 
         Returns:
             Parsed response object/list or streaming response
@@ -178,9 +180,13 @@ class ApiRequestHandler:
         if json_data:
             logger.debug(f"Request body: {json_data}")
 
+        headers = self._get_headers()
+        if extra_headers:
+            headers.update(extra_headers)
+
         response = requests.post(
             url=f"{self._base_url}{endpoint}",
-            headers=self._get_headers(),
+            headers=headers,
             json=json_data,
             verify=self._verify_ssl,
             stream=stream,
@@ -242,6 +248,7 @@ class ApiRequestHandler:
         json_data: Dict[str, Any],
         params: Optional[Dict[str, Any]] = None,
         wrap_response: bool = True,
+        extra_headers: Optional[Dict[str, str]] = None,
     ) -> T:
         """Makes a PUT request and parses the response.
 
@@ -251,14 +258,18 @@ class ApiRequestHandler:
             json_data: JSON request body
             params: Query parameters
             wrap_response: Whether response is wrapped in 'data' field
+            extra_headers: Optional additional HTTP headers to include (e.g., X-* for MCP propagation)
 
         Returns:
             Parsed response object or list of objects
         """
         logger.debug(f"Request body: {json_data}")
+        headers = self._get_headers()
+        if extra_headers:
+            headers.update(extra_headers)
         response = requests.put(
             url=f"{self._base_url}{endpoint}",
-            headers=self._get_headers(),
+            headers=headers,
             json=json_data,
             params=params,
             verify=self._verify_ssl,

{codemie_sdk_python-0.1.212.dist-info → codemie_sdk_python-0.1.230.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: codemie-sdk-python
-Version: 0.1.212
+Version: 0.1.230
 Summary: CodeMie SDK for Python
 Author: Vadym Vlasenko
 Author-email: vadym_vlasenko@epam.com
@@ -59,7 +59,7 @@ Python SDK for CodeMie services. This SDK provides a comprehensive interface to
 ## Installation
 
 ```sh
-pip install codemie-sdk
+pip install codemie-sdk-python
 ```
 OR
 ```sh
@@ -174,16 +174,41 @@ result = client.assistant.delete("assistant-id")
 
 #### Advanced Features
 
-6. **Chat with Assistant**
+6. **Chat with Assistant (with MCP header propagation)**
 ```python
 from codemie_sdk.models.assistant import AssistantChatRequest
 
 chat_request = AssistantChatRequest(
     text="Your message here",
     stream=False,  # Set to True for streaming response
-    # Additional parameters
+    propagate_headers=True,  # Enable propagation of X-* headers to MCP servers
+)
+# Pass X-* headers to forward to MCP servers
+response = client.assistant.chat(
+    "assistant-id",
+    chat_request,
+    headers={
+        "X-Tenant-ID": "tenant-abc-123",
+        "X-User-ID": "user-456",
+        "X-Request-ID": "req-123",
+    },
+)
+```
+
+6.a. **Chat with Assistant by slug (with MCP header propagation)**
+```python
+chat_request = AssistantChatRequest(
+    text="Your message here",
+    propagate_headers=True,
+)
+response = client.assistants.chat_by_slug(
+    "assistant-slug",
+    chat_request,
+    headers={
+        "X-Environment": "production",
+        "X-Feature-Flag-Beta": "true",
+    },
 )
-response = client.assistant.chat("assistant-id", chat_request)
 ```
 
 7. **Utilize structured outputs with Assistant**
@@ -548,10 +573,18 @@ result = client.workflow.delete("workflow-id")
 
 The SDK provides comprehensive workflow execution management through the WorkflowExecutionService:
 
-1. **Run Workflow**
+1. **Run Workflow (with MCP header propagation)**
 ```python
-# Simple workflow execution
-execution = client.workflow.run("workflow-id", user_input="optional input")
+# Enable propagation in payload and pass X-* headers to forward to MCP servers
+execution = client.workflow.run(
+    "workflow-id",
+    user_input="optional input",
+    propagate_headers=True,
+    headers={
+        "X-Request-ID": "req-abc-123",
+        "X-Source-App": "analytics-ui",
+    },
+)
 
 # Get execution service for advanced operations
 execution_service = client.workflow.executions("workflow-id")
@@ -571,8 +604,14 @@ execution = execution_service.get("execution-id")
 # Abort running execution
 result = execution_service.abort("execution-id")
 
-# Resume interrupted execution
-result = execution_service.resume("execution-id")
+# Resume interrupted execution with header propagation (query param + headers)
+result = execution_service.resume(
+    "execution-id",
+    propagate_headers=True,
+    headers={
+        "X-Correlation-ID": "corr-456",
+    },
+)
 
 # Delete all executions
 result = execution_service.delete_all()
@@ -831,6 +870,50 @@ client = CodeMieClient(
 )
 ```
 
+### Assistant Versioning via SDK
+
+The Python SDK exposes full assistant versioning capabilities delivered in EPMCDME-8285.
+
+- List versions
+```python
+versions = client.assistants.list_versions("assistant-id", page=0, per_page=20)
+print([v.version_number for v in versions])
+```
+
+- Get specific version
+```python
+version = client.assistants.get_version("assistant-id", 2)
+print(version.system_prompt)
+```
+
+- Compare two versions
+```python
+from codemie_sdk.models.assistant import AssistantVersionDiff
+
+diff: AssistantVersionDiff = client.assistants.compare_versions("assistant-id", 1, 3)
+print(diff.summary)
+```
+
+- Rollback to a version
+```python
+resp = client.assistants.rollback_to_version("assistant-id", 2)
+print(resp)
+```
+
+- Chat with a specific version
+```python
+from codemie_sdk.models.assistant import AssistantChatRequest
+
+req = AssistantChatRequest(text="Hi", stream=False)
+resp = client.assistants.chat_with_version("assistant-id", 2, req)
+print(resp.generated)
+```
+
+Quick CLI example
+```bash
+python codemie-sdk/examples/assistant_versions.py <assistant_id> [version_number]
+```
+
 ## Support
 For providing credentials please contact AI/Run CodeMie Team: Vadym_Vlasenko@epam.com or Nikita_Levyankov@epam.com