codemie-sdk-python 0.1.342__tar.gz → 0.1.344__tar.gz

{codemie_sdk_python-0.1.342 → codemie_sdk_python-0.1.344}/PKG-INFO

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

{codemie_sdk_python-0.1.342 → codemie_sdk_python-0.1.344}/pyproject.toml

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

{codemie_sdk_python-0.1.342 → codemie_sdk_python-0.1.344}/src/codemie_sdk/models/assistant.py

@@ -200,7 +200,7 @@ class Assistant(AssistantBase):
     assistant_ids: List[str] = Field(default_factory=list)
     version_count: Optional[int] = None
     prompt_variables: Optional[List[PromptVariable]] = Field(default=None)
-    categories: List[Category] = Field(
+    categories: List[Union[str, Category]] = Field(
         default_factory=list,
         description="List of categories for marketplace classification (e.g., 'quality-assurance', 'data-analysis')",
     )

{codemie_sdk_python-0.1.342 → codemie_sdk_python-0.1.344}/src/codemie_sdk/models/workflow_execution_payload.py

@@ -22,3 +22,6 @@ class WorkflowExecutionCreateRequest(BaseModel):
         default=False,
         description="Enable propagation of X-* HTTP headers to MCP servers during tool execution",
     )
+    tags: Optional[list[str]] = Field(
+        None, description="Tags to attach to the workflow execution trace in Langfuse"
+    )

{codemie_sdk_python-0.1.342 → codemie_sdk_python-0.1.344}/src/codemie_sdk/services/workflow.py

@@ -137,6 +137,7 @@ class WorkflowService:
         file_name: Optional[str] = None,
         propagate_headers: bool = False,
         headers: Optional[dict[str, str]] = None,
+        tags: Optional[List[str]] = None,
     ) -> dict:
         """Run a workflow with optional input parameters.
 
@@ -146,6 +147,7 @@ class WorkflowService:
             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)
+            tags: Optional tags to attach to the workflow execution trace in Langfuse.
 
         Returns:
             dict: Created workflow execution details
@@ -155,6 +157,7 @@ class WorkflowService:
             file_name=file_name,
             propagate_headers=propagate_headers,
             headers=headers,
+            tags=tags,
         )
 
     def executions(self, workflow_id: str) -> WorkflowExecutionService:

{codemie_sdk_python-0.1.342 → codemie_sdk_python-0.1.344}/src/codemie_sdk/services/workflow_execution.py

@@ -52,6 +52,7 @@ class WorkflowExecutionService:
         conversation_id: Optional[str] = None,
         propagate_headers: bool = False,
         headers: Optional[dict[str, str]] = None,
+        tags: Optional[list[str]] = None,
     ) -> dict:
         """Create a new workflow execution.
 
@@ -61,6 +62,7 @@ class WorkflowExecutionService:
             conversation_id: Optional conversation ID for workflow chat mode.
             propagate_headers: Enable propagation of X-* HTTP headers to MCP servers.
             headers: Optional additional HTTP headers (e.g., X-* for MCP propagation)
+            tags: Optional tags to attach to the workflow execution trace in Langfuse.
 
         Returns:
             dict: Created workflow execution details
@@ -70,6 +72,7 @@ class WorkflowExecutionService:
             file_name=file_name,
             conversation_id=conversation_id,
             propagate_headers=propagate_headers,
+            tags=tags,
         )
         return self._api.post(
             f"/v1/workflows/{self._workflow_id}/executions",