use prefix to deploy multiple stack (#43)

llm-jp · Jan 22, 2024 · 11363d1 · 11363d1
1 parent 9cc2ab7
commit 11363d1
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 4 deletions.
diff --git a/packages/cdk/bin/generative-ai-use-cases.ts b/packages/cdk/bin/generative-ai-use-cases.ts
@@ -5,4 +5,6 @@ import { GenerativeAiUseCasesStack } from '../lib/generative-ai-use-cases-stack'
 
 const app = new cdk.App();
 
-new GenerativeAiUseCasesStack(app, 'GenerativeAiUseCasesStack');
+const stage = app.node.tryGetContext('stage');
+
+new GenerativeAiUseCasesStack(app, `${stage}GenerativeAiUseCasesStack`);
diff --git a/packages/cdk/lib/construct/llm.ts b/packages/cdk/lib/construct/llm.ts
@@ -19,13 +19,18 @@ export class LLM extends Construct {
   public readonly models: Model[] = models;
   public readonly deploy_suffix: string =
     '-' + new Date().toISOString().replace(/[:T-]/g, '').split('.')[0];
-  public readonly endpointConfigName =
-    'llm-jp-endpoint-config' + this.deploy_suffix;
-  public readonly endpointName = 'llm-jp-endpoint';
+  public readonly endpointConfigName;
+  public readonly endpointName;
 
   constructor(scope: Construct, id: string) {
     super(scope, id);
 
+    // Specify Endpoint with stage suffix
+    const stage = this.node.tryGetContext('stage');
+    const prefix = stage ? `${stage}-` : '';
+    this.endpointConfigName = prefix + 'llm-jp-endpoint-config' + this.deploy_suffix;
+    this.endpointName = prefix + 'llm-jp-endpoint';
+
     // Get Container Image
     // https://github.com/aws/deep-learning-containers/blob/master/available_images.md
     const repositoryName = 'djl-inference';