chore: add Cloudflare AI gateway authentication token

Author: derogab

README.md

@@ -31,6 +31,7 @@ You need to create a personal telegram bot and connect it to your worker.
 1. First of all you need to create a bot with [BotFather](https://t.me/BotFather).
 2. Go to _Cloudflare > Workers & Pages > Your Worker > Settings > Variables > Add Encrypted Environment Variables_
    - `AI_GATEWAY`: (optional) set the Cloudflare AI gateway ID.
+   - `CF_AIG_AUTHORIZATION`: (optional) set the Cloudflare AI gateway authentication token.
    - `OPENAI_API_KEY`: (optional) set the OpenAI API KEY. If available GPT will be used, otherwise Llama4.
    - `OPENAI_BASE_URL`: (optional) set a custom OpenAI compatible BASE URL. If available it will be used, otherwise the openai default one.
    - `TELEGRAM_BOT_API_KEY`: (mandatory) set the API KEY of the telegram bot.

src/env.ts

@@ -6,4 +6,5 @@ export interface Env {
   TZ: string;
   OPENAI_API_KEY: string;
   OPENAI_BASE_URL: string | undefined;
+  CF_AIG_AUTHORIZATION: string | undefined;
 }

src/llm.ts

@@ -27,7 +27,13 @@ export async function llm(messages: MessageParam[], env: Env) : Promise<MessageP
     // Try to make a call to the OpenAI API.
     try {
       // Initialize OpenAI.
-      const openai = new OpenAI({ apiKey: env.OPENAI_API_KEY, baseURL: env.OPENAI_BASE_URL });
+      const openai = new OpenAI({
+        apiKey: env.OPENAI_API_KEY,
+        baseURL: env.OPENAI_BASE_URL,
+        defaultHeaders: {
+          "cf-aig-authorization": `Bearer ${env.CF_AIG_AUTHORIZATION}`,
+        }
+      });
       // Call the OpenAI API.
       const chatCompletion = await openai.chat.completions.create({ model: 'gpt-4o-mini', messages: messages });
       const response = chatCompletion.choices[0].message;