kailash 0.3.0__py3-none-any.whl → 0.3.1__py3-none-any.whl

kailash/api/studio.py

@@ -13,9 +13,9 @@ import json
 import logging
 import os
 import uuid
-from datetime import datetime, timezone
+from datetime import UTC, datetime
 from pathlib import Path
-from typing import Any, Dict, List, Optional
+from typing import Any
 
 import uvicorn
 from fastapi import FastAPI, HTTPException, Query, WebSocket, WebSocketDisconnect
@@ -49,25 +49,25 @@ class NodeDefinition(BaseModel):
     category: str
     name: str
     description: str
-    parameters: List[Dict[str, Any]]
-    inputs: List[Dict[str, Any]]
-    outputs: List[Dict[str, Any]]
+    parameters: list[dict[str, Any]]
+    inputs: list[dict[str, Any]]
+    outputs: list[dict[str, Any]]
 
 
 class WorkflowCreate(BaseModel):
     """Workflow creation request"""
 
     name: str
-    description: Optional[str] = None
-    definition: Dict[str, Any]
+    description: str | None = None
+    definition: dict[str, Any]
 
 
 class WorkflowUpdate(BaseModel):
     """Workflow update request"""
 
-    name: Optional[str] = None
-    description: Optional[str] = None
-    definition: Optional[Dict[str, Any]] = None
+    name: str | None = None
+    description: str | None = None
+    definition: dict[str, Any] | None = None
 
 
 class WorkflowResponse(BaseModel):
@@ -75,8 +75,8 @@ class WorkflowResponse(BaseModel):
 
     id: str
     name: str
-    description: Optional[str]
-    definition: Dict[str, Any]
+    description: str | None
+    definition: dict[str, Any]
     created_at: datetime
     updated_at: datetime
 
@@ -84,7 +84,7 @@ class WorkflowResponse(BaseModel):
 class ExecutionRequest(BaseModel):
     """Workflow execution request"""
 
-    parameters: Optional[Dict[str, Any]] = None
+    parameters: dict[str, Any] | None = None
 
 
 class ExecutionResponse(BaseModel):
@@ -94,16 +94,16 @@ class ExecutionResponse(BaseModel):
     workflow_id: str
     status: str
     started_at: datetime
-    completed_at: Optional[datetime]
-    result: Optional[Dict[str, Any]]
-    error: Optional[str]
+    completed_at: datetime | None
+    result: dict[str, Any] | None
+    error: str | None
 
 
 class WorkflowImportRequest(BaseModel):
     """Workflow import request"""
 
     name: str
-    description: Optional[str] = None
+    description: str | None = None
     format: str = Field(..., pattern="^(yaml|json|python)$")
     content: str
 
@@ -113,10 +113,10 @@ class WorkflowImportResponse(BaseModel):
 
     id: str
     name: str
-    description: Optional[str]
-    definition: Dict[str, Any]
+    description: str | None
+    definition: dict[str, Any]
     created_at: datetime
-    warnings: List[str] = []
+    warnings: list[str] = []
 
 
 class WorkflowStudioAPI:
@@ -135,8 +135,8 @@ class WorkflowStudioAPI:
         self.setup_middleware()
         self.setup_routes()
         self.setup_storage()
-        self.active_executions: Dict[str, asyncio.Task] = {}
-        self.websocket_connections: Dict[str, List[WebSocket]] = {}
+        self.active_executions: dict[str, asyncio.Task] = {}
+        self.websocket_connections: dict[str, list[WebSocket]] = {}
 
         # Register custom nodes on startup
         self.app.add_event_handler("startup", self._register_custom_nodes)
@@ -231,7 +231,7 @@ class WorkflowStudioAPI:
             return {"status": "healthy", "tenant_id": self.tenant_id}
 
         # Node discovery endpoints
-        @self.app.get("/api/nodes", response_model=Dict[str, List[NodeDefinition]])
+        @self.app.get("/api/nodes", response_model=dict[str, list[NodeDefinition]])
         async def list_nodes():
             """List all available nodes grouped by category"""
             try:
@@ -385,7 +385,7 @@ class WorkflowStudioAPI:
 
         # Add alias for backward compatibility
         @self.app.get(
-            "/api/nodes/discover", response_model=Dict[str, List[NodeDefinition]]
+            "/api/nodes/discover", response_model=dict[str, list[NodeDefinition]]
         )
         async def discover_nodes():
             """Alias for list_nodes endpoint for backward compatibility"""
@@ -420,7 +420,7 @@ class WorkflowStudioAPI:
             )
 
         # Workflow management endpoints
-        @self.app.get("/api/workflows", response_model=List[WorkflowResponse])
+        @self.app.get("/api/workflows", response_model=list[WorkflowResponse])
         async def list_workflows(
             limit: int = Query(100, ge=1, le=1000), offset: int = Query(0, ge=0)
         ):
@@ -434,7 +434,7 @@ class WorkflowStudioAPI:
 
             for workflow_file in workflow_files[offset : offset + limit]:
                 try:
-                    with open(workflow_file, "r") as f:
+                    with open(workflow_file) as f:
                         data = json.load(f)
                         workflows.append(WorkflowResponse(**data))
                 except Exception as e:
@@ -446,7 +446,7 @@ class WorkflowStudioAPI:
         async def create_workflow(workflow: WorkflowCreate):
             """Create a new workflow"""
             workflow_id = str(uuid.uuid4())
-            now = datetime.now(timezone.utc)
+            now = datetime.now(UTC)
 
             workflow_data = {
                 "id": workflow_id,
@@ -471,7 +471,7 @@ class WorkflowStudioAPI:
             if not workflow_file.exists():
                 raise HTTPException(status_code=404, detail="Workflow not found")
 
-            with open(workflow_file, "r") as f:
+            with open(workflow_file) as f:
                 data = json.load(f)
 
             return WorkflowResponse(**data)
@@ -484,7 +484,7 @@ class WorkflowStudioAPI:
                 raise HTTPException(status_code=404, detail="Workflow not found")
 
             # Load existing workflow
-            with open(workflow_file, "r") as f:
+            with open(workflow_file) as f:
                 data = json.load(f)
 
             # Update fields
@@ -495,7 +495,7 @@ class WorkflowStudioAPI:
             if update.definition is not None:
                 data["definition"] = update.definition
 
-            data["updated_at"] = datetime.now(timezone.utc).isoformat()
+            data["updated_at"] = datetime.now(UTC).isoformat()
 
             # Save updated workflow
             with open(workflow_file, "w") as f:
@@ -524,7 +524,7 @@ class WorkflowStudioAPI:
             if not workflow_file.exists():
                 raise HTTPException(status_code=404, detail="Workflow not found")
 
-            with open(workflow_file, "r") as f:
+            with open(workflow_file) as f:
                 workflow_data = json.load(f)
 
             # Create execution record
@@ -533,7 +533,7 @@ class WorkflowStudioAPI:
                 "id": execution_id,
                 "workflow_id": workflow_id,
                 "status": "running",
-                "started_at": datetime.now(timezone.utc).isoformat(),
+                "started_at": datetime.now(UTC).isoformat(),
                 "completed_at": None,
                 "result": None,
                 "error": None,
@@ -560,7 +560,7 @@ class WorkflowStudioAPI:
             except Exception as e:
                 execution_data["status"] = "failed"
                 execution_data["error"] = str(e)
-                execution_data["completed_at"] = datetime.now(timezone.utc).isoformat()
+                execution_data["completed_at"] = datetime.now(UTC).isoformat()
 
                 with open(execution_file, "w") as f:
                     json.dump(execution_data, f, indent=2)
@@ -576,7 +576,7 @@ class WorkflowStudioAPI:
             if not execution_file.exists():
                 raise HTTPException(status_code=404, detail="Execution not found")
 
-            with open(execution_file, "r") as f:
+            with open(execution_file) as f:
                 data = json.load(f)
 
             return ExecutionResponse(**data)
@@ -602,7 +602,7 @@ class WorkflowStudioAPI:
                         break
 
                     # Send current status
-                    with open(execution_file, "r") as f:
+                    with open(execution_file) as f:
                         data = json.load(f)
                     await websocket.send_json(data)
 
@@ -633,7 +633,7 @@ class WorkflowStudioAPI:
             if not workflow_file.exists():
                 raise HTTPException(status_code=404, detail="Workflow not found")
 
-            with open(workflow_file, "r") as f:
+            with open(workflow_file) as f:
                 workflow_data = json.load(f)
 
             # Create workflow from definition
@@ -686,7 +686,7 @@ class WorkflowStudioAPI:
                     warnings.append(f"Workflow validation warning: {str(e)}")
 
                 # Create workflow record
-                now = datetime.now(timezone.utc)
+                now = datetime.now(UTC)
                 workflow_data = {
                     "id": workflow_id,
                     "name": request.name,
@@ -718,7 +718,7 @@ class WorkflowStudioAPI:
         execution_id: str,
         workflow: Workflow,
         runtime: LocalRuntime,
-        parameters: Dict[str, Any],
+        parameters: dict[str, Any],
     ):
         """Execute workflow asynchronously and update status"""
         execution_file = self.executions_path / f"{execution_id}.json"
@@ -728,11 +728,11 @@ class WorkflowStudioAPI:
             result, run_id = runtime.execute(workflow, parameters=parameters)
 
             # Update execution record
-            with open(execution_file, "r") as f:
+            with open(execution_file) as f:
                 execution_data = json.load(f)
 
             execution_data["status"] = "completed"
-            execution_data["completed_at"] = datetime.now(timezone.utc).isoformat()
+            execution_data["completed_at"] = datetime.now(UTC).isoformat()
             execution_data["result"] = result
 
             with open(execution_file, "w") as f:
@@ -743,11 +743,11 @@ class WorkflowStudioAPI:
 
         except Exception as e:
             # Update execution record with error
-            with open(execution_file, "r") as f:
+            with open(execution_file) as f:
                 execution_data = json.load(f)
 
             execution_data["status"] = "failed"
-            execution_data["completed_at"] = datetime.now(timezone.utc).isoformat()
+            execution_data["completed_at"] = datetime.now(UTC).isoformat()
             execution_data["error"] = str(e)
 
             with open(execution_file, "w") as f:
@@ -761,7 +761,7 @@ class WorkflowStudioAPI:
             if execution_id in self.active_executions:
                 del self.active_executions[execution_id]
 
-    async def _notify_websocket_clients(self, execution_id: str, data: Dict[str, Any]):
+    async def _notify_websocket_clients(self, execution_id: str, data: dict[str, Any]):
         """Notify all WebSocket clients watching this execution"""
         if execution_id in self.websocket_connections:
             for websocket in self.websocket_connections[execution_id]:
@@ -851,7 +851,7 @@ class WorkflowStudioAPI:
 
         return "\n".join(lines)
 
-    def _parse_python_workflow(self, python_code: str) -> Dict[str, Any]:
+    def _parse_python_workflow(self, python_code: str) -> dict[str, Any]:
         """Parse Python code to extract workflow definition.
 
         This is a simplified parser that extracts workflow structure from Python code.
@@ -868,7 +868,7 @@ class WorkflowStudioAPI:
             },
         }
 
-    def _format_config(self, config: Dict[str, Any]) -> str:
+    def _format_config(self, config: dict[str, Any]) -> str:
         """Format config dict as Python code"""
         if not config:
             return ""

kailash/api/workflow_api.py

@@ -8,7 +8,7 @@ workflow as a REST API with minimal configuration.
 import asyncio
 from contextlib import asynccontextmanager
 from enum import Enum
-from typing import Any, Dict, List, Optional, Union
+from typing import Any
 
 import uvicorn
 from fastapi import BackgroundTasks, FastAPI, HTTPException
@@ -31,8 +31,8 @@ class ExecutionMode(str, Enum):
 class WorkflowRequest(BaseModel):
     """Base request model for workflow execution."""
 
-    inputs: Dict[str, Any] = Field(..., description="Input data for workflow nodes")
-    config: Optional[Dict[str, Any]] = Field(
+    inputs: dict[str, Any] = Field(..., description="Input data for workflow nodes")
+    config: dict[str, Any] | None = Field(
         None, description="Node configuration overrides"
     )
     mode: ExecutionMode = Field(ExecutionMode.SYNC, description="Execution mode")
@@ -41,7 +41,7 @@ class WorkflowRequest(BaseModel):
 class WorkflowResponse(BaseModel):
     """Base response model for workflow execution."""
 
-    outputs: Dict[str, Any] = Field(..., description="Output data from workflow nodes")
+    outputs: dict[str, Any] = Field(..., description="Output data from workflow nodes")
     execution_time: float = Field(..., description="Execution time in seconds")
     workflow_id: str = Field(..., description="Workflow identifier")
     version: str = Field(..., description="Workflow version")
@@ -63,7 +63,7 @@ class WorkflowAPI:
 
     def __init__(
         self,
-        workflow: Union[WorkflowBuilder, Workflow],
+        workflow: WorkflowBuilder | Workflow,
         app_name: str = "Kailash Workflow API",
         version: str = "1.0.0",
         description: str = "API wrapper for Kailash workflow execution",
@@ -102,7 +102,7 @@ class WorkflowAPI:
         self._setup_routes()
 
         # Cache for async executions
-        self._execution_cache: Dict[str, Dict[str, Any]] = {}
+        self._execution_cache: dict[str, dict[str, Any]] = {}
 
     @asynccontextmanager
     async def _lifespan(self, app: FastAPI):
@@ -317,12 +317,12 @@ class HierarchicalRAGAPI(WorkflowAPI):
 
         class RAGResponse(BaseModel):
             answer: str
-            sources: List[Dict[str, Any]]
+            sources: list[dict[str, Any]]
             query: str
             execution_time: float
 
         @self.app.post("/documents")
-        async def add_documents(documents: List[Document]):
+        async def add_documents(documents: list[Document]):
             """Add documents to the knowledge base."""
             # This would integrate with document storage
             return {"message": f"Added {len(documents)} documents"}

kailash/cli/commands.py

@@ -4,7 +4,6 @@ import json
 import logging
 import sys
 from pathlib import Path
-from typing import Optional
 
 import click
 
@@ -77,7 +76,7 @@ def init(name: str, template: str):
 @click.option("--params", "-p", help="JSON file with parameter overrides")
 @click.option("--debug", is_flag=True, help="Enable debug mode")
 @click.option("--no-tracking", is_flag=True, help="Disable task tracking")
-def run(workflow_file: str, params: Optional[str], debug: bool, no_tracking: bool):
+def run(workflow_file: str, params: str | None, debug: bool, no_tracking: bool):
     """Run a workflow locally."""
     try:
         # Validate workflow file exists
@@ -97,7 +96,7 @@ def run(workflow_file: str, params: Optional[str], debug: bool, no_tracking: boo
         parameters = {}
         if params:
             try:
-                with open(params, "r") as f:
+                with open(params) as f:
                     parameters = json.load(f)
             except FileNotFoundError:
                 raise CLIException(f"Parameters file not found: {params}")
@@ -190,7 +189,7 @@ def validate(workflow_file: str):
     "--format", default="yaml", type=click.Choice(["yaml", "json", "manifest"])
 )
 @click.option("--registry", help="Container registry URL")
-def export(workflow_file: str, output_file: str, format: str, registry: Optional[str]):
+def export(workflow_file: str, output_file: str, format: str, registry: str | None):
     """Export workflow to Kailash format."""
     try:
         # Validate workflow file exists
@@ -233,14 +232,13 @@ def export(workflow_file: str, output_file: str, format: str, registry: Optional
 @cli.group()
 def tasks():
     """Task tracking commands."""
-    pass
 
 
 @tasks.command("list")
 @click.option("--workflow", help="Filter by workflow name")
 @click.option("--status", help="Filter by status")
 @click.option("--limit", default=10, help="Number of runs to show")
-def list_tasks(workflow: Optional[str], status: Optional[str], limit: int):
+def list_tasks(workflow: str | None, status: str | None, limit: int):
     """List workflow runs."""
     try:
         task_manager = TaskManager()
@@ -380,7 +378,6 @@ def clear_tasks():
 @cli.group()
 def nodes():
     """Node management commands."""
-    pass
 
 
 @nodes.command("list")
@@ -497,7 +494,7 @@ def _load_python_workflow(workflow_file: str) -> Workflow:
     try:
         # Read and execute Python file
         global_scope = {}
-        with open(workflow_file, "r") as f:
+        with open(workflow_file) as f:
             code = f.read()
 
         exec(code, global_scope)