-
Notifications
You must be signed in to change notification settings - Fork 5.1k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[AIG] Adding AI Gateway to Workers AI Binding (#17595)
* AI Gateway binding * Update src/content/docs/ai-gateway/integration/aig-workers-ai-binding.mdx Co-authored-by: hyperlint-ai[bot] <154288675+hyperlint-ai[bot]@users.noreply.github.com> * Update src/content/docs/ai-gateway/integration/aig-workers-ai-binding.mdx Co-authored-by: Jun Lee <junlee@cloudflare.com> * Update src/content/docs/ai-gateway/integration/aig-workers-ai-binding.mdx Co-authored-by: Jun Lee <junlee@cloudflare.com> * Update src/content/docs/ai-gateway/integration/aig-workers-ai-binding.mdx Co-authored-by: Jun Lee <junlee@cloudflare.com> * minor edits --------- Co-authored-by: hyperlint-ai[bot] <154288675+hyperlint-ai[bot]@users.noreply.github.com> Co-authored-by: Jun Lee <junlee@cloudflare.com>
- Loading branch information
1 parent
c579dad
commit 41ab20f
Showing
1 changed file
with
142 additions
and
0 deletions.
There are no files selected for viewing
142 changes: 142 additions & 0 deletions
142
src/content/docs/ai-gateway/integration/aig-workers-ai-binding.mdx
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,142 @@ | ||
--- | ||
title: Workers AI | ||
pcx_content_type: tutorial | ||
updated: 2024-10-17 | ||
--- | ||
|
||
import { Render, PackageManagers } from "~/components"; | ||
|
||
This guide will walk you through setting up and deploying a Workers AI project. You will use [Workers](/workers/), an AI Gateway binding, and a large language model (LLM), to deploy your first AI-powered application on the Cloudflare global network. | ||
|
||
## Prerequisites | ||
|
||
<Render file="prereqs" product="workers" /> | ||
|
||
## 1. Create a Worker Project | ||
|
||
You will create a new Worker project using the create-Cloudflare CLI (C3). C3 is a command-line tool designed to help you set up and deploy new applications to Cloudflare. | ||
|
||
Create a new project named `hello-ai` by running: | ||
|
||
<PackageManagers type="create" pkg="cloudflare@latest" args={"hello-ai"} /> | ||
|
||
Running `npm create cloudflare@latest` will prompt you to install the create-cloudflare package and lead you through setup. C3 will also install [Wrangler](/workers/wrangler/), the Cloudflare Developer Platform CLI. | ||
|
||
<Render | ||
file="c3-post-run-steps" | ||
product="workers" | ||
params={{ | ||
category: "hello-world", | ||
type: "Hello World Worker", | ||
lang: "TypeScript", | ||
}} | ||
/> | ||
|
||
This will create a new `hello-ai` directory. Your new `hello-ai` directory will include: | ||
|
||
- A "Hello World" Worker at `src/index.ts`. | ||
- A `wrangler.toml` configuration file. | ||
|
||
Go to your application directory: | ||
|
||
```bash | ||
cd hello-ai | ||
``` | ||
|
||
## 2. Connect your Worker to Workers AI | ||
|
||
You must create an AI binding for your Worker to connect to Workers AI. Bindings allow your Workers to interact with resources, like Workers AI, on the Cloudflare Developer Platform. | ||
|
||
To bind Workers AI to your Worker, add the following to the end of your `wrangler.toml` file: | ||
|
||
```toml title="wrangler.toml" | ||
[ai] | ||
binding = "AI" | ||
``` | ||
|
||
Your binding is [available in your Worker code](/workers/reference/migrate-to-module-workers/#bindings-in-es-modules-format) on [`env.AI`](/workers/runtime-apis/handlers/fetch/). | ||
|
||
You will need to have your `gateway id` for the next step. You can learn [how to create an AI Gateway in this tutorial](/ai-gateway/get-started/). | ||
|
||
## 3. Run an inference task containing AI Gateway in your Worker | ||
|
||
You are now ready to run an inference task in your Worker. In this case, you will use an LLM, [`llama-3.1-8b-instruct-fast`](/workers-ai/models/llama-3.1-8b-instruct-fast/), to answer a question. Your gateway ID is found on the dashboard. | ||
|
||
Update the `index.ts` file in your `hello-ai` application directory with the following code: | ||
|
||
```typescript title="src/index.ts" {78-81} | ||
export interface Env { | ||
// If you set another name in wrangler.toml as the value for 'binding', | ||
// replace "AI" with the variable name you defined. | ||
AI: Ai; | ||
} | ||
|
||
export default { | ||
async fetch(request, env): Promise<Response> { | ||
// Specify the gateway label and other options here | ||
const response = await env.AI.run("@cf/meta/llama-3.1-8b-instruct-fast", { | ||
prompt: "What is the origin of the phrase Hello, World", | ||
gateway: { | ||
id: "GATEWAYID", // Use your gateway label here | ||
skipCache: true, // Optional: Skip cache if needed | ||
}, | ||
}); | ||
|
||
// Return the AI response as a JSON object | ||
return new Response(JSON.stringify(response), { | ||
headers: { "Content-Type": "application/json" }, | ||
}); | ||
}, | ||
} satisfies ExportedHandler<Env>; | ||
``` | ||
|
||
Up to this point, you have created an AI binding for your Worker and configured your Worker to be able to execute the Llama 3.1 model. You can now test your project locally before you deploy globally. | ||
|
||
## 4. Develop locally with Wrangler | ||
|
||
While in your project directory, test Workers AI locally by running [`wrangler dev`](/workers/wrangler/commands/#dev): | ||
|
||
```bash | ||
npx wrangler dev | ||
``` | ||
|
||
<Render file="ai-local-usage-charges" product="workers" /> | ||
|
||
You will be prompted to log in after you run `wrangler dev`. When you run `npx wrangler dev`, Wrangler will give you a URL (most likely `localhost:8787`) to review your Worker. After you go to the URL Wrangler provides, you will see a message that resembles the following example: | ||
|
||
````json | ||
{ | ||
"response": "A fascinating question!\n\nThe phrase \"Hello, World!\" originates from a simple computer program written in the early days of programming. It is often attributed to Brian Kernighan, a Canadian computer scientist and a pioneer in the field of computer programming.\n\nIn the early 1970s, Kernighan, along with his colleague Dennis Ritchie, were working on the C programming language. They wanted to create a simple program that would output a message to the screen to demonstrate the basic structure of a program. They chose the phrase \"Hello, World!\" because it was a simple and recognizable message that would illustrate how a program could print text to the screen.\n\nThe exact code was written in the 5th edition of Kernighan and Ritchie's book \"The C Programming Language,\" published in 1988. The code, literally known as \"Hello, World!\" is as follows:\n\n``` | ||
main() | ||
{ | ||
printf(\"Hello, World!\"); | ||
} | ||
```\n\nThis code is still often used as a starting point for learning programming languages, as it demonstrates how to output a simple message to the console.\n\nThe phrase \"Hello, World!\" has since become a catch-all phrase to indicate the start of a new program or a small test program, and is widely used in computer science and programming education.\n\nSincerely, I'm glad I could help clarify the origin of this iconic phrase for you!" | ||
} | ||
```` | ||
|
||
## 5. Deploy your AI Worker | ||
|
||
Before deploying your AI Worker globally, log in with your Cloudflare account by running: | ||
|
||
```bash | ||
npx wrangler login | ||
``` | ||
|
||
You will be directed to a web page asking you to log in to the Cloudflare dashboard. After you have logged in, you will be asked if Wrangler can make changes to your Cloudflare account. Scroll down and select **Allow** to continue. | ||
|
||
Finally, deploy your Worker to make your project accessible on the Internet. To deploy your Worker, run: | ||
|
||
```bash | ||
npx wrangler deploy | ||
``` | ||
|
||
Once deployed, your Worker will be available at a URL like: | ||
|
||
```bash | ||
https://hello-ai.<YOUR_SUBDOMAIN>.workers.dev | ||
``` | ||
|
||
Your Worker will be deployed to your custom [`workers.dev`](/workers/configuration/routing/workers-dev/) subdomain. You can now visit the URL to run your AI Worker. | ||
|
||
By completing this tutorial, you have created a Worker, connected it to Workers AI through an AI Gateway binding, and successfully ran an inference task using the Llama 3.1 model. |