codemie-sdk-python 0.1.515__tar.gz → 0.1.516__tar.gz

{codemie_sdk_python-0.1.515 → codemie_sdk_python-0.1.516}/PKG-INFO

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

{codemie_sdk_python-0.1.515 → codemie_sdk_python-0.1.516}/pyproject.toml

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

{codemie_sdk_python-0.1.515 → codemie_sdk_python-0.1.516}/src/codemie_sdk/models/workflow.py

@@ -100,3 +100,73 @@ class WorkflowEvaluationRequest(BaseModel):
     max_concurrency: int = Field(
         default=1, ge=1, le=5, description="Maximum number of concurrent evaluations"
     )
+
+
+class WorkflowGeneratorRequest(BaseModel):
+    """Request model for generating a workflow from a natural language description.
+
+    Maps to the backend's ``WorkflowGeneratorRequest``
+    (``src/codemie/rest_api/models/workflow_generator.py``), where the wire field
+    for the natural language description is ``text`` (not ``nl_query`` — that name
+    is only used internally as the service-layer parameter).
+    """
+
+    text: str = Field(..., min_length=1)
+    llm_model: str | None = None
+    persist: bool = False
+    guardrail_ids: list[str] | None = None
+
+
+class WorkflowGeneratedState(BaseModel):
+    """A single generated state (node) inside a generated workflow config."""
+
+    model_config = ConfigDict(extra="allow")
+
+    id: str | None = None
+    assistant_id: str | None = None
+    tool_id: str | None = None
+    custom_node_id: str | None = None
+    task: str | None = None
+    next: dict | None = None
+
+
+class WorkflowGeneratedAssistant(BaseModel):
+    """A single generated assistant reference inside a generated workflow config."""
+
+    model_config = ConfigDict(extra="allow")
+
+    id: str | None = None
+    assistant_id: str | None = None
+
+
+class WorkflowGeneratorConfig(BaseModel):
+    """The generated workflow configuration returned by the generator.
+
+    Mirrors the backend's ``CreateWorkflowRequest`` shape
+    (``src/codemie/core/workflow_models/workflow_models.py``). Uses
+    ``extra="allow"`` since the backend model carries additional fields (e.g.
+    ``guardrail_assignments``) that are not required for test assertions.
+    """
+
+    model_config = ConfigDict(populate_by_name=True, extra="allow")
+
+    name: str
+    description: str | None = None
+    project: str | None = None
+    mode: WorkflowMode = WorkflowMode.SEQUENTIAL
+    yaml_config: str | None = None
+    shared: bool = True
+    assistants: list[WorkflowGeneratedAssistant] = Field(default_factory=list)
+    tools: list[dict] = Field(default_factory=list)
+    states: list[WorkflowGeneratedState] = Field(default_factory=list)
+    supervisor_prompt: str | None = ""
+    meta_config: str | None = None
+
+
+class WorkflowGeneratorResponse(BaseModel):
+    """Response model for ``POST /v1/workflows/generate``."""
+
+    model_config = ConfigDict(populate_by_name=True)
+
+    workflow_config: WorkflowGeneratorConfig
+    workflow_id: str | None = None

{codemie_sdk_python-0.1.515 → codemie_sdk_python-0.1.516}/src/codemie_sdk/services/workflow.py

@@ -11,6 +11,8 @@ from ..models.workflow import (
     WorkflowCreateRequest,
     WorkflowUpdateRequest,
     WorkflowEvaluationRequest,
+    WorkflowGeneratorRequest,
+    WorkflowGeneratorResponse,
     Workflow,
 )
 from ..utils import ApiRequestHandler, TokenSource
@@ -56,6 +58,45 @@ class WorkflowService:
         """
         return self._api.post("/v1/workflows", dict, json_data=request.model_dump())
 
+    def generate(
+        self,
+        text: str,
+        llm_model: str | None = None,
+        persist: bool = False,
+        guardrail_ids: list[str] | None = None,
+    ) -> WorkflowGeneratorResponse:
+        """Generate a workflow configuration from a natural language description.
+
+        Args:
+            text: Natural language description of the desired workflow.
+            llm_model: Optional LLM model to use for generation. Falls back to
+                the platform default (``WORKFLOW_GENERATOR_LLM_MODEL`` or the
+                global default LLM) when not provided.
+            persist: When True, the generated workflow is validated
+                (``WorkflowExecutor.validate_workflow``) and saved on the
+                backend, and the response includes ``workflow_id``. When False
+                (default), only the in-memory ``workflow_config`` is returned
+                for client-side review/edit before an explicit
+                ``create_workflow`` call.
+            guardrail_ids: Optional guardrail IDs to attach to the generated
+                workflow's guardrail assignments.
+
+        Returns:
+            WorkflowGeneratorResponse containing the generated ``workflow_config``
+            and, when ``persist=True``, the created ``workflow_id``.
+        """
+        request = WorkflowGeneratorRequest(
+            text=text,
+            llm_model=llm_model,
+            persist=persist,
+            guardrail_ids=guardrail_ids,
+        )
+        return self._api.post(
+            "/v1/workflows/generate",
+            WorkflowGeneratorResponse,
+            json_data=request.model_dump(exclude_none=True),
+        )
+
     def update(self, workflow_id: str, request: WorkflowUpdateRequest) -> dict:
         """Update an existing workflow.