@azure/ai-projects 1.0.0-beta.1

package/dist/commonjs/tracing.js

@@ -0,0 +1,197 @@
+"use strict";
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TracingUtility = exports.TracingOperationName = exports.TracingAttributes = void 0;
+const core_tracing_1 = require("@azure/core-tracing");
+const constants_js_1 = require("./constants.js");
+const core_util_1 = require("@azure/core-util");
+const logger_js_1 = require("./logger.js");
+var TracingAttributes;
+(function (TracingAttributes) {
+    TracingAttributes["GEN_AI_MESSAGE_ID"] = "gen_ai.message.id";
+    TracingAttributes["GEN_AI_MESSAGE_STATUS"] = "gen_ai.message.status";
+    TracingAttributes["GEN_AI_THREAD_ID"] = "gen_ai.thread.id";
+    TracingAttributes["GEN_AI_THREAD_RUN_ID"] = "gen_ai.thread.run.id";
+    TracingAttributes["GEN_AI_AGENT_ID"] = "gen_ai.agent.id";
+    TracingAttributes["GEN_AI_AGENT_NAME"] = "gen_ai.agent.name";
+    TracingAttributes["GEN_AI_AGENT_DESCRIPTION"] = "gen_ai.agent.description";
+    TracingAttributes["GEN_AI_OPERATION_NAME"] = "gen_ai.operation.name";
+    TracingAttributes["GEN_AI_THREAD_RUN_STATUS"] = "gen_ai.thread.run.status";
+    TracingAttributes["GEN_AI_REQUEST_MODEL"] = "gen_ai.request.model";
+    TracingAttributes["GEN_AI_REQUEST_TEMPERATURE"] = "gen_ai.request.temperature";
+    TracingAttributes["GEN_AI_REQUEST_TOP_P"] = "gen_ai.request.top_p";
+    TracingAttributes["GEN_AI_REQUEST_MAX_INPUT_TOKENS"] = "gen_ai.request.max_input_tokens";
+    TracingAttributes["GEN_AI_REQUEST_MAX_OUTPUT_TOKENS"] = "gen_ai.request.max_output_tokens";
+    TracingAttributes["GEN_AI_RESPONSE_MODEL"] = "gen_ai.response.model";
+    TracingAttributes["GEN_AI_SYSTEM"] = "gen_ai.system";
+    TracingAttributes["SERVER_ADDRESS"] = "server.address";
+    TracingAttributes["AZ_AI_AGENT_SYSTEM"] = "az.ai.agents";
+    TracingAttributes["GEN_AI_TOOL_NAME"] = "gen_ai.tool.name";
+    TracingAttributes["GEN_AI_TOOL_CALL_ID"] = "gen_ai.tool.call.id";
+    TracingAttributes["GEN_AI_REQUEST_RESPONSE_FORMAT"] = "gen_ai.request.response_format";
+    TracingAttributes["GEN_AI_USAGE_INPUT_TOKENS"] = "gen_ai.usage.input_tokens";
+    TracingAttributes["GEN_AI_USAGE_OUTPUT_TOKENS"] = "gen_ai.usage.output_tokens";
+    TracingAttributes["GEN_AI_SYSTEM_MESSAGE"] = "gen_ai.system.message";
+    TracingAttributes["GEN_AI_EVENT_CONTENT"] = "gen_ai.event.content";
+    TracingAttributes["ERROR_TYPE"] = "error.type";
+})(TracingAttributes || (exports.TracingAttributes = TracingAttributes = {}));
+var TracingOperationName;
+(function (TracingOperationName) {
+    TracingOperationName["CREATE_AGENT"] = "create_agent";
+    TracingOperationName["CREATE_UPDATE_AGENT"] = "update_agent";
+    TracingOperationName["CREATE_THREAD"] = "create_thread";
+    TracingOperationName["CREATE_MESSAGE"] = "create_message";
+    TracingOperationName["CREATE_RUN"] = "create_run";
+    TracingOperationName["START_THREAD_RUN"] = "start_thread_run";
+    TracingOperationName["EXECUTE_TOOL"] = "execute_tool";
+    TracingOperationName["LIST_MESSAGES"] = "list_messages";
+    TracingOperationName["SUBMIT_TOOL_OUTPUTS"] = "submit_tool_outputs";
+    TracingOperationName["CREATE_THREAD_RUN"] = "create_thread_run";
+})(TracingOperationName || (exports.TracingOperationName = TracingOperationName = {}));
+class TracingUtility {
+    static async withSpan(name, options, request, startTrace, endTrace) {
+        return TracingUtility.tracingClient.withSpan(name, options, async (updatedOptions, span) => {
+            if (startTrace) {
+                try {
+                    updatedOptions.tracingAttributeOptions = Object.assign(Object.assign({}, updatedOptions.tracingAttributeOptions), { operationName: name });
+                    startTrace(span, updatedOptions);
+                }
+                catch (e) {
+                    logger_js_1.logger.warning(`Skipping updating span before request execution due to an error: ${(0, core_util_1.getErrorMessage)(e)}`);
+                }
+            }
+            let result;
+            try {
+                result = await request(updatedOptions);
+            }
+            catch (error) {
+                const errorStatus = { status: "error" };
+                if (error instanceof Error) {
+                    errorStatus.error = error;
+                }
+                throw error;
+            }
+            if (endTrace && result !== undefined) {
+                try {
+                    endTrace(span, updatedOptions, result);
+                }
+                catch (e) {
+                    logger_js_1.logger.warning(`Skipping updating span after request execution due to an error: ${(0, core_util_1.getErrorMessage)(e)}`);
+                }
+            }
+            return result;
+        }, { spanKind: "client" });
+    }
+    static updateSpanAttributes(span, attributeOptions) {
+        TracingUtility.setAttributes(span, attributeOptions);
+    }
+    static setSpanAttributes(span, operationName, attributeOptions) {
+        attributeOptions.operationName = operationName;
+        TracingUtility.setAttributes(span, attributeOptions);
+    }
+    static setAttributes(span, attributeOptions) {
+        if (span.isRecording()) {
+            const { name, operationName, description, serverAddress, threadId, agentId, messageId, runId, model, temperature, topP, maxPromptTokens, maxCompletionTokens, responseFormat, runStatus, responseModel, usageCompletionTokens, usagePromptTokens, genAiSystem = TracingAttributes.AZ_AI_AGENT_SYSTEM, } = attributeOptions;
+            if (genAiSystem) {
+                span.setAttribute(TracingAttributes.GEN_AI_SYSTEM, genAiSystem);
+            }
+            if (name) {
+                span.setAttribute(TracingAttributes.GEN_AI_AGENT_NAME, name);
+            }
+            if (description) {
+                span.setAttribute(TracingAttributes.GEN_AI_AGENT_DESCRIPTION, description);
+            }
+            if (serverAddress) {
+                span.setAttribute(TracingAttributes.SERVER_ADDRESS, serverAddress);
+            }
+            if (threadId) {
+                span.setAttribute(TracingAttributes.GEN_AI_THREAD_ID, threadId);
+            }
+            if (agentId) {
+                span.setAttribute(TracingAttributes.GEN_AI_AGENT_ID, agentId);
+            }
+            if (runId) {
+                span.setAttribute(TracingAttributes.GEN_AI_THREAD_RUN_ID, runId);
+            }
+            if (messageId) {
+                span.setAttribute(TracingAttributes.GEN_AI_MESSAGE_ID, messageId);
+            }
+            if (model) {
+                span.setAttribute(TracingAttributes.GEN_AI_REQUEST_MODEL, model);
+            }
+            if (temperature !== null) {
+                span.setAttribute(TracingAttributes.GEN_AI_REQUEST_TEMPERATURE, temperature);
+            }
+            if (topP !== null) {
+                span.setAttribute(TracingAttributes.GEN_AI_REQUEST_TOP_P, topP);
+            }
+            if (maxPromptTokens !== null) {
+                span.setAttribute(TracingAttributes.GEN_AI_REQUEST_MAX_INPUT_TOKENS, maxPromptTokens);
+            }
+            if (maxCompletionTokens !== null) {
+                span.setAttribute(TracingAttributes.GEN_AI_REQUEST_MAX_OUTPUT_TOKENS, maxCompletionTokens);
+            }
+            if (responseFormat) {
+                span.setAttribute(TracingAttributes.GEN_AI_REQUEST_RESPONSE_FORMAT, responseFormat);
+            }
+            if (runStatus) {
+                span.setAttribute(TracingAttributes.GEN_AI_THREAD_RUN_STATUS, runStatus);
+            }
+            if (responseModel) {
+                span.setAttribute(TracingAttributes.GEN_AI_RESPONSE_MODEL, responseModel);
+            }
+            if (usagePromptTokens) {
+                span.setAttribute(TracingAttributes.GEN_AI_USAGE_INPUT_TOKENS, usagePromptTokens);
+            }
+            if (usageCompletionTokens) {
+                span.setAttribute(TracingAttributes.GEN_AI_USAGE_OUTPUT_TOKENS, usageCompletionTokens);
+            }
+            if (operationName) {
+                span.setAttribute(TracingAttributes.GEN_AI_OPERATION_NAME, operationName);
+            }
+        }
+        return;
+    }
+    static addSpanEvent(span, eventName, attributeOptions) {
+        if (span.isRecording()) {
+            const { threadId, agentId, runId, messageId, eventContent, usageCompletionTokens, usagePromptTokens, messageStatus, } = attributeOptions;
+            const attributes = {};
+            if (eventContent) {
+                attributes[TracingAttributes.GEN_AI_EVENT_CONTENT] = eventContent;
+            }
+            if (threadId) {
+                attributes[TracingAttributes.GEN_AI_THREAD_ID] = threadId;
+            }
+            if (agentId) {
+                attributes[TracingAttributes.GEN_AI_AGENT_ID] = agentId;
+            }
+            if (runId) {
+                attributes[TracingAttributes.GEN_AI_THREAD_RUN_ID] = runId;
+            }
+            if (messageId) {
+                attributes[TracingAttributes.GEN_AI_MESSAGE_ID] = messageId;
+            }
+            if (messageStatus) {
+                attributes[TracingAttributes.GEN_AI_MESSAGE_STATUS] = messageStatus;
+            }
+            if (usagePromptTokens) {
+                attributes[TracingAttributes.GEN_AI_USAGE_INPUT_TOKENS] = usagePromptTokens;
+            }
+            if (usageCompletionTokens) {
+                attributes[TracingAttributes.GEN_AI_USAGE_OUTPUT_TOKENS] = usageCompletionTokens;
+            }
+            if (span.addEvent) {
+                span.addEvent(eventName, { attributes });
+            }
+        }
+        return;
+    }
+}
+exports.TracingUtility = TracingUtility;
+TracingUtility.tracingClient = (0, core_tracing_1.createTracingClient)({
+    namespace: "Microsoft.CognitiveServices",
+    packageName: constants_js_1.PACKAGE_NAME,
+    packageVersion: constants_js_1.SDK_VERSION,
+});
+//# sourceMappingURL=tracing.js.map

package/dist/commonjs/tracing.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tracing.js","sourceRoot":"","sources":["../../src/tracing.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;;AAQlC,sDAA0D;AAC1D,iDAA2D;AAC3D,gDAAmD;AACnD,2CAAqC;AAErC,IAAY,iBA2BX;AA3BD,WAAY,iBAAiB;IAC3B,4DAAuC,CAAA;IACvC,oEAA+C,CAAA;IAC/C,0DAAqC,CAAA;IACrC,kEAA6C,CAAA;IAC7C,wDAAmC,CAAA;IACnC,4DAAuC,CAAA;IACvC,0EAAqD,CAAA;IACrD,oEAA+C,CAAA;IAC/C,0EAAqD,CAAA;IACrD,kEAA6C,CAAA;IAC7C,8EAAyD,CAAA;IACzD,kEAA6C,CAAA;IAC7C,wFAAmE,CAAA;IACnE,0FAAqE,CAAA;IACrE,oEAA+C,CAAA;IAC/C,oDAA+B,CAAA;IAC/B,sDAAiC,CAAA;IACjC,wDAAmC,CAAA;IACnC,0DAAqC,CAAA;IACrC,gEAA2C,CAAA;IAC3C,sFAAiE,CAAA;IACjE,4EAAuD,CAAA;IACvD,8EAAyD,CAAA;IACzD,oEAA+C,CAAA;IAC/C,kEAA6C,CAAA;IAC7C,8CAAyB,CAAA;AAC3B,CAAC,EA3BW,iBAAiB,iCAAjB,iBAAiB,QA2B5B;AACD,IAAY,oBAWX;AAXD,WAAY,oBAAoB;IAC9B,qDAA6B,CAAA;IAC7B,4DAAoC,CAAA;IACpC,uDAA+B,CAAA;IAC/B,yDAAiC,CAAA;IACjC,iDAAyB,CAAA;IACzB,6DAAqC,CAAA;IACrC,qDAA6B,CAAA;IAC7B,uDAA+B,CAAA;IAC/B,mEAA2C,CAAA;IAC3C,+DAAuC,CAAA;AACzC,CAAC,EAXW,oBAAoB,oCAApB,oBAAoB,QAW/B;AAkCD,MAAa,cAAc;IAOzB,MAAM,CAAC,KAAK,CAAC,QAAQ,CAInB,IAAY,EACZ,OAAgB,EAChB,OAAgB,EAChB,UAA0D,EAC1D,QAAqF;QAErF,OAAO,cAAc,CAAC,aAAa,CAAC,QAAQ,CAC1C,IAAI,EACJ,OAAO,EACP,KAAK,EAAE,cAAuB,EAAE,IAAU,EAAE,EAAE;YAC5C,IAAI,UAAU,EAAE,CAAC;gBACf,IAAI,CAAC;oBACH,cAAc,CAAC,uBAAuB,mCACjC,cAAc,CAAC,uBAAuB,KACzC,aAAa,EAAE,IAAI,GACpB,CAAC;oBACF,UAAU,CAAC,IAAI,EAAE,cAAc,CAAC,CAAC;gBACnC,CAAC;gBAAC,OAAO,CAAC,EAAE,CAAC;oBACX,kBAAM,CAAC,OAAO,CACZ,oEAAoE,IAAA,2BAAe,EAAC,CAAC,CAAC,EAAE,CACzF,CAAC;gBACJ,CAAC;YACH,CAAC;YACD,IAAI,MAAuC,CAAC;YAC5C,IAAI,CAAC;gBACH,MAAM,GAAG,MAAM,OAAO,CAAC,cAAc,CAAC,CAAC;YACzC,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,WAAW,GAAoB,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC;gBACzD,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;oBAC3B,WAAW,CAAC,KAAK,GAAG,KAAK,CAAC;gBAC5B,CAAC;gBACD,MAAM,KAAK,CAAC;YACd,CAAC;YAED,IAAI,QAAQ,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;gBACrC,IAAI,CAAC;oBACH,QAAQ,CAAC,IAAI,EAAE,cAAc,EAAE,MAAM,CAAC,CAAC;gBACzC,CAAC;gBAAC,OAAO,CAAC,EAAE,CAAC;oBACX,kBAAM,CAAC,OAAO,CACZ,mEAAmE,IAAA,2BAAe,EAAC,CAAC,CAAC,EAAE,CACxF,CAAC;gBACJ,CAAC;YACH,CAAC;YACD,OAAO,MAAM,CAAC;QAChB,CAAC,EACD,EAAE,QAAQ,EAAE,QAAQ,EAAE,CACvB,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,oBAAoB,CACzB,IAAU,EACV,gBAAgE;QAEhE,cAAc,CAAC,aAAa,CAAC,IAAI,EAAE,gBAAgB,CAAC,CAAC;IACvD,CAAC;IAED,MAAM,CAAC,iBAAiB,CACtB,IAAU,EACV,aAAqB,EACrB,gBAAyC;QAEzC,gBAAgB,CAAC,aAAa,GAAG,aAAa,CAAC;QAC/C,cAAc,CAAC,aAAa,CAAC,IAAI,EAAE,gBAAgB,CAAC,CAAC;IACvD,CAAC;IAED,MAAM,CAAC,aAAa,CAAC,IAAU,EAAE,gBAAyC;QACxE,IAAI,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC;YACvB,MAAM,EACJ,IAAI,EACJ,aAAa,EACb,WAAW,EACX,aAAa,EACb,QAAQ,EACR,OAAO,EACP,SAAS,EACT,KAAK,EACL,KAAK,EACL,WAAW,EACX,IAAI,EACJ,eAAe,EACf,mBAAmB,EACnB,cAAc,EACd,SAAS,EACT,aAAa,EACb,qBAAqB,EACrB,iBAAiB,EACjB,WAAW,GAAG,iBAAiB,CAAC,kBAAkB,GACnD,GAAG,gBAAgB,CAAC;YAErB,IAAI,WAAW,EAAE,CAAC;gBAChB,IAAI,CAAC,YAAY,CAAC,iBAAiB,CAAC,aAAa,EAAE,WAAW,CAAC,CAAC;YAClE,CAAC;YACD,IAAI,IAAI,EAAE,CAAC;gBACT,IAAI,CAAC,YAAY,CAAC,iBAAiB,CAAC,iBAAiB,EAAE,IAAI,CAAC,CAAC;YAC/D,CAAC;YACD,IAAI,WAAW,EAAE,CAAC;gBAChB,IAAI,CAAC,YAAY,CAAC,iBAAiB,CAAC,wBAAwB,EAAE,WAAW,CAAC,CAAC;YAC7E,CAAC;YAED,IAAI,aAAa,EAAE,CAAC;gBAClB,IAAI,CAAC,YAAY,CAAC,iBAAiB,CAAC,cAAc,EAAE,aAAa,CAAC,CAAC;YACrE,CAAC;YAED,IAAI,QAAQ,EAAE,CAAC;gBACb,IAAI,CAAC,YAAY,CAAC,iBAAiB,CAAC,gBAAgB,EAAE,QAAQ,CAAC,CAAC;YAClE,CAAC;YAED,IAAI,OAAO,EAAE,CAAC;gBACZ,IAAI,CAAC,YAAY,CAAC,iBAAiB,CAAC,eAAe,EAAE,OAAO,CAAC,CAAC;YAChE,CAAC;YAED,IAAI,KAAK,EAAE,CAAC;gBACV,IAAI,CAAC,YAAY,CAAC,iBAAiB,CAAC,oBAAoB,EAAE,KAAK,CAAC,CAAC;YACnE,CAAC;YAED,IAAI,SAAS,EAAE,CAAC;gBACd,IAAI,CAAC,YAAY,CAAC,iBAAiB,CAAC,iBAAiB,EAAE,SAAS,CAAC,CAAC;YACpE,CAAC;YACD,IAAI,KAAK,EAAE,CAAC;gBACV,IAAI,CAAC,YAAY,CAAC,iBAAiB,CAAC,oBAAoB,EAAE,KAAK,CAAC,CAAC;YACnE,CAAC;YAED,IAAI,WAAW,KAAK,IAAI,EAAE,CAAC;gBACzB,IAAI,CAAC,YAAY,CAAC,iBAAiB,CAAC,0BAA0B,EAAE,WAAW,CAAC,CAAC;YAC/E,CAAC;YAED,IAAI,IAAI,KAAK,IAAI,EAAE,CAAC;gBAClB,IAAI,CAAC,YAAY,CAAC,iBAAiB,CAAC,oBAAoB,EAAE,IAAI,CAAC,CAAC;YAClE,CAAC;YAED,IAAI,eAAe,KAAK,IAAI,EAAE,CAAC;gBAC7B,IAAI,CAAC,YAAY,CAAC,iBAAiB,CAAC,+BAA+B,EAAE,eAAe,CAAC,CAAC;YACxF,CAAC;YAED,IAAI,mBAAmB,KAAK,IAAI,EAAE,CAAC;gBACjC,IAAI,CAAC,YAAY,CAAC,iBAAiB,CAAC,gCAAgC,EAAE,mBAAmB,CAAC,CAAC;YAC7F,CAAC;YAED,IAAI,cAAc,EAAE,CAAC;gBACnB,IAAI,CAAC,YAAY,CAAC,iBAAiB,CAAC,8BAA8B,EAAE,cAAc,CAAC,CAAC;YACtF,CAAC;YAED,IAAI,SAAS,EAAE,CAAC;gBACd,IAAI,CAAC,YAAY,CAAC,iBAAiB,CAAC,wBAAwB,EAAE,SAAS,CAAC,CAAC;YAC3E,CAAC;YAED,IAAI,aAAa,EAAE,CAAC;gBAClB,IAAI,CAAC,YAAY,CAAC,iBAAiB,CAAC,qBAAqB,EAAE,aAAa,CAAC,CAAC;YAC5E,CAAC;YAED,IAAI,iBAAiB,EAAE,CAAC;gBACtB,IAAI,CAAC,YAAY,CAAC,iBAAiB,CAAC,yBAAyB,EAAE,iBAAiB,CAAC,CAAC;YACpF,CAAC;YAED,IAAI,qBAAqB,EAAE,CAAC;gBAC1B,IAAI,CAAC,YAAY,CAAC,iBAAiB,CAAC,0BAA0B,EAAE,qBAAqB,CAAC,CAAC;YACzF,CAAC;YACD,IAAI,aAAa,EAAE,CAAC;gBAClB,IAAI,CAAC,YAAY,CAAC,iBAAiB,CAAC,qBAAqB,EAAE,aAAa,CAAC,CAAC;YAC5E,CAAC;QACH,CAAC;QACD,OAAO;IACT,CAAC;IAED,MAAM,CAAC,YAAY,CACjB,IAAU,EACV,SAAiB,EACjB,gBAAgE;QAEhE,IAAI,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC;YACvB,MAAM,EACJ,QAAQ,EACR,OAAO,EACP,KAAK,EACL,SAAS,EACT,YAAY,EACZ,qBAAqB,EACrB,iBAAiB,EACjB,aAAa,GACd,GAAG,gBAAgB,CAAC;YACrB,MAAM,UAAU,GAA4B,EAAE,CAAC;YAE/C,IAAI,YAAY,EAAE,CAAC;gBACjB,UAAU,CAAC,iBAAiB,CAAC,oBAAoB,CAAC,GAAG,YAAY,CAAC;YACpE,CAAC;YAED,IAAI,QAAQ,EAAE,CAAC;gBACb,UAAU,CAAC,iBAAiB,CAAC,gBAAgB,CAAC,GAAG,QAAQ,CAAC;YAC5D,CAAC;YAED,IAAI,OAAO,EAAE,CAAC;gBACZ,UAAU,CAAC,iBAAiB,CAAC,eAAe,CAAC,GAAG,OAAO,CAAC;YAC1D,CAAC;YAED,IAAI,KAAK,EAAE,CAAC;gBACV,UAAU,CAAC,iBAAiB,CAAC,oBAAoB,CAAC,GAAG,KAAK,CAAC;YAC7D,CAAC;YAED,IAAI,SAAS,EAAE,CAAC;gBACd,UAAU,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,GAAG,SAAS,CAAC;YAC9D,CAAC;YACD,IAAI,aAAa,EAAE,CAAC;gBAClB,UAAU,CAAC,iBAAiB,CAAC,qBAAqB,CAAC,GAAG,aAAa,CAAC;YACtE,CAAC;YAED,IAAI,iBAAiB,EAAE,CAAC;gBACtB,UAAU,CAAC,iBAAiB,CAAC,yBAAyB,CAAC,GAAG,iBAAiB,CAAC;YAC9E,CAAC;YAED,IAAI,qBAAqB,EAAE,CAAC;gBAC1B,UAAU,CAAC,iBAAiB,CAAC,0BAA0B,CAAC,GAAG,qBAAqB,CAAC;YACnF,CAAC;YAED,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;gBAClB,IAAI,CAAC,QAAQ,CAAC,SAAS,EAAE,EAAE,UAAU,EAAE,CAAC,CAAC;YAC3C,CAAC;QACH,CAAC;QACD,OAAO;IACT,CAAC;;AArOH,wCAsOC;AArOgB,4BAAa,GAAG,IAAA,kCAAmB,EAAC;IACjD,SAAS,EAAE,6BAA6B;IACxC,WAAW,EAAE,2BAAY;IACzB,cAAc,EAAE,0BAAW;CAC5B,CAAC,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport type {\n  OperationTracingOptions,\n  Resolved,\n  SpanStatusError,\n  TracingSpan,\n} from \"@azure/core-tracing\";\nimport { createTracingClient } from \"@azure/core-tracing\";\nimport { PACKAGE_NAME, SDK_VERSION } from \"./constants.js\";\nimport { getErrorMessage } from \"@azure/core-util\";\nimport { logger } from \"./logger.js\";\n\nexport enum TracingAttributes {\n  GEN_AI_MESSAGE_ID = \"gen_ai.message.id\",\n  GEN_AI_MESSAGE_STATUS = \"gen_ai.message.status\",\n  GEN_AI_THREAD_ID = \"gen_ai.thread.id\",\n  GEN_AI_THREAD_RUN_ID = \"gen_ai.thread.run.id\",\n  GEN_AI_AGENT_ID = \"gen_ai.agent.id\",\n  GEN_AI_AGENT_NAME = \"gen_ai.agent.name\",\n  GEN_AI_AGENT_DESCRIPTION = \"gen_ai.agent.description\",\n  GEN_AI_OPERATION_NAME = \"gen_ai.operation.name\",\n  GEN_AI_THREAD_RUN_STATUS = \"gen_ai.thread.run.status\",\n  GEN_AI_REQUEST_MODEL = \"gen_ai.request.model\",\n  GEN_AI_REQUEST_TEMPERATURE = \"gen_ai.request.temperature\",\n  GEN_AI_REQUEST_TOP_P = \"gen_ai.request.top_p\",\n  GEN_AI_REQUEST_MAX_INPUT_TOKENS = \"gen_ai.request.max_input_tokens\",\n  GEN_AI_REQUEST_MAX_OUTPUT_TOKENS = \"gen_ai.request.max_output_tokens\",\n  GEN_AI_RESPONSE_MODEL = \"gen_ai.response.model\",\n  GEN_AI_SYSTEM = \"gen_ai.system\",\n  SERVER_ADDRESS = \"server.address\",\n  AZ_AI_AGENT_SYSTEM = \"az.ai.agents\",\n  GEN_AI_TOOL_NAME = \"gen_ai.tool.name\",\n  GEN_AI_TOOL_CALL_ID = \"gen_ai.tool.call.id\",\n  GEN_AI_REQUEST_RESPONSE_FORMAT = \"gen_ai.request.response_format\",\n  GEN_AI_USAGE_INPUT_TOKENS = \"gen_ai.usage.input_tokens\",\n  GEN_AI_USAGE_OUTPUT_TOKENS = \"gen_ai.usage.output_tokens\",\n  GEN_AI_SYSTEM_MESSAGE = \"gen_ai.system.message\",\n  GEN_AI_EVENT_CONTENT = \"gen_ai.event.content\",\n  ERROR_TYPE = \"error.type\",\n}\nexport enum TracingOperationName {\n  CREATE_AGENT = \"create_agent\",\n  CREATE_UPDATE_AGENT = \"update_agent\",\n  CREATE_THREAD = \"create_thread\",\n  CREATE_MESSAGE = \"create_message\",\n  CREATE_RUN = \"create_run\",\n  START_THREAD_RUN = \"start_thread_run\",\n  EXECUTE_TOOL = \"execute_tool\",\n  LIST_MESSAGES = \"list_messages\",\n  SUBMIT_TOOL_OUTPUTS = \"submit_tool_outputs\",\n  CREATE_THREAD_RUN = \"create_thread_run\",\n}\n\nexport interface TracingAttributeOptions {\n  operationName?: string;\n  name?: string;\n  description?: string;\n  serverAddress?: string;\n  threadId?: string;\n  agentId?: string;\n  instructions?: string;\n  additional_instructions?: string;\n  runId?: string;\n  runStatus?: string;\n  responseModel?: string;\n  model?: string;\n  temperature?: number;\n  topP?: number;\n  maxPromptTokens?: number;\n  maxCompletionTokens?: number;\n  responseFormat?: string;\n  genAiSystem?: string;\n  messageId?: string;\n  messageStatus?: string;\n  eventContent?: string;\n  usagePromptTokens?: number;\n  usageCompletionTokens?: number;\n}\n\nexport type Span = Omit<TracingSpan, \"end\">;\nexport type OptionsWithTracing = {\n  tracingOptions?: OperationTracingOptions;\n  tracingAttributeOptions?: TracingAttributeOptions;\n};\n\nexport class TracingUtility {\n  private static tracingClient = createTracingClient({\n    namespace: \"Microsoft.CognitiveServices\",\n    packageName: PACKAGE_NAME,\n    packageVersion: SDK_VERSION,\n  });\n\n  static async withSpan<\n    Options extends OptionsWithTracing,\n    Request extends (updatedOptions: Options) => ReturnType<Request>,\n  >(\n    name: string,\n    options: Options,\n    request: Request,\n    startTrace?: (span: Span, updatedOptions: Options) => void,\n    endTrace?: (span: Span, updatedOptions: Options, result: ReturnType<Request>) => void,\n  ): Promise<Resolved<ReturnType<Request>>> {\n    return TracingUtility.tracingClient.withSpan(\n      name,\n      options,\n      async (updatedOptions: Options, span: Span) => {\n        if (startTrace) {\n          try {\n            updatedOptions.tracingAttributeOptions = {\n              ...updatedOptions.tracingAttributeOptions,\n              operationName: name,\n            };\n            startTrace(span, updatedOptions);\n          } catch (e) {\n            logger.warning(\n              `Skipping updating span before request execution due to an error: ${getErrorMessage(e)}`,\n            );\n          }\n        }\n        let result: ReturnType<Request> | undefined;\n        try {\n          result = await request(updatedOptions);\n        } catch (error) {\n          const errorStatus: SpanStatusError = { status: \"error\" };\n          if (error instanceof Error) {\n            errorStatus.error = error;\n          }\n          throw error;\n        }\n\n        if (endTrace && result !== undefined) {\n          try {\n            endTrace(span, updatedOptions, result);\n          } catch (e) {\n            logger.warning(\n              `Skipping updating span after request execution due to an error: ${getErrorMessage(e)}`,\n            );\n          }\n        }\n        return result;\n      },\n      { spanKind: \"client\" },\n    );\n  }\n\n  static updateSpanAttributes(\n    span: Span,\n    attributeOptions: Omit<TracingAttributeOptions, \"operationName\">,\n  ): void {\n    TracingUtility.setAttributes(span, attributeOptions);\n  }\n\n  static setSpanAttributes(\n    span: Span,\n    operationName: string,\n    attributeOptions: TracingAttributeOptions,\n  ): void {\n    attributeOptions.operationName = operationName;\n    TracingUtility.setAttributes(span, attributeOptions);\n  }\n\n  static setAttributes(span: Span, attributeOptions: TracingAttributeOptions): void {\n    if (span.isRecording()) {\n      const {\n        name,\n        operationName,\n        description,\n        serverAddress,\n        threadId,\n        agentId,\n        messageId,\n        runId,\n        model,\n        temperature,\n        topP,\n        maxPromptTokens,\n        maxCompletionTokens,\n        responseFormat,\n        runStatus,\n        responseModel,\n        usageCompletionTokens,\n        usagePromptTokens,\n        genAiSystem = TracingAttributes.AZ_AI_AGENT_SYSTEM,\n      } = attributeOptions;\n\n      if (genAiSystem) {\n        span.setAttribute(TracingAttributes.GEN_AI_SYSTEM, genAiSystem);\n      }\n      if (name) {\n        span.setAttribute(TracingAttributes.GEN_AI_AGENT_NAME, name);\n      }\n      if (description) {\n        span.setAttribute(TracingAttributes.GEN_AI_AGENT_DESCRIPTION, description);\n      }\n\n      if (serverAddress) {\n        span.setAttribute(TracingAttributes.SERVER_ADDRESS, serverAddress);\n      }\n\n      if (threadId) {\n        span.setAttribute(TracingAttributes.GEN_AI_THREAD_ID, threadId);\n      }\n\n      if (agentId) {\n        span.setAttribute(TracingAttributes.GEN_AI_AGENT_ID, agentId);\n      }\n\n      if (runId) {\n        span.setAttribute(TracingAttributes.GEN_AI_THREAD_RUN_ID, runId);\n      }\n\n      if (messageId) {\n        span.setAttribute(TracingAttributes.GEN_AI_MESSAGE_ID, messageId);\n      }\n      if (model) {\n        span.setAttribute(TracingAttributes.GEN_AI_REQUEST_MODEL, model);\n      }\n\n      if (temperature !== null) {\n        span.setAttribute(TracingAttributes.GEN_AI_REQUEST_TEMPERATURE, temperature);\n      }\n\n      if (topP !== null) {\n        span.setAttribute(TracingAttributes.GEN_AI_REQUEST_TOP_P, topP);\n      }\n\n      if (maxPromptTokens !== null) {\n        span.setAttribute(TracingAttributes.GEN_AI_REQUEST_MAX_INPUT_TOKENS, maxPromptTokens);\n      }\n\n      if (maxCompletionTokens !== null) {\n        span.setAttribute(TracingAttributes.GEN_AI_REQUEST_MAX_OUTPUT_TOKENS, maxCompletionTokens);\n      }\n\n      if (responseFormat) {\n        span.setAttribute(TracingAttributes.GEN_AI_REQUEST_RESPONSE_FORMAT, responseFormat);\n      }\n\n      if (runStatus) {\n        span.setAttribute(TracingAttributes.GEN_AI_THREAD_RUN_STATUS, runStatus);\n      }\n\n      if (responseModel) {\n        span.setAttribute(TracingAttributes.GEN_AI_RESPONSE_MODEL, responseModel);\n      }\n\n      if (usagePromptTokens) {\n        span.setAttribute(TracingAttributes.GEN_AI_USAGE_INPUT_TOKENS, usagePromptTokens);\n      }\n\n      if (usageCompletionTokens) {\n        span.setAttribute(TracingAttributes.GEN_AI_USAGE_OUTPUT_TOKENS, usageCompletionTokens);\n      }\n      if (operationName) {\n        span.setAttribute(TracingAttributes.GEN_AI_OPERATION_NAME, operationName);\n      }\n    }\n    return;\n  }\n\n  static addSpanEvent(\n    span: Span,\n    eventName: string,\n    attributeOptions: Omit<TracingAttributeOptions, \"operationName\">,\n  ): void {\n    if (span.isRecording()) {\n      const {\n        threadId,\n        agentId,\n        runId,\n        messageId,\n        eventContent,\n        usageCompletionTokens,\n        usagePromptTokens,\n        messageStatus,\n      } = attributeOptions;\n      const attributes: Record<string, unknown> = {};\n\n      if (eventContent) {\n        attributes[TracingAttributes.GEN_AI_EVENT_CONTENT] = eventContent;\n      }\n\n      if (threadId) {\n        attributes[TracingAttributes.GEN_AI_THREAD_ID] = threadId;\n      }\n\n      if (agentId) {\n        attributes[TracingAttributes.GEN_AI_AGENT_ID] = agentId;\n      }\n\n      if (runId) {\n        attributes[TracingAttributes.GEN_AI_THREAD_RUN_ID] = runId;\n      }\n\n      if (messageId) {\n        attributes[TracingAttributes.GEN_AI_MESSAGE_ID] = messageId;\n      }\n      if (messageStatus) {\n        attributes[TracingAttributes.GEN_AI_MESSAGE_STATUS] = messageStatus;\n      }\n\n      if (usagePromptTokens) {\n        attributes[TracingAttributes.GEN_AI_USAGE_INPUT_TOKENS] = usagePromptTokens;\n      }\n\n      if (usageCompletionTokens) {\n        attributes[TracingAttributes.GEN_AI_USAGE_OUTPUT_TOKENS] = usageCompletionTokens;\n      }\n\n      if (span.addEvent) {\n        span.addEvent(eventName, { attributes });\n      }\n    }\n    return;\n  }\n}\n"]}

package/dist/commonjs/tsdoc-metadata.json

@@ -0,0 +1,11 @@
+// This file is read by tools that parse documentation comments conforming to the TSDoc standard.
+// It should be published with your NPM package.  It should not be tracked by Git.
+{
+  "tsdocVersion": "0.12",
+  "toolPackages": [
+    {
+      "packageName": "@microsoft/api-extractor",
+      "packageVersion": "7.48.0"
+    }
+  ]
+}

package/dist/esm/agents/assistants.d.ts

@@ -0,0 +1,14 @@
+import type { Client } from "@azure-rest/core-client";
+import type { AgentDeletionStatusOutput, AgentOutput, OpenAIPageableListOfAgentOutput } from "../customization/outputModels.js";
+import type { CreateAgentOptionalParams, DeleteAgentOptionalParams, GetAgentOptionalParams, ListAgentsOptionalParams, UpdateAgentOptionalParams } from "./customModels.js";
+/** Creates a new agent. */
+export declare function createAgent(context: Client, model: string, options?: CreateAgentOptionalParams): Promise<AgentOutput>;
+/** Gets a list of agents that were previously created. */
+export declare function listAgents(context: Client, options?: ListAgentsOptionalParams): Promise<OpenAIPageableListOfAgentOutput>;
+/** Retrieves an existing agent. */
+export declare function getAgent(context: Client, assistantId: string, options?: GetAgentOptionalParams): Promise<AgentOutput>;
+/** Modifies an existing agent. */
+export declare function updateAgent(context: Client, assistantId: string, options?: UpdateAgentOptionalParams): Promise<AgentOutput>;
+/** Deletes an agent. */
+export declare function deleteAgent(context: Client, assistantId: string, options?: DeleteAgentOptionalParams): Promise<AgentDeletionStatusOutput>;
+//# sourceMappingURL=assistants.d.ts.map

package/dist/esm/agents/assistants.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"assistants.d.ts","sourceRoot":"","sources":["../../../src/agents/assistants.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,yBAAyB,CAAC;AAEtD,OAAO,KAAK,EACV,yBAAyB,EACzB,WAAW,EACX,+BAA+B,EAChC,MAAM,kCAAkC,CAAC;AAU1C,OAAO,KAAK,EACV,yBAAyB,EACzB,yBAAyB,EACzB,sBAAsB,EACtB,wBAAwB,EACxB,yBAAyB,EAC1B,MAAM,mBAAmB,CAAC;AAmB3B,2BAA2B;AAC3B,wBAAsB,WAAW,CAC/B,OAAO,EAAE,MAAM,EACf,KAAK,EAAE,MAAM,EACb,OAAO,GAAE,yBAA8B,GACtC,OAAO,CAAC,WAAW,CAAC,CAuBtB;AAED,0DAA0D;AAC1D,wBAAsB,UAAU,CAC9B,OAAO,EAAE,MAAM,EACf,OAAO,GAAE,wBAA6B,GACrC,OAAO,CAAC,+BAA+B,CAAC,CAoB1C;AAED,mCAAmC;AACnC,wBAAsB,QAAQ,CAC5B,OAAO,EAAE,MAAM,EACf,WAAW,EAAE,MAAM,EACnB,OAAO,GAAE,sBAA2B,GACnC,OAAO,CAAC,WAAW,CAAC,CA0BtB;AAED,kCAAkC;AAClC,wBAAsB,WAAW,CAC/B,OAAO,EAAE,MAAM,EACf,WAAW,EAAE,MAAM,EACnB,OAAO,GAAE,yBAA8B,GACtC,OAAO,CAAC,WAAW,CAAC,CAyBtB;AAED,wBAAwB;AACxB,wBAAsB,WAAW,CAC/B,OAAO,EAAE,MAAM,EACf,WAAW,EAAE,MAAM,EACnB,OAAO,GAAE,yBAA8B,GACtC,OAAO,CAAC,yBAAyB,CAAC,CAsBpC"}

package/dist/esm/agents/assistants.js

@@ -0,0 +1,161 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+import { operationOptionsToRequestParameters } from "@azure-rest/core-client";
+import { validateLimit, validateMetadata, validateOrder, validateVectorStoreDataType, } from "./inputValidations.js";
+import { TracingUtility } from "../tracing.js";
+import { traceEndCreateOrUpdateAgent, traceStartCreateOrUpdateAgent } from "./assistantsTrace.js";
+import { traceEndAgentGeneric, traceStartAgentGeneric } from "./traceUtility.js";
+import * as ConvertFromWire from "../customization/convertOutputModelsFromWire.js";
+import * as ConverterToWire from "../customization/convertModelsToWrite.js";
+import { convertToListQueryParameters } from "../customization/convertParametersToWire.js";
+import { createOpenAIError } from "./openAIError.js";
+const expectedStatuses = ["200"];
+var Tools;
+(function (Tools) {
+    Tools["CodeInterpreter"] = "code_interpreter";
+    Tools["FileSearch"] = "file_search";
+    Tools["Function"] = "function";
+    Tools["BingGrounding"] = "bing_grounding";
+    Tools["MicrosoftFabric"] = "microsoft_fabric";
+    Tools["SharepointGrounding"] = "sharepoint_grounding";
+    Tools["AzureAISearch"] = "azure_ai_search";
+})(Tools || (Tools = {}));
+/** Creates a new agent. */
+export async function createAgent(context, model, options = {}) {
+    const createOptions = Object.assign(Object.assign({}, operationOptionsToRequestParameters(options)), { body: Object.assign({}, ConverterToWire.convertCreateAgentOptions(Object.assign(Object.assign({}, options), { model }))) });
+    validateCreateAgentParameters(createOptions);
+    const response = await TracingUtility.withSpan("CreateAgent", createOptions, async (updatedOptions) => {
+        const result = await context.path("/assistants").post(updatedOptions);
+        if (!expectedStatuses.includes(result.status)) {
+            throw createOpenAIError(result);
+        }
+        return result.body;
+    }, traceStartCreateOrUpdateAgent, traceEndCreateOrUpdateAgent);
+    return ConvertFromWire.convertAgentOutput(response);
+}
+/** Gets a list of agents that were previously created. */
+export async function listAgents(context, options = {}) {
+    const listAgentsOptions = Object.assign(Object.assign({}, operationOptionsToRequestParameters(options)), { queryParameters: convertToListQueryParameters(options) });
+    validateListAgentsParameters(listAgentsOptions);
+    const response = await TracingUtility.withSpan("ListAgents", listAgentsOptions || {}, async (updateOptions) => {
+        const result = await context.path("/assistants").get(updateOptions);
+        if (!expectedStatuses.includes(result.status)) {
+            throw createOpenAIError(result);
+        }
+        return result.body;
+    });
+    return ConvertFromWire.convertOpenAIPageableListOfAgentOutput(response);
+}
+/** Retrieves an existing agent. */
+export async function getAgent(context, assistantId, options = {}) {
+    const getAgentOptions = Object.assign({}, operationOptionsToRequestParameters(options));
+    validateAssistantId(assistantId);
+    const response = await TracingUtility.withSpan("GetAgent", getAgentOptions, async (updateOptions) => {
+        const result = await context
+            .path("/assistants/{assistantId}", assistantId)
+            .get(updateOptions);
+        if (!expectedStatuses.includes(result.status)) {
+            throw createOpenAIError(result);
+        }
+        return result.body;
+    }, (span, updatedOptions) => traceStartAgentGeneric(span, Object.assign(Object.assign({}, updatedOptions), { tracingAttributeOptions: { agentId: assistantId } })));
+    return ConvertFromWire.convertAgentOutput(response);
+}
+/** Modifies an existing agent. */
+export async function updateAgent(context, assistantId, options = {}) {
+    const updateAgentOptions = Object.assign(Object.assign({}, operationOptionsToRequestParameters(options)), { body: Object.assign({}, ConverterToWire.convertUpdateAgentOptions(options)) });
+    validateUpdateAgentParameters(assistantId, updateAgentOptions);
+    const response = await TracingUtility.withSpan("UpdateAgent", updateAgentOptions, async (updateOptions) => {
+        const result = await context
+            .path("/assistants/{assistantId}", assistantId)
+            .post(updateOptions);
+        if (!expectedStatuses.includes(result.status)) {
+            throw createOpenAIError(result);
+        }
+        return result.body;
+    }, (span, updatedOptions) => traceStartCreateOrUpdateAgent(span, updatedOptions, assistantId), traceEndCreateOrUpdateAgent);
+    return ConvertFromWire.convertAgentOutput(response);
+}
+/** Deletes an agent. */
+export async function deleteAgent(context, assistantId, options = {}) {
+    const deleteAgentOptions = Object.assign({}, operationOptionsToRequestParameters(options));
+    validateAssistantId(assistantId);
+    const response = await TracingUtility.withSpan("DeleteAgent", deleteAgentOptions, async (updateOptions) => {
+        const result = await context
+            .path("/assistants/{assistantId}", assistantId)
+            .delete(updateOptions);
+        if (!expectedStatuses.includes(result.status)) {
+            throw createOpenAIError(result);
+        }
+        return result.body;
+    }, traceStartAgentGeneric, traceEndAgentGeneric);
+    return ConvertFromWire.convertAgentDeletionStatusOutput(response);
+}
+function validateCreateAgentParameters(options) {
+    var _a;
+    if (options.body.tools) {
+        if (options.body.tools.some((value) => !Object.values(Tools).includes(value.type))) {
+            throw new Error("Tool type must be one of 'code_interpreter', 'file_search', 'function', 'bing_grounding', 'microsoft_fabric', 'sharepoint_grounding', 'azure_ai_search'");
+        }
+    }
+    if (options.body.tool_resources) {
+        if (options.body.tool_resources.code_interpreter) {
+            if (options.body.tool_resources.code_interpreter.file_ids &&
+                options.body.tool_resources.code_interpreter.data_sources) {
+                throw new Error("Only file_ids or data_sources can be provided, not both");
+            }
+            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.code_interpreter.data_sources) {
+                validateVectorStoreDataType(options.body.tool_resources.code_interpreter.data_sources);
+            }
+        }
+        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.file_search.vector_stores) {
+                if (options.body.tool_resources.file_search.vector_stores.length > 1) {
+                    throw new Error("Only one vector store is allowed");
+                }
+                validateVectorStoreDataType((_a = options.body.tool_resources.file_search.vector_stores[0]) === null || _a === void 0 ? void 0 : _a.configuration.data_sources);
+            }
+        }
+        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");
+            }
+        }
+    }
+    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.metadata) {
+        validateMetadata(options.body.metadata);
+    }
+}
+function validateListAgentsParameters(options) {
+    var _a, _b;
+    if ((_a = options === null || options === void 0 ? void 0 : options.queryParameters) === null || _a === void 0 ? void 0 : _a.limit) {
+        validateLimit(options.queryParameters.limit);
+    }
+    if ((_b = options === null || options === void 0 ? void 0 : options.queryParameters) === null || _b === void 0 ? void 0 : _b.order) {
+        validateOrder(options.queryParameters.order);
+    }
+}
+function validateAssistantId(assistantId) {
+    if (!assistantId) {
+        throw new Error("Assistant ID is required");
+    }
+}
+function validateUpdateAgentParameters(assistantId, options) {
+    validateAssistantId(assistantId);
+    if (options) {
+        validateCreateAgentParameters(options);
+    }
+}
+//# sourceMappingURL=assistants.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"assistants.js","sourceRoot":"","sources":["../../../src/agents/assistants.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAGlC,OAAO,EAAE,mCAAmC,EAAE,MAAM,yBAAyB,CAAC;AAM9E,OAAO,EACL,aAAa,EACb,gBAAgB,EAChB,aAAa,EACb,2BAA2B,GAC5B,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAC/C,OAAO,EAAE,2BAA2B,EAAE,6BAA6B,EAAE,MAAM,sBAAsB,CAAC;AAClG,OAAO,EAAE,oBAAoB,EAAE,sBAAsB,EAAE,MAAM,mBAAmB,CAAC;AASjF,OAAO,KAAK,eAAe,MAAM,iDAAiD,CAAC;AACnF,OAAO,KAAK,eAAe,MAAM,0CAA0C,CAAC;AAC5E,OAAO,EAAE,4BAA4B,EAAE,MAAM,6CAA6C,CAAC;AAC3F,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AAErD,MAAM,gBAAgB,GAAG,CAAC,KAAK,CAAC,CAAC;AAEjC,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,2BAA2B;AAC3B,MAAM,CAAC,KAAK,UAAU,WAAW,CAC/B,OAAe,EACf,KAAa,EACb,UAAqC,EAAE;IAEvC,MAAM,aAAa,mCACd,mCAAmC,CAAC,OAAO,CAAC,KAC/C,IAAI,oBACC,eAAe,CAAC,yBAAyB,iCAAM,OAAO,KAAE,KAAK,IAAG,IAEtE,CAAC;IAEF,6BAA6B,CAAC,aAAa,CAAC,CAAC;IAC7C,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,QAAQ,CAC5C,aAAa,EACb,aAAa,EACb,KAAK,EAAE,cAAc,EAAE,EAAE;QACvB,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACtE,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,6BAA6B,EAC7B,2BAA2B,CAC5B,CAAC;IACF,OAAO,eAAe,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC;AACtD,CAAC;AAED,0DAA0D;AAC1D,MAAM,CAAC,KAAK,UAAU,UAAU,CAC9B,OAAe,EACf,UAAoC,EAAE;IAEtC,MAAM,iBAAiB,mCAClB,mCAAmC,CAAC,OAAO,CAAC,KAC/C,eAAe,EAAE,4BAA4B,CAAC,OAAO,CAAC,GACvD,CAAC;IAEF,4BAA4B,CAAC,iBAAiB,CAAC,CAAC;IAChD,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,QAAQ,CAC5C,YAAY,EACZ,iBAAiB,IAAI,EAAE,EACvB,KAAK,EAAE,aAAa,EAAE,EAAE;QACtB,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;QACpE,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,CACF,CAAC;IAEF,OAAO,eAAe,CAAC,sCAAsC,CAAC,QAAQ,CAAC,CAAC;AAC1E,CAAC;AAED,mCAAmC;AACnC,MAAM,CAAC,KAAK,UAAU,QAAQ,CAC5B,OAAe,EACf,WAAmB,EACnB,UAAkC,EAAE;IAEpC,MAAM,eAAe,qBAChB,mCAAmC,CAAC,OAAO,CAAC,CAChD,CAAC;IAEF,mBAAmB,CAAC,WAAW,CAAC,CAAC;IACjC,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,QAAQ,CAC5C,UAAU,EACV,eAAe,EACf,KAAK,EAAE,aAAa,EAAE,EAAE;QACtB,MAAM,MAAM,GAAG,MAAM,OAAO;aACzB,IAAI,CAAC,2BAA2B,EAAE,WAAW,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,CACvB,sBAAsB,CAAC,IAAI,kCACtB,cAAc,KACjB,uBAAuB,EAAE,EAAE,OAAO,EAAE,WAAW,EAAE,IACjD,CACL,CAAC;IAEF,OAAO,eAAe,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC;AACtD,CAAC;AAED,kCAAkC;AAClC,MAAM,CAAC,KAAK,UAAU,WAAW,CAC/B,OAAe,EACf,WAAmB,EACnB,UAAqC,EAAE;IAEvC,MAAM,kBAAkB,mCACnB,mCAAmC,CAAC,OAAO,CAAC,KAC/C,IAAI,oBACC,eAAe,CAAC,yBAAyB,CAAC,OAAO,CAAC,IAExD,CAAC;IAEF,6BAA6B,CAAC,WAAW,EAAE,kBAAkB,CAAC,CAAC;IAC/D,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,QAAQ,CAC5C,aAAa,EACb,kBAAkB,EAClB,KAAK,EAAE,aAAa,EAAE,EAAE;QACtB,MAAM,MAAM,GAAG,MAAM,OAAO;aACzB,IAAI,CAAC,2BAA2B,EAAE,WAAW,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,6BAA6B,CAAC,IAAI,EAAE,cAAc,EAAE,WAAW,CAAC,EAC1F,2BAA2B,CAC5B,CAAC;IACF,OAAO,eAAe,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC;AACtD,CAAC;AAED,wBAAwB;AACxB,MAAM,CAAC,KAAK,UAAU,WAAW,CAC/B,OAAe,EACf,WAAmB,EACnB,UAAqC,EAAE;IAEvC,MAAM,kBAAkB,qBACnB,mCAAmC,CAAC,OAAO,CAAC,CAChD,CAAC;IAEF,mBAAmB,CAAC,WAAW,CAAC,CAAC;IACjC,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,QAAQ,CAC5C,aAAa,EACb,kBAAkB,EAClB,KAAK,EAAE,aAAa,EAAE,EAAE;QACtB,MAAM,MAAM,GAAG,MAAM,OAAO;aACzB,IAAI,CAAC,2BAA2B,EAAE,WAAW,CAAC;aAC9C,MAAM,CAAC,aAAa,CAAC,CAAC;QACzB,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,sBAAsB,EACtB,oBAAoB,CACrB,CAAC;IACF,OAAO,eAAe,CAAC,gCAAgC,CAAC,QAAQ,CAAC,CAAC;AACpE,CAAC;AAED,SAAS,6BAA6B,CACpC,OAA8F;;IAE9F,IAAI,OAAO,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;QACvB,IAAI,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAa,CAAC,CAAC,EAAE,CAAC;YAC5F,MAAM,IAAI,KAAK,CACb,yJAAyJ,CAC1J,CAAC;QACJ,CAAC;IACH,CAAC;IACD,IAAI,OAAO,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC;QAChC,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,YAAY,EACzD,CAAC;gBACD,MAAM,IAAI,KAAK,CAAC,yDAAyD,CAAC,CAAC;YAC7E,CAAC;YACD,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;YACD,IAAI,OAAO,CAAC,IAAI,CAAC,cAAc,CAAC,gBAAgB,CAAC,YAAY,EAAE,CAAC;gBAC9D,2BAA2B,CAAC,OAAO,CAAC,IAAI,CAAC,cAAc,CAAC,gBAAgB,CAAC,YAAY,CAAC,CAAC;YACzF,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;YACD,IAAI,OAAO,CAAC,IAAI,CAAC,cAAc,CAAC,WAAW,CAAC,aAAa,EAAE,CAAC;gBAC1D,IAAI,OAAO,CAAC,IAAI,CAAC,cAAc,CAAC,WAAW,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACrE,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;gBACtD,CAAC;gBACD,2BAA2B,CACzB,MAAA,OAAO,CAAC,IAAI,CAAC,cAAc,CAAC,WAAW,CAAC,aAAa,CAAC,CAAC,CAAC,0CAAE,aAAa,CAAC,YAAY,CACrF,CAAC;YACJ,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;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,QAAQ,EAAE,CAAC;QAC1B,gBAAgB,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC1C,CAAC;AACH,CAAC;AAED,SAAS,4BAA4B,CAAC,OAAkD;;IACtF,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,mBAAmB,CAAC,WAAmB;IAC9C,IAAI,CAAC,WAAW,EAAE,CAAC;QACjB,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;IAC9C,CAAC;AACH,CAAC;AAED,SAAS,6BAA6B,CACpC,WAAmB,EACnB,OAAmD;IAEnD,mBAAmB,CAAC,WAAW,CAAC,CAAC;IACjC,IAAI,OAAO,EAAE,CAAC;QACZ,6BAA6B,CAAC,OAAO,CAAC,CAAC;IACzC,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  AgentDeletionStatusOutput,\n  AgentOutput,\n  OpenAIPageableListOfAgentOutput,\n} from \"../customization/outputModels.js\";\nimport {\n  validateLimit,\n  validateMetadata,\n  validateOrder,\n  validateVectorStoreDataType,\n} from \"./inputValidations.js\";\nimport { TracingUtility } from \"../tracing.js\";\nimport { traceEndCreateOrUpdateAgent, traceStartCreateOrUpdateAgent } from \"./assistantsTrace.js\";\nimport { traceEndAgentGeneric, traceStartAgentGeneric } from \"./traceUtility.js\";\nimport type {\n  CreateAgentOptionalParams,\n  DeleteAgentOptionalParams,\n  GetAgentOptionalParams,\n  ListAgentsOptionalParams,\n  UpdateAgentOptionalParams,\n} from \"./customModels.js\";\nimport type * as GeneratedParameters from \"../generated/src/parameters.js\";\nimport * as ConvertFromWire from \"../customization/convertOutputModelsFromWire.js\";\nimport * as ConverterToWire from \"../customization/convertModelsToWrite.js\";\nimport { convertToListQueryParameters } from \"../customization/convertParametersToWire.js\";\nimport { createOpenAIError } from \"./openAIError.js\";\n\nconst expectedStatuses = [\"200\"];\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\n/** Creates a new agent. */\nexport async function createAgent(\n  context: Client,\n  model: string,\n  options: CreateAgentOptionalParams = {},\n): Promise<AgentOutput> {\n  const createOptions: GeneratedParameters.CreateAgentParameters = {\n    ...operationOptionsToRequestParameters(options),\n    body: {\n      ...ConverterToWire.convertCreateAgentOptions({ ...options, model }),\n    },\n  };\n\n  validateCreateAgentParameters(createOptions);\n  const response = await TracingUtility.withSpan(\n    \"CreateAgent\",\n    createOptions,\n    async (updatedOptions) => {\n      const result = await context.path(\"/assistants\").post(updatedOptions);\n      if (!expectedStatuses.includes(result.status)) {\n        throw createOpenAIError(result);\n      }\n      return result.body;\n    },\n    traceStartCreateOrUpdateAgent,\n    traceEndCreateOrUpdateAgent,\n  );\n  return ConvertFromWire.convertAgentOutput(response);\n}\n\n/** Gets a list of agents that were previously created. */\nexport async function listAgents(\n  context: Client,\n  options: ListAgentsOptionalParams = {},\n): Promise<OpenAIPageableListOfAgentOutput> {\n  const listAgentsOptions: GeneratedParameters.ListAgentsParameters = {\n    ...operationOptionsToRequestParameters(options),\n    queryParameters: convertToListQueryParameters(options),\n  };\n\n  validateListAgentsParameters(listAgentsOptions);\n  const response = await TracingUtility.withSpan(\n    \"ListAgents\",\n    listAgentsOptions || {},\n    async (updateOptions) => {\n      const result = await context.path(\"/assistants\").get(updateOptions);\n      if (!expectedStatuses.includes(result.status)) {\n        throw createOpenAIError(result);\n      }\n      return result.body;\n    },\n  );\n\n  return ConvertFromWire.convertOpenAIPageableListOfAgentOutput(response);\n}\n\n/** Retrieves an existing agent. */\nexport async function getAgent(\n  context: Client,\n  assistantId: string,\n  options: GetAgentOptionalParams = {},\n): Promise<AgentOutput> {\n  const getAgentOptions: GeneratedParameters.GetAgentParameters = {\n    ...operationOptionsToRequestParameters(options),\n  };\n\n  validateAssistantId(assistantId);\n  const response = await TracingUtility.withSpan(\n    \"GetAgent\",\n    getAgentOptions,\n    async (updateOptions) => {\n      const result = await context\n        .path(\"/assistants/{assistantId}\", assistantId)\n        .get(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: { agentId: assistantId },\n      }),\n  );\n\n  return ConvertFromWire.convertAgentOutput(response);\n}\n\n/** Modifies an existing agent. */\nexport async function updateAgent(\n  context: Client,\n  assistantId: string,\n  options: UpdateAgentOptionalParams = {},\n): Promise<AgentOutput> {\n  const updateAgentOptions: GeneratedParameters.UpdateAgentParameters = {\n    ...operationOptionsToRequestParameters(options),\n    body: {\n      ...ConverterToWire.convertUpdateAgentOptions(options),\n    },\n  };\n\n  validateUpdateAgentParameters(assistantId, updateAgentOptions);\n  const response = await TracingUtility.withSpan(\n    \"UpdateAgent\",\n    updateAgentOptions,\n    async (updateOptions) => {\n      const result = await context\n        .path(\"/assistants/{assistantId}\", assistantId)\n        .post(updateOptions);\n      if (!expectedStatuses.includes(result.status)) {\n        throw createOpenAIError(result);\n      }\n      return result.body;\n    },\n    (span, updatedOptions) => traceStartCreateOrUpdateAgent(span, updatedOptions, assistantId),\n    traceEndCreateOrUpdateAgent,\n  );\n  return ConvertFromWire.convertAgentOutput(response);\n}\n\n/** Deletes an agent. */\nexport async function deleteAgent(\n  context: Client,\n  assistantId: string,\n  options: DeleteAgentOptionalParams = {},\n): Promise<AgentDeletionStatusOutput> {\n  const deleteAgentOptions: GeneratedParameters.DeleteAgentParameters = {\n    ...operationOptionsToRequestParameters(options),\n  };\n\n  validateAssistantId(assistantId);\n  const response = await TracingUtility.withSpan(\n    \"DeleteAgent\",\n    deleteAgentOptions,\n    async (updateOptions) => {\n      const result = await context\n        .path(\"/assistants/{assistantId}\", assistantId)\n        .delete(updateOptions);\n      if (!expectedStatuses.includes(result.status)) {\n        throw createOpenAIError(result);\n      }\n      return result.body;\n    },\n    traceStartAgentGeneric,\n    traceEndAgentGeneric,\n  );\n  return ConvertFromWire.convertAgentDeletionStatusOutput(response);\n}\n\nfunction validateCreateAgentParameters(\n  options: GeneratedParameters.CreateAgentParameters | GeneratedParameters.UpdateAgentParameters,\n): void {\n  if (options.body.tools) {\n    if (options.body.tools.some((value) => !Object.values(Tools).includes(value.type 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  if (options.body.tool_resources) {\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.data_sources\n      ) {\n        throw new Error(\"Only file_ids or data_sources can be provided, not both\");\n      }\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      if (options.body.tool_resources.code_interpreter.data_sources) {\n        validateVectorStoreDataType(options.body.tool_resources.code_interpreter.data_sources);\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      if (options.body.tool_resources.file_search.vector_stores) {\n        if (options.body.tool_resources.file_search.vector_stores.length > 1) {\n          throw new Error(\"Only one vector store is allowed\");\n        }\n        validateVectorStoreDataType(\n          options.body.tool_resources.file_search.vector_stores[0]?.configuration.data_sources,\n        );\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  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.metadata) {\n    validateMetadata(options.body.metadata);\n  }\n}\n\nfunction validateListAgentsParameters(options?: GeneratedParameters.ListAgentsParameters): void {\n  if (options?.queryParameters?.limit) {\n    validateLimit(options.queryParameters.limit);\n  }\n  if (options?.queryParameters?.order) {\n    validateOrder(options.queryParameters.order);\n  }\n}\n\nfunction validateAssistantId(assistantId: string): void {\n  if (!assistantId) {\n    throw new Error(\"Assistant ID is required\");\n  }\n}\n\nfunction validateUpdateAgentParameters(\n  assistantId: string,\n  options?: GeneratedParameters.UpdateAgentParameters,\n): void {\n  validateAssistantId(assistantId);\n  if (options) {\n    validateCreateAgentParameters(options);\n  }\n}\n"]}

package/dist/esm/agents/assistantsTrace.d.ts

@@ -0,0 +1,17 @@
+import type { AgentOutput } from "../generated/src/outputModels.js";
+import type { Span } from "../tracing.js";
+import type { CreateAgentParameters, UpdateAgentParameters } from "../generated/src/parameters.js";
+/**
+ * Traces the start of creating or updating an agent.
+ * @param span - The span to trace.
+ * @param options - The options for creating an agent.
+ */
+export declare function traceStartCreateOrUpdateAgent(span: Span, options: CreateAgentParameters | UpdateAgentParameters, agentId?: string): void;
+/**
+ * Traces the end of creating an agent.
+ * @param span - The span to trace.
+ * @param _options - The options for creating an agent.
+ * @param result - The result of creating an agent.
+ */
+export declare function traceEndCreateOrUpdateAgent(span: Span, _options: CreateAgentParameters | UpdateAgentParameters, result: Promise<AgentOutput>): Promise<void>;
+//# sourceMappingURL=assistantsTrace.d.ts.map

package/dist/esm/agents/assistantsTrace.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"assistantsTrace.d.ts","sourceRoot":"","sources":["../../../src/agents/assistantsTrace.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,kCAAkC,CAAC;AACpE,OAAO,KAAK,EAA2B,IAAI,EAAE,MAAM,eAAe,CAAC;AAEnE,OAAO,KAAK,EAAE,qBAAqB,EAAE,qBAAqB,EAAE,MAAM,gCAAgC,CAAC;AAOnG;;;;GAIG;AACH,wBAAgB,6BAA6B,CAC3C,IAAI,EAAE,IAAI,EACV,OAAO,EAAE,qBAAqB,GAAG,qBAAqB,EACtD,OAAO,CAAC,EAAE,MAAM,GACf,IAAI,CAqBN;AAED;;;;;GAKG;AACH,wBAAsB,2BAA2B,CAC/C,IAAI,EAAE,IAAI,EACV,QAAQ,EAAE,qBAAqB,GAAG,qBAAqB,EACvD,MAAM,EAAE,OAAO,CAAC,WAAW,CAAC,GAC3B,OAAO,CAAC,IAAI,CAAC,CAMf"}

package/dist/esm/agents/assistantsTrace.js

@@ -0,0 +1,42 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+import { TracingUtility, TracingOperationName } from "../tracing.js";
+import { addInstructionsEvent, formatAgentApiResponse, UpdateWithAgentAttributes, } from "./traceUtility.js";
+/**
+ * Traces the start of creating or updating an agent.
+ * @param span - The span to trace.
+ * @param options - The options for creating an agent.
+ */
+export function traceStartCreateOrUpdateAgent(span, options, agentId) {
+    var _a, _b, _c, _d, _e;
+    const attributes = {
+        operationName: TracingOperationName.CREATE_AGENT,
+        name: (_a = options.body.name) !== null && _a !== void 0 ? _a : undefined,
+        model: options.body.model,
+        description: (_b = options.body.description) !== null && _b !== void 0 ? _b : undefined,
+        instructions: (_c = options.body.instructions) !== null && _c !== void 0 ? _c : undefined,
+        topP: (_d = options.body.top_p) !== null && _d !== void 0 ? _d : undefined,
+        temperature: (_e = options.body.temperature) !== null && _e !== void 0 ? _e : undefined,
+        responseFormat: formatAgentApiResponse(options.body.response_format),
+    };
+    if (agentId) {
+        attributes.operationName = TracingOperationName.CREATE_UPDATE_AGENT;
+        attributes.agentId = agentId;
+    }
+    TracingUtility.setSpanAttributes(span, TracingOperationName.CREATE_AGENT, UpdateWithAgentAttributes(attributes));
+    addInstructionsEvent(span, options.body);
+}
+/**
+ * Traces the end of creating an agent.
+ * @param span - The span to trace.
+ * @param _options - The options for creating an agent.
+ * @param result - The result of creating an agent.
+ */
+export async function traceEndCreateOrUpdateAgent(span, _options, result) {
+    const resolvedResult = await result;
+    const attributes = {
+        agentId: resolvedResult.id,
+    };
+    TracingUtility.updateSpanAttributes(span, attributes);
+}
+//# sourceMappingURL=assistantsTrace.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"assistantsTrace.js","sourceRoot":"","sources":["../../../src/agents/assistantsTrace.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAIlC,OAAO,EAAE,cAAc,EAAE,oBAAoB,EAAE,MAAM,eAAe,CAAC;AAErE,OAAO,EACL,oBAAoB,EACpB,sBAAsB,EACtB,yBAAyB,GAC1B,MAAM,mBAAmB,CAAC;AAE3B;;;;GAIG;AACH,MAAM,UAAU,6BAA6B,CAC3C,IAAU,EACV,OAAsD,EACtD,OAAgB;;IAEhB,MAAM,UAAU,GAA4B;QAC1C,aAAa,EAAE,oBAAoB,CAAC,YAAY;QAChD,IAAI,EAAE,MAAA,OAAO,CAAC,IAAI,CAAC,IAAI,mCAAI,SAAS;QACpC,KAAK,EAAE,OAAO,CAAC,IAAI,CAAC,KAAK;QACzB,WAAW,EAAE,MAAA,OAAO,CAAC,IAAI,CAAC,WAAW,mCAAI,SAAS;QAClD,YAAY,EAAE,MAAA,OAAO,CAAC,IAAI,CAAC,YAAY,mCAAI,SAAS;QACpD,IAAI,EAAE,MAAA,OAAO,CAAC,IAAI,CAAC,KAAK,mCAAI,SAAS;QACrC,WAAW,EAAE,MAAA,OAAO,CAAC,IAAI,CAAC,WAAW,mCAAI,SAAS;QAClD,cAAc,EAAE,sBAAsB,CAAC,OAAO,CAAC,IAAI,CAAC,eAAe,CAAC;KACrE,CAAC;IACF,IAAI,OAAO,EAAE,CAAC;QACZ,UAAU,CAAC,aAAa,GAAG,oBAAoB,CAAC,mBAAmB,CAAC;QACpE,UAAU,CAAC,OAAO,GAAG,OAAO,CAAC;IAC/B,CAAC;IACD,cAAc,CAAC,iBAAiB,CAC9B,IAAI,EACJ,oBAAoB,CAAC,YAAY,EACjC,yBAAyB,CAAC,UAAU,CAAC,CACtC,CAAC;IACF,oBAAoB,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;AAC3C,CAAC;AAED;;;;;GAKG;AACH,MAAM,CAAC,KAAK,UAAU,2BAA2B,CAC/C,IAAU,EACV,QAAuD,EACvD,MAA4B;IAE5B,MAAM,cAAc,GAAG,MAAM,MAAM,CAAC;IACpC,MAAM,UAAU,GAA4B;QAC1C,OAAO,EAAE,cAAc,CAAC,EAAE;KAC3B,CAAC;IACF,cAAc,CAAC,oBAAoB,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;AACxD,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport type { AgentOutput } from \"../generated/src/outputModels.js\";\nimport type { TracingAttributeOptions, Span } from \"../tracing.js\";\nimport { TracingUtility, TracingOperationName } from \"../tracing.js\";\nimport type { CreateAgentParameters, UpdateAgentParameters } from \"../generated/src/parameters.js\";\nimport {\n  addInstructionsEvent,\n  formatAgentApiResponse,\n  UpdateWithAgentAttributes,\n} from \"./traceUtility.js\";\n\n/**\n * Traces the start of creating or updating an agent.\n * @param span - The span to trace.\n * @param options - The options for creating an agent.\n */\nexport function traceStartCreateOrUpdateAgent(\n  span: Span,\n  options: CreateAgentParameters | UpdateAgentParameters,\n  agentId?: string,\n): void {\n  const attributes: TracingAttributeOptions = {\n    operationName: TracingOperationName.CREATE_AGENT,\n    name: options.body.name ?? undefined,\n    model: options.body.model,\n    description: options.body.description ?? undefined,\n    instructions: options.body.instructions ?? undefined,\n    topP: options.body.top_p ?? undefined,\n    temperature: options.body.temperature ?? undefined,\n    responseFormat: formatAgentApiResponse(options.body.response_format),\n  };\n  if (agentId) {\n    attributes.operationName = TracingOperationName.CREATE_UPDATE_AGENT;\n    attributes.agentId = agentId;\n  }\n  TracingUtility.setSpanAttributes(\n    span,\n    TracingOperationName.CREATE_AGENT,\n    UpdateWithAgentAttributes(attributes),\n  );\n  addInstructionsEvent(span, options.body);\n}\n\n/**\n * Traces the end of creating an agent.\n * @param span - The span to trace.\n * @param _options - The options for creating an agent.\n * @param result - The result of creating an agent.\n */\nexport async function traceEndCreateOrUpdateAgent(\n  span: Span,\n  _options: CreateAgentParameters | UpdateAgentParameters,\n  result: Promise<AgentOutput>,\n): Promise<void> {\n  const resolvedResult = await result;\n  const attributes: TracingAttributeOptions = {\n    agentId: resolvedResult.id,\n  };\n  TracingUtility.updateSpanAttributes(span, attributes);\n}\n"]}