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

package/dist/esm/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/esm/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/esm/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/esm/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"}

package/dist/esm/agents/vectorStoresFiles.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 { validateFileId, validateFileStatusFilter, validateLimit, validateOrder, validateVectorStoreId, } from "./inputValidations.js";
 import { convertToListQueryParameters } from "../customization/convertParametersToWire.js";
 import * as ConvertFromWire from "../customization/convertOutputModelsFromWire.js";
@@ -22,17 +22,23 @@ export async function listVectorStoreFiles(context, vectorStoreId, options = {})
     return ConvertFromWire.convertOpenAIPageableListOfVectorStoreFileOutput(result.body);
 }
 /** Create a vector store file by attaching a file to a vector store. */
-export async function createVectorStoreFile(context, vectorStoreId, options = {}) {
-    const createOptions = Object.assign(Object.assign({}, operationOptionsToRequestParameters(options)), ConvertParamsToWire.convertCreateVectorStoreFileParam({ body: options }));
-    validateVectorStoreId(vectorStoreId);
-    validateCreateVectorStoreFileParameters(createOptions);
-    const result = await context
-        .path("/vector_stores/{vectorStoreId}/files", vectorStoreId)
-        .post(createOptions);
-    if (!expectedStatuses.includes(result.status)) {
-        throw createOpenAIError(result);
-    }
-    return ConvertFromWire.convertVectorStoreFileOutput(result.body);
+export function createVectorStoreFile(context, vectorStoreId, options = {}) {
+    var _a;
+    return createPoller({
+        initOperation: async () => {
+            return createVectorStoreFileInternal(context, vectorStoreId, options);
+        },
+        pollOperation: async (currentResult) => {
+            return getVectorStoreFile(context, vectorStoreId, currentResult.id, options);
+        },
+        getOperationStatus: getLroOperationStatus,
+        getOperationError: (result) => {
+            return result.status === "failed" && result.lastError
+                ? new Error(`Operation failed with code ${result.lastError.code}: ${result.lastError.message}`)
+                : undefined;
+        },
+        intervalInMs: (_a = options.pollingOptions) === null || _a === void 0 ? void 0 : _a.sleepIntervalInMs,
+    });
 }
 /** Retrieves a vector store file. */
 export async function getVectorStoreFile(context, vectorStoreId, fileId, options = {}) {
@@ -65,24 +71,47 @@ export async function deleteVectorStoreFile(context, vectorStoreId, fileId, opti
 }
 /** Create a vector store file by attaching a file to a vector store and poll. */
 export function createVectorStoreFileAndPoll(context, vectorStoreId, options = {}) {
-    async function updateCreateVectorStoreFilePoll(currentResult) {
-        let vectorStoreFile;
-        if (!currentResult) {
-            vectorStoreFile = await createVectorStoreFile(context, vectorStoreId, options);
-        }
-        else {
-            vectorStoreFile = await getVectorStoreFile(context, vectorStoreId, currentResult.id, options);
-        }
-        return {
-            result: vectorStoreFile,
-            completed: vectorStoreFile.status !== "in_progress",
-        };
-    }
-    return new AgentsPoller({
-        update: updateCreateVectorStoreFilePoll,
-        pollingOptions: options.pollingOptions,
+    var _a;
+    return createPoller({
+        initOperation: async () => {
+            return createVectorStoreFileInternal(context, vectorStoreId, options);
+        },
+        pollOperation: async (currentResult) => {
+            return getVectorStoreFile(context, vectorStoreId, currentResult.id, options);
+        },
+        getOperationStatus: getLroOperationStatus,
+        getOperationError: (result) => {
+            return result.status === "failed" && result.lastError
+                ? new Error(`Operation failed with code ${result.lastError.code}: ${result.lastError.message}`)
+                : undefined;
+        },
+        intervalInMs: (_a = options.pollingOptions) === null || _a === void 0 ? void 0 : _a.sleepIntervalInMs,
     });
 }
+async function createVectorStoreFileInternal(context, vectorStoreId, options = {}) {
+    const createOptions = Object.assign(Object.assign({}, operationOptionsToRequestParameters(options)), ConvertParamsToWire.convertCreateVectorStoreFileParam({ body: options }));
+    validateVectorStoreId(vectorStoreId);
+    validateCreateVectorStoreFileParameters(createOptions);
+    const result = await context
+        .path("/vector_stores/{vectorStoreId}/files", vectorStoreId)
+        .post(createOptions);
+    if (!expectedStatuses.includes(result.status)) {
+        throw createOpenAIError(result);
+    }
+    return ConvertFromWire.convertVectorStoreFileOutput(result.body);
+}
+function getLroOperationStatus(result) {
+    switch (result.status) {
+        case "in_progress":
+            return "running";
+        case "completed":
+            return "succeeded";
+        case "cancelled":
+            return "canceled";
+        default:
+            return "failed";
+    }
+}
 function validateListVectorStoreFilesParameters(options) {
     var _a, _b, _c;
     if ((_a = options === null || options === void 0 ? void 0 : options.queryParameters) === null || _a === void 0 ? void 0 : _a.filter) {

package/dist/esm/agents/vectorStoresFiles.js.map

@@ -1 +1 @@
-{"version":3,"file":"vectorStoresFiles.js","sourceRoot":"","sources":["../../../src/agents/vectorStoresFiles.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAGlC,OAAO,EAAE,mCAAmC,EAAE,MAAM,yBAAyB,CAAC;AAU9E,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAQ3C,OAAO,EACL,cAAc,EACd,wBAAwB,EACxB,aAAa,EACb,aAAa,EACb,qBAAqB,GACtB,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EAAE,4BAA4B,EAAE,MAAM,6CAA6C,CAAC;AAE3F,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,4CAA4C;AAC5C,MAAM,CAAC,KAAK,UAAU,oBAAoB,CACxC,OAAe,EACf,aAAqB,EACrB,UAA8C,EAAE;IAEhD,qBAAqB,CAAC,aAAa,CAAC,CAAC;IAErC,MAAM,WAAW,mCACZ,mCAAmC,CAAC,OAAO,CAAC,KAC/C,eAAe,EAAE,4BAA4B,CAAC,OAAO,CAAC,GACvD,CAAC;IAEF,sCAAsC,CAAC,WAAW,CAAC,CAAC;IACpD,MAAM,MAAM,GAAG,MAAM,OAAO;SACzB,IAAI,CAAC,sCAAsC,EAAE,aAAa,CAAC;SAC3D,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,wEAAwE;AACxE,MAAM,CAAC,KAAK,UAAU,qBAAqB,CACzC,OAAe,EACf,aAAqB,EACrB,UAA+C,EAAE;IAEjD,MAAM,aAAa,mCACd,mCAAmC,CAAC,OAAO,CAAC,GAC5C,mBAAmB,CAAC,iCAAiC,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAC5E,CAAC;IAEF,qBAAqB,CAAC,aAAa,CAAC,CAAC;IACrC,uCAAuC,CAAC,aAAa,CAAC,CAAC;IACvD,MAAM,MAAM,GAAG,MAAM,OAAO;SACzB,IAAI,CAAC,sCAAsC,EAAE,aAAa,CAAC;SAC3D,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,4BAA4B,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AACnE,CAAC;AAED,qCAAqC;AACrC,MAAM,CAAC,KAAK,UAAU,kBAAkB,CACtC,OAAe,EACf,aAAqB,EACrB,MAAc,EACd,UAA4C,EAAE;IAE9C,MAAM,UAAU,qBACX,mCAAmC,CAAC,OAAO,CAAC,CAChD,CAAC;IAEF,qBAAqB,CAAC,aAAa,CAAC,CAAC;IACrC,cAAc,CAAC,MAAM,CAAC,CAAC;IACvB,MAAM,MAAM,GAAG,MAAM,OAAO;SACzB,IAAI,CAAC,+CAA+C,EAAE,aAAa,EAAE,MAAM,CAAC;SAC5E,GAAG,CAAC,UAAU,CAAC,CAAC;IACnB,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,4BAA4B,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AACnE,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,qBAAqB,CACzC,OAAe,EACf,aAAqB,EACrB,MAAc,EACd,UAA+C,EAAE;IAEjD,qBAAqB,CAAC,aAAa,CAAC,CAAC;IACrC,cAAc,CAAC,MAAM,CAAC,CAAC;IACvB,MAAM,aAAa,qBACd,mCAAmC,CAAC,OAAO,CAAC,CAChD,CAAC;IACF,MAAM,MAAM,GAAG,MAAM,OAAO;SACzB,IAAI,CAAC,+CAA+C,EAAE,aAAa,EAAE,MAAM,CAAC;SAC5E,MAAM,CAAC,aAAa,CAAC,CAAC;IACzB,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,0CAA0C,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AACjF,CAAC;AAED,iFAAiF;AACjF,MAAM,UAAU,4BAA4B,CAC1C,OAAe,EACf,aAAqB,EACrB,UAA0D,EAAE;IAE5D,KAAK,UAAU,+BAA+B,CAC5C,aAAqC;QAErC,IAAI,eAAsC,CAAC;QAC3C,IAAI,CAAC,aAAa,EAAE,CAAC;YACnB,eAAe,GAAG,MAAM,qBAAqB,CAAC,OAAO,EAAE,aAAa,EAAE,OAAO,CAAC,CAAC;QACjF,CAAC;aAAM,CAAC;YACN,eAAe,GAAG,MAAM,kBAAkB,CAAC,OAAO,EAAE,aAAa,EAAE,aAAa,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QAChG,CAAC;QACD,OAAO;YACL,MAAM,EAAE,eAAe;YACvB,SAAS,EAAE,eAAe,CAAC,MAAM,KAAK,aAAa;SACpD,CAAC;IACJ,CAAC;IAED,OAAO,IAAI,YAAY,CAAwB;QAC7C,MAAM,EAAE,+BAA+B;QACvC,cAAc,EAAE,OAAO,CAAC,cAAc;KACvC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,sCAAsC,CAAC,OAAwC;;IACtF,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;AAED,SAAS,uCAAuC,CAAC,OAAyC;;IACxF,IAAI,CAAA,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,IAAI,0CAAE,iBAAiB,KAAI,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;QAC9D,MAAM,IAAI,KAAK,CAAC,4DAA4D,CAAC,CAAC;IAChF,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  ListVectorStoreFilesParameters,\n  CreateVectorStoreFileParameters,\n} from \"../generated/src/parameters.js\";\nimport type {\n  OpenAIPageableListOfVectorStoreFileOutput,\n  VectorStoreFileDeletionStatusOutput,\n  VectorStoreFileOutput,\n} from \"../customization/outputModels.js\";\nimport { AgentsPoller } from \"./poller.js\";\nimport type {\n  CreateVectorStoreFileOptionalParams,\n  CreateVectorStoreFileWithPollingOptionalParams,\n  DeleteVectorStoreFileOptionalParams,\n  GetVectorStoreFileOptionalParams,\n  ListVectorStoreFilesOptionalParams,\n} from \"./customModels.js\";\nimport {\n  validateFileId,\n  validateFileStatusFilter,\n  validateLimit,\n  validateOrder,\n  validateVectorStoreId,\n} from \"./inputValidations.js\";\nimport { convertToListQueryParameters } from \"../customization/convertParametersToWire.js\";\nimport type * as GeneratedParameters 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/** Returns a list of vector store files. */\nexport async function listVectorStoreFiles(\n  context: Client,\n  vectorStoreId: string,\n  options: ListVectorStoreFilesOptionalParams = {},\n): Promise<OpenAIPageableListOfVectorStoreFileOutput> {\n  validateVectorStoreId(vectorStoreId);\n\n  const listOptions: GeneratedParameters.ListVectorStoreFilesParameters = {\n    ...operationOptionsToRequestParameters(options),\n    queryParameters: convertToListQueryParameters(options),\n  };\n\n  validateListVectorStoreFilesParameters(listOptions);\n  const result = await context\n    .path(\"/vector_stores/{vectorStoreId}/files\", vectorStoreId)\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 by attaching a file to a vector store. */\nexport async function createVectorStoreFile(\n  context: Client,\n  vectorStoreId: string,\n  options: CreateVectorStoreFileOptionalParams = {},\n): Promise<VectorStoreFileOutput> {\n  const createOptions: CreateVectorStoreFileParameters = {\n    ...operationOptionsToRequestParameters(options),\n    ...ConvertParamsToWire.convertCreateVectorStoreFileParam({ body: options }),\n  };\n\n  validateVectorStoreId(vectorStoreId);\n  validateCreateVectorStoreFileParameters(createOptions);\n  const result = await context\n    .path(\"/vector_stores/{vectorStoreId}/files\", vectorStoreId)\n    .post(createOptions);\n  if (!expectedStatuses.includes(result.status)) {\n    throw createOpenAIError(result);\n  }\n  return ConvertFromWire.convertVectorStoreFileOutput(result.body);\n}\n\n/** Retrieves a vector store file. */\nexport async function getVectorStoreFile(\n  context: Client,\n  vectorStoreId: string,\n  fileId: string,\n  options: GetVectorStoreFileOptionalParams = {},\n): Promise<VectorStoreFileOutput> {\n  const getOptions: GeneratedParameters.GetVectorStoreFileParameters = {\n    ...operationOptionsToRequestParameters(options),\n  };\n\n  validateVectorStoreId(vectorStoreId);\n  validateFileId(fileId);\n  const result = await context\n    .path(\"/vector_stores/{vectorStoreId}/files/{fileId}\", vectorStoreId, fileId)\n    .get(getOptions);\n  if (!expectedStatuses.includes(result.status)) {\n    throw createOpenAIError(result);\n  }\n  return ConvertFromWire.convertVectorStoreFileOutput(result.body);\n}\n\n/**\n * Delete a vector store file. This will remove the file from the vector store but the file itself will not be deleted.\n * To delete the file, use the delete file endpoint.\n */\nexport async function deleteVectorStoreFile(\n  context: Client,\n  vectorStoreId: string,\n  fileId: string,\n  options: DeleteVectorStoreFileOptionalParams = {},\n): Promise<VectorStoreFileDeletionStatusOutput> {\n  validateVectorStoreId(vectorStoreId);\n  validateFileId(fileId);\n  const deleteOptions: GeneratedParameters.GetVectorStoreFileParameters = {\n    ...operationOptionsToRequestParameters(options),\n  };\n  const result = await context\n    .path(\"/vector_stores/{vectorStoreId}/files/{fileId}\", vectorStoreId, fileId)\n    .delete(deleteOptions);\n  if (!expectedStatuses.includes(result.status)) {\n    throw createOpenAIError(result);\n  }\n  return ConvertFromWire.convertVectorStoreFileDeletionStatusOutput(result.body);\n}\n\n/** Create a vector store file by attaching a file to a vector store and poll. */\nexport function createVectorStoreFileAndPoll(\n  context: Client,\n  vectorStoreId: string,\n  options: CreateVectorStoreFileWithPollingOptionalParams = {},\n): PollerLike<PollOperationState<VectorStoreFileOutput>, VectorStoreFileOutput> {\n  async function updateCreateVectorStoreFilePoll(\n    currentResult?: VectorStoreFileOutput,\n  ): Promise<{ result: VectorStoreFileOutput; completed: boolean }> {\n    let vectorStoreFile: VectorStoreFileOutput;\n    if (!currentResult) {\n      vectorStoreFile = await createVectorStoreFile(context, vectorStoreId, options);\n    } else {\n      vectorStoreFile = await getVectorStoreFile(context, vectorStoreId, currentResult.id, options);\n    }\n    return {\n      result: vectorStoreFile,\n      completed: vectorStoreFile.status !== \"in_progress\",\n    };\n  }\n\n  return new AgentsPoller<VectorStoreFileOutput>({\n    update: updateCreateVectorStoreFilePoll,\n    pollingOptions: options.pollingOptions,\n  });\n}\n\nfunction validateListVectorStoreFilesParameters(options?: ListVectorStoreFilesParameters): 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\nfunction validateCreateVectorStoreFileParameters(options?: CreateVectorStoreFileParameters): void {\n  if (options?.body?.chunking_strategy && !options.body.file_id) {\n    throw new Error(\"Chunking strategy is only applicable if fileId is included\");\n  }\n}\n"]}
+{"version":3,"file":"vectorStoresFiles.js","sourceRoot":"","sources":["../../../src/agents/vectorStoresFiles.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAGlC,OAAO,EAAE,mCAAmC,EAAE,MAAM,yBAAyB,CAAC;AAU9E,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAO3C,OAAO,EACL,cAAc,EACd,wBAAwB,EACxB,aAAa,EACb,aAAa,EACb,qBAAqB,GACtB,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EAAE,4BAA4B,EAAE,MAAM,6CAA6C,CAAC;AAE3F,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,4CAA4C;AAC5C,MAAM,CAAC,KAAK,UAAU,oBAAoB,CACxC,OAAe,EACf,aAAqB,EACrB,UAA8C,EAAE;IAEhD,qBAAqB,CAAC,aAAa,CAAC,CAAC;IAErC,MAAM,WAAW,mCACZ,mCAAmC,CAAC,OAAO,CAAC,KAC/C,eAAe,EAAE,4BAA4B,CAAC,OAAO,CAAC,GACvD,CAAC;IAEF,sCAAsC,CAAC,WAAW,CAAC,CAAC;IACpD,MAAM,MAAM,GAAG,MAAM,OAAO;SACzB,IAAI,CAAC,sCAAsC,EAAE,aAAa,CAAC;SAC3D,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,wEAAwE;AACxE,MAAM,UAAU,qBAAqB,CACnC,OAAe,EACf,aAAqB,EACrB,UAA+C,EAAE;;IAEjD,OAAO,YAAY,CAAwB;QACzC,aAAa,EAAE,KAAK,IAAI,EAAE;YACxB,OAAO,6BAA6B,CAAC,OAAO,EAAE,aAAa,EAAE,OAAO,CAAC,CAAC;QACxE,CAAC;QACD,aAAa,EAAE,KAAK,EAAE,aAAoC,EAAE,EAAE;YAC5D,OAAO,kBAAkB,CAAC,OAAO,EAAE,aAAa,EAAE,aAAa,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QAC/E,CAAC;QACD,kBAAkB,EAAE,qBAAqB;QACzC,iBAAiB,EAAE,CAAC,MAA6B,EAAE,EAAE;YACnD,OAAO,MAAM,CAAC,MAAM,KAAK,QAAQ,IAAI,MAAM,CAAC,SAAS;gBACnD,CAAC,CAAC,IAAI,KAAK,CACP,8BAA8B,MAAM,CAAC,SAAS,CAAC,IAAI,KAAK,MAAM,CAAC,SAAS,CAAC,OAAO,EAAE,CACnF;gBACH,CAAC,CAAC,SAAS,CAAC;QAChB,CAAC;QACD,YAAY,EAAE,MAAA,OAAO,CAAC,cAAc,0CAAE,iBAAiB;KACxD,CAAC,CAAC;AACL,CAAC;AAED,qCAAqC;AACrC,MAAM,CAAC,KAAK,UAAU,kBAAkB,CACtC,OAAe,EACf,aAAqB,EACrB,MAAc,EACd,UAA4C,EAAE;IAE9C,MAAM,UAAU,qBACX,mCAAmC,CAAC,OAAO,CAAC,CAChD,CAAC;IAEF,qBAAqB,CAAC,aAAa,CAAC,CAAC;IACrC,cAAc,CAAC,MAAM,CAAC,CAAC;IACvB,MAAM,MAAM,GAAG,MAAM,OAAO;SACzB,IAAI,CAAC,+CAA+C,EAAE,aAAa,EAAE,MAAM,CAAC;SAC5E,GAAG,CAAC,UAAU,CAAC,CAAC;IACnB,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,4BAA4B,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AACnE,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,qBAAqB,CACzC,OAAe,EACf,aAAqB,EACrB,MAAc,EACd,UAA+C,EAAE;IAEjD,qBAAqB,CAAC,aAAa,CAAC,CAAC;IACrC,cAAc,CAAC,MAAM,CAAC,CAAC;IACvB,MAAM,aAAa,qBACd,mCAAmC,CAAC,OAAO,CAAC,CAChD,CAAC;IACF,MAAM,MAAM,GAAG,MAAM,OAAO;SACzB,IAAI,CAAC,+CAA+C,EAAE,aAAa,EAAE,MAAM,CAAC;SAC5E,MAAM,CAAC,aAAa,CAAC,CAAC;IACzB,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,0CAA0C,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AACjF,CAAC;AAED,iFAAiF;AACjF,MAAM,UAAU,4BAA4B,CAC1C,OAAe,EACf,aAAqB,EACrB,UAA+C,EAAE;;IAEjD,OAAO,YAAY,CAAwB;QACzC,aAAa,EAAE,KAAK,IAAI,EAAE;YACxB,OAAO,6BAA6B,CAAC,OAAO,EAAE,aAAa,EAAE,OAAO,CAAC,CAAC;QACxE,CAAC;QACD,aAAa,EAAE,KAAK,EAAE,aAAoC,EAAE,EAAE;YAC5D,OAAO,kBAAkB,CAAC,OAAO,EAAE,aAAa,EAAE,aAAa,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QAC/E,CAAC;QACD,kBAAkB,EAAE,qBAAqB;QACzC,iBAAiB,EAAE,CAAC,MAA6B,EAAE,EAAE;YACnD,OAAO,MAAM,CAAC,MAAM,KAAK,QAAQ,IAAI,MAAM,CAAC,SAAS;gBACnD,CAAC,CAAC,IAAI,KAAK,CACP,8BAA8B,MAAM,CAAC,SAAS,CAAC,IAAI,KAAK,MAAM,CAAC,SAAS,CAAC,OAAO,EAAE,CACnF;gBACH,CAAC,CAAC,SAAS,CAAC;QAChB,CAAC;QACD,YAAY,EAAE,MAAA,OAAO,CAAC,cAAc,0CAAE,iBAAiB;KACxD,CAAC,CAAC;AACL,CAAC;AAED,KAAK,UAAU,6BAA6B,CAC1C,OAAe,EACf,aAAqB,EACrB,UAA+C,EAAE;IAEjD,MAAM,aAAa,mCACd,mCAAmC,CAAC,OAAO,CAAC,GAC5C,mBAAmB,CAAC,iCAAiC,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAC5E,CAAC;IAEF,qBAAqB,CAAC,aAAa,CAAC,CAAC;IACrC,uCAAuC,CAAC,aAAa,CAAC,CAAC;IACvD,MAAM,MAAM,GAAG,MAAM,OAAO;SACzB,IAAI,CAAC,sCAAsC,EAAE,aAAa,CAAC;SAC3D,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,4BAA4B,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AACnE,CAAC;AAED,SAAS,qBAAqB,CAAC,MAA6B;IAC1D,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,sCAAsC,CAAC,OAAwC;;IACtF,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;AAED,SAAS,uCAAuC,CAAC,OAAyC;;IACxF,IAAI,CAAA,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,IAAI,0CAAE,iBAAiB,KAAI,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;QAC9D,MAAM,IAAI,KAAK,CAAC,4DAA4D,CAAC,CAAC;IAChF,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  ListVectorStoreFilesParameters,\n  CreateVectorStoreFileParameters,\n} from \"../generated/src/parameters.js\";\nimport type {\n  OpenAIPageableListOfVectorStoreFileOutput,\n  VectorStoreFileDeletionStatusOutput,\n  VectorStoreFileOutput,\n} from \"../customization/outputModels.js\";\nimport { createPoller } from \"./poller.js\";\nimport type {\n  CreateVectorStoreFileOptionalParams,\n  DeleteVectorStoreFileOptionalParams,\n  GetVectorStoreFileOptionalParams,\n  ListVectorStoreFilesOptionalParams,\n} from \"./customModels.js\";\nimport {\n  validateFileId,\n  validateFileStatusFilter,\n  validateLimit,\n  validateOrder,\n  validateVectorStoreId,\n} from \"./inputValidations.js\";\nimport { convertToListQueryParameters } from \"../customization/convertParametersToWire.js\";\nimport type * as GeneratedParameters 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 { OperationState, OperationStatus, PollerLike } from \"@azure/core-lro\";\n\nconst expectedStatuses = [\"200\"];\n\n/** Returns a list of vector store files. */\nexport async function listVectorStoreFiles(\n  context: Client,\n  vectorStoreId: string,\n  options: ListVectorStoreFilesOptionalParams = {},\n): Promise<OpenAIPageableListOfVectorStoreFileOutput> {\n  validateVectorStoreId(vectorStoreId);\n\n  const listOptions: GeneratedParameters.ListVectorStoreFilesParameters = {\n    ...operationOptionsToRequestParameters(options),\n    queryParameters: convertToListQueryParameters(options),\n  };\n\n  validateListVectorStoreFilesParameters(listOptions);\n  const result = await context\n    .path(\"/vector_stores/{vectorStoreId}/files\", vectorStoreId)\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 by attaching a file to a vector store. */\nexport function createVectorStoreFile(\n  context: Client,\n  vectorStoreId: string,\n  options: CreateVectorStoreFileOptionalParams = {},\n): PollerLike<OperationState<VectorStoreFileOutput>, VectorStoreFileOutput> {\n  return createPoller<VectorStoreFileOutput>({\n    initOperation: async () => {\n      return createVectorStoreFileInternal(context, vectorStoreId, options);\n    },\n    pollOperation: async (currentResult: VectorStoreFileOutput) => {\n      return getVectorStoreFile(context, vectorStoreId, currentResult.id, options);\n    },\n    getOperationStatus: getLroOperationStatus,\n    getOperationError: (result: VectorStoreFileOutput) => {\n      return result.status === \"failed\" && result.lastError\n        ? new Error(\n            `Operation failed with code ${result.lastError.code}: ${result.lastError.message}`,\n          )\n        : undefined;\n    },\n    intervalInMs: options.pollingOptions?.sleepIntervalInMs,\n  });\n}\n\n/** Retrieves a vector store file. */\nexport async function getVectorStoreFile(\n  context: Client,\n  vectorStoreId: string,\n  fileId: string,\n  options: GetVectorStoreFileOptionalParams = {},\n): Promise<VectorStoreFileOutput> {\n  const getOptions: GeneratedParameters.GetVectorStoreFileParameters = {\n    ...operationOptionsToRequestParameters(options),\n  };\n\n  validateVectorStoreId(vectorStoreId);\n  validateFileId(fileId);\n  const result = await context\n    .path(\"/vector_stores/{vectorStoreId}/files/{fileId}\", vectorStoreId, fileId)\n    .get(getOptions);\n  if (!expectedStatuses.includes(result.status)) {\n    throw createOpenAIError(result);\n  }\n  return ConvertFromWire.convertVectorStoreFileOutput(result.body);\n}\n\n/**\n * Delete a vector store file. This will remove the file from the vector store but the file itself will not be deleted.\n * To delete the file, use the delete file endpoint.\n */\nexport async function deleteVectorStoreFile(\n  context: Client,\n  vectorStoreId: string,\n  fileId: string,\n  options: DeleteVectorStoreFileOptionalParams = {},\n): Promise<VectorStoreFileDeletionStatusOutput> {\n  validateVectorStoreId(vectorStoreId);\n  validateFileId(fileId);\n  const deleteOptions: GeneratedParameters.GetVectorStoreFileParameters = {\n    ...operationOptionsToRequestParameters(options),\n  };\n  const result = await context\n    .path(\"/vector_stores/{vectorStoreId}/files/{fileId}\", vectorStoreId, fileId)\n    .delete(deleteOptions);\n  if (!expectedStatuses.includes(result.status)) {\n    throw createOpenAIError(result);\n  }\n  return ConvertFromWire.convertVectorStoreFileDeletionStatusOutput(result.body);\n}\n\n/** Create a vector store file by attaching a file to a vector store and poll. */\nexport function createVectorStoreFileAndPoll(\n  context: Client,\n  vectorStoreId: string,\n  options: CreateVectorStoreFileOptionalParams = {},\n): PollerLike<OperationState<VectorStoreFileOutput>, VectorStoreFileOutput> {\n  return createPoller<VectorStoreFileOutput>({\n    initOperation: async () => {\n      return createVectorStoreFileInternal(context, vectorStoreId, options);\n    },\n    pollOperation: async (currentResult: VectorStoreFileOutput) => {\n      return getVectorStoreFile(context, vectorStoreId, currentResult.id, options);\n    },\n    getOperationStatus: getLroOperationStatus,\n    getOperationError: (result: VectorStoreFileOutput) => {\n      return result.status === \"failed\" && result.lastError\n        ? new Error(\n            `Operation failed with code ${result.lastError.code}: ${result.lastError.message}`,\n          )\n        : undefined;\n    },\n    intervalInMs: options.pollingOptions?.sleepIntervalInMs,\n  });\n}\n\nasync function createVectorStoreFileInternal(\n  context: Client,\n  vectorStoreId: string,\n  options: CreateVectorStoreFileOptionalParams = {},\n): Promise<VectorStoreFileOutput> {\n  const createOptions: CreateVectorStoreFileParameters = {\n    ...operationOptionsToRequestParameters(options),\n    ...ConvertParamsToWire.convertCreateVectorStoreFileParam({ body: options }),\n  };\n\n  validateVectorStoreId(vectorStoreId);\n  validateCreateVectorStoreFileParameters(createOptions);\n  const result = await context\n    .path(\"/vector_stores/{vectorStoreId}/files\", vectorStoreId)\n    .post(createOptions);\n  if (!expectedStatuses.includes(result.status)) {\n    throw createOpenAIError(result);\n  }\n  return ConvertFromWire.convertVectorStoreFileOutput(result.body);\n}\n\nfunction getLroOperationStatus(result: VectorStoreFileOutput): 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 validateListVectorStoreFilesParameters(options?: ListVectorStoreFilesParameters): 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\nfunction validateCreateVectorStoreFileParameters(options?: CreateVectorStoreFileParameters): void {\n  if (options?.body?.chunking_strategy && !options.body.file_id) {\n    throw new Error(\"Chunking strategy is only applicable if fileId is included\");\n  }\n}\n"]}

package/dist/esm/constants.d.ts

@@ -1,7 +1,7 @@
 /**
  * Current version of the `@azure/ai-projects` package.
  */
-export declare const SDK_VERSION = "1.0.0-beta.2";
+export declare const SDK_VERSION = "1.0.0-beta.4";
 /**
  * The package name of the `@azure/ai-projects` package.
  */

package/dist/esm/constants.js

@@ -3,7 +3,7 @@
 /**
  * Current version of the `@azure/ai-projects` package.
  */
-export const SDK_VERSION = `1.0.0-beta.2`;
+export const SDK_VERSION = `1.0.0-beta.4`;
 /**
  * The package name of the `@azure/ai-projects` package.
  */

package/dist/esm/constants.js.map

@@ -1 +1 @@
-{"version":3,"file":"constants.js","sourceRoot":"","sources":["../../src/constants.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAElC;;GAEG;AACH,MAAM,CAAC,MAAM,WAAW,GAAG,cAAc,CAAC;AAE1C;;GAEG;AACH,MAAM,CAAC,MAAM,YAAY,GAAG,oBAAoB,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\n/**\n * Current version of the `@azure/ai-projects` package.\n */\nexport const SDK_VERSION = `1.0.0-beta.2`;\n\n/**\n * The package name of the `@azure/ai-projects` package.\n */\nexport const PACKAGE_NAME = \"@azure/ai-projects\";\n"]}
+{"version":3,"file":"constants.js","sourceRoot":"","sources":["../../src/constants.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAElC;;GAEG;AACH,MAAM,CAAC,MAAM,WAAW,GAAG,cAAc,CAAC;AAE1C;;GAEG;AACH,MAAM,CAAC,MAAM,YAAY,GAAG,oBAAoB,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\n/**\n * Current version of the `@azure/ai-projects` package.\n */\nexport const SDK_VERSION = `1.0.0-beta.4`;\n\n/**\n * The package name of the `@azure/ai-projects` package.\n */\nexport const PACKAGE_NAME = \"@azure/ai-projects\";\n"]}

package/dist/esm/customization/convertModelsToWrite.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"convertModelsToWrite.d.ts","sourceRoot":"","sources":["../../../src/customization/convertModelsToWrite.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,KAAK,YAAY,MAAM,aAAa,CAAC;AACjD,OAAO,KAAK,KAAK,eAAe,MAAM,4BAA4B,CAAC;AAGnE,wBAAgB,yBAAyB,CACvC,MAAM,EAAE,YAAY,CAAC,kBAAkB,GACtC,eAAe,CAAC,kBAAkB,CAapC;AAsJD,wBAAgB,4BAA4B,CAC1C,MAAM,EAAE,YAAY,CAAC,qBAAqB,GACzC,eAAe,CAAC,qBAAqB,CAKvC;AA6CD,wBAAgB,yBAAyB,CACvC,MAAM,EAAE,YAAY,CAAC,kBAAkB,GACtC,eAAe,CAAC,kBAAkB,CAapC;AAED,wBAAgB,iCAAiC,CAC/C,MAAM,EAAE,YAAY,CAAC,0BAA0B,GAC9C,eAAe,CAAC,0BAA0B,CAM5C;AAED,wBAAgB,+BAA+B,CAC7C,MAAM,EAAE,YAAY,CAAC,wBAAwB,GAC5C,eAAe,CAAC,wBAAwB,CAK1C;AAyBD,wBAAgB,uBAAuB,CACrC,MAAM,EAAE,YAAY,CAAC,gBAAgB,GACpC,eAAe,CAAC,gBAAgB,CA0BlC;AA2CD,wBAAgB,iBAAiB,CAAC,MAAM,EAAE,YAAY,CAAC,UAAU,GAAG,eAAe,CAAC,UAAU,CAK7F;AAED,wBAAgB,gCAAgC,CAC9C,MAAM,EAAE,YAAY,CAAC,yBAAyB,GAC7C,eAAe,CAAC,yBAAyB,CAwB3C;AAWD,wBAAgB,yBAAyB,CACvC,MAAM,EAAE,YAAY,CAAC,kBAAkB,GACtC,eAAe,CAAC,kBAAkB,CAepC;AAED,wBAAgB,yCAAyC,CACvD,MAAM,EAAE,YAAY,CAAC,kCAAkC,GACtD,eAAe,CAAC,kCAAkC,CAWpD;AAoBD,wBAAgB,+BAA+B,CAC7C,MAAM,EAAE,YAAY,CAAC,wBAAwB,GAC5C,eAAe,CAAC,wBAAwB,CAQ1C"}
+{"version":3,"file":"convertModelsToWrite.d.ts","sourceRoot":"","sources":["../../../src/customization/convertModelsToWrite.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,KAAK,YAAY,MAAM,aAAa,CAAC;AACjD,OAAO,KAAK,KAAK,eAAe,MAAM,4BAA4B,CAAC;AAGnE,wBAAgB,yBAAyB,CACvC,MAAM,EAAE,YAAY,CAAC,kBAAkB,GACtC,eAAe,CAAC,kBAAkB,CAapC;AAsKD,wBAAgB,4BAA4B,CAC1C,MAAM,EAAE,YAAY,CAAC,qBAAqB,GACzC,eAAe,CAAC,qBAAqB,CAKvC;AA8ED,wBAAgB,yBAAyB,CACvC,MAAM,EAAE,YAAY,CAAC,kBAAkB,GACtC,eAAe,CAAC,kBAAkB,CAapC;AAED,wBAAgB,iCAAiC,CAC/C,MAAM,EAAE,YAAY,CAAC,0BAA0B,GAC9C,eAAe,CAAC,0BAA0B,CAM5C;AAED,wBAAgB,+BAA+B,CAC7C,MAAM,EAAE,YAAY,CAAC,wBAAwB,GAC5C,eAAe,CAAC,wBAAwB,CAK1C;AAyBD,wBAAgB,uBAAuB,CACrC,MAAM,EAAE,YAAY,CAAC,gBAAgB,GACpC,eAAe,CAAC,gBAAgB,CA0BlC;AA2CD,wBAAgB,iBAAiB,CAAC,MAAM,EAAE,YAAY,CAAC,UAAU,GAAG,eAAe,CAAC,UAAU,CAK7F;AAED,wBAAgB,gCAAgC,CAC9C,MAAM,EAAE,YAAY,CAAC,yBAAyB,GAC7C,eAAe,CAAC,yBAAyB,CAwB3C;AAWD,wBAAgB,yBAAyB,CACvC,MAAM,EAAE,YAAY,CAAC,kBAAkB,GACtC,eAAe,CAAC,kBAAkB,CAepC;AAED,wBAAgB,yCAAyC,CACvD,MAAM,EAAE,YAAY,CAAC,kCAAkC,GACtD,eAAe,CAAC,kCAAkC,CAWpD;AAoBD,wBAAgB,+BAA+B,CAC7C,MAAM,EAAE,YAAY,CAAC,wBAAwB,GAC5C,eAAe,CAAC,wBAAwB,CAQ1C"}

package/dist/esm/customization/convertModelsToWrite.js

@@ -5,10 +5,12 @@ export function convertCreateAgentOptions(source) {
     return Object.assign(Object.assign(Object.assign(Object.assign(Object.assign(Object.assign(Object.assign(Object.assign(Object.assign({ model: source.model }, (source.name && { name: source.name })), (source.description && { description: source.description })), (source.instructions && { instructions: source.instructions })), (source.tools && { tools: source.tools.map(convertToolDefinition) })), (source.toolResources && { tool_resources: convertToolResources(source.toolResources) })), (source.temperature !== undefined && { temperature: source.temperature })), (source.topP !== undefined && { top_p: source.topP })), (source.responseFormat && { response_format: source.responseFormat })), (source.metadata && { metadata: source.metadata }));
 }
 function convertToolResources(source) {
-    return Object.assign(Object.assign(Object.assign({}, (source.codeInterpreter && {
+    return Object.assign(Object.assign(Object.assign(Object.assign({}, (source.codeInterpreter && {
         code_interpreter: convertCodeInterpreterToolResource(source.codeInterpreter),
     })), (source.fileSearch && { file_search: convertFileSearchToolResource(source.fileSearch) })), (source.azureAISearch && {
         azure_ai_search: convertAzureAISearchResource(source.azureAISearch),
+    })), (source.azureFunction && {
+        name: source.azureFunction.name,
     }));
 }
 function convertMessageAttachmentToolDefinition(source) {
@@ -31,12 +33,16 @@ function convertToolDefinition(source) {
             return convertFunctionToolDefinition(source);
         case "bing_grounding":
             return convertBingGroundingToolDefinition(source);
-        case "microsoft_fabric":
+        case "fabric_dataagent":
             return convertMicrosoftFabricToolDefinition(source);
         case "sharepoint_grounding":
             return convertSharepointToolDefinition(source);
         case "azure_ai_search":
             return convertAzureAISearchToolDefinition(source);
+        case "openapi":
+            return convertOpenApiToolDefinition(source);
+        case "azure_function":
+            return convertAzureFunctionToolDefinition(source);
         default:
             throw new Error(`Unknown tool type: ${source.type}`);
     }
@@ -66,7 +72,7 @@ function convertBingGroundingToolDefinition(source) {
 function convertMicrosoftFabricToolDefinition(source) {
     return {
         type: source.type,
-        microsoft_fabric: convertToolConnectionList(source.microsoftFabric),
+        fabric_dataagent: convertToolConnectionList(source.fabricDataAgent),
     };
 }
 function convertSharepointToolDefinition(source) {
@@ -80,6 +86,12 @@ function convertAzureAISearchToolDefinition(source) {
         type: source.type,
     };
 }
+function convertOpenApiToolDefinition(source) {
+    return {
+        type: source.type,
+        openapi: source.openapi,
+    };
+}
 function convertFileSearchToolDefinitionDetails(source) {
     return Object.assign(Object.assign({}, (source.maxNumResults && { max_num_results: source.maxNumResults })), (source.rankingOptions && {
         ranking_options: convertFileSearchRankingOptions(source.rankingOptions),
@@ -107,6 +119,31 @@ function convertFileSearchToolResource(source) {
         vector_stores: source.vectorStores.map(convertVectorStoreConfigurations),
     }));
 }
+function convertAzureFunctionToolDefinition(source) {
+    return {
+        type: source.type,
+        azure_function: convertAzureFunctionDefinition(source.azureFunction),
+    };
+}
+function convertAzureFunctionDefinition(source) {
+    return {
+        function: source.function,
+        input_binding: {
+            type: source.inputBinding.type,
+            storage_queue: {
+                queue_service_endpoint: source.inputBinding.storageQueue.queueServiceEndpoint,
+                queue_name: source.inputBinding.storageQueue.queueName,
+            },
+        },
+        output_binding: {
+            type: source.outputBinding.type,
+            storage_queue: {
+                queue_service_endpoint: source.outputBinding.storageQueue.queueServiceEndpoint,
+                queue_name: source.outputBinding.storageQueue.queueName,
+            },
+        },
+    };
+}
 function convertVectorStoreConfigurations(source) {
     return {
         name: source.name,
@@ -140,15 +177,15 @@ function convertThreadMessageOptions(source) {
     return Object.assign(Object.assign({ role: source.role, content: source.content }, (source.attachments && { attachments: source.attachments.map(convertMessageAttachment) })), (source.metadata && { metadata: source.metadata }));
 }
 function convertMessageAttachment(source) {
-    return Object.assign(Object.assign({ file_id: source.fileId }, (source.dataSources && {
-        data_sources: source.dataSources.map(convertVectorStoreDataSource),
+    return Object.assign(Object.assign({ file_id: source.fileId }, (source.dataSource && {
+        data_source: convertVectorStoreDataSource(source.dataSource),
     })), (source.tools && { tools: source.tools.map(convertMessageAttachmentToolDefinition) }));
 }
 export function convertCreateRunOptions(source) {
     return Object.assign(Object.assign(Object.assign(Object.assign(Object.assign(Object.assign(Object.assign(Object.assign(Object.assign(Object.assign(Object.assign(Object.assign(Object.assign(Object.assign({ assistant_id: source.assistantId }, (source.model && { model: source.model })), (source.instructions && { instructions: source.instructions })), (source.additionalInstructions && {
         additional_instructions: source.additionalInstructions,
     })), (source.additionalMessages && {
-        additional_messages: source.additionalMessages.map(convertThreadMessage),
+        additional_messages: source.additionalMessages.map(convertThreadMessageOptions),
     })), (source.tools && { tools: source.tools.map(convertToolDefinition) })), (source.stream !== undefined && { stream: source.stream })), (source.temperature !== undefined && { temperature: source.temperature })), (source.topP !== undefined && { top_p: source.topP })), (source.maxPromptTokens !== undefined && { max_prompt_tokens: source.maxPromptTokens })), (source.maxCompletionTokens !== undefined && {
         max_completion_tokens: source.maxCompletionTokens,
     })), (source.truncationStrategy && {
@@ -235,92 +272,4 @@ function convertToolConnection(source) {
         connection_id: source.connectionId,
     };
 }
-function convertThreadMessage(source) {
-    var _a;
-    return {
-        id: source.id,
-        object: source.object,
-        created_at: source.createdAt,
-        thread_id: source.threadId,
-        status: source.status,
-        incomplete_details: !source.incompleteDetails
-            ? source.incompleteDetails
-            : convertMessageIncompleteDetails(source.incompleteDetails),
-        completed_at: source.completedAt,
-        incomplete_at: source.incompleteAt,
-        role: source.role,
-        content: source.content.map(convertMessageContent),
-        assistant_id: source.assistantId,
-        run_id: source.runId,
-        attachments: !source.attachments
-            ? source.attachments
-            : (_a = source.attachments) === null || _a === void 0 ? void 0 : _a.map(convertMessageAttachment),
-        metadata: source.metadata,
-    };
-}
-function convertMessageIncompleteDetails(source) {
-    return {
-        reason: source.reason,
-    };
-}
-function convertMessageContent(source) {
-    switch (source.type) {
-        case "text":
-            return convertMessageTextContent(source);
-        case "image_file":
-            return convertMessageImageFileContent(source);
-        default:
-            throw new Error(`Unknown message content type: ${source.type}`);
-    }
-}
-function convertMessageTextContent(source) {
-    return {
-        type: "text",
-        text: convertMessageTextDetails(source.text),
-    };
-}
-function convertMessageTextDetails(source) {
-    return {
-        value: source.value,
-        annotations: source.annotations.map(convertMessageTextAnnotation),
-    };
-}
-function convertMessageTextAnnotation(source) {
-    switch (source.type) {
-        case "file_citation":
-            return convertMessageTextFileCitationAnnotation(source);
-        case "file_path":
-            return convertMessageTextFilePathAnnotation(source);
-        default:
-            throw new Error(`Unknown message text annotation type: ${source.type}`);
-    }
-}
-function convertMessageTextFileCitationAnnotation(source) {
-    return Object.assign(Object.assign({ text: source.text, type: "file_citation", file_citation: convertMessageTextFileCitationDetails(source.fileCitation) }, (source.startIndex && { start_index: source.startIndex })), (source.endIndex && { end_index: source.endIndex }));
-}
-function convertMessageTextFileCitationDetails(source) {
-    return {
-        file_id: source.fileId,
-        quote: source.quote,
-    };
-}
-function convertMessageTextFilePathAnnotation(source) {
-    return Object.assign(Object.assign({ text: source.text, type: "file_path", file_path: convertMessageTextFilePathDetails(source.filePath) }, (source.startIndex && { start_index: source.startIndex })), (source.endIndex && { end_index: source.endIndex }));
-}
-function convertMessageTextFilePathDetails(source) {
-    return {
-        file_id: source.fileId,
-    };
-}
-function convertMessageImageFileContent(source) {
-    return {
-        type: "image_file",
-        image_file: convertMessageImageFileDetails(source.imageFile),
-    };
-}
-function convertMessageImageFileDetails(source) {
-    return {
-        file_id: source.fileId,
-    };
-}
 //# sourceMappingURL=convertModelsToWrite.js.map