@azure/ai-projects 1.0.0-beta.2 → 1.0.0-beta.4

package/dist/commonjs/agents/utils.d.ts

@@ -1,8 +1,8 @@
-import type { AzureAISearchToolDefinition, CodeInterpreterToolDefinition, FileSearchToolDefinition, FileSearchToolDefinitionDetails, FunctionDefinition, FunctionToolDefinition, RequiredActionOutput, RequiredToolCallOutput, ToolDefinition, ToolDefinitionOutputParent, ToolResources, VectorStoreConfigurations, VectorStoreDataSource } from "./inputOutputs.js";
+import type { AzureAISearchToolDefinition, AzureFunctionToolDefinition, CodeInterpreterToolDefinition, FileSearchToolDefinition, FileSearchToolDefinitionDetails, FunctionDefinition, FunctionToolDefinition, OpenApiFunctionDefinition, OpenApiToolDefinition, RequiredActionOutput, RequiredToolCallOutput, ToolDefinition, ToolDefinitionOutputParent, ToolResources, VectorStoreConfigurations, VectorStoreDataSource, AzureFunctionStorageQueue, AzureFunctionDefinition } from "./inputOutputs.js";
 /**
  * Determines if the given output is of the specified type.
  *
- * @template T - The type to check against, which extends one of the possible output parent types.
+ * @typeParam T - The type to check against, which extends one of the possible output parent types.
  * @param output - The action to check, which can be of type `RequiredActionOutput`, `RequiredToolCallOutput`, or `ToolDefinitionOutputParent`.
  * @param type - The type to check the action against.
  * @returns A boolean indicating whether the action is of the specified type.
@@ -15,9 +15,11 @@ export declare enum connectionToolType {
     /** Bing grounding search tool */
     BingGrounding = "bing_grounding",
     /** Microsoft Fabric tool */
-    MicrosoftFabric = "microsoft_fabric",
+    MicrosoftFabric = "fabric_dataagent",
     /** Sharepoint tool */
-    SharepointGrounding = "sharepoint_grounding"
+    SharepointGrounding = "sharepoint_grounding",
+    /** Azure Function tool */
+    AzureFunction = "azure_function"
 }
 /**
  * Utility class for creating various tools.
@@ -58,6 +60,19 @@ export declare class ToolUtility {
         definition: CodeInterpreterToolDefinition;
         resources: ToolResources;
     };
+    /**
+     * Creates an Azure Function tool
+     * @param name - The name of the Azure Function.
+     * @param description - The description of the Azure Function.
+     * @param parameters - The parameters of the Azure Function.
+     * @param inputQueue - The input queue configuration.
+     * @param outputQueue - The output queue configuration.
+     * @returns An object containing the definition and resources for the Azure Function tool.
+     */
+    static createAzureFunctionTool(name: string, description: string, parameters: unknown, inputQueue: AzureFunctionStorageQueue, outputQueue: AzureFunctionStorageQueue, definitionDetails: AzureFunctionDefinition): {
+        definition: AzureFunctionToolDefinition;
+        resources: ToolResources;
+    };
     /**
      * Creates an Azure AI search tool
      *
@@ -70,6 +85,15 @@ export declare class ToolUtility {
         definition: AzureAISearchToolDefinition;
         resources: ToolResources;
     };
+    /**
+     * Creates a Microsoft Fabric tool
+     *
+     * @param connectionIds - A list of the IDs of the Fabric connections to use.
+     * @returns An object containing the definition for the Microsoft Fabric tool
+     */
+    static createFabricTool(connectionId: string): {
+        definition: ToolDefinition;
+    };
     /**
      * Creates a function tool
      *
@@ -80,6 +104,16 @@ export declare class ToolUtility {
     static createFunctionTool(functionDefinition: FunctionDefinition): {
         definition: FunctionToolDefinition;
     };
+    /**
+     * Creates an OpenApi tool
+     *
+     * @param openApiFunctionDefinition - The OpenApi function definition to use.
+     *
+     * @returns An object containing the definition for the OpenApi tool.
+     */
+    static createOpenApiTool(openApiFunctionDefinition: OpenApiFunctionDefinition): {
+        definition: OpenApiToolDefinition;
+    };
 }
 /**
  * Represents a set of tools with their definitions and resources.
@@ -137,5 +171,39 @@ export declare class ToolSet {
         definition: AzureAISearchToolDefinition;
         resources: ToolResources;
     };
+    /**
+     * Adds an OpenApi tool to the tool set.
+     *
+     * @param openApiFunctionDefinition - The OpenApi function definition to use.
+     *
+     * @returns An object containing the definition for the OpenApi tool
+     */
+    addOpenApiTool(openApiFunctionDefinition: OpenApiFunctionDefinition): {
+        definition: OpenApiToolDefinition;
+    };
+    /**
+     * Adds an Azure Function tool to the tool set.
+     *
+     * @param name - The name of the Azure Function.
+     * @param description - The description of the Azure Function.
+     * @param parameters - The parameters of the Azure Function.
+     * @param inputQueue - The input queue configuration.
+     * @param outputQueue - The output queue configuration.
+     *
+     * @returns An object containing the definition and resources for the Azure Function tool.
+     */
+    addAzureFunctionTool(name: string, description: string, parameters: unknown, inputQueue: AzureFunctionStorageQueue, outputQueue: AzureFunctionStorageQueue, definitionDetails: AzureFunctionDefinition): {
+        definition: AzureFunctionToolDefinition;
+        resources: ToolResources;
+    };
+    /**
+     * Adds a Microsoft Fabric tool to the tool set.
+     *
+     * @param connectionId - The ID of the Fabric connection to use.
+     * @returns An object containing the definition for the Microsoft Fabric tool
+     */
+    addFabricTool(connectionId: string): {
+        definition: ToolDefinition;
+    };
 }
 //# sourceMappingURL=utils.d.ts.map

package/dist/commonjs/agents/utils.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../src/agents/utils.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EACV,2BAA2B,EAC3B,6BAA6B,EAC7B,wBAAwB,EACxB,+BAA+B,EAC/B,kBAAkB,EAClB,sBAAsB,EACtB,oBAAoB,EACpB,sBAAsB,EACtB,cAAc,EACd,0BAA0B,EAC1B,aAAa,EACb,yBAAyB,EACzB,qBAAqB,EACtB,MAAM,mBAAmB,CAAC;AAE3B;;;;;;;GAOG;AACH,wBAAgB,cAAc,CAAC,CAAC,SAAS;IAAE,IAAI,EAAE,MAAM,CAAA;CAAE,EACvD,MAAM,EAAE,oBAAoB,GAAG,sBAAsB,GAAG,0BAA0B,EAClF,IAAI,EAAE,MAAM,GACX,MAAM,IAAI,CAAC,CAEb;AAED,2DAA2D;AAC3D,oBAAY,kBAAkB;IAC5B,iCAAiC;IACjC,aAAa,mBAAmB;IAChC,4BAA4B;IAC5B,eAAe,qBAAqB;IACpC,sBAAsB;IACtB,mBAAmB,yBAAyB;CAC7C;AAQD;;GAEG;AACH,qBAAa,WAAW;IACtB;;;;;;OAMG;IACH,MAAM,CAAC,oBAAoB,CACzB,QAAQ,EAAE,kBAAkB,EAC5B,aAAa,EAAE,MAAM,EAAE,GACtB;QAAE,UAAU,EAAE,cAAc,CAAA;KAAE;IAWjC;;;;;;;;OAQG;IACH,MAAM,CAAC,oBAAoB,CACzB,cAAc,CAAC,EAAE,MAAM,EAAE,EACzB,YAAY,CAAC,EAAE,KAAK,CAAC,yBAAyB,CAAC,EAC/C,iBAAiB,CAAC,EAAE,+BAA+B,GAClD;QAAE,UAAU,EAAE,wBAAwB,CAAC;QAAC,SAAS,EAAE,aAAa,CAAA;KAAE;IAOrE;;;;;;;OAOG;IACH,MAAM,CAAC,yBAAyB,CAC9B,OAAO,CAAC,EAAE,MAAM,EAAE,EAClB,WAAW,CAAC,EAAE,KAAK,CAAC,qBAAqB,CAAC,GACzC;QAAE,UAAU,EAAE,6BAA6B,CAAC;QAAC,SAAS,EAAE,aAAa,CAAA;KAAE;IAW1E;;;;;;;OAOG;IACH,MAAM,CAAC,uBAAuB,CAC5B,iBAAiB,EAAE,MAAM,EACzB,SAAS,EAAE,MAAM,GAChB;QAAE,UAAU,EAAE,2BAA2B,CAAC;QAAC,SAAS,EAAE,aAAa,CAAA;KAAE;IAWxE;;;;;;OAMG;IACH,MAAM,CAAC,kBAAkB,CAAC,kBAAkB,EAAE,kBAAkB,GAAG;QACjE,UAAU,EAAE,sBAAsB,CAAC;KACpC;CAQF;AAED;;GAEG;AACH,qBAAa,OAAO;IAClB,uEAAuE;IACvE,eAAe,EAAE,cAAc,EAAE,CAAM;IAEvC,2EAA2E;IAC3E,aAAa,EAAE,aAAa,CAAM;IAElC;;;;;;;OAOG;IACH,iBAAiB,CACf,QAAQ,EAAE,kBAAkB,EAC5B,aAAa,EAAE,MAAM,EAAE,GACtB;QAAE,UAAU,EAAE,cAAc,CAAA;KAAE;IAMjC;;;;;;;;OAQG;IACH,iBAAiB,CACf,cAAc,CAAC,EAAE,MAAM,EAAE,EACzB,YAAY,CAAC,EAAE,KAAK,CAAC,yBAAyB,CAAC,EAC/C,iBAAiB,CAAC,EAAE,+BAA+B,GAClD;QAAE,UAAU,EAAE,wBAAwB,CAAC;QAAC,SAAS,EAAE,aAAa,CAAA;KAAE;IAOrE;;;;;;;OAOG;IACH,sBAAsB,CACpB,OAAO,CAAC,EAAE,MAAM,EAAE,EAClB,WAAW,CAAC,EAAE,KAAK,CAAC,qBAAqB,CAAC,GACzC;QAAE,UAAU,EAAE,6BAA6B,CAAC;QAAC,SAAS,EAAE,aAAa,CAAA;KAAE;IAO1E;;;;;;;OAOG;IACH,oBAAoB,CAClB,iBAAiB,EAAE,MAAM,EACzB,SAAS,EAAE,MAAM,GAChB;QAAE,UAAU,EAAE,2BAA2B,CAAC;QAAC,SAAS,EAAE,aAAa,CAAA;KAAE;CAMzE"}
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../src/agents/utils.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EACV,2BAA2B,EAC3B,2BAA2B,EAC3B,6BAA6B,EAC7B,wBAAwB,EACxB,+BAA+B,EAC/B,kBAAkB,EAClB,sBAAsB,EACtB,yBAAyB,EACzB,qBAAqB,EACrB,oBAAoB,EACpB,sBAAsB,EACtB,cAAc,EACd,0BAA0B,EAC1B,aAAa,EACb,yBAAyB,EACzB,qBAAqB,EACrB,yBAAyB,EACzB,uBAAuB,EACxB,MAAM,mBAAmB,CAAC;AAE3B;;;;;;;GAOG;AACH,wBAAgB,cAAc,CAAC,CAAC,SAAS;IAAE,IAAI,EAAE,MAAM,CAAA;CAAE,EACvD,MAAM,EAAE,oBAAoB,GAAG,sBAAsB,GAAG,0BAA0B,EAClF,IAAI,EAAE,MAAM,GACX,MAAM,IAAI,CAAC,CAEb;AAED,2DAA2D;AAC3D,oBAAY,kBAAkB;IAC5B,iCAAiC;IACjC,aAAa,mBAAmB;IAChC,4BAA4B;IAC5B,eAAe,qBAAqB;IACpC,sBAAsB;IACtB,mBAAmB,yBAAyB;IAC5C,0BAA0B;IAC1B,aAAa,mBAAmB;CACjC;AASD;;GAEG;AACH,qBAAa,WAAW;IACtB;;;;;;OAMG;IACH,MAAM,CAAC,oBAAoB,CACzB,QAAQ,EAAE,kBAAkB,EAC5B,aAAa,EAAE,MAAM,EAAE,GACtB;QAAE,UAAU,EAAE,cAAc,CAAA;KAAE;IAWjC;;;;;;;;OAQG;IACH,MAAM,CAAC,oBAAoB,CACzB,cAAc,CAAC,EAAE,MAAM,EAAE,EACzB,YAAY,CAAC,EAAE,KAAK,CAAC,yBAAyB,CAAC,EAC/C,iBAAiB,CAAC,EAAE,+BAA+B,GAClD;QAAE,UAAU,EAAE,wBAAwB,CAAC;QAAC,SAAS,EAAE,aAAa,CAAA;KAAE;IAOrE;;;;;;;OAOG;IACH,MAAM,CAAC,yBAAyB,CAC9B,OAAO,CAAC,EAAE,MAAM,EAAE,EAClB,WAAW,CAAC,EAAE,KAAK,CAAC,qBAAqB,CAAC,GACzC;QAAE,UAAU,EAAE,6BAA6B,CAAC;QAAC,SAAS,EAAE,aAAa,CAAA;KAAE;IAW1E;;;;;;;;OAQG;IACH,MAAM,CAAC,uBAAuB,CAC5B,IAAI,EAAE,MAAM,EACZ,WAAW,EAAE,MAAM,EACnB,UAAU,EAAE,OAAO,EACnB,UAAU,EAAE,yBAAyB,EACrC,WAAW,EAAE,yBAAyB,EACtC,iBAAiB,EAAE,uBAAuB,GACzC;QAAE,UAAU,EAAE,2BAA2B,CAAC;QAAC,SAAS,EAAE,aAAa,CAAA;KAAE;IAcxE;;;;;;;OAOG;IACH,MAAM,CAAC,uBAAuB,CAC5B,iBAAiB,EAAE,MAAM,EACzB,SAAS,EAAE,MAAM,GAChB;QAAE,UAAU,EAAE,2BAA2B,CAAC;QAAC,SAAS,EAAE,aAAa,CAAA;KAAE;IAWxE;;;;;OAKG;IACH,MAAM,CAAC,gBAAgB,CAAC,YAAY,EAAE,MAAM,GAAG;QAAE,UAAU,EAAE,cAAc,CAAA;KAAE;IAW7E;;;;;;OAMG;IACH,MAAM,CAAC,kBAAkB,CAAC,kBAAkB,EAAE,kBAAkB,GAAG;QACjE,UAAU,EAAE,sBAAsB,CAAC;KACpC;IASD;;;;;;OAMG;IACH,MAAM,CAAC,iBAAiB,CAAC,yBAAyB,EAAE,yBAAyB,GAAG;QAC9E,UAAU,EAAE,qBAAqB,CAAC;KACnC;CAQF;AAED;;GAEG;AACH,qBAAa,OAAO;IAClB,uEAAuE;IACvE,eAAe,EAAE,cAAc,EAAE,CAAM;IAEvC,2EAA2E;IAC3E,aAAa,EAAE,aAAa,CAAM;IAElC;;;;;;;OAOG;IACH,iBAAiB,CACf,QAAQ,EAAE,kBAAkB,EAC5B,aAAa,EAAE,MAAM,EAAE,GACtB;QAAE,UAAU,EAAE,cAAc,CAAA;KAAE;IAMjC;;;;;;;;OAQG;IACH,iBAAiB,CACf,cAAc,CAAC,EAAE,MAAM,EAAE,EACzB,YAAY,CAAC,EAAE,KAAK,CAAC,yBAAyB,CAAC,EAC/C,iBAAiB,CAAC,EAAE,+BAA+B,GAClD;QAAE,UAAU,EAAE,wBAAwB,CAAC;QAAC,SAAS,EAAE,aAAa,CAAA;KAAE;IAOrE;;;;;;;OAOG;IACH,sBAAsB,CACpB,OAAO,CAAC,EAAE,MAAM,EAAE,EAClB,WAAW,CAAC,EAAE,KAAK,CAAC,qBAAqB,CAAC,GACzC;QAAE,UAAU,EAAE,6BAA6B,CAAC;QAAC,SAAS,EAAE,aAAa,CAAA;KAAE;IAO1E;;;;;;;OAOG;IACH,oBAAoB,CAClB,iBAAiB,EAAE,MAAM,EACzB,SAAS,EAAE,MAAM,GAChB;QAAE,UAAU,EAAE,2BAA2B,CAAC;QAAC,SAAS,EAAE,aAAa,CAAA;KAAE;IAOxE;;;;;;OAMG;IACH,cAAc,CAAC,yBAAyB,EAAE,yBAAyB,GAAG;QACpE,UAAU,EAAE,qBAAqB,CAAC;KACnC;IAMD;;;;;;;;;;OAUG;IACH,oBAAoB,CAClB,IAAI,EAAE,MAAM,EACZ,WAAW,EAAE,MAAM,EACnB,UAAU,EAAE,OAAO,EACnB,UAAU,EAAE,yBAAyB,EACrC,WAAW,EAAE,yBAAyB,EACtC,iBAAiB,EAAE,uBAAuB,GACzC;QAAE,UAAU,EAAE,2BAA2B,CAAC;QAAC,SAAS,EAAE,aAAa,CAAA;KAAE;IAaxE;;;;;OAKG;IACH,aAAa,CAAC,YAAY,EAAE,MAAM,GAAG;QAAE,UAAU,EAAE,cAAc,CAAA;KAAE;CAKpE"}

package/dist/commonjs/agents/utils.js

@@ -7,7 +7,7 @@ exports.isOutputOfType = isOutputOfType;
 /**
  * Determines if the given output is of the specified type.
  *
- * @template T - The type to check against, which extends one of the possible output parent types.
+ * @typeParam T - The type to check against, which extends one of the possible output parent types.
  * @param output - The action to check, which can be of type `RequiredActionOutput`, `RequiredToolCallOutput`, or `ToolDefinitionOutputParent`.
  * @param type - The type to check the action against.
  * @returns A boolean indicating whether the action is of the specified type.
@@ -21,14 +21,17 @@ var connectionToolType;
     /** Bing grounding search tool */
     connectionToolType["BingGrounding"] = "bing_grounding";
     /** Microsoft Fabric tool */
-    connectionToolType["MicrosoftFabric"] = "microsoft_fabric";
+    connectionToolType["MicrosoftFabric"] = "fabric_dataagent";
     /** Sharepoint tool */
     connectionToolType["SharepointGrounding"] = "sharepoint_grounding";
+    /** Azure Function tool */
+    connectionToolType["AzureFunction"] = "azure_function";
 })(connectionToolType || (exports.connectionToolType = connectionToolType = {}));
 const toolMap = {
     bing_grounding: "bingGrounding",
-    microsoft_fabric: "microsoftFabric",
+    fabric_dataagent: "fabric_dataagent",
     sharepoint_grounding: "sharepointGrounding",
+    azure_function: "azureFunction",
 };
 /**
  * Utility class for creating various tools.
@@ -83,6 +86,29 @@ class ToolUtility {
             resources: { codeInterpreter: { fileIds: fileIds, dataSources: dataSources } },
         };
     }
+    /**
+     * Creates an Azure Function tool
+     * @param name - The name of the Azure Function.
+     * @param description - The description of the Azure Function.
+     * @param parameters - The parameters of the Azure Function.
+     * @param inputQueue - The input queue configuration.
+     * @param outputQueue - The output queue configuration.
+     * @returns An object containing the definition and resources for the Azure Function tool.
+     */
+    static createAzureFunctionTool(name, description, parameters, inputQueue, outputQueue, definitionDetails) {
+        return {
+            definition: { type: "azure_function", azureFunction: definitionDetails },
+            resources: {
+                azureFunction: {
+                    name: name,
+                    description: description,
+                    parameters: parameters,
+                    inputQueue: inputQueue,
+                    outputQueue: outputQueue,
+                },
+            },
+        };
+    }
     /**
      * Creates an Azure AI search tool
      *
@@ -101,6 +127,22 @@ class ToolUtility {
             },
         };
     }
+    /**
+     * Creates a Microsoft Fabric tool
+     *
+     * @param connectionIds - A list of the IDs of the Fabric connections to use.
+     * @returns An object containing the definition for the Microsoft Fabric tool
+     */
+    static createFabricTool(connectionId) {
+        return {
+            definition: {
+                type: "fabric_dataagent",
+                fabricDataAgent: {
+                    connections: [{ connectionId: connectionId }],
+                },
+            },
+        };
+    }
     /**
      * Creates a function tool
      *
@@ -116,6 +158,21 @@ class ToolUtility {
             },
         };
     }
+    /**
+     * Creates an OpenApi tool
+     *
+     * @param openApiFunctionDefinition - The OpenApi function definition to use.
+     *
+     * @returns An object containing the definition for the OpenApi tool.
+     */
+    static createOpenApiTool(openApiFunctionDefinition) {
+        return {
+            definition: {
+                type: "openapi",
+                openapi: openApiFunctionDefinition,
+            },
+        };
+    }
 }
 exports.ToolUtility = ToolUtility;
 /**
@@ -184,6 +241,46 @@ class ToolSet {
         this.toolResources = Object.assign(Object.assign({}, this.toolResources), tool.resources);
         return tool;
     }
+    /**
+     * Adds an OpenApi tool to the tool set.
+     *
+     * @param openApiFunctionDefinition - The OpenApi function definition to use.
+     *
+     * @returns An object containing the definition for the OpenApi tool
+     */
+    addOpenApiTool(openApiFunctionDefinition) {
+        const tool = ToolUtility.createOpenApiTool(openApiFunctionDefinition);
+        this.toolDefinitions.push(tool.definition);
+        return tool;
+    }
+    /**
+     * Adds an Azure Function tool to the tool set.
+     *
+     * @param name - The name of the Azure Function.
+     * @param description - The description of the Azure Function.
+     * @param parameters - The parameters of the Azure Function.
+     * @param inputQueue - The input queue configuration.
+     * @param outputQueue - The output queue configuration.
+     *
+     * @returns An object containing the definition and resources for the Azure Function tool.
+     */
+    addAzureFunctionTool(name, description, parameters, inputQueue, outputQueue, definitionDetails) {
+        const tool = ToolUtility.createAzureFunctionTool(name, description, parameters, inputQueue, outputQueue, definitionDetails);
+        this.toolDefinitions.push(tool.definition);
+        this.toolResources = Object.assign(Object.assign({}, this.toolResources), tool.resources);
+        return tool;
+    }
+    /**
+     * Adds a Microsoft Fabric tool to the tool set.
+     *
+     * @param connectionId - The ID of the Fabric connection to use.
+     * @returns An object containing the definition for the Microsoft Fabric tool
+     */
+    addFabricTool(connectionId) {
+        const tool = ToolUtility.createFabricTool(connectionId);
+        this.toolDefinitions.push(tool.definition);
+        return tool;
+    }
 }
 exports.ToolSet = ToolSet;
 //# sourceMappingURL=utils.js.map

package/dist/commonjs/agents/utils.js.map

@@ -1 +1 @@
-{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../../src/agents/utils.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;;AA0BlC,wCAKC;AAbD;;;;;;;GAOG;AACH,SAAgB,cAAc,CAC5B,MAAkF,EAClF,IAAY;IAEZ,OAAO,MAAM,CAAC,IAAI,KAAK,IAAI,CAAC;AAC9B,CAAC;AAED,2DAA2D;AAC3D,IAAY,kBAOX;AAPD,WAAY,kBAAkB;IAC5B,iCAAiC;IACjC,sDAAgC,CAAA;IAChC,4BAA4B;IAC5B,0DAAoC,CAAA;IACpC,sBAAsB;IACtB,kEAA4C,CAAA;AAC9C,CAAC,EAPW,kBAAkB,kCAAlB,kBAAkB,QAO7B;AAED,MAAM,OAAO,GAAG;IACd,cAAc,EAAE,eAAe;IAC/B,gBAAgB,EAAE,iBAAiB;IACnC,oBAAoB,EAAE,qBAAqB;CAC5C,CAAC;AAEF;;GAEG;AACH,MAAa,WAAW;IACtB;;;;;;OAMG;IACH,MAAM,CAAC,oBAAoB,CACzB,QAA4B,EAC5B,aAAuB;QAEvB,OAAO;YACL,UAAU,EAAE;gBACV,IAAI,EAAE,QAAQ;gBACd,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,EAAE;oBACnB,WAAW,EAAE,aAAa,CAAC,GAAG,CAAC,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC,EAAE,YAAY,EAAE,YAAY,EAAE,CAAC,CAAC;iBACnF;aACF;SACF,CAAC;IACJ,CAAC;IAED;;;;;;;;OAQG;IACH,MAAM,CAAC,oBAAoB,CACzB,cAAyB,EACzB,YAA+C,EAC/C,iBAAmD;QAEnD,OAAO;YACL,UAAU,EAAE,EAAE,IAAI,EAAE,aAAa,EAAE,UAAU,EAAE,iBAAiB,EAAE;YAClE,SAAS,EAAE,EAAE,UAAU,EAAE,EAAE,cAAc,EAAE,cAAc,EAAE,YAAY,EAAE,YAAY,EAAE,EAAE;SAC1F,CAAC;IACJ,CAAC;IAED;;;;;;;OAOG;IACH,MAAM,CAAC,yBAAyB,CAC9B,OAAkB,EAClB,WAA0C;QAE1C,IAAI,OAAO,IAAI,WAAW,EAAE,CAAC;YAC3B,MAAM,IAAI,KAAK,CAAC,6CAA6C,CAAC,CAAC;QACjE,CAAC;QAED,OAAO;YACL,UAAU,EAAE,EAAE,IAAI,EAAE,kBAAkB,EAAE;YACxC,SAAS,EAAE,EAAE,eAAe,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,WAAW,EAAE,WAAW,EAAE,EAAE;SAC/E,CAAC;IACJ,CAAC;IAED;;;;;;;OAOG;IACH,MAAM,CAAC,uBAAuB,CAC5B,iBAAyB,EACzB,SAAiB;QAEjB,OAAO;YACL,UAAU,EAAE,EAAE,IAAI,EAAE,iBAAiB,EAAE;YACvC,SAAS,EAAE;gBACT,aAAa,EAAE;oBACb,OAAO,EAAE,CAAC,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC;iBAC1E;aACF;SACF,CAAC;IACJ,CAAC;IAED;;;;;;OAMG;IACH,MAAM,CAAC,kBAAkB,CAAC,kBAAsC;QAG9D,OAAO;YACL,UAAU,EAAE;gBACV,IAAI,EAAE,UAAU;gBAChB,QAAQ,EAAE,kBAAkB;aAC7B;SACF,CAAC;IACJ,CAAC;CACF;AAvGD,kCAuGC;AAED;;GAEG;AACH,MAAa,OAAO;IAApB;QACE,uEAAuE;QACvE,oBAAe,GAAqB,EAAE,CAAC;QAEvC,2EAA2E;QAC3E,kBAAa,GAAkB,EAAE,CAAC;IA0EpC,CAAC;IAxEC;;;;;;;OAOG;IACH,iBAAiB,CACf,QAA4B,EAC5B,aAAuB;QAEvB,MAAM,IAAI,GAAG,WAAW,CAAC,oBAAoB,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC;QACvE,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC3C,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;;;;;;;OAQG;IACH,iBAAiB,CACf,cAAyB,EACzB,YAA+C,EAC/C,iBAAmD;QAEnD,MAAM,IAAI,GAAG,WAAW,CAAC,oBAAoB,CAAC,cAAc,EAAE,YAAY,EAAE,iBAAiB,CAAC,CAAC;QAC/F,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC3C,IAAI,CAAC,aAAa,mCAAQ,IAAI,CAAC,aAAa,GAAK,IAAI,CAAC,SAAS,CAAE,CAAC;QAClE,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;;;;;;OAOG;IACH,sBAAsB,CACpB,OAAkB,EAClB,WAA0C;QAE1C,MAAM,IAAI,GAAG,WAAW,CAAC,yBAAyB,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;QACzE,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC3C,IAAI,CAAC,aAAa,mCAAQ,IAAI,CAAC,aAAa,GAAK,IAAI,CAAC,SAAS,CAAE,CAAC;QAClE,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;;;;;;OAOG;IACH,oBAAoB,CAClB,iBAAyB,EACzB,SAAiB;QAEjB,MAAM,IAAI,GAAG,WAAW,CAAC,uBAAuB,CAAC,iBAAiB,EAAE,SAAS,CAAC,CAAC;QAC/E,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC3C,IAAI,CAAC,aAAa,mCAAQ,IAAI,CAAC,aAAa,GAAK,IAAI,CAAC,SAAS,CAAE,CAAC;QAClE,OAAO,IAAI,CAAC;IACd,CAAC;CACF;AA/ED,0BA+EC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport type {\n  AzureAISearchToolDefinition,\n  CodeInterpreterToolDefinition,\n  FileSearchToolDefinition,\n  FileSearchToolDefinitionDetails,\n  FunctionDefinition,\n  FunctionToolDefinition,\n  RequiredActionOutput,\n  RequiredToolCallOutput,\n  ToolDefinition,\n  ToolDefinitionOutputParent,\n  ToolResources,\n  VectorStoreConfigurations,\n  VectorStoreDataSource,\n} from \"./inputOutputs.js\";\n\n/**\n * Determines if the given output is of the specified type.\n *\n * @template T - The type to check against, which extends one of the possible output parent types.\n * @param output - The action to check, which can be of type `RequiredActionOutput`, `RequiredToolCallOutput`, or `ToolDefinitionOutputParent`.\n * @param type - The type to check the action against.\n * @returns A boolean indicating whether the action is of the specified type.\n */\nexport function isOutputOfType<T extends { type: string }>(\n  output: RequiredActionOutput | RequiredToolCallOutput | ToolDefinitionOutputParent,\n  type: string,\n): output is T {\n  return output.type === type;\n}\n\n/** Types of connection tools used to configure an agent */\nexport enum connectionToolType {\n  /** Bing grounding search tool */\n  BingGrounding = \"bing_grounding\",\n  /** Microsoft Fabric tool */\n  MicrosoftFabric = \"microsoft_fabric\",\n  /** Sharepoint tool */\n  SharepointGrounding = \"sharepoint_grounding\",\n}\n\nconst toolMap = {\n  bing_grounding: \"bingGrounding\",\n  microsoft_fabric: \"microsoftFabric\",\n  sharepoint_grounding: \"sharepointGrounding\",\n};\n\n/**\n * Utility class for creating various tools.\n */\nexport class ToolUtility {\n  /**\n   * Creates a connection tool\n   *\n   * @param toolType - The type of the connection tool.\n   * @param connectionIds - A list of the IDs of the connections to use.\n   * @returns An object containing the definition for the connection tool\n   */\n  static createConnectionTool(\n    toolType: connectionToolType,\n    connectionIds: string[],\n  ): { definition: ToolDefinition } {\n    return {\n      definition: {\n        type: toolType,\n        [toolMap[toolType]]: {\n          connections: connectionIds.map((connectionId) => ({ connectionId: connectionId })),\n        },\n      },\n    };\n  }\n\n  /**\n   * Creates a file search tool\n   *\n   * @param vectorStoreIds - The ID of the vector store attached to this agent. There can be a maximum of 1 vector store attached to the agent.\n   * @param vectorStores - The list of vector store configuration objects from Azure. This list is limited to one element. The only element of this list contains the list of azure asset IDs used by the search tool.\n   * @param definitionDetails - The input definition information for a file search tool as used to configure an agent.\n   *\n   * @returns An object containing the definition and resources for the file search tool\n   */\n  static createFileSearchTool(\n    vectorStoreIds?: string[],\n    vectorStores?: Array<VectorStoreConfigurations>,\n    definitionDetails?: FileSearchToolDefinitionDetails,\n  ): { definition: FileSearchToolDefinition; resources: ToolResources } {\n    return {\n      definition: { type: \"file_search\", fileSearch: definitionDetails },\n      resources: { fileSearch: { vectorStoreIds: vectorStoreIds, vectorStores: vectorStores } },\n    };\n  }\n\n  /**\n   * Creates a code interpreter tool\n   *\n   * @param fileIds - A list of file IDs made available to the `code_interpreter` tool. There can be a maximum of 20 files associated with the tool.\n   * @param dataSources - The data sources to be used. This option is mutually exclusive with fileIds.\n   *\n   * @returns An object containing the definition and resources for the code interpreter tool.\n   */\n  static createCodeInterpreterTool(\n    fileIds?: string[],\n    dataSources?: Array<VectorStoreDataSource>,\n  ): { definition: CodeInterpreterToolDefinition; resources: ToolResources } {\n    if (fileIds && dataSources) {\n      throw new Error(\"Cannot specify both fileIds and dataSources\");\n    }\n\n    return {\n      definition: { type: \"code_interpreter\" },\n      resources: { codeInterpreter: { fileIds: fileIds, dataSources: dataSources } },\n    };\n  }\n\n  /**\n   * Creates an Azure AI search tool\n   *\n   * @param indexConnectionId - The connection ID of the Azure AI search index.\n   * @param indexName - The name of the Azure AI search index.\n   *\n   * @returns An object containing the definition and resources for the Azure AI search tool.\n   */\n  static createAzureAISearchTool(\n    indexConnectionId: string,\n    indexName: string,\n  ): { definition: AzureAISearchToolDefinition; resources: ToolResources } {\n    return {\n      definition: { type: \"azure_ai_search\" },\n      resources: {\n        azureAISearch: {\n          indexes: [{ indexConnectionId: indexConnectionId, indexName: indexName }],\n        },\n      },\n    };\n  }\n\n  /**\n   * Creates a function tool\n   *\n   * @param functionDefinition - The function definition to use.\n   *\n   * @returns An object containing the definition for the function tool.\n   */\n  static createFunctionTool(functionDefinition: FunctionDefinition): {\n    definition: FunctionToolDefinition;\n  } {\n    return {\n      definition: {\n        type: \"function\",\n        function: functionDefinition,\n      },\n    };\n  }\n}\n\n/**\n * Represents a set of tools with their definitions and resources.\n */\nexport class ToolSet {\n  /** A list of tool definitions that have been added to the tool set. */\n  toolDefinitions: ToolDefinition[] = [];\n\n  /** A collection of resources associated with the tools in the tool set. */\n  toolResources: ToolResources = {};\n\n  /**\n   * Adds a connection tool to the tool set.\n   *\n   * @param toolType - The type of the connection tool.\n   * @param connectionIds - A list of the IDs of the connections to use.\n   *\n   * @returns An object containing the definition for the connection tool\n   */\n  addConnectionTool(\n    toolType: connectionToolType,\n    connectionIds: string[],\n  ): { definition: ToolDefinition } {\n    const tool = ToolUtility.createConnectionTool(toolType, connectionIds);\n    this.toolDefinitions.push(tool.definition);\n    return tool;\n  }\n\n  /**\n   * Adds a file search tool to the tool set.\n   *\n   * @param vectorStoreIds - The ID of the vector store attached to this agent. There can be a maximum of 1 vector store attached to the agent.\n   * @param vectorStores -  The list of vector store configuration objects from Azure. This list is limited to one element. The only element of this list contains the list of azure asset IDs used by the search tool.\n   * @param definitionDetails - The input definition information for a file search tool as used to configure an agent.\n   *\n   * @returns An object containing the definition and resources for the file search tool\n   */\n  addFileSearchTool(\n    vectorStoreIds?: string[],\n    vectorStores?: Array<VectorStoreConfigurations>,\n    definitionDetails?: FileSearchToolDefinitionDetails,\n  ): { definition: FileSearchToolDefinition; resources: ToolResources } {\n    const tool = ToolUtility.createFileSearchTool(vectorStoreIds, vectorStores, definitionDetails);\n    this.toolDefinitions.push(tool.definition);\n    this.toolResources = { ...this.toolResources, ...tool.resources };\n    return tool;\n  }\n\n  /**\n   * Adds a code interpreter tool to the tool set.\n   *\n   * @param fileIds - A list of file IDs made available to the `code_interpreter` tool. There can be a maximum of 20 files associated with the tool.\n   * @param dataSources - The data sources to be used. This option is mutually exclusive with fileIds.\n   *\n   * @returns An object containing the definition and resources for the code interpreter tool\n   */\n  addCodeInterpreterTool(\n    fileIds?: string[],\n    dataSources?: Array<VectorStoreDataSource>,\n  ): { definition: CodeInterpreterToolDefinition; resources: ToolResources } {\n    const tool = ToolUtility.createCodeInterpreterTool(fileIds, dataSources);\n    this.toolDefinitions.push(tool.definition);\n    this.toolResources = { ...this.toolResources, ...tool.resources };\n    return tool;\n  }\n\n  /**\n   * Adds an Azure AI search tool to the tool set.\n   *\n   * @param indexConnectionId - The connection ID of the Azure AI search index.\n   * @param indexName - The name of the Azure AI search index.\n   *\n   * @returns An object containing the definition and resources for the Azure AI search tool\n   */\n  addAzureAISearchTool(\n    indexConnectionId: string,\n    indexName: string,\n  ): { definition: AzureAISearchToolDefinition; resources: ToolResources } {\n    const tool = ToolUtility.createAzureAISearchTool(indexConnectionId, indexName);\n    this.toolDefinitions.push(tool.definition);\n    this.toolResources = { ...this.toolResources, ...tool.resources };\n    return tool;\n  }\n}\n"]}
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../../src/agents/utils.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;;AA+BlC,wCAKC;AAbD;;;;;;;GAOG;AACH,SAAgB,cAAc,CAC5B,MAAkF,EAClF,IAAY;IAEZ,OAAO,MAAM,CAAC,IAAI,KAAK,IAAI,CAAC;AAC9B,CAAC;AAED,2DAA2D;AAC3D,IAAY,kBASX;AATD,WAAY,kBAAkB;IAC5B,iCAAiC;IACjC,sDAAgC,CAAA;IAChC,4BAA4B;IAC5B,0DAAoC,CAAA;IACpC,sBAAsB;IACtB,kEAA4C,CAAA;IAC5C,0BAA0B;IAC1B,sDAAgC,CAAA;AAClC,CAAC,EATW,kBAAkB,kCAAlB,kBAAkB,QAS7B;AAED,MAAM,OAAO,GAAG;IACd,cAAc,EAAE,eAAe;IAC/B,gBAAgB,EAAE,kBAAkB;IACpC,oBAAoB,EAAE,qBAAqB;IAC3C,cAAc,EAAE,eAAe;CAChC,CAAC;AAEF;;GAEG;AACH,MAAa,WAAW;IACtB;;;;;;OAMG;IACH,MAAM,CAAC,oBAAoB,CACzB,QAA4B,EAC5B,aAAuB;QAEvB,OAAO;YACL,UAAU,EAAE;gBACV,IAAI,EAAE,QAAQ;gBACd,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,EAAE;oBACnB,WAAW,EAAE,aAAa,CAAC,GAAG,CAAC,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC,EAAE,YAAY,EAAE,YAAY,EAAE,CAAC,CAAC;iBACnF;aACF;SACF,CAAC;IACJ,CAAC;IAED;;;;;;;;OAQG;IACH,MAAM,CAAC,oBAAoB,CACzB,cAAyB,EACzB,YAA+C,EAC/C,iBAAmD;QAEnD,OAAO;YACL,UAAU,EAAE,EAAE,IAAI,EAAE,aAAa,EAAE,UAAU,EAAE,iBAAiB,EAAE;YAClE,SAAS,EAAE,EAAE,UAAU,EAAE,EAAE,cAAc,EAAE,cAAc,EAAE,YAAY,EAAE,YAAY,EAAE,EAAE;SAC1F,CAAC;IACJ,CAAC;IAED;;;;;;;OAOG;IACH,MAAM,CAAC,yBAAyB,CAC9B,OAAkB,EAClB,WAA0C;QAE1C,IAAI,OAAO,IAAI,WAAW,EAAE,CAAC;YAC3B,MAAM,IAAI,KAAK,CAAC,6CAA6C,CAAC,CAAC;QACjE,CAAC;QAED,OAAO;YACL,UAAU,EAAE,EAAE,IAAI,EAAE,kBAAkB,EAAE;YACxC,SAAS,EAAE,EAAE,eAAe,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,WAAW,EAAE,WAAW,EAAE,EAAE;SAC/E,CAAC;IACJ,CAAC;IAED;;;;;;;;OAQG;IACH,MAAM,CAAC,uBAAuB,CAC5B,IAAY,EACZ,WAAmB,EACnB,UAAmB,EACnB,UAAqC,EACrC,WAAsC,EACtC,iBAA0C;QAE1C,OAAO;YACL,UAAU,EAAE,EAAE,IAAI,EAAE,gBAAgB,EAAE,aAAa,EAAE,iBAAiB,EAAE;YACxE,SAAS,EAAE;gBACT,aAAa,EAAE;oBACb,IAAI,EAAE,IAAI;oBACV,WAAW,EAAE,WAAW;oBACxB,UAAU,EAAE,UAAU;oBACtB,UAAU,EAAE,UAAU;oBACtB,WAAW,EAAE,WAAW;iBACzB;aACF;SACF,CAAC;IACJ,CAAC;IACD;;;;;;;OAOG;IACH,MAAM,CAAC,uBAAuB,CAC5B,iBAAyB,EACzB,SAAiB;QAEjB,OAAO;YACL,UAAU,EAAE,EAAE,IAAI,EAAE,iBAAiB,EAAE;YACvC,SAAS,EAAE;gBACT,aAAa,EAAE;oBACb,OAAO,EAAE,CAAC,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC;iBAC1E;aACF;SACF,CAAC;IACJ,CAAC;IAED;;;;;OAKG;IACH,MAAM,CAAC,gBAAgB,CAAC,YAAoB;QAC1C,OAAO;YACL,UAAU,EAAE;gBACV,IAAI,EAAE,kBAAkB;gBACxB,eAAe,EAAE;oBACf,WAAW,EAAE,CAAC,EAAE,YAAY,EAAE,YAAY,EAAE,CAAC;iBAC9C;aACF;SACF,CAAC;IACJ,CAAC;IAED;;;;;;OAMG;IACH,MAAM,CAAC,kBAAkB,CAAC,kBAAsC;QAG9D,OAAO;YACL,UAAU,EAAE;gBACV,IAAI,EAAE,UAAU;gBAChB,QAAQ,EAAE,kBAAkB;aAC7B;SACF,CAAC;IACJ,CAAC;IAED;;;;;;OAMG;IACH,MAAM,CAAC,iBAAiB,CAAC,yBAAoD;QAG3E,OAAO;YACL,UAAU,EAAE;gBACV,IAAI,EAAE,SAAS;gBACf,OAAO,EAAE,yBAAyB;aACnC;SACF,CAAC;IACJ,CAAC;CACF;AAxKD,kCAwKC;AAED;;GAEG;AACH,MAAa,OAAO;IAApB;QACE,uEAAuE;QACvE,oBAAe,GAAqB,EAAE,CAAC;QAEvC,2EAA2E;QAC3E,kBAAa,GAAkB,EAAE,CAAC;IAoIpC,CAAC;IAlIC;;;;;;;OAOG;IACH,iBAAiB,CACf,QAA4B,EAC5B,aAAuB;QAEvB,MAAM,IAAI,GAAG,WAAW,CAAC,oBAAoB,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC;QACvE,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC3C,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;;;;;;;OAQG;IACH,iBAAiB,CACf,cAAyB,EACzB,YAA+C,EAC/C,iBAAmD;QAEnD,MAAM,IAAI,GAAG,WAAW,CAAC,oBAAoB,CAAC,cAAc,EAAE,YAAY,EAAE,iBAAiB,CAAC,CAAC;QAC/F,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC3C,IAAI,CAAC,aAAa,mCAAQ,IAAI,CAAC,aAAa,GAAK,IAAI,CAAC,SAAS,CAAE,CAAC;QAClE,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;;;;;;OAOG;IACH,sBAAsB,CACpB,OAAkB,EAClB,WAA0C;QAE1C,MAAM,IAAI,GAAG,WAAW,CAAC,yBAAyB,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;QACzE,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC3C,IAAI,CAAC,aAAa,mCAAQ,IAAI,CAAC,aAAa,GAAK,IAAI,CAAC,SAAS,CAAE,CAAC;QAClE,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;;;;;;OAOG;IACH,oBAAoB,CAClB,iBAAyB,EACzB,SAAiB;QAEjB,MAAM,IAAI,GAAG,WAAW,CAAC,uBAAuB,CAAC,iBAAiB,EAAE,SAAS,CAAC,CAAC;QAC/E,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC3C,IAAI,CAAC,aAAa,mCAAQ,IAAI,CAAC,aAAa,GAAK,IAAI,CAAC,SAAS,CAAE,CAAC;QAClE,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;;;;;OAMG;IACH,cAAc,CAAC,yBAAoD;QAGjE,MAAM,IAAI,GAAG,WAAW,CAAC,iBAAiB,CAAC,yBAAyB,CAAC,CAAC;QACtE,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC3C,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;;;;;;;;;OAUG;IACH,oBAAoB,CAClB,IAAY,EACZ,WAAmB,EACnB,UAAmB,EACnB,UAAqC,EACrC,WAAsC,EACtC,iBAA0C;QAE1C,MAAM,IAAI,GAAG,WAAW,CAAC,uBAAuB,CAC9C,IAAI,EACJ,WAAW,EACX,UAAU,EACV,UAAU,EACV,WAAW,EACX,iBAAiB,CAClB,CAAC;QACF,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC3C,IAAI,CAAC,aAAa,mCAAQ,IAAI,CAAC,aAAa,GAAK,IAAI,CAAC,SAAS,CAAE,CAAC;QAClE,OAAO,IAAI,CAAC;IACd,CAAC;IACD;;;;;OAKG;IACH,aAAa,CAAC,YAAoB;QAChC,MAAM,IAAI,GAAG,WAAW,CAAC,gBAAgB,CAAC,YAAY,CAAC,CAAC;QACxD,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC3C,OAAO,IAAI,CAAC;IACd,CAAC;CACF;AAzID,0BAyIC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport type {\n  AzureAISearchToolDefinition,\n  AzureFunctionToolDefinition,\n  CodeInterpreterToolDefinition,\n  FileSearchToolDefinition,\n  FileSearchToolDefinitionDetails,\n  FunctionDefinition,\n  FunctionToolDefinition,\n  OpenApiFunctionDefinition,\n  OpenApiToolDefinition,\n  RequiredActionOutput,\n  RequiredToolCallOutput,\n  ToolDefinition,\n  ToolDefinitionOutputParent,\n  ToolResources,\n  VectorStoreConfigurations,\n  VectorStoreDataSource,\n  AzureFunctionStorageQueue,\n  AzureFunctionDefinition,\n} from \"./inputOutputs.js\";\n\n/**\n * Determines if the given output is of the specified type.\n *\n * @typeParam T - The type to check against, which extends one of the possible output parent types.\n * @param output - The action to check, which can be of type `RequiredActionOutput`, `RequiredToolCallOutput`, or `ToolDefinitionOutputParent`.\n * @param type - The type to check the action against.\n * @returns A boolean indicating whether the action is of the specified type.\n */\nexport function isOutputOfType<T extends { type: string }>(\n  output: RequiredActionOutput | RequiredToolCallOutput | ToolDefinitionOutputParent,\n  type: string,\n): output is T {\n  return output.type === type;\n}\n\n/** Types of connection tools used to configure an agent */\nexport enum connectionToolType {\n  /** Bing grounding search tool */\n  BingGrounding = \"bing_grounding\",\n  /** Microsoft Fabric tool */\n  MicrosoftFabric = \"fabric_dataagent\",\n  /** Sharepoint tool */\n  SharepointGrounding = \"sharepoint_grounding\",\n  /** Azure Function tool */\n  AzureFunction = \"azure_function\",\n}\n\nconst toolMap = {\n  bing_grounding: \"bingGrounding\",\n  fabric_dataagent: \"fabric_dataagent\",\n  sharepoint_grounding: \"sharepointGrounding\",\n  azure_function: \"azureFunction\",\n};\n\n/**\n * Utility class for creating various tools.\n */\nexport class ToolUtility {\n  /**\n   * Creates a connection tool\n   *\n   * @param toolType - The type of the connection tool.\n   * @param connectionIds - A list of the IDs of the connections to use.\n   * @returns An object containing the definition for the connection tool\n   */\n  static createConnectionTool(\n    toolType: connectionToolType,\n    connectionIds: string[],\n  ): { definition: ToolDefinition } {\n    return {\n      definition: {\n        type: toolType,\n        [toolMap[toolType]]: {\n          connections: connectionIds.map((connectionId) => ({ connectionId: connectionId })),\n        },\n      },\n    };\n  }\n\n  /**\n   * Creates a file search tool\n   *\n   * @param vectorStoreIds - The ID of the vector store attached to this agent. There can be a maximum of 1 vector store attached to the agent.\n   * @param vectorStores - The list of vector store configuration objects from Azure. This list is limited to one element. The only element of this list contains the list of azure asset IDs used by the search tool.\n   * @param definitionDetails - The input definition information for a file search tool as used to configure an agent.\n   *\n   * @returns An object containing the definition and resources for the file search tool\n   */\n  static createFileSearchTool(\n    vectorStoreIds?: string[],\n    vectorStores?: Array<VectorStoreConfigurations>,\n    definitionDetails?: FileSearchToolDefinitionDetails,\n  ): { definition: FileSearchToolDefinition; resources: ToolResources } {\n    return {\n      definition: { type: \"file_search\", fileSearch: definitionDetails },\n      resources: { fileSearch: { vectorStoreIds: vectorStoreIds, vectorStores: vectorStores } },\n    };\n  }\n\n  /**\n   * Creates a code interpreter tool\n   *\n   * @param fileIds - A list of file IDs made available to the `code_interpreter` tool. There can be a maximum of 20 files associated with the tool.\n   * @param dataSources - The data sources to be used. This option is mutually exclusive with fileIds.\n   *\n   * @returns An object containing the definition and resources for the code interpreter tool.\n   */\n  static createCodeInterpreterTool(\n    fileIds?: string[],\n    dataSources?: Array<VectorStoreDataSource>,\n  ): { definition: CodeInterpreterToolDefinition; resources: ToolResources } {\n    if (fileIds && dataSources) {\n      throw new Error(\"Cannot specify both fileIds and dataSources\");\n    }\n\n    return {\n      definition: { type: \"code_interpreter\" },\n      resources: { codeInterpreter: { fileIds: fileIds, dataSources: dataSources } },\n    };\n  }\n\n  /**\n   * Creates an Azure Function tool\n   * @param name - The name of the Azure Function.\n   * @param description - The description of the Azure Function.\n   * @param parameters - The parameters of the Azure Function.\n   * @param inputQueue - The input queue configuration.\n   * @param outputQueue - The output queue configuration.\n   * @returns An object containing the definition and resources for the Azure Function tool.\n   */\n  static createAzureFunctionTool(\n    name: string,\n    description: string,\n    parameters: unknown,\n    inputQueue: AzureFunctionStorageQueue,\n    outputQueue: AzureFunctionStorageQueue,\n    definitionDetails: AzureFunctionDefinition,\n  ): { definition: AzureFunctionToolDefinition; resources: ToolResources } {\n    return {\n      definition: { type: \"azure_function\", azureFunction: definitionDetails },\n      resources: {\n        azureFunction: {\n          name: name,\n          description: description,\n          parameters: parameters,\n          inputQueue: inputQueue,\n          outputQueue: outputQueue,\n        },\n      },\n    };\n  }\n  /**\n   * Creates an Azure AI search tool\n   *\n   * @param indexConnectionId - The connection ID of the Azure AI search index.\n   * @param indexName - The name of the Azure AI search index.\n   *\n   * @returns An object containing the definition and resources for the Azure AI search tool.\n   */\n  static createAzureAISearchTool(\n    indexConnectionId: string,\n    indexName: string,\n  ): { definition: AzureAISearchToolDefinition; resources: ToolResources } {\n    return {\n      definition: { type: \"azure_ai_search\" },\n      resources: {\n        azureAISearch: {\n          indexes: [{ indexConnectionId: indexConnectionId, indexName: indexName }],\n        },\n      },\n    };\n  }\n\n  /**\n   * Creates a Microsoft Fabric tool\n   *\n   * @param connectionIds - A list of the IDs of the Fabric connections to use.\n   * @returns An object containing the definition for the Microsoft Fabric tool\n   */\n  static createFabricTool(connectionId: string): { definition: ToolDefinition } {\n    return {\n      definition: {\n        type: \"fabric_dataagent\",\n        fabricDataAgent: {\n          connections: [{ connectionId: connectionId }],\n        },\n      },\n    };\n  }\n\n  /**\n   * Creates a function tool\n   *\n   * @param functionDefinition - The function definition to use.\n   *\n   * @returns An object containing the definition for the function tool.\n   */\n  static createFunctionTool(functionDefinition: FunctionDefinition): {\n    definition: FunctionToolDefinition;\n  } {\n    return {\n      definition: {\n        type: \"function\",\n        function: functionDefinition,\n      },\n    };\n  }\n\n  /**\n   * Creates an OpenApi tool\n   *\n   * @param openApiFunctionDefinition - The OpenApi function definition to use.\n   *\n   * @returns An object containing the definition for the OpenApi tool.\n   */\n  static createOpenApiTool(openApiFunctionDefinition: OpenApiFunctionDefinition): {\n    definition: OpenApiToolDefinition;\n  } {\n    return {\n      definition: {\n        type: \"openapi\",\n        openapi: openApiFunctionDefinition,\n      },\n    };\n  }\n}\n\n/**\n * Represents a set of tools with their definitions and resources.\n */\nexport class ToolSet {\n  /** A list of tool definitions that have been added to the tool set. */\n  toolDefinitions: ToolDefinition[] = [];\n\n  /** A collection of resources associated with the tools in the tool set. */\n  toolResources: ToolResources = {};\n\n  /**\n   * Adds a connection tool to the tool set.\n   *\n   * @param toolType - The type of the connection tool.\n   * @param connectionIds - A list of the IDs of the connections to use.\n   *\n   * @returns An object containing the definition for the connection tool\n   */\n  addConnectionTool(\n    toolType: connectionToolType,\n    connectionIds: string[],\n  ): { definition: ToolDefinition } {\n    const tool = ToolUtility.createConnectionTool(toolType, connectionIds);\n    this.toolDefinitions.push(tool.definition);\n    return tool;\n  }\n\n  /**\n   * Adds a file search tool to the tool set.\n   *\n   * @param vectorStoreIds - The ID of the vector store attached to this agent. There can be a maximum of 1 vector store attached to the agent.\n   * @param vectorStores -  The list of vector store configuration objects from Azure. This list is limited to one element. The only element of this list contains the list of azure asset IDs used by the search tool.\n   * @param definitionDetails - The input definition information for a file search tool as used to configure an agent.\n   *\n   * @returns An object containing the definition and resources for the file search tool\n   */\n  addFileSearchTool(\n    vectorStoreIds?: string[],\n    vectorStores?: Array<VectorStoreConfigurations>,\n    definitionDetails?: FileSearchToolDefinitionDetails,\n  ): { definition: FileSearchToolDefinition; resources: ToolResources } {\n    const tool = ToolUtility.createFileSearchTool(vectorStoreIds, vectorStores, definitionDetails);\n    this.toolDefinitions.push(tool.definition);\n    this.toolResources = { ...this.toolResources, ...tool.resources };\n    return tool;\n  }\n\n  /**\n   * Adds a code interpreter tool to the tool set.\n   *\n   * @param fileIds - A list of file IDs made available to the `code_interpreter` tool. There can be a maximum of 20 files associated with the tool.\n   * @param dataSources - The data sources to be used. This option is mutually exclusive with fileIds.\n   *\n   * @returns An object containing the definition and resources for the code interpreter tool\n   */\n  addCodeInterpreterTool(\n    fileIds?: string[],\n    dataSources?: Array<VectorStoreDataSource>,\n  ): { definition: CodeInterpreterToolDefinition; resources: ToolResources } {\n    const tool = ToolUtility.createCodeInterpreterTool(fileIds, dataSources);\n    this.toolDefinitions.push(tool.definition);\n    this.toolResources = { ...this.toolResources, ...tool.resources };\n    return tool;\n  }\n\n  /**\n   * Adds an Azure AI search tool to the tool set.\n   *\n   * @param indexConnectionId - The connection ID of the Azure AI search index.\n   * @param indexName - The name of the Azure AI search index.\n   *\n   * @returns An object containing the definition and resources for the Azure AI search tool\n   */\n  addAzureAISearchTool(\n    indexConnectionId: string,\n    indexName: string,\n  ): { definition: AzureAISearchToolDefinition; resources: ToolResources } {\n    const tool = ToolUtility.createAzureAISearchTool(indexConnectionId, indexName);\n    this.toolDefinitions.push(tool.definition);\n    this.toolResources = { ...this.toolResources, ...tool.resources };\n    return tool;\n  }\n\n  /**\n   * Adds an OpenApi tool to the tool set.\n   *\n   * @param openApiFunctionDefinition - The OpenApi function definition to use.\n   *\n   * @returns An object containing the definition for the OpenApi tool\n   */\n  addOpenApiTool(openApiFunctionDefinition: OpenApiFunctionDefinition): {\n    definition: OpenApiToolDefinition;\n  } {\n    const tool = ToolUtility.createOpenApiTool(openApiFunctionDefinition);\n    this.toolDefinitions.push(tool.definition);\n    return tool;\n  }\n\n  /**\n   * Adds an Azure Function tool to the tool set.\n   *\n   * @param name - The name of the Azure Function.\n   * @param description - The description of the Azure Function.\n   * @param parameters - The parameters of the Azure Function.\n   * @param inputQueue - The input queue configuration.\n   * @param outputQueue - The output queue configuration.\n   *\n   * @returns An object containing the definition and resources for the Azure Function tool.\n   */\n  addAzureFunctionTool(\n    name: string,\n    description: string,\n    parameters: unknown,\n    inputQueue: AzureFunctionStorageQueue,\n    outputQueue: AzureFunctionStorageQueue,\n    definitionDetails: AzureFunctionDefinition,\n  ): { definition: AzureFunctionToolDefinition; resources: ToolResources } {\n    const tool = ToolUtility.createAzureFunctionTool(\n      name,\n      description,\n      parameters,\n      inputQueue,\n      outputQueue,\n      definitionDetails,\n    );\n    this.toolDefinitions.push(tool.definition);\n    this.toolResources = { ...this.toolResources, ...tool.resources };\n    return tool;\n  }\n  /**\n   * Adds a Microsoft Fabric tool to the tool set.\n   *\n   * @param connectionId - The ID of the Fabric connection to use.\n   * @returns An object containing the definition for the Microsoft Fabric tool\n   */\n  addFabricTool(connectionId: string): { definition: ToolDefinition } {\n    const tool = ToolUtility.createFabricTool(connectionId);\n    this.toolDefinitions.push(tool.definition);\n    return tool;\n  }\n}\n"]}

package/dist/commonjs/agents/vectorStores.d.ts

@@ -1,12 +1,11 @@
 import type { Client } from "@azure-rest/core-client";
 import type { OpenAIPageableListOfVectorStoreOutput, VectorStoreDeletionStatusOutput, VectorStoreOutput } from "../customization/outputModels.js";
-import type { CreateVectorStoreWithPollingOptionalParams } from "./customModels.js";
 import { type CreateVectorStoreOptionalParams, type DeleteVectorStoreOptionalParams, type GetVectorStoreOptionalParams, type ListVectorStoresOptionalParams, type UpdateVectorStoreOptionalParams } from "./customModels.js";
-import type { PollerLike, PollOperationState } from "@azure/core-lro";
+import type { OperationState, PollerLike } from "@azure/core-lro";
 /** Returns a list of vector stores. */
 export declare function listVectorStores(context: Client, options?: ListVectorStoresOptionalParams): Promise<OpenAIPageableListOfVectorStoreOutput>;
 /** Creates a vector store. */
-export declare function createVectorStore(context: Client, options?: CreateVectorStoreOptionalParams): Promise<VectorStoreOutput>;
+export declare function createVectorStore(context: Client, options?: CreateVectorStoreOptionalParams): PollerLike<OperationState<VectorStoreOutput>, VectorStoreOutput>;
 /** Returns the vector store object matching the specified ID. */
 export declare function getVectorStore(context: Client, vectorStoreId: string, options?: GetVectorStoreOptionalParams): Promise<VectorStoreOutput>;
 /** The ID of the vector store to modify. */
@@ -16,5 +15,5 @@ export declare function deleteVectorStore(context: Client, vectorStoreId: string
 /**
  * Creates a vector store and poll.
  */
-export declare function createVectorStoreAndPoll(context: Client, options?: CreateVectorStoreWithPollingOptionalParams): PollerLike<PollOperationState<VectorStoreOutput>, VectorStoreOutput>;
+export declare function createVectorStoreAndPoll(context: Client, options?: CreateVectorStoreOptionalParams): PollerLike<OperationState<VectorStoreOutput>, VectorStoreOutput>;
 //# sourceMappingURL=vectorStores.d.ts.map

package/dist/commonjs/agents/vectorStores.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"vectorStores.d.ts","sourceRoot":"","sources":["../../../src/agents/vectorStores.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,yBAAyB,CAAC;AAOtD,OAAO,KAAK,EACV,qCAAqC,EACrC,+BAA+B,EAC/B,iBAAiB,EAClB,MAAM,kCAAkC,CAAC;AAE1C,OAAO,KAAK,EAAE,0CAA0C,EAAE,MAAM,mBAAmB,CAAC;AACpF,OAAO,EACL,KAAK,+BAA+B,EACpC,KAAK,+BAA+B,EACpC,KAAK,4BAA4B,EACjC,KAAK,8BAA8B,EACnC,KAAK,+BAA+B,EACrC,MAAM,mBAAmB,CAAC;AAY3B,OAAO,KAAK,EAAE,UAAU,EAAE,kBAAkB,EAAE,MAAM,iBAAiB,CAAC;AAItE,uCAAuC;AACvC,wBAAsB,gBAAgB,CACpC,OAAO,EAAE,MAAM,EACf,OAAO,GAAE,8BAAmC,GAC3C,OAAO,CAAC,qCAAqC,CAAC,CAahD;AAED,8BAA8B;AAC9B,wBAAsB,iBAAiB,CACrC,OAAO,EAAE,MAAM,EACf,OAAO,GAAE,+BAAoC,GAC5C,OAAO,CAAC,iBAAiB,CAAC,CAa5B;AAED,iEAAiE;AACjE,wBAAsB,cAAc,CAClC,OAAO,EAAE,MAAM,EACf,aAAa,EAAE,MAAM,EACrB,OAAO,GAAE,4BAAiC,GACzC,OAAO,CAAC,iBAAiB,CAAC,CAc5B;AAED,4CAA4C;AAC5C,wBAAsB,iBAAiB,CACrC,OAAO,EAAE,MAAM,EACf,aAAa,EAAE,MAAM,EACrB,OAAO,GAAE,+BAAoC,GAC5C,OAAO,CAAC,iBAAiB,CAAC,CAgB5B;AAED,iEAAiE;AACjE,wBAAsB,iBAAiB,CACrC,OAAO,EAAE,MAAM,EACf,aAAa,EAAE,MAAM,EACrB,OAAO,GAAE,+BAAoC,GAC5C,OAAO,CAAC,+BAA+B,CAAC,CAc1C;AAED;;GAEG;AACH,wBAAgB,wBAAwB,CACtC,OAAO,EAAE,MAAM,EACf,OAAO,GAAE,0CAA+C,GACvD,UAAU,CAAC,kBAAkB,CAAC,iBAAiB,CAAC,EAAE,iBAAiB,CAAC,CAuBtE"}
+{"version":3,"file":"vectorStores.d.ts","sourceRoot":"","sources":["../../../src/agents/vectorStores.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,yBAAyB,CAAC;AAOtD,OAAO,KAAK,EACV,qCAAqC,EACrC,+BAA+B,EAC/B,iBAAiB,EAClB,MAAM,kCAAkC,CAAC;AAE1C,OAAO,EACL,KAAK,+BAA+B,EACpC,KAAK,+BAA+B,EACpC,KAAK,4BAA4B,EACjC,KAAK,8BAA8B,EACnC,KAAK,+BAA+B,EACrC,MAAM,mBAAmB,CAAC;AAY3B,OAAO,KAAK,EAAE,cAAc,EAAmB,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAInF,uCAAuC;AACvC,wBAAsB,gBAAgB,CACpC,OAAO,EAAE,MAAM,EACf,OAAO,GAAE,8BAAmC,GAC3C,OAAO,CAAC,qCAAqC,CAAC,CAahD;AAED,8BAA8B;AAC9B,wBAAgB,iBAAiB,CAC/B,OAAO,EAAE,MAAM,EACf,OAAO,GAAE,+BAAoC,GAC5C,UAAU,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,iBAAiB,CAAC,CAWlE;AAED,iEAAiE;AACjE,wBAAsB,cAAc,CAClC,OAAO,EAAE,MAAM,EACf,aAAa,EAAE,MAAM,EACrB,OAAO,GAAE,4BAAiC,GACzC,OAAO,CAAC,iBAAiB,CAAC,CAc5B;AAED,4CAA4C;AAC5C,wBAAsB,iBAAiB,CACrC,OAAO,EAAE,MAAM,EACf,aAAa,EAAE,MAAM,EACrB,OAAO,GAAE,+BAAoC,GAC5C,OAAO,CAAC,iBAAiB,CAAC,CAgB5B;AAED,iEAAiE;AACjE,wBAAsB,iBAAiB,CACrC,OAAO,EAAE,MAAM,EACf,aAAa,EAAE,MAAM,EACrB,OAAO,GAAE,+BAAoC,GAC5C,OAAO,CAAC,+BAA+B,CAAC,CAc1C;AAED;;GAEG;AACH,wBAAgB,wBAAwB,CACtC,OAAO,EAAE,MAAM,EACf,OAAO,GAAE,+BAAoC,GAC5C,UAAU,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,iBAAiB,CAAC,CAWlE"}

package/dist/commonjs/agents/vectorStores.js

@@ -28,14 +28,18 @@ async function listVectorStores(context, options = {}) {
     return ConvertFromWire.convertOpenAIPageableListOfVectorStoreOutput(result.body);
 }
 /** Creates a vector store. */
-async function createVectorStore(context, options = {}) {
-    const createOptions = Object.assign(Object.assign({}, (0, core_client_1.operationOptionsToRequestParameters)(options)), { body: ConvertToWire.convertVectorStoreOptions(options) });
-    validateCreateVectorStoreParameters(createOptions);
-    const result = await context.path("/vector_stores").post(createOptions);
-    if (!expectedStatuses.includes(result.status)) {
-        throw (0, openAIError_js_1.createOpenAIError)(result);
-    }
-    return ConvertFromWire.convertVectorStoreOutput(result.body);
+function createVectorStore(context, options = {}) {
+    var _a;
+    return (0, poller_js_1.createPoller)({
+        initOperation: async () => {
+            return createVectorStoreInternal(context, options);
+        },
+        pollOperation: async (currentResult) => {
+            return getVectorStore(context, currentResult.id, options);
+        },
+        getOperationStatus: getLroOperationStatus,
+        intervalInMs: (_a = options.pollingOptions) === null || _a === void 0 ? void 0 : _a.sleepIntervalInMs,
+    });
 }
 /** Returns the vector store object matching the specified ID. */
 async function getVectorStore(context, vectorStoreId, options = {}) {
@@ -78,25 +82,39 @@ async function deleteVectorStore(context, vectorStoreId, options = {}) {
  * Creates a vector store and poll.
  */
 function createVectorStoreAndPoll(context, options = {}) {
-    async function updateCreateVectorStorePoll(currentResult) {
-        let vectorStore;
-        if (!currentResult) {
-            vectorStore = await createVectorStore(context, options);
-        }
-        else {
-            const getOptions = Object.assign({}, (0, core_client_1.operationOptionsToRequestParameters)(options));
-            vectorStore = await getVectorStore(context, currentResult.id, getOptions);
-        }
-        return {
-            result: vectorStore,
-            completed: vectorStore.status !== "in_progress",
-        };
-    }
-    return new poller_js_1.AgentsPoller({
-        update: updateCreateVectorStorePoll,
-        pollingOptions: options.pollingOptions,
+    var _a;
+    return (0, poller_js_1.createPoller)({
+        initOperation: async () => {
+            return createVectorStoreInternal(context, options);
+        },
+        pollOperation: async (currentResult) => {
+            return getVectorStore(context, currentResult.id, options);
+        },
+        getOperationStatus: getLroOperationStatus,
+        intervalInMs: (_a = options.pollingOptions) === null || _a === void 0 ? void 0 : _a.sleepIntervalInMs,
     });
 }
+async function createVectorStoreInternal(context, options = {}) {
+    const createOptions = Object.assign(Object.assign({}, (0, core_client_1.operationOptionsToRequestParameters)(options)), { body: ConvertToWire.convertVectorStoreOptions(options) });
+    validateCreateVectorStoreParameters(createOptions);
+    const result = await context.path("/vector_stores").post(createOptions);
+    if (!expectedStatuses.includes(result.status)) {
+        throw (0, openAIError_js_1.createOpenAIError)(result);
+    }
+    return ConvertFromWire.convertVectorStoreOutput(result.body);
+}
+function getLroOperationStatus(result) {
+    switch (result.status) {
+        case "in_progress":
+            return "running";
+        case "completed":
+            return "succeeded";
+        case "expired":
+            return "failed";
+        default:
+            return "failed";
+    }
+}
 function validateListVectorStoresParameters(options) {
     var _a, _b;
     if ((_a = options === null || options === void 0 ? void 0 : options.queryParameters) === null || _a === void 0 ? void 0 : _a.limit) {

package/dist/commonjs/agents/vectorStores.js.map

@@ -1 +1 @@
-{"version":3,"file":"vectorStores.js","sourceRoot":"","sources":["../../../src/agents/vectorStores.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;AAuClC,4CAgBC;AAGD,8CAgBC;AAGD,wCAkBC;AAGD,8CAoBC;AAGD,8CAkBC;AAKD,4DA0BC;;AAvKD,yDAA8E;AAW9E,2CAA2C;AAS3C,+DAK+B;AAE/B,yGAAmF;AACnF,gGAA0E;AAC1E,4FAA2F;AAC3F,qDAAqD;AAGrD,MAAM,gBAAgB,GAAG,CAAC,KAAK,CAAC,CAAC;AAEjC,uCAAuC;AAChC,KAAK,UAAU,gBAAgB,CACpC,OAAe,EACf,UAA0C,EAAE;IAE5C,MAAM,WAAW,mCACZ,IAAA,iDAAmC,EAAC,OAAO,CAAC,KAC/C,eAAe,EAAE,IAAA,yDAA4B,EAAC,OAAO,CAAC,GACvD,CAAC;IAEF,kCAAkC,CAAC,WAAW,CAAC,CAAC;IAChD,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;IAErE,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;QAC9C,MAAM,IAAA,kCAAiB,EAAC,MAAM,CAAC,CAAC;IAClC,CAAC;IACD,OAAO,eAAe,CAAC,4CAA4C,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AACnF,CAAC;AAED,8BAA8B;AACvB,KAAK,UAAU,iBAAiB,CACrC,OAAe,EACf,UAA2C,EAAE;IAE7C,MAAM,aAAa,mCACd,IAAA,iDAAmC,EAAC,OAAO,CAAC,KAC/C,IAAI,EAAE,aAAa,CAAC,yBAAyB,CAAC,OAAO,CAAC,GACvD,CAAC;IAEF,mCAAmC,CAAC,aAAa,CAAC,CAAC;IACnD,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IAExE,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;QAC9C,MAAM,IAAA,kCAAiB,EAAC,MAAM,CAAC,CAAC;IAClC,CAAC;IACD,OAAO,eAAe,CAAC,wBAAwB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AAC/D,CAAC;AAED,iEAAiE;AAC1D,KAAK,UAAU,cAAc,CAClC,OAAe,EACf,aAAqB,EACrB,UAAwC,EAAE;IAE1C,MAAM,UAAU,qBACX,IAAA,iDAAmC,EAAC,OAAO,CAAC,CAChD,CAAC;IAEF,IAAA,2CAAqB,EAAC,aAAa,CAAC,CAAC;IACrC,MAAM,MAAM,GAAG,MAAM,OAAO;SACzB,IAAI,CAAC,gCAAgC,EAAE,aAAa,CAAC;SACrD,GAAG,CAAC,UAAU,CAAC,CAAC;IAEnB,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;QAC9C,MAAM,IAAA,kCAAiB,EAAC,MAAM,CAAC,CAAC;IAClC,CAAC;IACD,OAAO,eAAe,CAAC,wBAAwB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AAC/D,CAAC;AAED,4CAA4C;AACrC,KAAK,UAAU,iBAAiB,CACrC,OAAe,EACf,aAAqB,EACrB,UAA2C,EAAE;IAE7C,MAAM,aAAa,mCACd,IAAA,iDAAmC,EAAC,OAAO,CAAC,KAC/C,IAAI,EAAE,aAAa,CAAC,+BAA+B,CAAC,OAAO,CAAC,GAC7D,CAAC;IAEF,IAAA,2CAAqB,EAAC,aAAa,CAAC,CAAC;IACrC,mCAAmC,CAAC,aAAa,CAAC,CAAC;IACnD,MAAM,MAAM,GAAG,MAAM,OAAO;SACzB,IAAI,CAAC,gCAAgC,EAAE,aAAa,CAAC;SACrD,IAAI,CAAC,aAAa,CAAC,CAAC;IAEvB,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;QAC9C,MAAM,IAAA,kCAAiB,EAAC,MAAM,CAAC,CAAC;IAClC,CAAC;IACD,OAAO,eAAe,CAAC,wBAAwB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AAC/D,CAAC;AAED,iEAAiE;AAC1D,KAAK,UAAU,iBAAiB,CACrC,OAAe,EACf,aAAqB,EACrB,UAA2C,EAAE;IAE7C,MAAM,aAAa,qBACd,IAAA,iDAAmC,EAAC,OAAO,CAAC,CAChD,CAAC;IAEF,IAAA,2CAAqB,EAAC,aAAa,CAAC,CAAC;IACrC,MAAM,MAAM,GAAG,MAAM,OAAO;SACzB,IAAI,CAAC,gCAAgC,EAAE,aAAa,CAAC;SACrD,MAAM,CAAC,aAAa,CAAC,CAAC;IAEzB,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;QAC9C,MAAM,IAAA,kCAAiB,EAAC,MAAM,CAAC,CAAC;IAClC,CAAC;IACD,OAAO,eAAe,CAAC,sCAAsC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AAC7E,CAAC;AAED;;GAEG;AACH,SAAgB,wBAAwB,CACtC,OAAe,EACf,UAAsD,EAAE;IAExD,KAAK,UAAU,2BAA2B,CACxC,aAAiC;QAEjC,IAAI,WAA8B,CAAC;QACnC,IAAI,CAAC,aAAa,EAAE,CAAC;YACnB,WAAW,GAAG,MAAM,iBAAiB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QAC1D,CAAC;aAAM,CAAC;YACN,MAAM,UAAU,qBACX,IAAA,iDAAmC,EAAC,OAAO,CAAC,CAChD,CAAC;YACF,WAAW,GAAG,MAAM,cAAc,CAAC,OAAO,EAAE,aAAa,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;QAC5E,CAAC;QACD,OAAO;YACL,MAAM,EAAE,WAAW;YACnB,SAAS,EAAE,WAAW,CAAC,MAAM,KAAK,aAAa;SAChD,CAAC;IACJ,CAAC;IAED,OAAO,IAAI,wBAAY,CAAoB;QACzC,MAAM,EAAE,2BAA2B;QACnC,cAAc,EAAE,OAAO,CAAC,cAAc;KACvC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,kCAAkC,CAAC,OAAoC;;IAC9E,IAAI,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,eAAe,0CAAE,KAAK,EAAE,CAAC;QACpC,IAAA,mCAAa,EAAC,OAAO,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;IAC/C,CAAC;IACD,IAAI,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,eAAe,0CAAE,KAAK,EAAE,CAAC;QACpC,IAAA,mCAAa,EAAC,OAAO,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;IAC/C,CAAC;AACH,CAAC;AAED,SAAS,mCAAmC,CAAC,OAAqC;;IAChF,IACE,CAAA,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,IAAI,0CAAE,iBAAiB;QAChC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC,CAAC,EAC9D,CAAC;QACD,MAAM,IAAI,KAAK,CAAC,8DAA8D,CAAC,CAAC;IAClF,CAAC;IACD,IAAI,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,IAAI,0CAAE,QAAQ,EAAE,CAAC;QAC5B,IAAA,sCAAgB,EAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC1C,CAAC;AACH,CAAC;AAED,SAAS,mCAAmC,CAAC,OAAqC;;IAChF,IAAI,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,IAAI,0CAAE,QAAQ,EAAE,CAAC;QAC5B,IAAA,sCAAgB,EAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC1C,CAAC;AACH,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport type { Client } from \"@azure-rest/core-client\";\nimport { operationOptionsToRequestParameters } from \"@azure-rest/core-client\";\nimport type {\n  ListVectorStoresParameters,\n  CreateVectorStoreParameters,\n  ModifyVectorStoreParameters,\n} from \"../generated/src/parameters.js\";\nimport type {\n  OpenAIPageableListOfVectorStoreOutput,\n  VectorStoreDeletionStatusOutput,\n  VectorStoreOutput,\n} from \"../customization/outputModels.js\";\nimport { AgentsPoller } from \"./poller.js\";\nimport type { CreateVectorStoreWithPollingOptionalParams } from \"./customModels.js\";\nimport {\n  type CreateVectorStoreOptionalParams,\n  type DeleteVectorStoreOptionalParams,\n  type GetVectorStoreOptionalParams,\n  type ListVectorStoresOptionalParams,\n  type UpdateVectorStoreOptionalParams,\n} from \"./customModels.js\";\nimport {\n  validateLimit,\n  validateMetadata,\n  validateOrder,\n  validateVectorStoreId,\n} from \"./inputValidations.js\";\nimport type * as GeneratedParameters from \"../generated/src/parameters.js\";\nimport * as ConvertFromWire from \"../customization/convertOutputModelsFromWire.js\";\nimport * as ConvertToWire from \"../customization/convertModelsToWrite.js\";\nimport { convertToListQueryParameters } from \"../customization/convertParametersToWire.js\";\nimport { createOpenAIError } from \"./openAIError.js\";\nimport type { PollerLike, PollOperationState } from \"@azure/core-lro\";\n\nconst expectedStatuses = [\"200\"];\n\n/** Returns a list of vector stores. */\nexport async function listVectorStores(\n  context: Client,\n  options: ListVectorStoresOptionalParams = {},\n): Promise<OpenAIPageableListOfVectorStoreOutput> {\n  const listOptions: GeneratedParameters.ListVectorStoresParameters = {\n    ...operationOptionsToRequestParameters(options),\n    queryParameters: convertToListQueryParameters(options),\n  };\n\n  validateListVectorStoresParameters(listOptions);\n  const result = await context.path(\"/vector_stores\").get(listOptions);\n\n  if (!expectedStatuses.includes(result.status)) {\n    throw createOpenAIError(result);\n  }\n  return ConvertFromWire.convertOpenAIPageableListOfVectorStoreOutput(result.body);\n}\n\n/** Creates a vector store. */\nexport async function createVectorStore(\n  context: Client,\n  options: CreateVectorStoreOptionalParams = {},\n): Promise<VectorStoreOutput> {\n  const createOptions: GeneratedParameters.CreateVectorStoreParameters = {\n    ...operationOptionsToRequestParameters(options),\n    body: ConvertToWire.convertVectorStoreOptions(options),\n  };\n\n  validateCreateVectorStoreParameters(createOptions);\n  const result = await context.path(\"/vector_stores\").post(createOptions);\n\n  if (!expectedStatuses.includes(result.status)) {\n    throw createOpenAIError(result);\n  }\n  return ConvertFromWire.convertVectorStoreOutput(result.body);\n}\n\n/** Returns the vector store object matching the specified ID. */\nexport async function getVectorStore(\n  context: Client,\n  vectorStoreId: string,\n  options: GetVectorStoreOptionalParams = {},\n): Promise<VectorStoreOutput> {\n  const getOptions: GeneratedParameters.GetVectorStoreParameters = {\n    ...operationOptionsToRequestParameters(options),\n  };\n\n  validateVectorStoreId(vectorStoreId);\n  const result = await context\n    .path(\"/vector_stores/{vectorStoreId}\", vectorStoreId)\n    .get(getOptions);\n\n  if (!expectedStatuses.includes(result.status)) {\n    throw createOpenAIError(result);\n  }\n  return ConvertFromWire.convertVectorStoreOutput(result.body);\n}\n\n/** The ID of the vector store to modify. */\nexport async function modifyVectorStore(\n  context: Client,\n  vectorStoreId: string,\n  options: UpdateVectorStoreOptionalParams = {},\n): Promise<VectorStoreOutput> {\n  const modifyOptions: GeneratedParameters.ModifyVectorStoreParameters = {\n    ...operationOptionsToRequestParameters(options),\n    body: ConvertToWire.convertVectorStoreUpdateOptions(options),\n  };\n\n  validateVectorStoreId(vectorStoreId);\n  validateModifyVectorStoreParameters(modifyOptions);\n  const result = await context\n    .path(\"/vector_stores/{vectorStoreId}\", vectorStoreId)\n    .post(modifyOptions);\n\n  if (!expectedStatuses.includes(result.status)) {\n    throw createOpenAIError(result);\n  }\n  return ConvertFromWire.convertVectorStoreOutput(result.body);\n}\n\n/** Deletes the vector store object matching the specified ID. */\nexport async function deleteVectorStore(\n  context: Client,\n  vectorStoreId: string,\n  options: DeleteVectorStoreOptionalParams = {},\n): Promise<VectorStoreDeletionStatusOutput> {\n  const deleteOptions: GeneratedParameters.DeleteVectorStoreParameters = {\n    ...operationOptionsToRequestParameters(options),\n  };\n\n  validateVectorStoreId(vectorStoreId);\n  const result = await context\n    .path(\"/vector_stores/{vectorStoreId}\", vectorStoreId)\n    .delete(deleteOptions);\n\n  if (!expectedStatuses.includes(result.status)) {\n    throw createOpenAIError(result);\n  }\n  return ConvertFromWire.convertVectorStoreDeletionStatusOutput(result.body);\n}\n\n/**\n * Creates a vector store and poll.\n */\nexport function createVectorStoreAndPoll(\n  context: Client,\n  options: CreateVectorStoreWithPollingOptionalParams = {},\n): PollerLike<PollOperationState<VectorStoreOutput>, VectorStoreOutput> {\n  async function updateCreateVectorStorePoll(\n    currentResult?: VectorStoreOutput,\n  ): Promise<{ result: VectorStoreOutput; completed: boolean }> {\n    let vectorStore: VectorStoreOutput;\n    if (!currentResult) {\n      vectorStore = await createVectorStore(context, options);\n    } else {\n      const getOptions: GetVectorStoreOptionalParams = {\n        ...operationOptionsToRequestParameters(options),\n      };\n      vectorStore = await getVectorStore(context, currentResult.id, getOptions);\n    }\n    return {\n      result: vectorStore,\n      completed: vectorStore.status !== \"in_progress\",\n    };\n  }\n\n  return new AgentsPoller<VectorStoreOutput>({\n    update: updateCreateVectorStorePoll,\n    pollingOptions: options.pollingOptions,\n  });\n}\n\nfunction validateListVectorStoresParameters(options?: ListVectorStoresParameters): void {\n  if (options?.queryParameters?.limit) {\n    validateLimit(options.queryParameters.limit);\n  }\n  if (options?.queryParameters?.order) {\n    validateOrder(options.queryParameters.order);\n  }\n}\n\nfunction validateCreateVectorStoreParameters(options?: CreateVectorStoreParameters): void {\n  if (\n    options?.body?.chunking_strategy &&\n    (!options.body.file_ids || options.body.file_ids.length === 0)\n  ) {\n    throw new Error(\"Chunking strategy is only applicable if fileIds is non-empty\");\n  }\n  if (options?.body?.metadata) {\n    validateMetadata(options.body.metadata);\n  }\n}\n\nfunction validateModifyVectorStoreParameters(options?: ModifyVectorStoreParameters): void {\n  if (options?.body?.metadata) {\n    validateMetadata(options.body.metadata);\n  }\n}\n"]}
+{"version":3,"file":"vectorStores.js","sourceRoot":"","sources":["../../../src/agents/vectorStores.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;AAsClC,4CAgBC;AAGD,8CAcC;AAGD,wCAkBC;AAGD,8CAoBC;AAGD,8CAkBC;AAKD,4DAcC;;AAxJD,yDAA8E;AAW9E,2CAA2C;AAQ3C,+DAK+B;AAE/B,yGAAmF;AACnF,gGAA0E;AAC1E,4FAA2F;AAC3F,qDAAqD;AAGrD,MAAM,gBAAgB,GAAG,CAAC,KAAK,CAAC,CAAC;AAEjC,uCAAuC;AAChC,KAAK,UAAU,gBAAgB,CACpC,OAAe,EACf,UAA0C,EAAE;IAE5C,MAAM,WAAW,mCACZ,IAAA,iDAAmC,EAAC,OAAO,CAAC,KAC/C,eAAe,EAAE,IAAA,yDAA4B,EAAC,OAAO,CAAC,GACvD,CAAC;IAEF,kCAAkC,CAAC,WAAW,CAAC,CAAC;IAChD,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;IAErE,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;QAC9C,MAAM,IAAA,kCAAiB,EAAC,MAAM,CAAC,CAAC;IAClC,CAAC;IACD,OAAO,eAAe,CAAC,4CAA4C,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AACnF,CAAC;AAED,8BAA8B;AAC9B,SAAgB,iBAAiB,CAC/B,OAAe,EACf,UAA2C,EAAE;;IAE7C,OAAO,IAAA,wBAAY,EAAoB;QACrC,aAAa,EAAE,KAAK,IAAI,EAAE;YACxB,OAAO,yBAAyB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QACrD,CAAC;QACD,aAAa,EAAE,KAAK,EAAE,aAAgC,EAAE,EAAE;YACxD,OAAO,cAAc,CAAC,OAAO,EAAE,aAAa,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QAC5D,CAAC;QACD,kBAAkB,EAAE,qBAAqB;QACzC,YAAY,EAAE,MAAA,OAAO,CAAC,cAAc,0CAAE,iBAAiB;KACxD,CAAC,CAAC;AACL,CAAC;AAED,iEAAiE;AAC1D,KAAK,UAAU,cAAc,CAClC,OAAe,EACf,aAAqB,EACrB,UAAwC,EAAE;IAE1C,MAAM,UAAU,qBACX,IAAA,iDAAmC,EAAC,OAAO,CAAC,CAChD,CAAC;IAEF,IAAA,2CAAqB,EAAC,aAAa,CAAC,CAAC;IACrC,MAAM,MAAM,GAAG,MAAM,OAAO;SACzB,IAAI,CAAC,gCAAgC,EAAE,aAAa,CAAC;SACrD,GAAG,CAAC,UAAU,CAAC,CAAC;IAEnB,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;QAC9C,MAAM,IAAA,kCAAiB,EAAC,MAAM,CAAC,CAAC;IAClC,CAAC;IACD,OAAO,eAAe,CAAC,wBAAwB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AAC/D,CAAC;AAED,4CAA4C;AACrC,KAAK,UAAU,iBAAiB,CACrC,OAAe,EACf,aAAqB,EACrB,UAA2C,EAAE;IAE7C,MAAM,aAAa,mCACd,IAAA,iDAAmC,EAAC,OAAO,CAAC,KAC/C,IAAI,EAAE,aAAa,CAAC,+BAA+B,CAAC,OAAO,CAAC,GAC7D,CAAC;IAEF,IAAA,2CAAqB,EAAC,aAAa,CAAC,CAAC;IACrC,mCAAmC,CAAC,aAAa,CAAC,CAAC;IACnD,MAAM,MAAM,GAAG,MAAM,OAAO;SACzB,IAAI,CAAC,gCAAgC,EAAE,aAAa,CAAC;SACrD,IAAI,CAAC,aAAa,CAAC,CAAC;IAEvB,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;QAC9C,MAAM,IAAA,kCAAiB,EAAC,MAAM,CAAC,CAAC;IAClC,CAAC;IACD,OAAO,eAAe,CAAC,wBAAwB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AAC/D,CAAC;AAED,iEAAiE;AAC1D,KAAK,UAAU,iBAAiB,CACrC,OAAe,EACf,aAAqB,EACrB,UAA2C,EAAE;IAE7C,MAAM,aAAa,qBACd,IAAA,iDAAmC,EAAC,OAAO,CAAC,CAChD,CAAC;IAEF,IAAA,2CAAqB,EAAC,aAAa,CAAC,CAAC;IACrC,MAAM,MAAM,GAAG,MAAM,OAAO;SACzB,IAAI,CAAC,gCAAgC,EAAE,aAAa,CAAC;SACrD,MAAM,CAAC,aAAa,CAAC,CAAC;IAEzB,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;QAC9C,MAAM,IAAA,kCAAiB,EAAC,MAAM,CAAC,CAAC;IAClC,CAAC;IACD,OAAO,eAAe,CAAC,sCAAsC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AAC7E,CAAC;AAED;;GAEG;AACH,SAAgB,wBAAwB,CACtC,OAAe,EACf,UAA2C,EAAE;;IAE7C,OAAO,IAAA,wBAAY,EAAoB;QACrC,aAAa,EAAE,KAAK,IAAI,EAAE;YACxB,OAAO,yBAAyB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QACrD,CAAC;QACD,aAAa,EAAE,KAAK,EAAE,aAAgC,EAAE,EAAE;YACxD,OAAO,cAAc,CAAC,OAAO,EAAE,aAAa,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QAC5D,CAAC;QACD,kBAAkB,EAAE,qBAAqB;QACzC,YAAY,EAAE,MAAA,OAAO,CAAC,cAAc,0CAAE,iBAAiB;KACxD,CAAC,CAAC;AACL,CAAC;AAED,KAAK,UAAU,yBAAyB,CACtC,OAAe,EACf,UAA2C,EAAE;IAE7C,MAAM,aAAa,mCACd,IAAA,iDAAmC,EAAC,OAAO,CAAC,KAC/C,IAAI,EAAE,aAAa,CAAC,yBAAyB,CAAC,OAAO,CAAC,GACvD,CAAC;IAEF,mCAAmC,CAAC,aAAa,CAAC,CAAC;IACnD,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IAExE,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;QAC9C,MAAM,IAAA,kCAAiB,EAAC,MAAM,CAAC,CAAC;IAClC,CAAC;IACD,OAAO,eAAe,CAAC,wBAAwB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AAC/D,CAAC;AAED,SAAS,qBAAqB,CAAC,MAAyB;IACtD,QAAQ,MAAM,CAAC,MAAM,EAAE,CAAC;QACtB,KAAK,aAAa;YAChB,OAAO,SAAS,CAAC;QACnB,KAAK,WAAW;YACd,OAAO,WAAW,CAAC;QACrB,KAAK,SAAS;YACZ,OAAO,QAAQ,CAAC;QAClB;YACE,OAAO,QAAQ,CAAC;IACpB,CAAC;AACH,CAAC;AAED,SAAS,kCAAkC,CAAC,OAAoC;;IAC9E,IAAI,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,eAAe,0CAAE,KAAK,EAAE,CAAC;QACpC,IAAA,mCAAa,EAAC,OAAO,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;IAC/C,CAAC;IACD,IAAI,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,eAAe,0CAAE,KAAK,EAAE,CAAC;QACpC,IAAA,mCAAa,EAAC,OAAO,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;IAC/C,CAAC;AACH,CAAC;AAED,SAAS,mCAAmC,CAAC,OAAqC;;IAChF,IACE,CAAA,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,IAAI,0CAAE,iBAAiB;QAChC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC,CAAC,EAC9D,CAAC;QACD,MAAM,IAAI,KAAK,CAAC,8DAA8D,CAAC,CAAC;IAClF,CAAC;IACD,IAAI,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,IAAI,0CAAE,QAAQ,EAAE,CAAC;QAC5B,IAAA,sCAAgB,EAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC1C,CAAC;AACH,CAAC;AAED,SAAS,mCAAmC,CAAC,OAAqC;;IAChF,IAAI,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,IAAI,0CAAE,QAAQ,EAAE,CAAC;QAC5B,IAAA,sCAAgB,EAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC1C,CAAC;AACH,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport type { Client } from \"@azure-rest/core-client\";\nimport { operationOptionsToRequestParameters } from \"@azure-rest/core-client\";\nimport type {\n  ListVectorStoresParameters,\n  CreateVectorStoreParameters,\n  ModifyVectorStoreParameters,\n} from \"../generated/src/parameters.js\";\nimport type {\n  OpenAIPageableListOfVectorStoreOutput,\n  VectorStoreDeletionStatusOutput,\n  VectorStoreOutput,\n} from \"../customization/outputModels.js\";\nimport { createPoller } from \"./poller.js\";\nimport {\n  type CreateVectorStoreOptionalParams,\n  type DeleteVectorStoreOptionalParams,\n  type GetVectorStoreOptionalParams,\n  type ListVectorStoresOptionalParams,\n  type UpdateVectorStoreOptionalParams,\n} from \"./customModels.js\";\nimport {\n  validateLimit,\n  validateMetadata,\n  validateOrder,\n  validateVectorStoreId,\n} from \"./inputValidations.js\";\nimport type * as GeneratedParameters from \"../generated/src/parameters.js\";\nimport * as ConvertFromWire from \"../customization/convertOutputModelsFromWire.js\";\nimport * as ConvertToWire from \"../customization/convertModelsToWrite.js\";\nimport { convertToListQueryParameters } from \"../customization/convertParametersToWire.js\";\nimport { createOpenAIError } from \"./openAIError.js\";\nimport type { OperationState, OperationStatus, PollerLike } from \"@azure/core-lro\";\n\nconst expectedStatuses = [\"200\"];\n\n/** Returns a list of vector stores. */\nexport async function listVectorStores(\n  context: Client,\n  options: ListVectorStoresOptionalParams = {},\n): Promise<OpenAIPageableListOfVectorStoreOutput> {\n  const listOptions: GeneratedParameters.ListVectorStoresParameters = {\n    ...operationOptionsToRequestParameters(options),\n    queryParameters: convertToListQueryParameters(options),\n  };\n\n  validateListVectorStoresParameters(listOptions);\n  const result = await context.path(\"/vector_stores\").get(listOptions);\n\n  if (!expectedStatuses.includes(result.status)) {\n    throw createOpenAIError(result);\n  }\n  return ConvertFromWire.convertOpenAIPageableListOfVectorStoreOutput(result.body);\n}\n\n/** Creates a vector store. */\nexport function createVectorStore(\n  context: Client,\n  options: CreateVectorStoreOptionalParams = {},\n): PollerLike<OperationState<VectorStoreOutput>, VectorStoreOutput> {\n  return createPoller<VectorStoreOutput>({\n    initOperation: async () => {\n      return createVectorStoreInternal(context, options);\n    },\n    pollOperation: async (currentResult: VectorStoreOutput) => {\n      return getVectorStore(context, currentResult.id, options);\n    },\n    getOperationStatus: getLroOperationStatus,\n    intervalInMs: options.pollingOptions?.sleepIntervalInMs,\n  });\n}\n\n/** Returns the vector store object matching the specified ID. */\nexport async function getVectorStore(\n  context: Client,\n  vectorStoreId: string,\n  options: GetVectorStoreOptionalParams = {},\n): Promise<VectorStoreOutput> {\n  const getOptions: GeneratedParameters.GetVectorStoreParameters = {\n    ...operationOptionsToRequestParameters(options),\n  };\n\n  validateVectorStoreId(vectorStoreId);\n  const result = await context\n    .path(\"/vector_stores/{vectorStoreId}\", vectorStoreId)\n    .get(getOptions);\n\n  if (!expectedStatuses.includes(result.status)) {\n    throw createOpenAIError(result);\n  }\n  return ConvertFromWire.convertVectorStoreOutput(result.body);\n}\n\n/** The ID of the vector store to modify. */\nexport async function modifyVectorStore(\n  context: Client,\n  vectorStoreId: string,\n  options: UpdateVectorStoreOptionalParams = {},\n): Promise<VectorStoreOutput> {\n  const modifyOptions: GeneratedParameters.ModifyVectorStoreParameters = {\n    ...operationOptionsToRequestParameters(options),\n    body: ConvertToWire.convertVectorStoreUpdateOptions(options),\n  };\n\n  validateVectorStoreId(vectorStoreId);\n  validateModifyVectorStoreParameters(modifyOptions);\n  const result = await context\n    .path(\"/vector_stores/{vectorStoreId}\", vectorStoreId)\n    .post(modifyOptions);\n\n  if (!expectedStatuses.includes(result.status)) {\n    throw createOpenAIError(result);\n  }\n  return ConvertFromWire.convertVectorStoreOutput(result.body);\n}\n\n/** Deletes the vector store object matching the specified ID. */\nexport async function deleteVectorStore(\n  context: Client,\n  vectorStoreId: string,\n  options: DeleteVectorStoreOptionalParams = {},\n): Promise<VectorStoreDeletionStatusOutput> {\n  const deleteOptions: GeneratedParameters.DeleteVectorStoreParameters = {\n    ...operationOptionsToRequestParameters(options),\n  };\n\n  validateVectorStoreId(vectorStoreId);\n  const result = await context\n    .path(\"/vector_stores/{vectorStoreId}\", vectorStoreId)\n    .delete(deleteOptions);\n\n  if (!expectedStatuses.includes(result.status)) {\n    throw createOpenAIError(result);\n  }\n  return ConvertFromWire.convertVectorStoreDeletionStatusOutput(result.body);\n}\n\n/**\n * Creates a vector store and poll.\n */\nexport function createVectorStoreAndPoll(\n  context: Client,\n  options: CreateVectorStoreOptionalParams = {},\n): PollerLike<OperationState<VectorStoreOutput>, VectorStoreOutput> {\n  return createPoller<VectorStoreOutput>({\n    initOperation: async () => {\n      return createVectorStoreInternal(context, options);\n    },\n    pollOperation: async (currentResult: VectorStoreOutput) => {\n      return getVectorStore(context, currentResult.id, options);\n    },\n    getOperationStatus: getLroOperationStatus,\n    intervalInMs: options.pollingOptions?.sleepIntervalInMs,\n  });\n}\n\nasync function createVectorStoreInternal(\n  context: Client,\n  options: CreateVectorStoreOptionalParams = {},\n): Promise<VectorStoreOutput> {\n  const createOptions: GeneratedParameters.CreateVectorStoreParameters = {\n    ...operationOptionsToRequestParameters(options),\n    body: ConvertToWire.convertVectorStoreOptions(options),\n  };\n\n  validateCreateVectorStoreParameters(createOptions);\n  const result = await context.path(\"/vector_stores\").post(createOptions);\n\n  if (!expectedStatuses.includes(result.status)) {\n    throw createOpenAIError(result);\n  }\n  return ConvertFromWire.convertVectorStoreOutput(result.body);\n}\n\nfunction getLroOperationStatus(result: VectorStoreOutput): OperationStatus {\n  switch (result.status) {\n    case \"in_progress\":\n      return \"running\";\n    case \"completed\":\n      return \"succeeded\";\n    case \"expired\":\n      return \"failed\";\n    default:\n      return \"failed\";\n  }\n}\n\nfunction validateListVectorStoresParameters(options?: ListVectorStoresParameters): void {\n  if (options?.queryParameters?.limit) {\n    validateLimit(options.queryParameters.limit);\n  }\n  if (options?.queryParameters?.order) {\n    validateOrder(options.queryParameters.order);\n  }\n}\n\nfunction validateCreateVectorStoreParameters(options?: CreateVectorStoreParameters): void {\n  if (\n    options?.body?.chunking_strategy &&\n    (!options.body.file_ids || options.body.file_ids.length === 0)\n  ) {\n    throw new Error(\"Chunking strategy is only applicable if fileIds is non-empty\");\n  }\n  if (options?.body?.metadata) {\n    validateMetadata(options.body.metadata);\n  }\n}\n\nfunction validateModifyVectorStoreParameters(options?: ModifyVectorStoreParameters): void {\n  if (options?.body?.metadata) {\n    validateMetadata(options.body.metadata);\n  }\n}\n"]}

package/dist/commonjs/agents/vectorStoresFileBatches.d.ts

@@ -1,9 +1,9 @@
 import type { Client } from "@azure-rest/core-client";
 import type { OpenAIPageableListOfVectorStoreFileOutput, VectorStoreFileBatchOutput } from "../customization/outputModels.js";
-import type { CancelVectorStoreFileBatchOptionalParams, CreateVectorStoreFileBatchOptionalParams, CreateVectorStoreFileBatchWithPollingOptionalParams, GetVectorStoreFileBatchOptionalParams, ListVectorStoreFileBatchFilesOptionalParams } from "./customModels.js";
-import type { PollerLike, PollOperationState } from "@azure/core-lro";
+import type { CancelVectorStoreFileBatchOptionalParams, CreateVectorStoreFileBatchOptionalParams, GetVectorStoreFileBatchOptionalParams, ListVectorStoreFileBatchFilesOptionalParams } from "./customModels.js";
+import type { PollerLike, OperationState } from "@azure/core-lro";
 /** Create a vector store file batch. */
-export declare function createVectorStoreFileBatch(context: Client, vectorStoreId: string, options?: CreateVectorStoreFileBatchOptionalParams): Promise<VectorStoreFileBatchOutput>;
+export declare function createVectorStoreFileBatch(context: Client, vectorStoreId: string, options?: CreateVectorStoreFileBatchOptionalParams): PollerLike<OperationState<VectorStoreFileBatchOutput>, VectorStoreFileBatchOutput>;
 /** Retrieve a vector store file batch. */
 export declare function getVectorStoreFileBatch(context: Client, vectorStoreId: string, batchId: string, options?: GetVectorStoreFileBatchOptionalParams): Promise<VectorStoreFileBatchOutput>;
 /** Cancel a vector store file batch. This attempts to cancel the processing of files in this batch as soon as possible. */
@@ -11,5 +11,6 @@ export declare function cancelVectorStoreFileBatch(context: Client, vectorStoreI
 /** Returns a list of vector store files in a batch. */
 export declare function listVectorStoreFileBatchFiles(context: Client, vectorStoreId: string, batchId: string, options?: ListVectorStoreFileBatchFilesOptionalParams): Promise<OpenAIPageableListOfVectorStoreFileOutput>;
 /** Create a vector store file batch and poll. */
-export declare function createVectorStoreFileBatchAndPoll(context: Client, vectorStoreId: string, options?: CreateVectorStoreFileBatchWithPollingOptionalParams): PollerLike<PollOperationState<VectorStoreFileBatchOutput>, VectorStoreFileBatchOutput>;
+export declare function createVectorStoreFileBatchAndPoll(context: Client, vectorStoreId: string, options?: CreateVectorStoreFileBatchOptionalParams): PollerLike<OperationState<VectorStoreFileBatchOutput>, VectorStoreFileBatchOutput>;
+export declare function createVectorStoreFileBatchInternal(context: Client, vectorStoreId: string, options?: CreateVectorStoreFileBatchOptionalParams): Promise<VectorStoreFileBatchOutput>;
 //# sourceMappingURL=vectorStoresFileBatches.d.ts.map

package/dist/commonjs/agents/vectorStoresFileBatches.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"vectorStoresFileBatches.d.ts","sourceRoot":"","sources":["../../../src/agents/vectorStoresFileBatches.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,yBAAyB,CAAC;AAEtD,OAAO,KAAK,EACV,yCAAyC,EACzC,0BAA0B,EAC3B,MAAM,kCAAkC,CAAC;AAE1C,OAAO,KAAK,EACV,wCAAwC,EACxC,wCAAwC,EACxC,mDAAmD,EACnD,qCAAqC,EACrC,2CAA2C,EAC5C,MAAM,mBAAmB,CAAC;AAc3B,OAAO,KAAK,EAAE,UAAU,EAAE,kBAAkB,EAAE,MAAM,iBAAiB,CAAC;AAItE,wCAAwC;AACxC,wBAAsB,0BAA0B,CAC9C,OAAO,EAAE,MAAM,EACf,aAAa,EAAE,MAAM,EACrB,OAAO,GAAE,wCAA6C,GACrD,OAAO,CAAC,0BAA0B,CAAC,CAerC;AAED,0CAA0C;AAC1C,wBAAsB,uBAAuB,CAC3C,OAAO,EAAE,MAAM,EACf,aAAa,EAAE,MAAM,EACrB,OAAO,EAAE,MAAM,EACf,OAAO,GAAE,qCAA0C,GAClD,OAAO,CAAC,0BAA0B,CAAC,CASrC;AAED,2HAA2H;AAC3H,wBAAsB,0BAA0B,CAC9C,OAAO,EAAE,MAAM,EACf,aAAa,EAAE,MAAM,EACrB,OAAO,EAAE,MAAM,EACf,OAAO,GAAE,wCAA6C,GACrD,OAAO,CAAC,0BAA0B,CAAC,CASrC;AAED,uDAAuD;AACvD,wBAAsB,6BAA6B,CACjD,OAAO,EAAE,MAAM,EACf,aAAa,EAAE,MAAM,EACrB,OAAO,EAAE,MAAM,EACf,OAAO,GAAE,2CAAgD,GACxD,OAAO,CAAC,yCAAyC,CAAC,CAkBpD;AAED,iDAAiD;AACjD,wBAAgB,iCAAiC,CAC/C,OAAO,EAAE,MAAM,EACf,aAAa,EAAE,MAAM,EACrB,OAAO,GAAE,mDAAwD,GAChE,UAAU,CAAC,kBAAkB,CAAC,0BAA0B,CAAC,EAAE,0BAA0B,CAAC,CAiCxF"}
+{"version":3,"file":"vectorStoresFileBatches.d.ts","sourceRoot":"","sources":["../../../src/agents/vectorStoresFileBatches.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,yBAAyB,CAAC;AAEtD,OAAO,KAAK,EACV,yCAAyC,EACzC,0BAA0B,EAC3B,MAAM,kCAAkC,CAAC;AAE1C,OAAO,KAAK,EACV,wCAAwC,EACxC,wCAAwC,EACxC,qCAAqC,EACrC,2CAA2C,EAC5C,MAAM,mBAAmB,CAAC;AAc3B,OAAO,KAAK,EAAE,UAAU,EAAE,cAAc,EAAmB,MAAM,iBAAiB,CAAC;AAInF,wCAAwC;AACxC,wBAAgB,0BAA0B,CACxC,OAAO,EAAE,MAAM,EACf,aAAa,EAAE,MAAM,EACrB,OAAO,GAAE,wCAA6C,GACrD,UAAU,CAAC,cAAc,CAAC,0BAA0B,CAAC,EAAE,0BAA0B,CAAC,CAWpF;AAED,0CAA0C;AAC1C,wBAAsB,uBAAuB,CAC3C,OAAO,EAAE,MAAM,EACf,aAAa,EAAE,MAAM,EACrB,OAAO,EAAE,MAAM,EACf,OAAO,GAAE,qCAA0C,GAClD,OAAO,CAAC,0BAA0B,CAAC,CASrC;AAED,2HAA2H;AAC3H,wBAAsB,0BAA0B,CAC9C,OAAO,EAAE,MAAM,EACf,aAAa,EAAE,MAAM,EACrB,OAAO,EAAE,MAAM,EACf,OAAO,GAAE,wCAA6C,GACrD,OAAO,CAAC,0BAA0B,CAAC,CASrC;AAED,uDAAuD;AACvD,wBAAsB,6BAA6B,CACjD,OAAO,EAAE,MAAM,EACf,aAAa,EAAE,MAAM,EACrB,OAAO,EAAE,MAAM,EACf,OAAO,GAAE,2CAAgD,GACxD,OAAO,CAAC,yCAAyC,CAAC,CAkBpD;AAED,iDAAiD;AACjD,wBAAgB,iCAAiC,CAC/C,OAAO,EAAE,MAAM,EACf,aAAa,EAAE,MAAM,EACrB,OAAO,GAAE,wCAA6C,GACrD,UAAU,CAAC,cAAc,CAAC,0BAA0B,CAAC,EAAE,0BAA0B,CAAC,CAWpF;AAED,wBAAsB,kCAAkC,CACtD,OAAO,EAAE,MAAM,EACf,aAAa,EAAE,MAAM,EACrB,OAAO,GAAE,wCAA6C,GACrD,OAAO,CAAC,0BAA0B,CAAC,CAerC"}