codemie-sdk-python 0.1.229__tar.gz → 0.1.231__tar.gz

{codemie_sdk_python-0.1.229 → codemie_sdk_python-0.1.231}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: codemie-sdk-python
-Version: 0.1.229
+Version: 0.1.231
 Summary: CodeMie SDK for Python
 Author: Vadym Vlasenko
 Author-email: vadym_vlasenko@epam.com
@@ -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()

{codemie_sdk_python-0.1.229 → codemie_sdk_python-0.1.231}/README.md

@@ -161,16 +161,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**
@@ -535,10 +560,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")
@@ -558,8 +591,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()

{codemie_sdk_python-0.1.229 → codemie_sdk_python-0.1.231}/pyproject.toml

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

{codemie_sdk_python-0.1.229 → codemie_sdk_python-0.1.231}/src/codemie_sdk/__init__.py

@@ -78,7 +78,7 @@ from .services.vendor_workflow import VendorWorkflowService
 from .services.vendor_knowledgebase import VendorKnowledgeBaseService
 from .services.vendor_guardrail import VendorGuardrailService
 
-__version__ = "0.2.11"
+__version__ = "0.2.12"
 __all__ = [
     "CodeMieClient",
     "VendorType",

{codemie_sdk_python-0.1.229 → codemie_sdk_python-0.1.231}/src/codemie_sdk/models/assistant.py

@@ -284,6 +284,14 @@ class AssistantChatRequest(BaseModel):
         default=None, description="DataSource in conversation history"
     )
     stream: bool = Field(default=False, description="Enable streaming response")
+    propagate_headers: bool = Field(
+        default=False,
+        description="Enable propagation of X-* HTTP headers to MCP servers during tool execution",
+    )
+    custom_metadata: Optional[dict[str, Any]] = Field(
+        default=None,
+        description="Custom metadata for the AI Assistant",
+    )
     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")

{codemie_sdk_python-0.1.229 → codemie_sdk_python-0.1.231}/src/codemie_sdk/models/workflow_execution_payload.py

@@ -15,3 +15,7 @@ class WorkflowExecutionCreateRequest(BaseModel):
     file_name: Optional[str] = Field(
         None, description="File name associated with the workflow execution"
     )
+    propagate_headers: bool = Field(
+        default=False,
+        description="Enable propagation of X-* HTTP headers to MCP servers during tool execution",
+    )

{codemie_sdk_python-0.1.229 → codemie_sdk_python-0.1.231}/src/codemie_sdk/services/assistant.py

@@ -309,12 +309,14 @@ class AssistantService:
         self,
         assistant_id: str,
         request: AssistantChatRequest,
+        headers: Optional[Dict[str, str]] = None,
     ) -> Union[requests.Response, BaseModelResponse]:
         """Send a chat request to an assistant.
 
         Args:
             assistant_id: ID of the assistant to chat with
             request: Chat request details
+            headers: Optional additional HTTP headers (e.g., X-* for MCP propagation)
 
         Returns:
             Chat response or streaming response
@@ -333,6 +335,7 @@ class AssistantService:
             BaseModelResponse,
             json_data=request.model_dump(exclude_none=True, by_alias=True),
             stream=request.stream,
+            extra_headers=headers,
         )
         if not request.stream and pydantic_schema:
             # we do conversion to the BaseModel here because self._parse_response don't see actual request model,
@@ -380,6 +383,43 @@ class AssistantService:
 
         return response
 
+    def chat_by_slug(
+        self,
+        assistant_slug: str,
+        request: AssistantChatRequest,
+        headers: Optional[Dict[str, str]] = None,
+    ) -> Union[requests.Response, BaseModelResponse]:
+        """Send a chat request to an assistant by slug.
+
+        Args:
+            assistant_slug: Slug of the assistant to chat with
+            request: Chat request details
+            headers: Optional additional HTTP headers (e.g., X-* for MCP propagation)
+
+        Returns:
+            Chat response or streaming response
+        """
+        pydantic_schema = None
+        if (
+            request.output_schema is not None
+            and inspect.isclass(request.output_schema)
+            and issubclass(request.output_schema, BaseModel)
+        ):
+            pydantic_schema = deepcopy(request.output_schema)
+            request.output_schema = request.output_schema.model_json_schema()
+
+        response = self._api.post(
+            f"/v1/assistants/slug/{assistant_slug}/model",
+            BaseModelResponse,
+            json_data=request.model_dump(exclude_none=True, by_alias=True),
+            stream=request.stream,
+            extra_headers=headers,
+        )
+        if not request.stream and pydantic_schema:
+            response.generated = pydantic_schema.model_validate(response.generated)
+
+        return response
+
     def upload_file_to_chat(self, file_path: Path):
         """Upload a file to assistant chat and return the response containing file_url."""
 

{codemie_sdk_python-0.1.229 → codemie_sdk_python-0.1.231}/src/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_python-0.1.229 → codemie_sdk_python-0.1.231}/src/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_python-0.1.229 → codemie_sdk_python-0.1.231}/src/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,