@azure/ai-projects 1.0.0-beta.1

package/dist/browser/agents/runs.js

@@ -0,0 +1,222 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+import { operationOptionsToRequestParameters } from "@azure-rest/core-client";
+import { validateLimit, validateMessages, validateMetadata, validateOrder, validateRunId, validateThreadId, validateTools, validateTruncationStrategy, } from "./inputValidations.js";
+import { TracingUtility } from "../tracing.js";
+import { traceEndCreateOrUpdateRun, traceEndSubmitToolOutputsToRun, traceStartCreateRun, traceStartCreateThreadAndRun, traceStartSubmitToolOutputsToRun, } from "./runTrace.js";
+import { traceStartAgentGeneric } from "./traceUtility.js";
+import { createRunStreaming, createThreadAndRunStreaming, submitToolOutputsToRunStreaming, } from "./streaming.js";
+import * as ConverterToWire from "../customization/convertModelsToWrite.js";
+import * as ConvertFromWire from "../customization/convertOutputModelsFromWire.js";
+import { convertToListQueryParameters } from "../customization/convertParametersToWire.js";
+import { createOpenAIError } from "./openAIError.js";
+const expectedStatuses = ["200"];
+/** Creates and starts a new run of the specified thread using the specified agent. */
+export function createRun(context, threadId, assistantId, options) {
+    const createRunOptions = Object.assign(Object.assign({}, operationOptionsToRequestParameters(options)), { body: Object.assign(Object.assign({}, ConverterToWire.convertCreateRunOptions(Object.assign(Object.assign({}, options), { assistantId }))), { stream: false }) });
+    validateThreadId(threadId);
+    validateCreateRunParameters(createRunOptions);
+    async function executeCreateRun() {
+        const output = await TracingUtility.withSpan("CreateRun", createRunOptions, async (updateOptions) => {
+            const result = await context.path("/threads/{threadId}/runs", threadId).post(updateOptions);
+            if (!expectedStatuses.includes(result.status)) {
+                const error = createOpenAIError(result);
+                throw error;
+            }
+            return result.body;
+        }, (span, updatedOptions) => traceStartCreateRun(span, updatedOptions, threadId), traceEndCreateOrUpdateRun);
+        return ConvertFromWire.convertThreadRunOutput(output);
+    }
+    return {
+        then: function (onFulfilled, onRejected) {
+            return executeCreateRun().then(onFulfilled, onRejected).catch(onRejected);
+        },
+        async stream() {
+            return createRunStreaming(context, threadId, createRunOptions);
+        },
+    };
+}
+/** Gets a list of runs for a specified thread. */
+export async function listRuns(context, threadId, options = {}) {
+    const listRunOptions = Object.assign(Object.assign({}, operationOptionsToRequestParameters(options)), { queryParameters: convertToListQueryParameters(options) });
+    validateListRunsParameters(threadId, options);
+    return TracingUtility.withSpan("ListRuns", listRunOptions || {}, async (updateOptions) => {
+        const result = await context.path("/threads/{threadId}/runs", threadId).get(updateOptions);
+        if (!expectedStatuses.includes(result.status)) {
+            throw createOpenAIError(result);
+        }
+        return ConvertFromWire.convertOpenAIPageableListOfThreadRunOutput(result.body);
+    }, (span, updatedOptions) => traceStartAgentGeneric(span, Object.assign(Object.assign({}, updatedOptions), { tracingAttributeOptions: { threadId: threadId } })));
+}
+/** Gets an existing run from an existing thread. */
+export async function getRun(context, threadId, runId, options = {}) {
+    validateThreadId(threadId);
+    validateRunId(runId);
+    const getRunOptions = Object.assign({}, operationOptionsToRequestParameters(options));
+    return 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 createOpenAIError(result);
+        }
+        return ConvertFromWire.convertThreadRunOutput(result.body);
+    }, (span, updatedOptions) => traceStartAgentGeneric(span, Object.assign(Object.assign({}, updatedOptions), { tracingAttributeOptions: { threadId: threadId, runId: runId } })));
+}
+/** Modifies an existing thread run. */
+export async function updateRun(context, threadId, runId, options = {}) {
+    const updateRunOptions = Object.assign(Object.assign({}, operationOptionsToRequestParameters(options)), { body: {
+            metadata: options === null || options === void 0 ? void 0 : options.metadata,
+        } });
+    validateUpdateRunParameters(threadId, runId, updateRunOptions);
+    const response = await 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 createOpenAIError(result);
+        }
+        return result.body;
+    }, (span, updatedOptions) => traceStartAgentGeneric(span, Object.assign(Object.assign({}, updatedOptions), { tracingAttributeOptions: { threadId: threadId, runId: runId } })), 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'. */
+export function submitToolOutputsToRun(context, threadId, runId, toolOutputs, options = {}) {
+    validateThreadId(threadId);
+    validateRunId(runId);
+    const submitToolOutputsOptions = Object.assign(Object.assign({}, operationOptionsToRequestParameters(options)), { body: {
+            tool_outputs: toolOutputs === null || toolOutputs === void 0 ? void 0 : toolOutputs.map(ConverterToWire.convertToolOutput),
+            stream: false,
+        } });
+    async function executeSubmitToolOutputsToRun() {
+        const response = await 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 createOpenAIError(result);
+            }
+            return result.body;
+        }, (span, updatedOptions) => traceStartSubmitToolOutputsToRun(span, updatedOptions, threadId, runId), traceEndSubmitToolOutputsToRun);
+        return ConvertFromWire.convertThreadRunOutput(response);
+    }
+    return {
+        then: function (onFulfilled, onrejected) {
+            return executeSubmitToolOutputsToRun().then(onFulfilled, onrejected).catch(onrejected);
+        },
+        async stream() {
+            return submitToolOutputsToRunStreaming(context, threadId, runId, submitToolOutputsOptions);
+        },
+    };
+}
+/** Cancels a run of an in progress thread. */
+export async function cancelRun(context, threadId, runId, options = {}) {
+    validateThreadId(threadId);
+    validateRunId(runId);
+    const cancelRunOptions = Object.assign({}, operationOptionsToRequestParameters(options));
+    return 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 createOpenAIError(result);
+        }
+        return ConvertFromWire.convertThreadRunOutput(result.body);
+    });
+}
+/** Creates a new thread and immediately starts a run of that thread. */
+export function createThreadAndRun(context, assistantId, options) {
+    const createThreadAndRunOptions = Object.assign(Object.assign({}, 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 TracingUtility.withSpan("CreateThreadAndRun", createThreadAndRunOptions, async (updateOptions) => {
+            const result = await context.path("/threads/runs").post(updateOptions);
+            if (!expectedStatuses.includes(result.status)) {
+                throw createOpenAIError(result);
+            }
+            return result.body;
+        }, traceStartCreateThreadAndRun, traceEndCreateOrUpdateRun);
+        return ConvertFromWire.convertThreadRunOutput(response);
+    }
+    return {
+        then: function (onFulfilled, onrejected) {
+            return executeCreateThreadAndRun().then(onFulfilled, onrejected).catch(onrejected);
+        },
+        async stream() {
+            return createThreadAndRunStreaming(context, createThreadAndRunOptions);
+        },
+    };
+}
+function validateListRunsParameters(thread_id, options) {
+    var _a, _b, _c;
+    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) {
+        validateLimit(options.queryParameters.limit);
+    }
+    if ((_c = options === null || options === void 0 ? void 0 : options.queryParameters) === null || _c === void 0 ? void 0 : _c.order) {
+        validateOrder(options.queryParameters.order);
+    }
+}
+function validateUpdateRunParameters(thread_id, run_id, options) {
+    validateThreadId(thread_id);
+    validateRunId(run_id);
+    if (options === null || options === void 0 ? void 0 : options.body.metadata) {
+        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) => validateMessages(message.role));
+    }
+    if (options.body.tools) {
+        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") {
+        validateTools([options.body.tool_choice]);
+    }
+    if ((_a = options.body.truncation_strategy) === null || _a === void 0 ? void 0 : _a.type) {
+        validateTruncationStrategy(options.body.truncation_strategy.type);
+    }
+    if (options.body.metadata) {
+        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) => validateMessages(message.role));
+    }
+    if (options.body.tools) {
+        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/browser/agents/runs.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"runs.js","sourceRoot":"","sources":["../../../src/agents/runs.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAGlC,OAAO,EAAE,mCAAmC,EAAE,MAAM,yBAAyB,CAAC;AAI9E,OAAO,EACL,aAAa,EACb,gBAAgB,EAChB,gBAAgB,EAChB,aAAa,EACb,aAAa,EACb,gBAAgB,EAChB,aAAa,EACb,0BAA0B,GAC3B,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAC/C,OAAO,EACL,yBAAyB,EACzB,8BAA8B,EAC9B,mBAAmB,EACnB,4BAA4B,EAC5B,gCAAgC,GACjC,MAAM,eAAe,CAAC;AACvB,OAAO,EAAE,sBAAsB,EAAE,MAAM,mBAAmB,CAAC;AAC3D,OAAO,EACL,kBAAkB,EAClB,2BAA2B,EAC3B,+BAA+B,GAChC,MAAM,gBAAgB,CAAC;AAYxB,OAAO,KAAK,eAAe,MAAM,0CAA0C,CAAC;AAC5E,OAAO,KAAK,eAAe,MAAM,iDAAiD,CAAC;AACnF,OAAO,EAAE,4BAA4B,EAAE,MAAM,6CAA6C,CAAC;AAC3F,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AAErD,MAAM,gBAAgB,GAAG,CAAC,KAAK,CAAC,CAAC;AAEjC,sFAAsF;AACtF,MAAM,UAAU,SAAS,CACvB,OAAe,EACf,QAAgB,EAChB,WAAmB,EACnB,OAAgC;IAEhC,MAAM,gBAAgB,mCACjB,mCAAmC,CAAC,OAAO,CAAC,KAC/C,IAAI,kCACC,eAAe,CAAC,uBAAuB,iCAAM,OAAO,KAAE,WAAW,IAAG,KACvE,MAAM,EAAE,KAAK,MAEhB,CAAC;IACF,gBAAgB,CAAC,QAAQ,CAAC,CAAC;IAC3B,2BAA2B,CAAC,gBAAgB,CAAC,CAAC;IAE9C,KAAK,UAAU,gBAAgB;QAC7B,MAAM,MAAM,GAAG,MAAM,cAAc,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,iBAAiB,CAAC,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,mBAAmB,CAAC,IAAI,EAAE,cAAc,EAAE,QAAQ,CAAC,EAC7E,yBAAyB,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,kBAAkB,CAAC,OAAO,EAAE,QAAQ,EAAE,gBAAgB,CAAC,CAAC;QACjE,CAAC;KACF,CAAC;AACJ,CAAC;AAED,kDAAkD;AAClD,MAAM,CAAC,KAAK,UAAU,QAAQ,CAC5B,OAAe,EACf,QAAgB,EAChB,UAAsC,EAAE;IAExC,MAAM,cAAc,mCACf,mCAAmC,CAAC,OAAO,CAAC,KAC/C,eAAe,EAAE,4BAA4B,CAAC,OAAO,CAAC,GACvD,CAAC;IAEF,0BAA0B,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IAC9C,OAAO,cAAc,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,iBAAiB,CAAC,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,sBAAsB,CAAC,IAAI,kCACtB,cAAc,KACjB,uBAAuB,EAAE,EAAE,QAAQ,EAAE,QAAQ,EAAE,IAC/C,CACL,CAAC;AACJ,CAAC;AAED,oDAAoD;AACpD,MAAM,CAAC,KAAK,UAAU,MAAM,CAC1B,OAAe,EACf,QAAgB,EAChB,KAAa,EACb,UAAgC,EAAE;IAElC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;IAC3B,aAAa,CAAC,KAAK,CAAC,CAAC;IACrB,MAAM,aAAa,qBACd,mCAAmC,CAAC,OAAO,CAAC,CAChD,CAAC;IACF,OAAO,cAAc,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,iBAAiB,CAAC,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,sBAAsB,CAAC,IAAI,kCACtB,cAAc,KACjB,uBAAuB,EAAE,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,KAAK,EAAE,IAC7D,CACL,CAAC;AACJ,CAAC;AAED,uCAAuC;AACvC,MAAM,CAAC,KAAK,UAAU,SAAS,CAC7B,OAAe,EACf,QAAgB,EAChB,KAAa,EACb,UAAmC,EAAE;IAErC,MAAM,gBAAgB,mCACjB,mCAAmC,CAAC,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,cAAc,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,iBAAiB,CAAC,MAAM,CAAC,CAAC;QAClC,CAAC;QACD,OAAO,MAAM,CAAC,IAAI,CAAC;IACrB,CAAC,EACD,CAAC,IAAI,EAAE,cAAc,EAAE,EAAE,CACvB,sBAAsB,CAAC,IAAI,kCACtB,cAAc,KACjB,uBAAuB,EAAE,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,KAAK,EAAE,IAC7D,EACJ,yBAAyB,CAC1B,CAAC;IAEF,OAAO,eAAe,CAAC,sBAAsB,CAAC,QAAQ,CAAC,CAAC;AAC1D,CAAC;AAED,0MAA0M;AAC1M,MAAM,UAAU,sBAAsB,CACpC,OAAe,EACf,QAAgB,EAChB,KAAa,EACb,WAA2C,EAC3C,UAAgD,EAAE;IAElD,gBAAgB,CAAC,QAAQ,CAAC,CAAC;IAC3B,aAAa,CAAC,KAAK,CAAC,CAAC;IACrB,MAAM,wBAAwB,mCACzB,mCAAmC,CAAC,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,cAAc,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,iBAAiB,CAAC,MAAM,CAAC,CAAC;YAClC,CAAC;YACD,OAAO,MAAM,CAAC,IAAI,CAAC;QACrB,CAAC,EACD,CAAC,IAAI,EAAE,cAAc,EAAE,EAAE,CACvB,gCAAgC,CAAC,IAAI,EAAE,cAAc,EAAE,QAAQ,EAAE,KAAK,CAAC,EACzE,8BAA8B,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,+BAA+B,CAAC,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,wBAAwB,CAAC,CAAC;QAC7F,CAAC;KACF,CAAC;AACJ,CAAC;AAED,8CAA8C;AAC9C,MAAM,CAAC,KAAK,UAAU,SAAS,CAC7B,OAAe,EACf,QAAgB,EAChB,KAAa,EACb,UAAmC,EAAE;IAErC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;IAC3B,aAAa,CAAC,KAAK,CAAC,CAAC;IACrB,MAAM,gBAAgB,qBACjB,mCAAmC,CAAC,OAAO,CAAC,CAChD,CAAC;IACF,OAAO,cAAc,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,iBAAiB,CAAC,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,MAAM,UAAU,kBAAkB,CAChC,OAAe,EACf,WAAmB,EACnB,OAAyC;IAEzC,MAAM,yBAAyB,mCAC1B,mCAAmC,CAAC,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,cAAc,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,iBAAiB,CAAC,MAAM,CAAC,CAAC;YAClC,CAAC;YAED,OAAO,MAAM,CAAC,IAAI,CAAC;QACrB,CAAC,EACD,4BAA4B,EAC5B,yBAAyB,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,2BAA2B,CAAC,OAAO,EAAE,yBAAyB,CAAC,CAAC;QACzE,CAAC;KACF,CAAC;AACJ,CAAC;AAED,SAAS,0BAA0B,CACjC,SAAiB,EACjB,OAAgD;;IAEhD,gBAAgB,CAAC,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,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,2BAA2B,CAClC,SAAiB,EACjB,MAAc,EACd,OAAiD;IAEjD,gBAAgB,CAAC,SAAS,CAAC,CAAC;IAC5B,aAAa,CAAC,MAAM,CAAC,CAAC;IACtB,IAAI,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,IAAI,CAAC,QAAQ,EAAE,CAAC;QAC3B,gBAAgB,CAAC,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,gBAAgB,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC;IACxF,CAAC;IACD,IAAI,OAAO,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;QACvB,aAAa,CAAC,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,aAAa,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC;IAC5C,CAAC;IACD,IAAI,MAAA,OAAO,CAAC,IAAI,CAAC,mBAAmB,0CAAE,IAAI,EAAE,CAAC;QAC3C,0BAA0B,CAAC,OAAO,CAAC,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC;IACpE,CAAC;IACD,IAAI,OAAO,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;QAC1B,gBAAgB,CAAC,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,gBAAgB,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC;IACrF,CAAC;IACD,IAAI,OAAO,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;QACvB,aAAa,CAAC,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/browser/agents/streaming.d.ts

@@ -0,0 +1,9 @@
+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/browser/agents/streaming.d.ts.map

@@ -0,0 +1 @@
+{"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/browser/agents/streaming.js

@@ -0,0 +1,135 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+import { __asyncGenerator, __asyncValues, __await } from "tslib";
+import { MessageStreamEvent, RunStepStreamEvent, RunStreamEvent, ThreadStreamEvent, } from "./streamingModels.js";
+import { createSseStream } from "@azure/core-sse";
+import { isNodeLike } from "@azure/core-util";
+import { validateMessages, validateMetadata, validateRunId, validateThreadId, validateToolResources, validateTools, validateTruncationStrategy, } from "./inputValidations.js";
+import { createOpenAIError } from "./openAIError.js";
+import { convertAgentThreadOutput, convertMessageDeltaChunkOutput, convertRunStepDeltaChunk, convertRunStepOutput, convertThreadMessageOutput, convertThreadRunOutput, } from "../customization/convertOutputModelsFromWire.js";
+import { logger } from "../logger.js";
+const expectedStatuses = ["200"];
+const handlers = [
+    { events: Object.values(ThreadStreamEvent), converter: convertAgentThreadOutput },
+    { events: Object.values(RunStreamEvent), converter: convertThreadRunOutput },
+    { events: Object.values(RunStepStreamEvent), converter: convertRunStepOutput },
+    { events: Object.values(MessageStreamEvent), converter: convertThreadMessageOutput },
+];
+function createAgentStream(stream) {
+    const asyncIterator = toAsyncIterable(stream);
+    const asyncDisposable = stream;
+    return Object.assign(asyncIterator, asyncDisposable);
+}
+function toAsyncIterable(stream) {
+    return __asyncGenerator(this, arguments, function* toAsyncIterable_1() {
+        var _a, e_1, _b, _c;
+        try {
+            for (var _d = true, stream_1 = __asyncValues(stream), stream_1_1; stream_1_1 = yield __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 __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 __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 MessageStreamEvent.ThreadMessageDelta:
+                return convertMessageDeltaChunkOutput(jsonData);
+            case RunStepStreamEvent.ThreadRunStepDelta:
+                return convertRunStepDeltaChunk(jsonData);
+            default: {
+                for (const { events, converter } of handlers) {
+                    if (events.includes(event.event)) {
+                        return converter(jsonData);
+                    }
+                }
+                return jsonData;
+            }
+        }
+    }
+    catch (ex) {
+        logger.error(`Failed to parse event data  ${event.event} - error: ${ex}`);
+        return event.data;
+    }
+}
+async function processStream(streamResponse) {
+    const result = isNodeLike
+        ? await streamResponse.asNodeStream()
+        : await streamResponse.asBrowserStream();
+    if (!expectedStatuses.includes(result.status)) {
+        throw createOpenAIError(result);
+    }
+    if (!result.body) {
+        throw new Error("No body in response");
+    }
+    const stream = isNodeLike
+        ? createSseStream(result.body)
+        : createSseStream(result.body);
+    return createAgentStream(stream);
+}
+/** Create a run and stream the events */
+export async function createRunStreaming(context, threadId, options) {
+    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 */
+export async function createThreadAndRunStreaming(context, options) {
+    validateCreateThreadAndRunBodyParam(options);
+    options.body.stream = true;
+    return processStream(context.path("/threads/runs").post(options));
+}
+export async function submitToolOutputsToRunStreaming(context, threadId, runId, options) {
+    validateThreadId(threadId);
+    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) => 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) => validateMessages(message.role));
+    }
+    if (options.body.tools) {
+        validateTools(options.body.tools);
+    }
+    if ("tool_resources" in options.body && (options === null || options === void 0 ? void 0 : options.body.tool_resources)) {
+        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") {
+        validateTools([options.body.tool_choice]);
+    }
+    if ((_c = options.body.truncation_strategy) === null || _c === void 0 ? void 0 : _c.type) {
+        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) {
+        validateMetadata(options.body.metadata);
+    }
+}
+//# sourceMappingURL=streaming.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"streaming.js","sourceRoot":"","sources":["../../../src/agents/streaming.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;;AAQlC,OAAO,EACL,kBAAkB,EAClB,kBAAkB,EAClB,cAAc,EACd,iBAAiB,GAIlB,MAAM,sBAAsB,CAAC;AAE9B,OAAO,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAC;AAClD,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAE9C,OAAO,EACL,gBAAgB,EAChB,gBAAgB,EAChB,aAAa,EACb,gBAAgB,EAChB,qBAAqB,EACrB,aAAa,EACb,0BAA0B,GAC3B,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AACrD,OAAO,EACL,wBAAwB,EACxB,8BAA8B,EAC9B,wBAAwB,EACxB,oBAAoB,EACpB,0BAA0B,EAC1B,sBAAsB,GACvB,MAAM,iDAAiD,CAAC;AACzD,OAAO,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AAEtC,MAAM,gBAAgB,GAAG,CAAC,KAAK,CAAC,CAAC;AAEjC,MAAM,QAAQ,GAAG;IACf,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,iBAAiB,CAAa,EAAE,SAAS,EAAE,wBAAwB,EAAE;IAC7F,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,cAAc,CAAa,EAAE,SAAS,EAAE,sBAAsB,EAAE;IACxF,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,kBAAkB,CAAa,EAAE,SAAS,EAAE,oBAAoB,EAAE;IAC1F,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,kBAAkB,CAAa,EAAE,SAAS,EAAE,0BAA0B,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,cAAA,MAAM,CAAA,YAAA,qFAAE,CAAC;gBAAT,sBAAM;gBAAN,WAAM;gBAArB,MAAM,KAAK,KAAA,CAAA;gBACpB,MAAM,IAAI,GAAG,oBAAoB,CAAC,KAAK,CAAC,CAAC;gBACzC,oBAAM,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,kBAAkB,CAAC,kBAAkB;gBACxC,OAAO,8BAA8B,CAAC,QAAQ,CAAC,CAAC;YAClD,KAAK,kBAAkB,CAAC,kBAAkB;gBACxC,OAAO,wBAAwB,CAAC,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,MAAM,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,UAAU;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,iBAAiB,CAAC,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,UAAU;QACvB,CAAC,CAAC,eAAe,CAAC,MAAM,CAAC,IAAuB,CAAC;QACjD,CAAC,CAAC,eAAe,CAAC,MAAM,CAAC,IAAkC,CAAC,CAAC;IAC/D,OAAO,iBAAiB,CAAC,MAAM,CAAC,CAAC;AACnC,CAAC;AAED,yCAAyC;AACzC,MAAM,CAAC,KAAK,UAAU,kBAAkB,CACtC,OAAe,EACf,QAAgB,EAChB,OAA4B;IAE5B,gBAAgB,CAAC,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;AACpD,MAAM,CAAC,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;AAED,MAAM,CAAC,KAAK,UAAU,+BAA+B,CACnD,OAAe,EACf,QAAgB,EAChB,KAAa,EACb,OAAyC;IAEzC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;IAC3B,aAAa,CAAC,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,gBAAgB,CAAC,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,gBAAgB,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC;IACrF,CAAC;IACD,IAAI,OAAO,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;QACvB,aAAa,CAAC,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,qBAAqB,CAAC,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,aAAa,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC;IAC5C,CAAC;IACD,IAAI,MAAA,OAAO,CAAC,IAAI,CAAC,mBAAmB,0CAAE,IAAI,EAAE,CAAC;QAC3C,0BAA0B,CAAC,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,gBAAgB,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC1C,CAAC;AACH,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport type { Client, 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/browser/agents/streamingModels.d.ts

@@ -0,0 +1,101 @@
+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/browser/agents/streamingModels.d.ts.map

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

package/dist/browser/agents/streamingModels.js

@@ -0,0 +1,75 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+/** Thread operation related streaming events */
+export var ThreadStreamEvent;
+(function (ThreadStreamEvent) {
+    /** Event sent when a new thread is created. The data of this event is of type AgentThread */
+    ThreadStreamEvent["ThreadCreated"] = "thread.created";
+})(ThreadStreamEvent || (ThreadStreamEvent = {}));
+/** Run operation related streaming events */
+export var RunStreamEvent;
+(function (RunStreamEvent) {
+    /** Event sent when a new run is created. The data of this event is of type ThreadRun */
+    RunStreamEvent["ThreadRunCreated"] = "thread.run.created";
+    /** Event sent when a run moves to `queued` status. The data of this event is of type ThreadRun */
+    RunStreamEvent["ThreadRunQueued"] = "thread.run.queued";
+    /** Event sent when a run moves to `in_progress` status. The data of this event is of type ThreadRun */
+    RunStreamEvent["ThreadRunInProgress"] = "thread.run.in_progress";
+    /** Event sent when a run moves to `requires_action` status. The data of this event is of type ThreadRun */
+    RunStreamEvent["ThreadRunRequiresAction"] = "thread.run.requires_action";
+    /** Event sent when a run is completed. The data of this event is of type ThreadRun */
+    RunStreamEvent["ThreadRunCompleted"] = "thread.run.completed";
+    /** Event sent when a run fails. The data of this event is of type ThreadRun */
+    RunStreamEvent["ThreadRunFailed"] = "thread.run.failed";
+    /** Event sent when a run moves to `cancelling` status. The data of this event is of type ThreadRun */
+    RunStreamEvent["ThreadRunCancelling"] = "thread.run.cancelling";
+    /** Event sent when a run is cancelled. The data of this event is of type ThreadRun */
+    RunStreamEvent["ThreadRunCancelled"] = "thread.run.cancelled";
+    /** Event sent when a run is expired. The data of this event is of type ThreadRun */
+    RunStreamEvent["ThreadRunExpired"] = "thread.run.expired";
+})(RunStreamEvent || (RunStreamEvent = {}));
+/** Run step operation related streaming events */
+export var RunStepStreamEvent;
+(function (RunStepStreamEvent) {
+    /** Event sent when a new thread run step is created. The data of this event is of type RunStep */
+    RunStepStreamEvent["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 */
+    RunStepStreamEvent["ThreadRunStepInProgress"] = "thread.run.step.in_progress";
+    /** Event sent when a run step is being streamed. The data of this event is of type RunStepDeltaChunk */
+    RunStepStreamEvent["ThreadRunStepDelta"] = "thread.run.step.delta";
+    /** Event sent when a run step is completed. The data of this event is of type RunStep */
+    RunStepStreamEvent["ThreadRunStepCompleted"] = "thread.run.step.completed";
+    /** Event sent when a run step fails. The data of this event is of type RunStep */
+    RunStepStreamEvent["ThreadRunStepFailed"] = "thread.run.step.failed";
+    /** Event sent when a run step is cancelled. The data of this event is of type RunStep */
+    RunStepStreamEvent["ThreadRunStepCancelled"] = "thread.run.step.cancelled";
+    /** Event sent when a run step is expired. The data of this event is of type RunStep */
+    RunStepStreamEvent["ThreadRunStepExpired"] = "thread.run.step.expired";
+})(RunStepStreamEvent || (RunStepStreamEvent = {}));
+/** Message operation related streaming events */
+export var MessageStreamEvent;
+(function (MessageStreamEvent) {
+    /** Event sent when a new message is created. The data of this event is of type ThreadMessage */
+    MessageStreamEvent["ThreadMessageCreated"] = "thread.message.created";
+    /** Event sent when a message moves to `in_progress` status. The data of this event is of type ThreadMessage */
+    MessageStreamEvent["ThreadMessageInProgress"] = "thread.message.in_progress";
+    /** Event sent when a message is being streamed. The data of this event is of type MessageDeltaChunk */
+    MessageStreamEvent["ThreadMessageDelta"] = "thread.message.delta";
+    /** Event sent when a message is completed. The data of this event is of type ThreadMessage */
+    MessageStreamEvent["ThreadMessageCompleted"] = "thread.message.completed";
+    /** Event sent before a message is completed. The data of this event is of type ThreadMessage */
+    MessageStreamEvent["ThreadMessageIncomplete"] = "thread.message.incomplete";
+})(MessageStreamEvent || (MessageStreamEvent = {}));
+/** Terminal event indicating a server side error while streaming. */
+export var ErrorEvent;
+(function (ErrorEvent) {
+    /** Event sent when an error occurs, such as an internal server error or a timeout. */
+    ErrorEvent["Error"] = "error";
+})(ErrorEvent || (ErrorEvent = {}));
+/** Terminal event indicating the successful end of a stream. */
+export var DoneEvent;
+(function (DoneEvent) {
+    /** Event sent when the stream is done. */
+    DoneEvent["Done"] = "done";
+})(DoneEvent || (DoneEvent = {}));
+//# sourceMappingURL=streamingModels.js.map