armoriq-sdk-dev 0.3.1__py3-none-any.whl

armoriq_sdk/__init__.py

@@ -0,0 +1,129 @@
+"""
+ArmorIQ SDK - Build Secure AI Agents
+
+A Python SDK for building AI agents with cryptographic intent verification.
+Provides simple APIs for plan capture, token management, and secure MCP
+tool invocation with built-in security.
+
+Author: ArmorIQ Team <license@armoriq.io>
+"""
+
+from .client import ArmorIQClient
+from .session import (
+    ArmorIQSession,
+    EnforceResult,
+    ReportOptions,
+    SessionMode,
+    SessionOptions,
+)
+from .plan_builder import (
+    ToolNameParser,
+    build_plan_from_tool_calls,
+    default_tool_name_parser,
+    hash_tool_calls,
+)
+from .exceptions import (
+    ArmorIQException,
+    ConfigurationException,
+    DelegationException,
+    IntentMismatchException,
+    InvalidTokenException,
+    MCPInvocationException,
+    PolicyBlockedException,
+    PolicyHoldException,
+    TokenExpiredException,
+)
+from .models import (
+    ApprovedDelegation,
+    DelegationRequest,
+    DelegationRequestParams,
+    DelegationRequestResult,
+    DelegationResult,
+    HoldInfo,
+    IntentToken,
+    InvokeOptions,
+    MCPInvocation,
+    MCPInvocationResult,
+    MCPSemanticMetadata,
+    McpCredential,
+    McpCredentialMap,
+    PlanCapture,
+    PolicyContext,
+    SDKConfig,
+    ToolCall,
+    ToolSemanticEntry,
+)
+
+__version__ = "0.2.12"
+VERSION = __version__
+__author__ = "ArmorIQ Team"
+AUTHOR = __author__
+__email__ = "license@armoriq.io"
+EMAIL = __email__
+
+__all__ = [
+    # Core
+    "ArmorIQClient",
+    "ArmorIQSession",
+    "SessionOptions",
+    "SessionMode",
+    "EnforceResult",
+    "ReportOptions",
+    # Plan helpers
+    "build_plan_from_tool_calls",
+    "default_tool_name_parser",
+    "hash_tool_calls",
+    "ToolNameParser",
+    # Exceptions
+    "ArmorIQException",
+    "InvalidTokenException",
+    "IntentMismatchException",
+    "MCPInvocationException",
+    "TokenExpiredException",
+    "DelegationException",
+    "ConfigurationException",
+    "PolicyBlockedException",
+    "PolicyHoldException",
+    # Models
+    "IntentToken",
+    "PlanCapture",
+    "MCPInvocation",
+    "MCPInvocationResult",
+    "DelegationRequest",
+    "DelegationResult",
+    "SDKConfig",
+    "MCPSemanticMetadata",
+    "ToolSemanticEntry",
+    "PolicyContext",
+    "InvokeOptions",
+    "HoldInfo",
+    "DelegationRequestParams",
+    "DelegationRequestResult",
+    "ApprovedDelegation",
+    "ToolCall",
+    "McpCredential",
+    "McpCredentialMap",
+    # Metadata
+    "VERSION",
+    "AUTHOR",
+    "EMAIL",
+]
+
+# Optional framework integrations (each requires its own extra to be installed)
+try:
+    from .integrations import (
+        ArmorIQAnthropic,
+        ArmorIQCrew,
+        ArmorIQGoogleADK,
+        ArmorIQLangChain,
+        ArmorIQOpenAI,
+    )
+    __all__ += [
+        "ArmorIQCrew",
+        "ArmorIQLangChain",
+        "ArmorIQGoogleADK",
+        "ArmorIQOpenAI",
+        "ArmorIQAnthropic",
+    ]
+except Exception:
+    pass

armoriq_sdk/_build_env.py

@@ -0,0 +1,45 @@
+"""
+Build-time environment marker.
+
+This file is the ONLY difference between the `dev` and `main` branches —
+merging dev → main conflicts on this single constant, which is intentional
+so the release owner consciously flips the default before publishing prod.
+
+  main branch  →  ARMORIQ_ENV = "production"  (prod URLs; published as stable)
+  dev  branch  →  ARMORIQ_ENV = "staging"     (staging URLs; published as -dev)
+
+The baked constant is the ONLY source of truth for which environment's
+URLs to use — no runtime env-var override. To point the SDK at staging,
+install the dev build; to override a specific endpoint for testing,
+pass `backend_endpoint=...` to the ArmorIQClient constructor or set
+BACKEND_ENDPOINT / IAP_ENDPOINT / PROXY_ENDPOINT env vars.
+"""
+
+ARMORIQ_ENV: str = "staging"
+
+# Endpoint table — keep in sync with GCP Cloud Run domain mappings
+# (verified via `gcloud beta run domain-mappings describe --domain=...`).
+#   prod:
+#     api.armoriq.ai        → conmap-auto            (us-central1)
+#     iap.armoriq.ai        → csrg-execution-service (us-central1)
+#     proxy.armoriq.ai      → armoriq-proxy-server   (us-central1)
+#   staging:
+#     staging-api.armoriq.ai          → conmap-auto-staging            (us-central1)
+#     iap-staging.armoriq.ai          → csrg-execution-service-staging (us-central1)
+#     cloud-run-proxy.armoriq.io      → armoriq-proxy-dev              (europe-west1)
+ENDPOINTS = {
+    "production": {
+        "backend": "https://api.armoriq.ai",
+        "proxy": "https://proxy.armoriq.ai",
+        "iap": "https://iap.armoriq.ai",
+    },
+    "staging": {
+        "backend": "https://staging-api.armoriq.ai",
+        "proxy": "https://cloud-run-proxy.armoriq.io",
+        "iap": "https://iap-staging.armoriq.ai",
+    },
+}
+
+
+def resolve(kind: str) -> str:
+    return ENDPOINTS[ARMORIQ_ENV][kind]