diff --git a/diagrams/aws/ml.py b/diagrams/aws/ml.py index a08da89c..632ea2dc 100644 --- a/diagrams/aws/ml.py +++ b/diagrams/aws/ml.py @@ -8,6 +8,14 @@ class _ML(_AWS): _icon_dir = "resources/aws/ml" +class AmazonCodewhisperer(_ML): + _icon = "amazon-codewhisperer.png" + + +class AmazonNova(_ML): + _icon = "amazon-nova.png" + + class ApacheMxnetOnAWS(_ML): _icon = "apache-mxnet-on-aws.png" @@ -16,6 +24,14 @@ class AugmentedAi(_ML): _icon = "augmented-ai.png" +class AWSAppStudio(_ML): + _icon = "aws-app-studio.png" + + +class AWSNeuron(_ML): + _icon = "aws-neuron.png" + + class Bedrock(_ML): _icon = "bedrock.png" diff --git a/resources/aws/ml/amazon-codewhisperer.png b/resources/aws/ml/amazon-codewhisperer.png new file mode 100644 index 00000000..31ba06a9 Binary files /dev/null and b/resources/aws/ml/amazon-codewhisperer.png differ diff --git a/resources/aws/ml/amazon-nova.png b/resources/aws/ml/amazon-nova.png new file mode 100644 index 00000000..ab69fd59 Binary files /dev/null and b/resources/aws/ml/amazon-nova.png differ diff --git a/resources/aws/ml/aws-app-studio.png b/resources/aws/ml/aws-app-studio.png new file mode 100644 index 00000000..c8ffedd8 Binary files /dev/null and b/resources/aws/ml/aws-app-studio.png differ diff --git a/resources/aws/ml/aws-neuron.png b/resources/aws/ml/aws-neuron.png new file mode 100644 index 00000000..4984baf4 Binary files /dev/null and b/resources/aws/ml/aws-neuron.png differ diff --git a/test_nova_icons.py b/test_nova_icons.py new file mode 100644 index 00000000..0c89b0c5 --- /dev/null +++ b/test_nova_icons.py @@ -0,0 +1,29 @@ +#!/usr/bin/env python3 + +""" +Test script for new AWS AI/ML service icons +Tests Amazon Nova, AWS App Studio, Amazon CodeWhisperer, and AWS Neuron +""" + +from diagrams import Diagram +from diagrams.aws.ml import AmazonNova, AWSAppStudio, AmazonCodewhisperer, AWSNeuron + +def test_nova_icons(): + """Test the new Nova and AI service icons""" + + with Diagram("New AWS AI/ML Services", show=False, filename="nova_test"): + # Create instances of the new services + nova = AmazonNova("Amazon Nova") + app_studio = AWSAppStudio("AWS App Studio") + codewhisperer = AmazonCodewhisperer("CodeWhisperer") + neuron = AWSNeuron("AWS Neuron") + + # Create a simple flow + nova >> app_studio >> codewhisperer >> neuron + + print("✅ Test diagram created successfully!") + print("📁 Generated file: nova_test.png") + print("🔍 Please verify the icons render correctly in the diagram") + +if __name__ == "__main__": + test_nova_icons()