Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
79 changes: 79 additions & 0 deletions src/components/models/code/Bge-Reranker-Base.astro
Original file line number Diff line number Diff line change
@@ -0,0 +1,79 @@
---
import { z } from "astro:schema";
import { Code } from "@astrojs/starlight/components";
import Details from "~/components/Details.astro";

type Props = z.infer<typeof props>;

const props = z.object({
name: z.string(),
});

const { name } = props.parse(Astro.props);

const worker = `
export interface Env {
AI: Ai;
}

export default {
async fetch(request, env): Promise<Response> {
const query = 'Which one is cooler?'
const contexts = [
{
text: 'a cyberpunk lizzard'
},
{
text: 'a cyberpunk cat'
}
];

const response = await env.AI.run('${name}', { query, contexts });

return Response.json(response);
},
} satisfies ExportedHandler<Env>;

`;

const python = `
import os
import requests

ACCOUNT_ID = "your-account-id"
AUTH_TOKEN = os.environ.get("CLOUDFLARE_AUTH_TOKEN")

response = requests.post(
f"https://api.cloudflare.com/client/v4/accounts/{ACCOUNT_ID}/ai/run/${name}",
headers={"Authorization": f"Bearer {AUTH_TOKEN}"},
json={
"query": "Which one is better?"
"context": [
{"text": "a cyberpunk lizzard"},
{"text": "a cyberpunk car"},
]
}
)
result = response.json()
print(result)
`;

const curl = `
curl https://api.cloudflare.com/client/v4/accounts/$CLOUDFLARE_ACCOUNT_ID/ai/run/${name} \\
-X POST \\
-H "Authorization: Bearer $CLOUDFLARE_AUTH_TOKEN" \\
-d '{ "query": "Which one is better?", "contexts": [{ "text": "a cyberpunk lizzard" }, {"text": "a cyberpunk cat"}]}'
`;
---

<Details header="Worker">
<Code code={worker} lang="ts" />
</Details>

<Details header="Python">
<Code code={python} lang="py" />
</Details>

<Details header="curl">
<Code code={curl} lang="sh" />
</Details>
5 changes: 5 additions & 0 deletions src/pages/workers-ai/models/[name].astro
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,7 @@ import StableDiffusionV15InpaintingCode from "~/components/models/code/StableDif
import Flux1Schnell from "~/components/models/code/Flux-1-Schnell.astro";
import WhisperBase64Code from "~/components/models/code/WhisperBase64Code.astro";
import LlamaGuard from "~/components/models/code/LlamaGuard.astro";
import BgeRerankerBase from "~/components/models/code/Bge-Reranker-Base.astro";

import { authorData } from "~/components/models/data";

Expand Down Expand Up @@ -98,6 +99,10 @@ if (model.name === "@cf/meta/llama-guard-3-8b") {
CodeExamples = LlamaGuard;
}

if (model.name === "@cf/baai/bge-reranker-base") {
CodeExamples = BgeRerankerBase;
}

const description = model.description;

const isBeta = model.properties.find(
Expand Down