@azure/ai-projects 1.0.0-beta.1

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

@@ -0,0 +1,141 @@
+import type { AzureAISearchToolDefinition, CodeInterpreterToolDefinition, FileSearchToolDefinition, FileSearchToolDefinitionDetails, FunctionDefinition, FunctionToolDefinition, RequiredActionOutput, RequiredToolCallOutput, ToolDefinition, ToolDefinitionOutputParent, ToolResources, VectorStoreConfigurations, VectorStoreDataSource } 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.
+ * @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.
+ */
+export declare function isOutputOfType<T extends {
+    type: string;
+}>(output: RequiredActionOutput | RequiredToolCallOutput | ToolDefinitionOutputParent, type: string): output is T;
+/** Types of connection tools used to configure an agent */
+export declare enum connectionToolType {
+    /** Bing grounding search tool */
+    BingGrounding = "bing_grounding",
+    /** Microsoft Fabric tool */
+    MicrosoftFabric = "microsoft_fabric",
+    /** Sharepoint tool */
+    SharepointGrounding = "sharepoint_grounding"
+}
+/**
+ * Utility class for creating various tools.
+ */
+export declare class ToolUtility {
+    /**
+     * Creates a connection tool
+     *
+     * @param toolType - The type of the connection tool.
+     * @param connectionIds - A list of the IDs of the connections to use.
+     * @returns An object containing the definition for the connection tool
+     */
+    static createConnectionTool(toolType: connectionToolType, connectionIds: string[]): {
+        definition: ToolDefinition;
+    };
+    /**
+     * Creates a file search tool
+     *
+     * @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.
+     * @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.
+     * @param definitionDetails - The input definition information for a file search tool as used to configure an agent.
+     *
+     * @returns An object containing the definition and resources for the file search tool
+     */
+    static createFileSearchTool(vectorStoreIds?: string[], vectorStores?: Array<VectorStoreConfigurations>, definitionDetails?: FileSearchToolDefinitionDetails): {
+        definition: FileSearchToolDefinition;
+        resources: ToolResources;
+    };
+    /**
+     * Creates a code interpreter tool
+     *
+     * @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.
+     * @param dataSources - The data sources to be used. This option is mutually exclusive with fileIds.
+     *
+     * @returns An object containing the definition and resources for the code interpreter tool.
+     */
+    static createCodeInterpreterTool(fileIds?: string[], dataSources?: Array<VectorStoreDataSource>): {
+        definition: CodeInterpreterToolDefinition;
+        resources: ToolResources;
+    };
+    /**
+     * Creates an Azure AI search tool
+     *
+     * @param indexConnectionId - The connection ID of the Azure AI search index.
+     * @param indexName - The name of the Azure AI search index.
+     *
+     * @returns An object containing the definition and resources for the Azure AI search tool.
+     */
+    static createAzureAISearchTool(indexConnectionId: string, indexName: string): {
+        definition: AzureAISearchToolDefinition;
+        resources: ToolResources;
+    };
+    /**
+     * Creates a function tool
+     *
+     * @param functionDefinition - The function definition to use.
+     *
+     * @returns An object containing the definition for the function tool.
+     */
+    static createFunctionTool(functionDefinition: FunctionDefinition): {
+        definition: FunctionToolDefinition;
+    };
+}
+/**
+ * Represents a set of tools with their definitions and resources.
+ */
+export declare class ToolSet {
+    /** A list of tool definitions that have been added to the tool set. */
+    toolDefinitions: ToolDefinition[];
+    /** A collection of resources associated with the tools in the tool set. */
+    toolResources: ToolResources;
+    /**
+     * Adds a connection tool to the tool set.
+     *
+     * @param toolType - The type of the connection tool.
+     * @param connectionIds - A list of the IDs of the connections to use.
+     *
+     * @returns An object containing the definition for the connection tool
+     */
+    addConnectionTool(toolType: connectionToolType, connectionIds: string[]): {
+        definition: ToolDefinition;
+    };
+    /**
+     * Adds a file search tool to the tool set.
+     *
+     * @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.
+     * @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.
+     * @param definitionDetails - The input definition information for a file search tool as used to configure an agent.
+     *
+     * @returns An object containing the definition and resources for the file search tool
+     */
+    addFileSearchTool(vectorStoreIds?: string[], vectorStores?: Array<VectorStoreConfigurations>, definitionDetails?: FileSearchToolDefinitionDetails): {
+        definition: FileSearchToolDefinition;
+        resources: ToolResources;
+    };
+    /**
+     * Adds a code interpreter tool to the tool set.
+     *
+     * @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.
+     * @param dataSources - The data sources to be used. This option is mutually exclusive with fileIds.
+     *
+     * @returns An object containing the definition and resources for the code interpreter tool
+     */
+    addCodeInterpreterTool(fileIds?: string[], dataSources?: Array<VectorStoreDataSource>): {
+        definition: CodeInterpreterToolDefinition;
+        resources: ToolResources;
+    };
+    /**
+     * Adds an Azure AI search tool to the tool set.
+     *
+     * @param indexConnectionId - The connection ID of the Azure AI search index.
+     * @param indexName - The name of the Azure AI search index.
+     *
+     * @returns An object containing the definition and resources for the Azure AI search tool
+     */
+    addAzureAISearchTool(indexConnectionId: string, indexName: string): {
+        definition: AzureAISearchToolDefinition;
+        resources: ToolResources;
+    };
+}
+//# sourceMappingURL=utils.d.ts.map

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

@@ -0,0 +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;AAED;;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"}

package/dist/commonjs/agents/utils.js

@@ -0,0 +1,184 @@
+"use strict";
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ToolSet = exports.ToolUtility = exports.connectionToolType = void 0;
+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.
+ * @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.
+ */
+function isOutputOfType(output, type) {
+    return output.type === type;
+}
+/** Types of connection tools used to configure an agent */
+var connectionToolType;
+(function (connectionToolType) {
+    /** Bing grounding search tool */
+    connectionToolType["BingGrounding"] = "bing_grounding";
+    /** Microsoft Fabric tool */
+    connectionToolType["MicrosoftFabric"] = "microsoft_fabric";
+    /** Sharepoint tool */
+    connectionToolType["SharepointGrounding"] = "sharepoint_grounding";
+})(connectionToolType || (exports.connectionToolType = connectionToolType = {}));
+/**
+ * Utility class for creating various tools.
+ */
+class ToolUtility {
+    /**
+     * Creates a connection tool
+     *
+     * @param toolType - The type of the connection tool.
+     * @param connectionIds - A list of the IDs of the connections to use.
+     * @returns An object containing the definition for the connection tool
+     */
+    static createConnectionTool(toolType, connectionIds) {
+        return {
+            definition: {
+                type: toolType,
+                [toolType]: {
+                    connections: connectionIds.map((connectionId) => ({ connectionId: connectionId })),
+                },
+            },
+        };
+    }
+    /**
+     * Creates a file search tool
+     *
+     * @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.
+     * @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.
+     * @param definitionDetails - The input definition information for a file search tool as used to configure an agent.
+     *
+     * @returns An object containing the definition and resources for the file search tool
+     */
+    static createFileSearchTool(vectorStoreIds, vectorStores, definitionDetails) {
+        return {
+            definition: { type: "file_search", fileSearch: definitionDetails },
+            resources: { fileSearch: { vectorStoreIds: vectorStoreIds, vectorStores: vectorStores } },
+        };
+    }
+    /**
+     * Creates a code interpreter tool
+     *
+     * @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.
+     * @param dataSources - The data sources to be used. This option is mutually exclusive with fileIds.
+     *
+     * @returns An object containing the definition and resources for the code interpreter tool.
+     */
+    static createCodeInterpreterTool(fileIds, dataSources) {
+        if (fileIds && dataSources) {
+            throw new Error("Cannot specify both fileIds and dataSources");
+        }
+        return {
+            definition: { type: "code_interpreter" },
+            resources: { codeInterpreter: { fileIds: fileIds, dataSources: dataSources } },
+        };
+    }
+    /**
+     * Creates an Azure AI search tool
+     *
+     * @param indexConnectionId - The connection ID of the Azure AI search index.
+     * @param indexName - The name of the Azure AI search index.
+     *
+     * @returns An object containing the definition and resources for the Azure AI search tool.
+     */
+    static createAzureAISearchTool(indexConnectionId, indexName) {
+        return {
+            definition: { type: "azure_ai_search" },
+            resources: {
+                azureAISearch: {
+                    indexes: [{ indexConnectionId: indexConnectionId, indexName: indexName }],
+                },
+            },
+        };
+    }
+    /**
+     * Creates a function tool
+     *
+     * @param functionDefinition - The function definition to use.
+     *
+     * @returns An object containing the definition for the function tool.
+     */
+    static createFunctionTool(functionDefinition) {
+        return {
+            definition: {
+                type: "function",
+                function: functionDefinition,
+            },
+        };
+    }
+}
+exports.ToolUtility = ToolUtility;
+/**
+ * Represents a set of tools with their definitions and resources.
+ */
+class ToolSet {
+    constructor() {
+        /** A list of tool definitions that have been added to the tool set. */
+        this.toolDefinitions = [];
+        /** A collection of resources associated with the tools in the tool set. */
+        this.toolResources = {};
+    }
+    /**
+     * Adds a connection tool to the tool set.
+     *
+     * @param toolType - The type of the connection tool.
+     * @param connectionIds - A list of the IDs of the connections to use.
+     *
+     * @returns An object containing the definition for the connection tool
+     */
+    addConnectionTool(toolType, connectionIds) {
+        const tool = ToolUtility.createConnectionTool(toolType, connectionIds);
+        this.toolDefinitions.push(tool.definition);
+        return tool;
+    }
+    /**
+     * Adds a file search tool to the tool set.
+     *
+     * @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.
+     * @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.
+     * @param definitionDetails - The input definition information for a file search tool as used to configure an agent.
+     *
+     * @returns An object containing the definition and resources for the file search tool
+     */
+    addFileSearchTool(vectorStoreIds, vectorStores, definitionDetails) {
+        const tool = ToolUtility.createFileSearchTool(vectorStoreIds, vectorStores, definitionDetails);
+        this.toolDefinitions.push(tool.definition);
+        this.toolResources = Object.assign(Object.assign({}, this.toolResources), tool.resources);
+        return tool;
+    }
+    /**
+     * Adds a code interpreter tool to the tool set.
+     *
+     * @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.
+     * @param dataSources - The data sources to be used. This option is mutually exclusive with fileIds.
+     *
+     * @returns An object containing the definition and resources for the code interpreter tool
+     */
+    addCodeInterpreterTool(fileIds, dataSources) {
+        const tool = ToolUtility.createCodeInterpreterTool(fileIds, dataSources);
+        this.toolDefinitions.push(tool.definition);
+        this.toolResources = Object.assign(Object.assign({}, this.toolResources), tool.resources);
+        return tool;
+    }
+    /**
+     * Adds an Azure AI search tool to the tool set.
+     *
+     * @param indexConnectionId - The connection ID of the Azure AI search index.
+     * @param indexName - The name of the Azure AI search index.
+     *
+     * @returns An object containing the definition and resources for the Azure AI search tool
+     */
+    addAzureAISearchTool(indexConnectionId, indexName) {
+        const tool = ToolUtility.createAzureAISearchTool(indexConnectionId, indexName);
+        this.toolDefinitions.push(tool.definition);
+        this.toolResources = Object.assign(Object.assign({}, this.toolResources), tool.resources);
+        return tool;
+    }
+}
+exports.ToolSet = ToolSet;
+//# sourceMappingURL=utils.js.map

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

@@ -0,0 +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;;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,QAAQ,CAAC,EAAE;oBACV,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\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        [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"]}

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

@@ -0,0 +1,20 @@
+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";
+/** 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>;
+/** 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. */
+export declare function modifyVectorStore(context: Client, vectorStoreId: string, options?: UpdateVectorStoreOptionalParams): Promise<VectorStoreOutput>;
+/** Deletes the vector store object matching the specified ID. */
+export declare function deleteVectorStore(context: Client, vectorStoreId: string, options?: DeleteVectorStoreOptionalParams): Promise<VectorStoreDeletionStatusOutput>;
+/**
+ * Creates a vector store and poll.
+ */
+export declare function createVectorStoreAndPoll(context: Client, options?: CreateVectorStoreWithPollingOptionalParams): PollerLike<PollOperationState<VectorStoreOutput>, VectorStoreOutput>;
+//# sourceMappingURL=vectorStores.d.ts.map

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

@@ -0,0 +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"}

package/dist/commonjs/agents/vectorStores.js

@@ -0,0 +1,125 @@
+"use strict";
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.listVectorStores = listVectorStores;
+exports.createVectorStore = createVectorStore;
+exports.getVectorStore = getVectorStore;
+exports.modifyVectorStore = modifyVectorStore;
+exports.deleteVectorStore = deleteVectorStore;
+exports.createVectorStoreAndPoll = createVectorStoreAndPoll;
+const tslib_1 = require("tslib");
+const core_client_1 = require("@azure-rest/core-client");
+const poller_js_1 = require("./poller.js");
+const inputValidations_js_1 = require("./inputValidations.js");
+const ConvertFromWire = tslib_1.__importStar(require("../customization/convertOutputModelsFromWire.js"));
+const ConvertToWire = tslib_1.__importStar(require("../customization/convertModelsToWrite.js"));
+const convertParametersToWire_js_1 = require("../customization/convertParametersToWire.js");
+const openAIError_js_1 = require("./openAIError.js");
+const expectedStatuses = ["200"];
+/** Returns a list of vector stores. */
+async function listVectorStores(context, options = {}) {
+    const listOptions = Object.assign(Object.assign({}, (0, core_client_1.operationOptionsToRequestParameters)(options)), { queryParameters: (0, convertParametersToWire_js_1.convertToListQueryParameters)(options) });
+    validateListVectorStoresParameters(listOptions);
+    const result = await context.path("/vector_stores").get(listOptions);
+    if (!expectedStatuses.includes(result.status)) {
+        throw (0, openAIError_js_1.createOpenAIError)(result);
+    }
+    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);
+}
+/** Returns the vector store object matching the specified ID. */
+async function getVectorStore(context, vectorStoreId, options = {}) {
+    const getOptions = Object.assign({}, (0, core_client_1.operationOptionsToRequestParameters)(options));
+    (0, inputValidations_js_1.validateVectorStoreId)(vectorStoreId);
+    const result = await context
+        .path("/vector_stores/{vectorStoreId}", vectorStoreId)
+        .get(getOptions);
+    if (!expectedStatuses.includes(result.status)) {
+        throw (0, openAIError_js_1.createOpenAIError)(result);
+    }
+    return ConvertFromWire.convertVectorStoreOutput(result.body);
+}
+/** The ID of the vector store to modify. */
+async function modifyVectorStore(context, vectorStoreId, options = {}) {
+    const modifyOptions = Object.assign(Object.assign({}, (0, core_client_1.operationOptionsToRequestParameters)(options)), { body: ConvertToWire.convertVectorStoreUpdateOptions(options) });
+    (0, inputValidations_js_1.validateVectorStoreId)(vectorStoreId);
+    validateModifyVectorStoreParameters(modifyOptions);
+    const result = await context
+        .path("/vector_stores/{vectorStoreId}", vectorStoreId)
+        .post(modifyOptions);
+    if (!expectedStatuses.includes(result.status)) {
+        throw (0, openAIError_js_1.createOpenAIError)(result);
+    }
+    return ConvertFromWire.convertVectorStoreOutput(result.body);
+}
+/** Deletes the vector store object matching the specified ID. */
+async function deleteVectorStore(context, vectorStoreId, options = {}) {
+    const deleteOptions = Object.assign({}, (0, core_client_1.operationOptionsToRequestParameters)(options));
+    (0, inputValidations_js_1.validateVectorStoreId)(vectorStoreId);
+    const result = await context
+        .path("/vector_stores/{vectorStoreId}", vectorStoreId)
+        .delete(deleteOptions);
+    if (!expectedStatuses.includes(result.status)) {
+        throw (0, openAIError_js_1.createOpenAIError)(result);
+    }
+    return ConvertFromWire.convertVectorStoreDeletionStatusOutput(result.body);
+}
+/**
+ * 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,
+    });
+}
+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) {
+        (0, inputValidations_js_1.validateLimit)(options.queryParameters.limit);
+    }
+    if ((_b = options === null || options === void 0 ? void 0 : options.queryParameters) === null || _b === void 0 ? void 0 : _b.order) {
+        (0, inputValidations_js_1.validateOrder)(options.queryParameters.order);
+    }
+}
+function validateCreateVectorStoreParameters(options) {
+    var _a, _b;
+    if (((_a = options === null || options === void 0 ? void 0 : options.body) === null || _a === void 0 ? void 0 : _a.chunking_strategy) &&
+        (!options.body.file_ids || options.body.file_ids.length === 0)) {
+        throw new Error("Chunking strategy is only applicable if fileIds is non-empty");
+    }
+    if ((_b = options === null || options === void 0 ? void 0 : options.body) === null || _b === void 0 ? void 0 : _b.metadata) {
+        (0, inputValidations_js_1.validateMetadata)(options.body.metadata);
+    }
+}
+function validateModifyVectorStoreParameters(options) {
+    var _a;
+    if ((_a = options === null || options === void 0 ? void 0 : options.body) === null || _a === void 0 ? void 0 : _a.metadata) {
+        (0, inputValidations_js_1.validateMetadata)(options.body.metadata);
+    }
+}
+//# sourceMappingURL=vectorStores.js.map

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

@@ -0,0 +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"]}

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

@@ -0,0 +1,15 @@
+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";
+/** Create a vector store file batch. */
+export declare function createVectorStoreFileBatch(context: Client, vectorStoreId: string, options?: CreateVectorStoreFileBatchOptionalParams): Promise<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. */
+export declare function cancelVectorStoreFileBatch(context: Client, vectorStoreId: string, batchId: string, options?: CancelVectorStoreFileBatchOptionalParams): Promise<VectorStoreFileBatchOutput>;
+/** 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>;
+//# sourceMappingURL=vectorStoresFileBatches.d.ts.map

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

@@ -0,0 +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"}