@azure/ai-projects 1.0.0-beta.1

package/dist/browser/tracing.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tracing.js","sourceRoot":"","sources":["../../src/tracing.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAQlC,OAAO,EAAE,mBAAmB,EAAE,MAAM,qBAAqB,CAAC;AAC1D,OAAO,EAAE,YAAY,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAC3D,OAAO,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AACnD,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AAErC,MAAM,CAAN,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,KAAjB,iBAAiB,QA2B5B;AACD,MAAM,CAAN,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,KAApB,oBAAoB,QAW/B;AAkCD,MAAM,OAAO,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,MAAM,CAAC,OAAO,CACZ,oEAAoE,eAAe,CAAC,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,MAAM,CAAC,OAAO,CACZ,mEAAmE,eAAe,CAAC,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;;AApOc,4BAAa,GAAG,mBAAmB,CAAC;IACjD,SAAS,EAAE,6BAA6B;IACxC,WAAW,EAAE,YAAY;IACzB,cAAc,EAAE,WAAW;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/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/commonjs/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/commonjs/agents/assistants.js

@@ -0,0 +1,169 @@
+"use strict";
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createAgent = createAgent;
+exports.listAgents = listAgents;
+exports.getAgent = getAgent;
+exports.updateAgent = updateAgent;
+exports.deleteAgent = deleteAgent;
+const tslib_1 = require("tslib");
+const core_client_1 = require("@azure-rest/core-client");
+const inputValidations_js_1 = require("./inputValidations.js");
+const tracing_js_1 = require("../tracing.js");
+const assistantsTrace_js_1 = require("./assistantsTrace.js");
+const traceUtility_js_1 = require("./traceUtility.js");
+const ConvertFromWire = tslib_1.__importStar(require("../customization/convertOutputModelsFromWire.js"));
+const ConverterToWire = tslib_1.__importStar(require("../customization/convertModelsToWrite.js"));
+const convertParametersToWire_js_1 = require("../customization/convertParametersToWire.js");
+const openAIError_js_1 = require("./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. */
+async function createAgent(context, model, options = {}) {
+    const createOptions = Object.assign(Object.assign({}, (0, core_client_1.operationOptionsToRequestParameters)(options)), { body: Object.assign({}, ConverterToWire.convertCreateAgentOptions(Object.assign(Object.assign({}, options), { model }))) });
+    validateCreateAgentParameters(createOptions);
+    const response = await tracing_js_1.TracingUtility.withSpan("CreateAgent", createOptions, async (updatedOptions) => {
+        const result = await context.path("/assistants").post(updatedOptions);
+        if (!expectedStatuses.includes(result.status)) {
+            throw (0, openAIError_js_1.createOpenAIError)(result);
+        }
+        return result.body;
+    }, assistantsTrace_js_1.traceStartCreateOrUpdateAgent, assistantsTrace_js_1.traceEndCreateOrUpdateAgent);
+    return ConvertFromWire.convertAgentOutput(response);
+}
+/** Gets a list of agents that were previously created. */
+async function listAgents(context, options = {}) {
+    const listAgentsOptions = Object.assign(Object.assign({}, (0, core_client_1.operationOptionsToRequestParameters)(options)), { queryParameters: (0, convertParametersToWire_js_1.convertToListQueryParameters)(options) });
+    validateListAgentsParameters(listAgentsOptions);
+    const response = await tracing_js_1.TracingUtility.withSpan("ListAgents", listAgentsOptions || {}, async (updateOptions) => {
+        const result = await context.path("/assistants").get(updateOptions);
+        if (!expectedStatuses.includes(result.status)) {
+            throw (0, openAIError_js_1.createOpenAIError)(result);
+        }
+        return result.body;
+    });
+    return ConvertFromWire.convertOpenAIPageableListOfAgentOutput(response);
+}
+/** Retrieves an existing agent. */
+async function getAgent(context, assistantId, options = {}) {
+    const getAgentOptions = Object.assign({}, (0, core_client_1.operationOptionsToRequestParameters)(options));
+    validateAssistantId(assistantId);
+    const response = await tracing_js_1.TracingUtility.withSpan("GetAgent", getAgentOptions, async (updateOptions) => {
+        const result = await context
+            .path("/assistants/{assistantId}", assistantId)
+            .get(updateOptions);
+        if (!expectedStatuses.includes(result.status)) {
+            throw (0, openAIError_js_1.createOpenAIError)(result);
+        }
+        return result.body;
+    }, (span, updatedOptions) => (0, traceUtility_js_1.traceStartAgentGeneric)(span, Object.assign(Object.assign({}, updatedOptions), { tracingAttributeOptions: { agentId: assistantId } })));
+    return ConvertFromWire.convertAgentOutput(response);
+}
+/** Modifies an existing agent. */
+async function updateAgent(context, assistantId, options = {}) {
+    const updateAgentOptions = Object.assign(Object.assign({}, (0, core_client_1.operationOptionsToRequestParameters)(options)), { body: Object.assign({}, ConverterToWire.convertUpdateAgentOptions(options)) });
+    validateUpdateAgentParameters(assistantId, updateAgentOptions);
+    const response = await tracing_js_1.TracingUtility.withSpan("UpdateAgent", updateAgentOptions, async (updateOptions) => {
+        const result = await context
+            .path("/assistants/{assistantId}", assistantId)
+            .post(updateOptions);
+        if (!expectedStatuses.includes(result.status)) {
+            throw (0, openAIError_js_1.createOpenAIError)(result);
+        }
+        return result.body;
+    }, (span, updatedOptions) => (0, assistantsTrace_js_1.traceStartCreateOrUpdateAgent)(span, updatedOptions, assistantId), assistantsTrace_js_1.traceEndCreateOrUpdateAgent);
+    return ConvertFromWire.convertAgentOutput(response);
+}
+/** Deletes an agent. */
+async function deleteAgent(context, assistantId, options = {}) {
+    const deleteAgentOptions = Object.assign({}, (0, core_client_1.operationOptionsToRequestParameters)(options));
+    validateAssistantId(assistantId);
+    const response = await tracing_js_1.TracingUtility.withSpan("DeleteAgent", deleteAgentOptions, async (updateOptions) => {
+        const result = await context
+            .path("/assistants/{assistantId}", assistantId)
+            .delete(updateOptions);
+        if (!expectedStatuses.includes(result.status)) {
+            throw (0, openAIError_js_1.createOpenAIError)(result);
+        }
+        return result.body;
+    }, traceUtility_js_1.traceStartAgentGeneric, traceUtility_js_1.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) {
+                (0, inputValidations_js_1.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");
+                }
+                (0, inputValidations_js_1.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) {
+        (0, inputValidations_js_1.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) {
+        (0, inputValidations_js_1.validateLimit)(options.queryParameters.limit);
+    }
+    if ((_b = options === null || options === void 0 ? void 0 : options.queryParameters) === null || _b === void 0 ? void 0 : _b.order) {
+        (0, inputValidations_js_1.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/commonjs/agents/assistants.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"assistants.js","sourceRoot":"","sources":["../../../src/agents/assistants.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;AA4ClC,kCA2BC;AAGD,gCAuBC;AAGD,4BA8BC;AAGD,kCA6BC;AAGD,kCA0BC;;AA5LD,yDAA8E;AAM9E,+DAK+B;AAC/B,8CAA+C;AAC/C,6DAAkG;AAClG,uDAAiF;AASjF,yGAAmF;AACnF,kGAA4E;AAC5E,4FAA2F;AAC3F,qDAAqD;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;AACpB,KAAK,UAAU,WAAW,CAC/B,OAAe,EACf,KAAa,EACb,UAAqC,EAAE;IAEvC,MAAM,aAAa,mCACd,IAAA,iDAAmC,EAAC,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,2BAAc,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,IAAA,kCAAiB,EAAC,MAAM,CAAC,CAAC;QAClC,CAAC;QACD,OAAO,MAAM,CAAC,IAAI,CAAC;IACrB,CAAC,EACD,kDAA6B,EAC7B,gDAA2B,CAC5B,CAAC;IACF,OAAO,eAAe,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC;AACtD,CAAC;AAED,0DAA0D;AACnD,KAAK,UAAU,UAAU,CAC9B,OAAe,EACf,UAAoC,EAAE;IAEtC,MAAM,iBAAiB,mCAClB,IAAA,iDAAmC,EAAC,OAAO,CAAC,KAC/C,eAAe,EAAE,IAAA,yDAA4B,EAAC,OAAO,CAAC,GACvD,CAAC;IAEF,4BAA4B,CAAC,iBAAiB,CAAC,CAAC;IAChD,MAAM,QAAQ,GAAG,MAAM,2BAAc,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,IAAA,kCAAiB,EAAC,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;AAC5B,KAAK,UAAU,QAAQ,CAC5B,OAAe,EACf,WAAmB,EACnB,UAAkC,EAAE;IAEpC,MAAM,eAAe,qBAChB,IAAA,iDAAmC,EAAC,OAAO,CAAC,CAChD,CAAC;IAEF,mBAAmB,CAAC,WAAW,CAAC,CAAC;IACjC,MAAM,QAAQ,GAAG,MAAM,2BAAc,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,IAAA,kCAAiB,EAAC,MAAM,CAAC,CAAC;QAClC,CAAC;QACD,OAAO,MAAM,CAAC,IAAI,CAAC;IACrB,CAAC,EACD,CAAC,IAAI,EAAE,cAAc,EAAE,EAAE,CACvB,IAAA,wCAAsB,EAAC,IAAI,kCACtB,cAAc,KACjB,uBAAuB,EAAE,EAAE,OAAO,EAAE,WAAW,EAAE,IACjD,CACL,CAAC;IAEF,OAAO,eAAe,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC;AACtD,CAAC;AAED,kCAAkC;AAC3B,KAAK,UAAU,WAAW,CAC/B,OAAe,EACf,WAAmB,EACnB,UAAqC,EAAE;IAEvC,MAAM,kBAAkB,mCACnB,IAAA,iDAAmC,EAAC,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,2BAAc,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,IAAA,kCAAiB,EAAC,MAAM,CAAC,CAAC;QAClC,CAAC;QACD,OAAO,MAAM,CAAC,IAAI,CAAC;IACrB,CAAC,EACD,CAAC,IAAI,EAAE,cAAc,EAAE,EAAE,CAAC,IAAA,kDAA6B,EAAC,IAAI,EAAE,cAAc,EAAE,WAAW,CAAC,EAC1F,gDAA2B,CAC5B,CAAC;IACF,OAAO,eAAe,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC;AACtD,CAAC;AAED,wBAAwB;AACjB,KAAK,UAAU,WAAW,CAC/B,OAAe,EACf,WAAmB,EACnB,UAAqC,EAAE;IAEvC,MAAM,kBAAkB,qBACnB,IAAA,iDAAmC,EAAC,OAAO,CAAC,CAChD,CAAC;IAEF,mBAAmB,CAAC,WAAW,CAAC,CAAC;IACjC,MAAM,QAAQ,GAAG,MAAM,2BAAc,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,IAAA,kCAAiB,EAAC,MAAM,CAAC,CAAC;QAClC,CAAC;QACD,OAAO,MAAM,CAAC,IAAI,CAAC;IACrB,CAAC,EACD,wCAAsB,EACtB,sCAAoB,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,IAAA,iDAA2B,EAAC,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,IAAA,iDAA2B,EACzB,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,IAAA,sCAAgB,EAAC,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,IAAA,mCAAa,EAAC,OAAO,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;IAC/C,CAAC;IACD,IAAI,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,eAAe,0CAAE,KAAK,EAAE,CAAC;QACpC,IAAA,mCAAa,EAAC,OAAO,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;IAC/C,CAAC;AACH,CAAC;AAED,SAAS,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/commonjs/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/commonjs/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/commonjs/agents/assistantsTrace.js

@@ -0,0 +1,46 @@
+"use strict";
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.traceStartCreateOrUpdateAgent = traceStartCreateOrUpdateAgent;
+exports.traceEndCreateOrUpdateAgent = traceEndCreateOrUpdateAgent;
+const tracing_js_1 = require("../tracing.js");
+const traceUtility_js_1 = require("./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.
+ */
+function traceStartCreateOrUpdateAgent(span, options, agentId) {
+    var _a, _b, _c, _d, _e;
+    const attributes = {
+        operationName: tracing_js_1.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: (0, traceUtility_js_1.formatAgentApiResponse)(options.body.response_format),
+    };
+    if (agentId) {
+        attributes.operationName = tracing_js_1.TracingOperationName.CREATE_UPDATE_AGENT;
+        attributes.agentId = agentId;
+    }
+    tracing_js_1.TracingUtility.setSpanAttributes(span, tracing_js_1.TracingOperationName.CREATE_AGENT, (0, traceUtility_js_1.UpdateWithAgentAttributes)(attributes));
+    (0, traceUtility_js_1.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.
+ */
+async function traceEndCreateOrUpdateAgent(span, _options, result) {
+    const resolvedResult = await result;
+    const attributes = {
+        agentId: resolvedResult.id,
+    };
+    tracing_js_1.TracingUtility.updateSpanAttributes(span, attributes);
+}
+//# sourceMappingURL=assistantsTrace.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"assistantsTrace.js","sourceRoot":"","sources":["../../../src/agents/assistantsTrace.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;AAiBlC,sEAyBC;AAQD,kEAUC;AAxDD,8CAAqE;AAErE,uDAI2B;AAE3B;;;;GAIG;AACH,SAAgB,6BAA6B,CAC3C,IAAU,EACV,OAAsD,EACtD,OAAgB;;IAEhB,MAAM,UAAU,GAA4B;QAC1C,aAAa,EAAE,iCAAoB,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,IAAA,wCAAsB,EAAC,OAAO,CAAC,IAAI,CAAC,eAAe,CAAC;KACrE,CAAC;IACF,IAAI,OAAO,EAAE,CAAC;QACZ,UAAU,CAAC,aAAa,GAAG,iCAAoB,CAAC,mBAAmB,CAAC;QACpE,UAAU,CAAC,OAAO,GAAG,OAAO,CAAC;IAC/B,CAAC;IACD,2BAAc,CAAC,iBAAiB,CAC9B,IAAI,EACJ,iCAAoB,CAAC,YAAY,EACjC,IAAA,2CAAyB,EAAC,UAAU,CAAC,CACtC,CAAC;IACF,IAAA,sCAAoB,EAAC,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;AAC3C,CAAC;AAED;;;;;GAKG;AACI,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,2BAAc,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"]}

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

@@ -0,0 +1,288 @@
+import type { OperationOptions, RequestParameters } from "@azure-rest/core-client";
+import type { AbortSignalLike } from "@azure/abort-controller";
+import type { ThreadRunOutput } from "../customization/outputModels.js";
+import type { AgentEventMessageStream } from "./streamingModels.js";
+import type { AgentThreadCreationOptions, CreateAgentOptions, CreateAndRunThreadOptions, CreateRunOptions, UpdateAgentOptions, UpdateAgentThreadOptions, VectorStoreFileStatusFilter, VectorStoreOptions, VectorStoreUpdateOptions } from "../customization/models.js";
+import type { ListMessagesQueryParamProperties, ListFilesQueryParamProperties } from "../customization/parameters.js";
+import type { CreateVectorStoreFileBatchOptions, CreateVectorStoreFileOptions } from "./vectorStoresModels.js";
+/**
+ * Optional request parameters support passing headers, abort signal, etc.
+ */
+export type OptionalRequestParameters = Pick<RequestParameters, "headers" | "timeout" | "abortSignal" | "tracingOptions">;
+/**
+ * Request options for list requests.
+ */
+export interface ListQueryParameters {
+    /**
+     * A limit on the number of objects to be returned. Limit can range between 1 and 100, and the default is 20.
+     */
+    limit?: number;
+    /**
+     * Sort order by the created_at timestamp of the objects. asc for ascending order and desc for descending order.
+     */
+    order?: "asc" | "desc";
+    /**
+     * A cursor for use in pagination. after is an object ID that defines your place in the list. For instance, if you make a list request and receive 100 objects, ending with obj_foo, your subsequent call can include after=obj_foo in order to fetch the next page of the list.
+     */
+    after?: string;
+    /**
+     * A cursor for use in pagination. before is an object ID that defines your place in the list. For instance, if you make a list request and receive 100 objects, ending with obj_foo, your subsequent call can include before=obj_foo in order to fetch the previous page of the list.
+     */
+    before?: string;
+}
+/**
+ * Options for configuring polling behavior.
+ */
+export interface PollingOptions {
+    /**
+     * The interval, in milliseconds, to wait between polling attempts. If not specified, a default interval of 1000ms will be used.
+     */
+    sleepIntervalInMs?: number;
+    /**
+     * An AbortSignalLike object (as defined by @azure/abort-controller) that can be used to cancel the polling operation.
+     */
+    abortSignal?: AbortSignalLike;
+}
+/**
+ * Optional parameters configuring polling behavior.
+ */
+export interface PollingOptionsParams {
+    /** Options for configuring polling behavior. */
+    pollingOptions?: PollingOptions;
+}
+/**
+ * Agent run response with support to stream.
+ */
+export type AgentRunResponse = PromiseLike<ThreadRunOutput> & {
+    /**
+     * Function to start streaming the agent event messages.
+     * @returns A promise that resolves to an AgentEventMessageStream.
+     */
+    stream: () => Promise<AgentEventMessageStream>;
+};
+/**
+ * Optional parameters for creating and running a thread, excluding the assistantId.
+ */
+export type CreateRunOptionalParams = Omit<CreateRunOptions & OperationOptions, "assistantId"> & OperationOptions;
+/**
+ * Optional parameters for creating and running a thread, excluding the assistantId.
+ */
+export type CreateAndRunThreadOptionalParams = Omit<CreateAndRunThreadOptions, "assistantId"> & OperationOptions;
+/**
+ * Optional parameters for listing run queries.
+ */
+export interface ListRunQueryOptionalParams extends ListQueryParameters, OperationOptions {
+}
+/**
+ * Optional parameters for getting a run.
+ */
+export interface GetRunOptionalParams extends OperationOptions {
+}
+/**
+ * Optional parameters for canceling a run.
+ */
+export interface CancelRunOptionalParams extends OperationOptions {
+}
+/**
+ * Optional parameters for submitting tool outputs to a run.
+ */
+export interface SubmitToolOutputsToRunOptionalParams extends OperationOptions {
+    /**
+     * Whether to stream the tool outputs.
+     */
+    stream?: boolean;
+}
+/**
+ * Optional parameters for updating a run.
+ */
+export interface UpdateRunOptionalParams extends OperationOptions {
+    /**  Metadata to update in the run.  */
+    metadata?: Record<string, string> | null;
+}
+/**
+ * Optional parameters for creating an agent thread.
+ */
+export interface CreateAgentThreadOptionalParams extends AgentThreadCreationOptions, OperationOptions {
+}
+/**
+ * Optional parameters for getting an agent thread.
+ */
+export interface GetAgentThreadOptionalParams extends OperationOptions {
+}
+/**
+ * Optional parameters for updating an agent thread.
+ */
+export interface UpdateAgentThreadOptionalParams extends UpdateAgentThreadOptions, OperationOptions {
+}
+/**
+ * Optional parameters for deleting an agent thread.
+ */
+export interface DeleteAgentThreadOptionalParams extends OperationOptions {
+}
+/**
+ * Optional parameters for getting an run step.
+ */
+export interface GetRunStepOptionalParams extends OperationOptions {
+}
+/**
+ * Optional parameters for listing run steps.
+ */
+export interface ListRunStepsOptionalParams extends ListQueryParameters, OperationOptions {
+}
+/**
+ * Optional parameters for creating a message.
+ */
+export interface CreateMessageOptionalParams extends OperationOptions {
+}
+/**
+ * Optional parameters for updating a message.
+ */
+export interface UpdateMessageOptionalParams extends OperationOptions {
+    /** Metadata to update in the message. */
+    metadata?: Record<string, string> | null;
+}
+/**
+ * Optional parameters for listing messages.
+ */
+export interface ListMessagesOptionalParams extends ListMessagesQueryParamProperties, OperationOptions {
+}
+/**
+ * Optional parameters creating vector store.
+ */
+export interface CreateVectorStoreOptionalParams extends VectorStoreOptions, OperationOptions {
+}
+/**
+ * Optional parameters for creating vector store with polling.
+ */
+export interface CreateVectorStoreWithPollingOptionalParams extends CreateVectorStoreOptionalParams, PollingOptionsParams {
+}
+/**
+ * Optional parameters for listing vector stores.
+ */
+export interface ListVectorStoresOptionalParams extends ListQueryParameters, OperationOptions {
+}
+/**
+ * Optional parameters for updating a vector store.
+ */
+export interface UpdateVectorStoreOptionalParams extends VectorStoreUpdateOptions, OperationOptions {
+}
+/**
+ * Optional parameters for deleting a vector store.
+ */
+export interface DeleteVectorStoreOptionalParams extends OperationOptions {
+}
+/**
+ * Optional parameters for getting a vector store.
+ */
+export interface GetVectorStoreOptionalParams extends OperationOptions {
+}
+/**
+ * Optional parameters for listing vector store files.
+ */
+export interface ListVectorStoreFilesOptionalParams extends ListQueryParameters, OperationOptions {
+}
+/**
+ * Optional parameters for creating a vector store file.
+ */
+export interface CreateVectorStoreFileOptionalParams extends CreateVectorStoreFileOptions, OperationOptions {
+}
+/**
+ * Optional parameters for getting a vector store file.
+ */
+export interface GetVectorStoreFileOptionalParams extends OperationOptions {
+}
+/**
+ * Optional parameters for deleting a vector store file.
+ */
+export interface DeleteVectorStoreFileOptionalParams extends OperationOptions {
+}
+/**
+ * Optional parameters for creating a vector store file with polling.
+ */
+export interface CreateVectorStoreFileWithPollingOptionalParams extends CreateVectorStoreFileOptions, PollingOptionsParams, OperationOptions {
+}
+/**
+ * Optional parameters for listing vector store file batches.
+ */
+export interface ListVectorStoreFileBatchFilesOptionalParams extends ListQueryParameters, OperationOptions {
+    /** Filter by file status. */
+    filter?: VectorStoreFileStatusFilter;
+}
+/**
+ * Optional parameters for getting a vector store file batch.
+ */
+export interface GetVectorStoreFileBatchOptionalParams extends OperationOptions {
+}
+/**
+ * Optional parameters for canceling a vector store file batch.
+ */
+export interface CancelVectorStoreFileBatchOptionalParams extends OperationOptions {
+}
+/**
+ * Optional parameters for creating a vector store file batch.
+ */
+export interface CreateVectorStoreFileBatchOptionalParams extends CreateVectorStoreFileBatchOptions, OperationOptions {
+}
+/**
+ * Optional parameters for creating a vector store file batch with polling.
+ */
+export interface CreateVectorStoreFileBatchWithPollingOptionalParams extends CreateVectorStoreFileBatchOptionalParams, PollingOptionsParams {
+}
+/**
+ * Optional parameters for creating agent.
+ */
+export interface CreateAgentOptionalParams extends Omit<CreateAgentOptions, "model">, OperationOptions {
+}
+/**
+ * Optional parameters for updating agent.
+ */
+export interface UpdateAgentOptionalParams extends UpdateAgentOptions, OperationOptions {
+}
+/**
+ * Optional parameters for deleting agent.
+ */
+export interface DeleteAgentOptionalParams extends OperationOptions {
+}
+/**
+ * Optional parameters for getting agent.
+ */
+export interface GetAgentOptionalParams extends OperationOptions {
+}
+/**
+ * Optional parameters for listing agents.
+ */
+export interface ListAgentsOptionalParams extends ListQueryParameters, OperationOptions {
+}
+/**
+ * Optional parameters for listing files.
+ */
+export interface ListFilesOptionalParams extends ListFilesQueryParamProperties, OperationOptions {
+}
+/**
+ * Optional parameters for deleting a file.
+ */
+export interface DeleteFileOptionalParams extends OperationOptions {
+}
+/**
+ * Optional parameters for getting a file.
+ */
+export interface GetFileOptionalParams extends OperationOptions {
+}
+/**
+ * Optional parameters for getting file content.
+ */
+export interface GetFileContentOptionalParams extends OperationOptions {
+}
+/**
+ * Optional parameters for uploading a file.
+ */
+export interface UploadFileOptionalParams extends OperationOptions {
+    /** The name of the file. */
+    fileName?: string;
+}
+/**
+ * Optional parameters for uploading a file with polling.
+ */
+export interface UploadFileWithPollingOptionalParams extends UploadFileOptionalParams, PollingOptionsParams {
+}
+//# sourceMappingURL=customModels.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"customModels.d.ts","sourceRoot":"","sources":["../../../src/agents/customModels.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,gBAAgB,EAAE,iBAAiB,EAAE,MAAM,yBAAyB,CAAC;AACnF,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAC/D,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,kCAAkC,CAAC;AACxE,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,sBAAsB,CAAC;AACpE,OAAO,KAAK,EACV,0BAA0B,EAC1B,kBAAkB,EAClB,yBAAyB,EACzB,gBAAgB,EAChB,kBAAkB,EAClB,wBAAwB,EACxB,2BAA2B,EAC3B,kBAAkB,EAClB,wBAAwB,EACzB,MAAM,4BAA4B,CAAC;AACpC,OAAO,KAAK,EACV,gCAAgC,EAChC,6BAA6B,EAC9B,MAAM,gCAAgC,CAAC;AACxC,OAAO,KAAK,EACV,iCAAiC,EACjC,4BAA4B,EAC7B,MAAM,yBAAyB,CAAC;AAEjC;;GAEG;AACH,MAAM,MAAM,yBAAyB,GAAG,IAAI,CAC1C,iBAAiB,EACjB,SAAS,GAAG,SAAS,GAAG,aAAa,GAAG,gBAAgB,CACzD,CAAC;AAEF;;GAEG;AACH,MAAM,WAAW,mBAAmB;IAClC;;OAEG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IAEf;;OAEG;IACH,KAAK,CAAC,EAAE,KAAK,GAAG,MAAM,CAAC;IAEvB;;OAEG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IAEf;;OAEG;IACH,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAED;;GAEG;AACH,MAAM,WAAW,cAAc;IAC7B;;OAEG;IACH,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAE3B;;OAEG;IACH,WAAW,CAAC,EAAE,eAAe,CAAC;CAC/B;AAED;;GAEG;AACH,MAAM,WAAW,oBAAoB;IACnC,gDAAgD;IAChD,cAAc,CAAC,EAAE,cAAc,CAAC;CACjC;AAED;;GAEG;AACH,MAAM,MAAM,gBAAgB,GAAG,WAAW,CAAC,eAAe,CAAC,GAAG;IAC5D;;;OAGG;IACH,MAAM,EAAE,MAAM,OAAO,CAAC,uBAAuB,CAAC,CAAC;CAChD,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,uBAAuB,GAAG,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,EAAE,aAAa,CAAC,GAC5F,gBAAgB,CAAC;AAEnB;;GAEG;AACH,MAAM,MAAM,gCAAgC,GAAG,IAAI,CAAC,yBAAyB,EAAE,aAAa,CAAC,GAC3F,gBAAgB,CAAC;AAEnB;;GAEG;AACH,MAAM,WAAW,0BAA2B,SAAQ,mBAAmB,EAAE,gBAAgB;CAAG;AAE5F;;GAEG;AACH,MAAM,WAAW,oBAAqB,SAAQ,gBAAgB;CAAG;AAEjE;;GAEG;AACH,MAAM,WAAW,uBAAwB,SAAQ,gBAAgB;CAAG;AAEpE;;GAEG;AACH,MAAM,WAAW,oCAAqC,SAAQ,gBAAgB;IAC5E;;OAEG;IACH,MAAM,CAAC,EAAE,OAAO,CAAC;CAClB;AAED;;GAEG;AACH,MAAM,WAAW,uBAAwB,SAAQ,gBAAgB;IAC/D,uCAAuC;IACvC,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,IAAI,CAAC;CAC1C;AAED;;GAEG;AACH,MAAM,WAAW,+BACf,SAAQ,0BAA0B,EAChC,gBAAgB;CAAG;AAEvB;;GAEG;AACH,MAAM,WAAW,4BAA6B,SAAQ,gBAAgB;CAAG;AAEzE;;GAEG;AACH,MAAM,WAAW,+BACf,SAAQ,wBAAwB,EAC9B,gBAAgB;CAAG;AAEvB;;GAEG;AACH,MAAM,WAAW,+BAAgC,SAAQ,gBAAgB;CAAG;AAE5E;;GAEG;AACH,MAAM,WAAW,wBAAyB,SAAQ,gBAAgB;CAAG;AAErE;;GAEG;AACH,MAAM,WAAW,0BAA2B,SAAQ,mBAAmB,EAAE,gBAAgB;CAAG;AAE5F;;GAEG;AACH,MAAM,WAAW,2BAA4B,SAAQ,gBAAgB;CAAG;AAExE;;GAEG;AACH,MAAM,WAAW,2BAA4B,SAAQ,gBAAgB;IACnE,yCAAyC;IACzC,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,IAAI,CAAC;CAC1C;AAED;;GAEG;AACH,MAAM,WAAW,0BACf,SAAQ,gCAAgC,EACtC,gBAAgB;CAAG;AAEvB;;GAEG;AACH,MAAM,WAAW,+BAAgC,SAAQ,kBAAkB,EAAE,gBAAgB;CAAG;AAEhG;;GAEG;AACH,MAAM,WAAW,0CACf,SAAQ,+BAA+B,EACrC,oBAAoB;CAAG;AAE3B;;GAEG;AACH,MAAM,WAAW,8BAA+B,SAAQ,mBAAmB,EAAE,gBAAgB;CAAG;AAEhG;;GAEG;AACH,MAAM,WAAW,+BACf,SAAQ,wBAAwB,EAC9B,gBAAgB;CAAG;AAEvB;;GAEG;AACH,MAAM,WAAW,+BAAgC,SAAQ,gBAAgB;CAAG;AAE5E;;GAEG;AACH,MAAM,WAAW,4BAA6B,SAAQ,gBAAgB;CAAG;AAEzE;;GAEG;AACH,MAAM,WAAW,kCAAmC,SAAQ,mBAAmB,EAAE,gBAAgB;CAAG;AAEpG;;GAEG;AACH,MAAM,WAAW,mCACf,SAAQ,4BAA4B,EAClC,gBAAgB;CAAG;AAEvB;;GAEG;AACH,MAAM,WAAW,gCAAiC,SAAQ,gBAAgB;CAAG;AAE7E;;GAEG;AACH,MAAM,WAAW,mCAAoC,SAAQ,gBAAgB;CAAG;AAEhF;;GAEG;AACH,MAAM,WAAW,8CACf,SAAQ,4BAA4B,EAClC,oBAAoB,EACpB,gBAAgB;CAAG;AAEvB;;GAEG;AACH,MAAM,WAAW,2CACf,SAAQ,mBAAmB,EACzB,gBAAgB;IAClB,6BAA6B;IAC7B,MAAM,CAAC,EAAE,2BAA2B,CAAC;CACtC;AAED;;GAEG;AACH,MAAM,WAAW,qCAAsC,SAAQ,gBAAgB;CAAG;AAElF;;GAEG;AACH,MAAM,WAAW,wCAAyC,SAAQ,gBAAgB;CAAG;AAErF;;GAEG;AACH,MAAM,WAAW,wCACf,SAAQ,iCAAiC,EACvC,gBAAgB;CAAG;AAEvB;;GAEG;AACH,MAAM,WAAW,mDACf,SAAQ,wCAAwC,EAC9C,oBAAoB;CAAG;AAE3B;;GAEG;AACH,MAAM,WAAW,yBACf,SAAQ,IAAI,CAAC,kBAAkB,EAAE,OAAO,CAAC,EACvC,gBAAgB;CAAG;AAEvB;;GAEG;AACH,MAAM,WAAW,yBAA0B,SAAQ,kBAAkB,EAAE,gBAAgB;CAAG;AAE1F;;GAEG;AACH,MAAM,WAAW,yBAA0B,SAAQ,gBAAgB;CAAG;AAEtE;;GAEG;AACH,MAAM,WAAW,sBAAuB,SAAQ,gBAAgB;CAAG;AAEnE;;GAEG;AACH,MAAM,WAAW,wBAAyB,SAAQ,mBAAmB,EAAE,gBAAgB;CAAG;AAE1F;;GAEG;AACH,MAAM,WAAW,uBAAwB,SAAQ,6BAA6B,EAAE,gBAAgB;CAAG;AAEnG;;GAEG;AACH,MAAM,WAAW,wBAAyB,SAAQ,gBAAgB;CAAG;AAErE;;GAEG;AACH,MAAM,WAAW,qBAAsB,SAAQ,gBAAgB;CAAG;AAElE;;GAEG;AACH,MAAM,WAAW,4BAA6B,SAAQ,gBAAgB;CAAG;AAEzE;;GAEG;AACH,MAAM,WAAW,wBAAyB,SAAQ,gBAAgB;IAChE,4BAA4B;IAC5B,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAED;;GAEG;AACH,MAAM,WAAW,mCACf,SAAQ,wBAAwB,EAC9B,oBAAoB;CAAG"}

package/dist/commonjs/agents/customModels.js

@@ -0,0 +1,5 @@
+"use strict";
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=customModels.js.map