codemie-sdk-python 0.1.191__tar.gz → 0.1.249__tar.gz

{codemie_sdk_python-0.1.191 → codemie_sdk_python-0.1.249}/PKG-INFO

@@ -1,13 +1,13 @@
 Metadata-Version: 2.1
 Name: codemie-sdk-python
-Version: 0.1.191
+Version: 0.1.249
 Summary: CodeMie SDK for Python
 Author: Vadym Vlasenko
 Author-email: vadym_vlasenko@epam.com
 Requires-Python: >=3.12,<4.0
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.12
-Requires-Dist: pydantic (>=2.11.1,<3.0.0)
+Requires-Dist: pydantic (>=2.12.2,<3.0.0)
 Requires-Dist: requests (>=2.31.0,<3.0.0)
 Description-Content-Type: text/markdown
 
@@ -25,6 +25,7 @@ Python SDK for CodeMie services. This SDK provides a comprehensive interface to
   - [Assistant Service](#assistant-service)
     - [Core Methods](#core-methods)
     - [Advanced Features](#advanced-features)
+    - [Prompt Variables Support](#prompt-variables-support)
   - [Datasource Service](#datasource-service)
     - [Supported Datasource Types](#supported-datasource-types)
     - [Core Methods](#core-methods-1)
@@ -43,6 +44,8 @@ Python SDK for CodeMie services. This SDK provides a comprehensive interface to
     - [Workflow Status Monitoring](#workflow-status-monitoring)
   - [Conversation Service](#conversation-service)
     - [Core Methods](#core-methods-4)
+  - [Webhook Service](#webhook-service)
+    - [Core Methods](#core-methods-5)
 - [Development](#development)
   - [Setup](#setup)
   - [Running Tests](#running-tests)
@@ -57,7 +60,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
@@ -172,16 +175,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**
@@ -241,6 +269,42 @@ prebuilt_assistant = client.assistant.get_prebuilt_by_slug("assistant-slug")
 tools = client.assistant.get_tools()
 ```
 
+### Prompt Variables Support
+The SDK supports assistant-level prompt variables that the backend already exposes via the `prompt_variables` field.
+
+Create and update an assistant with prompt variables:
+```python
+from codemie_sdk.models.assistant import AssistantCreateRequest, AssistantUpdateRequest, PromptVariable
+
+# Create
+create_req = AssistantCreateRequest(
+    name="My Assistant",
+    description="Assistant description",
+    system_prompt="Instructions. Use {{project_name}} in responses.",
+    toolkits=[],
+    project="my_project",
+    llm_model_type="gpt-4o",
+    context=[],
+    conversation_starters=[],
+    mcp_servers=[],
+    assistant_ids=[],
+    prompt_variables=[
+        PromptVariable(key="project_name", default_value="Delta", description="Current project"),
+        PromptVariable(key="region", default_value="eu"),
+    ],
+)
+client.assistants.create(create_req)
+
+# Update
+update_req = AssistantUpdateRequest(
+    **create_req.model_dump(),
+    prompt_variables=[
+        PromptVariable(key="project_name", default_value="Delta-Updated"),
+        PromptVariable(key="region", default_value="us"),
+    ],
+)
+client.assistants.update("assistant-id", update_req)
+
 ### Datasource Service
 
 The Datasource service enables managing various types of data sources in CodeMie, including code repositories, Confluence spaces, Jira projects, files, and Google documents.
@@ -546,10 +610,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")
@@ -569,8 +641,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()
@@ -735,6 +813,19 @@ client.conversations.list_by_assistant_id("assistant-id")
 client.conversations.delete("conversation-id")
 ```
 
+
+### Webhook Service
+
+The Webhook service provides access to trigger available webhook in CodeMie.
+
+#### Core Methods
+
+1. **Get All Conversations**
+```python
+# Trigger assistant/workflow/datasource by it's ID
+# Data - body of the post method
+response = client.webhook.trigger("resource_id", "data")
+```
 ## Error Handling
 
 The SDK implements comprehensive error handling. All API calls may raise exceptions for:
@@ -816,6 +907,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
 

{codemie_sdk_python-0.1.191 → codemie_sdk_python-0.1.249}/README.md

@@ -12,6 +12,7 @@ Python SDK for CodeMie services. This SDK provides a comprehensive interface to
   - [Assistant Service](#assistant-service)
     - [Core Methods](#core-methods)
     - [Advanced Features](#advanced-features)
+    - [Prompt Variables Support](#prompt-variables-support)
   - [Datasource Service](#datasource-service)
     - [Supported Datasource Types](#supported-datasource-types)
     - [Core Methods](#core-methods-1)
@@ -30,6 +31,8 @@ Python SDK for CodeMie services. This SDK provides a comprehensive interface to
     - [Workflow Status Monitoring](#workflow-status-monitoring)
   - [Conversation Service](#conversation-service)
     - [Core Methods](#core-methods-4)
+  - [Webhook Service](#webhook-service)
+    - [Core Methods](#core-methods-5)
 - [Development](#development)
   - [Setup](#setup)
   - [Running Tests](#running-tests)
@@ -44,7 +47,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
@@ -159,16 +162,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**
@@ -228,6 +256,42 @@ prebuilt_assistant = client.assistant.get_prebuilt_by_slug("assistant-slug")
 tools = client.assistant.get_tools()
 ```
 
+### Prompt Variables Support
+The SDK supports assistant-level prompt variables that the backend already exposes via the `prompt_variables` field.
+
+Create and update an assistant with prompt variables:
+```python
+from codemie_sdk.models.assistant import AssistantCreateRequest, AssistantUpdateRequest, PromptVariable
+
+# Create
+create_req = AssistantCreateRequest(
+    name="My Assistant",
+    description="Assistant description",
+    system_prompt="Instructions. Use {{project_name}} in responses.",
+    toolkits=[],
+    project="my_project",
+    llm_model_type="gpt-4o",
+    context=[],
+    conversation_starters=[],
+    mcp_servers=[],
+    assistant_ids=[],
+    prompt_variables=[
+        PromptVariable(key="project_name", default_value="Delta", description="Current project"),
+        PromptVariable(key="region", default_value="eu"),
+    ],
+)
+client.assistants.create(create_req)
+
+# Update
+update_req = AssistantUpdateRequest(
+    **create_req.model_dump(),
+    prompt_variables=[
+        PromptVariable(key="project_name", default_value="Delta-Updated"),
+        PromptVariable(key="region", default_value="us"),
+    ],
+)
+client.assistants.update("assistant-id", update_req)
+
 ### Datasource Service
 
 The Datasource service enables managing various types of data sources in CodeMie, including code repositories, Confluence spaces, Jira projects, files, and Google documents.
@@ -533,10 +597,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")
@@ -556,8 +628,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()
@@ -722,6 +800,19 @@ client.conversations.list_by_assistant_id("assistant-id")
 client.conversations.delete("conversation-id")
 ```
 
+
+### Webhook Service
+
+The Webhook service provides access to trigger available webhook in CodeMie.
+
+#### Core Methods
+
+1. **Get All Conversations**
+```python
+# Trigger assistant/workflow/datasource by it's ID
+# Data - body of the post method
+response = client.webhook.trigger("resource_id", "data")
+```
 ## Error Handling
 
 The SDK implements comprehensive error handling. All API calls may raise exceptions for:
@@ -803,6 +894,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
 

{codemie_sdk_python-0.1.191 → codemie_sdk_python-0.1.249}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "codemie-sdk-python"
-version = "0.1.191"
+version = "0.1.249"
 description = "CodeMie SDK for Python"
 authors = [
     "Vadym Vlasenko <vadym_vlasenko@epam.com>",
@@ -13,7 +13,7 @@ packages = [
 [tool.poetry.dependencies]
 python = "^3.12"
 requests = "^2.31.0"
-pydantic = "^2.11.1"
+pydantic = "^2.12.2"
 
 [tool.poetry.group.dev.dependencies]
 ruff = "^0.11.3"

codemie_sdk_python-0.1.249/src/codemie_sdk/__init__.py

@@ -0,0 +1,135 @@
+"""
+CodeMie SDK for Python
+~~~~~~~~~~~~~~~~~~~~~
+
+A Python SDK for interacting with CodeMie API.
+
+Basic usage:
+
+    >>> from codemie_sdk import CodeMieClient
+    >>> client = CodeMieClient(
+    ...     auth_server_url="https://auth.example.com",
+    ...     auth_client_id="client_id",
+    ...     auth_client_secret="secret",
+    ...     auth_realm_name="realm",
+    ...     codemie_api_domain="api.codemie.com"
+    ... )
+    >>> assistants = client.assistants.list()
+"""
+
+from .client.client import CodeMieClient
+from .models.vendor_assistant import (
+    VendorType,
+    VendorAssistantSetting,
+    VendorAssistantSettingsResponse,
+    VendorAssistant,
+    VendorAssistantVersion,
+    VendorAssistantStatus,
+    VendorAssistantsResponse,
+    VendorAssistantAlias,
+    VendorAssistantAliasesResponse,
+    VendorAssistantInstallRequest,
+    VendorAssistantInstallSummary,
+    VendorAssistantInstallResponse,
+    VendorAssistantUninstallResponse,
+    PaginationInfo,
+    TokenPagination,
+)
+from .models.vendor_workflow import (
+    VendorWorkflowSetting,
+    VendorWorkflowSettingsResponse,
+    VendorWorkflow,
+    VendorWorkflowStatus,
+    VendorWorkflowsResponse,
+    VendorWorkflowAlias,
+    VendorWorkflowAliasesResponse,
+    VendorWorkflowInstallRequest,
+    VendorWorkflowInstallSummary,
+    VendorWorkflowInstallResponse,
+    VendorWorkflowUninstallResponse,
+)
+from .models.vendor_knowledgebase import (
+    VendorKnowledgeBaseSetting,
+    VendorKnowledgeBaseSettingsResponse,
+    VendorKnowledgeBase,
+    VendorKnowledgeBaseStatus,
+    VendorKnowledgeBasesResponse,
+    VendorKnowledgeBaseDetail,
+    VendorKnowledgeBaseInstallRequest,
+    VendorKnowledgeBaseInstallSummary,
+    VendorKnowledgeBaseInstallResponse,
+    VendorKnowledgeBaseUninstallResponse,
+)
+from .models.vendor_guardrail import (
+    VendorGuardrailSetting,
+    VendorGuardrailSettingsResponse,
+    VendorGuardrail,
+    VendorGuardrailStatus,
+    VendorGuardrailsResponse,
+    VendorGuardrailVersion,
+    VendorGuardrailVersionsResponse,
+    VendorGuardrailInstallRequest,
+    VendorGuardrailInstallSummary,
+    VendorGuardrailInstallResponse,
+    VendorGuardrailUninstallResponse,
+)
+from .services.vendor_assistant import VendorAssistantService
+from .services.vendor_workflow import VendorWorkflowService
+from .services.vendor_knowledgebase import VendorKnowledgeBaseService
+from .services.vendor_guardrail import VendorGuardrailService
+
+__version__ = "0.2.12"
+__all__ = [
+    "CodeMieClient",
+    "VendorType",
+    "VendorAssistantSetting",
+    "VendorAssistantSettingsResponse",
+    "VendorAssistant",
+    "VendorAssistantVersion",
+    "VendorAssistantStatus",
+    "VendorAssistantsResponse",
+    "VendorAssistantAlias",
+    "VendorAssistantAliasesResponse",
+    "VendorAssistantInstallRequest",
+    "VendorAssistantInstallSummary",
+    "VendorAssistantInstallResponse",
+    "VendorAssistantUninstallResponse",
+    "PaginationInfo",
+    "TokenPagination",
+    "VendorAssistantService",
+    "VendorWorkflowSetting",
+    "VendorWorkflowSettingsResponse",
+    "VendorWorkflow",
+    "VendorWorkflowStatus",
+    "VendorWorkflowsResponse",
+    "VendorWorkflowAlias",
+    "VendorWorkflowAliasesResponse",
+    "VendorWorkflowInstallRequest",
+    "VendorWorkflowInstallSummary",
+    "VendorWorkflowInstallResponse",
+    "VendorWorkflowUninstallResponse",
+    "VendorWorkflowService",
+    "VendorKnowledgeBaseSetting",
+    "VendorKnowledgeBaseSettingsResponse",
+    "VendorKnowledgeBase",
+    "VendorKnowledgeBaseStatus",
+    "VendorKnowledgeBasesResponse",
+    "VendorKnowledgeBaseDetail",
+    "VendorKnowledgeBaseInstallRequest",
+    "VendorKnowledgeBaseInstallSummary",
+    "VendorKnowledgeBaseInstallResponse",
+    "VendorKnowledgeBaseUninstallResponse",
+    "VendorKnowledgeBaseService",
+    "VendorGuardrailSetting",
+    "VendorGuardrailSettingsResponse",
+    "VendorGuardrail",
+    "VendorGuardrailStatus",
+    "VendorGuardrailsResponse",
+    "VendorGuardrailVersion",
+    "VendorGuardrailVersionsResponse",
+    "VendorGuardrailInstallRequest",
+    "VendorGuardrailInstallSummary",
+    "VendorGuardrailInstallResponse",
+    "VendorGuardrailUninstallResponse",
+    "VendorGuardrailService",
+]

{codemie_sdk_python-0.1.191 → codemie_sdk_python-0.1.249}/src/codemie_sdk/client/client.py

@@ -11,6 +11,12 @@ from ..services.integration import IntegrationService
 from ..services.task import TaskService
 from ..services.user import UserService
 from ..services.workflow import WorkflowService
+from ..services.files import FileOperationService
+from ..services.webhook import WebhookService
+from ..services.vendor_assistant import VendorAssistantService
+from ..services.vendor_workflow import VendorWorkflowService
+from ..services.vendor_knowledgebase import VendorKnowledgeBaseService
+from ..services.vendor_guardrail import VendorGuardrailService
 
 
 class CodeMieClient:
@@ -81,6 +87,24 @@ class CodeMieClient:
         self.conversations = ConversationService(
             self._api_domain, self._token, verify_ssl=self._verify_ssl
         )
+        self.files = FileOperationService(
+            self._api_domain, self._token, verify_ssl=self._verify_ssl
+        )
+        self.webhook = WebhookService(
+            self._api_domain, self._token, verify_ssl=self._verify_ssl
+        )
+        self.vendor_assistants = VendorAssistantService(
+            self._api_domain, self._token, verify_ssl=self._verify_ssl
+        )
+        self.vendor_workflows = VendorWorkflowService(
+            self._api_domain, self._token, verify_ssl=self._verify_ssl
+        )
+        self.vendor_knowledgebases = VendorKnowledgeBaseService(
+            self._api_domain, self._token, verify_ssl=self._verify_ssl
+        )
+        self.vendor_guardrails = VendorGuardrailService(
+            self._api_domain, self._token, verify_ssl=self._verify_ssl
+        )
 
     @property
     def token(self) -> str:
@@ -96,6 +120,7 @@ class CodeMieClient:
             "localhost",
             "127.0.0.1",
             "0.0.0.0",
+            "192.168",
         ]
         return any(pattern in domain_lower for pattern in localhost_patterns)
 
@@ -127,4 +152,22 @@ class CodeMieClient:
         self.conversations = ConversationService(
             self._api_domain, self._token, verify_ssl=self._verify_ssl
         )
+        self.files = FileOperationService(
+            self._api_domain, self._token, verify_ssl=self._verify_ssl
+        )
+        self.webhook = WebhookService(
+            self._api_domain, self._token, verify_ssl=self._verify_ssl
+        )
+        self.vendor_assistants = VendorAssistantService(
+            self._api_domain, self._token, verify_ssl=self._verify_ssl
+        )
+        self.vendor_workflows = VendorWorkflowService(
+            self._api_domain, self._token, verify_ssl=self._verify_ssl
+        )
+        self.vendor_knowledgebases = VendorKnowledgeBaseService(
+            self._api_domain, self._token, verify_ssl=self._verify_ssl
+        )
+        self.vendor_guardrails = VendorGuardrailService(
+            self._api_domain, self._token, verify_ssl=self._verify_ssl
+        )
         return self._token