codemie-sdk-python 0.1.609__tar.gz → 0.1.610__tar.gz

{codemie_sdk_python-0.1.609 → codemie_sdk_python-0.1.610}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: codemie-sdk-python
-Version: 0.1.609
+Version: 0.1.610
 Summary: CodeMie SDK for Python
 Author: Vadym Vlasenko
 Author-email: vadym_vlasenko@epam.com

{codemie_sdk_python-0.1.609 → codemie_sdk_python-0.1.610}/pyproject.toml

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

{codemie_sdk_python-0.1.609 → codemie_sdk_python-0.1.610}/src/codemie_sdk/models/assistant.py

@@ -181,6 +181,44 @@ class AssistantListResponse(BaseModel):
     created_by: User | None = None
 
 
+class ToolCallPolicy(str, Enum):
+    """Policy governing when tool calls require user approval."""
+
+    AUTO_APPROVE = "auto_approve"
+    APPROVE_FOR_ME = "approve_for_me"
+    ASK_FOR_APPROVAL = "ask_for_approval"
+
+
+class ToolCallAction(str, Enum):
+    """User action on a pending tool-call confirmation."""
+
+    ALLOW = "allow"
+    DENY = "deny"
+
+
+class ToolPermissionsConfig(BaseModel):
+    """Configuration for tool-call approval policy on an assistant."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    tool_call_policy: ToolCallPolicy = ToolCallPolicy.AUTO_APPROVE
+    allow_override: bool = True
+
+
+class ToolCallResumeRequest(BaseModel):
+    """Request body for POST /v1/assistants/{id}/model/tool-call/resume."""
+
+    model_config = ConfigDict(extra="ignore", use_enum_values=True)
+
+    conversation_id: str = Field(
+        description="Conversation ID that has a pending tool-call checkpoint"
+    )
+    action: ToolCallAction = Field(
+        description="User decision: 'allow' executes the tool, 'deny' rejects it"
+    )
+    stream: bool = Field(default=True, description="Enable streaming NDJSON response")
+
+
 class Assistant(AssistantBase):
     """Full assistant model with additional fields."""
 
@@ -233,6 +271,7 @@ class Assistant(AssistantBase):
         default=None,
         description="Number of times this assistant has been cloned in the Marketplace",
     )
+    tool_permissions: ToolPermissionsConfig | None = None
 
 
 class AssistantRequestBase(AssistantBase):
@@ -268,6 +307,7 @@ class AssistantRequestBase(AssistantBase):
     enabled_builtin_subagents: list[str] | None = None
     prompt_variables: list[PromptVariable] = Field(default_factory=list)
     skip_integration_validation: bool | None = Field(default=False)
+    tool_permissions: ToolPermissionsConfig | None = None
 
 
 class AssistantCreateRequest(AssistantRequestBase):

{codemie_sdk_python-0.1.609 → codemie_sdk_python-0.1.610}/src/codemie_sdk/services/assistant.py

@@ -28,6 +28,7 @@ from ..models.assistant import (
     ExportAssistantPayload,
     AssistantEvaluationRequest,
     ReactionRequest,
+    ToolCallResumeRequest,
 )
 from ..models.common import PaginationParams
 from ..utils import ApiRequestHandler, TokenSource
@@ -586,6 +587,35 @@ class AssistantService:
             f"/v1/assistants/{assistant_id}/marketplace/publish/validate", dict
         )
 
+    def resume_tool_call(
+        self,
+        assistant_id: str,
+        request: ToolCallResumeRequest,
+    ):
+        """Resume a paused tool-call after user approval or denial.
+
+        Calls POST /v1/assistants/{id}/model/tool-call/resume.
+        When stream=True, returns the raw streaming requests.Response so the
+        caller can iterate the NDJSON chunks.  When stream=False, returns a
+        parsed dict containing the final generated text.
+
+        Args:
+            assistant_id: ID of the assistant that has a pending tool call
+            request: Resume request with conversation_id, action, and stream flag
+
+        Returns:
+            requests.Response (streaming) or dict (non-streaming)
+        """
+        import requests as _requests
+
+        response = self._api.post(
+            f"/v1/assistants/{assistant_id}/model/tool-call/resume",
+            _requests.Response,
+            json_data=request.model_dump(),
+            stream=request.stream,
+        )
+        return response
+
     def react(self, assistant_id: str, request: ReactionRequest) -> dict:
         """React to an assistant with like or dislike.