diff --git a/diagrams/aws/ml.py b/diagrams/aws/ml.py index 632ea2dc..eb7cc5fd 100644 --- a/diagrams/aws/ml.py +++ b/diagrams/aws/ml.py @@ -8,6 +8,14 @@ class _ML(_AWS): _icon_dir = "resources/aws/ml" +class Agentcore(_ML): + _icon = "agentcore.png" + + +class AiAgent(_ML): + _icon = "ai-agent.png" + + class AmazonCodewhisperer(_ML): _icon = "amazon-codewhisperer.png" @@ -36,6 +44,14 @@ class Bedrock(_ML): _icon = "bedrock.png" +class BrowserTool(_ML): + _icon = "browser-tool.png" + + +class CodeInterpreter(_ML): + _icon = "code-interpreter.png" + + class Comprehend(_ML): _icon = "comprehend.png" @@ -64,6 +80,10 @@ class ElasticInference(_ML): _icon = "elastic-inference.png" +class Evaluations(_ML): + _icon = "evaluations.png" + + class Forecast(_ML): _icon = "forecast.png" @@ -72,6 +92,14 @@ class FraudDetector(_ML): _icon = "fraud-detector.png" +class Gateway(_ML): + _icon = "gateway.png" + + +class Identity(_ML): + _icon = "identity.png" + + class Kendra(_ML): _icon = "kendra.png" @@ -84,10 +112,22 @@ class MachineLearning(_ML): _icon = "machine-learning.png" +class Memory(_ML): + _icon = "memory.png" + + +class Observability(_ML): + _icon = "observability.png" + + class Personalize(_ML): _icon = "personalize.png" +class PolicyEngineAgenticGuardrails(_ML): + _icon = "policy-engine-agentic-guardrails.png" + + class Polly(_ML): _icon = "polly.png" @@ -108,6 +148,10 @@ class Rekognition(_ML): _icon = "rekognition.png" +class Runtime(_ML): + _icon = "runtime.png" + + class SagemakerGroundTruth(_ML): _icon = "sagemaker-ground-truth.png" diff --git a/resources/aws/ml/agentcore.png b/resources/aws/ml/agentcore.png new file mode 100644 index 00000000..c00b77e8 Binary files /dev/null and b/resources/aws/ml/agentcore.png differ diff --git a/resources/aws/ml/ai-agent.png b/resources/aws/ml/ai-agent.png new file mode 100644 index 00000000..d33c27cd Binary files /dev/null and b/resources/aws/ml/ai-agent.png differ diff --git a/resources/aws/ml/browser-tool.png b/resources/aws/ml/browser-tool.png new file mode 100644 index 00000000..d2578710 Binary files /dev/null and b/resources/aws/ml/browser-tool.png differ diff --git a/resources/aws/ml/code-interpreter.png b/resources/aws/ml/code-interpreter.png new file mode 100644 index 00000000..c1bb9419 Binary files /dev/null and b/resources/aws/ml/code-interpreter.png differ diff --git a/resources/aws/ml/evaluations.png b/resources/aws/ml/evaluations.png new file mode 100644 index 00000000..4d41cf12 Binary files /dev/null and b/resources/aws/ml/evaluations.png differ diff --git a/resources/aws/ml/gateway.png b/resources/aws/ml/gateway.png new file mode 100644 index 00000000..3ca9fead Binary files /dev/null and b/resources/aws/ml/gateway.png differ diff --git a/resources/aws/ml/identity.png b/resources/aws/ml/identity.png new file mode 100644 index 00000000..ae818e7f Binary files /dev/null and b/resources/aws/ml/identity.png differ diff --git a/resources/aws/ml/memory.png b/resources/aws/ml/memory.png new file mode 100644 index 00000000..794b26be Binary files /dev/null and b/resources/aws/ml/memory.png differ diff --git a/resources/aws/ml/observability.png b/resources/aws/ml/observability.png new file mode 100644 index 00000000..1a8a8bbd Binary files /dev/null and b/resources/aws/ml/observability.png differ diff --git a/resources/aws/ml/policy-engine-agentic-guardrails.png b/resources/aws/ml/policy-engine-agentic-guardrails.png new file mode 100644 index 00000000..79d57443 Binary files /dev/null and b/resources/aws/ml/policy-engine-agentic-guardrails.png differ diff --git a/resources/aws/ml/runtime.png b/resources/aws/ml/runtime.png new file mode 100644 index 00000000..af5d0c3b Binary files /dev/null and b/resources/aws/ml/runtime.png differ diff --git a/test_agentcore_icons.py b/test_agentcore_icons.py new file mode 100644 index 00000000..dff6fd8b --- /dev/null +++ b/test_agentcore_icons.py @@ -0,0 +1,58 @@ +#!/usr/bin/env python3 + +""" +Test script for AWS Agentcore Bedrock icons +Validates that all 11 new Agentcore icons can be imported and used correctly. +""" + +from diagrams import Diagram +from diagrams.aws.ml import ( + Agentcore, + AiAgent, + Runtime, + Gateway, + Identity, + CodeInterpreter, + Observability, + BrowserTool, + Memory, + Evaluations, + PolicyEngineAgenticGuardrails +) + +def test_agentcore_icons(): + """Test all Agentcore icons in a sample diagram""" + + with Diagram("AWS Agentcore Bedrock Icons Test", show=False, filename="agentcore_test"): + # Core Agentcore components + agentcore = Agentcore("Agentcore") + ai_agent = AiAgent("AI Agent") + runtime = Runtime("Runtime") + gateway = Gateway("Gateway") + + # Identity and security + identity = Identity("Identity") + policy_engine = PolicyEngineAgenticGuardrails("Policy Engine") + + # Tools and capabilities + code_interpreter = CodeInterpreter("Code Interpreter") + browser_tool = BrowserTool("Browser Tool") + memory = Memory("Memory") + + # Monitoring and evaluation + observability = Observability("Observability") + evaluations = Evaluations("Evaluations") + + # Create a flow showing the relationships + agentcore >> ai_agent >> runtime + ai_agent >> [code_interpreter, browser_tool, memory] + gateway >> identity >> policy_engine + [runtime, code_interpreter, browser_tool] >> observability + observability >> evaluations + + print("✅ Test diagram created successfully!") + print("📁 Generated file: agentcore_test.png") + print("🔍 Please verify the icons render correctly in the diagram") + +if __name__ == "__main__": + test_agentcore_icons()