AI gateway -> request handling

src/content/changelogs/ai-gateway.yaml

@@ -5,6 +5,11 @@ productLink: "/ai-gateway/"
 productArea: Developer platform
 productAreaLink: /workers/platform/changelog/platform/
 entries:
+  - publish_date: "2025-02-06"
+    title: Added request handling
+    description: |-
+      * Added [request handling options](/ai-gateway/request-handling/) to help manage AI provider interactions effectively, ensuring your applications remain responsive and reliable.
+
   - publish_date: "2025-02-05"
     title: New AI Gateway providers
     description: |-

src/content/docs/ai-gateway/configuration/fallbacks.mdx

@@ -9,11 +9,15 @@ import { Render } from "~/components";
 
 Specify model or provider fallbacks with your [Universal endpoint](/ai-gateway/providers/universal/) to handle request failures and ensure reliability.
 
-Fallbacks are currently triggered only when a request encounters an error. We are working to expand fallback functionality to include time-based triggers, which will allow requests that exceed a predefined response time to timeout and fallback. 
+Cloudflare can trigger your fallback provider in response to [request errors](#request-failures) or [predetermined request timeouts](#request-timeouts). The [response header `cf-aig-step`](#response-headercf-aig-step) indicates which step successfully processed the request.
 
-## Example
+## Request failures
 
-In the following example, a request first goes to the [Workers AI](/workers-ai/) Inference API. If the request fails, it falls back to OpenAI. The response header `cf-aig-step` indicates which provider successfully processed the request. 
+By default, Cloudflare triggers your fallback if a model request returns an error.
+
+### Example
+
+In the following example, a request first goes to the [Workers AI](/workers-ai/) Inference API. If the request fails, it falls back to OpenAI. The response header `cf-aig-step` indicates which provider successfully processed the request.
 
 1. Sends a request to Workers AI Inference API.
 2. If that request fails, proceeds to OpenAI.

src/content/docs/ai-gateway/configuration/request-handling.mdx

@@ -0,0 +1,205 @@
+---
+pcx_content_type: configuration
+title: Request handling
+sidebar:
+  order: 4
+---
+
+import { Render, Aside } from "~/components";
+
+Your AI gateway supports different strategies for handling requests to providers, which allows you to manage AI interactions effectively and ensure your applications remain responsive and reliable.
+
+## Request timeouts
+
+A request timeout allows you to trigger fallbacks or a retry if a provider takes too long to respond.
+
+These timeouts help:
+
+- Improve user experience, by preventing users from waiting too long for a response
+- Proactively handle errors, by detecting unresponsive providers and triggering a fallback option
+
+Request timeouts can be set on a Universal Endpoint or directly on a request to any provider.
+
+### Definitions
+
+A timeout is set in milliseconds. Additionally, the timeout is based on when the first part of the response comes back. As long as the first part of the response returns within the specified timeframe - such as when streaming a response - your gateway will wait for the response.
+
+### Configuration
+
+#### Universal Endpoint
+
+If set on a [Universal Endpoint](/ai-gateway/providers/universal/), a request timeout specifies the timeout duration for requests and triggers a fallback.
+
+For a Universal Endpoint, configure the timeout value by setting a `requestTimeout` property within the provider-specific `config` object. Each provider can have a different `requestTimeout` value for granular customization.
+
+```bash title="Provider-level config" {11-13} collapse={15-48}
+curl 'https://gateway.ai.cloudflare.com/v1/{account_id}/{gateway_id}' \
+	--header 'Content-Type: application/json' \
+	--data '[
+    {
+        "provider": "workers-ai",
+        "endpoint": "@cf/meta/llama-3.1-8b-instruct",
+        "headers": {
+            "Authorization": "Bearer {cloudflare_token}",
+            "Content-Type": "application/json"
+        },
+        "config": {
+            "requestTimeout": 1000
+        },
+        "query": {
+            "messages": [
+                {
+                    "role": "system",
+                    "content": "You are a friendly assistant"
+                },
+                {
+                    "role": "user",
+                    "content": "What is Cloudflare?"
+                }
+            ]
+        }
+    },
+    {
+        "provider": "workers-ai",
+        "endpoint": "@cf/meta/llama-3.1-8b-instruct-fast",
+        "headers": {
+            "Authorization": "Bearer {cloudflare_token}",
+            "Content-Type": "application/json"
+        },
+        "query": {
+            "messages": [
+                {
+                    "role": "system",
+                    "content": "You are a friendly assistant"
+                },
+                {
+                    "role": "user",
+                    "content": "What is Cloudflare?"
+                }
+            ]
+        },
+				"config": {
+            "requestTimeout": 3000
+        },
+    }
+]'
+```
+
+#### Direct provider
+
+If set on a [provider](/ai-gateway/providers/) request, request timeout specifies the timeout duration for a request and - if exceeded - returns an error.
+
+For a provider-specific endpoint, configure the timeout value by adding a `cf-aig-request-timeout` header.
+
+```bash title="Provider-specific endpoint example" {4}
+curl https://gateway.ai.cloudflare.com/v1/{account_id}/{gateway_id}/workers-ai/@cf/meta/llama-3.1-8b-instruct \
+ --header 'Authorization: Bearer {cf_api_token}' \
+ --header 'Content-Type: application/json' \
+ --header 'cf-aig-request-timeout: 5000'
+ --data '{"prompt": "What is Cloudflare?"}'
+```
+
+---
+
+## Request retries
+
+AI Gateway also supports automatic retries for failed requests, with a maximum of five retry attempts.
+
+This feature improves your application's resiliency, ensuring you can recover from temporary issues without manual intervention.
+
+Request timeouts can be set on a Universal Endpoint or directly on a request to any provider.
+
+### Definitions
+
+With request retries, you can adjust a combination of three properties:
+
+- Number of attempts (maximum of 5 tries)
+- How long before retrying (in milliseconds, maximum of 5 seconds)
+- Backoff method (constant, linear, or exponential)
+
+On the final retry attempt, your gateway will wait until the request completes, regardless of how long it takes.
+
+### Configuration
+
+#### Universal endpoint
+
+If set on a [Universal Endpoint](/ai-gateway/providers/universal/), a request retry will automatically retry failed requests up to five times before triggering any configured fallbacks.
+
+For a Universal Endpoint, configure the retry settings with the following properties in the provider-specific `config`:
+
+```json
+config:{
+	maxAttempts?: number;
+	retryDelay?: number;
+	backoff?: "constant" | "linear" | "exponential";
+}
+```
+
+As with the [request timeout](/ai-gateway/configuration/request-handling/#universal-endpoint), each provider can have a different retry settings for granular customization.
+
+```bash title="Provider-level config" {11-15} collapse={16-55}
+curl 'https://gateway.ai.cloudflare.com/v1/{account_id}/{gateway_id}' \
+	--header 'Content-Type: application/json' \
+	--data '[
+    {
+        "provider": "workers-ai",
+        "endpoint": "@cf/meta/llama-3.1-8b-instruct",
+        "headers": {
+            "Authorization": "Bearer {cloudflare_token}",
+            "Content-Type": "application/json"
+        },
+        "config": {
+            "maxAttempts": 2,
+						"retryDelay": 1000,
+						"backoff": "constant"
+        },
+        "query": {
+            "messages": [
+                {
+                    "role": "system",
+                    "content": "You are a friendly assistant"
+                },
+                {
+                    "role": "user",
+                    "content": "What is Cloudflare?"
+                }
+            ]
+        }
+    },
+    {
+        "provider": "workers-ai",
+        "endpoint": "@cf/meta/llama-3.1-8b-instruct-fast",
+        "headers": {
+            "Authorization": "Bearer {cloudflare_token}",
+            "Content-Type": "application/json"
+        },
+        "query": {
+            "messages": [
+                {
+                    "role": "system",
+                    "content": "You are a friendly assistant"
+                },
+                {
+                    "role": "user",
+                    "content": "What is Cloudflare?"
+                }
+            ]
+        },
+				"config": {
+            "maxAttempts": 4,
+						"retryDelay": 1000,
+						"backoff": "exponential"
+        },
+    }
+]'
+```
+
+#### Direct provider
+
+If set on a [provider](/ai-gateway/providers/) request, a request retry will automatically retry failed requests up to five times. On the final retry attempt, your gateway will wait until the request completes, regardless of how long it takes.
+
+For a provider-specific endpoint, configure the retry settings by adding different header values:
+
+- `cf-aig-max-attempts` (number)
+- `cf-aig-retry-delay` (number)
+- `cf-aig-backoff` ("constant" | "linear" | "exponential)

src/content/glossary/ai-gateway.yaml

@@ -41,6 +41,22 @@ entries:
     general_definition: |-
       Header to [bypass caching for a specific request](/ai-gateway/configuration/caching/#skip-cache-cf-aig-skip-cache).
 
+  - term: cf-aig-request-timeout
+    general_definition: |-
+      Header to trigger a fallback provider based on a [predetermined response time](/ai-gateway/configuration/fallbacks/#request-timeouts) (measured in milliseconds).
+
+  - term: cf-aig-max-attempts
+    general_definition: |-
+      Header to customize the number of max attempts for [request retries](/ai-gateway/configuration/request-handling/#request-retries) of a request.
+
+  - term: cf-aig-retry-delay
+    general_definition: |-
+      Header to customize the retry delay for [request retries](/ai-gateway/configuration/request-handling/#request-retries) of a request.
+
+  - term: cf-aig-backoff
+    general_definition: |-
+      Header to customize the backoff type for [request retries](/ai-gateway/configuration/request-handling/#request-retries) of a request.
+
   # Deprecated headers
   - term: cf-cache-ttl
     general_definition: |-