Add returnRawResponse to AI binding options

cloudflare · Jan 22, 2025 · 3a48e2a · 3a48e2a
1 parent deec614
commit 3a48e2a
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/cloudflare/internal/ai-api.ts b/src/cloudflare/internal/ai-api.ts
@@ -23,7 +23,8 @@ export type SessionOptions = {
 
 export type AiOptions = {
   gateway?: GatewayOptions;
-
+  /** If true it will return a Response object */
+  returnRawResponse?: boolean;
   prefix?: string;
   extraHeaders?: object;
   /*
@@ -142,6 +143,10 @@ export class Ai {
     this.aiGatewayLogId = res.headers.get('cf-aig-log-id');
     this.lastRequestHttpStatusCode = res.status;
 
+    if (this.options.returnRawResponse) {
+      return res;
+    }
+
     if (!res.ok) {
       throw await this._parseError(res);
     }