@macaw-ai/client 0.0.0-stage → 0.9.11

package/payload/secureAI/examples/anthropic/anthropic_1b_multiuser_bind.py

@@ -0,0 +1,190 @@
+#!/usr/bin/env python3
+"""
+anthropic_1b_multiuser_bind.py - Multi-user Anthropic with bind_to_user
+
+Each user's JWT identity flows through for policy evaluation.
+Different users get different permissions based on their policies.
+
+Prerequisites:
+    - MACAW SDK installed (pip install macaw-client macaw-adapters)
+    - ANTHROPIC_API_KEY environment variable
+    - Identity Provider configured (Console -> Settings -> Identity Bridge)
+    - Test users: alice/Alice123!, bob/Bob@123!
+
+Run:
+    export ANTHROPIC_API_KEY=sk-ant-...
+    python anthropic_1b_multiuser_bind.py
+
+No IdP configured? Run simpler example first:
+    python anthropic_1a_dropin_simple.py
+"""
+
+import os
+import sys
+
+from macaw_adapters.anthropic import SecureAnthropic
+from macaw_client import MACAWClient, RemoteIdentityProvider
+
+
+# Test configurations based on user policies
+# Alice: Haiku only, max 500 tokens
+# Bob: Haiku/Sonnet/Opus, max 2000 tokens
+USER_TESTS = {
+    "alice": {
+        "password": "Alice123!",
+        "policy_desc": "Claude Haiku only, max 500 tokens",
+        "tests": [
+            # (model, max_tokens, should_succeed)
+            ("claude-3-haiku-20240307", 400, True),   # ALLOWED - correct model, within limit
+            ("claude-opus-4-5-20251101", 400, False),   # BLOCKED - alice can't use Opus
+            ("claude-3-haiku-20240307", 600, False),  # BLOCKED - exceeds max_tokens
+        ]
+    },
+    "bob": {
+        "password": "Bob@123!",
+        "policy_desc": "Claude Haiku/Sonnet/Opus, max 2000 tokens",
+        "tests": [
+            ("claude-3-haiku-20240307", 400, True),   # ALLOWED
+            ("claude-opus-4-5-20251101", 400, True),    # ALLOWED - bob CAN use Opus
+            ("claude-3-haiku-20240307", 600, True),   # ALLOWED - within bob's 2000 limit
+            ("claude-opus-4-5-20251101", 2500, False),  # BLOCKED - exceeds max_tokens
+        ]
+    }
+}
+
+
+def create_user_client(username: str, password: str) -> MACAWClient:
+    """Create authenticated user client."""
+    print(f"  Authenticating {username}...")
+    jwt_token, _ = RemoteIdentityProvider().login(username, password)
+
+    user = MACAWClient(
+        user_name=username,
+        iam_token=jwt_token,
+        agent_type="user",
+        app_name="financial-app"
+    )
+
+    if not user.register():
+        raise RuntimeError(f"Failed to register user {username}")
+
+    print(f"  User agent: {user.agent_id}")
+    return user
+
+
+def test_user(username: str, anthropic_service: SecureAnthropic):
+    """Test a user with bind_to_user pattern."""
+    config = USER_TESTS[username]
+
+    print(f"\n{'=' * 60}")
+    print(f"Testing {username.upper()} via bind_to_user")
+    print(f"Policy: {config['policy_desc']}")
+    print("=" * 60)
+
+    # 1. Create user client with JWT
+    user_client = create_user_client(username, config["password"])
+
+    # 2. Bind user to service
+    print(f"\n  Binding to service: {anthropic_service.server_id}")
+    user_anthropic = anthropic_service.bind_to_user(user_client)
+
+    # 3. Test with different parameters
+    print(f"\n  Running tests:")
+    for model, max_tokens, should_succeed in config["tests"]:
+        expected = "SHOULD SUCCEED" if should_succeed else "SHOULD BE BLOCKED"
+        print(f"\n  -> {model}, {max_tokens} tokens ({expected})")
+
+        try:
+            response = user_anthropic.messages.create(
+                model=model,
+                max_tokens=max_tokens,
+                messages=[
+                    {"role": "user", "content": "What is revenue growth? Brief answer."}
+                ]
+            )
+
+            if should_succeed:
+                print(f"     SUCCESS (as expected)")
+                print(f"     Model: {response.model}")
+                print(f"     Response: {response.content[0].text[:60]}...")
+            else:
+                print(f"     UNEXPECTED SUCCESS - should have been blocked!")
+
+        except Exception as e:
+            error = str(e)
+            is_policy_block = any(word in error.lower() for word in
+                                  ["not in allowed", "not permitted", "policy",
+                                   "max_tokens", "model", "blocked", "denied"])
+            if is_policy_block:
+                if should_succeed:
+                    print(f"     UNEXPECTED BLOCK: {error[:60]}...")
+                else:
+                    print(f"     CORRECTLY BLOCKED by policy")
+            else:
+                print(f"     ERROR: {error[:60]}...")
+
+    # 4. Cleanup
+    print(f"\n  Unbinding user...")
+    user_anthropic.unbind()
+    print(f"  is_bound: {user_anthropic.is_bound}")
+
+
+def main():
+    # Check for API key
+    if not os.environ.get("ANTHROPIC_API_KEY"):
+        print("Set ANTHROPIC_API_KEY environment variable")
+        print("  export ANTHROPIC_API_KEY=sk-ant-...")
+        return
+
+    print("=" * 60)
+    print("Example 1b: Multi-user with bind_to_user (Anthropic)")
+    print("=" * 60)
+    print("\nPath: bind_to_user (per-user identity)")
+    print("Use when: SaaS app, different users need different permissions")
+    print("\nArchitecture:")
+    print("  - Single SecureAnthropic service (shared)")
+    print("  - Per-user MACAWClient with JWT identity")
+    print("  - bind_to_user() connects user to service")
+    print("  - User's identity flows through for policy evaluation")
+
+    # Create SINGLE service (shared across all users)
+    print("\n--- Creating SecureAnthropic service ---")
+    anthropic_service = SecureAnthropic(app_name="anthropic-service")
+    print(f"Service registered: {anthropic_service.server_id}")
+    print(f"Mode: {anthropic_service._mode}")
+
+    # Test alice (restricted)
+    try:
+        test_user("alice", anthropic_service)
+    except Exception as e:
+        print(f"\nFailed to test alice: {e}")
+
+    # Test bob (enhanced)
+    try:
+        test_user("bob", anthropic_service)
+    except Exception as e:
+        print(f"\nFailed to test bob: {e}")
+
+    print("\n" + "=" * 60)
+    print("Example complete!")
+    print("=" * 60)
+
+
+if __name__ == "__main__":
+    try:
+        main()
+    except Exception as e:
+        err = str(e)
+        print("\n" + "=" * 60)
+        if "ANTHROPIC_API_KEY" in err or "api_key" in err.lower():
+            print("ERROR: Anthropic API key not configured")
+            print("Fix: export ANTHROPIC_API_KEY=sk-ant-...")
+        elif "Local provider does not support" in err:
+            print("ERROR: Identity Provider not configured")
+            print("Fix: Console -> Settings -> Identity Bridge")
+            print("\nOr run simpler example first:")
+            print("  python anthropic_1a_dropin_simple.py")
+        else:
+            print(f"ERROR: {e}")
+        print("=" * 60)
+        sys.exit(1)

package/payload/secureAI/examples/anthropic/anthropic_1b_multiuser_bind_streaming.py

@@ -0,0 +1,190 @@
+#!/usr/bin/env python3
+"""
+anthropic_1b_multiuser_bind_streaming.py - Multi-user Anthropic with streaming
+
+Same as 1b but demonstrates streaming responses with bind_to_user pattern.
+Claude uses messages.stream() context manager pattern.
+
+Prerequisites:
+    - MACAW SDK installed (pip install macaw-client macaw-adapters)
+    - ANTHROPIC_API_KEY environment variable
+    - Identity Provider configured (Console -> Settings -> Identity Bridge)
+    - Test users: alice/Alice123!, bob/Bob@123!
+
+Run:
+    export ANTHROPIC_API_KEY=sk-ant-...
+    python anthropic_1b_multiuser_bind_streaming.py
+
+No IdP configured? Run simpler example first:
+    python anthropic_1a_dropin_simple.py
+"""
+
+import os
+import sys
+
+from macaw_adapters.anthropic import SecureAnthropic
+from macaw_client import MACAWClient, RemoteIdentityProvider
+
+
+# Test configurations based on user policies
+# Alice: Haiku only, max 500 tokens
+# Bob: Haiku/Sonnet/Opus, max 2000 tokens
+USER_TESTS = {
+    "alice": {
+        "password": "Alice123!",
+        "policy_desc": "Claude Haiku only, max 500 tokens",
+        "tests": [
+            # (model, max_tokens, should_succeed)
+            ("claude-3-haiku-20240307", 400, True),   # ALLOWED
+            ("claude-opus-4-5-20251101", 400, False),  # BLOCKED - alice can't use Opus
+        ]
+    },
+    "bob": {
+        "password": "Bob@123!",
+        "policy_desc": "Claude Haiku/Sonnet/Opus, max 2000 tokens",
+        "tests": [
+            ("claude-3-haiku-20240307", 400, True),   # ALLOWED
+            ("claude-opus-4-5-20251101", 400, True),   # ALLOWED - bob CAN use Opus
+        ]
+    }
+}
+
+
+def create_user_client(username: str, password: str) -> MACAWClient:
+    """Create authenticated user client."""
+    print(f"  Authenticating {username}...")
+    jwt_token, _ = RemoteIdentityProvider().login(username, password)
+
+    user = MACAWClient(
+        user_name=username,
+        iam_token=jwt_token,
+        agent_type="user",
+        app_name="financial-app"
+    )
+
+    if not user.register():
+        raise RuntimeError(f"Failed to register user {username}")
+
+    print(f"  User agent: {user.agent_id}")
+    return user
+
+
+def test_user(username: str, anthropic_service: SecureAnthropic):
+    """Test a user with streaming bind_to_user pattern."""
+    config = USER_TESTS[username]
+
+    print(f"\n{'=' * 60}")
+    print(f"Testing {username.upper()} via bind_to_user (STREAMING)")
+    print(f"Policy: {config['policy_desc']}")
+    print("=" * 60)
+
+    # 1. Create user client with JWT
+    user_client = create_user_client(username, config["password"])
+
+    # 2. Bind user to service
+    print(f"\n  Binding to service: {anthropic_service.server_id}")
+    user_anthropic = anthropic_service.bind_to_user(user_client)
+
+    # 3. Test with streaming
+    print(f"\n  Running streaming tests:")
+    for model, max_tokens, should_succeed in config["tests"]:
+        expected = "SHOULD SUCCEED" if should_succeed else "SHOULD BE BLOCKED"
+        print(f"\n  -> {model}, {max_tokens} tokens ({expected})")
+
+        try:
+            # Claude streaming uses context manager pattern
+            with user_anthropic.messages.stream(
+                model=model,
+                max_tokens=max_tokens,
+                messages=[
+                    {"role": "user", "content": "What is revenue growth? Brief answer."}
+                ]
+            ) as stream:
+                if should_succeed:
+                    print(f"     SUCCESS (as expected)")
+                    print(f"     Streaming: ", end="")
+                    collected = []
+                    for text in stream.text_stream:
+                        collected.append(text)
+                        print(text, end="", flush=True)
+                    print()  # newline
+                    print(f"     Total chars: {len(''.join(collected))}")
+                else:
+                    # If we got here, policy didn't block - unexpected
+                    for text in stream.text_stream:
+                        pass  # consume stream
+                    print(f"     UNEXPECTED SUCCESS - should have been blocked!")
+
+        except Exception as e:
+            error = str(e)
+            is_policy_block = any(word in error.lower() for word in
+                                  ["not in allowed", "not permitted", "policy",
+                                   "max_tokens", "model", "blocked", "denied"])
+            if is_policy_block:
+                if should_succeed:
+                    print(f"     UNEXPECTED BLOCK: {error[:60]}...")
+                else:
+                    print(f"     CORRECTLY BLOCKED by policy")
+            else:
+                print(f"     ERROR: {error[:60]}...")
+
+    # 4. Cleanup
+    print(f"\n  Unbinding user...")
+    user_anthropic.unbind()
+    print(f"  is_bound: {user_anthropic.is_bound}")
+
+
+def main():
+    # Check for API key
+    if not os.environ.get("ANTHROPIC_API_KEY"):
+        print("Set ANTHROPIC_API_KEY environment variable")
+        print("  export ANTHROPIC_API_KEY=sk-ant-...")
+        return
+
+    print("=" * 60)
+    print("Example 1b-stream: Multi-user with Streaming (Anthropic)")
+    print("=" * 60)
+    print("\nStreaming Pattern: messages.stream() context manager")
+    print("Policy enforcement: BEFORE first chunk is returned")
+
+    # Create SINGLE service (shared across all users)
+    print("\n--- Creating SecureAnthropic service ---")
+    anthropic_service = SecureAnthropic(app_name="anthropic-service")
+    print(f"Service registered: {anthropic_service.server_id}")
+    print(f"Mode: {anthropic_service._mode}")
+
+    # Test alice (restricted)
+    try:
+        test_user("alice", anthropic_service)
+    except Exception as e:
+        print(f"\nFailed to test alice: {e}")
+
+    # Test bob (enhanced)
+    try:
+        test_user("bob", anthropic_service)
+    except Exception as e:
+        print(f"\nFailed to test bob: {e}")
+
+    print("\n" + "=" * 60)
+    print("Example complete!")
+    print("=" * 60)
+
+
+if __name__ == "__main__":
+    try:
+        main()
+    except Exception as e:
+        err = str(e)
+        print("\n" + "=" * 60)
+        if "ANTHROPIC_API_KEY" in err or "api_key" in err.lower():
+            print("ERROR: Anthropic API key not configured")
+            print("Fix: export ANTHROPIC_API_KEY=sk-ant-...")
+        elif "Local provider does not support" in err:
+            print("ERROR: Identity Provider not configured")
+            print("Fix: Console -> Settings -> Identity Bridge")
+            print("\nOr run simpler example first:")
+            print("  python anthropic_1a_dropin_simple.py")
+        else:
+            print(f"ERROR: {e}")
+        print("=" * 60)
+        sys.exit(1)

package/payload/secureAI/examples/anthropic/anthropic_1c_a2a_invoke.py

@@ -0,0 +1,156 @@
+#!/usr/bin/env python3
+"""
+anthropic_1c_a2a_invoke.py - Agent-to-Agent invocation with invoke_tool
+
+Demonstrates explicit tool invocation for agent systems and cross-service calls.
+Use when you need fine-grained control over routing and tool invocation.
+
+Prerequisites:
+    - MACAW SDK installed (pip install macaw-client macaw-adapters)
+    - ANTHROPIC_API_KEY environment variable
+    - Identity Provider configured (Console -> Settings -> Identity Bridge)
+    - Test users: alice/Alice123!
+
+Run:
+    export ANTHROPIC_API_KEY=sk-ant-...
+    python anthropic_1c_a2a_invoke.py
+
+No IdP configured? Run simpler example first:
+    python anthropic_1a_dropin_simple.py
+"""
+
+import os
+import sys
+
+from macaw_adapters.anthropic import SecureAnthropic
+from macaw_client import MACAWClient, RemoteIdentityProvider
+
+
+def main():
+    # Check for API key
+    if not os.environ.get("ANTHROPIC_API_KEY"):
+        print("Set ANTHROPIC_API_KEY environment variable")
+        print("  export ANTHROPIC_API_KEY=sk-ant-...")
+        return
+
+    print("=" * 60)
+    print("Example 1c: A2A with invoke_tool (Anthropic)")
+    print("=" * 60)
+    print("\nPath: invoke_tool (explicit A2A control)")
+    print("Use when: Agent systems, explicit control, cross-service calls")
+    print("\nArchitecture:")
+    print("  - SecureAnthropic service registers tools")
+    print("  - User agent calls invoke_tool() directly")
+    print("  - Must know MAPL tool name: tool:<app>/generate")
+    print("  - Returns raw dict (not SDK types)")
+
+    # 1. Create Anthropic service
+    print("\n--- Creating SecureAnthropic service ---")
+    anthropic_service = SecureAnthropic(app_name="anthropic-service")
+    print(f"Service registered: {anthropic_service.server_id}")
+    print(f"Tools registered:")
+    for tool_name in anthropic_service.tools.keys():
+        print(f"  - {tool_name}")
+
+    # 2. Create user agent with JWT
+    print("\n--- Creating user agent ---")
+    print("  Authenticating alice...")
+    jwt_token, _ = RemoteIdentityProvider().login("alice", "Alice123!")
+
+    user = MACAWClient(
+        user_name="alice",
+        iam_token=jwt_token,
+        agent_type="user",
+        app_name="my-agent"
+    )
+
+    if not user.register():
+        print("  Failed to register user agent")
+        return
+
+    print(f"  User agent: {user.agent_id}")
+
+    # 3. Explicit invoke_tool
+    print("\n--- Test 1: Direct invoke_tool ---")
+    print(f"  Tool name: tool:{anthropic_service.app_name}/generate")
+    print(f"  Target: {anthropic_service.server_id}")
+
+    result = user.invoke_tool(
+        tool_name=f"tool:{anthropic_service.app_name}/generate",
+        parameters={
+            "model": "claude-3-haiku-20240307",
+            "max_tokens": 100,
+            "messages": [
+                {"role": "user", "content": "What is compound interest? Brief answer."}
+            ]
+        },
+        target_agent=anthropic_service.server_id
+    )
+
+    # Result is raw dict - you handle parsing
+    if isinstance(result, dict):
+        if "error" in result:
+            print(f"  Error: {result['error']}")
+        elif "content" in result:
+            # Claude response format
+            content = result["content"][0]["text"]
+            print(f"  Success!")
+            print(f"  Model: {result.get('model')}")
+            print(f"  Response: {content[:80]}...")
+        else:
+            print(f"  Unexpected result: {result}")
+
+    # 4. Test policy enforcement
+    print("\n--- Test 2: Policy enforcement (Opus blocked for alice) ---")
+    try:
+        result = user.invoke_tool(
+            tool_name=f"tool:{anthropic_service.app_name}/generate",
+            parameters={
+                "model": "claude-opus-4-5-20251101",  # Alice can't use Opus!
+                "max_tokens": 100,
+                "messages": [{"role": "user", "content": "Hello"}]
+            },
+            target_agent=anthropic_service.server_id
+        )
+
+        if isinstance(result, dict) and "error" in result:
+            print(f"  Correctly blocked: {result['error'][:60]}...")
+        else:
+            print(f"  Unexpected - should have been blocked!")
+    except Exception as e:
+        print(f"  Correctly blocked: {str(e)[:60]}...")
+
+    # 5. Service discovery (optional advanced feature)
+    print("\n--- Service Discovery ---")
+    print("  (Agents can discover services dynamically via registry)")
+    print(f"  Known service: {anthropic_service.server_id}")
+    print(f"  Tool pattern: tool:<app_name>/generate")
+
+    print("\n" + "=" * 60)
+    print("Example complete!")
+    print("=" * 60)
+    print("\nKey differences from bind_to_user:")
+    print("  - Must know MAPL tool names (tool:xxx/generate)")
+    print("  - Must know target agent server_id")
+    print("  - Returns raw dict (not Message)")
+    print("  - More control, but more responsibility")
+
+
+if __name__ == "__main__":
+    try:
+        main()
+    except Exception as e:
+        err = str(e)
+        print("\n" + "=" * 60)
+        if "ANTHROPIC_API_KEY" in err or "api_key" in err.lower():
+            print("ERROR: Anthropic API key not configured")
+            print("Fix: export ANTHROPIC_API_KEY=sk-ant-...")
+        elif "Local provider does not support" in err:
+            print("ERROR: Identity Provider not configured")
+            print("Fix: Console -> Settings -> Identity Bridge")
+            print("\nOr run simpler example first:")
+            print("  python anthropic_1a_dropin_simple.py")
+        else:
+            print(f"ERROR: {e}")
+        print("=" * 60)
+        sys.exit(1)