Skip to content

Llama

Unstable API

0.7.0

@project-lakechain/bedrock-text-processors

TypeScript Icon

Bedrock text processors expose different constructs that leverage the Generative AI text models powered by Amazon Bedrock. Using these constructs, you can use prompt engineering techniques to transform text documents, including, text summarization, text translation, information extraction, and more!

The Llama text processor is part of the Bedrock text processors and allows you to leverage the Llama family of large-language models provided by Meta within your pipelines.


📝 Text Generation

To start using Llama models in your pipelines, you import the LlamaTextProcessor construct in your CDK stack, and specify the specific text model you want to use.

💁 The below example demonstrates how to use the Llama text processor to summarize input documents uploaded to an S3 bucket using the Llama 3 70B model.

import { S3EventTrigger } from '@project-lakechain/s3-event-trigger';
import { LlamaTextProcessor, LlamaModel } from '@project-lakechain/bedrock-text-processors';
import { CacheStorage } from '@project-lakechain/core';
class Stack extends cdk.Stack {
constructor(scope: cdk.Construct, id: string) {
const cache = new CacheStorage(this, 'Cache');
// Create the S3 event trigger.
const trigger = new S3EventTrigger.Builder()
.withScope(this)
.withIdentifier('Trigger')
.withCacheStorage(cache)
.withBucket(bucket)
.build();
// Summarize input documents using Llama 3.
const llama = new LlamaTextProcessor.Builder()
.withScope(this)
.withIdentifier('LlamaTextProcessor')
.withCacheStorage(cache)
.withSource(source)
.withModel(LlamaModel.LLAMA3_70B_INSTRUCT_V1)
.withPrompt(`
Give a detailed summary of the text with the following constraints:
- Write the summary in the same language as the original text.
- Keep the original meaning, style, and tone of the text in the summary.
`)
.withModelParameters({
max_gen_len: 2048
})
.build();
}
}


🤖 Model Selection

You can select the specific Llama model to use with this middleware using the .withModel API.

import { LlamaTextProcessor, LlamaModel } from '@project-lakechain/bedrock-text-processors';
const llama = new LlamaTextProcessor.Builder()
.withScope(this)
.withIdentifier('LlamaTextProcessor')
.withCacheStorage(cache)
.withSource(source)
.withModel(LlamaModel.LLAMA2_13B_CHAT_V1) // 👈 Specify a model
.withPrompt(prompt)
.build();

💁 You can choose amongst the following models — see the Bedrock documentation for more information.

Model NameModel identifier
LLAMA2_13B_CHAT_V1meta.llama2-13b-chat-v1
LLAMA2_70B_CHAT_V1meta.llama2-70b-chat-v1
LLAMA3_8B_INSTRUCT_V1meta.llama3-8b-instruct-v1:0
LLAMA3_70B_INSTRUCT_V1meta.llama3-70b-instruct-v1:0


🌐 Region Selection

You can specify the AWS region in which you want to invoke Amazon Bedrock using the .withRegion API. This can be helpful if Amazon Bedrock is not yet available in your deployment region.

💁 By default, the middleware will use the current region in which it is deployed.

import { LlamaTextProcessor, LlamaModel } from '@project-lakechain/bedrock-text-processors';
const llama = new LlamaTextProcessor.Builder()
.withScope(this)
.withIdentifier('LlamaTextProcessor')
.withCacheStorage(cache)
.withSource(source)
.withRegion('eu-central-1') // 👈 Alternate region
.withModel(LlamaModel.LLAMA2_13B_CHAT_V1)
.withPrompt(prompt)
.build();


⚙️ Model Parameters

You can forward specific parameters to text models using the .withModelParameters method. See the Bedrock Inference Parameters for more information on the parameters supported by the different models.



🏗️ Architecture

This middleware is based on a Lambda compute running on an ARM64 architecture, and integrate with Amazon Bedrock to generate text based on the given prompt and input documents.

Architecture



🏷️ Properties


Supported Inputs
Mime TypeDescription
text/plainUTF-8 text documents.
text/markdownMarkdown documents.
text/csvCSV documents.
text/htmlHTML documents.
application/x-subripSubRip subtitles.
text/vttWeb Video Text Tracks (WebVTT) subtitles.
application/jsonJSON documents.
application/json+schedulerUsed by the Scheduler middleware.
Supported Outputs
Mime TypeDescription
text/plainUTF-8 text documents.
Supported Compute Types
TypeDescription
CPUThis middleware only supports CPU compute.