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

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

@@ -1,19 +0,0 @@
-import type { Client } from "@azure-rest/core-client";
-import type * as CustomOutputModels from "../customization/outputModels.js";
-import type * as CustomModels from "../customization/models.js";
-import type { AgentRunResponse, CreateRunOptionalParams, CancelRunOptionalParams, CreateAndRunThreadOptionalParams, GetRunOptionalParams, ListRunQueryOptionalParams, SubmitToolOutputsToRunOptionalParams, UpdateRunOptionalParams } from "./customModels.js";
-/** Creates and starts a new run of the specified thread using the specified agent. */
-export declare function createRun(context: Client, threadId: string, assistantId: string, options: CreateRunOptionalParams): AgentRunResponse;
-/** Gets a list of runs for a specified thread. */
-export declare function listRuns(context: Client, threadId: string, options?: ListRunQueryOptionalParams): Promise<CustomOutputModels.OpenAIPageableListOfThreadRunOutput>;
-/** Gets an existing run from an existing thread. */
-export declare function getRun(context: Client, threadId: string, runId: string, options?: GetRunOptionalParams): Promise<CustomOutputModels.ThreadRunOutput>;
-/** Modifies an existing thread run. */
-export declare function updateRun(context: Client, threadId: string, runId: string, options?: UpdateRunOptionalParams): Promise<CustomOutputModels.ThreadRunOutput>;
-/** Submits outputs from tools as requested by tool calls in a run. Runs that need submitted tool outputs will have a status of 'requires_action' with a required_action.type of 'submit_tool_outputs'. */
-export declare function submitToolOutputsToRun(context: Client, threadId: string, runId: string, toolOutputs: Array<CustomModels.ToolOutput>, options?: SubmitToolOutputsToRunOptionalParams): AgentRunResponse;
-/** Cancels a run of an in progress thread. */
-export declare function cancelRun(context: Client, threadId: string, runId: string, options?: CancelRunOptionalParams): Promise<CustomOutputModels.ThreadRunOutput>;
-/** Creates a new thread and immediately starts a run of that thread. */
-export declare function createThreadAndRun(context: Client, assistantId: string, options: CreateAndRunThreadOptionalParams): AgentRunResponse;
-//# sourceMappingURL=runs.d.ts.map

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

@@ -1 +0,0 @@
-{"version":3,"file":"runs.d.ts","sourceRoot":"","sources":["../../../src/agents/runs.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,yBAAyB,CAAC;AAGtD,OAAO,KAAK,KAAK,kBAAkB,MAAM,kCAAkC,CAAC;AAC5E,OAAO,KAAK,KAAK,YAAY,MAAM,4BAA4B,CAAC;AA0BhE,OAAO,KAAK,EACV,gBAAgB,EAChB,uBAAuB,EACvB,uBAAuB,EACvB,gCAAgC,EAChC,oBAAoB,EACpB,0BAA0B,EAC1B,oCAAoC,EACpC,uBAAuB,EACxB,MAAM,mBAAmB,CAAC;AAQ3B,sFAAsF;AACtF,wBAAgB,SAAS,CACvB,OAAO,EAAE,MAAM,EACf,QAAQ,EAAE,MAAM,EAChB,WAAW,EAAE,MAAM,EACnB,OAAO,EAAE,uBAAuB,GAC/B,gBAAgB,CAqClB;AAED,kDAAkD;AAClD,wBAAsB,QAAQ,CAC5B,OAAO,EAAE,MAAM,EACf,QAAQ,EAAE,MAAM,EAChB,OAAO,GAAE,0BAA+B,GACvC,OAAO,CAAC,kBAAkB,CAAC,mCAAmC,CAAC,CAuBjE;AAED,oDAAoD;AACpD,wBAAsB,MAAM,CAC1B,OAAO,EAAE,MAAM,EACf,QAAQ,EAAE,MAAM,EAChB,KAAK,EAAE,MAAM,EACb,OAAO,GAAE,oBAAyB,GACjC,OAAO,CAAC,kBAAkB,CAAC,eAAe,CAAC,CAwB7C;AAED,uCAAuC;AACvC,wBAAsB,SAAS,CAC7B,OAAO,EAAE,MAAM,EACf,QAAQ,EAAE,MAAM,EAChB,KAAK,EAAE,MAAM,EACb,OAAO,GAAE,uBAA4B,GACpC,OAAO,CAAC,kBAAkB,CAAC,eAAe,CAAC,CA8B7C;AAED,0MAA0M;AAC1M,wBAAgB,sBAAsB,CACpC,OAAO,EAAE,MAAM,EACf,QAAQ,EAAE,MAAM,EAChB,KAAK,EAAE,MAAM,EACb,WAAW,EAAE,KAAK,CAAC,YAAY,CAAC,UAAU,CAAC,EAC3C,OAAO,GAAE,oCAAyC,GACjD,gBAAgB,CAuClB;AAED,8CAA8C;AAC9C,wBAAsB,SAAS,CAC7B,OAAO,EAAE,MAAM,EACf,QAAQ,EAAE,MAAM,EAChB,KAAK,EAAE,MAAM,EACb,OAAO,GAAE,uBAA4B,GACpC,OAAO,CAAC,kBAAkB,CAAC,eAAe,CAAC,CAe7C;AAED,wEAAwE;AACxE,wBAAgB,kBAAkB,CAChC,OAAO,EAAE,MAAM,EACf,WAAW,EAAE,MAAM,EACnB,OAAO,EAAE,gCAAgC,GACxC,gBAAgB,CAsClB"}

package/dist/commonjs/agents/runs.js

@@ -1,232 +0,0 @@
-"use strict";
-// Copyright (c) Microsoft Corporation.
-// Licensed under the MIT License.
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.createRun = createRun;
-exports.listRuns = listRuns;
-exports.getRun = getRun;
-exports.updateRun = updateRun;
-exports.submitToolOutputsToRun = submitToolOutputsToRun;
-exports.cancelRun = cancelRun;
-exports.createThreadAndRun = createThreadAndRun;
-const tslib_1 = require("tslib");
-const core_client_1 = require("@azure-rest/core-client");
-const inputValidations_js_1 = require("./inputValidations.js");
-const tracing_js_1 = require("../tracing.js");
-const runTrace_js_1 = require("./runTrace.js");
-const traceUtility_js_1 = require("./traceUtility.js");
-const streaming_js_1 = require("./streaming.js");
-const ConverterToWire = tslib_1.__importStar(require("../customization/convertModelsToWrite.js"));
-const ConvertFromWire = tslib_1.__importStar(require("../customization/convertOutputModelsFromWire.js"));
-const convertParametersToWire_js_1 = require("../customization/convertParametersToWire.js");
-const openAIError_js_1 = require("./openAIError.js");
-const expectedStatuses = ["200"];
-/** Creates and starts a new run of the specified thread using the specified agent. */
-function createRun(context, threadId, assistantId, options) {
-    const createRunOptions = Object.assign(Object.assign({}, (0, core_client_1.operationOptionsToRequestParameters)(options)), { body: Object.assign(Object.assign({}, ConverterToWire.convertCreateRunOptions(Object.assign(Object.assign({}, options), { assistantId }))), { stream: false }) });
-    (0, inputValidations_js_1.validateThreadId)(threadId);
-    validateCreateRunParameters(createRunOptions);
-    async function executeCreateRun() {
-        const output = await tracing_js_1.TracingUtility.withSpan("CreateRun", createRunOptions, async (updateOptions) => {
-            const result = await context.path("/threads/{threadId}/runs", threadId).post(updateOptions);
-            if (!expectedStatuses.includes(result.status)) {
-                const error = (0, openAIError_js_1.createOpenAIError)(result);
-                throw error;
-            }
-            return result.body;
-        }, (span, updatedOptions) => (0, runTrace_js_1.traceStartCreateRun)(span, updatedOptions, threadId), runTrace_js_1.traceEndCreateOrUpdateRun);
-        return ConvertFromWire.convertThreadRunOutput(output);
-    }
-    return {
-        then: function (onFulfilled, onRejected) {
-            return executeCreateRun().then(onFulfilled, onRejected).catch(onRejected);
-        },
-        async stream() {
-            return (0, streaming_js_1.createRunStreaming)(context, threadId, createRunOptions);
-        },
-    };
-}
-/** Gets a list of runs for a specified thread. */
-async function listRuns(context, threadId, options = {}) {
-    const listRunOptions = Object.assign(Object.assign({}, (0, core_client_1.operationOptionsToRequestParameters)(options)), { queryParameters: (0, convertParametersToWire_js_1.convertToListQueryParameters)(options) });
-    validateListRunsParameters(threadId, options);
-    return tracing_js_1.TracingUtility.withSpan("ListRuns", listRunOptions || {}, async (updateOptions) => {
-        const result = await context.path("/threads/{threadId}/runs", threadId).get(updateOptions);
-        if (!expectedStatuses.includes(result.status)) {
-            throw (0, openAIError_js_1.createOpenAIError)(result);
-        }
-        return ConvertFromWire.convertOpenAIPageableListOfThreadRunOutput(result.body);
-    }, (span, updatedOptions) => (0, traceUtility_js_1.traceStartAgentGeneric)(span, Object.assign(Object.assign({}, updatedOptions), { tracingAttributeOptions: { threadId: threadId } })));
-}
-/** Gets an existing run from an existing thread. */
-async function getRun(context, threadId, runId, options = {}) {
-    (0, inputValidations_js_1.validateThreadId)(threadId);
-    (0, inputValidations_js_1.validateRunId)(runId);
-    const getRunOptions = Object.assign({}, (0, core_client_1.operationOptionsToRequestParameters)(options));
-    return tracing_js_1.TracingUtility.withSpan("GetRun", getRunOptions, async (updateOptions) => {
-        const result = await context
-            .path("/threads/{threadId}/runs/{runId}", threadId, runId)
-            .get(updateOptions);
-        if (!expectedStatuses.includes(result.status)) {
-            throw (0, openAIError_js_1.createOpenAIError)(result);
-        }
-        return ConvertFromWire.convertThreadRunOutput(result.body);
-    }, (span, updatedOptions) => (0, traceUtility_js_1.traceStartAgentGeneric)(span, Object.assign(Object.assign({}, updatedOptions), { tracingAttributeOptions: { threadId: threadId, runId: runId } })));
-}
-/** Modifies an existing thread run. */
-async function updateRun(context, threadId, runId, options = {}) {
-    const updateRunOptions = Object.assign(Object.assign({}, (0, core_client_1.operationOptionsToRequestParameters)(options)), { body: {
-            metadata: options === null || options === void 0 ? void 0 : options.metadata,
-        } });
-    validateUpdateRunParameters(threadId, runId, updateRunOptions);
-    const response = await tracing_js_1.TracingUtility.withSpan("UpdateRun", updateRunOptions, async (updateOptions) => {
-        const result = await context
-            .path("/threads/{threadId}/runs/{runId}", threadId, runId)
-            .post(updateOptions);
-        if (!expectedStatuses.includes(result.status)) {
-            throw (0, openAIError_js_1.createOpenAIError)(result);
-        }
-        return result.body;
-    }, (span, updatedOptions) => (0, traceUtility_js_1.traceStartAgentGeneric)(span, Object.assign(Object.assign({}, updatedOptions), { tracingAttributeOptions: { threadId: threadId, runId: runId } })), runTrace_js_1.traceEndCreateOrUpdateRun);
-    return ConvertFromWire.convertThreadRunOutput(response);
-}
-/** Submits outputs from tools as requested by tool calls in a run. Runs that need submitted tool outputs will have a status of 'requires_action' with a required_action.type of 'submit_tool_outputs'. */
-function submitToolOutputsToRun(context, threadId, runId, toolOutputs, options = {}) {
-    (0, inputValidations_js_1.validateThreadId)(threadId);
-    (0, inputValidations_js_1.validateRunId)(runId);
-    const submitToolOutputsOptions = Object.assign(Object.assign({}, (0, core_client_1.operationOptionsToRequestParameters)(options)), { body: {
-            tool_outputs: toolOutputs === null || toolOutputs === void 0 ? void 0 : toolOutputs.map(ConverterToWire.convertToolOutput),
-            stream: false,
-        } });
-    async function executeSubmitToolOutputsToRun() {
-        const response = await tracing_js_1.TracingUtility.withSpan("SubmitToolOutputsToRun", submitToolOutputsOptions, async (updateOptions) => {
-            const result = await context
-                .path("/threads/{threadId}/runs/{runId}/submit_tool_outputs", threadId, runId)
-                .post(updateOptions);
-            if (!expectedStatuses.includes(result.status)) {
-                throw (0, openAIError_js_1.createOpenAIError)(result);
-            }
-            return result.body;
-        }, (span, updatedOptions) => (0, runTrace_js_1.traceStartSubmitToolOutputsToRun)(span, updatedOptions, threadId, runId), runTrace_js_1.traceEndSubmitToolOutputsToRun);
-        return ConvertFromWire.convertThreadRunOutput(response);
-    }
-    return {
-        then: function (onFulfilled, onrejected) {
-            return executeSubmitToolOutputsToRun().then(onFulfilled, onrejected).catch(onrejected);
-        },
-        async stream() {
-            return (0, streaming_js_1.submitToolOutputsToRunStreaming)(context, threadId, runId, submitToolOutputsOptions);
-        },
-    };
-}
-/** Cancels a run of an in progress thread. */
-async function cancelRun(context, threadId, runId, options = {}) {
-    (0, inputValidations_js_1.validateThreadId)(threadId);
-    (0, inputValidations_js_1.validateRunId)(runId);
-    const cancelRunOptions = Object.assign({}, (0, core_client_1.operationOptionsToRequestParameters)(options));
-    return tracing_js_1.TracingUtility.withSpan("CancelRun", cancelRunOptions, async (updateOptions) => {
-        const result = await context
-            .path("/threads/{threadId}/runs/{runId}/cancel", threadId, runId)
-            .post(updateOptions);
-        if (!expectedStatuses.includes(result.status)) {
-            throw (0, openAIError_js_1.createOpenAIError)(result);
-        }
-        return ConvertFromWire.convertThreadRunOutput(result.body);
-    });
-}
-/** Creates a new thread and immediately starts a run of that thread. */
-function createThreadAndRun(context, assistantId, options) {
-    const createThreadAndRunOptions = Object.assign(Object.assign({}, (0, core_client_1.operationOptionsToRequestParameters)(options)), { body: Object.assign(Object.assign({}, ConverterToWire.convertCreateAndRunThreadOptions(Object.assign(Object.assign({}, options), { assistantId }))), { stream: false }) });
-    validateCreateThreadAndRunParameters(createThreadAndRunOptions);
-    async function executeCreateThreadAndRun() {
-        const response = await tracing_js_1.TracingUtility.withSpan("CreateThreadAndRun", createThreadAndRunOptions, async (updateOptions) => {
-            const result = await context.path("/threads/runs").post(updateOptions);
-            if (!expectedStatuses.includes(result.status)) {
-                throw (0, openAIError_js_1.createOpenAIError)(result);
-            }
-            return result.body;
-        }, runTrace_js_1.traceStartCreateThreadAndRun, runTrace_js_1.traceEndCreateOrUpdateRun);
-        return ConvertFromWire.convertThreadRunOutput(response);
-    }
-    return {
-        then: function (onFulfilled, onrejected) {
-            return executeCreateThreadAndRun().then(onFulfilled, onrejected).catch(onrejected);
-        },
-        async stream() {
-            return (0, streaming_js_1.createThreadAndRunStreaming)(context, createThreadAndRunOptions);
-        },
-    };
-}
-function validateListRunsParameters(thread_id, options) {
-    var _a, _b, _c;
-    (0, inputValidations_js_1.validateThreadId)(thread_id);
-    if (((_a = options === null || options === void 0 ? void 0 : options.queryParameters) === null || _a === void 0 ? void 0 : _a.limit) &&
-        (options.queryParameters.limit < 1 || options.queryParameters.limit > 100)) {
-        throw new Error("Limit must be between 1 and 100");
-    }
-    if ((_b = options === null || options === void 0 ? void 0 : options.queryParameters) === null || _b === void 0 ? void 0 : _b.limit) {
-        (0, inputValidations_js_1.validateLimit)(options.queryParameters.limit);
-    }
-    if ((_c = options === null || options === void 0 ? void 0 : options.queryParameters) === null || _c === void 0 ? void 0 : _c.order) {
-        (0, inputValidations_js_1.validateOrder)(options.queryParameters.order);
-    }
-}
-function validateUpdateRunParameters(thread_id, run_id, options) {
-    (0, inputValidations_js_1.validateThreadId)(thread_id);
-    (0, inputValidations_js_1.validateRunId)(run_id);
-    if (options === null || options === void 0 ? void 0 : options.body.metadata) {
-        (0, inputValidations_js_1.validateMetadata)(options.body.metadata);
-    }
-}
-function validateCreateRunParameters(options) {
-    var _a;
-    if ("additional_messages" in options.body && options.body.additional_messages) {
-        options.body.additional_messages.forEach((message) => (0, inputValidations_js_1.validateMessages)(message.role));
-    }
-    if (options.body.tools) {
-        (0, inputValidations_js_1.validateTools)(options.body.tools);
-    }
-    if (options.body.temperature && (options.body.temperature < 0 || options.body.temperature > 2)) {
-        throw new Error("Temperature must be between 0 and 2");
-    }
-    if (options.body.tool_choice && typeof options.body.tool_choice !== "string") {
-        (0, inputValidations_js_1.validateTools)([options.body.tool_choice]);
-    }
-    if ((_a = options.body.truncation_strategy) === null || _a === void 0 ? void 0 : _a.type) {
-        (0, inputValidations_js_1.validateTruncationStrategy)(options.body.truncation_strategy.type);
-    }
-    if (options.body.metadata) {
-        (0, inputValidations_js_1.validateMetadata)(options.body.metadata);
-    }
-}
-function validateCreateThreadAndRunParameters(options) {
-    var _a, _b, _c;
-    validateCreateRunParameters(options);
-    if ((_a = options.body.thread) === null || _a === void 0 ? void 0 : _a.messages) {
-        (_b = options.body.thread) === null || _b === void 0 ? void 0 : _b.messages.forEach((message) => (0, inputValidations_js_1.validateMessages)(message.role));
-    }
-    if (options.body.tools) {
-        (0, inputValidations_js_1.validateTools)(options.body.tools);
-    }
-    if ((_c = options.body.tool_resources) === null || _c === void 0 ? void 0 : _c.code_interpreter) {
-        if (options.body.tool_resources.code_interpreter) {
-            if (options.body.tool_resources.code_interpreter.file_ids &&
-                options.body.tool_resources.code_interpreter.file_ids.length > 20) {
-                throw new Error("A maximum of 20 file IDs are allowed");
-            }
-        }
-        if (options.body.tool_resources.file_search) {
-            if (options.body.tool_resources.file_search.vector_store_ids &&
-                options.body.tool_resources.file_search.vector_store_ids.length > 1) {
-                throw new Error("Only one vector store ID is allowed");
-            }
-        }
-        if (options.body.tool_resources.azure_ai_search) {
-            if (options.body.tool_resources.azure_ai_search.indexes &&
-                options.body.tool_resources.azure_ai_search.indexes.length > 1) {
-                throw new Error("Only one index is allowed");
-            }
-        }
-    }
-}
-//# sourceMappingURL=runs.js.map

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

@@ -1 +0,0 @@
-{"version":3,"file":"runs.js","sourceRoot":"","sources":["../../../src/agents/runs.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;AAkDlC,8BA0CC;AAGD,4BA2BC;AAGD,wBA6BC;AAGD,8BAmCC;AAGD,wDA6CC;AAGD,8BAoBC;AAGD,gDA0CC;;AAjTD,yDAA8E;AAI9E,+DAS+B;AAC/B,8CAA+C;AAC/C,+CAMuB;AACvB,uDAA2D;AAC3D,iDAIwB;AAYxB,kGAA4E;AAC5E,yGAAmF;AACnF,4FAA2F;AAC3F,qDAAqD;AAErD,MAAM,gBAAgB,GAAG,CAAC,KAAK,CAAC,CAAC;AAEjC,sFAAsF;AACtF,SAAgB,SAAS,CACvB,OAAe,EACf,QAAgB,EAChB,WAAmB,EACnB,OAAgC;IAEhC,MAAM,gBAAgB,mCACjB,IAAA,iDAAmC,EAAC,OAAO,CAAC,KAC/C,IAAI,kCACC,eAAe,CAAC,uBAAuB,iCAAM,OAAO,KAAE,WAAW,IAAG,KACvE,MAAM,EAAE,KAAK,MAEhB,CAAC;IACF,IAAA,sCAAgB,EAAC,QAAQ,CAAC,CAAC;IAC3B,2BAA2B,CAAC,gBAAgB,CAAC,CAAC;IAE9C,KAAK,UAAU,gBAAgB;QAC7B,MAAM,MAAM,GAAG,MAAM,2BAAc,CAAC,QAAQ,CAC1C,WAAW,EACX,gBAAgB,EAChB,KAAK,EAAE,aAAa,EAAE,EAAE;YACtB,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,0BAA0B,EAAE,QAAQ,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YAC5F,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;gBAC9C,MAAM,KAAK,GAAG,IAAA,kCAAiB,EAAC,MAAM,CAAC,CAAC;gBACxC,MAAM,KAAK,CAAC;YACd,CAAC;YACD,OAAO,MAAM,CAAC,IAAI,CAAC;QACrB,CAAC,EACD,CAAC,IAAI,EAAE,cAAc,EAAE,EAAE,CAAC,IAAA,iCAAmB,EAAC,IAAI,EAAE,cAAc,EAAE,QAAQ,CAAC,EAC7E,uCAAyB,CAC1B,CAAC;QACF,OAAO,eAAe,CAAC,sBAAsB,CAAC,MAAM,CAAC,CAAC;IACxD,CAAC;IAED,OAAO;QACL,IAAI,EAAE,UAAU,WAAW,EAAE,UAAU;YACrC,OAAO,gBAAgB,EAAE,CAAC,IAAI,CAAC,WAAW,EAAE,UAAU,CAAC,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;QAC5E,CAAC;QACD,KAAK,CAAC,MAAM;YACV,OAAO,IAAA,iCAAkB,EAAC,OAAO,EAAE,QAAQ,EAAE,gBAAgB,CAAC,CAAC;QACjE,CAAC;KACF,CAAC;AACJ,CAAC;AAED,kDAAkD;AAC3C,KAAK,UAAU,QAAQ,CAC5B,OAAe,EACf,QAAgB,EAChB,UAAsC,EAAE;IAExC,MAAM,cAAc,mCACf,IAAA,iDAAmC,EAAC,OAAO,CAAC,KAC/C,eAAe,EAAE,IAAA,yDAA4B,EAAC,OAAO,CAAC,GACvD,CAAC;IAEF,0BAA0B,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IAC9C,OAAO,2BAAc,CAAC,QAAQ,CAC5B,UAAU,EACV,cAAc,IAAI,EAAE,EACpB,KAAK,EAAE,aAAa,EAAE,EAAE;QACtB,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,0BAA0B,EAAE,QAAQ,CAAC,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;QAC3F,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;YAC9C,MAAM,IAAA,kCAAiB,EAAC,MAAM,CAAC,CAAC;QAClC,CAAC;QACD,OAAO,eAAe,CAAC,0CAA0C,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IACjF,CAAC,EACD,CAAC,IAAI,EAAE,cAAc,EAAE,EAAE,CACvB,IAAA,wCAAsB,EAAC,IAAI,kCACtB,cAAc,KACjB,uBAAuB,EAAE,EAAE,QAAQ,EAAE,QAAQ,EAAE,IAC/C,CACL,CAAC;AACJ,CAAC;AAED,oDAAoD;AAC7C,KAAK,UAAU,MAAM,CAC1B,OAAe,EACf,QAAgB,EAChB,KAAa,EACb,UAAgC,EAAE;IAElC,IAAA,sCAAgB,EAAC,QAAQ,CAAC,CAAC;IAC3B,IAAA,mCAAa,EAAC,KAAK,CAAC,CAAC;IACrB,MAAM,aAAa,qBACd,IAAA,iDAAmC,EAAC,OAAO,CAAC,CAChD,CAAC;IACF,OAAO,2BAAc,CAAC,QAAQ,CAC5B,QAAQ,EACR,aAAa,EACb,KAAK,EAAE,aAAa,EAAE,EAAE;QACtB,MAAM,MAAM,GAAG,MAAM,OAAO;aACzB,IAAI,CAAC,kCAAkC,EAAE,QAAQ,EAAE,KAAK,CAAC;aACzD,GAAG,CAAC,aAAa,CAAC,CAAC;QACtB,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;YAC9C,MAAM,IAAA,kCAAiB,EAAC,MAAM,CAAC,CAAC;QAClC,CAAC;QACD,OAAO,eAAe,CAAC,sBAAsB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IAC7D,CAAC,EACD,CAAC,IAAI,EAAE,cAAc,EAAE,EAAE,CACvB,IAAA,wCAAsB,EAAC,IAAI,kCACtB,cAAc,KACjB,uBAAuB,EAAE,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,KAAK,EAAE,IAC7D,CACL,CAAC;AACJ,CAAC;AAED,uCAAuC;AAChC,KAAK,UAAU,SAAS,CAC7B,OAAe,EACf,QAAgB,EAChB,KAAa,EACb,UAAmC,EAAE;IAErC,MAAM,gBAAgB,mCACjB,IAAA,iDAAmC,EAAC,OAAO,CAAC,KAC/C,IAAI,EAAE;YACJ,QAAQ,EAAE,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,QAAQ;SAC5B,GACF,CAAC;IAEF,2BAA2B,CAAC,QAAQ,EAAE,KAAK,EAAE,gBAAgB,CAAC,CAAC;IAC/D,MAAM,QAAQ,GAAG,MAAM,2BAAc,CAAC,QAAQ,CAC5C,WAAW,EACX,gBAAgB,EAChB,KAAK,EAAE,aAAa,EAAE,EAAE;QACtB,MAAM,MAAM,GAAG,MAAM,OAAO;aACzB,IAAI,CAAC,kCAAkC,EAAE,QAAQ,EAAE,KAAK,CAAC;aACzD,IAAI,CAAC,aAAa,CAAC,CAAC;QACvB,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;YAC9C,MAAM,IAAA,kCAAiB,EAAC,MAAM,CAAC,CAAC;QAClC,CAAC;QACD,OAAO,MAAM,CAAC,IAAI,CAAC;IACrB,CAAC,EACD,CAAC,IAAI,EAAE,cAAc,EAAE,EAAE,CACvB,IAAA,wCAAsB,EAAC,IAAI,kCACtB,cAAc,KACjB,uBAAuB,EAAE,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,KAAK,EAAE,IAC7D,EACJ,uCAAyB,CAC1B,CAAC;IAEF,OAAO,eAAe,CAAC,sBAAsB,CAAC,QAAQ,CAAC,CAAC;AAC1D,CAAC;AAED,0MAA0M;AAC1M,SAAgB,sBAAsB,CACpC,OAAe,EACf,QAAgB,EAChB,KAAa,EACb,WAA2C,EAC3C,UAAgD,EAAE;IAElD,IAAA,sCAAgB,EAAC,QAAQ,CAAC,CAAC;IAC3B,IAAA,mCAAa,EAAC,KAAK,CAAC,CAAC;IACrB,MAAM,wBAAwB,mCACzB,IAAA,iDAAmC,EAAC,OAAO,CAAC,KAC/C,IAAI,EAAE;YACJ,YAAY,EAAE,WAAW,aAAX,WAAW,uBAAX,WAAW,CAAE,GAAG,CAAC,eAAe,CAAC,iBAAiB,CAAC;YACjE,MAAM,EAAE,KAAK;SACd,GACF,CAAC;IAEF,KAAK,UAAU,6BAA6B;QAC1C,MAAM,QAAQ,GAAG,MAAM,2BAAc,CAAC,QAAQ,CAC5C,wBAAwB,EACxB,wBAAwB,EACxB,KAAK,EAAE,aAAa,EAAE,EAAE;YACtB,MAAM,MAAM,GAAG,MAAM,OAAO;iBACzB,IAAI,CAAC,sDAAsD,EAAE,QAAQ,EAAE,KAAK,CAAC;iBAC7E,IAAI,CAAC,aAAa,CAAC,CAAC;YACvB,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;gBAC9C,MAAM,IAAA,kCAAiB,EAAC,MAAM,CAAC,CAAC;YAClC,CAAC;YACD,OAAO,MAAM,CAAC,IAAI,CAAC;QACrB,CAAC,EACD,CAAC,IAAI,EAAE,cAAc,EAAE,EAAE,CACvB,IAAA,8CAAgC,EAAC,IAAI,EAAE,cAAc,EAAE,QAAQ,EAAE,KAAK,CAAC,EACzE,4CAA8B,CAC/B,CAAC;QACF,OAAO,eAAe,CAAC,sBAAsB,CAAC,QAAQ,CAAC,CAAC;IAC1D,CAAC;IAED,OAAO;QACL,IAAI,EAAE,UAAU,WAAW,EAAE,UAAU;YACrC,OAAO,6BAA6B,EAAE,CAAC,IAAI,CAAC,WAAW,EAAE,UAAU,CAAC,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;QACzF,CAAC;QACD,KAAK,CAAC,MAAM;YACV,OAAO,IAAA,8CAA+B,EAAC,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,wBAAwB,CAAC,CAAC;QAC7F,CAAC;KACF,CAAC;AACJ,CAAC;AAED,8CAA8C;AACvC,KAAK,UAAU,SAAS,CAC7B,OAAe,EACf,QAAgB,EAChB,KAAa,EACb,UAAmC,EAAE;IAErC,IAAA,sCAAgB,EAAC,QAAQ,CAAC,CAAC;IAC3B,IAAA,mCAAa,EAAC,KAAK,CAAC,CAAC;IACrB,MAAM,gBAAgB,qBACjB,IAAA,iDAAmC,EAAC,OAAO,CAAC,CAChD,CAAC;IACF,OAAO,2BAAc,CAAC,QAAQ,CAAC,WAAW,EAAE,gBAAgB,EAAE,KAAK,EAAE,aAAa,EAAE,EAAE;QACpF,MAAM,MAAM,GAAG,MAAM,OAAO;aACzB,IAAI,CAAC,yCAAyC,EAAE,QAAQ,EAAE,KAAK,CAAC;aAChE,IAAI,CAAC,aAAa,CAAC,CAAC;QACvB,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;YAC9C,MAAM,IAAA,kCAAiB,EAAC,MAAM,CAAC,CAAC;QAClC,CAAC;QACD,OAAO,eAAe,CAAC,sBAAsB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IAC7D,CAAC,CAAC,CAAC;AACL,CAAC;AAED,wEAAwE;AACxE,SAAgB,kBAAkB,CAChC,OAAe,EACf,WAAmB,EACnB,OAAyC;IAEzC,MAAM,yBAAyB,mCAC1B,IAAA,iDAAmC,EAAC,OAAO,CAAC,KAC/C,IAAI,kCACC,eAAe,CAAC,gCAAgC,iCAAM,OAAO,KAAE,WAAW,IAAG,KAChF,MAAM,EAAE,KAAK,MAEhB,CAAC;IAEF,oCAAoC,CAAC,yBAAyB,CAAC,CAAC;IAEhE,KAAK,UAAU,yBAAyB;QACtC,MAAM,QAAQ,GAAG,MAAM,2BAAc,CAAC,QAAQ,CAC5C,oBAAoB,EACpB,yBAAyB,EACzB,KAAK,EAAE,aAAa,EAAE,EAAE;YACtB,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YACvE,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;gBAC9C,MAAM,IAAA,kCAAiB,EAAC,MAAM,CAAC,CAAC;YAClC,CAAC;YAED,OAAO,MAAM,CAAC,IAAI,CAAC;QACrB,CAAC,EACD,0CAA4B,EAC5B,uCAAyB,CAC1B,CAAC;QAEF,OAAO,eAAe,CAAC,sBAAsB,CAAC,QAAQ,CAAC,CAAC;IAC1D,CAAC;IAED,OAAO;QACL,IAAI,EAAE,UAAU,WAAW,EAAE,UAAU;YACrC,OAAO,yBAAyB,EAAE,CAAC,IAAI,CAAC,WAAW,EAAE,UAAU,CAAC,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;QACrF,CAAC;QACD,KAAK,CAAC,MAAM;YACV,OAAO,IAAA,0CAA2B,EAAC,OAAO,EAAE,yBAAyB,CAAC,CAAC;QACzE,CAAC;KACF,CAAC;AACJ,CAAC;AAED,SAAS,0BAA0B,CACjC,SAAiB,EACjB,OAAgD;;IAEhD,IAAA,sCAAgB,EAAC,SAAS,CAAC,CAAC;IAC5B,IACE,CAAA,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,eAAe,0CAAE,KAAK;QAC/B,CAAC,OAAO,CAAC,eAAe,CAAC,KAAK,GAAG,CAAC,IAAI,OAAO,CAAC,eAAe,CAAC,KAAK,GAAG,GAAG,CAAC,EAC1E,CAAC;QACD,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC;IACrD,CAAC;IACD,IAAI,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,eAAe,0CAAE,KAAK,EAAE,CAAC;QACpC,IAAA,mCAAa,EAAC,OAAO,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;IAC/C,CAAC;IACD,IAAI,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,eAAe,0CAAE,KAAK,EAAE,CAAC;QACpC,IAAA,mCAAa,EAAC,OAAO,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;IAC/C,CAAC;AACH,CAAC;AAED,SAAS,2BAA2B,CAClC,SAAiB,EACjB,MAAc,EACd,OAAiD;IAEjD,IAAA,sCAAgB,EAAC,SAAS,CAAC,CAAC;IAC5B,IAAA,mCAAa,EAAC,MAAM,CAAC,CAAC;IACtB,IAAI,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,IAAI,CAAC,QAAQ,EAAE,CAAC;QAC3B,IAAA,sCAAgB,EAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC1C,CAAC;AACH,CAAC;AAED,SAAS,2BAA2B,CAClC,OAEoD;;IAEpD,IAAI,qBAAqB,IAAI,OAAO,CAAC,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC,mBAAmB,EAAE,CAAC;QAC9E,OAAO,CAAC,IAAI,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,IAAA,sCAAgB,EAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC;IACxF,CAAC;IACD,IAAI,OAAO,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;QACvB,IAAA,mCAAa,EAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACpC,CAAC;IACD,IAAI,OAAO,CAAC,IAAI,CAAC,WAAW,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,WAAW,GAAG,CAAC,IAAI,OAAO,CAAC,IAAI,CAAC,WAAW,GAAG,CAAC,CAAC,EAAE,CAAC;QAC/F,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;IACzD,CAAC;IACD,IAAI,OAAO,CAAC,IAAI,CAAC,WAAW,IAAI,OAAO,OAAO,CAAC,IAAI,CAAC,WAAW,KAAK,QAAQ,EAAE,CAAC;QAC7E,IAAA,mCAAa,EAAC,CAAC,OAAO,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC;IAC5C,CAAC;IACD,IAAI,MAAA,OAAO,CAAC,IAAI,CAAC,mBAAmB,0CAAE,IAAI,EAAE,CAAC;QAC3C,IAAA,gDAA0B,EAAC,OAAO,CAAC,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC;IACpE,CAAC;IACD,IAAI,OAAO,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;QAC1B,IAAA,sCAAgB,EAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC1C,CAAC;AACH,CAAC;AAED,SAAS,oCAAoC,CAC3C,OAAyD;;IAEzD,2BAA2B,CAAC,OAAO,CAAC,CAAC;IACrC,IAAI,MAAA,OAAO,CAAC,IAAI,CAAC,MAAM,0CAAE,QAAQ,EAAE,CAAC;QAClC,MAAA,OAAO,CAAC,IAAI,CAAC,MAAM,0CAAE,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,IAAA,sCAAgB,EAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC;IACrF,CAAC;IACD,IAAI,OAAO,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;QACvB,IAAA,mCAAa,EAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACpC,CAAC;IACD,IAAI,MAAA,OAAO,CAAC,IAAI,CAAC,cAAc,0CAAE,gBAAgB,EAAE,CAAC;QAClD,IAAI,OAAO,CAAC,IAAI,CAAC,cAAc,CAAC,gBAAgB,EAAE,CAAC;YACjD,IACE,OAAO,CAAC,IAAI,CAAC,cAAc,CAAC,gBAAgB,CAAC,QAAQ;gBACrD,OAAO,CAAC,IAAI,CAAC,cAAc,CAAC,gBAAgB,CAAC,QAAQ,CAAC,MAAM,GAAG,EAAE,EACjE,CAAC;gBACD,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;YAC1D,CAAC;QACH,CAAC;QACD,IAAI,OAAO,CAAC,IAAI,CAAC,cAAc,CAAC,WAAW,EAAE,CAAC;YAC5C,IACE,OAAO,CAAC,IAAI,CAAC,cAAc,CAAC,WAAW,CAAC,gBAAgB;gBACxD,OAAO,CAAC,IAAI,CAAC,cAAc,CAAC,WAAW,CAAC,gBAAgB,CAAC,MAAM,GAAG,CAAC,EACnE,CAAC;gBACD,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;YACzD,CAAC;QACH,CAAC;QACD,IAAI,OAAO,CAAC,IAAI,CAAC,cAAc,CAAC,eAAe,EAAE,CAAC;YAChD,IACE,OAAO,CAAC,IAAI,CAAC,cAAc,CAAC,eAAe,CAAC,OAAO;gBACnD,OAAO,CAAC,IAAI,CAAC,cAAc,CAAC,eAAe,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAC9D,CAAC;gBACD,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;YAC/C,CAAC;QACH,CAAC;IACH,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 * as GeneratedParameters from \"../generated/src/parameters.js\";\nimport type * as CustomOutputModels from \"../customization/outputModels.js\";\nimport type * as CustomModels from \"../customization/models.js\";\nimport {\n  validateLimit,\n  validateMessages,\n  validateMetadata,\n  validateOrder,\n  validateRunId,\n  validateThreadId,\n  validateTools,\n  validateTruncationStrategy,\n} from \"./inputValidations.js\";\nimport { TracingUtility } from \"../tracing.js\";\nimport {\n  traceEndCreateOrUpdateRun,\n  traceEndSubmitToolOutputsToRun,\n  traceStartCreateRun,\n  traceStartCreateThreadAndRun,\n  traceStartSubmitToolOutputsToRun,\n} from \"./runTrace.js\";\nimport { traceStartAgentGeneric } from \"./traceUtility.js\";\nimport {\n  createRunStreaming,\n  createThreadAndRunStreaming,\n  submitToolOutputsToRunStreaming,\n} from \"./streaming.js\";\nimport type { AgentEventMessageStream } from \"./streamingModels.js\";\nimport type {\n  AgentRunResponse,\n  CreateRunOptionalParams,\n  CancelRunOptionalParams,\n  CreateAndRunThreadOptionalParams,\n  GetRunOptionalParams,\n  ListRunQueryOptionalParams,\n  SubmitToolOutputsToRunOptionalParams,\n  UpdateRunOptionalParams,\n} from \"./customModels.js\";\nimport * as ConverterToWire from \"../customization/convertModelsToWrite.js\";\nimport * as ConvertFromWire from \"../customization/convertOutputModelsFromWire.js\";\nimport { convertToListQueryParameters } from \"../customization/convertParametersToWire.js\";\nimport { createOpenAIError } from \"./openAIError.js\";\n\nconst expectedStatuses = [\"200\"];\n\n/** Creates and starts a new run of the specified thread using the specified agent. */\nexport function createRun(\n  context: Client,\n  threadId: string,\n  assistantId: string,\n  options: CreateRunOptionalParams,\n): AgentRunResponse {\n  const createRunOptions: GeneratedParameters.CreateRunParameters = {\n    ...operationOptionsToRequestParameters(options),\n    body: {\n      ...ConverterToWire.convertCreateRunOptions({ ...options, assistantId }),\n      stream: false,\n    },\n  };\n  validateThreadId(threadId);\n  validateCreateRunParameters(createRunOptions);\n\n  async function executeCreateRun(): Promise<CustomOutputModels.ThreadRunOutput> {\n    const output = await TracingUtility.withSpan(\n      \"CreateRun\",\n      createRunOptions,\n      async (updateOptions) => {\n        const result = await context.path(\"/threads/{threadId}/runs\", threadId).post(updateOptions);\n        if (!expectedStatuses.includes(result.status)) {\n          const error = createOpenAIError(result);\n          throw error;\n        }\n        return result.body;\n      },\n      (span, updatedOptions) => traceStartCreateRun(span, updatedOptions, threadId),\n      traceEndCreateOrUpdateRun,\n    );\n    return ConvertFromWire.convertThreadRunOutput(output);\n  }\n\n  return {\n    then: function (onFulfilled, onRejected) {\n      return executeCreateRun().then(onFulfilled, onRejected).catch(onRejected);\n    },\n    async stream(): Promise<AgentEventMessageStream> {\n      return createRunStreaming(context, threadId, createRunOptions);\n    },\n  };\n}\n\n/** Gets a list of runs for a specified thread. */\nexport async function listRuns(\n  context: Client,\n  threadId: string,\n  options: ListRunQueryOptionalParams = {},\n): Promise<CustomOutputModels.OpenAIPageableListOfThreadRunOutput> {\n  const listRunOptions: GeneratedParameters.ListRunsParameters = {\n    ...operationOptionsToRequestParameters(options),\n    queryParameters: convertToListQueryParameters(options),\n  };\n\n  validateListRunsParameters(threadId, options);\n  return TracingUtility.withSpan(\n    \"ListRuns\",\n    listRunOptions || {},\n    async (updateOptions) => {\n      const result = await context.path(\"/threads/{threadId}/runs\", threadId).get(updateOptions);\n      if (!expectedStatuses.includes(result.status)) {\n        throw createOpenAIError(result);\n      }\n      return ConvertFromWire.convertOpenAIPageableListOfThreadRunOutput(result.body);\n    },\n    (span, updatedOptions) =>\n      traceStartAgentGeneric(span, {\n        ...updatedOptions,\n        tracingAttributeOptions: { threadId: threadId },\n      }),\n  );\n}\n\n/** Gets an existing run from an existing thread. */\nexport async function getRun(\n  context: Client,\n  threadId: string,\n  runId: string,\n  options: GetRunOptionalParams = {},\n): Promise<CustomOutputModels.ThreadRunOutput> {\n  validateThreadId(threadId);\n  validateRunId(runId);\n  const getRunOptions: GeneratedParameters.GetRunParameters = {\n    ...operationOptionsToRequestParameters(options),\n  };\n  return TracingUtility.withSpan(\n    \"GetRun\",\n    getRunOptions,\n    async (updateOptions) => {\n      const result = await context\n        .path(\"/threads/{threadId}/runs/{runId}\", threadId, runId)\n        .get(updateOptions);\n      if (!expectedStatuses.includes(result.status)) {\n        throw createOpenAIError(result);\n      }\n      return ConvertFromWire.convertThreadRunOutput(result.body);\n    },\n    (span, updatedOptions) =>\n      traceStartAgentGeneric(span, {\n        ...updatedOptions,\n        tracingAttributeOptions: { threadId: threadId, runId: runId },\n      }),\n  );\n}\n\n/** Modifies an existing thread run. */\nexport async function updateRun(\n  context: Client,\n  threadId: string,\n  runId: string,\n  options: UpdateRunOptionalParams = {},\n): Promise<CustomOutputModels.ThreadRunOutput> {\n  const updateRunOptions: GeneratedParameters.UpdateRunParameters = {\n    ...operationOptionsToRequestParameters(options),\n    body: {\n      metadata: options?.metadata,\n    },\n  };\n\n  validateUpdateRunParameters(threadId, runId, updateRunOptions);\n  const response = await TracingUtility.withSpan(\n    \"UpdateRun\",\n    updateRunOptions,\n    async (updateOptions) => {\n      const result = await context\n        .path(\"/threads/{threadId}/runs/{runId}\", threadId, runId)\n        .post(updateOptions);\n      if (!expectedStatuses.includes(result.status)) {\n        throw createOpenAIError(result);\n      }\n      return result.body;\n    },\n    (span, updatedOptions) =>\n      traceStartAgentGeneric(span, {\n        ...updatedOptions,\n        tracingAttributeOptions: { threadId: threadId, runId: runId },\n      }),\n    traceEndCreateOrUpdateRun,\n  );\n\n  return ConvertFromWire.convertThreadRunOutput(response);\n}\n\n/** Submits outputs from tools as requested by tool calls in a run. Runs that need submitted tool outputs will have a status of 'requires_action' with a required_action.type of 'submit_tool_outputs'. */\nexport function submitToolOutputsToRun(\n  context: Client,\n  threadId: string,\n  runId: string,\n  toolOutputs: Array<CustomModels.ToolOutput>,\n  options: SubmitToolOutputsToRunOptionalParams = {},\n): AgentRunResponse {\n  validateThreadId(threadId);\n  validateRunId(runId);\n  const submitToolOutputsOptions: GeneratedParameters.SubmitToolOutputsToRunParameters = {\n    ...operationOptionsToRequestParameters(options),\n    body: {\n      tool_outputs: toolOutputs?.map(ConverterToWire.convertToolOutput),\n      stream: false,\n    },\n  };\n\n  async function executeSubmitToolOutputsToRun(): Promise<CustomOutputModels.ThreadRunOutput> {\n    const response = await TracingUtility.withSpan(\n      \"SubmitToolOutputsToRun\",\n      submitToolOutputsOptions,\n      async (updateOptions) => {\n        const result = await context\n          .path(\"/threads/{threadId}/runs/{runId}/submit_tool_outputs\", threadId, runId)\n          .post(updateOptions);\n        if (!expectedStatuses.includes(result.status)) {\n          throw createOpenAIError(result);\n        }\n        return result.body;\n      },\n      (span, updatedOptions) =>\n        traceStartSubmitToolOutputsToRun(span, updatedOptions, threadId, runId),\n      traceEndSubmitToolOutputsToRun,\n    );\n    return ConvertFromWire.convertThreadRunOutput(response);\n  }\n\n  return {\n    then: function (onFulfilled, onrejected) {\n      return executeSubmitToolOutputsToRun().then(onFulfilled, onrejected).catch(onrejected);\n    },\n    async stream(): Promise<AgentEventMessageStream> {\n      return submitToolOutputsToRunStreaming(context, threadId, runId, submitToolOutputsOptions);\n    },\n  };\n}\n\n/** Cancels a run of an in progress thread. */\nexport async function cancelRun(\n  context: Client,\n  threadId: string,\n  runId: string,\n  options: CancelRunOptionalParams = {},\n): Promise<CustomOutputModels.ThreadRunOutput> {\n  validateThreadId(threadId);\n  validateRunId(runId);\n  const cancelRunOptions: GeneratedParameters.CancelRunParameters = {\n    ...operationOptionsToRequestParameters(options),\n  };\n  return TracingUtility.withSpan(\"CancelRun\", cancelRunOptions, async (updateOptions) => {\n    const result = await context\n      .path(\"/threads/{threadId}/runs/{runId}/cancel\", threadId, runId)\n      .post(updateOptions);\n    if (!expectedStatuses.includes(result.status)) {\n      throw createOpenAIError(result);\n    }\n    return ConvertFromWire.convertThreadRunOutput(result.body);\n  });\n}\n\n/** Creates a new thread and immediately starts a run of that thread. */\nexport function createThreadAndRun(\n  context: Client,\n  assistantId: string,\n  options: CreateAndRunThreadOptionalParams,\n): AgentRunResponse {\n  const createThreadAndRunOptions: GeneratedParameters.CreateThreadAndRunParameters = {\n    ...operationOptionsToRequestParameters(options),\n    body: {\n      ...ConverterToWire.convertCreateAndRunThreadOptions({ ...options, assistantId }),\n      stream: false,\n    },\n  };\n\n  validateCreateThreadAndRunParameters(createThreadAndRunOptions);\n\n  async function executeCreateThreadAndRun(): Promise<CustomOutputModels.ThreadRunOutput> {\n    const response = await TracingUtility.withSpan(\n      \"CreateThreadAndRun\",\n      createThreadAndRunOptions,\n      async (updateOptions) => {\n        const result = await context.path(\"/threads/runs\").post(updateOptions);\n        if (!expectedStatuses.includes(result.status)) {\n          throw createOpenAIError(result);\n        }\n\n        return result.body;\n      },\n      traceStartCreateThreadAndRun,\n      traceEndCreateOrUpdateRun,\n    );\n\n    return ConvertFromWire.convertThreadRunOutput(response);\n  }\n\n  return {\n    then: function (onFulfilled, onrejected) {\n      return executeCreateThreadAndRun().then(onFulfilled, onrejected).catch(onrejected);\n    },\n    async stream(): Promise<AgentEventMessageStream> {\n      return createThreadAndRunStreaming(context, createThreadAndRunOptions);\n    },\n  };\n}\n\nfunction validateListRunsParameters(\n  thread_id: string,\n  options?: GeneratedParameters.ListRunsParameters,\n): void {\n  validateThreadId(thread_id);\n  if (\n    options?.queryParameters?.limit &&\n    (options.queryParameters.limit < 1 || options.queryParameters.limit > 100)\n  ) {\n    throw new Error(\"Limit must be between 1 and 100\");\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 validateUpdateRunParameters(\n  thread_id: string,\n  run_id: string,\n  options?: GeneratedParameters.UpdateRunParameters,\n): void {\n  validateThreadId(thread_id);\n  validateRunId(run_id);\n  if (options?.body.metadata) {\n    validateMetadata(options.body.metadata);\n  }\n}\n\nfunction validateCreateRunParameters(\n  options:\n    | GeneratedParameters.CreateRunParameters\n    | GeneratedParameters.CreateThreadAndRunParameters,\n): void {\n  if (\"additional_messages\" in options.body && options.body.additional_messages) {\n    options.body.additional_messages.forEach((message) => validateMessages(message.role));\n  }\n  if (options.body.tools) {\n    validateTools(options.body.tools);\n  }\n  if (options.body.temperature && (options.body.temperature < 0 || options.body.temperature > 2)) {\n    throw new Error(\"Temperature must be between 0 and 2\");\n  }\n  if (options.body.tool_choice && typeof options.body.tool_choice !== \"string\") {\n    validateTools([options.body.tool_choice]);\n  }\n  if (options.body.truncation_strategy?.type) {\n    validateTruncationStrategy(options.body.truncation_strategy.type);\n  }\n  if (options.body.metadata) {\n    validateMetadata(options.body.metadata);\n  }\n}\n\nfunction validateCreateThreadAndRunParameters(\n  options: GeneratedParameters.CreateThreadAndRunParameters,\n): void {\n  validateCreateRunParameters(options);\n  if (options.body.thread?.messages) {\n    options.body.thread?.messages.forEach((message) => validateMessages(message.role));\n  }\n  if (options.body.tools) {\n    validateTools(options.body.tools);\n  }\n  if (options.body.tool_resources?.code_interpreter) {\n    if (options.body.tool_resources.code_interpreter) {\n      if (\n        options.body.tool_resources.code_interpreter.file_ids &&\n        options.body.tool_resources.code_interpreter.file_ids.length > 20\n      ) {\n        throw new Error(\"A maximum of 20 file IDs are allowed\");\n      }\n    }\n    if (options.body.tool_resources.file_search) {\n      if (\n        options.body.tool_resources.file_search.vector_store_ids &&\n        options.body.tool_resources.file_search.vector_store_ids.length > 1\n      ) {\n        throw new Error(\"Only one vector store ID is allowed\");\n      }\n    }\n    if (options.body.tool_resources.azure_ai_search) {\n      if (\n        options.body.tool_resources.azure_ai_search.indexes &&\n        options.body.tool_resources.azure_ai_search.indexes.length > 1\n      ) {\n        throw new Error(\"Only one index is allowed\");\n      }\n    }\n  }\n}\n"]}

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

@@ -1,9 +0,0 @@
-import type { Client } from "@azure-rest/core-client";
-import type { CreateRunParameters, CreateThreadAndRunBodyParam, SubmitToolOutputsToRunParameters } from "../generated/src/index.js";
-import { type AgentEventMessageStream } from "./streamingModels.js";
-/** Create a run and stream the events */
-export declare function createRunStreaming(context: Client, threadId: string, options: CreateRunParameters): Promise<AgentEventMessageStream>;
-/** Create a thread and run and stream the events */
-export declare function createThreadAndRunStreaming(context: Client, options: CreateThreadAndRunBodyParam): Promise<AgentEventMessageStream>;
-export declare function submitToolOutputsToRunStreaming(context: Client, threadId: string, runId: string, options: SubmitToolOutputsToRunParameters): Promise<AgentEventMessageStream>;
-//# sourceMappingURL=streaming.d.ts.map

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

@@ -1 +0,0 @@
-{"version":3,"file":"streaming.d.ts","sourceRoot":"","sources":["../../../src/agents/streaming.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,MAAM,EAAoB,MAAM,yBAAyB,CAAC;AACxE,OAAO,KAAK,EACV,mBAAmB,EACnB,2BAA2B,EAC3B,gCAAgC,EACjC,MAAM,2BAA2B,CAAC;AACnC,OAAO,EAML,KAAK,uBAAuB,EAE7B,MAAM,sBAAsB,CAAC;AAyF9B,yCAAyC;AACzC,wBAAsB,kBAAkB,CACtC,OAAO,EAAE,MAAM,EACf,QAAQ,EAAE,MAAM,EAChB,OAAO,EAAE,mBAAmB,GAC3B,OAAO,CAAC,uBAAuB,CAAC,CAMlC;AAED,oDAAoD;AACpD,wBAAsB,2BAA2B,CAC/C,OAAO,EAAE,MAAM,EACf,OAAO,EAAE,2BAA2B,GACnC,OAAO,CAAC,uBAAuB,CAAC,CAIlC;AAED,wBAAsB,+BAA+B,CACnD,OAAO,EAAE,MAAM,EACf,QAAQ,EAAE,MAAM,EAChB,KAAK,EAAE,MAAM,EACb,OAAO,EAAE,gCAAgC,GACxC,OAAO,CAAC,uBAAuB,CAAC,CAUlC"}

package/dist/commonjs/agents/streaming.js

@@ -1,140 +0,0 @@
-"use strict";
-// Copyright (c) Microsoft Corporation.
-// Licensed under the MIT License.
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.createRunStreaming = createRunStreaming;
-exports.createThreadAndRunStreaming = createThreadAndRunStreaming;
-exports.submitToolOutputsToRunStreaming = submitToolOutputsToRunStreaming;
-const tslib_1 = require("tslib");
-const streamingModels_js_1 = require("./streamingModels.js");
-const core_sse_1 = require("@azure/core-sse");
-const core_util_1 = require("@azure/core-util");
-const inputValidations_js_1 = require("./inputValidations.js");
-const openAIError_js_1 = require("./openAIError.js");
-const convertOutputModelsFromWire_js_1 = require("../customization/convertOutputModelsFromWire.js");
-const logger_js_1 = require("../logger.js");
-const expectedStatuses = ["200"];
-const handlers = [
-    { events: Object.values(streamingModels_js_1.ThreadStreamEvent), converter: convertOutputModelsFromWire_js_1.convertAgentThreadOutput },
-    { events: Object.values(streamingModels_js_1.RunStreamEvent), converter: convertOutputModelsFromWire_js_1.convertThreadRunOutput },
-    { events: Object.values(streamingModels_js_1.RunStepStreamEvent), converter: convertOutputModelsFromWire_js_1.convertRunStepOutput },
-    { events: Object.values(streamingModels_js_1.MessageStreamEvent), converter: convertOutputModelsFromWire_js_1.convertThreadMessageOutput },
-];
-function createAgentStream(stream) {
-    const asyncIterator = toAsyncIterable(stream);
-    const asyncDisposable = stream;
-    return Object.assign(asyncIterator, asyncDisposable);
-}
-function toAsyncIterable(stream) {
-    return tslib_1.__asyncGenerator(this, arguments, function* toAsyncIterable_1() {
-        var _a, e_1, _b, _c;
-        try {
-            for (var _d = true, stream_1 = tslib_1.__asyncValues(stream), stream_1_1; stream_1_1 = yield tslib_1.__await(stream_1.next()), _a = stream_1_1.done, !_a; _d = true) {
-                _c = stream_1_1.value;
-                _d = false;
-                const event = _c;
-                const data = deserializeEventData(event);
-                yield yield tslib_1.__await({ data: data, event: event.event });
-            }
-        }
-        catch (e_1_1) { e_1 = { error: e_1_1 }; }
-        finally {
-            try {
-                if (!_d && !_a && (_b = stream_1.return)) yield tslib_1.__await(_b.call(stream_1));
-            }
-            finally { if (e_1) throw e_1.error; }
-        }
-    });
-}
-function deserializeEventData(event) {
-    try {
-        const jsonData = JSON.parse(event.data);
-        switch (event.event) {
-            case streamingModels_js_1.MessageStreamEvent.ThreadMessageDelta:
-                return (0, convertOutputModelsFromWire_js_1.convertMessageDeltaChunkOutput)(jsonData);
-            case streamingModels_js_1.RunStepStreamEvent.ThreadRunStepDelta:
-                return (0, convertOutputModelsFromWire_js_1.convertRunStepDeltaChunk)(jsonData);
-            default: {
-                for (const { events, converter } of handlers) {
-                    if (events.includes(event.event)) {
-                        return converter(jsonData);
-                    }
-                }
-                return jsonData;
-            }
-        }
-    }
-    catch (ex) {
-        logger_js_1.logger.error(`Failed to parse event data  ${event.event} - error: ${ex}`);
-        return event.data;
-    }
-}
-async function processStream(streamResponse) {
-    const result = core_util_1.isNodeLike
-        ? await streamResponse.asNodeStream()
-        : await streamResponse.asBrowserStream();
-    if (!expectedStatuses.includes(result.status)) {
-        throw (0, openAIError_js_1.createOpenAIError)(result);
-    }
-    if (!result.body) {
-        throw new Error("No body in response");
-    }
-    const stream = core_util_1.isNodeLike
-        ? (0, core_sse_1.createSseStream)(result.body)
-        : (0, core_sse_1.createSseStream)(result.body);
-    return createAgentStream(stream);
-}
-/** Create a run and stream the events */
-async function createRunStreaming(context, threadId, options) {
-    (0, inputValidations_js_1.validateThreadId)(threadId);
-    validateCreateThreadAndRunBodyParam(options);
-    options.body.stream = true;
-    return processStream(context.path("/threads/{threadId}/runs", threadId).post(options));
-}
-/** Create a thread and run and stream the events */
-async function createThreadAndRunStreaming(context, options) {
-    validateCreateThreadAndRunBodyParam(options);
-    options.body.stream = true;
-    return processStream(context.path("/threads/runs").post(options));
-}
-async function submitToolOutputsToRunStreaming(context, threadId, runId, options) {
-    (0, inputValidations_js_1.validateThreadId)(threadId);
-    (0, inputValidations_js_1.validateRunId)(runId);
-    options.body.stream = true;
-    return processStream(context
-        .path("/threads/{threadId}/runs/{runId}/submit_tool_outputs", threadId, runId)
-        .post(options));
-}
-function validateCreateThreadAndRunBodyParam(options) {
-    var _a, _b, _c;
-    if ("additional_messages" in options.body && options.body.additional_messages) {
-        options.body.additional_messages.forEach((message) => (0, inputValidations_js_1.validateMessages)(message.role));
-    }
-    if ("thread" in options.body && ((_a = options.body.thread) === null || _a === void 0 ? void 0 : _a.messages)) {
-        (_b = options.body.thread) === null || _b === void 0 ? void 0 : _b.messages.forEach((message) => (0, inputValidations_js_1.validateMessages)(message.role));
-    }
-    if (options.body.tools) {
-        (0, inputValidations_js_1.validateTools)(options.body.tools);
-    }
-    if ("tool_resources" in options.body && (options === null || options === void 0 ? void 0 : options.body.tool_resources)) {
-        (0, inputValidations_js_1.validateToolResources)(options.body.tool_resources);
-    }
-    if (options.body.temperature && (options.body.temperature < 0 || options.body.temperature > 2)) {
-        throw new Error("Temperature must be between 0 and 2");
-    }
-    if (options.body.tool_choice && typeof options.body.tool_choice !== "string") {
-        (0, inputValidations_js_1.validateTools)([options.body.tool_choice]);
-    }
-    if ((_c = options.body.truncation_strategy) === null || _c === void 0 ? void 0 : _c.type) {
-        (0, inputValidations_js_1.validateTruncationStrategy)(options.body.truncation_strategy.type);
-    }
-    if (options.body.response_format) {
-        if (!["json", "text"].includes(options.body.response_format)) {
-            throw new Error("Response format must be either 'json' or 'text'");
-        }
-    }
-    if (options === null || options === void 0 ? void 0 : options.body.metadata) {
-        (0, inputValidations_js_1.validateMetadata)(options.body.metadata);
-    }
-}
-//# sourceMappingURL=streaming.js.map

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

@@ -1 +0,0 @@
-{"version":3,"file":"streaming.js","sourceRoot":"","sources":["../../../src/agents/streaming.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;AA0GlC,gDAUC;AAGD,kEAOC;AAED,0EAeC;;AAvID,6DAQ8B;AAE9B,8CAAkD;AAClD,gDAA8C;AAE9C,+DAQ+B;AAC/B,qDAAqD;AACrD,oGAOyD;AACzD,4CAAsC;AAEtC,MAAM,gBAAgB,GAAG,CAAC,KAAK,CAAC,CAAC;AAEjC,MAAM,QAAQ,GAAG;IACf,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,sCAAiB,CAAa,EAAE,SAAS,EAAE,yDAAwB,EAAE;IAC7F,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,mCAAc,CAAa,EAAE,SAAS,EAAE,uDAAsB,EAAE;IACxF,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,uCAAkB,CAAa,EAAE,SAAS,EAAE,qDAAoB,EAAE;IAC1F,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,uCAAkB,CAAa,EAAE,SAAS,EAAE,2DAA0B,EAAE;CACjG,CAAC;AAEF,SAAS,iBAAiB,CAAC,MAA0B;IACnD,MAAM,aAAa,GAAG,eAAe,CAAC,MAAM,CAAC,CAAC;IAC9C,MAAM,eAAe,GAAG,MAAyB,CAAC;IAClD,OAAO,MAAM,CAAC,MAAM,CAAC,aAAa,EAAE,eAAe,CAAC,CAAC;AACvD,CAAC;AAED,SAAgB,eAAe,CAAC,MAA0B;;;;YACxD,KAA0B,eAAA,WAAA,sBAAA,MAAM,CAAA,YAAA,6FAAE,CAAC;gBAAT,sBAAM;gBAAN,WAAM;gBAArB,MAAM,KAAK,KAAA,CAAA;gBACpB,MAAM,IAAI,GAAG,oBAAoB,CAAC,KAAK,CAAC,CAAC;gBACzC,4BAAM,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,CAAA,CAAC;YAC3C,CAAC;;;;;;;;;IACH,CAAC;CAAA;AAED,SAAS,oBAAoB,CAAC,KAAmB;IAC/C,IAAI,CAAC;QACH,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACxC,QAAQ,KAAK,CAAC,KAAK,EAAE,CAAC;YACpB,KAAK,uCAAkB,CAAC,kBAAkB;gBACxC,OAAO,IAAA,+DAA8B,EAAC,QAAQ,CAAC,CAAC;YAClD,KAAK,uCAAkB,CAAC,kBAAkB;gBACxC,OAAO,IAAA,yDAAwB,EAAC,QAAQ,CAAC,CAAC;YAC5C,OAAO,CAAC,CAAC,CAAC;gBACR,KAAK,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,QAAQ,EAAE,CAAC;oBAC7C,IAAI,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE,CAAC;wBACjC,OAAO,SAAS,CAAC,QAAQ,CAAC,CAAC;oBAC7B,CAAC;gBACH,CAAC;gBAED,OAAO,QAAQ,CAAC;YAClB,CAAC;QACH,CAAC;IACH,CAAC;IAAC,OAAO,EAAE,EAAE,CAAC;QACZ,kBAAM,CAAC,KAAK,CAAC,+BAA+B,KAAK,CAAC,KAAK,aAAa,EAAE,EAAE,CAAC,CAAC;QAC1E,OAAO,KAAK,CAAC,IAAI,CAAC;IACpB,CAAC;AACH,CAAC;AAED,KAAK,UAAU,aAAa,CAAC,cAAgC;IAC3D,MAAM,MAAM,GAAG,sBAAU;QACvB,CAAC,CAAC,MAAM,cAAc,CAAC,YAAY,EAAE;QACrC,CAAC,CAAC,MAAM,cAAc,CAAC,eAAe,EAAE,CAAC;IAE3C,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;QAC9C,MAAM,IAAA,kCAAiB,EAAC,MAAM,CAAC,CAAC;IAClC,CAAC;IACD,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;QACjB,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;IACzC,CAAC;IAED,MAAM,MAAM,GAAG,sBAAU;QACvB,CAAC,CAAC,IAAA,0BAAe,EAAC,MAAM,CAAC,IAAuB,CAAC;QACjD,CAAC,CAAC,IAAA,0BAAe,EAAC,MAAM,CAAC,IAAkC,CAAC,CAAC;IAC/D,OAAO,iBAAiB,CAAC,MAAM,CAAC,CAAC;AACnC,CAAC;AAED,yCAAyC;AAClC,KAAK,UAAU,kBAAkB,CACtC,OAAe,EACf,QAAgB,EAChB,OAA4B;IAE5B,IAAA,sCAAgB,EAAC,QAAQ,CAAC,CAAC;IAC3B,mCAAmC,CAAC,OAAO,CAAC,CAAC;IAC7C,OAAO,CAAC,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;IAE3B,OAAO,aAAa,CAAC,OAAO,CAAC,IAAI,CAAC,0BAA0B,EAAE,QAAQ,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC;AACzF,CAAC;AAED,oDAAoD;AAC7C,KAAK,UAAU,2BAA2B,CAC/C,OAAe,EACf,OAAoC;IAEpC,mCAAmC,CAAC,OAAO,CAAC,CAAC;IAC7C,OAAO,CAAC,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;IAC3B,OAAO,aAAa,CAAC,OAAO,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC;AACpE,CAAC;AAEM,KAAK,UAAU,+BAA+B,CACnD,OAAe,EACf,QAAgB,EAChB,KAAa,EACb,OAAyC;IAEzC,IAAA,sCAAgB,EAAC,QAAQ,CAAC,CAAC;IAC3B,IAAA,mCAAa,EAAC,KAAK,CAAC,CAAC;IACrB,OAAO,CAAC,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;IAE3B,OAAO,aAAa,CAClB,OAAO;SACJ,IAAI,CAAC,sDAAsD,EAAE,QAAQ,EAAE,KAAK,CAAC;SAC7E,IAAI,CAAC,OAAO,CAAC,CACjB,CAAC;AACJ,CAAC;AAED,SAAS,mCAAmC,CAC1C,OAA0D;;IAE1D,IAAI,qBAAqB,IAAI,OAAO,CAAC,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC,mBAAmB,EAAE,CAAC;QAC9E,OAAO,CAAC,IAAI,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,IAAA,sCAAgB,EAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC;IACxF,CAAC;IACD,IAAI,QAAQ,IAAI,OAAO,CAAC,IAAI,KAAI,MAAA,OAAO,CAAC,IAAI,CAAC,MAAM,0CAAE,QAAQ,CAAA,EAAE,CAAC;QAC9D,MAAA,OAAO,CAAC,IAAI,CAAC,MAAM,0CAAE,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,IAAA,sCAAgB,EAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC;IACrF,CAAC;IACD,IAAI,OAAO,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;QACvB,IAAA,mCAAa,EAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACpC,CAAC;IACD,IAAI,gBAAgB,IAAI,OAAO,CAAC,IAAI,KAAI,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,IAAI,CAAC,cAAc,CAAA,EAAE,CAAC;QACrE,IAAA,2CAAqB,EAAC,OAAO,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;IACrD,CAAC;IACD,IAAI,OAAO,CAAC,IAAI,CAAC,WAAW,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,WAAW,GAAG,CAAC,IAAI,OAAO,CAAC,IAAI,CAAC,WAAW,GAAG,CAAC,CAAC,EAAE,CAAC;QAC/F,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;IACzD,CAAC;IACD,IAAI,OAAO,CAAC,IAAI,CAAC,WAAW,IAAI,OAAO,OAAO,CAAC,IAAI,CAAC,WAAW,KAAK,QAAQ,EAAE,CAAC;QAC7E,IAAA,mCAAa,EAAC,CAAC,OAAO,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC;IAC5C,CAAC;IACD,IAAI,MAAA,OAAO,CAAC,IAAI,CAAC,mBAAmB,0CAAE,IAAI,EAAE,CAAC;QAC3C,IAAA,gDAA0B,EAAC,OAAO,CAAC,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC;IACpE,CAAC;IACD,IAAI,OAAO,CAAC,IAAI,CAAC,eAAe,EAAE,CAAC;QACjC,IAAI,CAAC,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,eAAyB,CAAC,EAAE,CAAC;YACvE,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC;QACrE,CAAC;IACH,CAAC;IACD,IAAI,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,IAAI,CAAC,QAAQ,EAAE,CAAC;QAC3B,IAAA,sCAAgB,EAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC1C,CAAC;AACH,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport type { Client, StreamableMethod } from \"@azure-rest/core-client\";\nimport type {\n  CreateRunParameters,\n  CreateThreadAndRunBodyParam,\n  SubmitToolOutputsToRunParameters,\n} from \"../generated/src/index.js\";\nimport {\n  MessageStreamEvent,\n  RunStepStreamEvent,\n  RunStreamEvent,\n  ThreadStreamEvent,\n  type AgentEventMessage,\n  type AgentEventMessageStream,\n  type AgentEventStreamDataOutput,\n} from \"./streamingModels.js\";\nimport type { EventMessage, EventMessageStream } from \"@azure/core-sse\";\nimport { createSseStream } from \"@azure/core-sse\";\nimport { isNodeLike } from \"@azure/core-util\";\nimport type { IncomingMessage } from \"http\";\nimport {\n  validateMessages,\n  validateMetadata,\n  validateRunId,\n  validateThreadId,\n  validateToolResources,\n  validateTools,\n  validateTruncationStrategy,\n} from \"./inputValidations.js\";\nimport { createOpenAIError } from \"./openAIError.js\";\nimport {\n  convertAgentThreadOutput,\n  convertMessageDeltaChunkOutput,\n  convertRunStepDeltaChunk,\n  convertRunStepOutput,\n  convertThreadMessageOutput,\n  convertThreadRunOutput,\n} from \"../customization/convertOutputModelsFromWire.js\";\nimport { logger } from \"../logger.js\";\n\nconst expectedStatuses = [\"200\"];\n\nconst handlers = [\n  { events: Object.values(ThreadStreamEvent) as string[], converter: convertAgentThreadOutput },\n  { events: Object.values(RunStreamEvent) as string[], converter: convertThreadRunOutput },\n  { events: Object.values(RunStepStreamEvent) as string[], converter: convertRunStepOutput },\n  { events: Object.values(MessageStreamEvent) as string[], converter: convertThreadMessageOutput },\n];\n\nfunction createAgentStream(stream: EventMessageStream): AgentEventMessageStream {\n  const asyncIterator = toAsyncIterable(stream);\n  const asyncDisposable = stream as AsyncDisposable;\n  return Object.assign(asyncIterator, asyncDisposable);\n}\n\nasync function* toAsyncIterable(stream: EventMessageStream): AsyncIterable<AgentEventMessage> {\n  for await (const event of stream) {\n    const data = deserializeEventData(event);\n    yield { data: data, event: event.event };\n  }\n}\n\nfunction deserializeEventData(event: EventMessage): AgentEventStreamDataOutput {\n  try {\n    const jsonData = JSON.parse(event.data);\n    switch (event.event) {\n      case MessageStreamEvent.ThreadMessageDelta:\n        return convertMessageDeltaChunkOutput(jsonData);\n      case RunStepStreamEvent.ThreadRunStepDelta:\n        return convertRunStepDeltaChunk(jsonData);\n      default: {\n        for (const { events, converter } of handlers) {\n          if (events.includes(event.event)) {\n            return converter(jsonData);\n          }\n        }\n\n        return jsonData;\n      }\n    }\n  } catch (ex) {\n    logger.error(`Failed to parse event data  ${event.event} - error: ${ex}`);\n    return event.data;\n  }\n}\n\nasync function processStream(streamResponse: StreamableMethod): Promise<AgentEventMessageStream> {\n  const result = isNodeLike\n    ? await streamResponse.asNodeStream()\n    : await streamResponse.asBrowserStream();\n\n  if (!expectedStatuses.includes(result.status)) {\n    throw createOpenAIError(result);\n  }\n  if (!result.body) {\n    throw new Error(\"No body in response\");\n  }\n\n  const stream = isNodeLike\n    ? createSseStream(result.body as IncomingMessage)\n    : createSseStream(result.body as ReadableStream<Uint8Array>);\n  return createAgentStream(stream);\n}\n\n/** Create a run and stream the events */\nexport async function createRunStreaming(\n  context: Client,\n  threadId: string,\n  options: CreateRunParameters,\n): Promise<AgentEventMessageStream> {\n  validateThreadId(threadId);\n  validateCreateThreadAndRunBodyParam(options);\n  options.body.stream = true;\n\n  return processStream(context.path(\"/threads/{threadId}/runs\", threadId).post(options));\n}\n\n/** Create a thread and run and stream the events */\nexport async function createThreadAndRunStreaming(\n  context: Client,\n  options: CreateThreadAndRunBodyParam,\n): Promise<AgentEventMessageStream> {\n  validateCreateThreadAndRunBodyParam(options);\n  options.body.stream = true;\n  return processStream(context.path(\"/threads/runs\").post(options));\n}\n\nexport async function submitToolOutputsToRunStreaming(\n  context: Client,\n  threadId: string,\n  runId: string,\n  options: SubmitToolOutputsToRunParameters,\n): Promise<AgentEventMessageStream> {\n  validateThreadId(threadId);\n  validateRunId(runId);\n  options.body.stream = true;\n\n  return processStream(\n    context\n      .path(\"/threads/{threadId}/runs/{runId}/submit_tool_outputs\", threadId, runId)\n      .post(options),\n  );\n}\n\nfunction validateCreateThreadAndRunBodyParam(\n  options: CreateRunParameters | CreateThreadAndRunBodyParam,\n): void {\n  if (\"additional_messages\" in options.body && options.body.additional_messages) {\n    options.body.additional_messages.forEach((message) => validateMessages(message.role));\n  }\n  if (\"thread\" in options.body && options.body.thread?.messages) {\n    options.body.thread?.messages.forEach((message) => validateMessages(message.role));\n  }\n  if (options.body.tools) {\n    validateTools(options.body.tools);\n  }\n  if (\"tool_resources\" in options.body && options?.body.tool_resources) {\n    validateToolResources(options.body.tool_resources);\n  }\n  if (options.body.temperature && (options.body.temperature < 0 || options.body.temperature > 2)) {\n    throw new Error(\"Temperature must be between 0 and 2\");\n  }\n  if (options.body.tool_choice && typeof options.body.tool_choice !== \"string\") {\n    validateTools([options.body.tool_choice]);\n  }\n  if (options.body.truncation_strategy?.type) {\n    validateTruncationStrategy(options.body.truncation_strategy.type);\n  }\n  if (options.body.response_format) {\n    if (![\"json\", \"text\"].includes(options.body.response_format as string)) {\n      throw new Error(\"Response format must be either 'json' or 'text'\");\n    }\n  }\n  if (options?.body.metadata) {\n    validateMetadata(options.body.metadata);\n  }\n}\n"]}

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

@@ -1,101 +0,0 @@
-import type { MessageDeltaChunk, RunStepDeltaChunk } from "../customization/streamingModels.js";
-import type { AgentThreadOutput, RunStepOutput, ThreadMessageOutput, ThreadRunOutput } from "../customization/outputModels.js";
-/**
-Each event in a server-sent events stream has an `event` and `data` property:
-  
-  ```
-  event: thread.created
-  data: {"id": "thread_123", "object": "thread", ...}
-  ```
-  
-  We emit events whenever a new object is created, transitions to a new state, or is being
-  streamed in parts (deltas). For example, we emit `thread.run.created` when a new run
-  is created, `thread.run.completed` when a run completes, and so on. When an Agent chooses
-  to create a message during a run, we emit a `thread.message.created event`, a
-  `thread.message.in_progress` event, many `thread.message.delta` events, and finally a
-  `thread.message.completed` event.
-  
-  We may add additional events over time, so we recommend handling unknown events gracefully
-  in your code.**/
-export interface AgentEventMessage {
-    /** The data of the event. The data can be of type AgentThreadOutput, ThreadRunOutput, RunStepOutput, ThreadMessageOutput, MessageDeltaChunk,RunStepDeltaChunk  */
-    data: AgentEventStreamDataOutput;
-    /** The type of the event. */
-    event: AgentStreamEventType | string;
-}
-/** Represents a stream event data in the agent. */
-export type AgentEventStreamDataOutput = AgentThreadOutput | ThreadRunOutput | RunStepOutput | ThreadMessageOutput | MessageDeltaChunk | RunStepDeltaChunk | string;
-/** Thread operation related streaming events */
-export declare enum ThreadStreamEvent {
-    /** Event sent when a new thread is created. The data of this event is of type AgentThread */
-    ThreadCreated = "thread.created"
-}
-/** Run operation related streaming events */
-export declare enum RunStreamEvent {
-    /** Event sent when a new run is created. The data of this event is of type ThreadRun */
-    ThreadRunCreated = "thread.run.created",
-    /** Event sent when a run moves to `queued` status. The data of this event is of type ThreadRun */
-    ThreadRunQueued = "thread.run.queued",
-    /** Event sent when a run moves to `in_progress` status. The data of this event is of type ThreadRun */
-    ThreadRunInProgress = "thread.run.in_progress",
-    /** Event sent when a run moves to `requires_action` status. The data of this event is of type ThreadRun */
-    ThreadRunRequiresAction = "thread.run.requires_action",
-    /** Event sent when a run is completed. The data of this event is of type ThreadRun */
-    ThreadRunCompleted = "thread.run.completed",
-    /** Event sent when a run fails. The data of this event is of type ThreadRun */
-    ThreadRunFailed = "thread.run.failed",
-    /** Event sent when a run moves to `cancelling` status. The data of this event is of type ThreadRun */
-    ThreadRunCancelling = "thread.run.cancelling",
-    /** Event sent when a run is cancelled. The data of this event is of type ThreadRun */
-    ThreadRunCancelled = "thread.run.cancelled",
-    /** Event sent when a run is expired. The data of this event is of type ThreadRun */
-    ThreadRunExpired = "thread.run.expired"
-}
-/** Run step operation related streaming events */
-export declare enum RunStepStreamEvent {
-    /** Event sent when a new thread run step is created. The data of this event is of type RunStep */
-    ThreadRunStepCreated = "thread.run.step.created",
-    /** Event sent when a run step moves to `in_progress` status. The data of this event is of type RunStep */
-    ThreadRunStepInProgress = "thread.run.step.in_progress",
-    /** Event sent when a run step is being streamed. The data of this event is of type RunStepDeltaChunk */
-    ThreadRunStepDelta = "thread.run.step.delta",
-    /** Event sent when a run step is completed. The data of this event is of type RunStep */
-    ThreadRunStepCompleted = "thread.run.step.completed",
-    /** Event sent when a run step fails. The data of this event is of type RunStep */
-    ThreadRunStepFailed = "thread.run.step.failed",
-    /** Event sent when a run step is cancelled. The data of this event is of type RunStep */
-    ThreadRunStepCancelled = "thread.run.step.cancelled",
-    /** Event sent when a run step is expired. The data of this event is of type RunStep */
-    ThreadRunStepExpired = "thread.run.step.expired"
-}
-/** Message operation related streaming events */
-export declare enum MessageStreamEvent {
-    /** Event sent when a new message is created. The data of this event is of type ThreadMessage */
-    ThreadMessageCreated = "thread.message.created",
-    /** Event sent when a message moves to `in_progress` status. The data of this event is of type ThreadMessage */
-    ThreadMessageInProgress = "thread.message.in_progress",
-    /** Event sent when a message is being streamed. The data of this event is of type MessageDeltaChunk */
-    ThreadMessageDelta = "thread.message.delta",
-    /** Event sent when a message is completed. The data of this event is of type ThreadMessage */
-    ThreadMessageCompleted = "thread.message.completed",
-    /** Event sent before a message is completed. The data of this event is of type ThreadMessage */
-    ThreadMessageIncomplete = "thread.message.incomplete"
-}
-/** Terminal event indicating a server side error while streaming. */
-export declare enum ErrorEvent {
-    /** Event sent when an error occurs, such as an internal server error or a timeout. */
-    Error = "error"
-}
-/** Terminal event indicating the successful end of a stream. */
-export declare enum DoneEvent {
-    /** Event sent when the stream is done. */
-    Done = "done"
-}
-/**
-  Represents the type of an agent stream event.
- */
-export type AgentStreamEventType = ThreadStreamEvent | RunStreamEvent | RunStepStreamEvent | MessageStreamEvent | ErrorEvent | DoneEvent;
-/** Represents a stream of agent event message. */
-export interface AgentEventMessageStream extends AsyncDisposable, AsyncIterable<AgentEventMessage> {
-}
-//# sourceMappingURL=streamingModels.d.ts.map

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

@@ -1 +0,0 @@
-{"version":3,"file":"streamingModels.d.ts","sourceRoot":"","sources":["../../../src/agents/streamingModels.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,MAAM,qCAAqC,CAAC;AAChG,OAAO,KAAK,EACV,iBAAiB,EACjB,aAAa,EACb,mBAAmB,EACnB,eAAe,EAChB,MAAM,kCAAkC,CAAC;AAE1C;;;;;;;;;;;;;;;;kBAgBkB;AAClB,MAAM,WAAW,iBAAiB;IAChC,kKAAkK;IAClK,IAAI,EAAE,0BAA0B,CAAC;IACjC,6BAA6B;IAC7B,KAAK,EAAE,oBAAoB,GAAG,MAAM,CAAC;CACtC;AAED,mDAAmD;AACnD,MAAM,MAAM,0BAA0B,GAClC,iBAAiB,GACjB,eAAe,GACf,aAAa,GACb,mBAAmB,GACnB,iBAAiB,GACjB,iBAAiB,GACjB,MAAM,CAAC;AAEX,gDAAgD;AAChD,oBAAY,iBAAiB;IAC3B,6FAA6F;IAC7F,aAAa,mBAAmB;CACjC;AAED,6CAA6C;AAC7C,oBAAY,cAAc;IACxB,wFAAwF;IACxF,gBAAgB,uBAAuB;IAEvC,kGAAkG;IAClG,eAAe,sBAAsB;IAErC,uGAAuG;IACvG,mBAAmB,2BAA2B;IAE9C,2GAA2G;IAC3G,uBAAuB,+BAA+B;IAEtD,sFAAsF;IACtF,kBAAkB,yBAAyB;IAE3C,+EAA+E;IAC/E,eAAe,sBAAsB;IAErC,sGAAsG;IACtG,mBAAmB,0BAA0B;IAE7C,sFAAsF;IACtF,kBAAkB,yBAAyB;IAE3C,oFAAoF;IACpF,gBAAgB,uBAAuB;CACxC;AAED,kDAAkD;AAClD,oBAAY,kBAAkB;IAC5B,kGAAkG;IAClG,oBAAoB,4BAA4B;IAEhD,0GAA0G;IAC1G,uBAAuB,gCAAgC;IAEvD,wGAAwG;IACxG,kBAAkB,0BAA0B;IAE5C,yFAAyF;IACzF,sBAAsB,8BAA8B;IAEpD,kFAAkF;IAClF,mBAAmB,2BAA2B;IAE9C,yFAAyF;IACzF,sBAAsB,8BAA8B;IAEpD,uFAAuF;IACvF,oBAAoB,4BAA4B;CACjD;AAED,iDAAiD;AACjD,oBAAY,kBAAkB;IAC5B,gGAAgG;IAChG,oBAAoB,2BAA2B;IAE/C,+GAA+G;IAC/G,uBAAuB,+BAA+B;IAEtD,uGAAuG;IACvG,kBAAkB,yBAAyB;IAE3C,8FAA8F;IAC9F,sBAAsB,6BAA6B;IAEnD,gGAAgG;IAChG,uBAAuB,8BAA8B;CACtD;AAED,qEAAqE;AACrE,oBAAY,UAAU;IACpB,sFAAsF;IACtF,KAAK,UAAU;CAChB;AAED,gEAAgE;AAChE,oBAAY,SAAS;IACnB,0CAA0C;IAC1C,IAAI,SAAS;CACd;AAED;;GAEG;AACH,MAAM,MAAM,oBAAoB,GAC5B,iBAAiB,GACjB,cAAc,GACd,kBAAkB,GAClB,kBAAkB,GAClB,UAAU,GACV,SAAS,CAAC;AAEd,kDAAkD;AAClD,MAAM,WAAW,uBACf,SAAQ,eAAe,EACrB,aAAa,CAAC,iBAAiB,CAAC;CAAG"}