codemie-sdk-python 0.1.52__py3-none-any.whl → 0.1.258__py3-none-any.whl

codemie_sdk/models/vendor_knowledgebase.py

@@ -0,0 +1,151 @@
+"""Models for vendor knowledge base settings."""
+
+from datetime import datetime
+from enum import Enum
+from typing import Optional, List
+
+from pydantic import BaseModel, ConfigDict, Field
+
+from .vendor_assistant import PaginationInfo, TokenPagination
+
+
+class VendorKnowledgeBaseSetting(BaseModel):
+    """Model representing a vendor knowledge base setting."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    setting_id: str = Field(..., description="Unique identifier for the setting")
+    setting_name: str = Field(..., description="Name of the setting")
+    project: str = Field(..., description="Project associated with the setting")
+    entities: List[str] = Field(
+        default_factory=list, description="List of entities associated with the setting"
+    )
+    invalid: Optional[bool] = Field(None, description="Whether the setting is invalid")
+    error: Optional[str] = Field(
+        None, description="Error message if the setting is invalid"
+    )
+
+
+class VendorKnowledgeBaseSettingsResponse(BaseModel):
+    """Response model for vendor knowledge base settings list."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    data: List[VendorKnowledgeBaseSetting] = Field(
+        ..., description="List of vendor knowledge base settings"
+    )
+    pagination: PaginationInfo = Field(..., description="Pagination information")
+
+
+class VendorKnowledgeBaseStatus(str, Enum):
+    """Status of vendor knowledge base."""
+
+    PREPARED = "PREPARED"
+    NOT_PREPARED = "NOT_PREPARED"
+
+
+class VendorKnowledgeBase(BaseModel):
+    """Model representing a vendor knowledge base."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    id: str = Field(..., description="Unique identifier for the knowledge base")
+    name: str = Field(..., description="Name of the knowledge base")
+    status: VendorKnowledgeBaseStatus = Field(
+        ..., description="Status of the knowledge base"
+    )
+    description: Optional[str] = Field(
+        None, description="Description of the knowledge base"
+    )
+    updatedAt: datetime = Field(
+        ..., description="Last update timestamp", alias="updatedAt"
+    )
+    aiRunId: Optional[str] = Field(
+        None,
+        description="AI run ID if the knowledge base is installed",
+        alias="aiRunId",
+    )
+
+
+class VendorKnowledgeBasesResponse(BaseModel):
+    """Response model for vendor knowledge bases list."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    data: List[VendorKnowledgeBase] = Field(
+        ..., description="List of vendor knowledge bases"
+    )
+    pagination: TokenPagination = Field(
+        ..., description="Token-based pagination information"
+    )
+
+
+class VendorKnowledgeBaseDetail(BaseModel):
+    """Model representing detailed information about a vendor knowledge base."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    id: str = Field(..., description="Unique identifier for the knowledge base")
+    name: str = Field(..., description="Name of the knowledge base")
+    description: Optional[str] = Field(
+        None, description="Description of the knowledge base"
+    )
+    type: str = Field(..., description="Type of knowledge base (e.g., VECTOR)")
+    status: VendorKnowledgeBaseStatus = Field(
+        ..., description="Status of the knowledge base"
+    )
+    embeddingModel: str = Field(
+        ...,
+        description="Embedding model used by the knowledge base",
+        alias="embeddingModel",
+    )
+    kendraIndexArn: Optional[str] = Field(
+        None, description="Kendra index ARN if applicable", alias="kendraIndexArn"
+    )
+    createdAt: datetime = Field(
+        ..., description="Creation timestamp", alias="createdAt"
+    )
+    updatedAt: datetime = Field(
+        ..., description="Last update timestamp", alias="updatedAt"
+    )
+    aiRunId: Optional[str] = Field(
+        None,
+        description="AI run ID if the knowledge base is installed",
+        alias="aiRunId",
+    )
+
+
+class VendorKnowledgeBaseInstallRequest(BaseModel):
+    """Model for a single knowledge base installation request."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    id: str = Field(..., description="Knowledge base ID to install")
+    setting_id: str = Field(..., description="Vendor setting ID")
+
+
+class VendorKnowledgeBaseInstallSummary(BaseModel):
+    """Model for knowledge base installation summary."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    knowledgeBaseId: str = Field(..., description="Installed knowledge base ID")
+    aiRunId: str = Field(..., description="AI run ID for the installation")
+
+
+class VendorKnowledgeBaseInstallResponse(BaseModel):
+    """Response model for knowledge base installation."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    summary: List[VendorKnowledgeBaseInstallSummary] = Field(
+        ..., description="List of installation summaries"
+    )
+
+
+class VendorKnowledgeBaseUninstallResponse(BaseModel):
+    """Response model for knowledge base uninstallation."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    success: bool = Field(..., description="Whether the uninstallation was successful")

codemie_sdk/models/vendor_workflow.py

@@ -0,0 +1,145 @@
+"""Models for vendor workflow settings."""
+
+from datetime import datetime
+from enum import Enum
+from typing import Optional, List
+
+from pydantic import BaseModel, ConfigDict, Field
+
+from .vendor_assistant import PaginationInfo, TokenPagination
+
+
+class VendorWorkflowSetting(BaseModel):
+    """Model representing a vendor workflow setting."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    setting_id: str = Field(..., description="Unique identifier for the setting")
+    setting_name: str = Field(..., description="Name of the setting")
+    project: str = Field(..., description="Project associated with the setting")
+    entities: List[str] = Field(
+        default_factory=list, description="List of entities associated with the setting"
+    )
+    invalid: Optional[bool] = Field(None, description="Whether the setting is invalid")
+    error: Optional[str] = Field(
+        None, description="Error message if the setting is invalid"
+    )
+
+
+class VendorWorkflowSettingsResponse(BaseModel):
+    """Response model for vendor workflow settings list."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    data: List[VendorWorkflowSetting] = Field(
+        ..., description="List of vendor workflow settings"
+    )
+    pagination: PaginationInfo = Field(..., description="Pagination information")
+
+
+class VendorWorkflowStatus(str, Enum):
+    """Status of vendor workflow."""
+
+    PREPARED = "PREPARED"
+    NOT_PREPARED = "NOT_PREPARED"
+
+
+class VendorWorkflow(BaseModel):
+    """Model representing a vendor workflow."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    id: str = Field(..., description="Unique identifier for the workflow")
+    name: str = Field(..., description="Name of the workflow")
+    status: VendorWorkflowStatus = Field(..., description="Status of the workflow")
+    description: Optional[str] = Field(None, description="Description of the workflow")
+    version: str = Field(..., description="Version of the workflow")
+    createdAt: datetime = Field(
+        ..., description="Creation timestamp", alias="createdAt"
+    )
+    updatedAt: datetime = Field(
+        ..., description="Last update timestamp", alias="updatedAt"
+    )
+
+
+class VendorWorkflowsResponse(BaseModel):
+    """Response model for vendor workflows list."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    data: List[VendorWorkflow] = Field(..., description="List of vendor workflows")
+    pagination: TokenPagination = Field(
+        ..., description="Token-based pagination information"
+    )
+
+
+class VendorWorkflowAlias(BaseModel):
+    """Model representing a vendor workflow alias."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    id: str = Field(..., description="Unique identifier for the alias")
+    name: str = Field(..., description="Name of the alias")
+    status: VendorWorkflowStatus = Field(..., description="Status of the alias")
+    description: Optional[str] = Field(None, description="Description of the alias")
+    version: str = Field(..., description="Version of the alias")
+    createdAt: datetime = Field(
+        ..., description="Creation timestamp", alias="createdAt"
+    )
+    updatedAt: datetime = Field(
+        ..., description="Last update timestamp", alias="updatedAt"
+    )
+    aiRunId: Optional[str] = Field(
+        None, description="AI run ID if the alias is installed", alias="aiRunId"
+    )
+
+
+class VendorWorkflowAliasesResponse(BaseModel):
+    """Response model for vendor workflow aliases list."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    data: List[VendorWorkflowAlias] = Field(
+        ..., description="List of vendor workflow aliases"
+    )
+    pagination: TokenPagination = Field(
+        ..., description="Token-based pagination information"
+    )
+
+
+class VendorWorkflowInstallRequest(BaseModel):
+    """Model for a single workflow installation request."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    id: str = Field(..., description="Workflow ID to install")
+    flowAliasId: str = Field(..., description="Flow alias ID to use for the workflow")
+    setting_id: str = Field(..., description="Vendor setting ID")
+
+
+class VendorWorkflowInstallSummary(BaseModel):
+    """Model for workflow installation summary."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    flowId: str = Field(..., description="Installed workflow ID")
+    flowAliasId: str = Field(..., description="Flow alias ID used for installation")
+    aiRunId: str = Field(..., description="AI run ID for the installation")
+
+
+class VendorWorkflowInstallResponse(BaseModel):
+    """Response model for workflow installation."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    summary: List[VendorWorkflowInstallSummary] = Field(
+        ..., description="List of installation summaries"
+    )
+
+
+class VendorWorkflowUninstallResponse(BaseModel):
+    """Response model for workflow uninstallation."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    success: bool = Field(..., description="Whether the uninstallation was successful")

codemie_sdk/models/workflow.py

@@ -59,9 +59,9 @@ class Workflow(BaseModel):
     mode: WorkflowMode = WorkflowMode.SEQUENTIAL
     shared: bool = False
     icon_url: Optional[str] = None
-    created_date: datetime = Field(None, alias="date")
-    update_date: datetime = Field(None)
-    created_by: User
+    created_date: Optional[datetime] = Field(None, alias="date")
+    update_date: Optional[datetime] = Field(None)
+    created_by: Optional[User] = None
 
 
 class ExecutionStatus(str, Enum):
@@ -84,6 +84,6 @@ class WorkflowExecution(BaseModel):
     status: ExecutionStatus = Field(alias="overall_status")
     created_date: datetime = Field(alias="date")
     prompt: str
-    updated_date: Optional[datetime] = None
+    updated_date: Optional[datetime] = Field(alias="update_date")
     created_by: User
     tokens_usage: Optional[TokensUsage] = None

codemie_sdk/models/workflow_execution_payload.py

@@ -0,0 +1,21 @@
+"""Workflow execution payload models."""
+
+from typing import Optional, Union
+from pydantic import BaseModel, ConfigDict, Field
+
+
+class WorkflowExecutionCreateRequest(BaseModel):
+    """Request model for workflow execution creation."""
+
+    model_config = ConfigDict(populate_by_name=True)
+
+    user_input: Optional[Union[str, dict, list, int, float, bool]] = Field(
+        None, description="User input for the workflow execution"
+    )
+    file_name: Optional[str] = Field(
+        None, description="File name associated with the workflow execution"
+    )
+    propagate_headers: bool = Field(
+        default=False,
+        description="Enable propagation of X-* HTTP headers to MCP servers during tool execution",
+    )

codemie_sdk/models/workflow_state.py

@@ -2,7 +2,7 @@
 
 from datetime import datetime
 from enum import Enum
-from typing import Optional
+from typing import List, Optional
 
 from pydantic import BaseModel, ConfigDict
 
@@ -24,9 +24,11 @@ class WorkflowExecutionStateThought(BaseModel):
     model_config = ConfigDict(populate_by_name=True)
 
     id: str
-    text: str
-    created_at: datetime
+    execution_state_id: str
     parent_id: Optional[str] = None
+    author_name: str
+    author_type: str
+    date: datetime
 
 
 class WorkflowExecutionState(BaseModel):
@@ -41,6 +43,7 @@ class WorkflowExecutionState(BaseModel):
     status: WorkflowExecutionStatusEnum = WorkflowExecutionStatusEnum.NOT_STARTED
     started_at: Optional[datetime] = None
     completed_at: Optional[datetime] = None
+    thoughts: Optional[List[WorkflowExecutionStateThought]] = None
 
 
 class WorkflowExecutionStateOutput(BaseModel):

codemie_sdk/models/workflow_thoughts.py

@@ -0,0 +1,26 @@
+"""Workflow execution thoughts models."""
+
+from datetime import datetime
+from typing import List, Optional
+
+from pydantic import BaseModel, ConfigDict
+
+
+class WorkflowExecutionThought(BaseModel):
+    """Model for workflow execution thought."""
+
+    model_config = ConfigDict(populate_by_name=True)
+
+    id: str
+    execution_state_id: str
+    parent_id: Optional[str] = None
+    author_name: str
+    author_type: str
+    input_text: str
+    content: str
+    date: datetime
+    children: List["WorkflowExecutionThought"] = []
+
+
+# Update forward references for recursive model
+WorkflowExecutionThought.model_rebuild()