fix(llm): dependency hierarchy of resources

DataChefHQ · Sep 23, 2024 · 43e2956 · 43e2956
1 parent 170bc8f
commit 43e2956
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/damavand/cloud/aws/resources/vllm_component.py b/src/damavand/cloud/aws/resources/vllm_component.py
@@ -234,6 +234,7 @@ def api(self) -> aws.apigateway.RestApi:
 
         return aws.apigateway.RestApi(
             resource_name=f"{self._name}-api",
+            opts=ResourceOptions(parent=self),
             endpoint_configuration=aws.apigateway.RestApiEndpointConfigurationArgs(
                 types="REGIONAL",
             ),
@@ -244,6 +245,7 @@ def api(self) -> aws.apigateway.RestApi:
     def api_resource(self) -> aws.apigateway.Resource:
         return aws.apigateway.Resource(
             resource_name=f"{self._name}-api-resource",
+            opts=ResourceOptions(parent=self),
             rest_api=self.api.id,
             parent_id=self.api.root_resource_id,
             path_part="completions",
@@ -254,6 +256,7 @@ def api_resource(self) -> aws.apigateway.Resource:
     def api_method(self) -> aws.apigateway.Method:
         return aws.apigateway.Method(
             resource_name=f"{self._name}-api-method",
+            opts=ResourceOptions(parent=self),
             rest_api=self.api.id,
             resource_id=self.api_resource.id,
             http_method="POST",
@@ -275,6 +278,7 @@ def api_access_sagemaker_role(self) -> aws.iam.Role:
 
         return aws.iam.Role(
             resource_name=f"{self._name}-api-sagemaker-access-role",
+            opts=ResourceOptions(parent=self),
             assume_role_policy=json.dumps(
                 self.get_service_assume_policy("apigateway.amazonaws.com")
             ),
@@ -289,6 +293,7 @@ def api_integration(self) -> aws.apigateway.Integration:
 
         return aws.apigateway.Integration(
             resource_name=f"{self._name}-api-integration",
+            opts=ResourceOptions(parent=self),
             rest_api=self.api.id,
             resource_id=self.api_resource.id,
             http_method=self.api_method.http_method,