@azure/ai-projects 1.0.0-beta.1

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

@@ -0,0 +1 @@
+{"version":3,"file":"inputValidations.js","sourceRoot":"","sources":["../../../src/agents/inputValidations.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAQlC,MAAM,UAAU,2BAA2B,CAAC,YAAqC;IAC/E,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,UAAU,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC;QACnF,MAAM,IAAI,KAAK,CAAC,+DAA+D,CAAC,CAAC;IACnF,CAAC;AACH,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,QAAgB;IAC/C,IAAI,CAAC,QAAQ,EAAE,CAAC;QACd,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;IAC3C,CAAC;AACH,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,KAAa;IACzC,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;IACxC,CAAC;AACH,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,KAAa;IACzC,IAAI,KAAK,GAAG,CAAC,IAAI,KAAK,GAAG,GAAG,EAAE,CAAC;QAC7B,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC;IACrD,CAAC;AACH,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,KAAa;IACzC,IAAI,CAAC,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;QACrC,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAC;IACnD,CAAC;AACH,CAAC;AAED,IAAK,KAQJ;AARD,WAAK,KAAK;IACR,6CAAoC,CAAA;IACpC,mCAA0B,CAAA;IAC1B,8BAAqB,CAAA;IACrB,yCAAgC,CAAA;IAChC,6CAAoC,CAAA;IACpC,qDAA4C,CAAA;IAC5C,0CAAiC,CAAA;AACnC,CAAC,EARI,KAAK,KAAL,KAAK,QAQT;AAED,MAAM,UAAU,aAAa,CAAC,KAA4B;IACxD,IAAI,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,IAAwB,CAAC,CAAC,EAAE,CAAC;QACnF,MAAM,IAAI,KAAK,CACb,yJAAyJ,CAC1J,CAAC;IACJ,CAAC;AACH,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,QAAgC;IAC/D,IAAI,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,MAAM,GAAG,EAAE,EAAE,CAAC;QACtC,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;IACzD,CAAC;IACD,IAAI,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,GAAG,EAAE,CAAC,EAAE,CAAC;QAC7D,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;IAC1D,CAAC;IACD,IAAI,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,GAAG,GAAG,CAAC,EAAE,CAAC;QAChE,MAAM,IAAI,KAAK,CAAC,yCAAyC,CAAC,CAAC;IAC7D,CAAC;AACH,CAAC;AAED,MAAM,UAAU,qBAAqB,CAAC,YAAwC;IAC5E,IAAI,YAAY,CAAC,gBAAgB,EAAE,CAAC;QAClC,IACE,YAAY,CAAC,gBAAgB,CAAC,QAAQ;YACtC,YAAY,CAAC,gBAAgB,CAAC,QAAQ,CAAC,MAAM,GAAG,EAAE,EAClD,CAAC;YACD,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC;IACD,IAAI,YAAY,CAAC,WAAW,EAAE,CAAC;QAC7B,IACE,YAAY,CAAC,WAAW,CAAC,gBAAgB;YACzC,YAAY,CAAC,WAAW,CAAC,gBAAgB,CAAC,MAAM,GAAG,CAAC,EACpD,CAAC;YACD,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;QACzD,CAAC;IACH,CAAC;IACD,IAAI,YAAY,CAAC,eAAe,EAAE,CAAC;QACjC,IAAI,YAAY,CAAC,eAAe,CAAC,OAAO,IAAI,YAAY,CAAC,eAAe,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5F,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;QAC/C,CAAC;IACH,CAAC;AACH,CAAC;AAED,MAAM,UAAU,qBAAqB,CAAC,aAAqB;IACzD,IAAI,CAAC,aAAa,EAAE,CAAC;QACnB,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;IACjD,CAAC;AACH,CAAC;AAED,MAAM,UAAU,cAAc,CAAC,MAAc;IAC3C,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;IACzC,CAAC;AACH,CAAC;AAED,IAAK,eAKJ;AALD,WAAK,eAAe;IAClB,6CAA0B,CAAA;IAC1B,0CAAuB,CAAA;IACvB,oCAAiB,CAAA;IACjB,0CAAuB,CAAA;AACzB,CAAC,EALI,eAAe,KAAf,eAAe,QAKnB;AAED,MAAM,UAAU,wBAAwB,CAAC,MAAc;IACrD,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC,QAAQ,CAAC,MAAyB,CAAC,EAAE,CAAC;QACxE,MAAM,IAAI,KAAK,CACb,qFAAqF,CACtF,CAAC;IACJ,CAAC;AACH,CAAC;AAED,IAAK,QAGJ;AAHD,WAAK,QAAQ;IACX,yBAAa,CAAA;IACb,oCAAwB,CAAA;AAC1B,CAAC,EAHI,QAAQ,KAAR,QAAQ,QAGZ;AAED,MAAM,UAAU,gBAAgB,CAAC,KAAa;IAC5C,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,QAAQ,CAAC,KAAiB,CAAC,EAAE,CAAC;QACzD,MAAM,IAAI,KAAK,CAAC,2CAA2C,CAAC,CAAC;IAC/D,CAAC;AACH,CAAC;AAED,IAAK,kBAGJ;AAHD,WAAK,kBAAkB;IACrB,mCAAa,CAAA;IACb,oDAA8B,CAAA;AAChC,CAAC,EAHI,kBAAkB,KAAlB,kBAAkB,QAGtB;AAED,MAAM,UAAU,0BAA0B,CAAC,KAAa;IACtD,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,kBAAkB,CAAC,CAAC,QAAQ,CAAC,KAA2B,CAAC,EAAE,CAAC;QAC7E,MAAM,IAAI,KAAK,CAAC,+CAA+C,CAAC,CAAC;IACnE,CAAC;AACH,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport type {\n  ToolDefinition,\n  UpdateToolResourcesOptions,\n  VectorStoreDataSource,\n} from \"../generated/src/models.js\";\n\nexport function validateVectorStoreDataType(data_sources: VectorStoreDataSource[]): void {\n  if (!data_sources.some((value) => ![\"uri_asset\", \"id_asset\"].includes(value.type))) {\n    throw new Error(\"Vector store data type must be one of 'uri_asset', 'id_asset'\");\n  }\n}\n\nexport function validateThreadId(threadId: string): void {\n  if (!threadId) {\n    throw new Error(\"Thread ID is required\");\n  }\n}\n\nexport function validateRunId(runId: string): void {\n  if (!runId) {\n    throw new Error(\"Run ID is required\");\n  }\n}\n\nexport function validateLimit(limit: number): void {\n  if (limit < 1 || limit > 100) {\n    throw new Error(\"Limit must be between 1 and 100\");\n  }\n}\n\nexport function validateOrder(order: string): void {\n  if (![\"asc\", \"desc\"].includes(order)) {\n    throw new Error(\"Order must be 'asc' or 'desc'\");\n  }\n}\n\nenum Tools {\n  CodeInterpreter = \"code_interpreter\",\n  FileSearch = \"file_search\",\n  Function = \"function\",\n  BingGrounding = \"bing_grounding\",\n  MicrosoftFabric = \"microsoft_fabric\",\n  SharepointGrounding = \"sharepoint_grounding\",\n  AzureAISearch = \"azure_ai_search\",\n}\n\nexport function validateTools(value: Array<ToolDefinition>): void {\n  if (value.some((tool) => !Object.values(Tools).includes(tool as unknown as Tools))) {\n    throw new Error(\n      \"Tool type must be one of 'code_interpreter', 'file_search', 'function', 'bing_grounding', 'microsoft_fabric', 'sharepoint_grounding', 'azure_ai_search'\",\n    );\n  }\n}\n\nexport function validateMetadata(metadata: Record<string, string>): void {\n  if (Object.keys(metadata).length > 16) {\n    throw new Error(\"Only 16 key/value pairs are allowed\");\n  }\n  if (Object.keys(metadata).some((value) => value.length > 64)) {\n    throw new Error(\"Keys must be less than 64 characters\");\n  }\n  if (Object.values(metadata).some((value) => value.length > 512)) {\n    throw new Error(\"Values must be less than 512 characters\");\n  }\n}\n\nexport function validateToolResources(toolResource: UpdateToolResourcesOptions): void {\n  if (toolResource.code_interpreter) {\n    if (\n      toolResource.code_interpreter.file_ids &&\n      toolResource.code_interpreter.file_ids.length > 20\n    ) {\n      throw new Error(\"A maximum of 20 file IDs are allowed\");\n    }\n  }\n  if (toolResource.file_search) {\n    if (\n      toolResource.file_search.vector_store_ids &&\n      toolResource.file_search.vector_store_ids.length > 1\n    ) {\n      throw new Error(\"Only one vector store ID is allowed\");\n    }\n  }\n  if (toolResource.azure_ai_search) {\n    if (toolResource.azure_ai_search.indexes && toolResource.azure_ai_search.indexes.length > 1) {\n      throw new Error(\"Only one index is allowed\");\n    }\n  }\n}\n\nexport function validateVectorStoreId(vectorStoreId: string): void {\n  if (!vectorStoreId) {\n    throw new Error(\"Vector store ID is required\");\n  }\n}\n\nexport function validateFileId(fileId: string): void {\n  if (!fileId) {\n    throw new Error(\"File ID is required\");\n  }\n}\n\nenum FileBatchStatus {\n  InProgress = \"in_progress\",\n  Completed = \"completed\",\n  Failed = \"failed\",\n  Cancelled = \"cancelled\",\n}\n\nexport function validateFileStatusFilter(filter: string): void {\n  if (!Object.values(FileBatchStatus).includes(filter as FileBatchStatus)) {\n    throw new Error(\n      \"File status filter must be one of 'in_progress', 'completed', 'failed', 'cancelled'\",\n    );\n  }\n}\n\nenum Messages {\n  User = \"user\",\n  Assistants = \"assistant\",\n}\n\nexport function validateMessages(value: string): void {\n  if (!Object.values(Messages).includes(value as Messages)) {\n    throw new Error(\"Role must be either 'user' or 'assistant'\");\n  }\n}\n\nenum TruncationStrategy {\n  Auto = \"auto\",\n  LastMessages = \"last_messages\",\n}\n\nexport function validateTruncationStrategy(value: string): void {\n  if (!Object.values(TruncationStrategy).includes(value as TruncationStrategy)) {\n    throw new Error(\"Role must be either 'auto' or 'last_messages'\");\n  }\n}\n"]}

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

@@ -0,0 +1,11 @@
+import type { Client } from "@azure-rest/core-client";
+import type { OpenAIPageableListOfThreadMessageOutput, ThreadMessageOutput } from "../customization/outputModels.js";
+import type { ThreadMessageOptions } from "../customization/models.js";
+import type { CreateMessageOptionalParams, ListMessagesOptionalParams, UpdateMessageOptionalParams } from "./customModels.js";
+/** Creates a new message on a specified thread. */
+export declare function createMessage(context: Client, threadId: string, messageOptions: ThreadMessageOptions, options?: CreateMessageOptionalParams): Promise<ThreadMessageOutput>;
+/** Gets a list of messages that exist on a thread. */
+export declare function listMessages(context: Client, threadId: string, options?: ListMessagesOptionalParams): Promise<OpenAIPageableListOfThreadMessageOutput>;
+/** Modifies an existing message on an existing thread. */
+export declare function updateMessage(context: Client, threadId: string, messageId: string, options?: UpdateMessageOptionalParams): Promise<ThreadMessageOutput>;
+//# sourceMappingURL=messages.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"messages.d.ts","sourceRoot":"","sources":["../../../src/agents/messages.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,yBAAyB,CAAC;AAEtD,OAAO,KAAK,EACV,uCAAuC,EACvC,mBAAmB,EACpB,MAAM,kCAAkC,CAAC;AAc1C,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,4BAA4B,CAAC;AACvE,OAAO,KAAK,EACV,2BAA2B,EAC3B,0BAA0B,EAC1B,2BAA2B,EAC5B,MAAM,mBAAmB,CAAC;AAO3B,mDAAmD;AACnD,wBAAsB,aAAa,CACjC,OAAO,EAAE,MAAM,EACf,QAAQ,EAAE,MAAM,EAChB,cAAc,EAAE,oBAAoB,EACpC,OAAO,GAAE,2BAAgC,GACxC,OAAO,CAAC,mBAAmB,CAAC,CA0B9B;AAED,sDAAsD;AACtD,wBAAsB,YAAY,CAChC,OAAO,EAAE,MAAM,EACf,QAAQ,EAAE,MAAM,EAChB,OAAO,GAAE,0BAA+B,GACvC,OAAO,CAAC,uCAAuC,CAAC,CA+BlD;AAED,0DAA0D;AAC1D,wBAAsB,aAAa,CACjC,OAAO,EAAE,MAAM,EACf,QAAQ,EAAE,MAAM,EAChB,SAAS,EAAE,MAAM,EACjB,OAAO,GAAE,2BAAgC,GACxC,OAAO,CAAC,mBAAmB,CAAC,CA6B9B"}

package/dist/browser/agents/messages.js

@@ -0,0 +1,101 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+import { operationOptionsToRequestParameters } from "@azure-rest/core-client";
+import { validateMetadata, validateVectorStoreDataType } from "./inputValidations.js";
+import { TracingUtility } from "../tracing.js";
+import { traceEndCreateMessage, traceEndListMessages, traceStartCreateMessage, traceStartListMessages, } from "./messagesTrace.js";
+import { traceStartAgentGeneric } from "./traceUtility.js";
+import * as ConvertFromWire from "../customization/convertOutputModelsFromWire.js";
+import { createOpenAIError } from "./openAIError.js";
+const expectedStatuses = ["200"];
+/** Creates a new message on a specified thread. */
+export async function createMessage(context, threadId, messageOptions, options = {}) {
+    const createOptions = Object.assign(Object.assign({}, operationOptionsToRequestParameters(options)), { body: Object.assign({}, messageOptions) });
+    validateThreadId(threadId);
+    validateCreateMessageParameters(createOptions);
+    const response = await TracingUtility.withSpan("CreateMessage", createOptions, async (updateOptions) => {
+        const result = await context
+            .path("/threads/{threadId}/messages", threadId)
+            .post(updateOptions);
+        if (!expectedStatuses.includes(result.status)) {
+            throw createOpenAIError(result);
+        }
+        return result.body;
+    }, (span, updatedOptions) => traceStartCreateMessage(span, threadId, updatedOptions), traceEndCreateMessage);
+    return ConvertFromWire.convertThreadMessageOutput(response);
+}
+/** Gets a list of messages that exist on a thread. */
+export async function listMessages(context, threadId, options = {}) {
+    const listOptions = Object.assign(Object.assign({}, operationOptionsToRequestParameters(options)), { queryParameters: Object.assign(Object.assign(Object.assign(Object.assign(Object.assign({}, (options.runId && { run_id: options.runId })), (options.limit && { run_id: options.limit })), (options.order && { run_id: options.order })), (options.after && { run_id: options.after })), (options.before && { run_id: options.before })) });
+    validateThreadId(threadId);
+    validateListMessagesParameters(listOptions);
+    const output = await TracingUtility.withSpan("ListMessages", listOptions, async (updateOptions) => {
+        const result = await context
+            .path("/threads/{threadId}/messages", threadId)
+            .get(updateOptions);
+        if (!expectedStatuses.includes(result.status)) {
+            throw createOpenAIError(result);
+        }
+        return result.body;
+    }, (span, updatedOptions) => traceStartListMessages(span, threadId, updatedOptions), traceEndListMessages);
+    return ConvertFromWire.convertOpenAIPageableListOfThreadMessageOutput(output);
+}
+/** Modifies an existing message on an existing thread. */
+export async function updateMessage(context, threadId, messageId, options = {}) {
+    const updateMessageOptions = Object.assign(Object.assign({}, operationOptionsToRequestParameters(options)), { body: Object.assign({}, (options.metadata ? { metadata: options.metadata } : {})) });
+    validateThreadId(threadId);
+    validateMessageId(messageId);
+    const response = await TracingUtility.withSpan("UpdateMessage", updateMessageOptions, async (updateOptions) => {
+        const result = await context
+            .path("/threads/{threadId}/messages/{messageId}", threadId, messageId)
+            .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, messageId: messageId } })));
+    return ConvertFromWire.convertThreadMessageOutput(response);
+}
+function validateThreadId(threadId) {
+    if (!threadId) {
+        throw new Error("Thread ID is required");
+    }
+}
+function validateMessageId(messageId) {
+    if (!messageId) {
+        throw new Error("Message ID is required");
+    }
+}
+function validateCreateMessageParameters(options) {
+    if (options.body.role && !["user", "assistant"].includes(options.body.role)) {
+        throw new Error("Role must be either 'user' or 'assistant'");
+    }
+    if (options.body.metadata) {
+        validateMetadata(options.body.metadata);
+    }
+    if (options.body.attachments) {
+        if (options.body.attachments.some((value) => {
+            value.tools.some((tool) => !["code_interpreter", "file_search"].includes(tool.type));
+        })) {
+            throw new Error("Tool type must be either 'code_interpreter' or 'file_search'");
+        }
+        if (options.body.attachments) {
+            options.body.attachments.forEach((value) => {
+                if (value.data_sources) {
+                    validateVectorStoreDataType(value.data_sources);
+                }
+            });
+        }
+    }
+}
+function validateListMessagesParameters(options) {
+    var _a, _b;
+    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.order) && !["asc", "desc"].includes(options.queryParameters.order)) {
+        throw new Error("Order must be either 'asc' or 'desc'");
+    }
+}
+//# sourceMappingURL=messages.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"messages.js","sourceRoot":"","sources":["../../../src/agents/messages.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAGlC,OAAO,EAAE,mCAAmC,EAAE,MAAM,yBAAyB,CAAC;AAS9E,OAAO,EAAE,gBAAgB,EAAE,2BAA2B,EAAE,MAAM,uBAAuB,CAAC;AACtF,OAAO,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAC/C,OAAO,EACL,qBAAqB,EACrB,oBAAoB,EACpB,uBAAuB,EACvB,sBAAsB,GACvB,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EAAE,sBAAsB,EAAE,MAAM,mBAAmB,CAAC;AAQ3D,OAAO,KAAK,eAAe,MAAM,iDAAiD,CAAC;AACnF,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AAErD,MAAM,gBAAgB,GAAG,CAAC,KAAK,CAAC,CAAC;AAEjC,mDAAmD;AACnD,MAAM,CAAC,KAAK,UAAU,aAAa,CACjC,OAAe,EACf,QAAgB,EAChB,cAAoC,EACpC,UAAuC,EAAE;IAEzC,MAAM,aAAa,mCACd,mCAAmC,CAAC,OAAO,CAAC,KAC/C,IAAI,oBACC,cAAc,IAEpB,CAAC;IAEF,gBAAgB,CAAC,QAAQ,CAAC,CAAC;IAC3B,+BAA+B,CAAC,aAAa,CAAC,CAAC;IAC/C,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,QAAQ,CAC5C,eAAe,EACf,aAAa,EACb,KAAK,EAAE,aAAa,EAAE,EAAE;QACtB,MAAM,MAAM,GAAG,MAAM,OAAO;aACzB,IAAI,CAAC,8BAA8B,EAAE,QAAQ,CAAC;aAC9C,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,CAAC,uBAAuB,CAAC,IAAI,EAAE,QAAQ,EAAE,cAAc,CAAC,EACjF,qBAAqB,CACtB,CAAC;IACF,OAAO,eAAe,CAAC,0BAA0B,CAAC,QAAQ,CAAC,CAAC;AAC9D,CAAC;AAED,sDAAsD;AACtD,MAAM,CAAC,KAAK,UAAU,YAAY,CAChC,OAAe,EACf,QAAgB,EAChB,UAAsC,EAAE;IAExC,MAAM,WAAW,mCACZ,mCAAmC,CAAC,OAAO,CAAC,KAC/C,eAAe,4EACV,CAAC,OAAO,CAAC,KAAK,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,KAAK,EAAE,CAAC,GAC5C,CAAC,OAAO,CAAC,KAAK,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,KAAK,EAAE,CAAC,GAC5C,CAAC,OAAO,CAAC,KAAK,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,KAAK,EAAE,CAAC,GAC5C,CAAC,OAAO,CAAC,KAAK,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,KAAK,EAAE,CAAC,GAC5C,CAAC,OAAO,CAAC,MAAM,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,IAEpD,CAAC;IAEF,gBAAgB,CAAC,QAAQ,CAAC,CAAC;IAC3B,8BAA8B,CAAC,WAAW,CAAC,CAAC;IAC5C,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,QAAQ,CAC1C,cAAc,EACd,WAAW,EACX,KAAK,EAAE,aAAa,EAAE,EAAE;QACtB,MAAM,MAAM,GAAG,MAAM,OAAO;aACzB,IAAI,CAAC,8BAA8B,EAAE,QAAQ,CAAC;aAC9C,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,MAAM,CAAC,IAAI,CAAC;IACrB,CAAC,EACD,CAAC,IAAI,EAAE,cAAc,EAAE,EAAE,CAAC,sBAAsB,CAAC,IAAI,EAAE,QAAQ,EAAE,cAAc,CAAC,EAChF,oBAAoB,CACrB,CAAC;IAEF,OAAO,eAAe,CAAC,8CAA8C,CAAC,MAAM,CAAC,CAAC;AAChF,CAAC;AAED,0DAA0D;AAC1D,MAAM,CAAC,KAAK,UAAU,aAAa,CACjC,OAAe,EACf,QAAgB,EAChB,SAAiB,EACjB,UAAuC,EAAE;IAEzC,MAAM,oBAAoB,mCACrB,mCAAmC,CAAC,OAAO,CAAC,KAC/C,IAAI,oBACC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,IAE9D,CAAC;IACF,gBAAgB,CAAC,QAAQ,CAAC,CAAC;IAC3B,iBAAiB,CAAC,SAAS,CAAC,CAAC;IAC7B,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,QAAQ,CAC5C,eAAe,EACf,oBAAoB,EACpB,KAAK,EAAE,aAAa,EAAE,EAAE;QACtB,MAAM,MAAM,GAAG,MAAM,OAAO;aACzB,IAAI,CAAC,0CAA0C,EAAE,QAAQ,EAAE,SAAS,CAAC;aACrE,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,SAAS,EAAE,SAAS,EAAE,IACrE,CACL,CAAC;IAEF,OAAO,eAAe,CAAC,0BAA0B,CAAC,QAAQ,CAAC,CAAC;AAC9D,CAAC;AAED,SAAS,gBAAgB,CAAC,QAAgB;IACxC,IAAI,CAAC,QAAQ,EAAE,CAAC;QACd,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;IAC3C,CAAC;AACH,CAAC;AAED,SAAS,iBAAiB,CAAC,SAAiB;IAC1C,IAAI,CAAC,SAAS,EAAE,CAAC;QACf,MAAM,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAC;IAC5C,CAAC;AACH,CAAC;AAED,SAAS,+BAA+B,CAAC,OAAgC;IACvE,IAAI,OAAO,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QAC5E,MAAM,IAAI,KAAK,CAAC,2CAA2C,CAAC,CAAC;IAC/D,CAAC;IACD,IAAI,OAAO,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;QAC1B,gBAAgB,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC1C,CAAC;IACD,IAAI,OAAO,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;QAC7B,IACE,OAAO,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE;YACtC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC,kBAAkB,EAAE,aAAa,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QACvF,CAAC,CAAC,EACF,CAAC;YACD,MAAM,IAAI,KAAK,CAAC,8DAA8D,CAAC,CAAC;QAClF,CAAC;QACD,IAAI,OAAO,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YAC7B,OAAO,CAAC,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE;gBACzC,IAAI,KAAK,CAAC,YAAY,EAAE,CAAC;oBACvB,2BAA2B,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;gBAClD,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC;IACH,CAAC;AACH,CAAC;AAED,SAAS,8BAA8B,CAAC,OAAgC;;IACtE,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,CAAA,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,eAAe,0CAAE,KAAK,KAAI,CAAC,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,eAAe,CAAC,KAAK,CAAC,EAAE,CAAC;QAChG,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;IAC1D,CAAC;AACH,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport type { Client } from \"@azure-rest/core-client\";\nimport { operationOptionsToRequestParameters } from \"@azure-rest/core-client\";\nimport type {\n  OpenAIPageableListOfThreadMessageOutput,\n  ThreadMessageOutput,\n} from \"../customization/outputModels.js\";\nimport type {\n  CreateMessageParameters,\n  ListMessagesParameters,\n} from \"../generated/src/parameters.js\";\nimport { validateMetadata, validateVectorStoreDataType } from \"./inputValidations.js\";\nimport { TracingUtility } from \"../tracing.js\";\nimport {\n  traceEndCreateMessage,\n  traceEndListMessages,\n  traceStartCreateMessage,\n  traceStartListMessages,\n} from \"./messagesTrace.js\";\nimport { traceStartAgentGeneric } from \"./traceUtility.js\";\nimport type { ThreadMessageOptions } from \"../customization/models.js\";\nimport type {\n  CreateMessageOptionalParams,\n  ListMessagesOptionalParams,\n  UpdateMessageOptionalParams,\n} from \"./customModels.js\";\nimport type * as GeneratedParameters from \"../generated/src/parameters.js\";\nimport * as ConvertFromWire from \"../customization/convertOutputModelsFromWire.js\";\nimport { createOpenAIError } from \"./openAIError.js\";\n\nconst expectedStatuses = [\"200\"];\n\n/** Creates a new message on a specified thread. */\nexport async function createMessage(\n  context: Client,\n  threadId: string,\n  messageOptions: ThreadMessageOptions,\n  options: CreateMessageOptionalParams = {},\n): Promise<ThreadMessageOutput> {\n  const createOptions: GeneratedParameters.CreateMessageParameters = {\n    ...operationOptionsToRequestParameters(options),\n    body: {\n      ...messageOptions,\n    },\n  };\n\n  validateThreadId(threadId);\n  validateCreateMessageParameters(createOptions);\n  const response = await TracingUtility.withSpan(\n    \"CreateMessage\",\n    createOptions,\n    async (updateOptions) => {\n      const result = await context\n        .path(\"/threads/{threadId}/messages\", threadId)\n        .post(updateOptions);\n      if (!expectedStatuses.includes(result.status)) {\n        throw createOpenAIError(result);\n      }\n      return result.body;\n    },\n    (span, updatedOptions) => traceStartCreateMessage(span, threadId, updatedOptions),\n    traceEndCreateMessage,\n  );\n  return ConvertFromWire.convertThreadMessageOutput(response);\n}\n\n/** Gets a list of messages that exist on a thread. */\nexport async function listMessages(\n  context: Client,\n  threadId: string,\n  options: ListMessagesOptionalParams = {},\n): Promise<OpenAIPageableListOfThreadMessageOutput> {\n  const listOptions: GeneratedParameters.ListMessagesParameters = {\n    ...operationOptionsToRequestParameters(options),\n    queryParameters: {\n      ...(options.runId && { run_id: options.runId }),\n      ...(options.limit && { run_id: options.limit }),\n      ...(options.order && { run_id: options.order }),\n      ...(options.after && { run_id: options.after }),\n      ...(options.before && { run_id: options.before }),\n    },\n  };\n\n  validateThreadId(threadId);\n  validateListMessagesParameters(listOptions);\n  const output = await TracingUtility.withSpan(\n    \"ListMessages\",\n    listOptions,\n    async (updateOptions) => {\n      const result = await context\n        .path(\"/threads/{threadId}/messages\", threadId)\n        .get(updateOptions);\n      if (!expectedStatuses.includes(result.status)) {\n        throw createOpenAIError(result);\n      }\n      return result.body;\n    },\n    (span, updatedOptions) => traceStartListMessages(span, threadId, updatedOptions),\n    traceEndListMessages,\n  );\n\n  return ConvertFromWire.convertOpenAIPageableListOfThreadMessageOutput(output);\n}\n\n/** Modifies an existing message on an existing thread. */\nexport async function updateMessage(\n  context: Client,\n  threadId: string,\n  messageId: string,\n  options: UpdateMessageOptionalParams = {},\n): Promise<ThreadMessageOutput> {\n  const updateMessageOptions: GeneratedParameters.UpdateMessageParameters = {\n    ...operationOptionsToRequestParameters(options),\n    body: {\n      ...(options.metadata ? { metadata: options.metadata } : {}),\n    },\n  };\n  validateThreadId(threadId);\n  validateMessageId(messageId);\n  const response = await TracingUtility.withSpan(\n    \"UpdateMessage\",\n    updateMessageOptions,\n    async (updateOptions) => {\n      const result = await context\n        .path(\"/threads/{threadId}/messages/{messageId}\", threadId, messageId)\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, messageId: messageId },\n      }),\n  );\n\n  return ConvertFromWire.convertThreadMessageOutput(response);\n}\n\nfunction validateThreadId(threadId: string): void {\n  if (!threadId) {\n    throw new Error(\"Thread ID is required\");\n  }\n}\n\nfunction validateMessageId(messageId: string): void {\n  if (!messageId) {\n    throw new Error(\"Message ID is required\");\n  }\n}\n\nfunction validateCreateMessageParameters(options: CreateMessageParameters): void {\n  if (options.body.role && ![\"user\", \"assistant\"].includes(options.body.role)) {\n    throw new Error(\"Role must be either 'user' or 'assistant'\");\n  }\n  if (options.body.metadata) {\n    validateMetadata(options.body.metadata);\n  }\n  if (options.body.attachments) {\n    if (\n      options.body.attachments.some((value) => {\n        value.tools.some((tool) => ![\"code_interpreter\", \"file_search\"].includes(tool.type));\n      })\n    ) {\n      throw new Error(\"Tool type must be either 'code_interpreter' or 'file_search'\");\n    }\n    if (options.body.attachments) {\n      options.body.attachments.forEach((value) => {\n        if (value.data_sources) {\n          validateVectorStoreDataType(value.data_sources);\n        }\n      });\n    }\n  }\n}\n\nfunction validateListMessagesParameters(options?: ListMessagesParameters): void {\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?.order && ![\"asc\", \"desc\"].includes(options.queryParameters.order)) {\n    throw new Error(\"Order must be either 'asc' or 'desc'\");\n  }\n}\n"]}

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

@@ -0,0 +1,9 @@
+import type { CreateMessageParameters, ListMessagesParameters } from "../generated/src/parameters.js";
+import type { Span } from "../tracing.js";
+import type { OpenAIPageableListOfThreadMessageOutput, ThreadMessageOutput as GeneratedThreadMessageOutput } from "../generated/src/outputModels.js";
+import type { ThreadMessageOutput } from "../customization/outputModels.js";
+export declare function traceStartCreateMessage(span: Span, threadId: string, options: CreateMessageParameters): void;
+export declare function traceEndCreateMessage(span: Span, _options: CreateMessageParameters, result: Promise<ThreadMessageOutput | GeneratedThreadMessageOutput>): Promise<void>;
+export declare function traceStartListMessages(span: Span, threadId: string, _options: ListMessagesParameters): void;
+export declare function traceEndListMessages(span: Span, _options: ListMessagesParameters, result: Promise<OpenAIPageableListOfThreadMessageOutput>): Promise<void>;
+//# sourceMappingURL=messagesTrace.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"messagesTrace.d.ts","sourceRoot":"","sources":["../../../src/agents/messagesTrace.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EACV,uBAAuB,EACvB,sBAAsB,EACvB,MAAM,gCAAgC,CAAC;AACxC,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,eAAe,CAAC;AAE1C,OAAO,KAAK,EACV,uCAAuC,EACvC,mBAAmB,IAAI,4BAA4B,EACpD,MAAM,kCAAkC,CAAC;AAE1C,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,kCAAkC,CAAC;AAE5E,wBAAgB,uBAAuB,CACrC,IAAI,EAAE,IAAI,EACV,QAAQ,EAAE,MAAM,EAChB,OAAO,EAAE,uBAAuB,GAC/B,IAAI,CAMN;AAED,wBAAsB,qBAAqB,CACzC,IAAI,EAAE,IAAI,EACV,QAAQ,EAAE,uBAAuB,EACjC,MAAM,EAAE,OAAO,CAAC,mBAAmB,GAAG,4BAA4B,CAAC,GAClE,OAAO,CAAC,IAAI,CAAC,CAGf;AAED,wBAAgB,sBAAsB,CACpC,IAAI,EAAE,IAAI,EACV,QAAQ,EAAE,MAAM,EAChB,QAAQ,EAAE,sBAAsB,GAC/B,IAAI,CAKN;AAED,wBAAsB,oBAAoB,CACxC,IAAI,EAAE,IAAI,EACV,QAAQ,EAAE,sBAAsB,EAChC,MAAM,EAAE,OAAO,CAAC,uCAAuC,CAAC,GACvD,OAAO,CAAC,IAAI,CAAC,CAKf"}

package/dist/browser/agents/messagesTrace.js

@@ -0,0 +1,29 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+import { TracingAttributes, TracingUtility, TracingOperationName } from "../tracing.js";
+import { addMessageEvent } from "./traceUtility.js";
+export function traceStartCreateMessage(span, threadId, options) {
+    TracingUtility.setSpanAttributes(span, TracingOperationName.CREATE_MESSAGE, {
+        threadId: threadId,
+        genAiSystem: TracingAttributes.AZ_AI_AGENT_SYSTEM,
+    });
+    addMessageEvent(span, Object.assign(Object.assign({}, options.body), { thread_id: threadId }));
+}
+export async function traceEndCreateMessage(span, _options, result) {
+    const resolvedResult = await result;
+    TracingUtility.updateSpanAttributes(span, { messageId: resolvedResult.id });
+}
+export function traceStartListMessages(span, threadId, _options) {
+    TracingUtility.setSpanAttributes(span, TracingOperationName.LIST_MESSAGES, {
+        threadId: threadId,
+        genAiSystem: TracingAttributes.AZ_AI_AGENT_SYSTEM,
+    });
+}
+export async function traceEndListMessages(span, _options, result) {
+    var _a;
+    const resolvedResult = await result;
+    (_a = resolvedResult.data) === null || _a === void 0 ? void 0 : _a.forEach((message) => {
+        addMessageEvent(span, message);
+    });
+}
+//# sourceMappingURL=messagesTrace.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"messagesTrace.js","sourceRoot":"","sources":["../../../src/agents/messagesTrace.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAOlC,OAAO,EAAE,iBAAiB,EAAE,cAAc,EAAE,oBAAoB,EAAE,MAAM,eAAe,CAAC;AAKxF,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAGpD,MAAM,UAAU,uBAAuB,CACrC,IAAU,EACV,QAAgB,EAChB,OAAgC;IAEhC,cAAc,CAAC,iBAAiB,CAAC,IAAI,EAAE,oBAAoB,CAAC,cAAc,EAAE;QAC1E,QAAQ,EAAE,QAAQ;QAClB,WAAW,EAAE,iBAAiB,CAAC,kBAAkB;KAClD,CAAC,CAAC;IACH,eAAe,CAAC,IAAI,kCAAO,OAAO,CAAC,IAAI,KAAE,SAAS,EAAE,QAAQ,IAAG,CAAC;AAClE,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,qBAAqB,CACzC,IAAU,EACV,QAAiC,EACjC,MAAmE;IAEnE,MAAM,cAAc,GAAG,MAAM,MAAM,CAAC;IACpC,cAAc,CAAC,oBAAoB,CAAC,IAAI,EAAE,EAAE,SAAS,EAAE,cAAc,CAAC,EAAE,EAAE,CAAC,CAAC;AAC9E,CAAC;AAED,MAAM,UAAU,sBAAsB,CACpC,IAAU,EACV,QAAgB,EAChB,QAAgC;IAEhC,cAAc,CAAC,iBAAiB,CAAC,IAAI,EAAE,oBAAoB,CAAC,aAAa,EAAE;QACzE,QAAQ,EAAE,QAAQ;QAClB,WAAW,EAAE,iBAAiB,CAAC,kBAAkB;KAClD,CAAC,CAAC;AACL,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,oBAAoB,CACxC,IAAU,EACV,QAAgC,EAChC,MAAwD;;IAExD,MAAM,cAAc,GAAG,MAAM,MAAM,CAAC;IACpC,MAAA,cAAc,CAAC,IAAI,0CAAE,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QACvC,eAAe,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;IACjC,CAAC,CAAC,CAAC;AACL,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport type {\n  CreateMessageParameters,\n  ListMessagesParameters,\n} from \"../generated/src/parameters.js\";\nimport type { Span } from \"../tracing.js\";\nimport { TracingAttributes, TracingUtility, TracingOperationName } from \"../tracing.js\";\nimport type {\n  OpenAIPageableListOfThreadMessageOutput,\n  ThreadMessageOutput as GeneratedThreadMessageOutput,\n} from \"../generated/src/outputModels.js\";\nimport { addMessageEvent } from \"./traceUtility.js\";\nimport type { ThreadMessageOutput } from \"../customization/outputModels.js\";\n\nexport function traceStartCreateMessage(\n  span: Span,\n  threadId: string,\n  options: CreateMessageParameters,\n): void {\n  TracingUtility.setSpanAttributes(span, TracingOperationName.CREATE_MESSAGE, {\n    threadId: threadId,\n    genAiSystem: TracingAttributes.AZ_AI_AGENT_SYSTEM,\n  });\n  addMessageEvent(span, { ...options.body, thread_id: threadId });\n}\n\nexport async function traceEndCreateMessage(\n  span: Span,\n  _options: CreateMessageParameters,\n  result: Promise<ThreadMessageOutput | GeneratedThreadMessageOutput>,\n): Promise<void> {\n  const resolvedResult = await result;\n  TracingUtility.updateSpanAttributes(span, { messageId: resolvedResult.id });\n}\n\nexport function traceStartListMessages(\n  span: Span,\n  threadId: string,\n  _options: ListMessagesParameters,\n): void {\n  TracingUtility.setSpanAttributes(span, TracingOperationName.LIST_MESSAGES, {\n    threadId: threadId,\n    genAiSystem: TracingAttributes.AZ_AI_AGENT_SYSTEM,\n  });\n}\n\nexport async function traceEndListMessages(\n  span: Span,\n  _options: ListMessagesParameters,\n  result: Promise<OpenAIPageableListOfThreadMessageOutput>,\n): Promise<void> {\n  const resolvedResult = await result;\n  resolvedResult.data?.forEach((message) => {\n    addMessageEvent(span, message);\n  });\n}\n"]}

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

@@ -0,0 +1,15 @@
+import type { PathUncheckedResponse } from "@azure-rest/core-client";
+import type { RestErrorOptions } from "@azure/core-rest-pipeline";
+import { RestError } from "@azure/core-rest-pipeline";
+interface OpenAIErrorOptions extends RestErrorOptions {
+    param?: string;
+    type?: string;
+}
+export declare class OpenAIError extends RestError {
+    readonly param?: string;
+    readonly type?: string;
+    constructor(message: string, OpenAIErrorOptions?: OpenAIErrorOptions);
+}
+export declare function createOpenAIError(response: PathUncheckedResponse): OpenAIError;
+export {};
+//# sourceMappingURL=openAIError.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"openAIError.d.ts","sourceRoot":"","sources":["../../../src/agents/openAIError.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,yBAAyB,CAAC;AAErE,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,2BAA2B,CAAC;AAClE,OAAO,EAAE,SAAS,EAAE,MAAM,2BAA2B,CAAC;AAEtD,UAAU,kBAAmB,SAAQ,gBAAgB;IACnD,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,IAAI,CAAC,EAAE,MAAM,CAAC;CACf;AAED,qBAAa,WAAY,SAAQ,SAAS;IACxC,QAAQ,CAAC,KAAK,CAAC,EAAE,MAAM,CAAC;IACxB,QAAQ,CAAC,IAAI,CAAC,EAAE,MAAM,CAAC;gBAEX,OAAO,EAAE,MAAM,EAAE,kBAAkB,GAAE,kBAAuB;CAKzE;AAED,wBAAgB,iBAAiB,CAAC,QAAQ,EAAE,qBAAqB,GAAG,WAAW,CAW9E"}

package/dist/browser/agents/openAIError.js

@@ -0,0 +1,24 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+import { createRestError } from "@azure-rest/core-client";
+import { RestError } from "@azure/core-rest-pipeline";
+export class OpenAIError extends RestError {
+    constructor(message, OpenAIErrorOptions = {}) {
+        super(message, OpenAIErrorOptions);
+        this.param = OpenAIErrorOptions === null || OpenAIErrorOptions === void 0 ? void 0 : OpenAIErrorOptions.param;
+        this.type = OpenAIErrorOptions === null || OpenAIErrorOptions === void 0 ? void 0 : OpenAIErrorOptions.type;
+    }
+}
+export function createOpenAIError(response) {
+    const internalError = response.body.error || response.body;
+    const restError = createRestError(internalError, response);
+    return new OpenAIError(restError.message, {
+        statusCode: restError === null || restError === void 0 ? void 0 : restError.statusCode,
+        code: restError === null || restError === void 0 ? void 0 : restError.code,
+        request: restError === null || restError === void 0 ? void 0 : restError.request,
+        response: restError === null || restError === void 0 ? void 0 : restError.response,
+        param: internalError === null || internalError === void 0 ? void 0 : internalError.param,
+        type: internalError === null || internalError === void 0 ? void 0 : internalError.type,
+    });
+}
+//# sourceMappingURL=openAIError.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"openAIError.js","sourceRoot":"","sources":["../../../src/agents/openAIError.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAGlC,OAAO,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAE1D,OAAO,EAAE,SAAS,EAAE,MAAM,2BAA2B,CAAC;AAOtD,MAAM,OAAO,WAAY,SAAQ,SAAS;IAIxC,YAAY,OAAe,EAAE,qBAAyC,EAAE;QACtE,KAAK,CAAC,OAAO,EAAE,kBAAkB,CAAC,CAAC;QACnC,IAAI,CAAC,KAAK,GAAG,kBAAkB,aAAlB,kBAAkB,uBAAlB,kBAAkB,CAAE,KAAK,CAAC;QACvC,IAAI,CAAC,IAAI,GAAG,kBAAkB,aAAlB,kBAAkB,uBAAlB,kBAAkB,CAAE,IAAI,CAAC;IACvC,CAAC;CACF;AAED,MAAM,UAAU,iBAAiB,CAAC,QAA+B;IAC/D,MAAM,aAAa,GAAG,QAAQ,CAAC,IAAI,CAAC,KAAK,IAAI,QAAQ,CAAC,IAAI,CAAC;IAC3D,MAAM,SAAS,GAAG,eAAe,CAAC,aAAa,EAAE,QAAQ,CAAC,CAAC;IAC3D,OAAO,IAAI,WAAW,CAAC,SAAS,CAAC,OAAO,EAAE;QACxC,UAAU,EAAE,SAAS,aAAT,SAAS,uBAAT,SAAS,CAAE,UAAU;QACjC,IAAI,EAAE,SAAS,aAAT,SAAS,uBAAT,SAAS,CAAE,IAAI;QACrB,OAAO,EAAE,SAAS,aAAT,SAAS,uBAAT,SAAS,CAAE,OAAO;QAC3B,QAAQ,EAAE,SAAS,aAAT,SAAS,uBAAT,SAAS,CAAE,QAAQ;QAC7B,KAAK,EAAE,aAAa,aAAb,aAAa,uBAAb,aAAa,CAAE,KAAK;QAC3B,IAAI,EAAE,aAAa,aAAb,aAAa,uBAAb,aAAa,CAAE,IAAI;KAC1B,CAAC,CAAC;AACL,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport type { PathUncheckedResponse } from \"@azure-rest/core-client\";\nimport { createRestError } from \"@azure-rest/core-client\";\nimport type { RestErrorOptions } from \"@azure/core-rest-pipeline\";\nimport { RestError } from \"@azure/core-rest-pipeline\";\n\ninterface OpenAIErrorOptions extends RestErrorOptions {\n  param?: string;\n  type?: string;\n}\n\nexport class OpenAIError extends RestError {\n  readonly param?: string;\n  readonly type?: string;\n\n  constructor(message: string, OpenAIErrorOptions: OpenAIErrorOptions = {}) {\n    super(message, OpenAIErrorOptions);\n    this.param = OpenAIErrorOptions?.param;\n    this.type = OpenAIErrorOptions?.type;\n  }\n}\n\nexport function createOpenAIError(response: PathUncheckedResponse): OpenAIError {\n  const internalError = response.body.error || response.body;\n  const restError = createRestError(internalError, response);\n  return new OpenAIError(restError.message, {\n    statusCode: restError?.statusCode,\n    code: restError?.code,\n    request: restError?.request,\n    response: restError?.response,\n    param: internalError?.param,\n    type: internalError?.type,\n  });\n}\n"]}

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

@@ -0,0 +1,34 @@
+import type { PollOperationState, PollOperation } from "@azure/core-lro";
+import { Poller } from "@azure/core-lro";
+import type { AbortSignalLike } from "@azure/abort-controller";
+import type { PollingOptions } from "./customModels.js";
+interface PollResult {
+    status?: string;
+}
+interface AgentsPollOperationState<T extends PollResult> extends PollOperationState<T> {
+    updateInternal: (state?: T) => Promise<{
+        result: T;
+        completed: boolean;
+    }>;
+    cancelInternal?: (state: T) => Promise<boolean>;
+    abortSignal?: AbortSignalLike;
+    cancelled?: boolean;
+}
+interface AgentsPollOperation<T extends PollResult> extends PollOperation<AgentsPollOperationState<T>, T> {
+}
+export declare class AgentsPoller<T extends PollResult> extends Poller<AgentsPollOperationState<T>, T> {
+    sleepIntervalInMs: number;
+    constructor({ update, pollingOptions, cancel, baseOperation, onProgress, }: {
+        update: (state?: T) => Promise<{
+            result: T;
+            completed: boolean;
+        }>;
+        pollingOptions?: PollingOptions;
+        cancel?: (state: T) => Promise<boolean>;
+        baseOperation?: AgentsPollOperation<T>;
+        onProgress?: (state: AgentsPollOperationState<T>) => void;
+    });
+    delay(): Promise<void>;
+}
+export {};
+//# sourceMappingURL=poller.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"poller.d.ts","sourceRoot":"","sources":["../../../src/agents/poller.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,kBAAkB,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AACzE,OAAO,EAAE,MAAM,EAAE,MAAM,iBAAiB,CAAC;AACzC,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAC/D,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AAExD,UAAU,UAAU;IAClB,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAED,UAAU,wBAAwB,CAAC,CAAC,SAAS,UAAU,CAAE,SAAQ,kBAAkB,CAAC,CAAC,CAAC;IACpF,cAAc,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,OAAO,CAAC;QAAE,MAAM,EAAE,CAAC,CAAC;QAAC,SAAS,EAAE,OAAO,CAAA;KAAE,CAAC,CAAC;IAC1E,cAAc,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC;IAChD,WAAW,CAAC,EAAE,eAAe,CAAC;IAC9B,SAAS,CAAC,EAAE,OAAO,CAAC;CACrB;AAED,UAAU,mBAAmB,CAAC,CAAC,SAAS,UAAU,CAChD,SAAQ,aAAa,CAAC,wBAAwB,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;CAAG;AAE1D,qBAAa,YAAY,CAAC,CAAC,SAAS,UAAU,CAAE,SAAQ,MAAM,CAAC,wBAAwB,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IAC5F,iBAAiB,EAAE,MAAM,CAAQ;gBAErB,EACV,MAAM,EACN,cAAc,EACd,MAAM,EACN,aAAa,EACb,UAAU,GACX,EAAE;QACD,MAAM,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,OAAO,CAAC;YAAE,MAAM,EAAE,CAAC,CAAC;YAAC,SAAS,EAAE,OAAO,CAAA;SAAE,CAAC,CAAC;QAClE,cAAc,CAAC,EAAE,cAAc,CAAC;QAChC,MAAM,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC;QACxC,aAAa,CAAC,EAAE,mBAAmB,CAAC,CAAC,CAAC,CAAC;QACvC,UAAU,CAAC,EAAE,CAAC,KAAK,EAAE,wBAAwB,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC;KAC3D;IAqBK,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;CAG7B"}

package/dist/browser/agents/poller.js

@@ -0,0 +1,60 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+import { delay } from "@azure/core-util";
+import { Poller } from "@azure/core-lro";
+export class AgentsPoller extends Poller {
+    constructor({ update, pollingOptions, cancel, baseOperation, onProgress, }) {
+        var _a;
+        let state = {
+            updateInternal: update,
+            cancelInternal: cancel,
+            abortSignal: pollingOptions === null || pollingOptions === void 0 ? void 0 : pollingOptions.abortSignal,
+        };
+        if (baseOperation) {
+            state = baseOperation.state;
+        }
+        const operation = makeOperation(state);
+        super(operation);
+        this.sleepIntervalInMs = 1000;
+        this.sleepIntervalInMs = (_a = pollingOptions === null || pollingOptions === void 0 ? void 0 : pollingOptions.sleepIntervalInMs) !== null && _a !== void 0 ? _a : this.sleepIntervalInMs;
+        if (onProgress) {
+            this.onProgress(onProgress);
+        }
+    }
+    async delay() {
+        await delay(this.sleepIntervalInMs);
+    }
+}
+/**
+ * Utility function to create a new instance of operation state.
+ * Recommended to avoid no references are left or manipulated by mistake
+ */
+function makeOperation(state) {
+    return {
+        state: Object.assign({}, state),
+        update: updateWrapper,
+        cancel: cancelWrapper,
+        toString: toString,
+    };
+}
+async function updateWrapper() {
+    var _a;
+    if ((_a = this.state.abortSignal) === null || _a === void 0 ? void 0 : _a.aborted) {
+        return makeOperation(this.state);
+    }
+    const { result, completed } = await this.state.updateInternal(this.state.result);
+    this.state.result = result;
+    this.state.isCompleted = completed;
+    return makeOperation(this.state);
+}
+async function cancelWrapper() {
+    if (!this.state.result || !this.state.cancelInternal) {
+        return makeOperation(Object.assign(Object.assign({}, this.state), { cancelled: false }));
+    }
+    const cancelled = await this.state.cancelInternal(this.state.result);
+    return makeOperation(Object.assign(Object.assign({}, this.state), { cancelled: cancelled }));
+}
+function toString() {
+    return JSON.stringify(this.state);
+}
+//# sourceMappingURL=poller.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"poller.js","sourceRoot":"","sources":["../../../src/agents/poller.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAElC,OAAO,EAAE,KAAK,EAAE,MAAM,kBAAkB,CAAC;AAEzC,OAAO,EAAE,MAAM,EAAE,MAAM,iBAAiB,CAAC;AAkBzC,MAAM,OAAO,YAAmC,SAAQ,MAAsC;IAG5F,YAAY,EACV,MAAM,EACN,cAAc,EACd,MAAM,EACN,aAAa,EACb,UAAU,GAOX;;QACC,IAAI,KAAK,GAAgC;YACvC,cAAc,EAAE,MAAM;YACtB,cAAc,EAAE,MAAM;YACtB,WAAW,EAAE,cAAc,aAAd,cAAc,uBAAd,cAAc,CAAE,WAAW;SACzC,CAAC;QAEF,IAAI,aAAa,EAAE,CAAC;YAClB,KAAK,GAAG,aAAa,CAAC,KAAK,CAAC;QAC9B,CAAC;QAED,MAAM,SAAS,GAAG,aAAa,CAAC,KAAK,CAAC,CAAC;QACvC,KAAK,CAAC,SAAS,CAAC,CAAC;QA1BnB,sBAAiB,GAAW,IAAI,CAAC;QA4B/B,IAAI,CAAC,iBAAiB,GAAG,MAAA,cAAc,aAAd,cAAc,uBAAd,cAAc,CAAE,iBAAiB,mCAAI,IAAI,CAAC,iBAAiB,CAAC;QAErF,IAAI,UAAU,EAAE,CAAC;YACf,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;QAC9B,CAAC;IACH,CAAC;IAED,KAAK,CAAC,KAAK;QACT,MAAM,KAAK,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;IACtC,CAAC;CACF;AAED;;;GAGG;AACH,SAAS,aAAa,CACpB,KAAkC;IAElC,OAAO;QACL,KAAK,oBACA,KAAK,CACT;QACD,MAAM,EAAE,aAAa;QACrB,MAAM,EAAE,aAAa;QACrB,QAAQ,EAAE,QAAQ;KACnB,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,aAAa;;IAG1B,IAAI,MAAA,IAAI,CAAC,KAAK,CAAC,WAAW,0CAAE,OAAO,EAAE,CAAC;QACpC,OAAO,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACnC,CAAC;IACD,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;IACjF,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,MAAM,CAAC;IAC3B,IAAI,CAAC,KAAK,CAAC,WAAW,GAAG,SAAS,CAAC;IACnC,OAAO,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;AACnC,CAAC;AAED,KAAK,UAAU,aAAa;IAG1B,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,cAAc,EAAE,CAAC;QACrD,OAAO,aAAa,iCACf,IAAI,CAAC,KAAK,KACb,SAAS,EAAE,KAAK,IAChB,CAAC;IACL,CAAC;IACD,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;IACrE,OAAO,aAAa,iCACf,IAAI,CAAC,KAAK,KACb,SAAS,EAAE,SAAS,IACpB,CAAC;AACL,CAAC;AAED,SAAS,QAAQ;IACf,OAAO,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;AACpC,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport { delay } from \"@azure/core-util\";\nimport type { PollOperationState, PollOperation } from \"@azure/core-lro\";\nimport { Poller } from \"@azure/core-lro\";\nimport type { AbortSignalLike } from \"@azure/abort-controller\";\nimport type { PollingOptions } from \"./customModels.js\";\n\ninterface PollResult {\n  status?: string;\n}\n\ninterface AgentsPollOperationState<T extends PollResult> extends PollOperationState<T> {\n  updateInternal: (state?: T) => Promise<{ result: T; completed: boolean }>;\n  cancelInternal?: (state: T) => Promise<boolean>;\n  abortSignal?: AbortSignalLike;\n  cancelled?: boolean;\n}\n\ninterface AgentsPollOperation<T extends PollResult>\n  extends PollOperation<AgentsPollOperationState<T>, T> {}\n\nexport class AgentsPoller<T extends PollResult> extends Poller<AgentsPollOperationState<T>, T> {\n  sleepIntervalInMs: number = 1000;\n\n  constructor({\n    update,\n    pollingOptions,\n    cancel,\n    baseOperation,\n    onProgress,\n  }: {\n    update: (state?: T) => Promise<{ result: T; completed: boolean }>;\n    pollingOptions?: PollingOptions;\n    cancel?: (state: T) => Promise<boolean>;\n    baseOperation?: AgentsPollOperation<T>;\n    onProgress?: (state: AgentsPollOperationState<T>) => void;\n  }) {\n    let state: AgentsPollOperationState<T> = {\n      updateInternal: update,\n      cancelInternal: cancel,\n      abortSignal: pollingOptions?.abortSignal,\n    };\n\n    if (baseOperation) {\n      state = baseOperation.state;\n    }\n\n    const operation = makeOperation(state);\n    super(operation);\n\n    this.sleepIntervalInMs = pollingOptions?.sleepIntervalInMs ?? this.sleepIntervalInMs;\n\n    if (onProgress) {\n      this.onProgress(onProgress);\n    }\n  }\n\n  async delay(): Promise<void> {\n    await delay(this.sleepIntervalInMs);\n  }\n}\n\n/**\n * Utility function to create a new instance of operation state.\n * Recommended to avoid no references are left or manipulated by mistake\n */\nfunction makeOperation<T extends PollResult>(\n  state: AgentsPollOperationState<T>,\n): PollOperation<AgentsPollOperationState<T>, T> {\n  return {\n    state: {\n      ...state,\n    },\n    update: updateWrapper,\n    cancel: cancelWrapper,\n    toString: toString,\n  };\n}\n\nasync function updateWrapper<T extends PollResult>(\n  this: AgentsPollOperation<T>,\n): Promise<PollOperation<AgentsPollOperationState<T>, T>> {\n  if (this.state.abortSignal?.aborted) {\n    return makeOperation(this.state);\n  }\n  const { result, completed } = await this.state.updateInternal(this.state.result);\n  this.state.result = result;\n  this.state.isCompleted = completed;\n  return makeOperation(this.state);\n}\n\nasync function cancelWrapper<T extends PollResult>(\n  this: AgentsPollOperation<T>,\n): Promise<PollOperation<AgentsPollOperationState<T>, T>> {\n  if (!this.state.result || !this.state.cancelInternal) {\n    return makeOperation({\n      ...this.state,\n      cancelled: false,\n    });\n  }\n  const cancelled = await this.state.cancelInternal(this.state.result);\n  return makeOperation({\n    ...this.state,\n    cancelled: cancelled,\n  });\n}\n\nfunction toString<T extends PollResult>(this: AgentsPollOperation<T>): string {\n  return JSON.stringify(this.state);\n}\n"]}

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

@@ -0,0 +1,8 @@
+import type { Client } from "@azure-rest/core-client";
+import type { OpenAIPageableListOfRunStepOutput, RunStepOutput } from "../customization/outputModels.js";
+import type { GetRunStepOptionalParams, ListRunStepsOptionalParams } from "./customModels.js";
+/** Gets a single run step from a thread run. */
+export declare function getRunStep(context: Client, threadId: string, runId: string, stepId: string, options?: GetRunStepOptionalParams): Promise<RunStepOutput>;
+/** Gets a list of run steps from a thread run. */
+export declare function listRunSteps(context: Client, threadId: string, runId: string, options?: ListRunStepsOptionalParams): Promise<OpenAIPageableListOfRunStepOutput>;
+//# sourceMappingURL=runSteps.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"runSteps.d.ts","sourceRoot":"","sources":["../../../src/agents/runSteps.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,yBAAyB,CAAC;AAEtD,OAAO,KAAK,EACV,iCAAiC,EACjC,aAAa,EACd,MAAM,kCAAkC,CAAC;AAS1C,OAAO,KAAK,EAAE,wBAAwB,EAAE,0BAA0B,EAAE,MAAM,mBAAmB,CAAC;AAM9F,gDAAgD;AAChD,wBAAsB,UAAU,CAC9B,OAAO,EAAE,MAAM,EACf,QAAQ,EAAE,MAAM,EAChB,KAAK,EAAE,MAAM,EACb,MAAM,EAAE,MAAM,EACd,OAAO,GAAE,wBAA6B,GACrC,OAAO,CAAC,aAAa,CAAC,CAgBxB;AAED,kDAAkD;AAClD,wBAAsB,YAAY,CAChC,OAAO,EAAE,MAAM,EACf,QAAQ,EAAE,MAAM,EAChB,KAAK,EAAE,MAAM,EACb,OAAO,GAAE,0BAA+B,GACvC,OAAO,CAAC,iCAAiC,CAAC,CAc5C"}

package/dist/browser/agents/runSteps.js

@@ -0,0 +1,55 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+import { operationOptionsToRequestParameters } from "@azure-rest/core-client";
+import * as ConverterFromWire from "../customization/convertOutputModelsFromWire.js";
+import { validateLimit, validateOrder, validateRunId, validateThreadId, } from "./inputValidations.js";
+import { convertToListQueryParameters } from "../customization/convertParametersToWire.js";
+import { createOpenAIError } from "./openAIError.js";
+const expectedStatuses = ["200"];
+/** Gets a single run step from a thread run. */
+export async function getRunStep(context, threadId, runId, stepId, options = {}) {
+    validateThreadId(threadId);
+    validateRunId(runId);
+    validateStepId(stepId);
+    const getOptions = Object.assign({}, operationOptionsToRequestParameters(options));
+    const result = await context
+        .path("/threads/{threadId}/runs/{runId}/steps/{stepId}", threadId, runId, stepId)
+        .get(getOptions);
+    if (!expectedStatuses.includes(result.status)) {
+        throw createOpenAIError(result);
+    }
+    return ConverterFromWire.convertRunStepOutput(result.body);
+}
+/** Gets a list of run steps from a thread run. */
+export async function listRunSteps(context, threadId, runId, options = {}) {
+    const listOptions = Object.assign(Object.assign({}, operationOptionsToRequestParameters(options)), { queryParameters: convertToListQueryParameters(options) });
+    validateListRunsParameters(threadId, runId, listOptions);
+    const result = await context
+        .path("/threads/{threadId}/runs/{runId}/steps", threadId, runId)
+        .get(listOptions);
+    if (!expectedStatuses.includes(result.status)) {
+        throw createOpenAIError(result);
+    }
+    return ConverterFromWire.convertOpenAIPageableListOfRunStepOutput(result.body);
+}
+function validateStepId(stepId) {
+    if (!stepId) {
+        throw new Error("Step ID is required");
+    }
+}
+function validateListRunsParameters(thread_id, runId, options) {
+    var _a, _b, _c;
+    validateThreadId(thread_id);
+    validateRunId(runId);
+    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);
+    }
+}
+//# sourceMappingURL=runSteps.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"runSteps.js","sourceRoot":"","sources":["../../../src/agents/runSteps.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAGlC,OAAO,EAAE,mCAAmC,EAAE,MAAM,yBAAyB,CAAC;AAM9E,OAAO,KAAK,iBAAiB,MAAM,iDAAiD,CAAC;AACrF,OAAO,EACL,aAAa,EACb,aAAa,EACb,aAAa,EACb,gBAAgB,GACjB,MAAM,uBAAuB,CAAC;AAE/B,OAAO,EAAE,4BAA4B,EAAE,MAAM,6CAA6C,CAAC;AAC3F,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AAErD,MAAM,gBAAgB,GAAG,CAAC,KAAK,CAAC,CAAC;AAEjC,gDAAgD;AAChD,MAAM,CAAC,KAAK,UAAU,UAAU,CAC9B,OAAe,EACf,QAAgB,EAChB,KAAa,EACb,MAAc,EACd,UAAoC,EAAE;IAEtC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;IAC3B,aAAa,CAAC,KAAK,CAAC,CAAC;IACrB,cAAc,CAAC,MAAM,CAAC,CAAC;IAEvB,MAAM,UAAU,qBACX,mCAAmC,CAAC,OAAO,CAAC,CAChD,CAAC;IAEF,MAAM,MAAM,GAAG,MAAM,OAAO;SACzB,IAAI,CAAC,iDAAiD,EAAE,QAAQ,EAAE,KAAK,EAAE,MAAM,CAAC;SAChF,GAAG,CAAC,UAAU,CAAC,CAAC;IACnB,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;QAC9C,MAAM,iBAAiB,CAAC,MAAM,CAAC,CAAC;IAClC,CAAC;IACD,OAAO,iBAAiB,CAAC,oBAAoB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AAC7D,CAAC;AAED,kDAAkD;AAClD,MAAM,CAAC,KAAK,UAAU,YAAY,CAChC,OAAe,EACf,QAAgB,EAChB,KAAa,EACb,UAAsC,EAAE;IAExC,MAAM,WAAW,mCACZ,mCAAmC,CAAC,OAAO,CAAC,KAC/C,eAAe,EAAE,4BAA4B,CAAC,OAAO,CAAC,GACvD,CAAC;IAEF,0BAA0B,CAAC,QAAQ,EAAE,KAAK,EAAE,WAAW,CAAC,CAAC;IACzD,MAAM,MAAM,GAAG,MAAM,OAAO;SACzB,IAAI,CAAC,wCAAwC,EAAE,QAAQ,EAAE,KAAK,CAAC;SAC/D,GAAG,CAAC,WAAW,CAAC,CAAC;IACpB,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;QAC9C,MAAM,iBAAiB,CAAC,MAAM,CAAC,CAAC;IAClC,CAAC;IACD,OAAO,iBAAiB,CAAC,wCAAwC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AACjF,CAAC;AAED,SAAS,cAAc,CAAC,MAAc;IACpC,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;IACzC,CAAC;AACH,CAAC;AAED,SAAS,0BAA0B,CACjC,SAAiB,EACjB,KAAa,EACb,OAAoD;;IAEpD,gBAAgB,CAAC,SAAS,CAAC,CAAC;IAC5B,aAAa,CAAC,KAAK,CAAC,CAAC;IACrB,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","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport type { Client } from \"@azure-rest/core-client\";\nimport { operationOptionsToRequestParameters } from \"@azure-rest/core-client\";\nimport type {\n  OpenAIPageableListOfRunStepOutput,\n  RunStepOutput,\n} from \"../customization/outputModels.js\";\nimport type * as GeneratedParameters from \"../generated/src/parameters.js\";\nimport * as ConverterFromWire from \"../customization/convertOutputModelsFromWire.js\";\nimport {\n  validateLimit,\n  validateOrder,\n  validateRunId,\n  validateThreadId,\n} from \"./inputValidations.js\";\nimport type { GetRunStepOptionalParams, ListRunStepsOptionalParams } from \"./customModels.js\";\nimport { convertToListQueryParameters } from \"../customization/convertParametersToWire.js\";\nimport { createOpenAIError } from \"./openAIError.js\";\n\nconst expectedStatuses = [\"200\"];\n\n/** Gets a single run step from a thread run. */\nexport async function getRunStep(\n  context: Client,\n  threadId: string,\n  runId: string,\n  stepId: string,\n  options: GetRunStepOptionalParams = {},\n): Promise<RunStepOutput> {\n  validateThreadId(threadId);\n  validateRunId(runId);\n  validateStepId(stepId);\n\n  const getOptions: GeneratedParameters.GetRunParameters = {\n    ...operationOptionsToRequestParameters(options),\n  };\n\n  const result = await context\n    .path(\"/threads/{threadId}/runs/{runId}/steps/{stepId}\", threadId, runId, stepId)\n    .get(getOptions);\n  if (!expectedStatuses.includes(result.status)) {\n    throw createOpenAIError(result);\n  }\n  return ConverterFromWire.convertRunStepOutput(result.body);\n}\n\n/** Gets a list of run steps from a thread run. */\nexport async function listRunSteps(\n  context: Client,\n  threadId: string,\n  runId: string,\n  options: ListRunStepsOptionalParams = {},\n): Promise<OpenAIPageableListOfRunStepOutput> {\n  const listOptions: GeneratedParameters.ListRunStepsParameters = {\n    ...operationOptionsToRequestParameters(options),\n    queryParameters: convertToListQueryParameters(options),\n  };\n\n  validateListRunsParameters(threadId, runId, listOptions);\n  const result = await context\n    .path(\"/threads/{threadId}/runs/{runId}/steps\", threadId, runId)\n    .get(listOptions);\n  if (!expectedStatuses.includes(result.status)) {\n    throw createOpenAIError(result);\n  }\n  return ConverterFromWire.convertOpenAIPageableListOfRunStepOutput(result.body);\n}\n\nfunction validateStepId(stepId: string): void {\n  if (!stepId) {\n    throw new Error(\"Step ID is required\");\n  }\n}\n\nfunction validateListRunsParameters(\n  thread_id: string,\n  runId: string,\n  options?: GeneratedParameters.ListRunStepsParameters,\n): void {\n  validateThreadId(thread_id);\n  validateRunId(runId);\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"]}

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

@@ -0,0 +1,9 @@
+import type { CreateRunParameters, CreateThreadAndRunParameters, SubmitToolOutputsToRunParameters, UpdateRunParameters } from "../generated/src/parameters.js";
+import type { ThreadRunOutput } from "../generated/src/outputModels.js";
+import type { Span } from "../tracing.js";
+export declare function traceStartCreateRun(span: Span, options: CreateRunParameters | CreateThreadAndRunParameters, threadId?: string, operationName?: string): void;
+export declare function traceStartCreateThreadAndRun(span: Span, options: CreateThreadAndRunParameters): void;
+export declare function traceEndCreateOrUpdateRun(span: Span, _options: CreateRunParameters | UpdateRunParameters, result: Promise<ThreadRunOutput>): Promise<void>;
+export declare function traceStartSubmitToolOutputsToRun(span: Span, options: SubmitToolOutputsToRunParameters, threadId: string, runId: string): void;
+export declare function traceEndSubmitToolOutputsToRun(span: Span, _options: SubmitToolOutputsToRunParameters, result: Promise<ThreadRunOutput>): Promise<void>;
+//# sourceMappingURL=runTrace.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"runTrace.d.ts","sourceRoot":"","sources":["../../../src/agents/runTrace.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EACV,mBAAmB,EACnB,4BAA4B,EAC5B,gCAAgC,EAChC,mBAAmB,EACpB,MAAM,gCAAgC,CAAC;AACxC,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,kCAAkC,CAAC;AACxE,OAAO,KAAK,EAA2B,IAAI,EAAE,MAAM,eAAe,CAAC;AAUnE,wBAAgB,mBAAmB,CACjC,IAAI,EAAE,IAAI,EACV,OAAO,EAAE,mBAAmB,GAAG,4BAA4B,EAC3D,QAAQ,CAAC,EAAE,MAAM,EACjB,aAAa,GAAE,MAAwC,GACtD,IAAI,CAkBN;AAED,wBAAgB,4BAA4B,CAC1C,IAAI,EAAE,IAAI,EACV,OAAO,EAAE,4BAA4B,GACpC,IAAI,CAEN;AAED,wBAAsB,yBAAyB,CAC7C,IAAI,EAAE,IAAI,EACV,QAAQ,EAAE,mBAAmB,GAAG,mBAAmB,EACnD,MAAM,EAAE,OAAO,CAAC,eAAe,CAAC,GAC/B,OAAO,CAAC,IAAI,CAAC,CAGf;AAED,wBAAgB,gCAAgC,CAC9C,IAAI,EAAE,IAAI,EACV,OAAO,EAAE,gCAAgC,EACzC,QAAQ,EAAE,MAAM,EAChB,KAAK,EAAE,MAAM,GACZ,IAAI,CAQN;AAED,wBAAsB,8BAA8B,CAClD,IAAI,EAAE,IAAI,EACV,QAAQ,EAAE,gCAAgC,EAC1C,MAAM,EAAE,OAAO,CAAC,eAAe,CAAC,GAC/B,OAAO,CAAC,IAAI,CAAC,CAGf"}

package/dist/browser/agents/runTrace.js

@@ -0,0 +1,62 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+import { TracingUtility, TracingOperationName } from "../tracing.js";
+import { addInstructionsEvent, addMessageEvent, addToolMessagesEvent, formatAgentApiResponse, UpdateWithAgentAttributes, } from "./traceUtility.js";
+export function traceStartCreateRun(span, options, threadId, operationName = TracingOperationName.CREATE_RUN) {
+    var _a, _b, _c, _d, _e, _f, _g;
+    const attributes = {
+        threadId: threadId,
+        agentId: options.body.assistant_id,
+        model: (_a = options.body.model) !== null && _a !== void 0 ? _a : undefined,
+        instructions: (_b = options.body.instructions) !== null && _b !== void 0 ? _b : undefined,
+        temperature: (_c = options.body.temperature) !== null && _c !== void 0 ? _c : undefined,
+        topP: (_d = options.body.top_p) !== null && _d !== void 0 ? _d : undefined,
+        maxCompletionTokens: (_e = options.body.max_completion_tokens) !== null && _e !== void 0 ? _e : undefined,
+        maxPromptTokens: (_f = options.body.max_prompt_tokens) !== null && _f !== void 0 ? _f : undefined,
+        responseFormat: formatAgentApiResponse(options.body.response_format),
+    };
+    if (options.body.additional_instructions) {
+        attributes.additional_instructions =
+            (_g = options.body.additional_instructions) !== null && _g !== void 0 ? _g : undefined;
+    }
+    TracingUtility.setSpanAttributes(span, operationName, UpdateWithAgentAttributes(attributes));
+    setSpanEvents(span, options);
+}
+export function traceStartCreateThreadAndRun(span, options) {
+    traceStartCreateRun(span, options, undefined, TracingOperationName.CREATE_THREAD_RUN);
+}
+export async function traceEndCreateOrUpdateRun(span, _options, result) {
+    const resolvedResult = await result;
+    updateSpanAttributesForRun(span, resolvedResult);
+}
+export function traceStartSubmitToolOutputsToRun(span, options, threadId, runId) {
+    const attributes = { threadId: threadId, runId: runId };
+    TracingUtility.setSpanAttributes(span, TracingOperationName.SUBMIT_TOOL_OUTPUTS, UpdateWithAgentAttributes(attributes));
+    addToolMessagesEvent(span, options.body.tool_outputs);
+}
+export async function traceEndSubmitToolOutputsToRun(span, _options, result) {
+    const resolvedResult = await result;
+    updateSpanAttributesForRun(span, resolvedResult);
+}
+function updateSpanAttributesForRun(span, output) {
+    TracingUtility.updateSpanAttributes(span, {
+        runId: output.id,
+        runStatus: output.status,
+        responseModel: output.model,
+    });
+    const usage = output.usage;
+    if (usage && "completion_tokens" in usage && usage.completion_tokens) {
+        TracingUtility.updateSpanAttributes(span, {
+            usageCompletionTokens: usage.completion_tokens,
+            usagePromptTokens: usage.prompt_tokens,
+        });
+    }
+}
+function setSpanEvents(span, options) {
+    var _a;
+    addInstructionsEvent(span, Object.assign(Object.assign({}, options.body), { agentId: options.body.assistant_id }));
+    (_a = options.body.additional_messages) === null || _a === void 0 ? void 0 : _a.forEach((message) => {
+        addMessageEvent(span, message);
+    });
+}
+//# sourceMappingURL=runTrace.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"runTrace.js","sourceRoot":"","sources":["../../../src/agents/runTrace.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAUlC,OAAO,EAAE,cAAc,EAAE,oBAAoB,EAAE,MAAM,eAAe,CAAC;AACrE,OAAO,EACL,oBAAoB,EACpB,eAAe,EACf,oBAAoB,EACpB,sBAAsB,EACtB,yBAAyB,GAC1B,MAAM,mBAAmB,CAAC;AAE3B,MAAM,UAAU,mBAAmB,CACjC,IAAU,EACV,OAA2D,EAC3D,QAAiB,EACjB,gBAAwB,oBAAoB,CAAC,UAAU;;IAEvD,MAAM,UAAU,GAA4B;QAC1C,QAAQ,EAAE,QAAQ;QAClB,OAAO,EAAE,OAAO,CAAC,IAAI,CAAC,YAAY;QAClC,KAAK,EAAE,MAAA,OAAO,CAAC,IAAI,CAAC,KAAK,mCAAI,SAAS;QACtC,YAAY,EAAE,MAAA,OAAO,CAAC,IAAI,CAAC,YAAY,mCAAI,SAAS;QACpD,WAAW,EAAE,MAAA,OAAO,CAAC,IAAI,CAAC,WAAW,mCAAI,SAAS;QAClD,IAAI,EAAE,MAAA,OAAO,CAAC,IAAI,CAAC,KAAK,mCAAI,SAAS;QACrC,mBAAmB,EAAE,MAAA,OAAO,CAAC,IAAI,CAAC,qBAAqB,mCAAI,SAAS;QACpE,eAAe,EAAE,MAAA,OAAO,CAAC,IAAI,CAAC,iBAAiB,mCAAI,SAAS;QAC5D,cAAc,EAAE,sBAAsB,CAAC,OAAO,CAAC,IAAI,CAAC,eAAe,CAAC;KACrE,CAAC;IACF,IAAK,OAA+B,CAAC,IAAI,CAAC,uBAAuB,EAAE,CAAC;QAClE,UAAU,CAAC,uBAAuB;YAChC,MAAC,OAA+B,CAAC,IAAI,CAAC,uBAAuB,mCAAI,SAAS,CAAC;IAC/E,CAAC;IACD,cAAc,CAAC,iBAAiB,CAAC,IAAI,EAAE,aAAa,EAAE,yBAAyB,CAAC,UAAU,CAAC,CAAC,CAAC;IAC7F,aAAa,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;AAC/B,CAAC;AAED,MAAM,UAAU,4BAA4B,CAC1C,IAAU,EACV,OAAqC;IAErC,mBAAmB,CAAC,IAAI,EAAE,OAAO,EAAE,SAAS,EAAE,oBAAoB,CAAC,iBAAiB,CAAC,CAAC;AACxF,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,yBAAyB,CAC7C,IAAU,EACV,QAAmD,EACnD,MAAgC;IAEhC,MAAM,cAAc,GAAG,MAAM,MAAM,CAAC;IACpC,0BAA0B,CAAC,IAAI,EAAE,cAAc,CAAC,CAAC;AACnD,CAAC;AAED,MAAM,UAAU,gCAAgC,CAC9C,IAAU,EACV,OAAyC,EACzC,QAAgB,EAChB,KAAa;IAEb,MAAM,UAAU,GAA4B,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;IACjF,cAAc,CAAC,iBAAiB,CAC9B,IAAI,EACJ,oBAAoB,CAAC,mBAAmB,EACxC,yBAAyB,CAAC,UAAU,CAAC,CACtC,CAAC;IACF,oBAAoB,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;AACxD,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,8BAA8B,CAClD,IAAU,EACV,QAA0C,EAC1C,MAAgC;IAEhC,MAAM,cAAc,GAAG,MAAM,MAAM,CAAC;IACpC,0BAA0B,CAAC,IAAI,EAAE,cAAc,CAAC,CAAC;AACnD,CAAC;AAED,SAAS,0BAA0B,CAAC,IAAU,EAAE,MAAuB;IACrE,cAAc,CAAC,oBAAoB,CAAC,IAAI,EAAE;QACxC,KAAK,EAAE,MAAM,CAAC,EAAE;QAChB,SAAS,EAAE,MAAM,CAAC,MAAM;QACxB,aAAa,EAAE,MAAM,CAAC,KAAK;KAC5B,CAAC,CAAC;IACH,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;IAC3B,IAAI,KAAK,IAAI,mBAAmB,IAAI,KAAK,IAAI,KAAK,CAAC,iBAAiB,EAAE,CAAC;QACrE,cAAc,CAAC,oBAAoB,CAAC,IAAI,EAAE;YACxC,qBAAqB,EAAE,KAAK,CAAC,iBAAiB;YAC9C,iBAAiB,EAAE,KAAK,CAAC,aAAa;SACvC,CAAC,CAAC;IACL,CAAC;AACH,CAAC;AAED,SAAS,aAAa,CAAC,IAAU,EAAE,OAA4B;;IAC7D,oBAAoB,CAAC,IAAI,kCAAO,OAAO,CAAC,IAAI,KAAE,OAAO,EAAE,OAAO,CAAC,IAAI,CAAC,YAAY,IAAG,CAAC;IACpF,MAAA,OAAO,CAAC,IAAI,CAAC,mBAAmB,0CAAE,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QACpD,eAAe,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;IACjC,CAAC,CAAC,CAAC;AACL,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport type {\n  CreateRunParameters,\n  CreateThreadAndRunParameters,\n  SubmitToolOutputsToRunParameters,\n  UpdateRunParameters,\n} from \"../generated/src/parameters.js\";\nimport type { ThreadRunOutput } from \"../generated/src/outputModels.js\";\nimport type { TracingAttributeOptions, Span } from \"../tracing.js\";\nimport { TracingUtility, TracingOperationName } from \"../tracing.js\";\nimport {\n  addInstructionsEvent,\n  addMessageEvent,\n  addToolMessagesEvent,\n  formatAgentApiResponse,\n  UpdateWithAgentAttributes,\n} from \"./traceUtility.js\";\n\nexport function traceStartCreateRun(\n  span: Span,\n  options: CreateRunParameters | CreateThreadAndRunParameters,\n  threadId?: string,\n  operationName: string = TracingOperationName.CREATE_RUN,\n): void {\n  const attributes: TracingAttributeOptions = {\n    threadId: threadId,\n    agentId: options.body.assistant_id,\n    model: options.body.model ?? undefined,\n    instructions: options.body.instructions ?? undefined,\n    temperature: options.body.temperature ?? undefined,\n    topP: options.body.top_p ?? undefined,\n    maxCompletionTokens: options.body.max_completion_tokens ?? undefined,\n    maxPromptTokens: options.body.max_prompt_tokens ?? undefined,\n    responseFormat: formatAgentApiResponse(options.body.response_format),\n  };\n  if ((options as CreateRunParameters).body.additional_instructions) {\n    attributes.additional_instructions =\n      (options as CreateRunParameters).body.additional_instructions ?? undefined;\n  }\n  TracingUtility.setSpanAttributes(span, operationName, UpdateWithAgentAttributes(attributes));\n  setSpanEvents(span, options);\n}\n\nexport function traceStartCreateThreadAndRun(\n  span: Span,\n  options: CreateThreadAndRunParameters,\n): void {\n  traceStartCreateRun(span, options, undefined, TracingOperationName.CREATE_THREAD_RUN);\n}\n\nexport async function traceEndCreateOrUpdateRun(\n  span: Span,\n  _options: CreateRunParameters | UpdateRunParameters,\n  result: Promise<ThreadRunOutput>,\n): Promise<void> {\n  const resolvedResult = await result;\n  updateSpanAttributesForRun(span, resolvedResult);\n}\n\nexport function traceStartSubmitToolOutputsToRun(\n  span: Span,\n  options: SubmitToolOutputsToRunParameters,\n  threadId: string,\n  runId: string,\n): void {\n  const attributes: TracingAttributeOptions = { threadId: threadId, runId: runId };\n  TracingUtility.setSpanAttributes(\n    span,\n    TracingOperationName.SUBMIT_TOOL_OUTPUTS,\n    UpdateWithAgentAttributes(attributes),\n  );\n  addToolMessagesEvent(span, options.body.tool_outputs);\n}\n\nexport async function traceEndSubmitToolOutputsToRun(\n  span: Span,\n  _options: SubmitToolOutputsToRunParameters,\n  result: Promise<ThreadRunOutput>,\n): Promise<void> {\n  const resolvedResult = await result;\n  updateSpanAttributesForRun(span, resolvedResult);\n}\n\nfunction updateSpanAttributesForRun(span: Span, output: ThreadRunOutput): void {\n  TracingUtility.updateSpanAttributes(span, {\n    runId: output.id,\n    runStatus: output.status,\n    responseModel: output.model,\n  });\n  const usage = output.usage;\n  if (usage && \"completion_tokens\" in usage && usage.completion_tokens) {\n    TracingUtility.updateSpanAttributes(span, {\n      usageCompletionTokens: usage.completion_tokens,\n      usagePromptTokens: usage.prompt_tokens,\n    });\n  }\n}\n\nfunction setSpanEvents(span: Span, options: CreateRunParameters): void {\n  addInstructionsEvent(span, { ...options.body, agentId: options.body.assistant_id });\n  options.body.additional_messages?.forEach((message) => {\n    addMessageEvent(span, message);\n  });\n}\n"]}

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

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

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