@workers-community/workers-types 4.20250313.0 → 4.20250317.0

package/index.d.ts

@@ -4341,6 +4341,7 @@ type AiModelListType = Record<string, any>;
 declare abstract class Ai<AiModelList extends AiModelListType = AiModels> {
   aiGatewayLogId: string | null;
   gateway(gatewayId: string): AiGateway;
+  autorag(autoragId: string): AutoRAG;
   run<Name extends keyof AiModelList, Options extends AiOptions>(
     model: Name,
     inputs: AiModelList[Name]["inputs"],
@@ -4360,6 +4361,7 @@ declare abstract class Ai<AiModelList extends AiModelListType = AiModels> {
     }[],
     options?: {
       gateway?: GatewayOptions;
+      extraHeaders?: object;
     },
   ): Promise<ConversionResponse[]>;
   public toMarkdown(
@@ -4369,6 +4371,7 @@ declare abstract class Ai<AiModelList extends AiModelListType = AiModels> {
     },
     options?: {
       gateway?: GatewayOptions;
+      extraHeaders?: object;
     },
   ): Promise<ConversionResponse>;
 }
@@ -4468,7 +4471,42 @@ declare abstract class AiGateway {
   run(
     data: AIGatewayUniversalRequest | AIGatewayUniversalRequest[],
   ): Promise<Response>;
-  getUrl(provider: AIGatewayProviders | string): Promise<string>; // eslint-disable-line
+  getUrl(provider?: AIGatewayProviders | string): Promise<string>; // eslint-disable-line
+}
+interface AutoRAGInternalError extends Error {}
+interface AutoRAGNotFoundError extends Error {}
+interface AutoRAGUnauthorizedError extends Error {}
+type AutoRagSearchRequest = {
+  query: string;
+  max_num_results?: number;
+  ranking_options?: {
+    ranker?: string;
+    score_threshold?: number;
+  };
+  rewrite_query?: boolean;
+};
+type AutoRagSearchResponse = {
+  object: "vector_store.search_results.page";
+  search_query: string;
+  data: {
+    file_id: string;
+    filename: string;
+    score: number;
+    attributes: Record<string, string | number | boolean | null>;
+    content: {
+      type: "text";
+      text: string;
+    }[];
+  }[];
+  has_more: boolean;
+  next_page: string | null;
+};
+type AutoRagAiSearchResponse = AutoRagSearchResponse & {
+  response: string;
+};
+declare abstract class AutoRAG {
+  search(params: AutoRagSearchRequest): Promise<AutoRagSearchResponse>;
+  aiSearch(params: AutoRagSearchRequest): Promise<AutoRagAiSearchResponse>;
 }
 interface BasicImageTransformations {
   /**

package/index.ts

@@ -4355,6 +4355,7 @@ export declare abstract class Ai<
 > {
   aiGatewayLogId: string | null;
   gateway(gatewayId: string): AiGateway;
+  autorag(autoragId: string): AutoRAG;
   run<Name extends keyof AiModelList, Options extends AiOptions>(
     model: Name,
     inputs: AiModelList[Name]["inputs"],
@@ -4374,6 +4375,7 @@ export declare abstract class Ai<
     }[],
     options?: {
       gateway?: GatewayOptions;
+      extraHeaders?: object;
     },
   ): Promise<ConversionResponse[]>;
   public toMarkdown(
@@ -4383,6 +4385,7 @@ export declare abstract class Ai<
     },
     options?: {
       gateway?: GatewayOptions;
+      extraHeaders?: object;
     },
   ): Promise<ConversionResponse>;
 }
@@ -4482,7 +4485,42 @@ export declare abstract class AiGateway {
   run(
     data: AIGatewayUniversalRequest | AIGatewayUniversalRequest[],
   ): Promise<Response>;
-  getUrl(provider: AIGatewayProviders | string): Promise<string>; // eslint-disable-line
+  getUrl(provider?: AIGatewayProviders | string): Promise<string>; // eslint-disable-line
+}
+export interface AutoRAGInternalError extends Error {}
+export interface AutoRAGNotFoundError extends Error {}
+export interface AutoRAGUnauthorizedError extends Error {}
+export type AutoRagSearchRequest = {
+  query: string;
+  max_num_results?: number;
+  ranking_options?: {
+    ranker?: string;
+    score_threshold?: number;
+  };
+  rewrite_query?: boolean;
+};
+export type AutoRagSearchResponse = {
+  object: "vector_store.search_results.page";
+  search_query: string;
+  data: {
+    file_id: string;
+    filename: string;
+    score: number;
+    attributes: Record<string, string | number | boolean | null>;
+    content: {
+      type: "text";
+      text: string;
+    }[];
+  }[];
+  has_more: boolean;
+  next_page: string | null;
+};
+export type AutoRagAiSearchResponse = AutoRagSearchResponse & {
+  response: string;
+};
+export declare abstract class AutoRAG {
+  search(params: AutoRagSearchRequest): Promise<AutoRagSearchResponse>;
+  aiSearch(params: AutoRagSearchRequest): Promise<AutoRagAiSearchResponse>;
 }
 export interface BasicImageTransformations {
   /**

package/package.json

@@ -7,7 +7,7 @@
   },
   "author": "Workers Community",
   "license": "MIT OR Apache-2.0",
-  "version": "4.20250313.0",
+  "version": "4.20250317.0",
   "exports": {
     ".": {
       "types": "./index.d.ts",