codemie-sdk-python 0.1.343__tar.gz → 0.1.345__tar.gz

{codemie_sdk_python-0.1.343 → codemie_sdk_python-0.1.345}/PKG-INFO

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

{codemie_sdk_python-0.1.343 → codemie_sdk_python-0.1.345}/pyproject.toml

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

{codemie_sdk_python-0.1.343 → codemie_sdk_python-0.1.345}/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.343 → codemie_sdk_python-0.1.345}/src/codemie_sdk/models/workflow_execution_payload.py

@@ -18,6 +18,10 @@ class WorkflowExecutionCreateRequest(BaseModel):
     conversation_id: Optional[str] = Field(
         None, description="Conversation ID for workflow chat mode"
     )
+    session_id: Optional[str] = Field(
+        None,
+        description="Session identifier for Langfuse tracing. If not provided, execution_id will be used as fallback.",
+    )
     propagate_headers: bool = Field(
         default=False,
         description="Enable propagation of X-* HTTP headers to MCP servers during tool execution",

{codemie_sdk_python-0.1.343 → codemie_sdk_python-0.1.345}/src/codemie_sdk/services/workflow.py

@@ -135,6 +135,7 @@ class WorkflowService:
         workflow_id: str,
         user_input: Optional[str] = None,
         file_name: Optional[str] = None,
+        session_id: Optional[str] = None,
         propagate_headers: bool = False,
         headers: Optional[dict[str, str]] = None,
         tags: Optional[List[str]] = None,
@@ -145,6 +146,7 @@ class WorkflowService:
             workflow_id: ID of the workflow to run
             user_input: Optional user input for the workflow execution
             file_name: Optional file name for the workflow execution
+            session_id: Optional session identifier for Langfuse tracing. If not provided, execution_id will be used.
             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.
@@ -155,6 +157,7 @@ class WorkflowService:
         return self.executions(workflow_id).create(
             user_input=user_input,
             file_name=file_name,
+            session_id=session_id,
             propagate_headers=propagate_headers,
             headers=headers,
             tags=tags,

{codemie_sdk_python-0.1.343 → codemie_sdk_python-0.1.345}/src/codemie_sdk/services/workflow_execution.py

@@ -50,6 +50,7 @@ class WorkflowExecutionService:
         user_input: Optional[str] = None,
         file_name: Optional[str] = None,
         conversation_id: Optional[str] = None,
+        session_id: Optional[str] = None,
         propagate_headers: bool = False,
         headers: Optional[dict[str, str]] = None,
         tags: Optional[list[str]] = None,
@@ -60,6 +61,7 @@ class WorkflowExecutionService:
             user_input: Optional input data for the workflow execution.
             file_name: Optional file name associated with the workflow execution.
             conversation_id: Optional conversation ID for workflow chat mode.
+            session_id: Optional session identifier for Langfuse tracing. If not provided, execution_id will be used.
             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.
@@ -71,6 +73,7 @@ class WorkflowExecutionService:
             user_input=user_input,
             file_name=file_name,
             conversation_id=conversation_id,
+            session_id=session_id,
             propagate_headers=propagate_headers,
             tags=tags,
         )