codemie-sdk-python 0.1.429__tar.gz → 0.1.430__tar.gz

{codemie_sdk_python-0.1.429 → codemie_sdk_python-0.1.430}/PKG-INFO

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

{codemie_sdk_python-0.1.429 → codemie_sdk_python-0.1.430}/pyproject.toml

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

{codemie_sdk_python-0.1.429 → codemie_sdk_python-0.1.430}/src/codemie_sdk/models/assistant.py

@@ -3,7 +3,7 @@
 import uuid
 from datetime import datetime
 from enum import Enum
-from typing import Any, Type
+from typing import Any, Type, Optional, List, Union, Dict
 
 from pydantic import BaseModel, Field, ConfigDict, model_validator
 
@@ -455,6 +455,65 @@ class AssistantChatRequest(BaseModel):
     )
 
 
+class AgentMode(str, Enum):
+    """Enum for agent execution mode."""
+
+    GENERAL = "general"
+    PLAN_EXECUTE = "plan_execute"
+
+
+class VirtualAssistantChatRequest(BaseModel):
+    """Request model for ephemeral virtual assistant inference.
+
+    The assistant definition is provided inline; no database record is used.
+    History is never persisted regardless of save_history value.
+    """
+
+    model_config = ConfigDict(extra="ignore", use_enum_values=True)
+
+    # --- Assistant definition fields ---
+    system_prompt: str = Field(
+        default="", description="System prompt for the virtual assistant."
+    )
+    llm_model_type: Optional[str] = Field(
+        default=None,
+        description="LLM model identifier. Falls back to deployment default if omitted.",
+    )
+    temperature: Optional[float] = Field(default=None, ge=0.0, le=2.0)
+    top_p: Optional[float] = Field(default=None, ge=0.0, le=1.0)
+    toolkits: List[ToolKitDetails] = Field(default_factory=list)
+    context: List[Context] = Field(default_factory=list)
+    mcp_servers: List[MCPServerDetails] = Field(default_factory=list)
+    skill_ids: List[str] = Field(default_factory=list)
+    assistant_ids: List[str] = Field(default_factory=list)
+    agent_mode: AgentMode = Field(default=AgentMode.GENERAL)
+    plan_prompt: Optional[str] = Field(default=None)
+    smart_tool_selection_enabled: bool = Field(default=False)
+    prompt_variables: List[PromptVariable] = Field(default_factory=list)
+
+    # --- Chat parameters ---
+    conversation_id: Optional[str] = Field(
+        default_factory=lambda: str(uuid.uuid4()),
+        description="Conversation identifier. Auto-generated if omitted.",
+    )
+    text: Optional[str] = Field(default=None, description="User message text.")
+    content_raw: Optional[str] = Field(default="")
+    file_names: Optional[List[str]] = Field(default_factory=list)
+    history: Union[List[ChatMessage], str] = Field(default_factory=list)
+    stream: bool = Field(default=False)
+    output_schema: Optional[Union[dict, Type[BaseModel]]] = Field(default=None)
+    tools_config: Optional[List[ToolConfig]] = Field(default=None)
+    metadata: Optional[Dict[str, Any]] = Field(default=None)
+    top_k: int = Field(default=10)
+    propagate_headers: bool = Field(default=False)
+    disable_cache: Optional[bool] = Field(default=False)
+    mcp_server_single_usage: Optional[bool] = Field(default=None)
+    save_history: bool = Field(
+        default=False,
+        description="Always False for virtual assistants. Field is accepted but ignored.",
+    )
+
+
 class BaseModelResponse(BaseModel):
     """
     Model for chat response from assistant.

{codemie_sdk_python-0.1.429 → codemie_sdk_python-0.1.430}/src/codemie_sdk/services/assistant.py

@@ -5,7 +5,7 @@ from __future__ import annotations
 import inspect
 import json
 from pathlib import Path
-from typing import Union, Any, Literal
+from typing import Union, Any, Literal, Optional, Dict
 from pydantic import BaseModel
 from copy import deepcopy
 
@@ -20,6 +20,7 @@ from ..models.assistant import (
     AssistantUpdateResponse,
     ToolKitDetails,
     AssistantChatRequest,
+    VirtualAssistantChatRequest,
     BaseModelResponse,
     AssistantBase,
     Context,
@@ -434,6 +435,43 @@ class AssistantService:
 
         return response
 
+    def ask_virtual(
+        self,
+        request: VirtualAssistantChatRequest,
+        headers: Optional[Dict[str, str]] = None,
+    ) -> Union[requests.Response, BaseModelResponse]:
+        """Run inference using an inline (virtual) assistant definition.
+
+        No database assistant record is required. History is never persisted.
+
+        Args:
+            request: Virtual assistant chat request with inline assistant definition
+            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(
+            "/v1/assistants/virtual/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."""