feat(construct): fixed defects abd added schema schanges warning

Author: Dinesh Sajwan

lambda/aws-qa-appsync-opensearch/question_answering/src/qa_agent/chain.py

@@ -48,7 +48,7 @@ def run_question_answering(arguments):
     image_url = arguments['presignedurl']
 
     #set deafult modality to text
-    qa_model= arguments['qa_model']['modality']
+    qa_model= arguments['qa_model']
     modality=qa_model.get('modality','Text')
 
     # Visual QA

lambda/aws-qa-appsync-opensearch/question_answering/src/qa_agent/image_qa.py

@@ -29,6 +29,7 @@
 logger = Logger(service="QUESTION_ANSWERING")
 tracer = Tracer(service="QUESTION_ANSWERING")
 metrics = Metrics(namespace="question_answering", service="QUESTION_ANSWERING")
+
 bucket_name = os.environ['INPUT_BUCKET']
 
 def run_qa_agent_on_image_no_memory(input_params):

lambda/aws-rag-appsync-stepfn-opensearch/s3_file_transformer/src/requirements.txt

@@ -5,6 +5,6 @@ typing-extensions
 boto3>=1.34.29
 requests
 langchain==0.1.4
-pypdf2==4.0.2
+pypdf2==3.0.1
 Pillow==10.2.0
 langchain-community==0.0.16

src/patterns/gen-ai/aws-qa-appsync-opensearch/index.ts

@@ -11,7 +11,7 @@
  *  and limitations under the License.
  */
 import * as path from 'path';
-import { Duration, Aws } from 'aws-cdk-lib';
+import { Duration, Aws, Annotations } from 'aws-cdk-lib';
 import * as appsync from 'aws-cdk-lib/aws-appsync';
 import * as cognito from 'aws-cdk-lib/aws-cognito';
 import * as ec2 from 'aws-cdk-lib/aws-ec2';
@@ -158,6 +158,21 @@ export interface QaAppsyncOpensearchProps {
  * @summary The QaAppsyncOpensearch class.
  */
 export class QaAppsyncOpensearch extends Construct {
+  /**
+   * Construct warning
+   */
+  public static readonly CONSTRUCT_SCHEMA_UPDATE_WARNING=`
+  Attention QaAppsyncOpensearch users, an update has been made to 
+  the GraphQL schema.To ensure continued functionality, please review 
+  and update your GraphQL mutations and subscriptions to align with 
+  the new schema.This schema update enables enhanced capabilities 
+  and optimizations,so adopting the changes is recommended. 
+  Please refer to the construct documentation for details 
+  on the schema changes and examples of updated GraphQL statements.
+  Reach out to the support team if you need assistance 
+  updating your integration codebase.  
+  `;
+
   /**
    * Returns the instance of ec2.IVpc used by the construct
    */
@@ -189,6 +204,7 @@ export class QaAppsyncOpensearch extends Construct {
    */
   public readonly qaLambdaFunction: lambda.DockerImageFunction;
 
+
   /**
    * @summary Constructs a new instance of the RagAppsyncStepfnOpensearch class.
    * @param {cdk.App} scope - represents the scope for all the resources.
@@ -200,6 +216,7 @@ export class QaAppsyncOpensearch extends Construct {
   constructor(scope: Construct, id: string, props: QaAppsyncOpensearchProps) {
     super(scope, id);
 
+    Annotations.of(scope).addWarning(QaAppsyncOpensearch.CONSTRUCT_SCHEMA_UPDATE_WARNING);
     // stage
     let stage = '-dev';
     if (props?.stage) {
@@ -425,7 +442,7 @@ export class QaAppsyncOpensearch extends Construct {
       },
     );
 
-    // Minimum permissions for a Lambda function to execute while accessing a resource within a VPC
+    // Minimum permissions for a Lambda functienvon to execute while accessing a resource within a VPC
     question_answering_function_role.addToPolicy(
       new iam.PolicyStatement({
         effect: iam.Effect.ALLOW,
@@ -449,6 +466,13 @@ export class QaAppsyncOpensearch extends Construct {
         resources: ['*'],
       }),
     );
+    question_answering_function_role.addToPolicy(
+      new iam.PolicyStatement({
+        effect: iam.Effect.ALLOW,
+        actions: ['sagemaker:InvokeEndpoint'],
+        resources: ['*'],
+      }),
+    );
 
     // The lambda will access the opensearch credentials
     if (props.openSearchSecret) {

src/patterns/gen-ai/aws-rag-appsync-stepfn-opensearch/index.ts

@@ -11,7 +11,7 @@
  *  and limitations under the License.
  */
 import * as path from 'path';
-import { Duration, Aws, Stack } from 'aws-cdk-lib';
+import { Duration, Aws, Stack, Annotations } from 'aws-cdk-lib';
 import * as appsync from 'aws-cdk-lib/aws-appsync';
 import * as cognito from 'aws-cdk-lib/aws-cognito';
 import * as ec2 from 'aws-cdk-lib/aws-ec2';
@@ -177,6 +177,22 @@ export interface RagAppsyncStepfnOpensearchProps {
    * @summary The RagAppsyncStepfnOpensearch class.
    */
 export class RagAppsyncStepfnOpensearch extends Construct {
+
+  /**
+   * Construct warning
+   */
+  public static readonly CONSTRUCT_SCHEMA_UPDATE_WARNING=`
+  Attention RagAppsyncStepfnOpensearch users, an update has been made to 
+  the GraphQL schema.To ensure continued functionality, please review 
+  and update your GraphQL mutations and subscriptions to align with 
+  the new schema.This schema update enables enhanced capabilities 
+  and optimizations,so adopting the changes is recommended. 
+  Please refer to the construct documentation for details 
+  on the schema changes and examples of updated GraphQL statements.
+  Reach out to the support team if you need assistance 
+  updating your integration codebase.  
+  `;
+
   /**
    * Returns the instance of ec2.IVpc used by the construct
    */
@@ -242,6 +258,7 @@ export class RagAppsyncStepfnOpensearch extends Construct {
   constructor(scope: Construct, id: string, props: RagAppsyncStepfnOpensearchProps) {
     super(scope, id);
 
+    Annotations.of(scope).addWarning(RagAppsyncStepfnOpensearch.CONSTRUCT_SCHEMA_UPDATE_WARNING);
     // stage
     let stage = '-dev';
     if (props?.stage) {