[prototype] bedrock anthropic with invokemodel

examples/ai-core/src/generate-object/amazon-bedrock-anthropic.ts

@@ -0,0 +1,34 @@
+import { bedrock } from '@ai-sdk/amazon-bedrock';
+import { generateObject } from 'ai';
+import dotenv from 'dotenv';
+import { z } from 'zod';
+
+dotenv.config();
+
+async function main() {
+  const result = await generateObject({
+    model: bedrock.anthropicLanguageModel(
+      'anthropic.claude-3-5-sonnet-20240620-v1:0',
+    ),
+    schema: z.object({
+      recipe: z.object({
+        name: z.string(),
+        ingredients: z.array(
+          z.object({
+            name: z.string(),
+            amount: z.string(),
+          }),
+        ),
+        steps: z.array(z.string()),
+      }),
+    }),
+    prompt: 'Generate a lasagna recipe.',
+  });
+
+  console.log(JSON.stringify(result.object, null, 2));
+  console.log();
+  console.log('Token usage:', result.usage);
+  console.log('Finish reason:', result.finishReason);
+}
+
+main().catch(console.error);

examples/ai-core/src/generate-text/amazon-bedrock-anthropic.ts

@@ -0,0 +1,21 @@
+import { bedrock } from '@ai-sdk/amazon-bedrock';
+import { generateText } from 'ai';
+import dotenv from 'dotenv';
+
+dotenv.config();
+
+async function main() {
+  const result = await generateText({
+    model: bedrock.anthropicLanguageModel(
+      'anthropic.claude-3-haiku-20240307-v1:0',
+    ),
+    prompt: 'Invent a new holiday and describe its traditions.',
+  });
+
+  console.log(result.text);
+  console.log();
+  console.log('Token usage:', result.usage);
+  console.log('Finish reason:', result.finishReason);
+}
+
+main().catch(console.error);

examples/ai-core/src/stream-object/amazon-bedrock-anthropic.ts

@@ -0,0 +1,34 @@
+import { bedrock } from '@ai-sdk/amazon-bedrock';
+import { streamObject } from 'ai';
+import dotenv from 'dotenv';
+import { z } from 'zod';
+
+dotenv.config();
+
+async function main() {
+  const result = await streamObject({
+    model: bedrock.anthropicLanguageModel(
+      'anthropic.claude-3-5-sonnet-20240620-v1:0',
+    ),
+    schema: z.object({
+      characters: z.array(
+        z.object({
+          name: z.string(),
+          class: z
+            .string()
+            .describe('Character class, e.g. warrior, mage, or thief.'),
+          description: z.string(),
+        }),
+      ),
+    }),
+    prompt:
+      'Generate 3 character descriptions for a fantasy role playing game.',
+  });
+
+  for await (const partialObject of result.partialObjectStream) {
+    console.clear();
+    console.log(partialObject);
+  }
+}
+
+main().catch(console.error);

examples/ai-core/src/stream-text/amazon-bedrock-anthropic.ts

@@ -0,0 +1,24 @@
+import { bedrock } from '@ai-sdk/amazon-bedrock';
+import { streamText } from 'ai';
+import dotenv from 'dotenv';
+
+dotenv.config();
+
+async function main() {
+  const result = await streamText({
+    model: bedrock.anthropicLanguageModel(
+      'anthropic.claude-3-haiku-20240307-v1:0',
+    ),
+    prompt: 'Invent a new holiday and describe its traditions.',
+  });
+
+  for await (const textPart of result.textStream) {
+    process.stdout.write(textPart);
+  }
+
+  console.log();
+  console.log('Token usage:', await result.usage);
+  console.log('Finish reason:', await result.finishReason);
+}
+
+main().catch(console.error);

packages/amazon-bedrock/src/anthropic/anthropic-messages-prompt.ts

@@ -0,0 +1,46 @@
+export type AnthropicMessagesPrompt = {
+  system?: string;
+  messages: AnthropicMessage[];
+};
+
+export type AnthropicMessage = AnthropicUserMessage | AnthropicAssistantMessage;
+
+export interface AnthropicUserMessage {
+  role: 'user';
+  content: Array<
+    AnthropicTextContent | AnthropicImageContent | AnthropicToolResultContent
+  >;
+}
+
+export interface AnthropicAssistantMessage {
+  role: 'assistant';
+  content: Array<AnthropicTextContent | AnthropicToolCallContent>;
+}
+
+export interface AnthropicTextContent {
+  type: 'text';
+  text: string;
+}
+
+export interface AnthropicImageContent {
+  type: 'image';
+  source: {
+    type: 'base64';
+    media_type: string;
+    data: string;
+  };
+}
+
+export interface AnthropicToolCallContent {
+  type: 'tool_use';
+  id: string;
+  name: string;
+  input: unknown;
+}
+
+export interface AnthropicToolResultContent {
+  type: 'tool_result';
+  tool_use_id: string;
+  content: unknown;
+  is_error?: boolean;
+}