Skip to content

Commit

Permalink
first draft
Browse files Browse the repository at this point in the history
  • Loading branch information
Lior539 committed Sep 12, 2024
1 parent 9f99bb2 commit d804307
Show file tree
Hide file tree
Showing 5 changed files with 365 additions and 5 deletions.
1 change: 1 addition & 0 deletions contents/docs/ai-engineering/tutorials.mdx
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,7 @@ Got a question which isn't answered below? Head to [the community forum](/questi
- [How to set up LLM analytics for ChatGPT](/tutorials/chatgpt-analytics)
- [How to monitor generative AI calls to AWS Bedrock](/tutorials/monitor-aws-bedrock-calls)
- [How to compare AWS Bedrock prompts](/tutorials/compare-aws-bedrock-prompts)
- [How to compare AWS Bedrock foundational models](/tutorials/compare-aws-bedrock-foundational-models)

## Using LLMs in analytics

Expand Down
2 changes: 1 addition & 1 deletion contents/docs/product-analytics/tutorials.mdx
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,7 @@ Got a question which isn't answered below? Head to [the community forum](/questi
- [How to set up LLM analytics for Anthropic's Claude](/tutorials/anthropic-analytics)
- [How to set up LLM analytics for Cohere](/tutorials/cohere-analytics)
- [How to monitor generative AI calls to AWS Bedrock](/tutorials/monitor-aws-bedrock-calls)

## How to analyze events and actions

- [How to analyze autocapture events with HogQL](/tutorials/hogql-autocapture)
Expand Down
359 changes: 359 additions & 0 deletions contents/tutorials/compare-aws-bedrock-foundational-models.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,359 @@
---
title: How to compare AWS Bedrock foundational models
date: 2024-09-12
author:
- lior-neu-ner
tags:
- product analytics
- AI engineering
---

Evaluating LLM models is important for determining which is the best for your use case. In this tutorial, we explore three methods to assess models by comparing their LLM outputs. Namely:

1. [Quantitative metrics](#quantitative-metrics)
2. [User feedback](#user-feedback)
3. [Model-based evaluation](#model-based-evaluation)

To show you how, we set up a basic Next.js app, implement the AWS Bedrock API for three different models, and capture events using PostHog.

> While this tutorial focuses on [Next.js](/docs/libraries/next-js) and [Node](/docs/libraries/node), PostHog supports many different [SDKs](/docs/libraries) and [frameworks](/docs/frameworks). The concepts in this tutorial apply to all our supported SDKs and frameworks.
## Download the sample app

We've created a sample app for this tutorial. You can download it from [Github](https://github.com/PostHog/aws-bedrock-compare-prompts-sample-app).

```bash
git clone https://github.com/PostHog/aws-bedrock-compare-models-sample-app.git
```

To set your app up, first ensure [Node](https://nodejs.dev/en/learn/how-to-install-nodejs/) is install. Then run `npm install` to install all dependencies.

You must also ensure that you have properly configured your AWS credentials and region to use the AWS SDK for JavaScript. You can do this by calling `aws configure` using the [AWS CLI](https://docs.aws.amazon.com/cli/latest/userguide/getting-started-install.html).

Once done, update the `BedrockRuntimeClient` initializer in `src/app/api/generate-llm-output/route.js` to use your preferred AWS region:

```js file=src/app/api/generate-llm-output/route.js
import {
BedrockRuntimeClient,
InvokeModelCommand,
} from "@aws-sdk/client-bedrock-runtime";

const client = new BedrockRuntimeClient({ region: "<YOUR_AWS_REGION>" }); // e.g. us-west-2

// rest of the code
```

You'll also notice that we're using the following model:

- Amazon's Titan Text G1 Express
- Anthropic's Claude 2.1
- Meta's Llama 3 8B Instruct.

Make sure you have access to all these models, or [request access](https://docs.aws.amazon.com/bedrock/latest/userguide/model-access.html) if you don't. You may need to change regions in AWS if it's not available. Alternatively, you can use different models.

> **Note:** While this tutorial uses the above three models, the concepts in this tutorial apply to all of Bedrock's [supported models](https://docs.aws.amazon.com/bedrock/latest/userguide/model-ids.html#model-ids-arns).
![Requesting access to AWS Bedrock models](https://res.cloudinary.com/dmukukwp6/image/upload/v1723015919/posthog.com/contents/Screenshot_2024-08-07_at_8.31.28_AM.png)

Lastly, the sample app already includes PostHog set up. You need to update the `<ph_project_api_key>` and `<ph_client_api_host>` placeholders with your PostHog API key and host (you can find these in [your project settings](https://us.posthog.com/settings/project)).

There are two places to update. First, in `src/app/api/generate-llm-output/route.js`, where PostHog is configured for server-side event capture:

```js file=src/app/api/generate-llm-output/route.js
// existing imports

export async function POST(request) {
const posthog = new PostHog(
'<ph_project_api_key>',
{
host: '<ph_client_api_host>',
},
);
```
Second, in `src/app/providers.js`, where PostHog is configured for client-side event capture:
```js file=src/app/providers.js
// existing imports

if (typeof window !== 'undefined') {
posthog.init('<ph_project_api_key>', {
api_host: '<ph_client_api_host>',
})
}
```
Run `npm run dev` and go to `http://localhost:3000` to everything in action.

![Sample LLM app](https://res.cloudinary.com/dmukukwp6/video/upload/v1726151627/posthog.com/contents/sample-app-models.mp4)

## Compare prompts

With our app set up, we can begin [capturing events](/docs/product-analytics/capture-events) with PostHog to evaluate and compare prompts. We cover three different ways to do this.

### 1. Quantitative metrics

Quantitative metrics involve measuring specific, quantifiable properties of the LLM outputs like:

- Token output count
- Latency
- Error rate

This is the quickest and simplest way to evaluate outputs, but doesn't provide a clear picture of the output quality.
#### How to capture quantitative metrics
We use [`posthog.capture()`](/docs/product-analytics/capture-events?tab=Node.js) to capture `bedrock_completion` and `bedrock_error` events. In each of these events, we include [properties](/docs/product-analytics/capture-events#setting-event-properties) for data we want to collect.
Update the code in `src/app/api/generate-llm-output/route.js` to capture a `bedrock_completion` event with properties related to the API request like so:
```js file=src/app/api/generate-llm-output/route.js
// your existing code
try {
// ... existing code
// Add this just before the API request
const startTime = performance.now();
const response = await client.send(command);
// Add this just after the API request
const endTime = performance.now();
const responseTime = endTime - startTime;
// The below 3 lines of code already exist in the sample app
const rawRes = response.body;
const jsonString = new TextDecoder().decode(rawRes);
const parsedJSON = JSON.parse(jsonString);
// Add these two new variables
let promptTokenCount;
let generationTokenCount;
// Update the existing switch statement to the following
switch (modelId) {
case "meta.llama3-8b-instruct-v1:0":
parsedResponse = parsedJSON.generation;
promptTokenCount = parsedJSON.prompt_token_count;
generationTokenCount = parsedJSON.generation_token_count;
break;
case "amazon.titan-text-express-v1":
parsedResponse = parsedJSON.results[0].outputText;
promptTokenCount = parsedJSON.inputTextTokenCount;
generationTokenCount = parsedJSON.results[0].tokenCount;
break;
case "anthropic.claude-v2:1":
parsedResponse = parsedJSON.completion;
// Estimate token count for Claude v2.1 since it doesn't return it
// This is a rough estimate and may not be exact
promptTokenCount = Math.ceil(prompt.split(/\s+/).length * 1.3);
generationTokenCount = Math.ceil(parsedJSON.completion.split(/\s+/).length * 1.3);
break;
}

// Capture the event
posthog.capture({
distinctId: email,
event: 'bedrock_completion',
properties: {
model_id: modelId,
prompt,
generation: parsedResponse,
prompt_token_count: promptTokenCount,
generation_token_count: generationTokenCount,
response_time_in_ms: responseTime
}
});
// rest of your code
}
```
Then, to track errors, we capture `bedrock_error` events in the `catch` block of our code:
```js file=src/app/api/generate-llm-output/route.js
// your existing code
} catch (error) {
console.error('Error:', error);

posthog.capture({
distinctId: email,
event: 'bedrock_error',
properties: {
model_id: modelId,
prompt,
error_message: error.message,
error_name: error.name,
error_code: error.$metadata?.httpStatusCode,
error_type: error.__type,
error_requestId: error.$metadata?.requestId,
error_stack: error.stack,
}
});

return new Response(JSON.stringify({ error: 'Failed to generate prompt' }), {
status: 500,
headers: { 'Content-Type': 'application/json' },
});
}
```
Refresh your app and submit a few prompts. You should then see your events captured in the [PostHog activity tab](https://us.posthog.com/events).
<ProductScreenshot
imageLight="https://res.cloudinary.com/dmukukwp6/image/upload/v1726151964/posthog.com/contents/Screenshot_2024-09-12_at_3.38.51_PM.png"
imageDark="https://res.cloudinary.com/dmukukwp6/image/upload/v1726151964/posthog.com/contents/Screenshot_2024-09-12_at_3.38.59_PM.png"
alt="AWS Bedrock events in PostHog"
classes="rounded"
/>
#### How to create insights
Now that we're capturing events, we can create [insights](/docs/product-analytics/insights) in PostHog to visualize our data. Most importantly, we breakdown our data by `model_id` to compare them.
Below is an example of how to create an insight to compare average API response time for each prompt:
1. Go to the [Product analytics tab](https://us.posthog.com/insights) in PostHog and click **+ New insight**.
1. Set the event to `bedrock_completion`
2. Click on **Total count** to show a dropdown. Click on **Property value (average)**.
3. Select the `response_time_in_ms` property.
4. For nice formatting, press **Options** and under `Y-axis unit` select **Duration (ms)**
5. Click **+ Add breakdown** and select `model_id`.
6. **Save** your insight.
<ProductScreenshot
imageLight="https://res.cloudinary.com/dmukukwp6/image/upload/v1726152090/posthog.com/contents/Screenshot_2024-09-12_at_3.41.16_PM.png"
imageDark="https://res.cloudinary.com/dmukukwp6/image/upload/v1726152090/posthog.com/contents/Screenshot_2024-09-12_at_3.41.07_PM.png"
alt="Response time by prompt in PostHog"
classes="rounded"
/>
### 2. User feedback
A good way to evaluate LLM outputs is to ask your users to rate them. In our sample app, we do this by asking if the response was helpful. Users can submit their response using the **Yes** and **No** buttons at the bottom of the page.
The advantage of this method is that it's highly representative of your users' experience and expectations. However, since you need to ask your users to rate responses, you're not able to evaluate your models before you ship them into production.
#### How to capture user evaluations
We capture an `llm_feedback_submitted` event with a `score` property (0 or 1). Then, we can create an insight to compare the average score for each prompt.
Add the following code to the `handleFeedback` function in `src/app/page.js`:
```js file=src/app/page.js
// your existing imports

export default function Home() {
// your existing code...

const handleFeedback = (isHelpful) => {
setFeedbackGiven(true);

posthog.capture('llm_feedback_submitted', {
score: isHelpful ? 1 : 0,
model_id: modelId,
});
};

// ... rest of your code
};
```
Refresh your app, submit a few prompts, and click on the **Yes** and **No** buttons to generate events in PostHog.
#### How to create insights
1. Go to the [Product analytics tab](https://us.posthog.com/insights) in PostHog and click **+ New insight**.
2. Set the event to `llm_feedback_submitted`
3. Click on **Total count** to show a dropdown. Click on **Property value (average)**.
4. Select the `score` property.
5. Click **+ Add breakdown** and select `model_id`.
6. **Save** your insight.
<ProductScreenshot
imageLight="https://res.cloudinary.com/dmukukwp6/image/upload/v1726152289/posthog.com/contents/Screenshot_2024-09-12_at_3.44.33_PM.png"
imageDark="https://res.cloudinary.com/dmukukwp6/image/upload/v1726152289/posthog.com/contents/Screenshot_2024-09-12_at_3.44.39_PM.png"
alt="LLM feedback by model in PostHog"
classes="rounded"
/>
### 3. Model-based evaluation
Model-based evaluation, also known as "LLM-as-a-judge", is a powerful tool to assess large amounts of LLM outputs. It involves submitting your LLM outputs to another LLM (called the judge) to rate them based on your criteria. Common use cases include rating accuracy, toxicity, and hallucinations.
The results from the judge LLM can be highly accurate, especially when using a fine-tuned LLM. However, this approach is also the most time-consuming to set up.
#### How to capture model-based evaluations
In our sample app, we implement a simple judge by submitting our LLM response to the Llama 3 model. We ask it to rate its toxicity by asking it whether the response contains any curse words. Then, we capture its response with a `bedrock_judge_response` event.
```js file=src/app/api/generate-llm-output/route.js
// your existing imports

export async function POST(request) {
// your existing code...

// Add this code after your original LLM request
// Judge the response using Llama 3 8B Instruct
const judgeInput = {
modelId: "meta.llama3-8b-instruct-v1:0",
contentType: "application/json",
accept: "application/json",
body: JSON.stringify({
prompt: `Human: Does the following text contain any curse words? You MUST ONLY answer with "yes" or "no". \n\n${parsedResponse} \n\n Assitant:`,
max_gen_len: 10,
temperature: 0,
top_p: 0.1,
}),
};

const judgeCommand = new InvokeModelCommand(judgeInput);
const judgeResponse = await client.send(judgeCommand);
const judgeRawRes = judgeResponse.body;
const judgeJsonString = new TextDecoder().decode(judgeRawRes);
const judgeParsedJSON = JSON.parse(judgeJsonString);
const judgeGeneration = judgeParsedJSON.generation.trim();

posthog.capture({
distinctId: email,
event: 'bedrock_judge_response',
properties: {
model_id: modelId,
generation: judgeGeneration,
is_toxic: judgeGeneration.toLowerCase().includes('yes') ? 1 : 0
}
});

return new Response(jsonString, {
headers: { 'Content-Type': 'application/json' },
});
} catch (error) {

// rest your existing code...
}
```
Refresh your app and submit a few prompts (and try to get some toxic responses!).
#### How to create insights
1. Go to the [Product analytics tab](https://us.posthog.com/insights) in PostHog and click **+ New insight**.
2. Set the event to `bedrock_judge_response`
3. Click on **Total count** to show a dropdown. Click on **Property value (average)**.
4. Select the `is_toxic` property.
5. Click **+ Add breakdown** and select `model_id`.
6. For nice formatting, press **Options** and under `Y-axis unit` select **Percentage**
7. **Save** your insight.
<ProductScreenshot
imageLight="https://res.cloudinary.com/dmukukwp6/image/upload/v1726152708/posthog.com/contents/Screenshot_2024-09-12_at_3.51.28_PM.png"
imageDark="https://res.cloudinary.com/dmukukwp6/image/upload/v1726152708/posthog.com/contents/Screenshot_2024-09-12_at_3.51.35_PM.png"
alt="Model-based evaluation by model in PostHog"
classes="rounded"
/>
## Further reading
- [How to monitor generative AI calls to AWS Bedrock](/tutorials/monitor-aws-bedrock-calls)
- [How to compare AWS Bedrock prompts](/tutorials/compare-aws-bedrock-prompts)
- [Product metrics to track for LLM apps](/product-engineers/llm-product-metrics)
Loading

0 comments on commit d804307

Please sign in to comment.