enablement-agent 1.6.0__py3-none-any.whl

enablement_agent/PKG-INFO

@@ -0,0 +1,184 @@
+Metadata-Version: 2.4
+Name: enablement-agent
+Version: 1.6.0
+Summary: Centralized Enablement Agent — the canonical knowledge hub for Partner Core enablement content.
+Project-URL: Homepage, https://w.amazon.com/bin/view/AWS/Teams/PartnerCore/EnablementAgent/
+Project-URL: Repository, https://code.amazon.com/packages/EnablementAgent
+Author: Partner Core - Enablement Shared Services
+License-Expression: MIT
+Keywords: enablement,knowledge-hub,mcp,partner-core
+Classifier: Development Status :: 4 - Beta
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Requires-Python: >=3.11
+Requires-Dist: aiohttp>=3.9.0
+Requires-Dist: boto3>=1.34.0
+Requires-Dist: fastmcp>=2.0.0
+Requires-Dist: mcp>=1.0.0
+Requires-Dist: opensearch-py>=2.4.0
+Requires-Dist: pdfplumber>=0.10.0
+Requires-Dist: python-docx>=1.0.0
+Requires-Dist: requests-aws4auth>=1.2.0
+Provides-Extra: dev
+Requires-Dist: moto[athena,s3]>=5.0.0; extra == 'dev'
+Requires-Dist: mypy>=1.9.0; extra == 'dev'
+Requires-Dist: pytest-asyncio>=0.23.0; extra == 'dev'
+Requires-Dist: pytest>=7.0; extra == 'dev'
+Requires-Dist: ruff>=0.4.0; extra == 'dev'
+Description-Content-Type: text/markdown
+
+# Centralized Enablement Agent
+
+The canonical knowledge hub for all enablement content across Partner Core.
+
+## What it does
+
+This agent ingests, indexes, and serves enablement content — user guides, FAQs, demo scripts, training decks, playbooks, and more. Anyone in Partner Core can query it for answers, search for content, and monitor content freshness.
+
+## Architecture
+
+```
+┌─────────────────────────────────────────────────────────────────┐
+│  Content IN                                                      │
+│  ┌──────────────┐ ┌──────────────┐ ┌────────────┐ ┌──────────┐ │
+│  │ Delivery     │ │ Manual       │ │ Wiki       │ │ Highspot │ │
+│  │ Agent Push   │ │ Upload       │ │ Scraper    │ │ Scraper  │ │
+│  └──────┬───────┘ └──────┬───────┘ └─────┬──────┘ └────┬─────┘ │
+│         │                │               │              │       │
+│         └────────────────┴───────────────┴──────────────┘       │
+│                                  │                               │
+│                          ┌───────▼───────┐                      │
+│                          │  Orchestrator  │                      │
+│                          └───────┬───────┘                      │
+│                                  │                               │
+│         ┌────────────────────────┼────────────────────┐         │
+│         │                        │                    │         │
+│  ┌──────▼──────┐  ┌─────────────▼────────┐  ┌───────▼───────┐ │
+│  │ S3 Bucket   │  │ OpenSearch Serverless │  │ Content       │ │
+│  │ (raw files) │  │ (vector embeddings)  │  │ Registry      │ │
+│  └─────────────┘  └──────────────────────┘  └───────────────┘ │
+│                                                                  │
+│  Content OUT (MCP Tools via SSE)                                │
+│  ┌──────────────────────────────────────────────────────────┐   │
+│  │ query · search · upload · status · stale · refresh       │   │
+│  │ check_permissions · request_access                       │   │
+│  └──────────────────────────────────────────────────────────┘   │
+└─────────────────────────────────────────────────────────────────┘
+```
+
+## Quick Start (for users)
+
+Install and configure in one step:
+
+```bash
+pip install enablement-agent
+enablement-agent configure
+```
+
+Or if you use `uvx` (recommended):
+
+```bash
+uvx enablement-agent configure
+```
+
+This writes your `~/.kiro/settings/mcp.json` with:
+
+```json
+{
+  "mcpServers": {
+    "enablement-agent": {
+      "command": "uvx",
+      "args": ["enablement-agent", "serve"],
+      "env": {
+        "AWS_REGION": "us-east-1"
+      },
+      "disabled": false,
+      "autoApprove": ["ask", "status"]
+    }
+  }
+}
+```
+
+Or add it manually — that's the whole config.
+
+**Prerequisites:** AWS credentials configured (`midway` or `~/.aws/config`) with access to the shared S3 bucket, OpenSearch, and Bedrock.
+
+## Deployment
+
+Two modes:
+
+### Local (stdio) — current default
+Users install from PyPI and run locally. Shared backend (S3, OpenSearch, Bedrock) is accessed directly via AWS credentials.
+
+### Centralized (SSE) — future
+Once a domain + HTTPS cert is provisioned, the server can run on ECS Fargate with SSE transport. Users would just add a URL to their mcp.json.
+
+### Backend Infrastructure (shared, required for both modes)
+
+Deploy with CDK:
+
+```bash
+cd infra/
+pip install aws-cdk-lib constructs
+cdk deploy
+```
+
+### Environment Variables
+
+| Variable | Description |
+|----------|-------------|
+| `OPENSEARCH_ENDPOINT` | OpenSearch Serverless collection endpoint |
+| `S3_BUCKET` | S3 bucket name for content storage |
+| `AWS_REGION` | AWS region (default: us-east-1) |
+| `HIGHSPOT_API_KEY` | Highspot API key (or loaded from Secrets Manager) |
+| `LOG_LEVEL` | Logging level (default: INFO) |
+
+## Development
+
+```bash
+# Install with dev dependencies
+pip install -e ".[dev]"
+
+# Run tests
+pytest
+
+# Lint
+ruff check .
+
+# Type check
+mypy enablement_agent/
+```
+
+## Modules
+
+| Module | Purpose |
+|--------|---------|
+| `orchestrator.py` | Central coordinator, request routing, background jobs |
+| `mcp_tools.py` | FastMCP tool registration (9 tools) |
+| `search.py` | Hybrid BM25 + vector + metadata search engine |
+| `vector_store.py` | OpenSearch Serverless knn integration |
+| `query_handler.py` | NL question answering with LLM synthesis |
+| `ingestion.py` | 4 ingestion pipelines |
+| `content_registry.py` | Metadata store with CRUD and bulk queries |
+| `freshness_monitor.py` | Staleness detection and auto-refresh |
+| `rls_guard.py` | Unified RLS enforcement via Galaxi |
+| `permissions_cache.py` | TTL cache for RLS permission lookups |
+| `wiki_scraper.py` | Wiki page discovery and change detection |
+| `highspot_scraper.py` | Highspot spot monitoring |
+| `config.py` | All configuration constants and enums |
+| `cli.py` | CLI entry point (serve, configure, health) |
+
+## RLS (Row-Level Security)
+
+Content containing customer data is automatically classified:
+
+- **NONE** — General enablement, open to all ASP/SMGS
+- **TERRITORY** — Contains territory-level customer data
+- **PARTNER** — Contains partner-specific data (SPMS IDs)
+- **BROAD** — Requires Geo+ level access (L8 approval)
+
+Blocked content is always *visible* (title + lock icon) but redacted. Users get links to request access.
+
+## Freshness
+
+Content is flagged stale after 4 months (120 days) without update. Wiki and Highspot sources auto-refresh when the source changes. Notifications go to the `enablement-freshness-alerts` channel.

enablement_agent/README.md

@@ -0,0 +1,155 @@
+# Centralized Enablement Agent
+
+The canonical knowledge hub for all enablement content across Partner Core.
+
+## What it does
+
+This agent ingests, indexes, and serves enablement content — user guides, FAQs, demo scripts, training decks, playbooks, and more. Anyone in Partner Core can query it for answers, search for content, and monitor content freshness.
+
+## Architecture
+
+```
+┌─────────────────────────────────────────────────────────────────┐
+│  Content IN                                                      │
+│  ┌──────────────┐ ┌──────────────┐ ┌────────────┐ ┌──────────┐ │
+│  │ Delivery     │ │ Manual       │ │ Wiki       │ │ Highspot │ │
+│  │ Agent Push   │ │ Upload       │ │ Scraper    │ │ Scraper  │ │
+│  └──────┬───────┘ └──────┬───────┘ └─────┬──────┘ └────┬─────┘ │
+│         │                │               │              │       │
+│         └────────────────┴───────────────┴──────────────┘       │
+│                                  │                               │
+│                          ┌───────▼───────┐                      │
+│                          │  Orchestrator  │                      │
+│                          └───────┬───────┘                      │
+│                                  │                               │
+│         ┌────────────────────────┼────────────────────┐         │
+│         │                        │                    │         │
+│  ┌──────▼──────┐  ┌─────────────▼────────┐  ┌───────▼───────┐ │
+│  │ S3 Bucket   │  │ OpenSearch Serverless │  │ Content       │ │
+│  │ (raw files) │  │ (vector embeddings)  │  │ Registry      │ │
+│  └─────────────┘  └──────────────────────┘  └───────────────┘ │
+│                                                                  │
+│  Content OUT (MCP Tools via SSE)                                │
+│  ┌──────────────────────────────────────────────────────────┐   │
+│  │ query · search · upload · status · stale · refresh       │   │
+│  │ check_permissions · request_access                       │   │
+│  └──────────────────────────────────────────────────────────┘   │
+└─────────────────────────────────────────────────────────────────┘
+```
+
+## Quick Start (for users)
+
+Install and configure in one step:
+
+```bash
+pip install enablement-agent
+enablement-agent configure
+```
+
+Or if you use `uvx` (recommended):
+
+```bash
+uvx enablement-agent configure
+```
+
+This writes your `~/.kiro/settings/mcp.json` with:
+
+```json
+{
+  "mcpServers": {
+    "enablement-agent": {
+      "command": "uvx",
+      "args": ["enablement-agent", "serve"],
+      "env": {
+        "AWS_REGION": "us-east-1"
+      },
+      "disabled": false,
+      "autoApprove": ["ask", "status"]
+    }
+  }
+}
+```
+
+Or add it manually — that's the whole config.
+
+**Prerequisites:** AWS credentials configured (`midway` or `~/.aws/config`) with access to the shared S3 bucket, OpenSearch, and Bedrock.
+
+## Deployment
+
+Two modes:
+
+### Local (stdio) — current default
+Users install from PyPI and run locally. Shared backend (S3, OpenSearch, Bedrock) is accessed directly via AWS credentials.
+
+### Centralized (SSE) — future
+Once a domain + HTTPS cert is provisioned, the server can run on ECS Fargate with SSE transport. Users would just add a URL to their mcp.json.
+
+### Backend Infrastructure (shared, required for both modes)
+
+Deploy with CDK:
+
+```bash
+cd infra/
+pip install aws-cdk-lib constructs
+cdk deploy
+```
+
+### Environment Variables
+
+| Variable | Description |
+|----------|-------------|
+| `OPENSEARCH_ENDPOINT` | OpenSearch Serverless collection endpoint |
+| `S3_BUCKET` | S3 bucket name for content storage |
+| `AWS_REGION` | AWS region (default: us-east-1) |
+| `HIGHSPOT_API_KEY` | Highspot API key (or loaded from Secrets Manager) |
+| `LOG_LEVEL` | Logging level (default: INFO) |
+
+## Development
+
+```bash
+# Install with dev dependencies
+pip install -e ".[dev]"
+
+# Run tests
+pytest
+
+# Lint
+ruff check .
+
+# Type check
+mypy enablement_agent/
+```
+
+## Modules
+
+| Module | Purpose |
+|--------|---------|
+| `orchestrator.py` | Central coordinator, request routing, background jobs |
+| `mcp_tools.py` | FastMCP tool registration (9 tools) |
+| `search.py` | Hybrid BM25 + vector + metadata search engine |
+| `vector_store.py` | OpenSearch Serverless knn integration |
+| `query_handler.py` | NL question answering with LLM synthesis |
+| `ingestion.py` | 4 ingestion pipelines |
+| `content_registry.py` | Metadata store with CRUD and bulk queries |
+| `freshness_monitor.py` | Staleness detection and auto-refresh |
+| `rls_guard.py` | Unified RLS enforcement via Galaxi |
+| `permissions_cache.py` | TTL cache for RLS permission lookups |
+| `wiki_scraper.py` | Wiki page discovery and change detection |
+| `highspot_scraper.py` | Highspot spot monitoring |
+| `config.py` | All configuration constants and enums |
+| `cli.py` | CLI entry point (serve, configure, health) |
+
+## RLS (Row-Level Security)
+
+Content containing customer data is automatically classified:
+
+- **NONE** — General enablement, open to all ASP/SMGS
+- **TERRITORY** — Contains territory-level customer data
+- **PARTNER** — Contains partner-specific data (SPMS IDs)
+- **BROAD** — Requires Geo+ level access (L8 approval)
+
+Blocked content is always *visible* (title + lock icon) but redacted. Users get links to request access.
+
+## Freshness
+
+Content is flagged stale after 4 months (120 days) without update. Wiki and Highspot sources auto-refresh when the source changes. Notifications go to the `enablement-freshness-alerts` channel.

enablement_agent/__init__.py

@@ -0,0 +1,182 @@
+"""
+enablement_agent
+================
+
+Centralized Enablement Agent — the canonical knowledge hub for all
+enablement content across Partner Core.
+
+This agent is a sister to the Project Delivery Agent and CM Engine.
+It serves as the single place anyone in Partner Core goes to learn
+about anything that's been launched or is being launched.
+
+Architecture:
+    Content IN:  4 ingestion pipelines (Delivery Agent push, manual upload,
+                 wiki scraper, Highspot scraper)
+    Storage:     S3 + vector embeddings index with unified metadata
+    Content OUT: MCP tools for query, search, upload, and management
+
+Key Modules:
+    orchestrator      - Main entry point and request router
+    ingestion         - 4 ingestion pipelines (delivery, manual, wiki, highspot)
+    wiki_scraper      - Wiki discovery and change-detection scraping
+    highspot_scraper  - Highspot spot monitoring and content pulling
+    search            - Hybrid BM25 + vector + metadata search engine
+    query_handler     - NL question answering with LLM synthesis
+    content_registry  - Metadata store with CRUD and bulk queries
+    freshness_monitor - Staleness detection and auto-refresh
+    mcp_tools         - FastMCP tool registration for external access
+    config            - All configuration constants and enums
+
+Usage:
+    # As MCP server
+    from enablement_agent.mcp_tools import serve
+    serve(port=8400)
+
+    # Programmatic access
+    from enablement_agent import get_orchestrator
+    orchestrator = get_orchestrator()
+    await orchestrator.initialize()
+    answer = await orchestrator.handle_query("How does PBP work?")
+
+    # Delivery Agent integration
+    from enablement_agent import ingest_from_delivery_agent
+    result = await orchestrator.handle_delivery_agent_push(asset)
+"""
+
+__version__ = "1.0.0"
+__author__ = "Partner Core — Enablement Shared Services"
+
+# ---------------------------------------------------------------------------
+# Public API Exports
+# ---------------------------------------------------------------------------
+
+from .config import (
+    AGENT,
+    Audience,
+    ContentSource,
+    ContentType,
+    EMBEDDING,
+    FRESHNESS,
+    FreshnessStatus,
+    HIGHSPOT_SPOTS,
+    Phase,
+    S3,
+    SCHEDULE,
+    SEARCH,
+)
+from .content_registry import ContentItem, ContentRegistry, compute_content_hash
+from .freshness_monitor import FreshnessMonitor, FreshnessReport
+from .highspot_scraper import (
+    HighspotClient,
+    HighspotItem,
+    extract_highspot_metadata,
+    fetch_highspot_item,
+    fetch_spot_items_since,
+    run_scheduled_sync as run_highspot_sync,
+)
+from .ingestion import (
+    IngestionResult,
+    TextChunk,
+    chunk_text,
+    extract_text,
+    ingest_from_delivery_agent,
+    ingest_from_highspot,
+    ingest_from_wiki,
+    ingest_manual_upload,
+)
+from .mcp_tools import create_mcp_server, serve
+from .orchestrator import Orchestrator, get_orchestrator
+from .permissions_cache import PermissionsCache, get_partner_cache, get_user_cache
+from .query_handler import Citation, EnablementAnswer, QueryHandler
+from .search import (
+    BM25Index,
+    SearchEngine,
+    SearchFilters,
+    SearchResponse,
+    SearchResult,
+    VectorIndex,
+)
+from .vector_store import EmbeddingService, OpenSearchConfig, VectorStore
+from .wiki_scraper import (
+    WikiPage,
+    crawl_wiki_tree,
+    discover_wikis_for_project,
+    fetch_wiki_page,
+    infer_project_from_url,
+    register_project_wiki,
+    run_scheduled_scrape as run_wiki_scrape,
+)
+
+__all__ = [
+    # Version
+    "__version__",
+    # Config & Enums
+    "AGENT",
+    "Audience",
+    "ContentSource",
+    "ContentType",
+    "EMBEDDING",
+    "FRESHNESS",
+    "FreshnessStatus",
+    "HIGHSPOT_SPOTS",
+    "Phase",
+    "S3",
+    "SCHEDULE",
+    "SEARCH",
+    # Orchestrator
+    "Orchestrator",
+    "get_orchestrator",
+    # Content Registry
+    "ContentItem",
+    "ContentRegistry",
+    "compute_content_hash",
+    # Ingestion
+    "IngestionResult",
+    "TextChunk",
+    "chunk_text",
+    "extract_text",
+    "ingest_from_delivery_agent",
+    "ingest_from_highspot",
+    "ingest_from_wiki",
+    "ingest_manual_upload",
+    # Wiki Scraper
+    "WikiPage",
+    "crawl_wiki_tree",
+    "discover_wikis_for_project",
+    "fetch_wiki_page",
+    "infer_project_from_url",
+    "register_project_wiki",
+    "run_wiki_scrape",
+    # Highspot Scraper
+    "HighspotClient",
+    "HighspotItem",
+    "extract_highspot_metadata",
+    "fetch_highspot_item",
+    "fetch_spot_items_since",
+    "run_highspot_sync",
+    # Search
+    "BM25Index",
+    "SearchEngine",
+    "SearchFilters",
+    "SearchResponse",
+    "SearchResult",
+    "VectorIndex",
+    # Query Handler
+    "Citation",
+    "EnablementAnswer",
+    "QueryHandler",
+    # Freshness
+    "FreshnessMonitor",
+    "FreshnessReport",
+    # MCP
+    "create_mcp_server",
+    "serve",
+    # Vector Store
+    "EmbeddingService",
+    "OpenSearchConfig",
+    "VectorStore",
+    # Permissions Cache
+    "PermissionsCache",
+    "get_user_cache",
+    "get_partner_cache",
+]