Welcome to Part 2 of our two-part blog series!
In this post, I'll elevate the concepts explored in Part 1 to create a scalable, production-ready solution. Using AWS Lambda functions and AWS CDK, you'll transform our notebook-based prototype into a robust, serverless architecture.
Together, we'll develop AWS Lambda functions for embedding generation and retrieval, leverage AWS CDK for infrastructure-as-code deployment, and integrate with Amazon S3 and Amazon Aurora PostgreSQL for efficient data storage and retrieval. By the end of this tutorial, you'll have a fully functional, serverless multimodal search engine capable of understanding and retrieving both textual and visual content.
โ
AWS Level: Advanced - 300
Prerequisites:
- Amazon Titan Embeddings V2
- Anthropic Claude 3 models (Haiku or Sonnet).
๐ฐ Cost to complete:
In the second part, you'll construct a Serverless Embedding App utilizing the AWS Cloud Development Kit (CDK) to create four Lambda Functions.
AWS Lambda Functions for Generating Embeddings for Text and Image Files:
To handle the embedding process, there is a dedicated Lambda Function for each file type:
Event to trigger:
{
"location": "REPLACE-YOU-KEY",
"vectorStoreLocation": "REPALCE-NAME.vdb",
"bucketName": "REPLACE-YOU-BUCKET",
"vectorStoreType": "faiss",
"splitStrategy": "semantic",
"fileType": "application/pdf",
"embeddingModel": "amazon.titan-embed-text-v1"
}
Event to trigger:
{
"location": "REPLACE-YOU-KEY-FOLDER",
"vectorStoreLocation": "REPLACE-NAME.vdb",
"bucketName": "REPLACE-YOU-BUCKET",
"vectorStoreType": "faiss",
"splitStrategy": "semantic",
"embeddingModel": "amazon.titan-embed-image-v1"
}
Event to trigger:
{
"location": "YOU-KEY",
"bucketName": "YOU-BUCKET-NAME",
"fileType": "pdf or image",
"embeddingModel": "amazon.titan-embed-text-v1",
"PGVECTOR_USER":"YOU-RDS-USER",
"PGVECTOR_PASSWORD":"YOU-RDS-PASSWORD",
"PGVECTOR_HOST":"YOU-RDS-ENDPOINT-PROXY",
"PGVECTOR_DATABASE":"YOU-RDS-DATABASE",
"PGVECTOR_PORT":"5432",
"collectioName": "YOU-collectioName",
"bedrock_endpoint": "https://vpce-...-.....bedrock-runtime.YOU-REGION.vpce.amazonaws.com"
}
AWS Lambda Funtions to Query for Text and Image Files in a Vector DB:
To handle the embedding process, there is a dedicated Lambda Function for each file type:
Event to trigger:
{
"vectorStoreLocation": "REPLACE-NAME.vdb",
"bucketName": "REPLACE-YOU-BUCKET",
"vectorStoreType": "faiss",
"query": "YOU-QUERY",
"numDocs": 5,
"embeddingModel": "amazon.titan-embed-text-v1"
}
You can search by text or by image
{
"vectorStoreLocation": "REPLACE-NAME.vdb",
"bucketName": "REPLACE-YOU-BUCKET",
"vectorStoreType": "faiss",
"InputType": "text",
"query":"TEXT-QUERY",
"embeddingModel": "amazon.titan-embed-text-v1"
}
Image event to trigger
{
"vectorStoreLocation": "REPLACE-NAME.vdb",
"bucketName": "REPLACE-YOU-BUCKET",
"vectorStoreType": "faiss",
"InputType": "image",
"query":"IMAGE-BUCKET-LOCATION-QUERY",
"embeddingModel": "amazon.titan-embed-text-v1"
}
๐ก The next step is to take the ย image_pathย value and download the file from Amazon S3 bucket with a ย download_file boto3ย method.
{
"location": "YOU-KEY",
"bucketName": "YOU-BUCKET-NAME",
"fileType": "pdf or image",
"embeddingModel": "amazon.titan-embed-text-v1",
"PGVECTOR_USER":"YOU-RDS-USER",
"PGVECTOR_PASSWORD":"YOU-RDS-PASSWORD",
"PGVECTOR_HOST":"YOU-RDS-ENDPOINT-PROXY",
"PGVECTOR_DATABASE":"YOU-RDS-DATABASE",
"PGVECTOR_PORT":"5432",
"collectioName": "YOU-collectioName",
"bedrock_endpoint": "https://vpce-...-.....bedrock-runtime.YOU-REGION.vpce.amazonaws.com",
"QUERY": "YOU-TEXT-QUESTION"
}
๐ก Use ย locationย and ย bucketNameย to deliver image location to make a query.
๐ Let's build!
The Amazon Lambdas that you build in this deployment are created with a ย container imagesย , you must have ย Docker Desktopย installed and active in your computer. Step 1: APP Set Up
โ
Clone the repo
git clone https://github.com/build-on-aws/langchain-embeddings
โ
Go to:
Step 2: Deploy architecture with CDK.
โ
Create The Virtual Environment: by following the steps in the ย READMEย python3 -m venv .venv
source .venv/bin/activate
for windows:
.venv\Scripts\activate.bat
โ
Install The Requirements:
pip install -r requirements.txt
โ
Synthesize The Cloudformation Template With The Following Command:
โ
๐ The Deployment:
๐งน Clean the house!:
If you finish testing and want to clean the application, you just have to follow these two steps:
- Delete the files from the Amazon S3 bucket created in the deployment.
- Run this command in your terminal:
Conclusion
In this post, you've demonstrated how to transform a notebook-based multimodal search solution into a scalable, serverless architecture using AWS services. You've walked through the process of developing Lambda functions for embedding tasks, utilizing AWS CDK for infrastructure deployment, and integrating with S3 and Aurora PostgreSQL for efficient data management.
By leveraging these serverless technologies, you can now deploy a robust, production-ready multimodal search engine capable of handling both textual and visual content. This approach not only enhances scalability but also reduces operational overhead, allowing you to focus on improving your search capabilities and user experience.
I encourage you to build upon this foundation, experiment with different embedding models, and explore additional AWS services to further enhance your multimodal search engine. Don't hesitate to share your experiences or ask questions in the comments below. Happy building!
ย
Loading article