parishad 0.1.0__py3-none-any.whl

parishad/config/pipeline.core.yaml

@@ -0,0 +1,128 @@
+# Parishad-Core Pipeline Configuration
+# 8 roles: Madhya Sabha (Balanced Council)
+# Uses 2 model slots: mid + big
+
+name: "parishad-core"
+version: "0.1.0"
+description: "Balanced 8-role pipeline: Madhya Sabha with extended council"
+
+# Role definitions
+roles:
+  darbari:
+    name: "darbari"
+    class: "Darbari"
+    description: "Analyzer: Deeply analyzes query"
+    slot: "mid"
+    max_tokens: 1000
+    temperature: 0.3
+    output_schema: "task_spec"
+
+  majumdar:
+    name: "majumdar"
+    class: "Majumdar"
+    description: "Planner: Creates detailed plan"
+    slot: "mid"
+    max_tokens: 1200
+    temperature: 0.5
+    output_schema: "plan"
+
+  sar_senapati:
+    name: "sar_senapati"
+    class: "SarSenapati"
+    description: "Coordinator: Orchestrates execution strategy"
+    slot: "mid"
+    max_tokens: 1000
+    temperature: 0.4
+    output_schema: "execution_strategy"
+
+  sainik:
+    name: "sainik"
+    class: "Sainik"
+    description: "Implementor: Executes plan"
+    slot: "big"
+    max_tokens: 2500
+    temperature: 0.6
+    output_schema: "candidate"
+
+  prerak:
+    name: "prerak"
+    class: "Prerak"
+    description: "Checker: Validates correctness"
+    slot: "mid"
+    max_tokens: 1200
+    temperature: 0.3
+    output_schema: "verdict"
+
+  sacheev:
+    name: "sacheev"
+    class: "Sacheev"
+    description: "Secretary: Documents and summarizes"
+    slot: "mid"
+    max_tokens: 800
+    temperature: 0.3
+    output_schema: "summary"
+
+  dandadhyaksha:
+    name: "dandadhyaksha"
+    class: "Dandadhyaksha"
+    description: "Magistrate: Enforces quality standards"
+    slot: "mid"
+    max_tokens: 1000
+    temperature: 0.3
+    output_schema: "enforcement"
+
+  raja:
+    name: "raja"
+    class: "Raja"
+    description: "Judge: Final answer & formatting"
+    slot: "big"
+    max_tokens: 1500
+    temperature: 0.3
+    output_schema: "final_answer"
+
+# Pipeline execution order (8 roles)
+pipeline:
+  - darbari
+  - majumdar
+  - sar_senapati
+  - sainik
+  - prerak
+  - sacheev
+  - dandadhyaksha
+  - raja
+
+# Data flow between roles
+data_flow:
+  darbari:
+    inputs: ["user_query"]
+    outputs: ["task_spec"]
+  majumdar:
+    inputs: ["user_query", "task_spec"]
+    outputs: ["plan"]
+  sar_senapati:
+    inputs: ["user_query", "task_spec", "plan"]
+    outputs: ["execution_strategy"]
+  sainik:
+    inputs: ["user_query", "task_spec", "plan", "execution_strategy"]
+    outputs: ["candidate"]
+  prerak:
+    inputs: ["user_query", "candidate", "plan"]
+    outputs: ["verdict"]
+  sacheev:
+    inputs: ["user_query", "candidate", "verdict"]
+    outputs: ["summary"]
+  dandadhyaksha:
+    inputs: ["user_query", "candidate", "verdict", "summary"]
+    outputs: ["enforcement"]
+  raja:
+    inputs: ["user_query", "candidate", "verdict", "summary", "enforcement"]
+    outputs: ["final_answer"]
+
+# Pipeline-specific settings
+settings:
+  allow_retry: true
+  strict_validation: true
+  enable_checker: true
+  enable_planner: true
+  truncation_policy: "moderate"
+  max_total_tokens: 10000

parishad/config/pipeline.extended.yaml

@@ -0,0 +1,172 @@
+# Parishad-Extended Pipeline Configuration
+# 10 roles: Maha Sabha (Full 10-Unique-Role Council)
+# Uses 3 model slots: small + mid + big
+
+name: "parishad-extended"
+version: "0.1.0"
+description: "Maha Sabha: Full 10-role council including Lateral Thinker (Vidushak)"
+
+# Role definitions
+roles:
+  pantapradhan:
+    name: "pantapradhan"
+    class: "Pantapradhan"
+    description: "Prime Minister: Meta-planning and strategic oversight"
+    slot: "big"
+    max_tokens: 1500
+    temperature: 0.4
+    output_schema: "meta_plan"
+
+  darbari:
+    name: "darbari"
+    class: "Darbari"
+    description: "Analyzer: Deep analysis of query"
+    slot: "mid"
+    max_tokens: 1200
+    temperature: 0.3
+    output_schema: "task_spec"
+
+  majumdar:
+    name: "majumdar"
+    class: "Majumdar"
+    description: "Planner: Creates detailed execution plan"
+    slot: "mid"
+    max_tokens: 1400
+    temperature: 0.5
+    output_schema: "plan"
+
+  sar_senapati:
+    name: "sar_senapati"
+    class: "SarSenapati"
+    description: "Coordinator: Execution strategy"
+    slot: "mid"
+    max_tokens: 1000
+    temperature: 0.4
+    output_schema: "execution_strategy"
+
+  vidushak:
+    name: "vidushak"
+    class: "Vidushak"
+    description: "Lateral Thinker: Challenges plan with creative alternatives"
+    slot: "mid"
+    max_tokens: 1000
+    temperature: 0.8
+    output_schema: "creative_challenge"
+
+  sainik:
+    name: "sainik"
+    class: "Sainik"
+    description: "Implementor: Executes optimized plan"
+    slot: "big"
+    max_tokens: 2500
+    temperature: 0.6
+    output_schema: "candidate"
+
+  prerak:
+    name: "prerak"
+    class: "Prerak"
+    description: "Checker: Detailed validation"
+    slot: "mid"
+    max_tokens: 1200
+    temperature: 0.3
+    output_schema: "verdict"
+
+  sacheev:
+    name: "sacheev"
+    class: "Sacheev"
+    description: "Secretary: Documents and summarizes"
+    slot: "small"
+    max_tokens: 800
+    temperature: 0.3
+    output_schema: "summary"
+
+  dandadhyaksha:
+    name: "dandadhyaksha"
+    class: "Dandadhyaksha"
+    description: "Magistrate: Enforces quality"
+    slot: "mid"
+    max_tokens: 1000
+    temperature: 0.3
+    output_schema: "enforcement"
+
+  raja:
+    name: "raja"
+    class: "Raja"
+    description: "Judge: Final answer & formatting"
+    slot: "big"
+    max_tokens: 1800
+    temperature: 0.3
+    output_schema: "final_answer"
+
+# Pipeline execution order (10 UNIQUE roles)
+pipeline:
+  - pantapradhan
+  - darbari
+  - majumdar
+  - sar_senapati
+  - vidushak
+  - sainik
+  - prerak
+  - sacheev
+  - dandadhyaksha
+  - raja
+
+# Data flow between roles
+data_flow:
+  pantapradhan:
+    inputs: ["user_query"]
+    outputs: ["meta_plan"]
+  darbari:
+    inputs: ["user_query", "meta_plan"]
+    outputs: ["task_spec"]
+  majumdar:
+    inputs: ["user_query", "meta_plan", "task_spec"]
+    outputs: ["plan"]
+  sar_senapati:
+    inputs: ["user_query", "task_spec", "plan"]
+    outputs: ["execution_strategy"]
+  vidushak:
+    inputs: ["user_query", "plan"]
+    outputs: ["creative_challenge"]
+  sainik:
+    inputs: ["user_query", "task_spec", "plan", "execution_strategy", "creative_challenge"]
+    outputs: ["candidate"]
+  prerak:
+    inputs: ["user_query", "candidate", "plan"]
+    outputs: ["verdict"]
+  sacheev:
+    inputs: ["user_query", "candidate", "verdict"]
+    outputs: ["summary"]
+  dandadhyaksha:
+    inputs: ["user_query", "candidate", "verdict", "summary"]
+    outputs: ["enforcement"]
+  raja:
+    inputs: ["user_query", "candidate", "verdict", "summary", "enforcement"]
+    outputs: ["final_answer"]
+
+# Pipeline-specific settings
+settings:
+  allow_retry: true
+  strict_validation: true
+  enable_checker: true
+  enable_planner: true
+  truncation_policy: "conservative"
+  max_total_tokens: 15000
+
+# Budget policies
+budget:
+  max_tokens_per_query: 15000
+  
+  # Per-role token allocations
+  role_budgets:
+    pantapradhan: 1500
+    darbari: 1200
+    majumdar: 1400
+    sar_senapati: 1000
+    vidushak: 1000
+    sainik: 2500
+    prerak: 1200
+    sacheev: 800
+    dandadhyaksha: 1000
+    raja: 1800
+    reserve: 100

parishad/config/pipeline.fast.yaml

@@ -0,0 +1,89 @@
+# Parishad-Fast Pipeline Configuration
+# 5 roles: Full Council (Small/Faster Models)
+# Laghu Sabha matches standard structure but uses smaller/faster models
+
+name: "parishad-fast"
+version: "0.1.0"
+description: "Fast 5-role pipeline: Analyze -> Plan -> Execute -> Check -> Judge"
+
+# Role definitions
+roles:
+  darbari:
+    name: "darbari"
+    class: "Darbari"
+    description: "Analyzer: Identifies requirements"
+    slot: "small"
+    max_tokens: 800
+    temperature: 0.3
+    output_schema: "task_spec"
+
+  majumdar:
+    name: "majumdar"
+    class: "Majumdar"
+    description: "Planner: Creates simple plan"
+    slot: "mid"
+    max_tokens: 1000
+    temperature: 0.5
+    output_schema: "plan"
+
+  sainik:
+    name: "sainik"
+    class: "Sainik"
+    description: "Implementor: Executes plan"
+    slot: "mid"
+    max_tokens: 1500
+    temperature: 0.6
+    output_schema: "candidate"
+
+  prerak:
+    name: "prerak"
+    class: "Prerak"
+    description: "Checker: Validates output"
+    slot: "small"
+    max_tokens: 800
+    temperature: 0.3
+    output_schema: "verdict"
+
+  raja:
+    name: "raja"
+    class: "Raja"
+    description: "Judge: Formats final answer"
+    slot: "small"
+    max_tokens: 1000
+    temperature: 0.3
+    output_schema: "final_answer"
+
+# Execution order
+pipeline:
+  - darbari
+  - majumdar
+  - sainik
+  - prerak
+  - raja
+
+# Data flow between roles
+data_flow:
+  darbari:
+    inputs: ["user_query"]
+    outputs: ["task_spec"]
+  majumdar:
+    inputs: ["user_query", "task_spec"]
+    outputs: ["plan"]
+  sainik:
+    inputs: ["user_query", "task_spec", "plan"]
+    outputs: ["candidate"]
+  prerak:
+    inputs: ["user_query", "candidate", "plan"]
+    outputs: ["verdict"]
+  raja:
+    inputs: ["user_query", "candidate", "verdict", "plan"]
+    outputs: ["final_answer"]
+
+# Pipeline-specific settings
+settings:
+  allow_retry: true
+  strict_validation: false
+  enable_checker: true
+  enable_planner: true
+  truncation_policy: "aggressive"
+  max_total_tokens: 4000

parishad/config/user_config.py

@@ -0,0 +1,115 @@
+"""
+User-level configuration management for Parishad.
+
+Handles user preferences stored in ~/.parishad/config.yaml:
+- default_profile: Which model profile to use by default
+- default_mode: Which execution mode to use by default
+- model_dir: Where models are stored
+"""
+
+from __future__ import annotations
+
+import logging
+from dataclasses import dataclass
+from pathlib import Path
+from typing import Optional
+
+import yaml
+
+logger = logging.getLogger(__name__)
+
+
+@dataclass
+class UserConfig:
+    """User configuration settings."""
+    default_profile: str
+    default_mode: str
+    model_dir: str
+
+
+def get_user_config_dir() -> Path:
+    """
+    Return ~/.parishad directory (cross-platform via Path.home()).
+    
+    Returns:
+        Path to user config directory
+    """
+    return Path.home() / ".parishad"
+
+
+def get_user_config_path() -> Path:
+    """
+    Return path to ~/.parishad/config.yaml.
+    
+    Returns:
+        Path to user config file
+    """
+    return get_user_config_dir() / "config.yaml"
+
+
+def load_user_config() -> UserConfig:
+    """
+    Load user config if present, else return sensible defaults.
+    
+    Defaults:
+        default_profile: "local_gpu"
+        default_mode: "balanced"
+        model_dir: "~/.parishad/models"
+    
+    Returns:
+        UserConfig with settings
+    """
+    path = get_user_config_path()
+    
+    if not path.exists():
+        # Return default config
+        logger.debug("No user config found, using defaults")
+        return UserConfig(
+            default_profile="local_gpu",
+            default_mode="balanced",
+            model_dir=str(get_user_config_dir() / "models"),
+        )
+    
+    try:
+        data = yaml.safe_load(path.read_text(encoding="utf-8")) or {}
+        
+        config = UserConfig(
+            default_profile=data.get("default_profile", "local_gpu"),
+            default_mode=data.get("default_mode", "balanced"),
+            model_dir=data.get("model_dir", str(get_user_config_dir() / "models")),
+        )
+        
+        logger.debug(f"Loaded user config: profile={config.default_profile}, mode={config.default_mode}")
+        return config
+        
+    except Exception as e:
+        logger.warning(f"Failed to load user config: {e}, using defaults")
+        return UserConfig(
+            default_profile="local_gpu",
+            default_mode="balanced",
+            model_dir=str(get_user_config_dir() / "models"),
+        )
+
+
+def save_user_config(cfg: UserConfig) -> None:
+    """
+    Save user config to ~/.parishad/config.yaml (create dir if needed).
+    
+    Args:
+        cfg: UserConfig to save
+    """
+    cfg_dir = get_user_config_dir()
+    cfg_dir.mkdir(parents=True, exist_ok=True)
+    
+    data = {
+        "default_profile": cfg.default_profile,
+        "default_mode": cfg.default_mode,
+        "model_dir": cfg.model_dir,
+    }
+    
+    get_user_config_path().write_text(
+        yaml.safe_dump(data, sort_keys=False),
+        encoding="utf-8"
+    )
+    
+    logger.info(f"Saved user config: profile={cfg.default_profile}, mode={cfg.default_mode}")

parishad/data/catalog.py

@@ -0,0 +1,118 @@
+"""
+Model Catalog for Parishad Marketplace.
+Defines recommended models and sabha configurations.
+"""
+from dataclasses import dataclass, field
+from typing import Literal, List, Dict
+
+@dataclass
+class ModelEntry:
+    """A recommended model."""
+    name: str  # Display name
+    backend: Literal["llama_cpp", "ollama", "lm_studio"] # Defaulting to GGUF-compatible backends
+    model_id: str  # The ID used by the backend or the lookup key
+    min_ram_gb: int
+    description: str
+    hw_tags: List[str] = field(default_factory=list) # e.g. ["cpu", "cuda", "mlx", "metal"]
+    download_info: Dict[str, str] = field(default_factory=dict) # Hints for downloader: repo_id, filename
+
+@dataclass
+class SabhaConfig:
+    """Configuration for a Sabha tier."""
+    name: str
+    roles: List[str]
+    description: str
+    min_tokens_req: int 
+
+# --- SABHA DEFINITIONS ---
+SABHAS = {
+    "laghu": SabhaConfig(
+        name="Laghu Sabha",
+        description="A concise council of 5 core roles. Fast and efficient.",
+        roles=["raja", "dandadhyaksha", "sacheev", "prerak", "sainik"],
+        min_tokens_req=4096
+    ),
+    "mantri": SabhaConfig(
+        name="Mantri Parishad",
+        description="Expanded council with 8 roles for better planning.",
+        roles=[
+            "raja", "dandadhyaksha", "sacheev", "prerak", "majumdar",
+            "pantapradhan", "darbari", "sainik"
+        ],
+        min_tokens_req=8192
+    ),
+    "maha": SabhaConfig(
+        name="Maha Sabha",
+        description="The full royal court of 10 roles for maximum capability.",
+        roles=[
+            "raja", "dandadhyaksha", "sacheev", "prerak", "majumdar",
+            "pantapradhan", "darbari", "sar_senapati", "guptachar", "sainik"
+        ],
+        min_tokens_req=16384
+    )
+}
+
+# --- MODEL RECOMMENDATIONS ---
+
+# Note: We prioritize GGUF (llama_cpp) for local ease (no gated repos, lower VRAM reqs).
+# 'model_id' here acts as the 'spec' for ModelManager.download()
+
+MODELS = {
+    "entry": [
+        ModelEntry(
+            name="Qwen 2.5 (0.5B)", backend="llama_cpp", model_id="qwen2.5:0.5b",
+            min_ram_gb=2, description="Ultra-lightweight. Good for basic testing.", 
+            hw_tags=["cpu", "cuda", "mlx"]
+        ),
+        ModelEntry(
+            name="Llama 3.2 (1B)", backend="llama_cpp", model_id="llama3.2:1b",
+            min_ram_gb=4, description="Meta's smallest instruction model.", 
+            hw_tags=["cpu", "cuda", "mlx"]
+        ),
+        ModelEntry(
+            name="Phi-3.5 Mini (3.8B)", backend="llama_cpp", model_id="phi3:mini", # Using phi3:mini as close approx or update downloader
+            min_ram_gb=8, description="High capability for size. Strong reasoning.", 
+            hw_tags=["cpu", "cuda", "mlx"],
+            download_info={"repo": "microsoft/Phi-3-mini-4k-instruct-gguf", "file": "Phi-3-mini-4k-instruct-q4.gguf"}
+        ),
+        ModelEntry(
+            name="Gemma 2 (2B)", backend="llama_cpp", model_id="gemma2:2b",
+            min_ram_gb=6, description="Google's lightweight efficient model.", 
+            hw_tags=["cpu", "cuda", "mlx"]
+        ),
+        ModelEntry(
+            name="Qwen 2.5 (1.5B)", backend="llama_cpp", model_id="qwen2.5:1.5b",
+            min_ram_gb=4, description="Balanced small model.", 
+            hw_tags=["cpu", "cuda", "mlx"]
+        ),
+    ],
+    "mid": [
+        ModelEntry(
+            name="Qwen 2.5 (7B)", backend="llama_cpp", model_id="qwen2.5:7b",
+            min_ram_gb=16, description="Excellent all-rounder. Best in class 7B.", 
+            hw_tags=["cuda", "mlx"]
+        ),
+        ModelEntry(
+            name="Llama 3.1 (8B)", backend="llama_cpp", model_id="llama3.1:8b",
+            min_ram_gb=16, description="Meta's state-of-the-art 8B model.", 
+            hw_tags=["cuda", "metal"]
+        ),
+        ModelEntry(
+            name="Mistral 7B (v0.3)", backend="llama_cpp", model_id="mistral:7b",
+            min_ram_gb=16, description="Reliable workhorse from Mistral AI.", 
+            hw_tags=["cuda", "mlx"]
+        ),
+    ],
+    "high": [
+        ModelEntry(
+            name="Qwen 2.5 (14B)", backend="llama_cpp", model_id="Qwen/Qwen2.5-14B-Instruct-GGUF/qwen2.5-14b-instruct-q4_k_m.gguf",
+            min_ram_gb=28, description="Heavyweight reasoning. Great for coding.", 
+            hw_tags=["cuda", "mlx"]
+        ),
+        ModelEntry(
+            name="Mixtral 8x7B", backend="llama_cpp", model_id="TheBloke/Mixtral-8x7B-Instruct-v0.1-GGUF/mixtral-8x7b-instruct-v0.1.Q4_K_M.gguf",
+            min_ram_gb=26, description="Top-tier sparse MoE. Very fast.", 
+            hw_tags=["cuda", "metal"]
+        ),
+    ]
+}