@azure/ai-projects 1.0.0-beta.1 → 1.0.0-beta.3

package/dist/browser/agents/utils.js

@@ -3,7 +3,7 @@
 /**
  * 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.
@@ -20,7 +20,15 @@ export var connectionToolType;
     connectionToolType["MicrosoftFabric"] = "microsoft_fabric";
     /** Sharepoint tool */
     connectionToolType["SharepointGrounding"] = "sharepoint_grounding";
+    /** Azure Function tool */
+    connectionToolType["AzureFunction"] = "azure_function";
 })(connectionToolType || (connectionToolType = {}));
+const toolMap = {
+    bing_grounding: "bingGrounding",
+    microsoft_fabric: "microsoftFabric",
+    sharepoint_grounding: "sharepointGrounding",
+    azure_function: "azureFunction",
+};
 /**
  * Utility class for creating various tools.
  */
@@ -36,7 +44,7 @@ export class ToolUtility {
         return {
             definition: {
                 type: toolType,
-                [toolType]: {
+                [toolMap[toolType]]: {
                     connections: connectionIds.map((connectionId) => ({ connectionId: connectionId })),
                 },
             },
@@ -74,6 +82,29 @@ export 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
      *
@@ -107,6 +138,21 @@ export 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,
+            },
+        };
+    }
 }
 /**
  * Represents a set of tools with their definitions and resources.
@@ -174,5 +220,34 @@ export 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;
+    }
 }
 //# sourceMappingURL=utils.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../../src/agents/utils.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAkBlC;;;;;;;GAOG;AACH,MAAM,UAAU,cAAc,CAC5B,MAAkF,EAClF,IAAY;IAEZ,OAAO,MAAM,CAAC,IAAI,KAAK,IAAI,CAAC;AAC9B,CAAC;AAED,2DAA2D;AAC3D,MAAM,CAAN,IAAY,kBAOX;AAPD,WAAY,kBAAkB;IAC5B,iCAAiC;IACjC,sDAAgC,CAAA;IAChC,4BAA4B;IAC5B,0DAAoC,CAAA;IACpC,sBAAsB;IACtB,kEAA4C,CAAA;AAC9C,CAAC,EAPW,kBAAkB,KAAlB,kBAAkB,QAO7B;AAED;;GAEG;AACH,MAAM,OAAO,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;AAED;;GAEG;AACH,MAAM,OAAO,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","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"]}
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../../src/agents/utils.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAuBlC;;;;;;;GAOG;AACH,MAAM,UAAU,cAAc,CAC5B,MAAkF,EAClF,IAAY;IAEZ,OAAO,MAAM,CAAC,IAAI,KAAK,IAAI,CAAC;AAC9B,CAAC;AAED,2DAA2D;AAC3D,MAAM,CAAN,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,KAAlB,kBAAkB,QAS7B;AAED,MAAM,OAAO,GAAG;IACd,cAAc,EAAE,eAAe;IAC/B,gBAAgB,EAAE,iBAAiB;IACnC,oBAAoB,EAAE,qBAAqB;IAC3C,cAAc,EAAE,eAAe;CAChC,CAAC;AAEF;;GAEG;AACH,MAAM,OAAO,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;;;;;;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;AAED;;GAEG;AACH,MAAM,OAAO,OAAO;IAApB;QACE,uEAAuE;QACvE,oBAAe,GAAqB,EAAE,CAAC;QAEvC,2EAA2E;QAC3E,kBAAa,GAAkB,EAAE,CAAC;IAyHpC,CAAC;IAvHC;;;;;;;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;CACF","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 = \"microsoft_fabric\",\n  /** Sharepoint tool */\n  SharepointGrounding = \"sharepoint_grounding\",\n  /** Azure Function tool */\n  AzureFunction = \"azure_function\",\n}\n\nconst toolMap = {\n  bing_grounding: \"bingGrounding\",\n  microsoft_fabric: \"microsoftFabric\",\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 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"]}

package/dist/browser/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/browser/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/browser/agents/vectorStores.js

@@ -1,7 +1,7 @@
 // Copyright (c) Microsoft Corporation.
 // Licensed under the MIT License.
 import { operationOptionsToRequestParameters } from "@azure-rest/core-client";
-import { AgentsPoller } from "./poller.js";
+import { createPoller } from "./poller.js";
 import { validateLimit, validateMetadata, validateOrder, validateVectorStoreId, } from "./inputValidations.js";
 import * as ConvertFromWire from "../customization/convertOutputModelsFromWire.js";
 import * as ConvertToWire from "../customization/convertModelsToWrite.js";
@@ -19,14 +19,18 @@ export async function listVectorStores(context, options = {}) {
     return ConvertFromWire.convertOpenAIPageableListOfVectorStoreOutput(result.body);
 }
 /** Creates a vector store. */
-export async function createVectorStore(context, options = {}) {
-    const createOptions = Object.assign(Object.assign({}, operationOptionsToRequestParameters(options)), { body: ConvertToWire.convertVectorStoreOptions(options) });
-    validateCreateVectorStoreParameters(createOptions);
-    const result = await context.path("/vector_stores").post(createOptions);
-    if (!expectedStatuses.includes(result.status)) {
-        throw createOpenAIError(result);
-    }
-    return ConvertFromWire.convertVectorStoreOutput(result.body);
+export function createVectorStore(context, options = {}) {
+    var _a;
+    return 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. */
 export async function getVectorStore(context, vectorStoreId, options = {}) {
@@ -69,25 +73,39 @@ export async function deleteVectorStore(context, vectorStoreId, options = {}) {
  * Creates a vector store and poll.
  */
 export function createVectorStoreAndPoll(context, options = {}) {
-    async function updateCreateVectorStorePoll(currentResult) {
-        let vectorStore;
-        if (!currentResult) {
-            vectorStore = await createVectorStore(context, options);
-        }
-        else {
-            const getOptions = Object.assign({}, operationOptionsToRequestParameters(options));
-            vectorStore = await getVectorStore(context, currentResult.id, getOptions);
-        }
-        return {
-            result: vectorStore,
-            completed: vectorStore.status !== "in_progress",
-        };
-    }
-    return new AgentsPoller({
-        update: updateCreateVectorStorePoll,
-        pollingOptions: options.pollingOptions,
+    var _a;
+    return 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({}, operationOptionsToRequestParameters(options)), { body: ConvertToWire.convertVectorStoreOptions(options) });
+    validateCreateVectorStoreParameters(createOptions);
+    const result = await context.path("/vector_stores").post(createOptions);
+    if (!expectedStatuses.includes(result.status)) {
+        throw 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/browser/agents/vectorStores.js.map

@@ -1 +1 @@
-{"version":3,"file":"vectorStores.js","sourceRoot":"","sources":["../../../src/agents/vectorStores.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAGlC,OAAO,EAAE,mCAAmC,EAAE,MAAM,yBAAyB,CAAC;AAW9E,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAS3C,OAAO,EACL,aAAa,EACb,gBAAgB,EAChB,aAAa,EACb,qBAAqB,GACtB,MAAM,uBAAuB,CAAC;AAE/B,OAAO,KAAK,eAAe,MAAM,iDAAiD,CAAC;AACnF,OAAO,KAAK,aAAa,MAAM,0CAA0C,CAAC;AAC1E,OAAO,EAAE,4BAA4B,EAAE,MAAM,6CAA6C,CAAC;AAC3F,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AAGrD,MAAM,gBAAgB,GAAG,CAAC,KAAK,CAAC,CAAC;AAEjC,uCAAuC;AACvC,MAAM,CAAC,KAAK,UAAU,gBAAgB,CACpC,OAAe,EACf,UAA0C,EAAE;IAE5C,MAAM,WAAW,mCACZ,mCAAmC,CAAC,OAAO,CAAC,KAC/C,eAAe,EAAE,4BAA4B,CAAC,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,iBAAiB,CAAC,MAAM,CAAC,CAAC;IAClC,CAAC;IACD,OAAO,eAAe,CAAC,4CAA4C,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AACnF,CAAC;AAED,8BAA8B;AAC9B,MAAM,CAAC,KAAK,UAAU,iBAAiB,CACrC,OAAe,EACf,UAA2C,EAAE;IAE7C,MAAM,aAAa,mCACd,mCAAmC,CAAC,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,iBAAiB,CAAC,MAAM,CAAC,CAAC;IAClC,CAAC;IACD,OAAO,eAAe,CAAC,wBAAwB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AAC/D,CAAC;AAED,iEAAiE;AACjE,MAAM,CAAC,KAAK,UAAU,cAAc,CAClC,OAAe,EACf,aAAqB,EACrB,UAAwC,EAAE;IAE1C,MAAM,UAAU,qBACX,mCAAmC,CAAC,OAAO,CAAC,CAChD,CAAC;IAEF,qBAAqB,CAAC,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,iBAAiB,CAAC,MAAM,CAAC,CAAC;IAClC,CAAC;IACD,OAAO,eAAe,CAAC,wBAAwB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AAC/D,CAAC;AAED,4CAA4C;AAC5C,MAAM,CAAC,KAAK,UAAU,iBAAiB,CACrC,OAAe,EACf,aAAqB,EACrB,UAA2C,EAAE;IAE7C,MAAM,aAAa,mCACd,mCAAmC,CAAC,OAAO,CAAC,KAC/C,IAAI,EAAE,aAAa,CAAC,+BAA+B,CAAC,OAAO,CAAC,GAC7D,CAAC;IAEF,qBAAqB,CAAC,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,iBAAiB,CAAC,MAAM,CAAC,CAAC;IAClC,CAAC;IACD,OAAO,eAAe,CAAC,wBAAwB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AAC/D,CAAC;AAED,iEAAiE;AACjE,MAAM,CAAC,KAAK,UAAU,iBAAiB,CACrC,OAAe,EACf,aAAqB,EACrB,UAA2C,EAAE;IAE7C,MAAM,aAAa,qBACd,mCAAmC,CAAC,OAAO,CAAC,CAChD,CAAC;IAEF,qBAAqB,CAAC,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,iBAAiB,CAAC,MAAM,CAAC,CAAC;IAClC,CAAC;IACD,OAAO,eAAe,CAAC,sCAAsC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AAC7E,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,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,mCAAmC,CAAC,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,YAAY,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,aAAa,CAAC,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,aAAa,CAAC,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,gBAAgB,CAAC,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,gBAAgB,CAAC,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;AAGlC,OAAO,EAAE,mCAAmC,EAAE,MAAM,yBAAyB,CAAC;AAW9E,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAQ3C,OAAO,EACL,aAAa,EACb,gBAAgB,EAChB,aAAa,EACb,qBAAqB,GACtB,MAAM,uBAAuB,CAAC;AAE/B,OAAO,KAAK,eAAe,MAAM,iDAAiD,CAAC;AACnF,OAAO,KAAK,aAAa,MAAM,0CAA0C,CAAC;AAC1E,OAAO,EAAE,4BAA4B,EAAE,MAAM,6CAA6C,CAAC;AAC3F,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AAGrD,MAAM,gBAAgB,GAAG,CAAC,KAAK,CAAC,CAAC;AAEjC,uCAAuC;AACvC,MAAM,CAAC,KAAK,UAAU,gBAAgB,CACpC,OAAe,EACf,UAA0C,EAAE;IAE5C,MAAM,WAAW,mCACZ,mCAAmC,CAAC,OAAO,CAAC,KAC/C,eAAe,EAAE,4BAA4B,CAAC,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,iBAAiB,CAAC,MAAM,CAAC,CAAC;IAClC,CAAC;IACD,OAAO,eAAe,CAAC,4CAA4C,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AACnF,CAAC;AAED,8BAA8B;AAC9B,MAAM,UAAU,iBAAiB,CAC/B,OAAe,EACf,UAA2C,EAAE;;IAE7C,OAAO,YAAY,CAAoB;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;AACjE,MAAM,CAAC,KAAK,UAAU,cAAc,CAClC,OAAe,EACf,aAAqB,EACrB,UAAwC,EAAE;IAE1C,MAAM,UAAU,qBACX,mCAAmC,CAAC,OAAO,CAAC,CAChD,CAAC;IAEF,qBAAqB,CAAC,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,iBAAiB,CAAC,MAAM,CAAC,CAAC;IAClC,CAAC;IACD,OAAO,eAAe,CAAC,wBAAwB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AAC/D,CAAC;AAED,4CAA4C;AAC5C,MAAM,CAAC,KAAK,UAAU,iBAAiB,CACrC,OAAe,EACf,aAAqB,EACrB,UAA2C,EAAE;IAE7C,MAAM,aAAa,mCACd,mCAAmC,CAAC,OAAO,CAAC,KAC/C,IAAI,EAAE,aAAa,CAAC,+BAA+B,CAAC,OAAO,CAAC,GAC7D,CAAC;IAEF,qBAAqB,CAAC,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,iBAAiB,CAAC,MAAM,CAAC,CAAC;IAClC,CAAC;IACD,OAAO,eAAe,CAAC,wBAAwB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AAC/D,CAAC;AAED,iEAAiE;AACjE,MAAM,CAAC,KAAK,UAAU,iBAAiB,CACrC,OAAe,EACf,aAAqB,EACrB,UAA2C,EAAE;IAE7C,MAAM,aAAa,qBACd,mCAAmC,CAAC,OAAO,CAAC,CAChD,CAAC;IAEF,qBAAqB,CAAC,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,iBAAiB,CAAC,MAAM,CAAC,CAAC;IAClC,CAAC;IACD,OAAO,eAAe,CAAC,sCAAsC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AAC7E,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,wBAAwB,CACtC,OAAe,EACf,UAA2C,EAAE;;IAE7C,OAAO,YAAY,CAAoB;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,mCAAmC,CAAC,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,iBAAiB,CAAC,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,aAAa,CAAC,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,aAAa,CAAC,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,gBAAgB,CAAC,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,gBAAgB,CAAC,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/browser/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/browser/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"}

package/dist/browser/agents/vectorStoresFileBatches.js

@@ -1,24 +1,25 @@
 // Copyright (c) Microsoft Corporation.
 // Licensed under the MIT License.
 import { operationOptionsToRequestParameters } from "@azure-rest/core-client";
-import { AgentsPoller } from "./poller.js";
+import { createPoller } from "./poller.js";
 import { validateFileStatusFilter, validateLimit, validateOrder, validateVectorStoreId, } from "./inputValidations.js";
 import * as ConvertFromWire from "../customization/convertOutputModelsFromWire.js";
 import * as ConvertParamsToWire from "../customization/convertParametersToWire.js";
 import { createOpenAIError } from "./openAIError.js";
 const expectedStatuses = ["200"];
 /** Create a vector store file batch. */
-export async function createVectorStoreFileBatch(context, vectorStoreId, options = {}) {
-    const createOptions = Object.assign(Object.assign({}, operationOptionsToRequestParameters(options)), ConvertParamsToWire.convertCreateVectorStoreFileBatchParam({ body: options }));
-    validateVectorStoreId(vectorStoreId);
-    validateCreateVectorStoreFileBatchParameters(createOptions);
-    const result = await context
-        .path("/vector_stores/{vectorStoreId}/file_batches", vectorStoreId)
-        .post(createOptions);
-    if (!expectedStatuses.includes(result.status)) {
-        throw createOpenAIError(result);
-    }
-    return ConvertFromWire.convertVectorStoreFileBatchOutput(result.body);
+export function createVectorStoreFileBatch(context, vectorStoreId, options = {}) {
+    var _a;
+    return createPoller({
+        initOperation: async () => {
+            return createVectorStoreFileBatchInternal(context, vectorStoreId, options);
+        },
+        pollOperation: async (currentResult) => {
+            return getVectorStoreFileBatch(context, vectorStoreId, currentResult.id, options);
+        },
+        getOperationStatus: getLroOperationStatus,
+        intervalInMs: (_a = options.pollingOptions) === null || _a === void 0 ? void 0 : _a.sleepIntervalInMs,
+    });
 }
 /** Retrieve a vector store file batch. */
 export async function getVectorStoreFileBatch(context, vectorStoreId, batchId, options = {}) {
@@ -58,28 +59,41 @@ export async function listVectorStoreFileBatchFiles(context, vectorStoreId, batc
 }
 /** Create a vector store file batch and poll. */
 export function createVectorStoreFileBatchAndPoll(context, vectorStoreId, options = {}) {
-    async function updateCreateVectorStoreFileBatchPoll(currentResult) {
-        let vectorStore;
-        if (!currentResult) {
-            vectorStore = await createVectorStoreFileBatch(context, vectorStoreId, options);
-        }
-        else {
-            vectorStore = await getVectorStoreFileBatch(context, vectorStoreId, currentResult.id, options);
-        }
-        return {
-            result: vectorStore,
-            completed: vectorStore.status !== "in_progress",
-        };
+    var _a;
+    return createPoller({
+        initOperation: async () => {
+            return createVectorStoreFileBatchInternal(context, vectorStoreId, options);
+        },
+        pollOperation: async (currentResult) => {
+            return getVectorStoreFileBatch(context, vectorStoreId, currentResult.id, options);
+        },
+        getOperationStatus: getLroOperationStatus,
+        intervalInMs: (_a = options.pollingOptions) === null || _a === void 0 ? void 0 : _a.sleepIntervalInMs,
+    });
+}
+export async function createVectorStoreFileBatchInternal(context, vectorStoreId, options = {}) {
+    const createOptions = Object.assign(Object.assign({}, operationOptionsToRequestParameters(options)), ConvertParamsToWire.convertCreateVectorStoreFileBatchParam({ body: options }));
+    validateVectorStoreId(vectorStoreId);
+    validateCreateVectorStoreFileBatchParameters(createOptions);
+    const result = await context
+        .path("/vector_stores/{vectorStoreId}/file_batches", vectorStoreId)
+        .post(createOptions);
+    if (!expectedStatuses.includes(result.status)) {
+        throw createOpenAIError(result);
     }
-    async function cancelCreateVectorStoreFileBatchPoll(currentResult) {
-        const result = await cancelVectorStoreFileBatch(context, vectorStoreId, currentResult.id);
-        return result.status === "cancelled";
+    return ConvertFromWire.convertVectorStoreFileBatchOutput(result.body);
+}
+function getLroOperationStatus(result) {
+    switch (result.status) {
+        case "in_progress":
+            return "running";
+        case "completed":
+            return "succeeded";
+        case "cancelled":
+            return "canceled";
+        default:
+            return "failed";
     }
-    return new AgentsPoller({
-        update: updateCreateVectorStoreFileBatchPoll,
-        cancel: cancelCreateVectorStoreFileBatchPoll,
-        pollingOptions: options.pollingOptions,
-    });
 }
 function validateBatchId(batchId) {
     if (!batchId) {

package/dist/browser/agents/vectorStoresFileBatches.js.map

@@ -1 +1 @@
-{"version":3,"file":"vectorStoresFileBatches.js","sourceRoot":"","sources":["../../../src/agents/vectorStoresFileBatches.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAGlC,OAAO,EAAE,mCAAmC,EAAE,MAAM,yBAAyB,CAAC;AAK9E,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAQ3C,OAAO,EACL,wBAAwB,EACxB,aAAa,EACb,aAAa,EACb,qBAAqB,GACtB,MAAM,uBAAuB,CAAC;AAK/B,OAAO,KAAK,eAAe,MAAM,iDAAiD,CAAC;AACnF,OAAO,KAAK,mBAAmB,MAAM,6CAA6C,CAAC;AACnF,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AAGrD,MAAM,gBAAgB,GAAG,CAAC,KAAK,CAAC,CAAC;AAEjC,wCAAwC;AACxC,MAAM,CAAC,KAAK,UAAU,0BAA0B,CAC9C,OAAe,EACf,aAAqB,EACrB,UAAoD,EAAE;IAEtD,MAAM,aAAa,mCACd,mCAAmC,CAAC,OAAO,CAAC,GAC5C,mBAAmB,CAAC,sCAAsC,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CACjF,CAAC;IAEF,qBAAqB,CAAC,aAAa,CAAC,CAAC;IACrC,4CAA4C,CAAC,aAAa,CAAC,CAAC;IAC5D,MAAM,MAAM,GAAG,MAAM,OAAO;SACzB,IAAI,CAAC,6CAA6C,EAAE,aAAa,CAAC;SAClE,IAAI,CAAC,aAAa,CAAC,CAAC;IACvB,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;QAC9C,MAAM,iBAAiB,CAAC,MAAM,CAAC,CAAC;IAClC,CAAC;IACD,OAAO,eAAe,CAAC,iCAAiC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AACxE,CAAC;AAED,0CAA0C;AAC1C,MAAM,CAAC,KAAK,UAAU,uBAAuB,CAC3C,OAAe,EACf,aAAqB,EACrB,OAAe,EACf,UAAiD,EAAE;IAEnD,qBAAqB,CAAC,aAAa,CAAC,CAAC;IACrC,MAAM,MAAM,GAAG,MAAM,OAAO;SACzB,IAAI,CAAC,uDAAuD,EAAE,aAAa,EAAE,OAAO,CAAC;SACrF,GAAG,CAAC,OAAO,CAAC,CAAC;IAChB,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;QAC9C,MAAM,iBAAiB,CAAC,MAAM,CAAC,CAAC;IAClC,CAAC;IACD,OAAO,eAAe,CAAC,iCAAiC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AACxE,CAAC;AAED,2HAA2H;AAC3H,MAAM,CAAC,KAAK,UAAU,0BAA0B,CAC9C,OAAe,EACf,aAAqB,EACrB,OAAe,EACf,UAAoD,EAAE;IAEtD,qBAAqB,CAAC,aAAa,CAAC,CAAC;IACrC,MAAM,MAAM,GAAG,MAAM,OAAO;SACzB,IAAI,CAAC,8DAA8D,EAAE,aAAa,EAAE,OAAO,CAAC;SAC5F,IAAI,CAAC,OAAO,CAAC,CAAC;IACjB,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;QAC9C,MAAM,iBAAiB,CAAC,MAAM,CAAC,CAAC;IAClC,CAAC;IACD,OAAO,eAAe,CAAC,iCAAiC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AACxE,CAAC;AAED,uDAAuD;AACvD,MAAM,CAAC,KAAK,UAAU,6BAA6B,CACjD,OAAe,EACf,aAAqB,EACrB,OAAe,EACf,UAAuD,EAAE;IAEzD,MAAM,WAAW,mCACZ,mCAAmC,CAAC,OAAO,CAAC,KAC/C,eAAe,EAAE,mBAAmB,CAAC,wDAAwD,CAC3F,OAAO,CACkB,GAC5B,CAAC;IAEF,qBAAqB,CAAC,aAAa,CAAC,CAAC;IACrC,eAAe,CAAC,OAAO,CAAC,CAAC;IACzB,+CAA+C,CAAC,WAAW,CAAC,CAAC;IAC7D,MAAM,MAAM,GAAG,MAAM,OAAO;SACzB,IAAI,CAAC,6DAA6D,EAAE,aAAa,EAAE,OAAO,CAAC;SAC3F,GAAG,CAAC,WAAW,CAAC,CAAC;IACpB,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;QAC9C,MAAM,iBAAiB,CAAC,MAAM,CAAC,CAAC;IAClC,CAAC;IACD,OAAO,eAAe,CAAC,gDAAgD,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AACvF,CAAC;AAED,iDAAiD;AACjD,MAAM,UAAU,iCAAiC,CAC/C,OAAe,EACf,aAAqB,EACrB,UAA+D,EAAE;IAEjE,KAAK,UAAU,oCAAoC,CACjD,aAA0C;QAE1C,IAAI,WAAuC,CAAC;QAC5C,IAAI,CAAC,aAAa,EAAE,CAAC;YACnB,WAAW,GAAG,MAAM,0BAA0B,CAAC,OAAO,EAAE,aAAa,EAAE,OAAO,CAAC,CAAC;QAClF,CAAC;aAAM,CAAC;YACN,WAAW,GAAG,MAAM,uBAAuB,CACzC,OAAO,EACP,aAAa,EACb,aAAa,CAAC,EAAE,EAChB,OAAO,CACR,CAAC;QACJ,CAAC;QACD,OAAO;YACL,MAAM,EAAE,WAAW;YACnB,SAAS,EAAE,WAAW,CAAC,MAAM,KAAK,aAAa;SAChD,CAAC;IACJ,CAAC;IAED,KAAK,UAAU,oCAAoC,CACjD,aAAyC;QAEzC,MAAM,MAAM,GAAG,MAAM,0BAA0B,CAAC,OAAO,EAAE,aAAa,EAAE,aAAa,CAAC,EAAE,CAAC,CAAC;QAC1F,OAAO,MAAM,CAAC,MAAM,KAAK,WAAW,CAAC;IACvC,CAAC;IAED,OAAO,IAAI,YAAY,CAA6B;QAClD,MAAM,EAAE,oCAAoC;QAC5C,MAAM,EAAE,oCAAoC;QAC5C,cAAc,EAAE,OAAO,CAAC,cAAc;KACvC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,eAAe,CAAC,OAAe;IACtC,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;IAC1C,CAAC;AACH,CAAC;AAED,SAAS,4CAA4C,CACnD,OAA8C;;IAE9C,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;AACH,CAAC;AAED,SAAS,+CAA+C,CACtD,OAAiD;;IAEjD,IAAI,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,eAAe,0CAAE,MAAM,EAAE,CAAC;QACrC,wBAAwB,CAAC,OAAO,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;IAC3D,CAAC;IACD,IAAI,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,eAAe,0CAAE,KAAK,EAAE,CAAC;QACpC,aAAa,CAAC,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,aAAa,CAAC,OAAO,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;IAC/C,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  OpenAIPageableListOfVectorStoreFileOutput,\n  VectorStoreFileBatchOutput,\n} from \"../customization/outputModels.js\";\nimport { AgentsPoller } from \"./poller.js\";\nimport type {\n  CancelVectorStoreFileBatchOptionalParams,\n  CreateVectorStoreFileBatchOptionalParams,\n  CreateVectorStoreFileBatchWithPollingOptionalParams,\n  GetVectorStoreFileBatchOptionalParams,\n  ListVectorStoreFileBatchFilesOptionalParams,\n} from \"./customModels.js\";\nimport {\n  validateFileStatusFilter,\n  validateLimit,\n  validateOrder,\n  validateVectorStoreId,\n} from \"./inputValidations.js\";\nimport type {\n  CreateVectorStoreFileBatchParameters,\n  ListVectorStoreFileBatchFilesParameters,\n} from \"../generated/src/parameters.js\";\nimport * as ConvertFromWire from \"../customization/convertOutputModelsFromWire.js\";\nimport * as ConvertParamsToWire from \"../customization/convertParametersToWire.js\";\nimport { createOpenAIError } from \"./openAIError.js\";\nimport type { PollerLike, PollOperationState } from \"@azure/core-lro\";\n\nconst expectedStatuses = [\"200\"];\n\n/** Create a vector store file batch. */\nexport async function createVectorStoreFileBatch(\n  context: Client,\n  vectorStoreId: string,\n  options: CreateVectorStoreFileBatchOptionalParams = {},\n): Promise<VectorStoreFileBatchOutput> {\n  const createOptions: CreateVectorStoreFileBatchParameters = {\n    ...operationOptionsToRequestParameters(options),\n    ...ConvertParamsToWire.convertCreateVectorStoreFileBatchParam({ body: options }),\n  };\n\n  validateVectorStoreId(vectorStoreId);\n  validateCreateVectorStoreFileBatchParameters(createOptions);\n  const result = await context\n    .path(\"/vector_stores/{vectorStoreId}/file_batches\", vectorStoreId)\n    .post(createOptions);\n  if (!expectedStatuses.includes(result.status)) {\n    throw createOpenAIError(result);\n  }\n  return ConvertFromWire.convertVectorStoreFileBatchOutput(result.body);\n}\n\n/** Retrieve a vector store file batch. */\nexport async function getVectorStoreFileBatch(\n  context: Client,\n  vectorStoreId: string,\n  batchId: string,\n  options: GetVectorStoreFileBatchOptionalParams = {},\n): Promise<VectorStoreFileBatchOutput> {\n  validateVectorStoreId(vectorStoreId);\n  const result = await context\n    .path(\"/vector_stores/{vectorStoreId}/file_batches/{batchId}\", vectorStoreId, batchId)\n    .get(options);\n  if (!expectedStatuses.includes(result.status)) {\n    throw createOpenAIError(result);\n  }\n  return ConvertFromWire.convertVectorStoreFileBatchOutput(result.body);\n}\n\n/** Cancel a vector store file batch. This attempts to cancel the processing of files in this batch as soon as possible. */\nexport async function cancelVectorStoreFileBatch(\n  context: Client,\n  vectorStoreId: string,\n  batchId: string,\n  options: CancelVectorStoreFileBatchOptionalParams = {},\n): Promise<VectorStoreFileBatchOutput> {\n  validateVectorStoreId(vectorStoreId);\n  const result = await context\n    .path(\"/vector_stores/{vectorStoreId}/file_batches/{batchId}/cancel\", vectorStoreId, batchId)\n    .post(options);\n  if (!expectedStatuses.includes(result.status)) {\n    throw createOpenAIError(result);\n  }\n  return ConvertFromWire.convertVectorStoreFileBatchOutput(result.body);\n}\n\n/** Returns a list of vector store files in a batch. */\nexport async function listVectorStoreFileBatchFiles(\n  context: Client,\n  vectorStoreId: string,\n  batchId: string,\n  options: ListVectorStoreFileBatchFilesOptionalParams = {},\n): Promise<OpenAIPageableListOfVectorStoreFileOutput> {\n  const listOptions: ListVectorStoreFileBatchFilesParameters = {\n    ...operationOptionsToRequestParameters(options),\n    queryParameters: ConvertParamsToWire.convertListVectorStoreFileBatchFilesQueryParamProperties(\n      options,\n    ) as Record<string, string>,\n  };\n\n  validateVectorStoreId(vectorStoreId);\n  validateBatchId(batchId);\n  validateListVectorStoreFileBatchFilesParameters(listOptions);\n  const result = await context\n    .path(\"/vector_stores/{vectorStoreId}/file_batches/{batchId}/files\", vectorStoreId, batchId)\n    .get(listOptions);\n  if (!expectedStatuses.includes(result.status)) {\n    throw createOpenAIError(result);\n  }\n  return ConvertFromWire.convertOpenAIPageableListOfVectorStoreFileOutput(result.body);\n}\n\n/** Create a vector store file batch and poll. */\nexport function createVectorStoreFileBatchAndPoll(\n  context: Client,\n  vectorStoreId: string,\n  options: CreateVectorStoreFileBatchWithPollingOptionalParams = {},\n): PollerLike<PollOperationState<VectorStoreFileBatchOutput>, VectorStoreFileBatchOutput> {\n  async function updateCreateVectorStoreFileBatchPoll(\n    currentResult?: VectorStoreFileBatchOutput,\n  ): Promise<{ result: VectorStoreFileBatchOutput; completed: boolean }> {\n    let vectorStore: VectorStoreFileBatchOutput;\n    if (!currentResult) {\n      vectorStore = await createVectorStoreFileBatch(context, vectorStoreId, options);\n    } else {\n      vectorStore = await getVectorStoreFileBatch(\n        context,\n        vectorStoreId,\n        currentResult.id,\n        options,\n      );\n    }\n    return {\n      result: vectorStore,\n      completed: vectorStore.status !== \"in_progress\",\n    };\n  }\n\n  async function cancelCreateVectorStoreFileBatchPoll(\n    currentResult: VectorStoreFileBatchOutput,\n  ): Promise<boolean> {\n    const result = await cancelVectorStoreFileBatch(context, vectorStoreId, currentResult.id);\n    return result.status === \"cancelled\";\n  }\n\n  return new AgentsPoller<VectorStoreFileBatchOutput>({\n    update: updateCreateVectorStoreFileBatchPoll,\n    cancel: cancelCreateVectorStoreFileBatchPoll,\n    pollingOptions: options.pollingOptions,\n  });\n}\n\nfunction validateBatchId(batchId: string): void {\n  if (!batchId) {\n    throw new Error(\"Batch ID is required\");\n  }\n}\n\nfunction validateCreateVectorStoreFileBatchParameters(\n  options?: CreateVectorStoreFileBatchParameters,\n): 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 are included\");\n  }\n}\n\nfunction validateListVectorStoreFileBatchFilesParameters(\n  options?: ListVectorStoreFileBatchFilesParameters,\n): void {\n  if (options?.queryParameters?.filter) {\n    validateFileStatusFilter(options.queryParameters.filter);\n  }\n  if (options?.queryParameters?.limit) {\n    validateLimit(options.queryParameters.limit);\n  }\n  if (options?.queryParameters?.order) {\n    validateOrder(options.queryParameters.order);\n  }\n}\n"]}
+{"version":3,"file":"vectorStoresFileBatches.js","sourceRoot":"","sources":["../../../src/agents/vectorStoresFileBatches.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAGlC,OAAO,EAAE,mCAAmC,EAAE,MAAM,yBAAyB,CAAC;AAK9E,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAO3C,OAAO,EACL,wBAAwB,EACxB,aAAa,EACb,aAAa,EACb,qBAAqB,GACtB,MAAM,uBAAuB,CAAC;AAK/B,OAAO,KAAK,eAAe,MAAM,iDAAiD,CAAC;AACnF,OAAO,KAAK,mBAAmB,MAAM,6CAA6C,CAAC;AACnF,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AAGrD,MAAM,gBAAgB,GAAG,CAAC,KAAK,CAAC,CAAC;AAEjC,wCAAwC;AACxC,MAAM,UAAU,0BAA0B,CACxC,OAAe,EACf,aAAqB,EACrB,UAAoD,EAAE;;IAEtD,OAAO,YAAY,CAA6B;QAC9C,aAAa,EAAE,KAAK,IAAI,EAAE;YACxB,OAAO,kCAAkC,CAAC,OAAO,EAAE,aAAa,EAAE,OAAO,CAAC,CAAC;QAC7E,CAAC;QACD,aAAa,EAAE,KAAK,EAAE,aAAyC,EAAE,EAAE;YACjE,OAAO,uBAAuB,CAAC,OAAO,EAAE,aAAa,EAAE,aAAa,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QACpF,CAAC;QACD,kBAAkB,EAAE,qBAAqB;QACzC,YAAY,EAAE,MAAA,OAAO,CAAC,cAAc,0CAAE,iBAAiB;KACxD,CAAC,CAAC;AACL,CAAC;AAED,0CAA0C;AAC1C,MAAM,CAAC,KAAK,UAAU,uBAAuB,CAC3C,OAAe,EACf,aAAqB,EACrB,OAAe,EACf,UAAiD,EAAE;IAEnD,qBAAqB,CAAC,aAAa,CAAC,CAAC;IACrC,MAAM,MAAM,GAAG,MAAM,OAAO;SACzB,IAAI,CAAC,uDAAuD,EAAE,aAAa,EAAE,OAAO,CAAC;SACrF,GAAG,CAAC,OAAO,CAAC,CAAC;IAChB,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;QAC9C,MAAM,iBAAiB,CAAC,MAAM,CAAC,CAAC;IAClC,CAAC;IACD,OAAO,eAAe,CAAC,iCAAiC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AACxE,CAAC;AAED,2HAA2H;AAC3H,MAAM,CAAC,KAAK,UAAU,0BAA0B,CAC9C,OAAe,EACf,aAAqB,EACrB,OAAe,EACf,UAAoD,EAAE;IAEtD,qBAAqB,CAAC,aAAa,CAAC,CAAC;IACrC,MAAM,MAAM,GAAG,MAAM,OAAO;SACzB,IAAI,CAAC,8DAA8D,EAAE,aAAa,EAAE,OAAO,CAAC;SAC5F,IAAI,CAAC,OAAO,CAAC,CAAC;IACjB,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;QAC9C,MAAM,iBAAiB,CAAC,MAAM,CAAC,CAAC;IAClC,CAAC;IACD,OAAO,eAAe,CAAC,iCAAiC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AACxE,CAAC;AAED,uDAAuD;AACvD,MAAM,CAAC,KAAK,UAAU,6BAA6B,CACjD,OAAe,EACf,aAAqB,EACrB,OAAe,EACf,UAAuD,EAAE;IAEzD,MAAM,WAAW,mCACZ,mCAAmC,CAAC,OAAO,CAAC,KAC/C,eAAe,EAAE,mBAAmB,CAAC,wDAAwD,CAC3F,OAAO,CACkB,GAC5B,CAAC;IAEF,qBAAqB,CAAC,aAAa,CAAC,CAAC;IACrC,eAAe,CAAC,OAAO,CAAC,CAAC;IACzB,+CAA+C,CAAC,WAAW,CAAC,CAAC;IAC7D,MAAM,MAAM,GAAG,MAAM,OAAO;SACzB,IAAI,CAAC,6DAA6D,EAAE,aAAa,EAAE,OAAO,CAAC;SAC3F,GAAG,CAAC,WAAW,CAAC,CAAC;IACpB,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;QAC9C,MAAM,iBAAiB,CAAC,MAAM,CAAC,CAAC;IAClC,CAAC;IACD,OAAO,eAAe,CAAC,gDAAgD,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AACvF,CAAC;AAED,iDAAiD;AACjD,MAAM,UAAU,iCAAiC,CAC/C,OAAe,EACf,aAAqB,EACrB,UAAoD,EAAE;;IAEtD,OAAO,YAAY,CAA6B;QAC9C,aAAa,EAAE,KAAK,IAAI,EAAE;YACxB,OAAO,kCAAkC,CAAC,OAAO,EAAE,aAAa,EAAE,OAAO,CAAC,CAAC;QAC7E,CAAC;QACD,aAAa,EAAE,KAAK,EAAE,aAAyC,EAAE,EAAE;YACjE,OAAO,uBAAuB,CAAC,OAAO,EAAE,aAAa,EAAE,aAAa,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QACpF,CAAC;QACD,kBAAkB,EAAE,qBAAqB;QACzC,YAAY,EAAE,MAAA,OAAO,CAAC,cAAc,0CAAE,iBAAiB;KACxD,CAAC,CAAC;AACL,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,kCAAkC,CACtD,OAAe,EACf,aAAqB,EACrB,UAAoD,EAAE;IAEtD,MAAM,aAAa,mCACd,mCAAmC,CAAC,OAAO,CAAC,GAC5C,mBAAmB,CAAC,sCAAsC,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CACjF,CAAC;IAEF,qBAAqB,CAAC,aAAa,CAAC,CAAC;IACrC,4CAA4C,CAAC,aAAa,CAAC,CAAC;IAC5D,MAAM,MAAM,GAAG,MAAM,OAAO;SACzB,IAAI,CAAC,6CAA6C,EAAE,aAAa,CAAC;SAClE,IAAI,CAAC,aAAa,CAAC,CAAC;IACvB,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;QAC9C,MAAM,iBAAiB,CAAC,MAAM,CAAC,CAAC;IAClC,CAAC;IACD,OAAO,eAAe,CAAC,iCAAiC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AACxE,CAAC;AAED,SAAS,qBAAqB,CAAC,MAAkC;IAC/D,QAAQ,MAAM,CAAC,MAAM,EAAE,CAAC;QACtB,KAAK,aAAa;YAChB,OAAO,SAAS,CAAC;QACnB,KAAK,WAAW;YACd,OAAO,WAAW,CAAC;QACrB,KAAK,WAAW;YACd,OAAO,UAAU,CAAC;QACpB;YACE,OAAO,QAAQ,CAAC;IACpB,CAAC;AACH,CAAC;AAED,SAAS,eAAe,CAAC,OAAe;IACtC,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;IAC1C,CAAC;AACH,CAAC;AAED,SAAS,4CAA4C,CACnD,OAA8C;;IAE9C,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;AACH,CAAC;AAED,SAAS,+CAA+C,CACtD,OAAiD;;IAEjD,IAAI,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,eAAe,0CAAE,MAAM,EAAE,CAAC;QACrC,wBAAwB,CAAC,OAAO,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;IAC3D,CAAC;IACD,IAAI,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,eAAe,0CAAE,KAAK,EAAE,CAAC;QACpC,aAAa,CAAC,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,aAAa,CAAC,OAAO,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;IAC/C,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  OpenAIPageableListOfVectorStoreFileOutput,\n  VectorStoreFileBatchOutput,\n} from \"../customization/outputModels.js\";\nimport { createPoller } from \"./poller.js\";\nimport type {\n  CancelVectorStoreFileBatchOptionalParams,\n  CreateVectorStoreFileBatchOptionalParams,\n  GetVectorStoreFileBatchOptionalParams,\n  ListVectorStoreFileBatchFilesOptionalParams,\n} from \"./customModels.js\";\nimport {\n  validateFileStatusFilter,\n  validateLimit,\n  validateOrder,\n  validateVectorStoreId,\n} from \"./inputValidations.js\";\nimport type {\n  CreateVectorStoreFileBatchParameters,\n  ListVectorStoreFileBatchFilesParameters,\n} from \"../generated/src/parameters.js\";\nimport * as ConvertFromWire from \"../customization/convertOutputModelsFromWire.js\";\nimport * as ConvertParamsToWire from \"../customization/convertParametersToWire.js\";\nimport { createOpenAIError } from \"./openAIError.js\";\nimport type { PollerLike, OperationState, OperationStatus } from \"@azure/core-lro\";\n\nconst expectedStatuses = [\"200\"];\n\n/** Create a vector store file batch. */\nexport function createVectorStoreFileBatch(\n  context: Client,\n  vectorStoreId: string,\n  options: CreateVectorStoreFileBatchOptionalParams = {},\n): PollerLike<OperationState<VectorStoreFileBatchOutput>, VectorStoreFileBatchOutput> {\n  return createPoller<VectorStoreFileBatchOutput>({\n    initOperation: async () => {\n      return createVectorStoreFileBatchInternal(context, vectorStoreId, options);\n    },\n    pollOperation: async (currentResult: VectorStoreFileBatchOutput) => {\n      return getVectorStoreFileBatch(context, vectorStoreId, currentResult.id, options);\n    },\n    getOperationStatus: getLroOperationStatus,\n    intervalInMs: options.pollingOptions?.sleepIntervalInMs,\n  });\n}\n\n/** Retrieve a vector store file batch. */\nexport async function getVectorStoreFileBatch(\n  context: Client,\n  vectorStoreId: string,\n  batchId: string,\n  options: GetVectorStoreFileBatchOptionalParams = {},\n): Promise<VectorStoreFileBatchOutput> {\n  validateVectorStoreId(vectorStoreId);\n  const result = await context\n    .path(\"/vector_stores/{vectorStoreId}/file_batches/{batchId}\", vectorStoreId, batchId)\n    .get(options);\n  if (!expectedStatuses.includes(result.status)) {\n    throw createOpenAIError(result);\n  }\n  return ConvertFromWire.convertVectorStoreFileBatchOutput(result.body);\n}\n\n/** Cancel a vector store file batch. This attempts to cancel the processing of files in this batch as soon as possible. */\nexport async function cancelVectorStoreFileBatch(\n  context: Client,\n  vectorStoreId: string,\n  batchId: string,\n  options: CancelVectorStoreFileBatchOptionalParams = {},\n): Promise<VectorStoreFileBatchOutput> {\n  validateVectorStoreId(vectorStoreId);\n  const result = await context\n    .path(\"/vector_stores/{vectorStoreId}/file_batches/{batchId}/cancel\", vectorStoreId, batchId)\n    .post(options);\n  if (!expectedStatuses.includes(result.status)) {\n    throw createOpenAIError(result);\n  }\n  return ConvertFromWire.convertVectorStoreFileBatchOutput(result.body);\n}\n\n/** Returns a list of vector store files in a batch. */\nexport async function listVectorStoreFileBatchFiles(\n  context: Client,\n  vectorStoreId: string,\n  batchId: string,\n  options: ListVectorStoreFileBatchFilesOptionalParams = {},\n): Promise<OpenAIPageableListOfVectorStoreFileOutput> {\n  const listOptions: ListVectorStoreFileBatchFilesParameters = {\n    ...operationOptionsToRequestParameters(options),\n    queryParameters: ConvertParamsToWire.convertListVectorStoreFileBatchFilesQueryParamProperties(\n      options,\n    ) as Record<string, string>,\n  };\n\n  validateVectorStoreId(vectorStoreId);\n  validateBatchId(batchId);\n  validateListVectorStoreFileBatchFilesParameters(listOptions);\n  const result = await context\n    .path(\"/vector_stores/{vectorStoreId}/file_batches/{batchId}/files\", vectorStoreId, batchId)\n    .get(listOptions);\n  if (!expectedStatuses.includes(result.status)) {\n    throw createOpenAIError(result);\n  }\n  return ConvertFromWire.convertOpenAIPageableListOfVectorStoreFileOutput(result.body);\n}\n\n/** Create a vector store file batch and poll. */\nexport function createVectorStoreFileBatchAndPoll(\n  context: Client,\n  vectorStoreId: string,\n  options: CreateVectorStoreFileBatchOptionalParams = {},\n): PollerLike<OperationState<VectorStoreFileBatchOutput>, VectorStoreFileBatchOutput> {\n  return createPoller<VectorStoreFileBatchOutput>({\n    initOperation: async () => {\n      return createVectorStoreFileBatchInternal(context, vectorStoreId, options);\n    },\n    pollOperation: async (currentResult: VectorStoreFileBatchOutput) => {\n      return getVectorStoreFileBatch(context, vectorStoreId, currentResult.id, options);\n    },\n    getOperationStatus: getLroOperationStatus,\n    intervalInMs: options.pollingOptions?.sleepIntervalInMs,\n  });\n}\n\nexport async function createVectorStoreFileBatchInternal(\n  context: Client,\n  vectorStoreId: string,\n  options: CreateVectorStoreFileBatchOptionalParams = {},\n): Promise<VectorStoreFileBatchOutput> {\n  const createOptions: CreateVectorStoreFileBatchParameters = {\n    ...operationOptionsToRequestParameters(options),\n    ...ConvertParamsToWire.convertCreateVectorStoreFileBatchParam({ body: options }),\n  };\n\n  validateVectorStoreId(vectorStoreId);\n  validateCreateVectorStoreFileBatchParameters(createOptions);\n  const result = await context\n    .path(\"/vector_stores/{vectorStoreId}/file_batches\", vectorStoreId)\n    .post(createOptions);\n  if (!expectedStatuses.includes(result.status)) {\n    throw createOpenAIError(result);\n  }\n  return ConvertFromWire.convertVectorStoreFileBatchOutput(result.body);\n}\n\nfunction getLroOperationStatus(result: VectorStoreFileBatchOutput): OperationStatus {\n  switch (result.status) {\n    case \"in_progress\":\n      return \"running\";\n    case \"completed\":\n      return \"succeeded\";\n    case \"cancelled\":\n      return \"canceled\";\n    default:\n      return \"failed\";\n  }\n}\n\nfunction validateBatchId(batchId: string): void {\n  if (!batchId) {\n    throw new Error(\"Batch ID is required\");\n  }\n}\n\nfunction validateCreateVectorStoreFileBatchParameters(\n  options?: CreateVectorStoreFileBatchParameters,\n): 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 are included\");\n  }\n}\n\nfunction validateListVectorStoreFileBatchFilesParameters(\n  options?: ListVectorStoreFileBatchFilesParameters,\n): void {\n  if (options?.queryParameters?.filter) {\n    validateFileStatusFilter(options.queryParameters.filter);\n  }\n  if (options?.queryParameters?.limit) {\n    validateLimit(options.queryParameters.limit);\n  }\n  if (options?.queryParameters?.order) {\n    validateOrder(options.queryParameters.order);\n  }\n}\n"]}

package/dist/browser/agents/vectorStoresFiles.d.ts

@@ -1,11 +1,11 @@
 import type { Client } from "@azure-rest/core-client";
 import type { OpenAIPageableListOfVectorStoreFileOutput, VectorStoreFileDeletionStatusOutput, VectorStoreFileOutput } from "../customization/outputModels.js";
-import type { CreateVectorStoreFileOptionalParams, CreateVectorStoreFileWithPollingOptionalParams, DeleteVectorStoreFileOptionalParams, GetVectorStoreFileOptionalParams, ListVectorStoreFilesOptionalParams } from "./customModels.js";
-import type { PollerLike, PollOperationState } from "@azure/core-lro";
+import type { CreateVectorStoreFileOptionalParams, DeleteVectorStoreFileOptionalParams, GetVectorStoreFileOptionalParams, ListVectorStoreFilesOptionalParams } from "./customModels.js";
+import type { OperationState, PollerLike } from "@azure/core-lro";
 /** Returns a list of vector store files. */
 export declare function listVectorStoreFiles(context: Client, vectorStoreId: string, options?: ListVectorStoreFilesOptionalParams): Promise<OpenAIPageableListOfVectorStoreFileOutput>;
 /** Create a vector store file by attaching a file to a vector store. */
-export declare function createVectorStoreFile(context: Client, vectorStoreId: string, options?: CreateVectorStoreFileOptionalParams): Promise<VectorStoreFileOutput>;
+export declare function createVectorStoreFile(context: Client, vectorStoreId: string, options?: CreateVectorStoreFileOptionalParams): PollerLike<OperationState<VectorStoreFileOutput>, VectorStoreFileOutput>;
 /** Retrieves a vector store file. */
 export declare function getVectorStoreFile(context: Client, vectorStoreId: string, fileId: string, options?: GetVectorStoreFileOptionalParams): Promise<VectorStoreFileOutput>;
 /**
@@ -14,5 +14,5 @@ export declare function getVectorStoreFile(context: Client, vectorStoreId: strin
  */
 export declare function deleteVectorStoreFile(context: Client, vectorStoreId: string, fileId: string, options?: DeleteVectorStoreFileOptionalParams): Promise<VectorStoreFileDeletionStatusOutput>;
 /** Create a vector store file by attaching a file to a vector store and poll. */
-export declare function createVectorStoreFileAndPoll(context: Client, vectorStoreId: string, options?: CreateVectorStoreFileWithPollingOptionalParams): PollerLike<PollOperationState<VectorStoreFileOutput>, VectorStoreFileOutput>;
+export declare function createVectorStoreFileAndPoll(context: Client, vectorStoreId: string, options?: CreateVectorStoreFileOptionalParams): PollerLike<OperationState<VectorStoreFileOutput>, VectorStoreFileOutput>;
 //# sourceMappingURL=vectorStoresFiles.d.ts.map

package/dist/browser/agents/vectorStoresFiles.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"vectorStoresFiles.d.ts","sourceRoot":"","sources":["../../../src/agents/vectorStoresFiles.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,yBAAyB,CAAC;AAMtD,OAAO,KAAK,EACV,yCAAyC,EACzC,mCAAmC,EACnC,qBAAqB,EACtB,MAAM,kCAAkC,CAAC;AAE1C,OAAO,KAAK,EACV,mCAAmC,EACnC,8CAA8C,EAC9C,mCAAmC,EACnC,gCAAgC,EAChC,kCAAkC,EACnC,MAAM,mBAAmB,CAAC;AAa3B,OAAO,KAAK,EAAE,UAAU,EAAE,kBAAkB,EAAE,MAAM,iBAAiB,CAAC;AAItE,4CAA4C;AAC5C,wBAAsB,oBAAoB,CACxC,OAAO,EAAE,MAAM,EACf,aAAa,EAAE,MAAM,EACrB,OAAO,GAAE,kCAAuC,GAC/C,OAAO,CAAC,yCAAyC,CAAC,CAgBpD;AAED,wEAAwE;AACxE,wBAAsB,qBAAqB,CACzC,OAAO,EAAE,MAAM,EACf,aAAa,EAAE,MAAM,EACrB,OAAO,GAAE,mCAAwC,GAChD,OAAO,CAAC,qBAAqB,CAAC,CAehC;AAED,qCAAqC;AACrC,wBAAsB,kBAAkB,CACtC,OAAO,EAAE,MAAM,EACf,aAAa,EAAE,MAAM,EACrB,MAAM,EAAE,MAAM,EACd,OAAO,GAAE,gCAAqC,GAC7C,OAAO,CAAC,qBAAqB,CAAC,CAchC;AAED;;;GAGG;AACH,wBAAsB,qBAAqB,CACzC,OAAO,EAAE,MAAM,EACf,aAAa,EAAE,MAAM,EACrB,MAAM,EAAE,MAAM,EACd,OAAO,GAAE,mCAAwC,GAChD,OAAO,CAAC,mCAAmC,CAAC,CAa9C;AAED,iFAAiF;AACjF,wBAAgB,4BAA4B,CAC1C,OAAO,EAAE,MAAM,EACf,aAAa,EAAE,MAAM,EACrB,OAAO,GAAE,8CAAmD,GAC3D,UAAU,CAAC,kBAAkB,CAAC,qBAAqB,CAAC,EAAE,qBAAqB,CAAC,CAoB9E"}
+{"version":3,"file":"vectorStoresFiles.d.ts","sourceRoot":"","sources":["../../../src/agents/vectorStoresFiles.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,yBAAyB,CAAC;AAMtD,OAAO,KAAK,EACV,yCAAyC,EACzC,mCAAmC,EACnC,qBAAqB,EACtB,MAAM,kCAAkC,CAAC;AAE1C,OAAO,KAAK,EACV,mCAAmC,EACnC,mCAAmC,EACnC,gCAAgC,EAChC,kCAAkC,EACnC,MAAM,mBAAmB,CAAC;AAa3B,OAAO,KAAK,EAAE,cAAc,EAAmB,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAInF,4CAA4C;AAC5C,wBAAsB,oBAAoB,CACxC,OAAO,EAAE,MAAM,EACf,aAAa,EAAE,MAAM,EACrB,OAAO,GAAE,kCAAuC,GAC/C,OAAO,CAAC,yCAAyC,CAAC,CAgBpD;AAED,wEAAwE;AACxE,wBAAgB,qBAAqB,CACnC,OAAO,EAAE,MAAM,EACf,aAAa,EAAE,MAAM,EACrB,OAAO,GAAE,mCAAwC,GAChD,UAAU,CAAC,cAAc,CAAC,qBAAqB,CAAC,EAAE,qBAAqB,CAAC,CAkB1E;AAED,qCAAqC;AACrC,wBAAsB,kBAAkB,CACtC,OAAO,EAAE,MAAM,EACf,aAAa,EAAE,MAAM,EACrB,MAAM,EAAE,MAAM,EACd,OAAO,GAAE,gCAAqC,GAC7C,OAAO,CAAC,qBAAqB,CAAC,CAchC;AAED;;;GAGG;AACH,wBAAsB,qBAAqB,CACzC,OAAO,EAAE,MAAM,EACf,aAAa,EAAE,MAAM,EACrB,MAAM,EAAE,MAAM,EACd,OAAO,GAAE,mCAAwC,GAChD,OAAO,CAAC,mCAAmC,CAAC,CAa9C;AAED,iFAAiF;AACjF,wBAAgB,4BAA4B,CAC1C,OAAO,EAAE,MAAM,EACf,aAAa,EAAE,MAAM,EACrB,OAAO,GAAE,mCAAwC,GAChD,UAAU,CAAC,cAAc,CAAC,qBAAqB,CAAC,EAAE,qBAAqB,CAAC,CAkB1E"}