onetrust-api 0.1.0__py3-none-any.whl

onetrust_api/__init__.py

@@ -0,0 +1,61 @@
+#!/usr/bin/env python
+
+import importlib
+import inspect
+import warnings
+
+# Suppress RequestsDependencyWarning due to chardet 6.x / requests 2.32.x mismatch
+# Centralized here to ensure it runs before any sub-package imports
+warnings.filterwarnings("ignore", message=".*urllib3.*or chardet.*")
+
+__all__: list[str] = []
+
+CORE_MODULES = [
+    "onetrust_api.api_client",
+]
+
+OPTIONAL_MODULES = {
+    "onetrust_api.agent_server": "agent",
+    "onetrust_api.mcp_server": "mcp",
+}
+
+
+def _import_module_safely(module_name: str):
+    """Try to import a module and return it, or None if not available."""
+    try:
+        return importlib.import_module(module_name)
+    except ImportError:
+        return None
+
+
+def _expose_members(module):
+    """Expose public classes and functions from a module into globals and __all__."""
+    for name, obj in inspect.getmembers(module):
+        if (inspect.isclass(obj) or inspect.isfunction(obj)) and not name.startswith(
+            "_"
+        ):
+            globals()[name] = obj
+            __all__.append(name)
+
+
+for module_name in CORE_MODULES:
+    try:
+        module = importlib.import_module(module_name)
+        _expose_members(module)
+    except ImportError:
+        pass
+
+for module_name, extra_name in OPTIONAL_MODULES.items():
+    module = _import_module_safely(module_name)
+    globals()[f"_{extra_name.upper()}_AVAILABLE"] = module is not None
+    if module is not None:
+        _expose_members(module)
+
+__all__.extend(["_MCP_AVAILABLE", "_AGENT_AVAILABLE"])
+
+
+"""
+onetrust-api
+
+Python OneTrust API client + MCP server + A2A agent with 100% API coverage
+"""

onetrust_api/__main__.py

@@ -0,0 +1,5 @@
+#!/usr/bin/python
+from onetrust_api.agent_server import agent_server
+
+if __name__ == "__main__":
+    agent_server()

onetrust_api/agent/AGENTS.md

@@ -0,0 +1,12 @@
+# AGENTS.md - Known A2A Peer Agents
+Last updated: 2026-06-15
+
+This file is the local registry of other A2A agents this agent can discover and call.
+
+## Registered A2A Peers
+
+| Name            | Endpoint URL                    | Description                          | Capabilities                     | Auth      | Notes / Last Connected |
+|-----------------|---------------------------------|--------------------------------------|----------------------------------|-----------|------------------------|
+| SearchMaster    | http://search-agent:9000/a2a    | Advanced web researcher              | web_search, summarize, browse    | none      | 2026-06-15             |
+
+*Add new rows manually or let the agent call `register_a2a_peer(...)`.*

onetrust_api/agent/CRON.md

@@ -0,0 +1,12 @@
+# CRON.md - Persistent Scheduled Tasks
+Last updated: 2026-06-15
+
+## Active Tasks
+
+| ID          | Name              | Interval (min) | Prompt                              | Last run          | Next approx |
+|-------------|-------------------|----------------|-------------------------------------|-------------------|-------------|
+| heartbeat   | Heartbeat         | 30             | @HEARTBEAT.md                       | —                 | —           |
+| log-cleanup | Log Cleanup       | 720            | __internal:cleanup_cron_log         | —                 | —           |
+
+*Edit this table to add/remove tasks. The agent reloads it periodically.*
+*Use `@filename.md` in the Prompt column to load a multi-line prompt from a workspace file.*

onetrust_api/agent/CRON_LOG.md

@@ -0,0 +1,5 @@
+# CRON_LOG.md - Scheduled Task History
+Last updated: 2026-06-15
+
+| Timestamp | Task ID | Status | Message |
+|-----------|---------|--------|---------|

onetrust_api/agent/HEARTBEAT.md

@@ -0,0 +1,28 @@
+# Heartbeat — Periodic Self-Check
+
+You are running a scheduled heartbeat. Perform these checks and report results concisely.
+
+## Checks
+
+1. **Tool Availability** — Call `list_tools` or equivalent to verify your MCP tools are reachable. Report any connection failures.
+2. **Memory Review** — Query the **Knowledge Graph** for any pending follow-up tasks, architectural decisions, or action items.
+3. **Cron Log** — Read `CRON_LOG.md` and check for recent errors (❌). Summarize any failures from the last 24 hours.
+4. **Peer Agents** — Read `AGENTS.md` and note if any registered peers need attention.
+5. **Domain-Specific Checks**:
+   - **Service Health**: Check service health status and scan recent logs for critical errors using available tools.
+6. **Self-Diagnostics** — Report your current model, available tool count, and any anomalies.
+
+## Response Format
+
+If everything is healthy:
+```
+HEARTBEAT_OK — All systems nominal. [tool_count] tools available. No pending actions.
+```
+
+If issues found:
+```
+HEARTBEAT_ALERT — [summary of issues found]
+- Issue 1: ...
+- Issue 2: ...
+- Action needed: ...
+```

onetrust_api/agent/IDENTITY.md

@@ -0,0 +1,15 @@
+# IDENTITY.md - OneTrust Api Agent Identity
+
+## [default]
+ * **Name:** OneTrust Api Agent
+ * **Role:** Python OneTrust API client + MCP server + A2A agent with 100% API coverage
+ * **Emoji:** 🤖
+
+ ### System Prompt
+ You are the OneTrust Api Agent.
+ You must always first run `list_skills` to show all skills.
+ Then, use the `mcp-client` universal skill and check the reference documentation for `onetrust-api.md` to discover the exact tags and tools available for your capabilities.
+
+ ### Capabilities
+ - **MCP Operations**: Leverage the `mcp-client` skill to interact with the target MCP server. Refer to `onetrust-api.md` for specific tool capabilities.
+ - **Custom Agent**: Handle custom tasks or general tasks.

onetrust_api/agent/MCP_AGENTS.md

@@ -0,0 +1,13 @@
+# MCP_AGENTS.md - Dynamic Agent Registry
+
+This file tracks the generated agents from MCP servers. You can manually modify the 'Tools' list to customize agent expertise.
+
+## Agent Mapping Table
+
+| Name | Description | System Prompt | Tools | Tag | Source MCP |
+|------|-------------|---------------|-------|-----|------------|
+
+## Tool Inventory Table
+
+| Tool Name | Description | Tag | Source |
+|-----------|-------------|-----|--------|

onetrust_api/agent/USER.md

@@ -0,0 +1,7 @@
+# USER.md - About the Human
+
+* **Name:** User
+* **Preferred name:** User
+* **Timezone:** America/Chicago
+* **Location:** Chicago, Illinois
+* **Style:** Technical, concise, no fluff

onetrust_api/agent/mcp_config.json

@@ -0,0 +1,11 @@
+{
+  "mcpServers": {
+    "onetrust": {
+      "command": "onetrust-mcp",
+      "env": {
+        "ONETRUST_URL": "${ONETRUST_URL:-http://localhost:8080}",
+        "ONETRUST_TOKEN": "${ONETRUST_TOKEN}"
+      }
+    }
+  }
+}

onetrust_api/agent_server.py

@@ -0,0 +1,84 @@
+#!/usr/bin/python
+import logging
+import os
+import sys
+import warnings
+
+__version__ = "0.1.0"
+
+logging.basicConfig(
+    level=logging.INFO,
+    format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
+    handlers=[logging.StreamHandler()],
+)
+logger = logging.getLogger(__name__)
+
+DEFAULT_AGENT_NAME = None
+DEFAULT_AGENT_DESCRIPTION = None
+DEFAULT_AGENT_SYSTEM_PROMPT = None
+
+
+def agent_server():
+    from agent_utilities import (
+        build_system_prompt_from_workspace,
+        create_agent_parser,
+        create_agent_server,
+        initialize_workspace,
+        load_identity,
+    )
+
+    global DEFAULT_AGENT_NAME, DEFAULT_AGENT_DESCRIPTION, DEFAULT_AGENT_SYSTEM_PROMPT
+    initialize_workspace()
+    meta = load_identity()
+    DEFAULT_AGENT_NAME = os.getenv(
+        "DEFAULT_AGENT_NAME", meta.get("name", "OneTrust Api")
+    )
+    DEFAULT_AGENT_DESCRIPTION = os.getenv(
+        "AGENT_DESCRIPTION",
+        meta.get(
+            "description", "AI agent for OneTrust privacy and compliance management."
+        ),
+    )
+    DEFAULT_AGENT_SYSTEM_PROMPT = os.getenv(
+        "AGENT_SYSTEM_PROMPT",
+        meta.get("content") or build_system_prompt_from_workspace(),
+    )
+
+    warnings.filterwarnings("ignore", message=".*urllib3.*or chardet.*")
+    warnings.filterwarnings("ignore", category=DeprecationWarning, module="fastmcp")
+
+    print(f"{DEFAULT_AGENT_NAME} v{__version__}", file=sys.stderr)
+    parser = create_agent_parser()
+    args = parser.parse_args()
+
+    if args.debug:
+        logging.getLogger().setLevel(logging.DEBUG)
+        logger.debug("Debug mode enabled")
+
+    # Start server using the auto-discovery pattern (from mcp_config.json).
+    # With many tool tags the agent-utilities graph router engages automatically.
+    create_agent_server(
+        mcp_url=args.mcp_url,
+        mcp_config=args.mcp_config or "mcp_config.json",
+        host=args.host,
+        port=args.port,
+        provider=args.provider,
+        model_id=args.model_id,
+        router_model=args.model_id,
+        agent_model=args.model_id,
+        base_url=args.base_url,
+        api_key=args.api_key,
+        custom_skills_directory=args.custom_skills_directory,
+        enable_web_ui=args.web,
+        enable_otel=args.otel,
+        otel_endpoint=args.otel_endpoint,
+        otel_headers=args.otel_headers,
+        otel_public_key=args.otel_public_key,
+        otel_secret_key=args.otel_secret_key,
+        otel_protocol=args.otel_protocol,
+        debug=args.debug,
+    )
+
+
+if __name__ == "__main__":
+    agent_server()

onetrust_api/api/__init__.py

@@ -0,0 +1,7 @@
+"""OneTrust API client layer.
+
+``api_client_base`` holds the hand-authored HTTP/auth/pagination machinery; the
+``api_client_<domain>`` modules are generated from the vendored OpenAPI specs by
+``scripts/generate_from_openapi.py`` and composed into the single ``Api`` class in
+``onetrust_api.api_client``.
+"""