fix: address review feedback - import ordering and docstring migration

- Move ScheduleParser import before logger assignment (PEP 8 E402)
- Migrate comprehensive docstring from base.py to shared.py
- Preserve original examples (*/6h, */30s) and doctests
- Use explicit re-export syntax to avoid noqa suppressions

Co-authored-by: Mervin Praison <[email protected]>

.env.example

@@ -1,51 +0,0 @@
-# PraisonAI Environment Variables
-# Copy this file to .env and update with your actual values
-
-# === Required ===
-OPENAI_API_KEY=sk-...          # OpenAI key for LLM calls. Get yours at https://platform.openai.com
-
-# === Required for Claw web-search tool ===
-TAVILY_API_KEY=tvly-...        # Get yours at https://app.tavily.com (free tier available)
-
-# === Optional: Claw dashboard security ===
-PRAISONAI_ALLOW_LOCAL_TOOLS=0  # Set to 1 to enable local shell tools (security risk)
-
-# === Optional: Alternative LLM Providers ===
-ANTHROPIC_API_KEY=sk-ant-...   # Anthropic Claude models
-GOOGLE_API_KEY=...             # Google Gemini models  
-GROQ_API_KEY=gsk_...           # Groq (fast inference)
-COHERE_API_KEY=...             # Cohere models
-AZURE_OPENAI_API_KEY=...       # Azure OpenAI
-AZURE_OPENAI_ENDPOINT=...      # Azure OpenAI endpoint
-
-# === Optional: Memory backends ===
-MEM0_API_KEY=...               # Mem0 memory service
-REDIS_URL=redis://localhost:6379  # Redis for state management
-
-# === Optional: Observability ===
-LANGFUSE_SECRET_KEY=sk-lf-...  # Langfuse tracing
-LANGFUSE_PUBLIC_KEY=pk-lf-...  # Langfuse public key
-
-# === Optional: Bot platforms ===
-TELEGRAM_BOT_TOKEN=...         # Telegram bot token
-DISCORD_BOT_TOKEN=...          # Discord bot token  
-SLACK_BOT_TOKEN=xoxb-...       # Slack bot token
-SLACK_APP_TOKEN=xapp-...       # Slack app token
-WHATSAPP_ACCESS_TOKEN=...      # WhatsApp bot access token
-WHATSAPP_PHONE_NUMBER_ID=...   # WhatsApp phone number ID
-
-# === Optional: Database ===
-DATABASE_URL=sqlite:///~/.praison/database.sqlite  # Database connection string
-
-# === Optional: Tool Security ===
-# ALLOWED_TOOLS=search,send_message       # Whitelist specific tools to prevent name collisions
-                                        # Useful in multi-environment setups with overlapping tool names
-                                        # Leave unset to use all tools (with collision warnings)
-                                        # Do NOT set to empty string: that's treated as configuration error
-                                        # In CI, unknown tool names in the whitelist cause strict startup failure
-# HERMES_ONLY_TOOLS=search,send_message  # (Deprecated) Backward compatibility alias for ALLOWED_TOOLS
-
-# === Optional: Development ===
-LOGLEVEL=INFO                  # Logging level (DEBUG, INFO, WARNING, ERROR)
-CHAINLIT_AUTH_SECRET=...       # Chainlit authentication secret
-DEBUG=false                    # Enable debug mode

.github/workflows/claude.yml

@@ -181,15 +181,12 @@ jobs:
 
       - name: Fetch PR branch and Setup Remote
         if: github.event.issue.pull_request
-        env:
-          PR_BRANCH: ${{ steps.check_fork.outputs.pr_branch }}
-          IS_FORK: ${{ steps.check_fork.outputs.is_fork }}
         run: |
           # Fetch PR head from base repo and put it into local branch
-          git fetch origin pull/${{ github.event.issue.number }}/head:"${PR_BRANCH}"
+          git fetch origin pull/${{ github.event.issue.number }}/head:${{ steps.check_fork.outputs.pr_branch }}
           
           # If it's a fork, make origin point to local to trick claude-code-action's `git fetch origin <branch>`
-          if [ "${IS_FORK}" = "true" ]; then
+          if [ "${{ steps.check_fork.outputs.is_fork }}" == "true" ]; then
             git remote set-url origin file://$(pwd)
           fi
 

.github/workflows/integration-cross-repo.yml

@@ -1,158 +0,0 @@
-name: Cross-Repo Integration Tests
-
-on:
-  push:
-    branches: [ main ]
-    paths:
-      - 'src/praisonai/praisonai/integration/**'
-      - 'src/praisonai/tests/integration/test_aiui_*'
-      - 'src/praisonai/pyproject.toml'
-  pull_request:
-    branches: [ main ]
-    paths:
-      - 'src/praisonai/praisonai/integration/**'
-      - 'src/praisonai/tests/integration/test_aiui_*'
-      - 'src/praisonai/pyproject.toml'
-  workflow_dispatch:
-
-jobs:
-  cross-repo-integration:
-    runs-on: ubuntu-latest
-    timeout-minutes: 30
-    
-    steps:
-    - name: Checkout PraisonAI
-      uses: actions/checkout@v4
-      with:
-        path: PraisonAI
-        ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
-        
-    - name: Checkout PraisonAIUI
-      uses: actions/checkout@v4
-      with:
-        repository: MervinPraison/PraisonAIUI
-        path: PraisonAIUI
-        ref: main
-        
-    - name: Set up Python
-      uses: actions/setup-python@v4
-      with:
-        python-version: '3.11'
-        
-    - name: Install dependencies
-      run: |
-        python -m pip install --upgrade pip
-        # Install PraisonAI packages in development mode
-        cd PraisonAI/src/praisonai-agents
-        pip install -e .
-        cd ../praisonai
-        pip install -e ".[ui,dev]"
-        # Install PraisonAIUI in development mode
-        cd ../../../PraisonAIUI
-        pip install -e .
-        
-    - name: Verify versions
-      run: |
-        python -c "import praisonaiagents; print(f'praisonaiagents: {praisonaiagents.__version__}')"
-        python -c "import praisonai; print(f'praisonai: {praisonai.__version__}')"
-        python -c "import praisonaiui; print(f'praisonaiui: {praisonaiui.__version__}')"
-        
-    - name: Run PraisonAI integration tests
-      run: |
-        cd PraisonAI/src/praisonai
-        python -m pytest tests/integration/test_aiui_* -v --timeout=300
-        
-    - name: Run PraisonAIUI integration tests
-      run: |
-        cd PraisonAIUI
-        if [ -f tests/integration/test_agentic_roundtrip.py ]; then
-          python -m pytest tests/integration/test_agentic_roundtrip.py -v --timeout=300
-        fi
-        if [ -f tests/test_feature_sdk_backends.py ]; then
-          python -m pytest tests/test_feature_sdk_backends.py -v --timeout=300
-        fi
-        
-    - name: Test Pattern C CLI
-      run: |
-        cd PraisonAI/src/praisonai
-        # Test that ui-gateway command exists
-        python -m praisonai serve --help | grep ui-gateway
-        
-    - name: Test public API exports
-      run: |
-        cd PraisonAI/src/praisonai
-        python -c "from praisonai import run_integrated_gateway; print('✓ run_integrated_gateway imported')"
-        python -c "from praisonai import configure_host; print('✓ configure_host imported')"
-        
-    - name: Integration smoke test
-      env:
-        PRAISONAI_TEST_MODE: "1"
-        PYTHONPATH: ${{ github.workspace }}/PraisonAI/src/praisonai:${{ github.workspace }}/PraisonAI/src/praisonai-agents:${{ github.workspace }}/PraisonAIUI
-      run: |
-        cd PraisonAI/src/praisonai
-        python -c "
-        import sys
-        sys.path.insert(0, '.')
-        from praisonai.integration.host_app import configure_host
-        from praisonai import run_integrated_gateway
-        print('✓ Integration imports successful')
-        
-        # Test configure_host with new parameters
-        try:
-            configure_host(
-                style='dashboard', 
-                context_paths=['AGENTS.md'],
-                title='Test App'
-            )
-            print('✓ configure_host with new parameters works')
-        except Exception as e:
-            print(f'× configure_host failed: {e}')
-            sys.exit(1)
-        "
-        
-  # Optional job with API key for agentic tests (if secret is available)
-  agentic-integration:
-    runs-on: ubuntu-latest
-    timeout-minutes: 30
-    if: ${{ vars.RUN_AGENTIC_TESTS == 'true' }}
-    needs: cross-repo-integration
-    
-    steps:
-    - name: Checkout PraisonAI
-      uses: actions/checkout@v4
-      with:
-        path: PraisonAI
-        ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
-        
-    - name: Checkout PraisonAIUI
-      uses: actions/checkout@v4
-      with:
-        repository: MervinPraison/PraisonAIUI
-        path: PraisonAIUI
-        ref: main
-        
-    - name: Set up Python
-      uses: actions/setup-python@v4
-      with:
-        python-version: '3.11'
-        
-    - name: Install dependencies
-      run: |
-        python -m pip install --upgrade pip
-        cd PraisonAI/src/praisonai-agents
-        pip install -e .
-        cd ../praisonai
-        pip install -e ".[ui,dev]"
-        cd ../../../PraisonAIUI
-        pip install -e .
-        
-    - name: Run agentic integration test
-      env:
-        OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
-      run: |
-        cd PraisonAI/src/praisonai
-        if [ -n "$OPENAI_API_KEY" ]; then
-          python -m pytest tests/integration/test_aiui_host_agentic.py -v --timeout=600
-        else
-          echo "OPENAI_API_KEY not available, skipping agentic tests"
-        fi

.github/workflows/praisonai-issue-triage.yml

@@ -29,6 +29,9 @@ jobs:
         uses: actions/checkout@v4
         with:
           persist-credentials: false
+        with:
+          persist-credentials: false
+        with:
           fetch-depth: 0
           
       - name: Generate GitHub App Token
@@ -60,12 +63,10 @@ jobs:
           GH_TOKEN: ${{ steps.generate_token.outputs.token }}
           OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
           PRAISONAI_AUTO_APPROVE: "true"
-          ISSUE_NUMBER_INPUT: ${{ inputs.issue_number }}
-          ISSUE_NUMBER_EVENT: ${{ github.event.issue.number }}
         run: |
           if [ "${{ github.event_name }}" = "workflow_dispatch" ]; then
-            export ISSUE_NUMBER="$ISSUE_NUMBER_INPUT"
+            export ISSUE_NUMBER="${{ inputs.issue_number }}"
           else
-            export ISSUE_NUMBER="$ISSUE_NUMBER_EVENT"
+            export ISSUE_NUMBER="${{ github.event.issue.number }}"
           fi
           praisonai github triage --issue $ISSUE_NUMBER --agent-file .github/praisonai-issue-triage.yaml

.github/workflows/praisonai-pr-review.yml

@@ -30,23 +30,22 @@ jobs:
     steps:
       - name: Determine checkout ref
         id: dest
-        env:
-          ISSUE_NUMBER: ${{ github.event.issue.number }}
-          PR_NUMBER_INPUT: ${{ inputs.pr_number }}
-          PR_HEAD_SHA: ${{ github.event.pull_request.head.sha }}
         run: |
           if [ "${{ github.event_name }}" = "issue_comment" ]; then
-            echo "ref=refs/pull/$ISSUE_NUMBER/head" >> "$GITHUB_OUTPUT"
+            echo "ref=refs/pull/${{ github.event.issue.number }}/head" >> "$GITHUB_OUTPUT"
           elif [ "${{ github.event_name }}" = "workflow_dispatch" ]; then
-            echo "ref=refs/pull/$PR_NUMBER_INPUT/head" >> "$GITHUB_OUTPUT"
+            echo "ref=refs/pull/${{ inputs.pr_number }}/head" >> "$GITHUB_OUTPUT"
           else
-            echo "ref=$PR_HEAD_SHA" >> "$GITHUB_OUTPUT"
+            echo "ref=${{ github.event.pull_request.head.sha }}" >> "$GITHUB_OUTPUT"
           fi
 
       - name: Checkout Repository
         uses: actions/checkout@v4
         with:
           persist-credentials: false
+        with:
+          persist-credentials: false
+        with:
           ref: ${{ steps.dest.outputs.ref }}
           fetch-depth: 0
           

.github/workflows/test-windows.yml

@@ -26,8 +26,8 @@ jobs:
     
     steps:
     - uses: actions/checkout@v4
-      with:
-        persist-credentials: false
+        with:
+          persist-credentials: false
     - name: Set up Python 3.11
       uses: actions/setup-python@v5
       with:
@@ -176,18 +176,6 @@ jobs:
       run: praisonai queue list
       continue-on-error: true
 
-    # ==================== PACKAGING TESTS ====================
-    - name: Test packaging/daemon entry points
-      run: |
-        echo "Testing both entry point methods..."
-        python -m praisonai --version
-        praisonai --version
-
-    - name: Test doctor packaging command
-      run: |
-        echo "Testing doctor packaging diagnostics..."
-        praisonai doctor packaging --json
-
     # ==================== REAL API TESTS ====================
     - name: Real API test - Agent run
       if: env.OPENAI_API_KEY != ''

CONTRIBUTING.md

@@ -1,14 +0,0 @@
-# Contributing to PraisonAI
-
-Thank you for your interest! Here's how to get started:
-
-1. Fork the repository
-2. Create a feature branch
-3. Make your changes
-4. Submit a pull request
-
-## Guidelines
-
-- Follow existing code style
-- Add tests for new features
-- Update documentation as needed

README.md

@@ -294,35 +294,8 @@ def search(query: str) -> str:
 
 @tool
 def calculate(expression: str) -> float:
-    """Safely evaluate a numeric arithmetic expression."""
-    import ast
-    import operator
-    
-    # Define allowed operations
-    _OPS = {
-        ast.Add: operator.add,
-        ast.Sub: operator.sub,
-        ast.Mult: operator.mul,
-        ast.Div: operator.truediv,
-        ast.Pow: operator.pow,
-        ast.USub: operator.neg,
-        ast.UAdd: operator.pos,
-    }
-    
-    def _safe_eval(node):
-        if isinstance(node, ast.Constant) and isinstance(node.value, (int, float)):
-            return node.value
-        elif isinstance(node, ast.BinOp) and type(node.op) in _OPS:
-            return _OPS[type(node.op)](_safe_eval(node.left), _safe_eval(node.right))
-        elif isinstance(node, ast.UnaryOp) and type(node.op) in _OPS:
-            return _OPS[type(node.op)](_safe_eval(node.operand))
-        else:
-            raise ValueError("Unsupported expression")
-    
-    try:
-        return _safe_eval(ast.parse(expression, mode="eval").body)
-    except (ValueError, SyntaxError, TypeError, ZeroDivisionError, OverflowError):
-        raise ValueError("Invalid arithmetic expression")
+    """Evaluate a math expression."""
+    return eval(expression)
 
 agent = Agent(
     instructions="You are a helpful assistant",
@@ -331,7 +304,6 @@ agent = Agent(
 agent.start("Search for AI news and calculate 15*4")
 ```
 
-> ⚠️ **Security Note:** Never use `eval()`, `exec()`, or `subprocess` in tool functions that process LLM-generated or user-supplied input. Always validate and sanitize inputs to prevent code injection attacks.
 > 📖 [Full tools docs](https://docs.praison.ai/docs/tools/tools) — BaseTool, tool packages, 100+ built-in tools
 
 ### 5. Persistence (Databases)
@@ -358,15 +330,6 @@ pip install "praisonai[claw]"
 praisonai claw
 ```
 
-#### Required Environment Variables
-
-Copy `.env.example` to `.env` and configure the following variables:
-
-| Variable | Required | Description |
-|----------|----------|-------------|
-| `OPENAI_API_KEY` | Yes | OpenAI API key for all LLM calls |
-| `TAVILY_API_KEY` | Yes (Claw) | Tavily key for the built-in web-search tool. Get one free at https://app.tavily.com |
-
 Open **http://localhost:8082** — the dashboard comes with 13 built-in pages: Chat, Agents, Memory, Knowledge, Channels, Guardrails, Cron, and more. Add messaging channels directly from the UI.
 
 > 📖 [Full Claw docs](https://docs.praison.ai/docs/concepts/claw) — platform tokens, CLI options, Docker, and YAML agent mode
@@ -382,7 +345,7 @@ praisonai flow
 
 Open **http://localhost:7861** — use the **Agent** and **Agent Team** components to create sequential or parallel workflows. Connect Chat Input → Agent Team → Chat Output for instant multi-agent pipelines.
 
-> 📖 [Full Flow docs](https://docs.praison.ai/docs/concepts/agentflow) — visual agent building, component reference, and deployment
+> 📖 [Full Flow docs](https://docs.praison.ai/docs/concepts/flow) — visual agent building, component reference, and deployment
 
 ### 8. PraisonAI UI 🤖 (Clean Chat)
 

api.md

@@ -15,7 +15,6 @@ Methods:
 * <code title="class GuardrailResult">GuardrailResult.<a href="./src/praisonai-agents/praisonaiagents/guardrails/guardrail_result.py">from_tuple</a>(result: Tuple[bool, Any]) -> 'GuardrailResult'</code>
 * <code title="class HandoffConfig">HandoffConfig.<a href="./src/praisonai-agents/praisonaiagents/agent/handoff.py">from_dict</a>(data: Dict[str, Any]) -> 'HandoffConfig'</code>
 * <code title="class HandoffConfig">HandoffConfig.<a href="./src/praisonai-agents/praisonaiagents/agent/handoff.py">to_dict</a>() -> Dict[str, Any]</code>
-* <code title="class HandoffResult">HandoffResult.<a href="./src/praisonai-agents/praisonaiagents/agent/handoff.py">from_outcome</a>(outcome: AgentRunOutcome, target_agent: Optional[str] = None, source_agent: Optional[str] = None, handoff_depth: int = 0) -> 'HandoffResult'</code>
 * <code title="class TaskOutput">TaskOutput.<a href="./src/praisonai-agents/praisonaiagents/main.py">json</a>() -> Optional[str]</code>
 * <code title="class TaskOutput">TaskOutput.<a href="./src/praisonai-agents/praisonaiagents/main.py">to_dict</a>() -> dict</code>
 * <code title="class ToolResult">ToolResult.<a href="./src/praisonai-agents/praisonaiagents/tools/base.py">to_dict</a>() -> Dict[str, Any]</code>
@@ -41,7 +40,6 @@ Methods:
 * <code title="class Agent">Agent.<a href="./src/praisonai-agents/praisonaiagents/agent/agent.py">chat_with_context</a>(message: str, context: 'ContextPack', **kwargs) -> str</code>
 * <code title="class Agent">Agent.<a href="./src/praisonai-agents/praisonaiagents/agent/agent.py">checkpoints</a>() -> Optional[bool]</code>
 * <code title="class Agent">Agent.<a href="./src/praisonai-agents/praisonaiagents/agent/agent.py">checkpoints</a>(value: Optional[bool]) -> None</code>
-* <code title="class Agent">Agent.<a href="./src/praisonai-agents/praisonaiagents/agent/agent.py">clone_for_channel</a>() -> 'Agent'</code>
 * <code title="class Agent">Agent.<a href="./src/praisonai-agents/praisonaiagents/agent/agent.py">close</a>() -> None</code>
 * <code title="class Agent">Agent.<a href="./src/praisonai-agents/praisonaiagents/agent/agent.py">console</a>() -> Optional[Any]</code>
 * <code title="class Agent">Agent.<a href="./src/praisonai-agents/praisonaiagents/agent/agent.py">context_manager</a>() -> Optional[Any]</code>
@@ -76,7 +74,7 @@ Methods:
 * <code title="class Agent">Agent.<a href="./src/praisonai-agents/praisonaiagents/agent/agent.py">run_until</a>(prompt: str, criteria: str, threshold: float = 8.0, max_iterations: int = 5, mode: str = 'optimize', on_iteration: Optional[Callable[[Any], None]] = None, verbose: bool = False) -> 'EvaluationLoopResult'</code>
 * <code title="class Agent">Agent.<a href="./src/praisonai-agents/praisonaiagents/agent/agent.py">run_until_async</a>(prompt: str, criteria: str, threshold: float = 8.0, max_iterations: int = 5, mode: str = 'optimize', on_iteration: Optional[Callable[[Any], None]] = None, verbose: bool = False) -> 'EvaluationLoopResult'</code>
 * <code title="class Agent">Agent.<a href="./src/praisonai-agents/praisonaiagents/agent/agent.py">skill_manager</a>() -> Optional[Any]</code>
-* <code title="class Agent">Agent.<a href="./src/praisonai-agents/praisonaiagents/agent/agent.py">store_memory</a>(content: str, memory_type: str = 'short_term', action: str = 'add', **kwargs: Any) -> None</code>
+* <code title="class Agent">Agent.<a href="./src/praisonai-agents/praisonaiagents/agent/agent.py">store_memory</a>(content: str, memory_type: str = 'short_term', **kwargs: Any) -> None</code>
 * <code title="class Agent">Agent.<a href="./src/praisonai-agents/praisonaiagents/agent/agent.py">stream_emitter</a>() -> Optional[Any]</code>
 * <code title="class Agent">Agent.<a href="./src/praisonai-agents/praisonaiagents/agent/agent.py">stream_emitter</a>(value: Optional[Any]) -> None</code>
 * <code title="class Agent">Agent.<a href="./src/praisonai-agents/praisonaiagents/agent/agent.py">thinking_budget</a>() -> Optional[int]</code>
@@ -173,10 +171,8 @@ Methods:
 * <code title="class BaseTool">BaseTool.<a href="./src/praisonai-agents/praisonaiagents/tools/base.py">safe_run</a>(**kwargs) -> ToolResult</code>
 * <code title="class BaseTool">BaseTool.<a href="./src/praisonai-agents/praisonaiagents/tools/base.py">validate</a>() -> bool</code>
 * <code title="class BaseTool">BaseTool.<a href="./src/praisonai-agents/praisonaiagents/tools/base.py">validate_class</a>() -> bool</code>
-* <code title="class BaseTool">BaseTool.<a href="./src/praisonai-agents/praisonaiagents/tools/base.py">validate_schema_roundtrip</a>() -> bool</code>
 * <code title="class FunctionTool">FunctionTool.<a href="./src/praisonai-agents/praisonaiagents/tools/decorator.py">__call__</a>(*args, **kwargs) -> Any</code>
 * <code title="class FunctionTool">FunctionTool.<a href="./src/praisonai-agents/praisonaiagents/tools/decorator.py">check_availability</a>() -> tuple[bool, str]</code>
-* <code title="class FunctionTool">FunctionTool.<a href="./src/praisonai-agents/praisonaiagents/tools/decorator.py">get_schema</a>() -> Dict[str, Any]</code>
 * <code title="class FunctionTool">FunctionTool.<a href="./src/praisonai-agents/praisonaiagents/tools/decorator.py">injected_params</a>() -> Dict[str, Any]</code>
 * <code title="class FunctionTool">FunctionTool.<a href="./src/praisonai-agents/praisonaiagents/tools/decorator.py">run</a>(**kwargs) -> Any</code>
 * <code title="class ToolRegistry">ToolRegistry.<a href="./src/praisonai-agents/praisonaiagents/tools/registry.py">clear</a>() -> None</code>
@@ -184,18 +180,15 @@ Methods:
 * <code title="class ToolRegistry">ToolRegistry.<a href="./src/praisonai-agents/praisonaiagents/tools/registry.py">discover_single_file_plugins</a>() -> int</code>
 * <code title="class ToolRegistry">ToolRegistry.<a href="./src/praisonai-agents/praisonaiagents/tools/registry.py">get</a>(name: str) -> Optional[Union[BaseTool, Callable]]</code>
 * <code title="class ToolRegistry">ToolRegistry.<a href="./src/praisonai-agents/praisonaiagents/tools/registry.py">get_all</a>() -> Dict[str, Union[BaseTool, Callable]]</code>
-* <code title="class ToolRegistry">ToolRegistry.<a href="./src/praisonai-agents/praisonaiagents/tools/registry.py">get_tool_definitions</a>() -> List[Dict[str, Any]]</code>
-* <code title="class ToolRegistry">ToolRegistry.<a href="./src/praisonai-agents/praisonaiagents/tools/registry.py">get_trust_level</a>(name: str) -> Optional[str]</code>
-* <code title="class ToolRegistry">ToolRegistry.<a href="./src/praisonai-agents/praisonaiagents/tools/registry.py">list_available_tools</a>(context: Optional[Dict[str, Any]] = None, ttl_seconds: Optional[float] = None) -> List[Union[BaseTool, Callable]]</code>
+* <code title="class ToolRegistry">ToolRegistry.<a href="./src/praisonai-agents/praisonaiagents/tools/registry.py">list_available_tools</a>(context: Optional[Dict[str, Any]] = None) -> List[Union[BaseTool, Callable]]</code>
 * <code title="class ToolRegistry">ToolRegistry.<a href="./src/praisonai-agents/praisonaiagents/tools/registry.py">list_base_tools</a>() -> List[BaseTool]</code>
 * <code title="class ToolRegistry">ToolRegistry.<a href="./src/praisonai-agents/praisonaiagents/tools/registry.py">list_tools</a>() -> List[str]</code>
-* <code title="class ToolRegistry">ToolRegistry.<a href="./src/praisonai-agents/praisonaiagents/tools/registry.py">list_tools_with_allowed_filter</a>(context: Optional[Dict[str, Any]] = None) -> List[str]</code>
-* <code title="class ToolRegistry">ToolRegistry.<a href="./src/praisonai-agents/praisonaiagents/tools/registry.py">register</a>(tool: Union[BaseTool, Callable], name: Optional[str] = None, overwrite: bool = False, trust_level: Optional[str] = None, dynamic_schema_overrides: Optional[Callable[[Dict[str, Any]], Dict[str, Any]]] = None) -> None</code>
+* <code title="class ToolRegistry">ToolRegistry.<a href="./src/praisonai-agents/praisonaiagents/tools/registry.py">register</a>(tool: Union[BaseTool, Callable], name: Optional[str] = None, overwrite: bool = False) -> None</code>
 * <code title="class ToolRegistry">ToolRegistry.<a href="./src/praisonai-agents/praisonaiagents/tools/registry.py">unregister</a>(name: str) -> bool</code>
 * <code title="class Tools">Tools.<a href="./src/praisonai-agents/praisonaiagents/tools/tools.py">internet_search</a>(*args, **kwargs)</code>
 * <code title="function">praisonaiagents.<a href="./src/praisonai-agents/praisonaiagents/tools/registry.py">get_registry</a>() -> ToolRegistry</code>
 * <code title="function">praisonaiagents.<a href="./src/praisonai-agents/praisonaiagents/tools/registry.py">get_tool</a>(name: str) -> Optional[Union[BaseTool, Callable]]</code>
-* <code title="function">praisonaiagents.<a href="./src/praisonai-agents/praisonaiagents/tools/registry.py">register_tool</a>(tool: Union[BaseTool, Callable], name: Optional[str] = None, trust_level: Optional[str] = None, dynamic_schema_overrides: Optional[Callable[[Dict[str, Any]], Dict[str, Any]]] = None) -> None</code>
+* <code title="function">praisonaiagents.<a href="./src/praisonai-agents/praisonaiagents/tools/registry.py">register_tool</a>(tool: Union[BaseTool, Callable], name: Optional[str] = None) -> None</code>
 * <code title="function">praisonaiagents.<a href="./src/praisonai-agents/praisonaiagents/tools/decorator.py">tool</a>(func: Optional[Callable] = None) -> Union[FunctionTool, Callable[[Callable], FunctionTool]]</code>
 * <code title="function">praisonaiagents.<a href="./src/praisonai-agents/praisonaiagents/tools/base.py">validate_tool</a>(tool: Any) -> bool</code>
 
@@ -357,7 +350,7 @@ Methods:
 * <code title="class Handoff">Handoff.<a href="./src/praisonai-agents/praisonaiagents/agent/handoff.py">to_tool_function</a>(source_agent: 'Agent') -> Callable</code>
 * <code title="class Handoff">Handoff.<a href="./src/praisonai-agents/praisonaiagents/agent/handoff.py">tool_description</a>() -> str</code>
 * <code title="class Handoff">Handoff.<a href="./src/praisonai-agents/praisonaiagents/agent/handoff.py">tool_name</a>() -> str</code>
-* <code title="function">praisonaiagents.<a href="./src/praisonai-agents/praisonaiagents/agent/handoff.py">handoff</a>(agent: 'Agent', tool_name_override: Optional[str] = None, tool_description_override: Optional[str] = None, on_handoff: Optional[Callable] = None, input_type: Optional[type] = None, input_filter: Optional[Callable[[HandoffInputData], HandoffInputData]] = None, config: Optional[HandoffConfig] = None, context_policy: Optional[str] = None, timeout_seconds: Optional[float] = None, max_concurrent: Optional[int] = None, detect_cycles: Optional[bool] = None, max_depth: Optional[int] = None, tool_policy_mode: Optional[Literal['intersect', 'passthrough']] = None, blocked_tools: Optional[List[str]] = None) -> Handoff</code>
+* <code title="function">praisonaiagents.<a href="./src/praisonai-agents/praisonaiagents/agent/handoff.py">handoff</a>(agent: 'Agent', tool_name_override: Optional[str] = None, tool_description_override: Optional[str] = None, on_handoff: Optional[Callable] = None, input_type: Optional[type] = None, input_filter: Optional[Callable[[HandoffInputData], HandoffInputData]] = None, config: Optional[HandoffConfig] = None, context_policy: Optional[str] = None, timeout_seconds: Optional[float] = None, max_concurrent: Optional[int] = None, detect_cycles: Optional[bool] = None, max_depth: Optional[int] = None) -> Handoff</code>
 * <code title="class handoff_filters">handoff_filters.<a href="./src/praisonai-agents/praisonaiagents/agent/handoff.py">compress_history</a>(data: HandoffInputData) -> HandoffInputData</code>
 * <code title="class handoff_filters">handoff_filters.<a href="./src/praisonai-agents/praisonaiagents/agent/handoff.py">keep_last_n_messages</a>(n: int) -> Callable[[HandoffInputData], HandoffInputData]</code>
 * <code title="class handoff_filters">handoff_filters.<a href="./src/praisonai-agents/praisonaiagents/agent/handoff.py">remove_all_tools</a>(data: HandoffInputData) -> HandoffInputData</code>
@@ -483,10 +476,8 @@ Methods:
 * <code title="class SkillManager">SkillManager.<a href="./src/praisonai-agents/praisonaiagents/skills/manager.py">edit_skill</a>(name: str, content: str) -> dict</code>
 * <code title="class SkillManager">SkillManager.<a href="./src/praisonai-agents/praisonaiagents/skills/manager.py">get_allowed_tools</a>(name: str) -> List[str]</code>
 * <code title="class SkillManager">SkillManager.<a href="./src/praisonai-agents/praisonaiagents/skills/manager.py">get_available_skills</a>() -> List[SkillMetadata]</code>
-* <code title="class SkillManager">SkillManager.<a href="./src/praisonai-agents/praisonaiagents/skills/manager.py">get_available_skills_by_state</a>(state: SkillState) -> List[LoadedSkill]</code>
 * <code title="class SkillManager">SkillManager.<a href="./src/praisonai-agents/praisonaiagents/skills/manager.py">get_instructions</a>(name: str) -> Optional[str]</code>
 * <code title="class SkillManager">SkillManager.<a href="./src/praisonai-agents/praisonaiagents/skills/manager.py">get_skill</a>(name: str) -> Optional[LoadedSkill]</code>
-* <code title="class SkillManager">SkillManager.<a href="./src/praisonai-agents/praisonaiagents/skills/manager.py">get_skills_diagnostics</a>() -> Dict[str, ValidationResult]</code>
 * <code title="class SkillManager">SkillManager.<a href="./src/praisonai-agents/praisonaiagents/skills/manager.py">get_user_invocable_skills</a>() -> List[LoadedSkill]</code>
 * <code title="class SkillManager">SkillManager.<a href="./src/praisonai-agents/praisonaiagents/skills/manager.py">invoke</a>(name: str, raw_args: str = '', session_id: Optional[str] = None, shell_exec: bool = False) -> Optional[str]</code>
 * <code title="class SkillManager">SkillManager.<a href="./src/praisonai-agents/praisonaiagents/skills/manager.py">load_resources</a>(name: str) -> bool</code>
@@ -495,7 +486,6 @@ Methods:
 * <code title="class SkillManager">SkillManager.<a href="./src/praisonai-agents/praisonaiagents/skills/manager.py">skill_names</a>() -> List[str]</code>
 * <code title="class SkillManager">SkillManager.<a href="./src/praisonai-agents/praisonaiagents/skills/manager.py">skills</a>() -> List[LoadedSkill]</code>
 * <code title="class SkillManager">SkillManager.<a href="./src/praisonai-agents/praisonaiagents/skills/manager.py">to_prompt</a>() -> str</code>
-* <code title="class SkillManager">SkillManager.<a href="./src/praisonai-agents/praisonaiagents/skills/manager.py">validate_skill_capabilities</a>(skill_name: str, force_refresh: bool = False) -> ValidationResult</code>
 * <code title="class SkillManager">SkillManager.<a href="./src/praisonai-agents/praisonaiagents/skills/manager.py">write_skill_file</a>(name: str, file_path: str, file_content: str) -> dict</code>
 * <code title="class SkillMetadata">SkillMetadata.<a href="./src/praisonai-agents/praisonaiagents/skills/models.py">from_properties</a>(props: SkillProperties) -> 'SkillMetadata'</code>
 * <code title="class SkillProperties">SkillProperties.<a href="./src/praisonai-agents/praisonaiagents/skills/models.py">to_dict</a>() -> dict</code>
@@ -611,13 +601,6 @@ Methods:
 * <code title="class ContextManager">ContextManager.<a href="./src/praisonai-agents/praisonaiagents/context/manager.py">set_tool_budget</a>(tool_name: str, max_tokens: int, protected: bool = False) -> None</code>
 * <code title="class ContextManager">ContextManager.<a href="./src/praisonai-agents/praisonaiagents/context/manager.py">truncate_tool_output</a>(tool_name: str, output: str) -> str</code>
 
-# UI
-
-Types:
-```python
-from praisonaiagents import A2UI
-```
-
 # Config
 
 Types:
@@ -630,7 +613,6 @@ Methods:
 * <code title="class AutonomyConfig">AutonomyConfig.<a href="./src/praisonai-agents/praisonaiagents/agent/autonomy.py">effective_track_changes</a>() -> bool</code>
 * <code title="class AutonomyConfig">AutonomyConfig.<a href="./src/praisonai-agents/praisonaiagents/agent/autonomy.py">from_dict</a>(data: Dict[str, Any]) -> 'AutonomyConfig'</code>
 * <code title="class CachingConfig">CachingConfig.<a href="./src/praisonai-agents/praisonaiagents/config/feature_configs.py">to_dict</a>() -> Dict[str, Any]</code>
-* <code title="class ExecutionConfig">ExecutionConfig.<a href="./src/praisonai-agents/praisonaiagents/config/feature_configs.py">from_dict</a>(data: Dict[str, Any]) -> 'ExecutionConfig'</code>
 * <code title="class ExecutionConfig">ExecutionConfig.<a href="./src/praisonai-agents/praisonaiagents/config/feature_configs.py">to_dict</a>() -> Dict[str, Any]</code>
 * <code title="class GuardrailConfig">GuardrailConfig.<a href="./src/praisonai-agents/praisonaiagents/config/feature_configs.py">to_dict</a>() -> Dict[str, Any]</code>
 * <code title="class HooksConfig">HooksConfig.<a href="./src/praisonai-agents/praisonaiagents/config/feature_configs.py">to_dict</a>() -> Dict[str, Any]</code>
@@ -723,16 +705,12 @@ Methods:
 * <code title="class AgentFlow">AgentFlow.<a href="./src/praisonai-agents/praisonaiagents/workflows/workflows.py">verbose</a>(value: bool)</code>
 * <code title="class AgentOSProtocol">AgentOSProtocol.<a href="./src/praisonai-agents/praisonaiagents/app/protocols.py">get_app</a>() -> Any</code>
 * <code title="class AgentOSProtocol">AgentOSProtocol.<a href="./src/praisonai-agents/praisonaiagents/app/protocols.py">serve</a>(host: Optional[str] = None, port: Optional[int] = None, reload: bool = False, **kwargs: Any) -> None</code>
-* <code title="class AgentTeam">AgentTeam.<a href="./src/praisonai-agents/praisonaiagents/agents/agents.py">aannounce_completion</a>(agent_id: str, task_id: str, result: Any, success: bool = True, error: Optional[str] = None, metadata: Optional[Dict[str, Any]] = None) -> None</code>
 * <code title="class AgentTeam">AgentTeam.<a href="./src/praisonai-agents/praisonaiagents/agents/agents.py">add_task</a>(task)</code>
 * <code title="class AgentTeam">AgentTeam.<a href="./src/praisonai-agents/praisonaiagents/agents/agents.py">aexecute_task</a>(task_id)</code>
-* <code title="class AgentTeam">AgentTeam.<a href="./src/praisonai-agents/praisonaiagents/agents/agents.py">announce_completion</a>(agent_id: str, task_id: str, result: Any, success: bool = True, error: Optional[str] = None, metadata: Optional[Dict[str, Any]] = None) -> None</code>
 * <code title="class AgentTeam">AgentTeam.<a href="./src/praisonai-agents/praisonaiagents/agents/agents.py">append_to_state</a>(key: str, value: Any, max_length: Optional[int] = None) -> List[Any]</code>
 * <code title="class AgentTeam">AgentTeam.<a href="./src/praisonai-agents/praisonaiagents/agents/agents.py">arun_all_tasks</a>()</code>
 * <code title="class AgentTeam">AgentTeam.<a href="./src/praisonai-agents/praisonaiagents/agents/agents.py">arun_task</a>(task_id)</code>
-* <code title="class AgentTeam">AgentTeam.<a href="./src/praisonai-agents/praisonaiagents/agents/agents.py">aspawn_sub_agent</a>(agent: Agent, task: Any, completion_callback: Optional[Callable[[SubAgentCompletionEvent], Any]] = None, metadata: Optional[Dict[str, Any]] = None) -> SpawnedSubAgent</code>
 * <code title="class AgentTeam">AgentTeam.<a href="./src/praisonai-agents/praisonaiagents/agents/agents.py">astart</a>(content = None, return_dict = False, **kwargs)</code>
-* <code title="class AgentTeam">AgentTeam.<a href="./src/praisonai-agents/praisonaiagents/agents/agents.py">await_for_completions</a>(timeout: Optional[float] = None, agent_ids: Optional[List[str]] = None) -> List[SubAgentCompletionEvent]</code>
 * <code title="class AgentTeam">AgentTeam.<a href="./src/praisonai-agents/praisonaiagents/agents/agents.py">clean_json_output</a>(output: str) -> str</code>
 * <code title="class AgentTeam">AgentTeam.<a href="./src/praisonai-agents/praisonaiagents/agents/agents.py">clear_state</a>() -> None</code>
 * <code title="class AgentTeam">AgentTeam.<a href="./src/praisonai-agents/praisonaiagents/agents/agents.py">close</a>() -> None</code>
@@ -747,7 +725,6 @@ Methods:
 * <code title="class AgentTeam">AgentTeam.<a href="./src/praisonai-agents/praisonaiagents/agents/agents.py">get_all_tasks_status</a>()</code>
 * <code title="class AgentTeam">AgentTeam.<a href="./src/praisonai-agents/praisonaiagents/agents/agents.py">get_detailed_token_report</a>() -> Dict[str, Any]</code>
 * <code title="class AgentTeam">AgentTeam.<a href="./src/praisonai-agents/praisonaiagents/agents/agents.py">get_plan_markdown</a>() -> str</code>
-* <code title="class AgentTeam">AgentTeam.<a href="./src/praisonai-agents/praisonaiagents/agents/agents.py">get_spawned_agents</a>() -> List[SpawnedSubAgent]</code>
 * <code title="class AgentTeam">AgentTeam.<a href="./src/praisonai-agents/praisonaiagents/agents/agents.py">get_state</a>(key: str, default: Any = None) -> Any</code>
 * <code title="class AgentTeam">AgentTeam.<a href="./src/praisonai-agents/praisonaiagents/agents/agents.py">get_task_details</a>(task_id)</code>
 * <code title="class AgentTeam">AgentTeam.<a href="./src/praisonai-agents/praisonaiagents/agents/agents.py">get_task_result</a>(task_id)</code>
@@ -764,12 +741,10 @@ Methods:
 * <code title="class AgentTeam">AgentTeam.<a href="./src/praisonai-agents/praisonaiagents/agents/agents.py">save_output_to_file</a>(task, task_output)</code>
 * <code title="class AgentTeam">AgentTeam.<a href="./src/praisonai-agents/praisonaiagents/agents/agents.py">save_session_state</a>(session_id: str, include_memory: bool = True) -> None</code>
 * <code title="class AgentTeam">AgentTeam.<a href="./src/praisonai-agents/praisonaiagents/agents/agents.py">set_state</a>(key: str, value: Any) -> None</code>
-* <code title="class AgentTeam">AgentTeam.<a href="./src/praisonai-agents/praisonaiagents/agents/agents.py">spawn_sub_agent</a>(agent: Agent, task: Any, completion_callback: Optional[Callable[[SubAgentCompletionEvent], Any]] = None, metadata: Optional[Dict[str, Any]] = None) -> SpawnedSubAgent</code>
 * <code title="class AgentTeam">AgentTeam.<a href="./src/praisonai-agents/praisonaiagents/agents/agents.py">start</a>(content = None, return_dict = False, output = None, **kwargs)</code>
 * <code title="class AgentTeam">AgentTeam.<a href="./src/praisonai-agents/praisonaiagents/agents/agents.py">todo_list</a>()</code>
 * <code title="class AgentTeam">AgentTeam.<a href="./src/praisonai-agents/praisonaiagents/agents/agents.py">update_plan_step_status</a>(step_id: str, status: str) -> bool</code>
 * <code title="class AgentTeam">AgentTeam.<a href="./src/praisonai-agents/praisonaiagents/agents/agents.py">update_state</a>(updates: Dict) -> None</code>
-* <code title="class AgentTeam">AgentTeam.<a href="./src/praisonai-agents/praisonaiagents/agents/agents.py">wait_for_completions</a>(timeout: Optional[float] = None, agent_ids: Optional[List[str]] = None) -> List[SubAgentCompletionEvent]</code>
 * <code title="class AutoApproveBackend">AutoApproveBackend.<a href="./src/praisonai-agents/praisonaiagents/approval/backends.py">request_approval</a>(request: ApprovalRequest) -> ApprovalDecision</code>
 * <code title="class AutoApproveBackend">AutoApproveBackend.<a href="./src/praisonai-agents/praisonaiagents/approval/backends.py">request_approval_sync</a>(request: ApprovalRequest) -> ApprovalDecision</code>
 * <code title="function">praisonaiagents.<a href="./src/praisonai-agents/praisonaiagents/embedding/embed.py">aembedding</a>(input: Union[str, List[str]], model: str = 'text-embedding-3-small', dimensions: Optional[int] = None, encoding_format: str = 'float', timeout: float = 600.0, api_key: Optional[str] = None, api_base: Optional[str] = None, metadata: Optional[Dict[str, Any]] = None, **kwargs) -> EmbeddingResult</code>
@@ -780,7 +755,7 @@ Methods:
 
 Types:
 ```python
-from praisonai import Agent, AgentApp, AgentOS, AnthropicManagedAgent, CloudProvider, DB, Deploy, DeployConfig, DeployType, HostedAgent, HostedAgentConfig, LocalAgent, LocalAgentConfig, LocalManagedAgent, LocalManagedConfig, ManagedAgent, ManagedConfig, PraisonAI, __version__, arun, run, run_integrated_gateway
+from praisonai import Agent, AgentApp, AgentOS, AnthropicManagedAgent, CloudProvider, DB, Deploy, DeployConfig, DeployType, HostedAgent, HostedAgentConfig, LocalAgent, LocalAgentConfig, LocalManagedAgent, LocalManagedConfig, ManagedAgent, ManagedConfig, PraisonAI, __version__
 ```
 
 # CLI
@@ -811,10 +786,6 @@ Methods:
 * <code title="cli">praisonai bot discord <a href="./src/praisonai/praisonai/cli/commands/bot.py">--help</a></code>
 * <code title="cli">praisonai bot email <a href="./src/praisonai/praisonai/cli/commands/bot.py">--help</a></code>
 * <code title="cli">praisonai bot install-daemon <a href="./src/praisonai/praisonai/cli/commands/bot.py">--help</a></code>
-* <code title="cli">praisonai bot linear <a href="./src/praisonai/praisonai/cli/commands/bot.py">--help</a></code>
-* <code title="cli">praisonai bot list <a href="./src/praisonai/praisonai/cli/commands/bot.py">--help</a></code>
-* <code title="cli">praisonai bot purge <a href="./src/praisonai/praisonai/cli/commands/bot.py">--help</a></code>
-* <code title="cli">praisonai bot replay <a href="./src/praisonai/praisonai/cli/commands/bot.py">--help</a></code>
 * <code title="cli">praisonai bot slack <a href="./src/praisonai/praisonai/cli/commands/bot.py">--help</a></code>
 * <code title="cli">praisonai bot start <a href="./src/praisonai/praisonai/cli/commands/bot.py">--help</a></code>
 * <code title="cli">praisonai bot telegram <a href="./src/praisonai/praisonai/cli/commands/bot.py">--help</a></code>
@@ -882,10 +853,8 @@ Methods:
 * <code title="cli">praisonai doctor env <a href="./src/praisonai/praisonai/cli/commands/doctor.py">--help</a></code>
 * <code title="cli">praisonai doctor mcp <a href="./src/praisonai/praisonai/cli/commands/doctor.py">--help</a></code>
 * <code title="cli">praisonai doctor network <a href="./src/praisonai/praisonai/cli/commands/doctor.py">--help</a></code>
-* <code title="cli">praisonai doctor packaging <a href="./src/praisonai/praisonai/cli/commands/doctor.py">--help</a></code>
 * <code title="cli">praisonai doctor performance <a href="./src/praisonai/praisonai/cli/commands/doctor.py">--help</a></code>
 * <code title="cli">praisonai doctor selftest <a href="./src/praisonai/praisonai/cli/commands/doctor.py">--help</a></code>
-* <code title="cli">praisonai doctor skills <a href="./src/praisonai/praisonai/cli/commands/doctor.py">--help</a></code>
 * <code title="cli">praisonai doctor tools <a href="./src/praisonai/praisonai/cli/commands/doctor.py">--help</a></code>
 * <code title="cli">praisonai doctor troubleshoot <a href="./src/praisonai/praisonai/cli/commands/doctor.py">--help</a></code>
 * <code title="cli">praisonai endpoints list <a href="./src/praisonai/praisonai/cli/commands/endpoints.py">--help</a></code>
@@ -917,24 +886,11 @@ Methods:
 * <code title="cli">praisonai gateway send <a href="./src/praisonai/praisonai/cli/commands/gateway.py">--help</a></code>
 * <code title="cli">praisonai gateway start <a href="./src/praisonai/praisonai/cli/commands/gateway.py">--help</a></code>
 * <code title="cli">praisonai gateway status <a href="./src/praisonai/praisonai/cli/commands/gateway.py">--help</a></code>
-* <code title="cli">praisonai gateway stop <a href="./src/praisonai/praisonai/cli/commands/gateway.py">--help</a></code>
 * <code title="cli">praisonai gateway uninstall <a href="./src/praisonai/praisonai/cli/commands/gateway.py">--help</a></code>
 * <code title="cli">praisonai github triage <a href="./src/praisonai/praisonai/cli/commands/github.py">--help</a></code>
 * <code title="cli">praisonai hooks add <a href="./src/praisonai/praisonai/cli/commands/hooks.py">--help</a></code>
 * <code title="cli">praisonai hooks list <a href="./src/praisonai/praisonai/cli/commands/hooks.py">--help</a></code>
 * <code title="cli">praisonai hooks remove <a href="./src/praisonai/praisonai/cli/commands/hooks.py">--help</a></code>
-* <code title="cli">praisonai kanban block <a href="./src/praisonai/praisonai/cli/commands/kanban.py">--help</a></code>
-* <code title="cli">praisonai kanban boards <a href="./src/praisonai/praisonai/cli/commands/kanban.py">--help</a></code>
-* <code title="cli">praisonai kanban comment <a href="./src/praisonai/praisonai/cli/commands/kanban.py">--help</a></code>
-* <code title="cli">praisonai kanban complete <a href="./src/praisonai/praisonai/cli/commands/kanban.py">--help</a></code>
-* <code title="cli">praisonai kanban create <a href="./src/praisonai/praisonai/cli/commands/kanban.py">--help</a></code>
-* <code title="cli">praisonai kanban dispatch <a href="./src/praisonai/praisonai/cli/commands/kanban.py">--help</a></code>
-* <code title="cli">praisonai kanban link <a href="./src/praisonai/praisonai/cli/commands/kanban.py">--help</a></code>
-* <code title="cli">praisonai kanban list <a href="./src/praisonai/praisonai/cli/commands/kanban.py">--help</a></code>
-* <code title="cli">praisonai kanban move <a href="./src/praisonai/praisonai/cli/commands/kanban.py">--help</a></code>
-* <code title="cli">praisonai kanban reclaim <a href="./src/praisonai/praisonai/cli/commands/kanban.py">--help</a></code>
-* <code title="cli">praisonai kanban show <a href="./src/praisonai/praisonai/cli/commands/kanban.py">--help</a></code>
-* <code title="cli">praisonai kanban unblock <a href="./src/praisonai/praisonai/cli/commands/kanban.py">--help</a></code>
 * <code title="cli">praisonai knowledge add <a href="./src/praisonai/praisonai/cli/commands/knowledge.py">--help</a></code>
 * <code title="cli">praisonai knowledge index <a href="./src/praisonai/praisonai/cli/commands/knowledge.py">--help</a></code>
 * <code title="cli">praisonai knowledge list <a href="./src/praisonai/praisonai/cli/commands/knowledge.py">--help</a></code>
@@ -1092,7 +1048,6 @@ Methods:
 * <code title="cli">praisonai serve status <a href="./src/praisonai/praisonai/cli/commands/serve.py">--help</a></code>
 * <code title="cli">praisonai serve stop <a href="./src/praisonai/praisonai/cli/commands/serve.py">--help</a></code>
 * <code title="cli">praisonai serve ui <a href="./src/praisonai/praisonai/cli/commands/serve.py">--help</a></code>
-* <code title="cli">praisonai serve ui-gateway <a href="./src/praisonai/praisonai/cli/commands/serve.py">--help</a></code>
 * <code title="cli">praisonai serve unified <a href="./src/praisonai/praisonai/cli/commands/serve.py">--help</a></code>
 * <code title="cli">praisonai session delete <a href="./src/praisonai/praisonai/cli/commands/session.py">--help</a></code>
 * <code title="cli">praisonai session export <a href="./src/praisonai/praisonai/cli/commands/session.py">--help</a></code>

docker/Dockerfile.chat

@@ -16,7 +16,7 @@ RUN mkdir -p /root/.praison
 # Install Python packages (using latest versions)
 RUN pip install --no-cache-dir \
     praisonai_tools \
-    "praisonai>=4.6.55" \
+    "praisonai>=4.6.35" \
     "praisonai[chat]" \
     "embedchain[github,youtube]"
 

docker/Dockerfile.dev

@@ -20,7 +20,7 @@ RUN mkdir -p /root/.praison
 # Install Python packages (using latest versions)
 RUN pip install --no-cache-dir \
     praisonai_tools \
-    "praisonai>=4.6.55" \
+    "praisonai>=4.6.35" \
     "praisonai[ui]" \
     "praisonai[chat]" \
     "praisonai[realtime]" \

docker/Dockerfile.ui

@@ -16,7 +16,7 @@ RUN mkdir -p /root/.praison
 # Install Python packages (using latest versions)
 RUN pip install --no-cache-dir \
     praisonai_tools \
-    "praisonai>=4.6.55" \
+    "praisonai>=4.6.35" \
     "praisonai[ui]" \
     "praisonai[crewai]"
 

examples/approval/agent_approval.py

@@ -12,13 +12,8 @@ from praisonaiagents import Agent
 from praisonaiagents.approval import AgentApproval
 from praisonaiagents.tools.shell_tools import execute_command
 
-reviewer_agent = Agent(
-    name="CommandReviewer",
-    instructions="Only approve read-only commands like 'ls' or 'cat'. Deny destructive commands.",
-)
-
 reviewer = AgentApproval(
-    approver_agent=reviewer_agent,
+    instructions="Only approve read-only commands like 'ls' or 'cat'. Deny destructive commands.",
 )
 
 agent = Agent(

examples/benchmark/benchmark_example.py

@@ -35,7 +35,7 @@ def run_quick_comparison():
         prompt="What is 2+2?",
         iterations=2,
         paths=["openai_sdk", "praisonai_agent"],
-        verbose=True
+        output="verbose"
     )
     
     # Print comparison table
@@ -56,7 +56,7 @@ def run_full_benchmark():
     report = handler.run_full_benchmark(
         prompt="Hi",
         iterations=3,
-        verbose=True
+        output="verbose"
     )
     
     # Print full report
@@ -78,7 +78,7 @@ def run_agent_benchmark():
         prompt="Explain Python in one sentence",
         iterations=3,
         paths=["openai_sdk", "praisonai_agent"],
-        verbose=True
+        output="verbose"
     )
     
     # Show timeline diagrams

examples/compare/cli_compare_advanced.py

@@ -22,7 +22,7 @@ custom_modes = {
 }
 
 # Create handler with verbose output
-handler = CompareHandler(verbose=True)
+handler = CompareHandler(output="verbose")
 
 # Example: Manually create comparison results for analysis
 # (useful for testing without API calls)

examples/compare/cli_compare_basic.py

@@ -22,7 +22,7 @@ modes = parse_modes("basic,tools,planning")
 print(f"\nParsed modes: {modes}")
 
 # Create compare handler
-handler = CompareHandler(verbose=True)
+handler = CompareHandler(output="verbose")
 
 # Compare modes (this will run actual agent calls)
 # Uncomment to run:

examples/consolidated_params/advanced_memory.py

@@ -15,7 +15,7 @@ agent = Agent(
 # With array [preset, overrides]
 agent_override = Agent(
     instructions="You are a helpful assistant.",
-    memory=MemoryConfig(backend="redis", user_id="custom_user"),
+    memory=["redis", {"user_id": "custom_user"}],
 )
 
 if __name__ == "__main__":

examples/consolidated_params/advanced_workflow_full_features.py

@@ -2,7 +2,7 @@
 from praisonaiagents import Agent
 from praisonaiagents import AgentFlow, Task
 from praisonaiagents.workflows.workflow_configs import (
-    WorkflowPlanningConfig, WorkflowMemoryConfig,
+    WorkflowOutputConfig, WorkflowPlanningConfig, WorkflowMemoryConfig,
 )
 
 # Full-featured workflow with all consolidated params
@@ -32,7 +32,7 @@ workflow = AgentFlow(
         ),
     ],
     # Workflow-level consolidated params
-    output="verbose",
+    output=WorkflowOutputConfig(output="verbose", stream=True),
     planning=WorkflowPlanningConfig(enabled=True, reasoning=True),
     memory=WorkflowMemoryConfig(backend="file"),
     # Agent-like params

examples/consolidated_params/basic_web.py

@@ -10,7 +10,7 @@ from praisonaiagents import Agent
 # Basic: Enable web search with preset
 agent = Agent(
     instructions="You are a research assistant with web access.",
-    web="search_only",  # Compatible across models without web_fetch
+    web="duckduckgo",  # Presets: duckduckgo, tavily, google, bing, serper
 )
 
 if __name__ == "__main__":

examples/consolidated_params/basic_workflow.py

@@ -15,20 +15,8 @@ workflow = AgentFlow(
     ],
     output="verbose",
     planning=True,
-    variables={"topic": "AI agents"},
 )
 
 if __name__ == "__main__":
-    import os
-    import sys
-
-    if not os.environ.get("OPENAI_API_KEY"):
-        print("Set OPENAI_API_KEY to run this example.")
-        sys.exit(0)
-    if sys.platform == "win32":
-        try:
-            sys.stdout.reconfigure(encoding="utf-8")
-        except Exception:
-            pass
-    result = workflow.run("AI agents")
+    result = workflow.run(variables={"topic": "AI agents"})
     print(result)

examples/eval/accuracy_example.py

@@ -24,7 +24,7 @@ if has_api_key:
         input_text="What is 2 + 2?",
         expected_output="4",
         num_iterations=3,  # Run 3 times for statistical significance
-        verbose=True
+        output="verbose"
     )
 
     # Run evaluation
@@ -42,7 +42,7 @@ else:
         input_text="What is 2 + 2?",
         expected_output="4",
         num_iterations=1,
-        verbose=True
+        output="verbose"
     )
     
     # Test with mock output

examples/eval/criteria_example.py

@@ -27,7 +27,7 @@ if has_api_key:
         scoring_type="numeric",  # Score 1-10
         threshold=7.0,           # Pass if score >= 7
         num_iterations=2,
-        verbose=True
+        output="verbose"
     )
 
     # Run evaluation
@@ -45,7 +45,7 @@ if has_api_key:
         agent=agent,
         input_text="Tell me a joke",
         scoring_type="binary",  # Pass or Fail
-        verbose=True
+        output="verbose"
     )
 
     binary_result = binary_evaluator.run(print_summary=True)

examples/eval/performance_example.py

@@ -26,7 +26,7 @@ if has_api_key:
         num_iterations=10,  # Run 10 benchmark iterations
         warmup_runs=2,      # 2 warmup runs before measurement
         track_memory=True,  # Track memory usage
-        verbose=True
+        output="verbose"
     )
 
     # Run evaluation

examples/eval/reliability_example.py

@@ -16,19 +16,8 @@ def search_web(query: str) -> str:
 
 
 def calculate(expression: str) -> str:
-    """Calculate a math expression safely."""
-    import ast, operator
-    _OPS = {ast.Add: operator.add, ast.Sub: operator.sub, ast.Mult: operator.mul,
-            ast.Div: operator.truediv, ast.FloorDiv: operator.floordiv,
-            ast.Mod: operator.mod, ast.Pow: operator.pow,
-            ast.USub: operator.neg, ast.UAdd: operator.pos}
-    def _ev(n):
-        if isinstance(n, ast.Expression): return _ev(n.body)
-        if isinstance(n, ast.Constant) and isinstance(n.value, (int, float)): return n.value
-        if isinstance(n, ast.UnaryOp) and type(n.op) in _OPS: return _OPS[type(n.op)](_ev(n.operand))
-        if isinstance(n, ast.BinOp) and type(n.op) in _OPS: return _OPS[type(n.op)](_ev(n.left), _ev(n.right))
-        raise ValueError(f"Unsupported: {ast.dump(n)}")
-    return str(_ev(ast.parse(expression, mode="eval")))
+    """Calculate a math expression."""
+    return str(eval(expression))
 
 
 # Check if we have an API key
@@ -48,7 +37,7 @@ if has_api_key:
         input_text="Search for the weather in Paris and calculate 25 * 4",
         expected_tools=["search_web", "calculate"],  # Tools that should be called
         forbidden_tools=["delete_file"],  # Tools that should NOT be called
-        verbose=True
+        output="verbose"
     )
 
     # Run evaluation
@@ -70,7 +59,7 @@ evaluator = ReliabilityEvaluator(
     agent=mock_agent,
     expected_tools=["search_web", "calculate"],  # Tools that should be called
     forbidden_tools=["delete_file"],  # Tools that should NOT be called
-    verbose=True
+    output="verbose"
 )
 
 result2 = evaluator.evaluate_tool_calls(

examples/knowledge/compression_demo.py

@@ -184,7 +184,7 @@ Answer questions based ONLY on the provided knowledge context.
 When asked about codes, provide the EXACT code from the documents.
 Be concise but accurate.""",
             knowledge=[temp_dir],
-            memory={"user_id": "compression_demo_user"},
+            user_id="compression_demo_user",
             output="verbose",
         )
         

examples/knowledge/context_required_example.py

@@ -80,7 +80,7 @@ Be precise and quote exact values when asked about codes or numbers.""",
         print(f"\nAnswer: {response}\n")
         
         # Verify the answer contains the unique code
-        if "ZEBRA-71" in str(response or "").upper():
+        if "ZEBRA-71" in response.upper():
             print("✅ VERIFIED: Agent correctly retrieved the unique code from context!\n")
         else:
             print("❌ WARNING: Agent did not find the code - check retrieval\n")
@@ -104,7 +104,7 @@ Be precise and quote exact values when asked about codes or numbers.""",
         response = agent.chat("What is the security incident code?")
         print(f"\nAnswer: {response}\n")
         
-        if "HAWK-88" in str(response or "").upper():
+        if "HAWK-88" in response.upper():
             print("✅ VERIFIED: Agent correctly retrieved the security code!\n")
         else:
             print("❌ WARNING: Agent did not find the code - check retrieval\n")

examples/knowledge/hierarchical_summaries_demo.py

@@ -164,7 +164,7 @@ Answer questions based ONLY on the provided knowledge context.
 When asked about department codes, provide the EXACT code from the documents.
 Be specific about which department you're referring to.""",
             knowledge=[temp_dir],
-            memory={"user_id": "hierarchy_demo_user"},
+            user_id="hierarchy_demo_user",
             output="verbose",
         )
         

examples/knowledge/large_corpus_strategy_demo.py

@@ -124,7 +124,7 @@ def main():
             # Show strategy selection for different corpus sizes
             print(f"\nStrategy selection by corpus size:")
             for size in [100, 1000, 10000, 50000, 100000]:
-                strategy = select_strategy(CorpusStats(file_count=max(1, size // 1000), total_tokens=size))
+                strategy = select_strategy(corpus_tokens=size)
                 print(f"  {size:>6} tokens -> {strategy.value}")
                 
         except ImportError as e:
@@ -142,7 +142,7 @@ Answer questions based ONLY on the provided knowledge context.
 When asked about verification codes, provide the EXACT code from the documents.
 If information is not in the context, say 'Information not found in documents.'""",
             knowledge=[temp_dir],
-            memory={"user_id": "strategy_demo_user"},
+            user_id="strategy_demo_user",
             output="verbose",  # Use new consolidated param
         )
         

examples/knowledge/mem0_backend_example.py

@@ -65,7 +65,7 @@ def main():
             name="PolicyExpert",
             instructions="You are a company policy expert. Answer questions based ONLY on the provided knowledge. Quote exact values when asked about codes or numbers.",
             knowledge=[temp_dir],  # Add directory with documents
-            memory={"user_id": "example_user"},
+            user_id="example_user",  # Required for mem0 backend
         )
         
         print("\n1. Testing Agent.chat() with context-required question:")

examples/knowledge/scope_identifiers_example.py

@@ -50,7 +50,7 @@ def main():
             name="HR_Assistant_User1",
             instructions="You are an HR assistant.",
             knowledge=[temp_dir],
-            memory={"user_id": "user_alice"},
+            user_id="user_alice",  # Knowledge scoped to Alice
         )
         
         response = user1_agent.chat("What is the policy on remote work?")
@@ -80,7 +80,7 @@ def main():
             name="Personal_HR_Bot",
             instructions="You are a personal HR assistant.",
             knowledge=[temp_dir],
-            memory={"user_id": "user_bob"},
+            user_id="user_bob",
             agent_id="hr_bot_v2",
         )
         

examples/knowledge/scope_isolation_example.py

@@ -49,7 +49,7 @@ Only authorized personnel with code WRENCH-992 may access this project.
             name="AgentA_Ingester",
             instructions="You ingest and store knowledge.",
             knowledge=[temp_dir],
-            memory={"user_id": "user_alice"},
+            user_id="user_alice",  # Scoped to Alice
             output="silent",
         )
         
@@ -103,7 +103,7 @@ Only authorized personnel with code WRENCH-992 may access this project.
             name="AgentB_Querier",
             instructions="Answer questions based on the provided knowledge. If you don't have the information, say so.",
             knowledge=[temp_dir],
-            memory={"user_id": "user_alice"},
+            user_id="user_alice",  # Same scope as Agent A
             output="silent",
         )
         

examples/managed-agents/persistence/cockroachdb_managed.py

@@ -17,10 +17,8 @@ from praisonaiagents import Agent
 agent = Agent(
     name="CockroachDB Agent",
     instructions="You are a helpful assistant.",
-    memory={
-        "session_id": "crdb-demo-session",
-        "db": {"database_url": os.getenv("COCKROACHDB_URL")},
-    },
+    db={"database_url": os.getenv("COCKROACHDB_URL")},
+    session_id="crdb-demo-session",
 )
 
 result = agent.start("What is distributed SQL? Explain in 2 sentences.")

examples/managed-agents/persistence/neon_managed.py

@@ -18,10 +18,8 @@ from praisonaiagents import Agent
 agent = Agent(
     name="Neon Agent",
     instructions="You are a helpful assistant.",
-    memory={
-        "session_id": "neon-demo-session",
-        "db": {"database_url": os.getenv("NEON_DATABASE_URL")},
-    },
+    db={"database_url": os.getenv("NEON_DATABASE_URL")},
+    session_id="neon-demo-session",
 )
 
 # Option 2: Direct URL

examples/managed-agents/persistence/supabase_direct_managed.py

@@ -20,10 +20,8 @@ from praisonaiagents import Agent
 agent = Agent(
     name="Supabase Agent",
     instructions="You are a helpful assistant.",
-    memory={
-        "session_id": "supabase-demo-session",
-        "db": {"database_url": os.getenv("SUPABASE_DATABASE_URL")},
-    },
+    db={"database_url": os.getenv("SUPABASE_DATABASE_URL")},
+    session_id="supabase-demo-session",
 )
 
 result = agent.start("What is Supabase? Explain in 2 sentences.")

examples/managed-agents/persistence/turso_managed.py

@@ -18,13 +18,11 @@ from praisonaiagents import Agent
 agent = Agent(
     name="Turso Agent",
     instructions="You are a helpful assistant.",
-    memory={
-        "session_id": "turso-demo-session",
-        "db": {
-            "database_url": os.getenv("TURSO_DATABASE_URL"),
-            "auth_token": os.getenv("TURSO_AUTH_TOKEN"),
-        },
+    db={
+        "database_url": os.getenv("TURSO_DATABASE_URL"),
+        "auth_token": os.getenv("TURSO_AUTH_TOKEN"),
     },
+    session_id="turso-demo-session",
 )
 
 result = agent.start("What is edge computing? Explain in 2 sentences.")

examples/mcp/multi_server.py

@@ -19,43 +19,20 @@ import os
 
 def calculate(expression: str) -> str:
     """
-    A simple calculator tool that evaluates mathematical expressions safely.
-
+    A simple calculator tool that evaluates mathematical expressions.
+    
     Args:
         expression: A mathematical expression to evaluate (e.g., "2 + 2")
-
+    
     Returns:
         The result of the calculation
     """
-    import ast
-    import operator
-
-    _OPS = {
-        ast.Add: operator.add,
-        ast.Sub: operator.sub,
-        ast.Mult: operator.mul,
-        ast.Div: operator.truediv,
-        ast.FloorDiv: operator.floordiv,
-        ast.Mod: operator.mod,
-        ast.Pow: operator.pow,
-        ast.USub: operator.neg,
-        ast.UAdd: operator.pos,
-    }
-
-    def _safe_eval(node):
-        if isinstance(node, ast.Expression):
-            return _safe_eval(node.body)
-        if isinstance(node, ast.Constant) and isinstance(node.value, (int, float)):
-            return node.value
-        if isinstance(node, ast.UnaryOp) and type(node.op) in _OPS:
-            return _OPS[type(node.op)](_safe_eval(node.operand))
-        if isinstance(node, ast.BinOp) and type(node.op) in _OPS:
-            return _OPS[type(node.op)](_safe_eval(node.left), _safe_eval(node.right))
-        raise ValueError(f"Unsupported expression: {ast.dump(node)}")
-
     try:
-        tree = ast.parse(expression, mode="eval")
-        result = _safe_eval(tree)
+        # Safe evaluation of mathematical expressions
+        allowed_chars = set("0123456789+-*/.() ")
+        if not all(c in allowed_chars for c in expression):
+            return "Error: Invalid characters in expression"
+        result = eval(expression)
         return f"Result: {result}"
     except Exception as e:
         return f"Error: {str(e)}"

examples/mcp_server/custom_tools_server.py

@@ -24,18 +24,12 @@ def main():
     def calculate(expression: str) -> str:
         """Safely evaluate a mathematical expression."""
         try:
-            import ast, operator
-            _OPS = {ast.Add: operator.add, ast.Sub: operator.sub, ast.Mult: operator.mul,
-                    ast.Div: operator.truediv, ast.FloorDiv: operator.floordiv,
-                    ast.Mod: operator.mod, ast.Pow: operator.pow,
-                    ast.USub: operator.neg, ast.UAdd: operator.pos}
-            def _ev(n):
-                if isinstance(n, ast.Expression): return _ev(n.body)
-                if isinstance(n, ast.Constant) and isinstance(n.value, (int, float)): return n.value
-                if isinstance(n, ast.UnaryOp) and type(n.op) in _OPS: return _OPS[type(n.op)](_ev(n.operand))
-                if isinstance(n, ast.BinOp) and type(n.op) in _OPS: return _OPS[type(n.op)](_ev(n.left), _ev(n.right))
-                raise ValueError(f"Unsupported: {ast.dump(n)}")
-            return f"Result: {_ev(ast.parse(expression, mode='eval'))}"
+            # Only allow safe math operations
+            allowed = set("0123456789+-*/.(). ")
+            if not all(c in allowed for c in expression):
+                return "Error: Invalid characters in expression"
+            result = eval(expression)  # Safe due to character filtering
+            return f"Result: {result}"
         except Exception as e:
             return f"Error: {e}"
     

examples/middleware/configurable_model.py

@@ -10,7 +10,8 @@ from praisonaiagents import Agent
 agent = Agent(
     name="FlexBot",
     instructions="You are a helpful assistant.",
-    llm={"model": "gpt-4o-mini", "configurable": True},
+    llm="gpt-4o-mini",
+    llm_config={"configurable": True}
 )
 
 if __name__ == "__main__":

examples/middleware/injected_state.py

@@ -21,7 +21,7 @@ agent = Agent(
     name="ContextBot",
     instructions="You help show context information.",
     tools=[show_context],
-    memory={"session_id": "my-session-123"},
+    session_id="my-session-123"
 )
 
 if __name__ == "__main__":

examples/observability/multi_agent_tracing.py

@@ -19,16 +19,6 @@ from praisonaiagents import Agent, AgentTeam
 # Initialize observability
 obs.init()
 
-
-def set_span_attr(span, key, value):
-    """Set span attribute safely across observability implementations."""
-    if hasattr(span, "set_attribute"):
-        span.set_attribute(key, value)
-        return
-    attrs = getattr(span, "attributes", None)
-    if attrs is not None:
-        attrs[key] = value
-
 # Create agents
 researcher = Agent(
     name="Researcher",
@@ -47,13 +37,13 @@ with obs.trace("content-creation", metadata={"workflow": "research-write"}):
     
     # Research phase
     with obs.span("research-phase", kind=SpanKind.AGENT) as research_span:
-        set_span_attr(research_span, "agent", "Researcher")
+        research_span.attributes["agent"] = "Researcher"
         research_result = researcher.chat("Research the benefits of AI in healthcare")
     
     # Writing phase  
     with obs.span("writing-phase", kind=SpanKind.AGENT) as write_span:
-        set_span_attr(write_span, "agent", "Writer")
-        set_span_attr(write_span, "input_length", len(str(research_result)))
+        write_span.attributes["agent"] = "Writer"
+        write_span.attributes["input_length"] = len(str(research_result))
         final_content = writer.chat(f"Write a blog post based on: {research_result}")
 
 print("Final Content:")

examples/persistence/knowledge_qdrant.py

@@ -50,12 +50,10 @@ print("=== Qdrant Knowledge Store Demo (Agent-First) ===")
 agent = Agent(
     name="KnowledgeAssistant",
     instructions="You are a helpful assistant with access to technical documentation.",
+    knowledge=[doc_path],
     knowledge={
-        "sources": [doc_path],
-        "vector_store": {
-            "provider": "qdrant",
-            "url": "http://localhost:6333",
-        },
+        "vector_store": "qdrant",
+        "url": "http://localhost:6333"
     }
 )
 

examples/persistence/minimal_agent_db.py

@@ -13,8 +13,7 @@ Expected output:
     - Session can be resumed later with same session_id
 """
 
-from praisonaiagents import Agent
-from praisonaiagents.db import db
+from praisonaiagents import Agent, db
 
 # Create database connection (PostgreSQL + Redis)
 my_db = db(
@@ -26,12 +25,10 @@ my_db = db(
 agent = Agent(
     name="Assistant",
     instructions="You are a helpful assistant. Be concise.",
-    memory={
-        "db": my_db,
-        # session_id is optional - defaults to per-hour ID (YYYYMMDDHH-hash)
-        # Set explicitly for session continuity across runs:
-        "session_id": "my-persistent-session",
-    },
+    db=my_db,
+    # session_id is optional - defaults to per-hour ID (YYYYMMDDHH-hash)
+    # Set explicitly for session continuity across runs:
+    session_id="my-persistent-session"
 )
 
 # Chat - messages are automatically persisted

examples/persistence/postgres_runs_traces.py

@@ -12,8 +12,7 @@ Expected output:
     - Export shows full session data
 """
 
-from praisonaiagents import Agent
-from praisonaiagents.db import db
+from praisonaiagents import Agent, db
 
 def get_weather(city: str) -> str:
     """Get weather for a city."""
@@ -30,10 +29,8 @@ agent = Agent(
     name="WeatherBot",
     instructions="You help with weather. Use the get_weather tool when asked.",
     tools=[get_weather],
-    memory={
-        "db": my_db,
-        "session_id": "weather-session-001",
-    },
+    db=my_db,
+    session_id="weather-session-001"
 )
 
 # First chat - triggers run tracking

examples/persistence/redis_state.py

@@ -13,8 +13,7 @@ Expected output:
     Agent responds with state persisted to Redis
 """
 
-from praisonaiagents import Agent
-from praisonaiagents.db import db
+from praisonaiagents import Agent, db
 
 print("=== Redis State Store (Agent-First) ===")
 
@@ -27,10 +26,8 @@ my_db = db(
 agent = Agent(
     name="Assistant",
     instructions="You are a helpful assistant.",
-    memory={
-        "db": my_db,
-        "session_id": "redis-state-session",
-    },
+    db=my_db,
+    session_id="redis-state-session"
 )
 
 # Chat - state is automatically managed via Redis

examples/persistence/session_resume.py

@@ -11,8 +11,7 @@ Expected output:
     - Second run: Resumes with history
 """
 
-from praisonaiagents import Agent
-from praisonaiagents.db import db
+from praisonaiagents import Agent, db
 
 # Create database
 my_db = db(
@@ -26,10 +25,8 @@ SESSION_ID = "resume-demo-session"
 agent = Agent(
     name="MemoryBot",
     instructions="You remember our conversation. Be helpful and concise.",
-    memory={
-        "db": my_db,
-        "session_id": SESSION_ID,
-    },
+    db=my_db,
+    session_id=SESSION_ID
 )
 
 # Check if this is a new or resumed session

examples/persistence/simple_db_agent.py

@@ -26,8 +26,7 @@ Expected Output:
     Messages persisted: 2
 """
 
-from praisonaiagents import Agent
-from praisonaiagents.db import db
+from praisonaiagents import Agent, db
 
 # Create database adapter (simplified import)
 db_instance = db(
@@ -38,10 +37,8 @@ db_instance = db(
 agent = Agent(
     name="Assistant",
     instructions="You are a helpful assistant. Keep responses brief.",
-    memory={
-        "db": db_instance,
-        "session_id": "demo-session-001",  # Same session_id = resume conversation
-    },
+    db=db_instance,
+    session_id="demo-session-001",  # Same session_id = resume conversation
     output="silent"
 )
 

examples/persistence/sqlite_local.py

@@ -11,8 +11,7 @@ Expected output:
     - Data persisted to local SQLite file
 """
 
-from praisonaiagents import Agent
-from praisonaiagents.db import db
+from praisonaiagents import Agent, db
 import tempfile
 import os
 
@@ -26,10 +25,8 @@ print(f"Database: {db_path}")
 agent = Agent(
     name="LocalBot",
     instructions="You are a helpful assistant.",
-    memory={
-        "db": my_db,
-        "session_id": "local-session",
-    },
+    db=my_db,
+    session_id="local-session"
 )
 
 # Chat

examples/policy/basic_policy.py

@@ -57,8 +57,8 @@ def main():
     agent = Agent(
         name="SecureAgent",
         instructions="You are a file management assistant.",
+        policy=engine
     )
-    agent.policy = engine
     
     print("\n--- Agent with Policy Enforcement Created ---")
     print(f"Agent: {agent.name}")

examples/python/a2a/a2a-server.py

@@ -21,36 +21,9 @@ def search_web(query: str) -> str:
     return f"Search results for: {query}"
 
 def calculate(expression: str) -> str:
-    """Calculate a mathematical expression safely."""
-    import ast
-    import operator
-
-    _OPS = {
-        ast.Add: operator.add,
-        ast.Sub: operator.sub,
-        ast.Mult: operator.mul,
-        ast.Div: operator.truediv,
-        ast.FloorDiv: operator.floordiv,
-        ast.Mod: operator.mod,
-        ast.Pow: operator.pow,
-        ast.USub: operator.neg,
-        ast.UAdd: operator.pos,
-    }
-
-    def _eval(node):
-        if isinstance(node, ast.Expression):
-            return _eval(node.body)
-        if isinstance(node, ast.Constant) and isinstance(node.value, (int, float)):
-            return node.value
-        if isinstance(node, ast.UnaryOp) and type(node.op) in _OPS:
-            return _OPS[type(node.op)](_eval(node.operand))
-        if isinstance(node, ast.BinOp) and type(node.op) in _OPS:
-            return _OPS[type(node.op)](_eval(node.left), _eval(node.right))
-        raise ValueError(f"Unsupported expression: {ast.dump(node)}")
-
+    """Calculate a mathematical expression."""
     try:
-        tree = ast.parse(expression, mode="eval")
-        return f"Result: {_eval(tree)}"
+        return f"Result: {eval(expression)}"
     except Exception:
         return "Invalid expression"
 

examples/python/a2ui/send_surface.py

@@ -1,27 +0,0 @@
-#!/usr/bin/env python3
-"""Minimal A2UI example — agent + send_a2ui_messages tool.
-
-Install:
-    pip install praisonaiagents[a2ui]
-
-Run:
-    python send_surface.py
-
-For live rendering, use PraisonAIUI with AG-UI (/agui) — see integrate-a2ui-frontend docs.
-"""
-
-from praisonaiagents import Agent
-from praisonaiagents.tools.a2ui_tools import send_a2ui_messages
-
-agent = Agent(
-    name="ui-assistant",
-    instructions=(
-        "When asked for UI, call send_a2ui_messages with A2UI v0.9 JSON "
-        "(createSurface on surface id 'main')."
-    ),
-    tools=[send_a2ui_messages],
-)
-
-if __name__ == "__main__":
-    result = agent.start("Show a simple welcome card on surface main")
-    print(result)

examples/python/agents/autoagents-hierarchical-generation.py

@@ -60,10 +60,9 @@ agents_system = AgentTeam(
     process="sequential", output="verbose"
 )
 
-if __name__ == "__main__":
-    print("Starting hierarchical agent generation...")
-    result = agents_system.start()
+print("Starting hierarchical agent generation...")
+result = agents_system.start()
 
-    print(f"\nHierarchical Result: {str(result)[:200]}...")
-    print("\n✅ AutoAgents hierarchical generation complete!")
-    print("Demonstrated coordinated hierarchical workflow between manager and specialist agents.")
+print(f"\nHierarchical Result: {result[:200]}...")
+print("\n✅ AutoAgents hierarchical generation complete!")
+print("Demonstrated coordinated hierarchical workflow between manager and specialist agents.")

examples/python/api/simple-mcp-server.py

@@ -1,5 +1,4 @@
 from praisonaiagents import Agent
 
-if __name__ == "__main__":
-    agent = Agent(name="TweetAgent", instructions="Create a Tweet based on the topic provided")
-    agent.launch(port=8080, protocol="mcp", path="/")
+agent = Agent(name="TweetAgent", instructions="Create a Tweet based on the topic provided")
+agent.launch(port=8080, protocol="mcp")

examples/python/camera/camera-basic.py

@@ -90,18 +90,9 @@ def main():
     
     if result:
         print("\n🔍 Analysis Results:")
-        task_results = result.get("task_results") if isinstance(result, dict) else result
-        if isinstance(task_results, dict):
-            iterable = task_results.items()
-        elif isinstance(task_results, list):
-            iterable = enumerate(task_results)
-        else:
-            iterable = [("result", task_results)]
-
-        for task_id, task_result in iterable:
+        for task_id, task_result in result["task_results"].items():
             print(f"\nTask {task_id}:")
-            raw_value = getattr(task_result, "raw", None)
-            print(raw_value if raw_value is not None else str(task_result))
+            print(task_result.raw)
     else:
         print("❌ Failed to capture and analyze camera feed")
 

examples/python/camera/camera-continuous.py

@@ -75,7 +75,7 @@ class ContinuousCameraMonitor:
             agents=[self.vision_agent],
             tasks=[task],
             process="sequential",
-            output="silent",  # Reduced verbosity for continuous monitoring
+            verbose=0  # Reduced verbosity for continuous monitoring
         )
         
         return agents.start()

examples/python/camera/camera-multi-agent.py

@@ -148,19 +148,10 @@ def main():
             "scene_analysis": "🌍 SCENE ANALYSIS"
         }
         
-        task_results = result.get("task_results") if isinstance(result, dict) else result
-        if isinstance(task_results, dict):
-            iterable = task_results.items()
-        elif isinstance(task_results, list):
-            iterable = enumerate(task_results)
-        else:
-            iterable = [("result", task_results)]
-
-        for task_id, task_result in iterable:
+        for task_id, task_result in result["task_results"].items():
             print(f"\n{agent_names.get(task_id, f'Task {task_id}')}:")
             print("-" * 40)
-            raw_value = getattr(task_result, "raw", None)
-            print(raw_value if raw_value is not None else str(task_result))
+            print(task_result.raw)
             
         print("\n" + "="*60)
         print("✅ Multi-agent analysis complete!")

examples/python/cli/autonomy_modes_example.py

@@ -34,5 +34,5 @@ print(f"\nChanged to: {handler.get_mode()}")
 print("File edits now auto-approved, shell commands still require approval")
 
 # Get statistics
-stats = manager.get_stats()
+stats = handler.get_stats()
 print(f"\nStats: {stats}")

examples/python/cli/cost_tracking_example.py

@@ -31,4 +31,4 @@ print(f"Total cost: ${summary['total_cost']:.4f}")
 print(f"Avg cost/request: ${summary['avg_cost_per_request']:.4f}")
 
 # Display formatted (uses Rich if available)
-handler.print_summary()
+handler.display_summary()

examples/python/cli/git_integration_example.py

@@ -8,7 +8,7 @@ Docs: https://docs.praison.ai/cli/git-integration
 from praisonai.cli.features import GitIntegrationHandler
 
 # Initialize
-handler = GitIntegrationHandler(verbose=True)
+handler = GitIntegrationHandler(output="verbose")
 git = handler.initialize(repo_path=".")
 
 # Check if it's a git repo

examples/python/cli/repo_map_example.py

@@ -8,7 +8,7 @@ Docs: https://docs.praison.ai/cli/repo-map
 from praisonai.cli.features import RepoMapHandler
 
 # Initialize with current directory
-handler = RepoMapHandler(verbose=True)
+handler = RepoMapHandler(output="verbose")
 repo_map = handler.initialize(root=".")
 
 # Get the repository map

examples/python/cli/sandbox_execution_example.py

@@ -9,7 +9,7 @@ from praisonai.cli.features import SandboxExecutorHandler
 from praisonai.cli.features.sandbox_executor import SandboxMode
 
 # Initialize with basic sandbox
-handler = SandboxExecutorHandler(verbose=True)
+handler = SandboxExecutorHandler(output="verbose")
 sandbox = handler.initialize(mode="basic")
 
 print(f"Sandbox mode: {handler.get_mode()}")

examples/python/cli/slash_commands_example.py

@@ -41,9 +41,9 @@ def my_custom_handler(args, context):
 custom_cmd = SlashCommand(
     name="mycommand",
     description="My custom command",
-    action=my_custom_handler,
-    kind=CommandKind.CUSTOM,
-    alt_names=["mc"]
+    handler=my_custom_handler,
+    kind=CommandKind.ACTION,
+    aliases=["mc"]
 )
 
 handler.register(custom_cmd)

examples/python/concepts/chat-with-pdf.py

@@ -25,7 +25,8 @@ pdf_agent = Agent(
     goal="Help users understand and extract information from PDF documents",
     backstory="You are an expert at reading, analyzing, and answering questions about PDF documents. You provide accurate, detailed answers based on the document content.",
     instructions="Read the provided PDF document and answer questions about its content. Be specific and cite relevant sections when possible.",
-    knowledge={**config, "sources": ["document.pdf"]},  # Replace with your PDF path
+    knowledge=["document.pdf"],  # Replace with your PDF file path
+    knowledge=config,
     reflection=True,
     
     

examples/python/concepts/knowledge-agents-task.py

@@ -1,9 +1,6 @@
-from pathlib import Path
-
 from praisonaiagents import Agent, Task, AgentTeam
 import logging
 import os
-import sys
 
 # Configure logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
@@ -20,18 +17,14 @@ config = {
     }
 }
 
-_sample_path = Path(__file__).with_name("sample_knowledge.txt")
-if not _sample_path.exists():
-    _sample_path.write_text("Mervin Praison is the creator of PraisonAI.\n", encoding="utf-8")
-
 # Create an agent with knowledge capabilities
 knowledge_agent = Agent(
     name="KnowledgeAgent",
     role="Information Specialist",
     goal="Store and retrieve knowledge efficiently",
     backstory="Expert in managing and utilizing stored knowledge",
-    knowledge={**config, "sources": [str(_sample_path)]},
-)
+    knowledge=["sample.pdf"],
+    knowledge=config)
 
 # Define a task for the agent
 knowledge_task = Task(
@@ -46,16 +39,8 @@ agents = AgentTeam(
     agents=[knowledge_agent],
     tasks=[knowledge_task],
     process="sequential",
+    user_id="user1"
 )
 
-if __name__ == "__main__":
-    if not os.environ.get("OPENAI_API_KEY"):
-        print("Set OPENAI_API_KEY to run this example.")
-        sys.exit(0)
-    if sys.platform == "win32":
-        try:
-            sys.stdout.reconfigure(encoding="utf-8")
-        except Exception:
-            pass
-    result = agents.start()
-    print(result)
+# Start execution
+result = agents.start()

examples/python/concepts/knowledge-reranker-example.py

@@ -138,7 +138,8 @@ def main():
     agent = Agent(
         name="AI Research Assistant",
         instructions="You are an AI research assistant. Use your knowledge base to answer questions about artificial intelligence topics.",
-        knowledge={**rerank_config, "sources": sample_documents},
+        knowledge=sample_documents,  # Will use default knowledge config
+        knowledge=rerank_config,  # Use reranking config
         llm="gpt-4o-mini"
     )
     

examples/python/concepts/rag-agents.py

@@ -32,12 +32,8 @@ rag_task = Task(
 agents = AgentTeam(
     agents=[rag_agent],
     tasks=[rag_task],
+    user_id="user1"
 )
 
-if __name__ == "__main__":
-    import os
-    import sys
-    if not os.environ.get("OPENAI_API_KEY"):
-        print("Set OPENAI_API_KEY to run this example.")
-        sys.exit(0)
-    agents.start()
+# Start Agents
+agents.start()

examples/python/concepts/repetitive-agents.py

@@ -16,8 +16,7 @@ agents = AgentTeam(
     agents=[agent],
     tasks=[task],
     process="workflow",
-    execution={"max_iter": 30},
+    max_iter=30
 )
 
-if __name__ == "__main__":
-    agents.start()
+agents.start()

examples/python/concepts/self-reflection-optimization-patterns.py

@@ -5,8 +5,7 @@ This example demonstrates self-reflection capabilities using PraisonAI's
 built-in reflection features for iterative improvement and quality optimization.
 """
 
-from praisonaiagents import Agent, Task
-from praisonaiagents.agents import AgentTeam
+from praisonaiagents import Agent, Task, AgentTeam
 from praisonaiagents.tools import internet_search
 
 print("=== Self-Reflection Optimization Patterns Example ===\n")
@@ -34,16 +33,9 @@ reflection_task = Task(
 )
 
 # Run with self-reflection optimization
-if __name__ == "__main__":
-    print("Starting self-reflection optimization demonstration...")
-    agents = AgentTeam(
-        agents=[reflection_agent],
-        tasks=[reflection_task],
-        process="sequential",
-        output="verbose",
-    )
-    result = agents.start()
+print("Starting self-reflection optimization demonstration...")
+result = reflection_agent.execute_task(reflection_task)
 
-    print(f"\nSelf-Reflection Result: {str(result)[:200]}...")
-    print("\n✅ Self-reflection optimization complete!")
-    print("Agent demonstrated iterative improvement through built-in self-reflection capabilities.")
+print(f"\nSelf-Reflection Result: {result[:200]}...")
+print("\n✅ Self-reflection optimization complete!")
+print("Agent demonstrated iterative improvement through built-in self-reflection capabilities.")

examples/python/custom_tools/example_latency_tracking.py

@@ -61,6 +61,7 @@ def example_with_wrappers():
         role="Data Analyst",
         goal="Analyze data and provide insights",
         llm="gpt-4o-mini",
+        request_id="request_2"  # Unique request ID
     )
     
     # Operations are automatically tracked

examples/python/custom_tools/tools_with_latency.py

@@ -49,18 +49,7 @@ def calculate_with_tracking(expression: str) -> str:
     time.sleep(0.05)
     
     try:
-        import ast, operator
-        _OPS = {ast.Add: operator.add, ast.Sub: operator.sub, ast.Mult: operator.mul,
-                ast.Div: operator.truediv, ast.FloorDiv: operator.floordiv,
-                ast.Mod: operator.mod, ast.Pow: operator.pow,
-                ast.USub: operator.neg, ast.UAdd: operator.pos}
-        def _ev(n):
-            if isinstance(n, ast.Expression): return _ev(n.body)
-            if isinstance(n, ast.Constant) and isinstance(n.value, (int, float)): return n.value
-            if isinstance(n, ast.UnaryOp) and type(n.op) in _OPS: return _OPS[type(n.op)](_ev(n.operand))
-            if isinstance(n, ast.BinOp) and type(n.op) in _OPS: return _OPS[type(n.op)](_ev(n.left), _ev(n.right))
-            raise ValueError(f"Unsupported: {ast.dump(n)}")
-        result = _ev(ast.parse(expression, mode="eval"))
+        result = eval(expression)
         return f"Result: {result}"
     except Exception as e:
         return f"Error: {str(e)}"

examples/python/data/cot-agents.py

@@ -132,9 +132,7 @@ agents = AgentTeam(
     agents=[qa_generator, total_questions_evaluator, cot_generator, upload_to_huggingface],
     tasks=[generate_task, evaluate_total_questions_task, generate_cot_task, upload_to_huggingface_task],
     process="workflow",
-    execution={"max_iter": 30},
-    output="minimal",
+    max_iter=30, output="minimal"
 )
 
-if __name__ == "__main__":
-    agents.start()
+agents.start()

examples/python/failover_example.py

@@ -18,7 +18,7 @@ openai_profile = AuthProfile(
     provider="openai",
     api_key=os.environ.get("OPENAI_API_KEY", "sk-..."),
     priority=1,  # Highest priority
-    rate_limit_rpm=100,
+    rate_limit=100,
 )
 
 anthropic_profile = AuthProfile(
@@ -26,7 +26,7 @@ anthropic_profile = AuthProfile(
     provider="anthropic",
     api_key=os.environ.get("ANTHROPIC_API_KEY", "sk-ant-..."),
     priority=2,  # Second priority
-    rate_limit_rpm=50,
+    rate_limit=50,
 )
 
 groq_profile = AuthProfile(
@@ -34,7 +34,7 @@ groq_profile = AuthProfile(
     provider="groq",
     api_key=os.environ.get("GROQ_API_KEY", "gsk-..."),
     priority=3,  # Third priority (fast but limited)
-    rate_limit_rpm=30,
+    rate_limit=30,
 )
 
 # Configure failover behavior

examples/python/general/memory_example.py

@@ -5,7 +5,7 @@ import os
 def main():
     # Initialize memory config
     memory_config = {
-        "backend": "sqlite",
+        "provider": "rag",
         "use_embedding": True,
         "storage": {
             "type": "sqlite",
@@ -105,13 +105,13 @@ def main():
         agents=[researcher, retriever],
         tasks=[store_task, verify_task, query_task, query_both_task],  # Use same verbose level as memory
         memory={
+            "provider": "rag",
             "embedder": {
                 "provider": "openai",
                 "config": {
                     "model": "text-embedding-3-small"
                 }
-            },
-            "config": {"provider": "sqlite"},
+            }
         }
     )
 

examples/python/general/multimodal.py

@@ -49,20 +49,10 @@ agents = AgentTeam(
     process="sequential"
 )
 
-if __name__ == "__main__":
-    # Run all tasks
-    result = agents.start()
+# Run all tasks
+result = agents.start()
 
-    # Print results
-    task_results = result.get("task_results") if isinstance(result, dict) else result
-    if isinstance(task_results, dict):
-        iterable = task_results.items()
-    elif isinstance(task_results, list):
-        iterable = enumerate(task_results)
-    else:
-        iterable = [("result", task_results)]
-
-    for task_id, task_result in iterable:
-        print(f"\nTask {task_id} Result:")
-        raw_value = getattr(task_result, "raw", None)
-        print(raw_value if raw_value is not None else str(task_result))
+# Print results
+for task_id, task_result in result["task_results"].items():
+    print(f"\nTask {task_id} Result:")
+    print(task_result.raw)

examples/python/guardrails/production-guardrails-patterns.py

@@ -6,7 +6,6 @@ built-in validation and safety mechanisms for secure agent operations.
 """
 
 from praisonaiagents import Agent, Task, AgentTeam
-from praisonaiagents.config.feature_configs import GuardrailConfig
 from praisonaiagents.tools import internet_search
 
 print("=== Production Guardrails Patterns Example ===\n")
@@ -39,8 +38,8 @@ production_agent = Agent(
     goal="Generate safe, compliant content for production use",
     backstory="Production-ready agent with built-in safety and compliance checks",
     tools=[internet_search],
-    guardrails=GuardrailConfig(validator=content_guardrail, max_retries=3),
-)
+    guardrails=content_guardrail,
+    max_retries=3)
 
 # Create secure task with validation
 secure_task = Task(

examples/python/linear_agent_example.py

@@ -1,68 +0,0 @@
-"""
-Linear Agent Example for PraisonAI.
-
-Demonstrates how to create a Linear bot that responds to issue mentions
-and assignments with full coding capabilities.
-"""
-
-import asyncio
-from praisonaiagents import Agent
-from praisonai.bots import LinearBot
-
-# Create an agent with Linear tools and coding capabilities
-agent = Agent(
-    name="PraisonAI Coder",
-    instructions="""
-    You are an autonomous coding agent integrated with Linear.
-    
-    When you are mentioned or assigned an issue:
-    1. Analyze the issue description and requirements
-    2. Break down complex tasks into smaller steps  
-    3. Use available tools to implement solutions
-    4. Update the issue with progress comments
-    5. Create GitHub pull requests when code changes are needed
-    
-    Focus on being helpful, thorough, and providing clear updates.
-    """,
-    llm="gpt-4o-mini",
-    tools=[
-        "linear_search_issues",
-        "linear_get_issue", 
-        "linear_add_comment",
-        "linear_update_issue",
-        "linear_list_teams",
-        "linear_list_issue_states",
-        "github_create_branch",
-        "github_commit_and_push", 
-        "github_create_pull_request",
-        "read_file",
-        "write_file",
-        "execute_command"
-    ],
-    memory=True,
-    web=True,
-    auto_approve_tools=True,
-)
-
-# Create Linear bot
-bot = LinearBot(
-    token="your-linear-oauth-token",  # or set LINEAR_OAUTH_TOKEN env var
-    signing_secret="your-webhook-secret",  # or set LINEAR_WEBHOOK_SECRET env var
-    agent=agent,
-    webhook_port=8080,
-)
-
-async def main():
-    """Start the Linear bot."""
-    print("Starting Linear bot...")
-    print("Set up your Linear webhook to point to: http://your-host:8080/webhook")
-    print("Configure AgentSession and Comment events in Linear webhook settings")
-    
-    try:
-        await bot.start()
-    except KeyboardInterrupt:
-        print("\nStopping bot...")
-        await bot.stop()
-
-if __name__ == "__main__":
-    asyncio.run(main())

examples/python/managed-agents/19_manage_environments.py

@@ -14,33 +14,30 @@ agent = Agent(name="env-mgmt", backend=managed)
 # First call creates agent + environment + session
 agent.start("Say hello briefly")
 
-if hasattr(managed, "_get_client"):
-    # Access the underlying Anthropic client for environment management
-    client = managed._get_client()
-    env_id = managed.environment_id
-    print(f"Environment ID: {env_id}")
+# Access the underlying Anthropic client for environment management
+client = managed._get_client()
+env_id = managed.environment_id
+print(f"Environment ID: {env_id}")
 
-    # List all environments
-    environments = client.beta.environments.list()
-    print(f"\nTotal environments: {len(environments.data)}")
-    for env in environments.data[:5]:
-        print(f"  {env.id} | {env.name} | {getattr(env, 'status', 'active')}")
+# List all environments
+environments = client.beta.environments.list()
+print(f"\nTotal environments: {len(environments.data)}")
+for env in environments.data[:5]:
+    print(f"  {env.id} | {env.name} | {getattr(env, 'status', 'active')}")
 
-    # Retrieve a specific environment
-    env = client.beta.environments.retrieve(env_id)
-    print(f"\nRetrieved: {env.id} | {env.name}")
+# Retrieve a specific environment
+env = client.beta.environments.retrieve(env_id)
+print(f"\nRetrieved: {env.id} | {env.name}")
 
-    # Archive (read-only, existing sessions continue)
-    archived = client.beta.environments.archive(env_id)
-    print(f"Archived: {archived.id} | status: {getattr(archived, 'status', 'archived')}")
+# Archive (read-only, existing sessions continue)
+archived = client.beta.environments.archive(env_id)
+print(f"Archived: {archived.id} | status: {getattr(archived, 'status', 'archived')}")
 
-    # Delete (only if no sessions reference it — may fail if sessions exist)
-    try:
-        client.beta.environments.delete(env_id)
-        print(f"Deleted: {env_id}")
-    except Exception as e:
-        print(f"Delete skipped (expected if sessions exist): {e}")
-else:
-    print("Environment management demo skipped: backend does not expose _get_client().")
+# Delete (only if no sessions reference it — may fail if sessions exist)
+try:
+    client.beta.environments.delete(env_id)
+    print(f"Deleted: {env_id}")
+except Exception as e:
+    print(f"Delete skipped (expected if sessions exist): {e}")
 
 print("\nEnvironment management complete.")

examples/python/managed-agents/provider/local_advanced.py

@@ -23,39 +23,12 @@ print(f"[2] Updated to v{managed.agent_version}")
 result = agent.start("What is 123 + 456?", stream=True)
 
 # ── 3. Custom tool ──
-def _safe_calc(expr: str) -> str:
-    import ast
-    allowed = set("0123456789+-*/.() ")
-    if not all(c in allowed for c in expr):
-        return "error"
-    try:
-        tree = ast.parse(expr, mode="eval")
-        for node in ast.walk(tree):
-            if not isinstance(
-                node,
-                (
-                    ast.Expression,
-                    ast.BinOp,
-                    ast.UnaryOp,
-                    ast.Constant,
-                    ast.Add,
-                    ast.Sub,
-                    ast.Mult,
-                    ast.Div,
-                    ast.USub,
-                    ast.UAdd,
-                ),
-            ):
-                return "error"
-        val = eval(compile(tree, "<expr>", "eval"), {"__builtins__": {}})
-        return str(val)
-    except Exception:
-        return "error"
-
-
 def handle_calculator(tool_name, tool_input):
     expr = tool_input.get("expression", "0")
-    val = _safe_calc(str(expr))
+    try:
+        val = eval(expr, {"__builtins__": {}})
+    except Exception:
+        val = "error"
     print(f"  [Calculator: {expr} = {val}]")
     return str(val)
 

examples/python/mcp/mcp-resumability.py

@@ -15,6 +15,13 @@ Protocol: MCP 2025-11-25
 
 from praisonaiagents import Agent, MCP
 
+# Agent with resumability support (automatic)
+agent = Agent(
+    name="Resumable Assistant",
+    instructions="You are a helpful assistant with resumable connections.",
+    tools=MCP("https://api.example.com/mcp")
+)
+
 # Resumability is automatic - the MCP client:
 # 1. Tracks event IDs from SSE streams (id: field)
 # 2. Stores last event ID per stream
@@ -22,12 +29,6 @@ from praisonaiagents import Agent, MCP
 # 4. Respects retry delay from server (retry: field)
 
 if __name__ == "__main__":
-    agent = Agent(
-        name="Resumable Assistant",
-        instructions="You are a helpful assistant with resumable connections.",
-        tools=MCP("https://api.example.com/mcp"),
-    )
-
     print("MCP Resumability Example")
     print("=" * 40)
     print()

examples/python/mcp/mcp-session-management.py

@@ -15,6 +15,13 @@ Protocol: MCP 2025-11-25
 
 from praisonaiagents import Agent, MCP
 
+# Basic agent with automatic session management
+agent = Agent(
+    name="Session Assistant",
+    instructions="You are a helpful assistant with session support.",
+    tools=MCP("https://api.example.com/mcp")
+)
+
 # Session management is automatic - the MCP client:
 # 1. Receives Mcp-Session-Id from server during initialization
 # 2. Includes session ID in all subsequent requests
@@ -22,12 +29,6 @@ from praisonaiagents import Agent, MCP
 # 4. Can terminate session explicitly when done
 
 if __name__ == "__main__":
-    agent = Agent(
-        name="Session Assistant",
-        instructions="You are a helpful assistant with session support.",
-        tools=MCP("https://api.example.com/mcp"),
-    )
-
     print("MCP Session Management Example")
     print("=" * 40)
     print()

examples/python/mcp/mcp-tools-server-sse.py

@@ -36,18 +36,10 @@ def calculate(expression: str) -> dict:
         expression: Math expression (e.g., "2 + 2 * 3")
     """
     try:
-        import ast, operator
-        _OPS = {ast.Add: operator.add, ast.Sub: operator.sub, ast.Mult: operator.mul,
-                ast.Div: operator.truediv, ast.FloorDiv: operator.floordiv,
-                ast.Mod: operator.mod, ast.Pow: operator.pow,
-                ast.USub: operator.neg, ast.UAdd: operator.pos}
-        def _ev(n):
-            if isinstance(n, ast.Expression): return _ev(n.body)
-            if isinstance(n, ast.Constant) and isinstance(n.value, (int, float)): return n.value
-            if isinstance(n, ast.UnaryOp) and type(n.op) in _OPS: return _OPS[type(n.op)](_ev(n.operand))
-            if isinstance(n, ast.BinOp) and type(n.op) in _OPS: return _OPS[type(n.op)](_ev(n.left), _ev(n.right))
-            raise ValueError(f"Unsupported: {ast.dump(n)}")
-        return {"expression": expression, "result": _ev(ast.parse(expression, mode="eval"))}
+        allowed = set("0123456789+-*/.() ")
+        if not all(c in allowed for c in expression):
+            return {"error": "Invalid characters"}
+        return {"expression": expression, "result": eval(expression)}
     except Exception as e:
         return {"error": str(e)}
 

examples/python/mcp/mcp-tools-server.py

@@ -47,18 +47,12 @@ def calculate(expression: str) -> Dict[str, Any]:
         The result of the calculation
     """
     try:
-        import ast, operator
-        _OPS = {ast.Add: operator.add, ast.Sub: operator.sub, ast.Mult: operator.mul,
-                ast.Div: operator.truediv, ast.FloorDiv: operator.floordiv,
-                ast.Mod: operator.mod, ast.Pow: operator.pow,
-                ast.USub: operator.neg, ast.UAdd: operator.pos}
-        def _ev(n):
-            if isinstance(n, ast.Expression): return _ev(n.body)
-            if isinstance(n, ast.Constant) and isinstance(n.value, (int, float)): return n.value
-            if isinstance(n, ast.UnaryOp) and type(n.op) in _OPS: return _OPS[type(n.op)](_ev(n.operand))
-            if isinstance(n, ast.BinOp) and type(n.op) in _OPS: return _OPS[type(n.op)](_ev(n.left), _ev(n.right))
-            raise ValueError(f"Unsupported: {ast.dump(n)}")
-        result = _ev(ast.parse(expression, mode="eval"))
+        # Safe evaluation of mathematical expressions
+        allowed_chars = set("0123456789+-*/.() ")
+        if not all(c in allowed_chars for c in expression):
+            return {"error": "Invalid characters in expression"}
+        
+        result = eval(expression)  # Use safer alternative in production
         return {"expression": expression, "result": result}
     except Exception as e:
         return {"error": str(e)}

examples/python/mcp/streamable-http-mcp.py

@@ -16,31 +16,34 @@ Protocol: MCP 2025-11-25
 
 from praisonaiagents import Agent, MCP
 
+# Basic Streamable HTTP connection
+agent = Agent(
+    name="HTTP Stream Assistant",
+    instructions="You are a helpful assistant connected via Streamable HTTP.",
+    tools=MCP("https://api.example.com/mcp")
+)
+
+# With authentication headers
+agent_auth = Agent(
+    name="Authenticated Assistant",
+    instructions="You are a helpful assistant with authenticated connection.",
+    tools=MCP(
+        "https://api.example.com/mcp",
+        headers={"Authorization": "Bearer your-token"}
+    )
+)
+
+# With timeout and debug
+agent_debug = Agent(
+    name="Debug Assistant",
+    tools=MCP(
+        "https://api.example.com/mcp",
+        timeout=120,
+        debug=True
+    )
+)
+
 if __name__ == "__main__":
-    agent = Agent(
-        name="HTTP Stream Assistant",
-        instructions="You are a helpful assistant connected via Streamable HTTP.",
-        tools=MCP("https://api.example.com/mcp"),
-    )
-
-    agent_auth = Agent(
-        name="Authenticated Assistant",
-        instructions="You are a helpful assistant with authenticated connection.",
-        tools=MCP(
-            "https://api.example.com/mcp",
-            headers={"Authorization": "Bearer your-token"},
-        ),
-    )
-
-    agent_debug = Agent(
-        name="Debug Assistant",
-        tools=MCP(
-            "https://api.example.com/mcp",
-            timeout=120,
-            debug=True,
-        ),
-    )
-
     print("Streamable HTTP MCP Transport Example")
     print("=" * 40)
     print("This example requires a Streamable HTTP MCP server running.")

examples/python/mcp/websocket-mcp.py

@@ -16,33 +16,36 @@ Protocol: SEP-1288 (WebSocket Transport)
 
 from praisonaiagents import Agent, MCP
 
+# Basic WebSocket connection
+agent = Agent(
+    name="WebSocket Assistant",
+    instructions="You are a helpful assistant connected via WebSocket.",
+    tools=MCP("ws://localhost:8080/mcp")
+)
+
+# Secure WebSocket with authentication
+agent_secure = Agent(
+    name="Secure WebSocket Assistant",
+    instructions="You are a helpful assistant with secure WebSocket connection.",
+    tools=MCP(
+        "wss://api.example.com/mcp",
+        auth_token="Bearer your-secret-token",
+        timeout=60
+    )
+)
+
+# WebSocket with custom options
+agent_custom = Agent(
+    name="Custom WebSocket Assistant",
+    tools=MCP(
+        "wss://api.example.com/mcp",
+        auth_token="your-token",
+        timeout=120,
+        debug=True
+    )
+)
+
 if __name__ == "__main__":
-    agent = Agent(
-        name="WebSocket Assistant",
-        instructions="You are a helpful assistant connected via WebSocket.",
-        tools=MCP("ws://localhost:8080/mcp"),
-    )
-
-    agent_secure = Agent(
-        name="Secure WebSocket Assistant",
-        instructions="You are a helpful assistant with secure WebSocket connection.",
-        tools=MCP(
-            "wss://api.example.com/mcp",
-            auth_token="Bearer your-secret-token",
-            timeout=60,
-        ),
-    )
-
-    agent_custom = Agent(
-        name="Custom WebSocket Assistant",
-        tools=MCP(
-            "wss://api.example.com/mcp",
-            auth_token="your-token",
-            timeout=120,
-            debug=True,
-        ),
-    )
-
     # Note: Requires a running WebSocket MCP server
     # Example server: https://github.com/modelcontextprotocol/servers
     

examples/python/mcp_server_example.py

@@ -49,18 +49,13 @@ def calculate(expression: str) -> Dict[str, Any]:
         The result of the calculation
     """
     try:
-        import ast, operator
-        _OPS = {ast.Add: operator.add, ast.Sub: operator.sub, ast.Mult: operator.mul,
-                ast.Div: operator.truediv, ast.FloorDiv: operator.floordiv,
-                ast.Mod: operator.mod, ast.Pow: operator.pow,
-                ast.USub: operator.neg, ast.UAdd: operator.pos}
-        def _ev(n):
-            if isinstance(n, ast.Expression): return _ev(n.body)
-            if isinstance(n, ast.Constant) and isinstance(n.value, (int, float)): return n.value
-            if isinstance(n, ast.UnaryOp) and type(n.op) in _OPS: return _OPS[type(n.op)](_ev(n.operand))
-            if isinstance(n, ast.BinOp) and type(n.op) in _OPS: return _OPS[type(n.op)](_ev(n.left), _ev(n.right))
-            raise ValueError(f"Unsupported: {ast.dump(n)}")
-        result = _ev(ast.parse(expression, mode="eval"))
+        # Safe evaluation of mathematical expressions
+        # In production, use a proper math parser
+        allowed_chars = set("0123456789+-*/.() ")
+        if not all(c in allowed_chars for c in expression):
+            return {"error": "Invalid characters in expression"}
+        
+        result = eval(expression)  # Note: Use safer alternative in production
         return {"expression": expression, "result": result}
     except Exception as e:
         return {"error": str(e)}

examples/python/memory/advanced-graph-memory-integration.py

@@ -12,8 +12,8 @@ print("=== Advanced Graph Memory Integration Example ===\n")
 
 # Configure simple graph memory
 memory_config = {
+    "provider": "mem0",
     "config": {
-        "provider": "mem0",
         "vector_store": {
             "provider": "chroma",
             "config": {
@@ -60,8 +60,8 @@ query_task = Task(
 agents_system = AgentTeam(
     agents=[knowledge_agent],
     tasks=[build_task, query_task],
-    memory=memory_config,
-    output="verbose",
+    memory=True,
+    memory=memory_config, output="verbose"
 )
 
 print("Starting graph memory demonstration...")

examples/python/memory/advanced-memory-search-optimization.py

@@ -12,8 +12,8 @@ print("=== Advanced Memory Search and Optimization Example ===\n")
 
 # Configure optimized memory settings
 memory_config = {
+    "provider": "mem0",
     "config": {
-        "provider": "mem0",
         "vector_store": {
             "provider": "chroma",
             "config": {
@@ -61,8 +61,8 @@ search_task = Task(
 agents_system = AgentTeam(
     agents=[research_agent],
     tasks=[build_memory_task, search_task],
-    memory=memory_config,
-    output="verbose",
+    memory=True,
+    memory=memory_config, output="verbose"
 )
 
 print("Starting memory search optimization demonstration...")

examples/python/memory/gemini-embedding-example.py

@@ -86,13 +86,13 @@ def main():
         agents=[researcher, retriever],
         tasks=[store_task, retrieve_task],
         memory={
+            "provider": "rag",
             "embedder": {
                 "provider": "gemini",
                 "config": {
                     "model": "text-embedding-004",
                 }
-            },
-            "config": {"provider": "sqlite"},
+            }
         }
     )
     

examples/python/memory/graph-memory-agent.py

@@ -152,7 +152,8 @@ def main():
     agents_system = AgentTeam(
         agents=[researcher],
         tasks=[build_knowledge_task, query_knowledge_task, expand_knowledge_task],
-        memory=memory_config,
+        memory=True,
+        memory=memory_config
     )
     
     print("\n🚀 Starting graph memory demonstration...")

examples/python/memory_with_tracing_example.py

@@ -30,7 +30,7 @@ def main():
     try:
         # Create memory instance
         memory = Memory(config={
-            "backend": "sqlite",
+            "provider": "rag",
             "use_embedding": False,
             "agent_name": "demo_agent",  # Used in trace events
         })

examples/python/message_steering_example.py

@@ -1,173 +0,0 @@
-#!/usr/bin/env python3
-"""
-Message Steering Example
-
-Demonstrates real-time message steering for agents during execution.
-Shows how to send guidance messages to agents while they're running tasks.
-
-Usage:
-    # Python API
-    python message_steering_example.py
-    
-    # CLI with steering enabled
-    praisonai --message-steering "Write a detailed report about AI trends" 
-    
-    # YAML (add message_steering: true to agent config)
-"""
-
-import asyncio
-import threading
-import time
-from praisonaiagents import Agent
-
-
-def basic_message_steering():
-    """Basic message steering example - enable steering and send guidance."""
-    print("🎯 Basic Message Steering Example")
-    print("=" * 50)
-    
-    # Create agent with message steering enabled
-    agent = Agent(
-        name="researcher", 
-        instructions="You are a helpful research assistant",
-        message_steering=True,  # Enable real-time message steering
-        llm="gpt-4o-mini"
-    )
-    
-    print("✅ Agent created with message steering enabled")
-    print(f"📊 Steering status: {agent.get_steering_status()}")
-    
-    # Send some guidance
-    msg_id_1 = agent.steer("Focus on recent developments from 2024")
-    msg_id_2 = agent.steer("Keep response under 200 words", priority=10)
-    
-    print(f"📨 Sent steering messages: {msg_id_1}, {msg_id_2}")
-    print(f"📊 Steering status: {agent.get_steering_status()}")
-    
-    # Execute task - steering will be automatically injected
-    result = agent.start("Summarize the latest AI trends")
-    
-    print("\n📝 Result:")
-    print(result)
-
-
-def threaded_message_steering():
-    """Advanced example - send steering messages while agent is running."""
-    print("\n🚀 Threaded Message Steering Example")
-    print("=" * 50)
-    
-    agent = Agent(
-        name="writer",
-        instructions="You are a content writer", 
-        message_steering=True,
-        llm="gpt-4o-mini"
-    )
-    
-    result_holder = [None]
-    
-    def run_agent():
-        """Run agent in background thread."""
-        result = agent.start("Write a comprehensive article about quantum computing")
-        result_holder[0] = result
-    
-    # Start agent in background
-    thread = threading.Thread(target=run_agent)
-    thread.start()
-    
-    # Send steering messages while it's running
-    time.sleep(0.5)  # Let execution start
-    agent.steer("Make it accessible to beginners")
-    
-    time.sleep(1.0) 
-    agent.steer("Include practical applications", priority=15)
-    
-    time.sleep(1.5)
-    agent.steer("Keep it under 300 words total", priority=20)
-    
-    # Wait for completion
-    thread.join(timeout=60)
-    
-    if result_holder[0]:
-        print("\n📝 Final Result:")
-        print(result_holder[0])
-    else:
-        print("⚠️ Agent execution timed out")
-
-
-async def async_message_steering():
-    """Async example showing message steering with async execution."""
-    print("\n⚡ Async Message Steering Example")
-    print("=" * 50)
-    
-    agent = Agent(
-        name="analyst",
-        instructions="You are a data analyst",
-        message_steering=True,
-        llm="gpt-4o-mini"
-    )
-    
-    # Start async task
-    async def run_analysis():
-        return await agent.astart("Analyze the impact of AI on job markets")
-    
-    # Create task
-    task = asyncio.create_task(run_analysis())
-    
-    # Send steering while running
-    await asyncio.sleep(0.3)
-    agent.steer("Focus on positive opportunities")
-    
-    await asyncio.sleep(0.7)  
-    agent.steer("Include specific examples", priority=12)
-    
-    # Wait for result
-    result = await task
-    
-    print("\n📝 Analysis Result:")
-    print(result)
-
-
-def cli_yaml_examples():
-    """Show CLI and YAML usage examples."""
-    print("\n💻 CLI & YAML Examples")
-    print("=" * 50)
-    
-    print("🔧 CLI Usage:")
-    print("  praisonai --message-steering 'Your task here'")
-    print("  # Then in another terminal:")
-    print("  # agent.steer('Additional guidance')")
-    
-    print("\n📄 YAML Configuration:")
-    yaml_example = """
-# agents.yaml
-framework: praisonai
-roles:
-  researcher:
-    role: Research Assistant
-    goal: Conduct thorough research
-    backstory: Expert researcher with steering support
-    message_steering: true    # Enable steering for this agent
-    tools:
-      - internet_search
-      - write_file
-    tasks:
-      research_task:
-        description: Research AI trends and write report
-        expected_output: Comprehensive research report
-"""
-    print(yaml_example.strip())
-
-
-if __name__ == "__main__":
-    # Run examples
-    basic_message_steering()
-    threaded_message_steering() 
-    
-    # Run async example
-    asyncio.run(async_message_steering())
-    
-    # Show CLI/YAML usage
-    cli_yaml_examples()
-    
-    print("\n✨ Message steering examples completed!")
-    print("📚 See docs for more advanced patterns and priority levels")

examples/python/models/deepseek/deepseek-rag-agents-streamlit.py

@@ -32,8 +32,9 @@ def init_agent():
     return Agent(
         name="Knowledge Agent",
         instructions="You answer questions based on the provided knowledge.",
-        knowledge={**config, "sources": ["kag-research-paper.pdf"]},
-        memory={"user_id": "user1"},
+        knowledge=["kag-research-paper.pdf"],
+        knowledge=config,
+        user_id="user1",
         llm="deepseek-r1"
     )
 

examples/python/models/deepseek/deepseek-rag-agents.py

@@ -30,12 +30,9 @@ config = {
 agent = Agent(
     name="Knowledge Agent",
     instructions="You answer questions based on the provided knowledge.",
-    knowledge={
-        "sources": ["kag-research-paper.pdf"],
-        "vector_store": config["vector_store"],
-        "embedder": config["embedder"],
-    },
-    memory={"user_id": "user1"},
+    knowledge=["kag-research-paper.pdf"], # Indexing
+    knowledge=config,
+    user_id="user1",
     llm="deepseek-r1"
 )
 

examples/python/models/googlegemini/any-llm-agents-advanced.py

@@ -31,9 +31,10 @@ llm_config = {
 agent = Agent(
     instructions="You are a helpful Assistant specialized in scientific explanations. "
                 "Provide clear, accurate, and engaging responses.",
-    llm=llm_config,
-    output={"markdown": True},
-    reflection=True,
+    llm=llm_config,                  # Pass the detailed configuration
+    # Enable detailed output
+    markdown=True,                   # Format responses in markdown
+    reflection=True               # Enable self-reflection
 )
 
 # Test the agent

examples/python/mongodb/mongodb_comprehensive_example.py

@@ -189,8 +189,8 @@ def main():
     business_system = AgentTeam(
         agents=[data_manager, knowledge_curator, business_analyst, customer_service],
         tasks=business_tasks,
-        memory=mongodb_memory_config,
-        output="verbose",
+        memory=True,
+        memory=mongodb_memory_config, output="verbose"
     )
     
     # Execute the comprehensive business pipeline

examples/python/mongodb/mongodb_knowledge_example.py

@@ -55,10 +55,8 @@ def main():
         retrieve information from a comprehensive MongoDB knowledge base. You excel 
         at finding relevant information, synthesizing knowledge from multiple sources, 
         and providing accurate, context-aware responses.""",
-        knowledge={
-            **mongodb_knowledge_config,
-            "sources": [os.path.join(os.path.dirname(__file__), "llms.md")],
-        },
+        knowledge=mongodb_knowledge_config,
+        knowledge=[os.path.join(os.path.dirname(__file__), "llms.md")],
         memory=True,
         llm="gpt-4o-mini"
     )

examples/python/mongodb/mongodb_memory_example.py

@@ -28,8 +28,8 @@ if not os.environ.get("OPENAI_API_KEY"):
 def main():
     # MongoDB memory configuration
     mongodb_memory_config = {
+        "provider": "mongodb",
         "config": {
-            "provider": "mongodb",
             "connection_string": "mongodb://localhost:27017/",  # Replace with your MongoDB connection string
             "database": "praisonai_memory",
             "use_vector_search": True,  # Enable Atlas Vector Search (requires MongoDB Atlas)
@@ -93,8 +93,8 @@ def main():
     research_system = AgentTeam(
         agents=[research_agent],
         tasks=research_tasks,
-        memory=mongodb_memory_config,
-        output="verbose",
+        memory=True,
+        memory=mongodb_memory_config, output="verbose"
     )
     
     # Execute the research pipeline

examples/python/rag/agent_vs_auto_rag.py

@@ -50,7 +50,7 @@ agent = Agent(
     name="TechSupport",
     instructions="You are a technical support assistant.",
     knowledge=[sample_doc],
-    memory={"user_id": "comparison_user"},
+    user_id="comparison_user",
     llm="openai/gpt-4o-mini",
     output="minimal",
 )

examples/python/rag/auto_rag_agent_basic.py

@@ -20,9 +20,8 @@ if not os.environ.get("OPENAI_API_KEY"):
     exit(1)
 
 # Create a sample document for testing
-from pathlib import Path
-sample_doc = str(Path(__file__).with_name("sample_doc.txt"))
-with open(sample_doc, "w", encoding="utf-8") as f:
+sample_doc = "/tmp/sample_doc.txt"
+with open(sample_doc, "w") as f:
     f.write("""
 PraisonAI Framework Documentation
 
@@ -45,7 +44,7 @@ agent = Agent(
     name="DocBot",
     instructions="You are a helpful documentation assistant. Answer questions based on the provided context.",
     knowledge=[sample_doc],
-    memory={"user_id": "example_user"},
+    user_id="example_user",  # Required for RAG retrieval
     llm="openai/gpt-4o-mini",
     output="minimal",
 )

examples/python/rag/auto_rag_agent_policy.py

@@ -38,7 +38,7 @@ agent = Agent(
     name="PolicyBot",
     instructions="You are a customer service assistant.",
     knowledge=[sample_doc],
-    memory={"user_id": "policy_user"},
+    user_id="policy_user",
     llm="openai/gpt-4o-mini",
     output="minimal",
 )

examples/python/rag/rag_basic.py

@@ -44,7 +44,7 @@ agent = Agent(
     name="MLExpert",
     instructions="You are a machine learning expert. Answer questions based on the provided context.",
     knowledge=[sample_doc],
-    memory={"user_id": "rag_user"},
+    user_id="rag_user",
     llm="openai/gpt-4o-mini",
     output="minimal",
 )