@azure/ai-projects 2.1.1 → 2.3.0

package/dist/esm/tracing/responseTracing.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"responseTracing.js","sourceRoot":"","sources":["../../../src/tracing/responseTracing.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAIlC,OAAO,EAAE,SAAS,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AACjE,OAAO,EACL,aAAa,EACb,mBAAmB,EACnB,uBAAuB,EACvB,yBAAyB,EACzB,kBAAkB,GACnB,MAAM,iBAAiB,CAAC;AACzB,OAAO,EACL,aAAa,EACb,sBAAsB,EACtB,4BAA4B,EAC5B,oBAAoB,EACpB,oBAAoB,EACpB,eAAe,GAChB,MAAM,gBAAgB,CAAC;AACxB,OAAO,EAAE,uBAAuB,EAAE,gBAAgB,EAAE,MAAM,cAAc,CAAC;AASzE,MAAM,CAAC,KAAK,UAAU,yBAAyB,CAC7C,eAAgD,EAChD,IAA6B,EAC7B,OAAgC,EAChC,QAAgB,EAChB,aAAqB,EACrB,QAAgB,EAChB,SAAkB,EAClB,mBAA4B,KAAK;IAEjC,MAAM,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;IACpC,MAAM,EAAE,aAAa,EAAE,UAAU,EAAE,GAAG,aAAa,CAAC,QAAQ,CAAC,CAAC;IAC9D,IAAI,SAA6B,CAAC;IAClC,IAAI,aAAiC,CAAC;IAEtC,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE,GAAG,SAAS,CAAC,QAAQ,CAAC,CAAC;IAC1C,IAAI,CAAC;QACH,uBAAuB,CACrB,IAAI,EACJ,aAAa,EACb,aAAa,EACb,UAAU,EACV,IAAI,EACJ,SAAS,EACT,gBAAgB,CACjB,CAAC;QACF,MAAM,QAAQ,GAAG,CAAC,MAAM,gBAAgB,CAAC,GAAG,EAAE,GAAG,EAAE,CACjD,eAAe,CAAC,IAAI,EAAE,OAAO,CAAC,CAC/B,CAAgB,CAAC;QAClB,yBAAyB,CAAC,IAAI,EAAE,QAAQ,EAAE,gBAAgB,CAAC,CAAC;QAC5D,uBAAuB,CAAC,IAAI,EAAE,QAAQ,EAAE,gBAAgB,CAAC,CAAC;QAC1D,aAAa,GAAG,OAAO,QAAQ,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC;QAEhF,iBAAiB;QACjB,MAAM,eAAe,GAAG,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,CAAC;QAC/D,uBAAuB,CAAC,eAAe,EAAE;YACvC,aAAa;YACb,aAAa;YACb,UAAU;YACV,aAAa;SACd,CAAC,CAAC;QACH,IAAI,QAAQ,CAAC,KAAK,EAAE,CAAC;YACnB,gBAAgB,CAAC,QAAQ,CAAC,KAAK,CAAC,YAAY,EAAE,QAAQ,CAAC,KAAK,CAAC,aAAa,EAAE;gBAC1E,aAAa;gBACb,aAAa;gBACb,UAAU;gBACV,aAAa;aACd,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,oBAAoB;QACjD,IAAI,CAAC,GAAG,EAAE,CAAC;QACX,OAAO,QAAQ,CAAC;IAClB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,kBAAkB,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QAChC,SAAS,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,IAAI,KAAK,CAAC,WAAW,EAAE,IAAI,IAAI,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC;QAChG,MAAM,eAAe,GAAG,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,CAAC;QAC/D,uBAAuB,CAAC,eAAe,EAAE;YACvC,aAAa;YACb,aAAa;YACb,UAAU;YACV,SAAS;SACV,CAAC,CAAC;QACH,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,uBAAuB;QACxE,IAAI,CAAC,GAAG,EAAE,CAAC;QACX,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAC1C,eAAgD,EAChD,IAA6B,EAC7B,OAAgC,EAChC,QAAgB,EAChB,aAAqB,EACrB,QAAgB,EAChB,SAAkB,EAClB,mBAA4B,KAAK;IAEjC,MAAM,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;IACpC,MAAM,EAAE,aAAa,EAAE,UAAU,EAAE,GAAG,aAAa,CAAC,QAAQ,CAAC,CAAC;IAC9D,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE,GAAG,SAAS,CAAC,QAAQ,CAAC,CAAC;IAC1C,uBAAuB,CACrB,IAAI,EACJ,aAAa,EACb,aAAa,EACb,UAAU,EACV,IAAI,EACJ,SAAS,EACT,gBAAgB,CACjB,CAAC;IAEF,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,gBAAgB,CACnC,GAAG,EACH,GAAG,EAAE,CAAC,eAAe,CAAC,IAAI,EAAE,OAAO,CAAoC,CACxE,CAAC;QACF,OAAO,UAAU,CACf,MAAM,EACN,IAAI,EACJ;YACE,SAAS;YACT,aAAa;YACb,aAAa;YACb,UAAU;SACX,EACD,gBAAgB,CACjB,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,kBAAkB,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QAChC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;QACvF,IAAI,CAAC,GAAG,EAAE,CAAC;QACX,MAAM,SAAS,GACb,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,IAAI,KAAK,CAAC,WAAW,EAAE,IAAI,IAAI,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC;QACtF,MAAM,eAAe,GAAG,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,CAAC;QAC/D,uBAAuB,CAAC,eAAe,EAAE;YACvC,aAAa;YACb,aAAa;YACb,UAAU;YACV,SAAS;SACV,CAAC,CAAC;QACH,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC;AAED,SAAS,UAAU,CACjB,WAAmC,EACnC,IAAU,EACV,UAAgC,EAChC,mBAA4B,KAAK;IAEjC,MAAM,QAAQ,GAAG,WAAW,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,CAAC;IACrD,IAAI,iBAA0C,CAAC;IAE/C,SAAS,mBAAmB,CAAC,SAAkB;QAC7C,MAAM,eAAe,GAAG,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,UAAU,CAAC,SAAS,CAAC,GAAG,IAAI,CAAC;QAC1E,MAAM,aAAa,GACjB,iBAAiB,IAAI,OAAO,iBAAiB,CAAC,KAAK,KAAK,QAAQ;YAC9D,CAAC,CAAC,iBAAiB,CAAC,KAAK;YACzB,CAAC,CAAC,SAAS,CAAC;QAChB,uBAAuB,CAAC,eAAe,EAAE;YACvC,aAAa,EAAE,UAAU,CAAC,aAAa;YACvC,aAAa,EAAE,UAAU,CAAC,aAAa;YACvC,UAAU,EAAE,UAAU,CAAC,UAAU;YACjC,aAAa;YACb,SAAS;SACV,CAAC,CAAC;QACH,IAAI,iBAAiB,EAAE,KAAK,EAAE,CAAC;YAC7B,gBAAgB,CACd,iBAAiB,CAAC,KAAK,CAAC,YAAY,EACpC,iBAAiB,CAAC,KAAK,CAAC,aAAa,EACrC;gBACE,aAAa,EAAE,UAAU,CAAC,aAAa;gBACvC,aAAa,EAAE,UAAU,CAAC,aAAa;gBACvC,UAAU,EAAE,UAAU,CAAC,UAAU;gBACjC,aAAa;aACd,CACF,CAAC;QACJ,CAAC;IACH,CAAC;IAED,MAAM,eAAe,GAA2B;QAC9C,KAAK,CAAC,IAAI;YACR,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;gBACrC,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;oBAChB,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK;oBAClC,IAAI,CAAC,GAAG,EAAE,CAAC;oBACX,mBAAmB,EAAE,CAAC;oBACtB,OAAO,MAAM,CAAC;gBAChB,CAAC;gBACD,iEAAiE;gBACjE,MAAM,KAAK,GAAG,MAAM,CAAC,KAAgC,CAAC;gBACtD,IAAI,KAAK,EAAE,IAAI,KAAK,2BAA2B,EAAE,CAAC;oBAChD,MAAM,IAAI,GAAI,KAA4C,CAAC,IAAI,CAAC;oBAChE,IAAI,IAAI,EAAE,IAAI,KAAK,iBAAiB,EAAE,CAAC;wBACrC,4BAA4B,CAAC,IAAI,EAAE,IAAI,EAAE,gBAAgB,CAAC,CAAC;oBAC7D,CAAC;gBACH,CAAC;gBACD,IAAI,KAAK,EAAE,IAAI,KAAK,oBAAoB,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;oBAC3D,iBAAiB,GAAG,KAAK,CAAC,QAAuB,CAAC;oBAClD,yBAAyB,CAAC,IAAI,EAAE,iBAAiB,EAAE,gBAAgB,CAAC,CAAC;gBACvE,CAAC;gBACD,OAAO,MAAM,CAAC;YAChB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,kBAAkB,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;gBAChC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;gBACvF,IAAI,CAAC,GAAG,EAAE,CAAC;gBACX,MAAM,OAAO,GACX,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,IAAI,KAAK,CAAC,WAAW,EAAE,IAAI,IAAI,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC;gBACtF,mBAAmB,CAAC,OAAO,CAAC,CAAC;gBAC7B,MAAM,KAAK,CAAC;YACd,CAAC;QACH,CAAC;QACD,KAAK,CAAC,MAAM,CAAC,KAAe;YAC1B,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK;YAClC,IAAI,CAAC,GAAG,EAAE,CAAC;YACX,mBAAmB,EAAE,CAAC;YACtB,IAAI,QAAQ,CAAC,MAAM,EAAE,CAAC;gBACpB,OAAO,QAAQ,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAChC,CAAC;YACD,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;QAC1C,CAAC;QACD,KAAK,CAAC,KAAK,CAAC,KAAe;YACzB,kBAAkB,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;YAChC,IAAI,CAAC,SAAS,CAAC;gBACb,IAAI,EAAE,CAAC;gBACP,OAAO,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAE,KAAe,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO;aACrE,CAAC,CAAC;YACH,IAAI,CAAC,GAAG,EAAE,CAAC;YACX,MAAM,OAAO,GACX,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,IAAI,KAAK,CAAC,WAAW,EAAE,IAAI,IAAI,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC;YACtF,mBAAmB,CAAC,OAAO,CAAC,CAAC;YAC7B,IAAI,QAAQ,CAAC,KAAK,EAAE,CAAC;gBACnB,OAAO,QAAQ,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YAC/B,CAAC;YACD,MAAM,KAAK,CAAC;QACd,CAAC;KACF,CAAC;IAEF,8FAA8F;IAC9F,MAAM,aAAa,GAAG,MAAM,CAAC,MAAM,CACjC,MAAM,CAAC,cAAc,CAAC,WAAW,CAAW,EAC5C,MAAM,CAAC,yBAAyB,CAAC,WAAW,CAAC,CACpB,CAAC;IAC3B,aAA6D,CAAC,MAAM,CAAC,aAAa,CAAC,GAAG,GAAG,EAAE,CAC1F,eAAe,CAAC;IAElB,OAAO,aAAa,CAAC;AACvB,CAAC;AAED;;GAEG;AACH,SAAS,4BAA4B,CACnC,IAAU,EACV,IAA6B,EAC7B,mBAA4B,KAAK;IAEjC,MAAM,eAAe,GAA4B,EAAE,CAAC;IACpD,IAAI,IAAI,CAAC,MAAM;QAAE,eAAe,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;IACtD,IAAI,gBAAgB,EAAE,CAAC;QACrB,IAAI,IAAI,CAAC,SAAS;YAAE,eAAe,CAAC,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC;QAC/D,IAAI,IAAI,CAAC,kBAAkB;YAAE,eAAe,CAAC,kBAAkB,GAAG,IAAI,CAAC,kBAAkB,CAAC;IAC5F,CAAC;IAED,MAAM,YAAY,GAAG;QACnB;YACE,IAAI,EAAE,UAAU;YAChB,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,iBAAiB,EAAE,OAAO,EAAE,eAAe,EAAE,CAAC;SAC/D;KACF,CAAC;IAEF,IAAI,CAAC,QAAQ,CAAC,4BAA4B,EAAE;QAC1C,CAAC,oBAAoB,CAAC,EAAE,eAAe;QACvC,CAAC,oBAAoB,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC;KACrD,CAAC,CAAC;AACL,CAAC;AAED;;GAEG;AACH,SAAS,uBAAuB,CAC9B,IAAU,EACV,QAAqB,EACrB,mBAA4B,KAAK;IAEjC,MAAM,MAAM,GAAI,QAA+C,CAAC,MAAM,CAAC;IACvE,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC;QAAE,OAAO;IACnC,KAAK,MAAM,IAAI,IAAI,MAAM,EAAE,CAAC;QAC1B,IACE,IAAI;YACJ,OAAO,IAAI,KAAK,QAAQ;YACvB,IAAgC,CAAC,IAAI,KAAK,iBAAiB,EAC5D,CAAC;YACD,4BAA4B,CAAC,IAAI,EAAE,IAA+B,EAAE,gBAAgB,CAAC,CAAC;QACxF,CAAC;IACH,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,uBAAuB,CAC3C,mBAAoD,EACpD,IAAe,EACf,QAAgB;IAEhB,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE,GAAG,SAAS,CAAC,aAAa,CAAC,mBAAmB,CAAC,CAAC;IACnE,IAAI,CAAC;QACH,mBAAmB,CAAC,IAAI,EAAE,aAAa,CAAC,mBAAmB,EAAE,QAAQ,CAAC,CAAC;QACvE,MAAM,MAAM,GAAG,MAAM,gBAAgB,CAAC,GAAG,EAAE,GAAG,EAAE,CAC7C,mBAA6D,CAAC,GAAG,IAAI,CAAC,CACxE,CAAC;QACF,MAAM,YAAY,GAAG,MAAiC,CAAC;QACvD,IAAI,OAAO,YAAY,CAAC,EAAE,KAAK,QAAQ,EAAE,CAAC;YACxC,IAAI,CAAC,YAAY,CAAC,sBAAsB,EAAE,YAAY,CAAC,EAAE,CAAC,CAAC;QAC7D,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;YAAS,CAAC;QACT,IAAI,CAAC,GAAG,EAAE,CAAC;IACb,CAAC;AACH,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport type { Response as OAIResponse } from \"openai/resources/responses/responses\";\nimport type { Span } from \"@opentelemetry/api\";\nimport { startSpan, runInSpanContext } from \"./tracingClient.js\";\nimport {\n  parseEndpoint,\n  setCommonAttributes,\n  setCommonSpanAttributes,\n  setResponseSpanAttributes,\n  setErrorAttributes,\n} from \"./attributes.js\";\nimport {\n  OperationName,\n  GEN_AI_CONVERSATION_ID,\n  GEN_AI_WORKFLOW_ACTION_EVENT,\n  GEN_AI_PROVIDER_NAME,\n  GEN_AI_EVENT_CONTENT,\n  AGENTS_PROVIDER,\n} from \"./constants.js\";\nimport { recordOperationDuration, recordTokenUsage } from \"./metrics.js\";\n\ninterface StreamMetricsContext {\n  startTime: number;\n  operationName: string;\n  serverAddress?: string;\n  serverPort?: number;\n}\n\nexport async function traceNonStreamingResponse(\n  responsesCreate: (...args: unknown[]) => unknown,\n  body: Record<string, unknown>,\n  options: Record<string, unknown>,\n  spanName: string,\n  operationName: string,\n  endpoint: string,\n  agentName?: string,\n  contentRecording: boolean = false,\n): Promise<unknown> {\n  const startTime = performance.now();\n  const { serverAddress, serverPort } = parseEndpoint(endpoint);\n  let errorType: string | undefined;\n  let responseModel: string | undefined;\n\n  const { span, ctx } = startSpan(spanName);\n  try {\n    setCommonSpanAttributes(\n      span,\n      operationName,\n      serverAddress,\n      serverPort,\n      body,\n      agentName,\n      contentRecording,\n    );\n    const response = (await runInSpanContext(ctx, () =>\n      responsesCreate(body, options),\n    )) as OAIResponse;\n    setResponseSpanAttributes(span, response, contentRecording);\n    addWorkflowActionEvents(span, response, contentRecording);\n    responseModel = typeof response.model === \"string\" ? response.model : undefined;\n\n    // Record metrics\n    const durationSeconds = (performance.now() - startTime) / 1000;\n    recordOperationDuration(durationSeconds, {\n      operationName,\n      serverAddress,\n      serverPort,\n      responseModel,\n    });\n    if (response.usage) {\n      recordTokenUsage(response.usage.input_tokens, response.usage.output_tokens, {\n        operationName,\n        serverAddress,\n        serverPort,\n        responseModel,\n      });\n    }\n\n    span.setStatus({ code: 1 }); // SpanStatusCode.OK\n    span.end();\n    return response;\n  } catch (error) {\n    setErrorAttributes(span, error);\n    errorType = error instanceof Error ? error.name || error.constructor?.name || \"Error\" : \"Error\";\n    const durationSeconds = (performance.now() - startTime) / 1000;\n    recordOperationDuration(durationSeconds, {\n      operationName,\n      serverAddress,\n      serverPort,\n      errorType,\n    });\n    span.setStatus({ code: 2, message: errorType }); // SpanStatusCode.ERROR\n    span.end();\n    throw error;\n  }\n}\n\nexport async function traceStreamingResponse(\n  responsesCreate: (...args: unknown[]) => unknown,\n  body: Record<string, unknown>,\n  options: Record<string, unknown>,\n  spanName: string,\n  operationName: string,\n  endpoint: string,\n  agentName?: string,\n  contentRecording: boolean = false,\n): Promise<unknown> {\n  const startTime = performance.now();\n  const { serverAddress, serverPort } = parseEndpoint(endpoint);\n  const { span, ctx } = startSpan(spanName);\n  setCommonSpanAttributes(\n    span,\n    operationName,\n    serverAddress,\n    serverPort,\n    body,\n    agentName,\n    contentRecording,\n  );\n\n  try {\n    const stream = await runInSpanContext(\n      ctx,\n      () => responsesCreate(body, options) as Promise<AsyncIterable<unknown>>,\n    );\n    return wrapStream(\n      stream,\n      span,\n      {\n        startTime,\n        operationName,\n        serverAddress,\n        serverPort,\n      },\n      contentRecording,\n    );\n  } catch (error) {\n    setErrorAttributes(span, error);\n    span.setStatus({ code: 2, message: error instanceof Error ? error.message : \"Error\" });\n    span.end();\n    const errorType =\n      error instanceof Error ? error.name || error.constructor?.name || \"Error\" : \"Error\";\n    const durationSeconds = (performance.now() - startTime) / 1000;\n    recordOperationDuration(durationSeconds, {\n      operationName,\n      serverAddress,\n      serverPort,\n      errorType,\n    });\n    throw error;\n  }\n}\n\nfunction wrapStream(\n  innerStream: AsyncIterable<unknown>,\n  span: Span,\n  metricsCtx: StreamMetricsContext,\n  contentRecording: boolean = false,\n): AsyncIterable<unknown> {\n  const iterator = innerStream[Symbol.asyncIterator]();\n  let completedResponse: OAIResponse | undefined;\n\n  function recordStreamMetrics(errorType?: string): void {\n    const durationSeconds = (performance.now() - metricsCtx.startTime) / 1000;\n    const responseModel =\n      completedResponse && typeof completedResponse.model === \"string\"\n        ? completedResponse.model\n        : undefined;\n    recordOperationDuration(durationSeconds, {\n      operationName: metricsCtx.operationName,\n      serverAddress: metricsCtx.serverAddress,\n      serverPort: metricsCtx.serverPort,\n      responseModel,\n      errorType,\n    });\n    if (completedResponse?.usage) {\n      recordTokenUsage(\n        completedResponse.usage.input_tokens,\n        completedResponse.usage.output_tokens,\n        {\n          operationName: metricsCtx.operationName,\n          serverAddress: metricsCtx.serverAddress,\n          serverPort: metricsCtx.serverPort,\n          responseModel,\n        },\n      );\n    }\n  }\n\n  const wrappedIterator: AsyncIterator<unknown> = {\n    async next() {\n      try {\n        const result = await iterator.next();\n        if (result.done) {\n          span.setStatus({ code: 1 }); // OK\n          span.end();\n          recordStreamMetrics();\n          return result;\n        }\n        // Check for response.completed event to capture final attributes\n        const event = result.value as Record<string, unknown>;\n        if (event?.type === \"response.output_item.done\") {\n          const item = (event as { item?: Record<string, unknown> }).item;\n          if (item?.type === \"workflow_action\") {\n            addSingleWorkflowActionEvent(span, item, contentRecording);\n          }\n        }\n        if (event?.type === \"response.completed\" && event.response) {\n          completedResponse = event.response as OAIResponse;\n          setResponseSpanAttributes(span, completedResponse, contentRecording);\n        }\n        return result;\n      } catch (error) {\n        setErrorAttributes(span, error);\n        span.setStatus({ code: 2, message: error instanceof Error ? error.message : \"Error\" });\n        span.end();\n        const errType =\n          error instanceof Error ? error.name || error.constructor?.name || \"Error\" : \"Error\";\n        recordStreamMetrics(errType);\n        throw error;\n      }\n    },\n    async return(value?: unknown) {\n      span.setStatus({ code: 1 }); // OK\n      span.end();\n      recordStreamMetrics();\n      if (iterator.return) {\n        return iterator.return(value);\n      }\n      return { done: true, value: undefined };\n    },\n    async throw(error?: unknown) {\n      setErrorAttributes(span, error);\n      span.setStatus({\n        code: 2,\n        message: error instanceof Error ? (error as Error).message : \"Error\",\n      });\n      span.end();\n      const errType =\n        error instanceof Error ? error.name || error.constructor?.name || \"Error\" : \"Error\";\n      recordStreamMetrics(errType);\n      if (iterator.throw) {\n        return iterator.throw(error);\n      }\n      throw error;\n    },\n  };\n\n  // Preserve the original stream object's shape — copy all properties and override the iterator\n  const wrappedStream = Object.create(\n    Object.getPrototypeOf(innerStream) as object,\n    Object.getOwnPropertyDescriptors(innerStream),\n  ) as AsyncIterable<unknown>;\n  (wrappedStream as unknown as Record<string | symbol, unknown>)[Symbol.asyncIterator] = () =>\n    wrappedIterator;\n\n  return wrappedStream;\n}\n\n/**\n * Emits a gen_ai.workflow.action event for a single workflow action item.\n */\nfunction addSingleWorkflowActionEvent(\n  span: Span,\n  item: Record<string, unknown>,\n  contentRecording: boolean = false,\n): void {\n  const workflowDetails: Record<string, unknown> = {};\n  if (item.status) workflowDetails.status = item.status;\n  if (contentRecording) {\n    if (item.action_id) workflowDetails.action_id = item.action_id;\n    if (item.previous_action_id) workflowDetails.previous_action_id = item.previous_action_id;\n  }\n\n  const contentArray = [\n    {\n      role: \"workflow\",\n      parts: [{ type: \"workflow_action\", content: workflowDetails }],\n    },\n  ];\n\n  span.addEvent(GEN_AI_WORKFLOW_ACTION_EVENT, {\n    [GEN_AI_PROVIDER_NAME]: AGENTS_PROVIDER,\n    [GEN_AI_EVENT_CONTENT]: JSON.stringify(contentArray),\n  });\n}\n\n/**\n * Emits workflow action events from a non-streaming response.\n */\nfunction addWorkflowActionEvents(\n  span: Span,\n  response: OAIResponse,\n  contentRecording: boolean = false,\n): void {\n  const output = (response as unknown as Record<string, unknown>).output;\n  if (!Array.isArray(output)) return;\n  for (const item of output) {\n    if (\n      item &&\n      typeof item === \"object\" &&\n      (item as Record<string, unknown>).type === \"workflow_action\"\n    ) {\n      addSingleWorkflowActionEvent(span, item as Record<string, unknown>, contentRecording);\n    }\n  }\n}\n\nexport async function traceConversationCreate(\n  conversationsCreate: (...args: unknown[]) => unknown,\n  args: unknown[],\n  endpoint: string,\n): Promise<unknown> {\n  const { span, ctx } = startSpan(OperationName.CREATE_CONVERSATION);\n  try {\n    setCommonAttributes(span, OperationName.CREATE_CONVERSATION, endpoint);\n    const result = await runInSpanContext(ctx, () =>\n      (conversationsCreate as (...a: unknown[]) => Promise<unknown>)(...args),\n    );\n    const conversation = result as Record<string, unknown>;\n    if (typeof conversation.id === \"string\") {\n      span.setAttribute(GEN_AI_CONVERSATION_ID, conversation.id);\n    }\n    return result;\n  } finally {\n    span.end();\n  }\n}\n"]}

package/dist/esm/tracing/tracingClient.d.ts

@@ -0,0 +1,22 @@
+import type { Span, Context } from "@opentelemetry/api";
+export type { Span as OTelSpan } from "@opentelemetry/api";
+export { SpanStatusCode } from "@opentelemetry/api";
+export interface StartSpanResult {
+    span: Span;
+    ctx: Context;
+}
+/**
+ * Starts a new client span using the global OTel TracerProvider.
+ */
+export declare function startSpan(name: string): StartSpanResult;
+/**
+ * Injects trace context headers (traceparent/tracestate) for the current active span.
+ */
+export declare function createRequestHeaders(): Record<string, string>;
+/**
+ * Runs a callback with the given span context as the active context,
+ * so that child operations (e.g. tracingFetch header injection) see this span
+ * as their parent.
+ */
+export declare function runInSpanContext<T>(ctx: Context, fn: () => T): T;
+//# sourceMappingURL=tracingClient.d.ts.map

package/dist/esm/tracing/tracingClient.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tracingClient.d.ts","sourceRoot":"","sources":["../../../src/tracing/tracingClient.ts"],"names":[],"mappings":"AAUA,OAAO,KAAK,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,oBAAoB,CAAC;AAGxD,YAAY,EAAE,IAAI,IAAI,QAAQ,EAAE,MAAM,oBAAoB,CAAC;AAC3D,OAAO,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AAEpD,MAAM,WAAW,eAAe;IAC9B,IAAI,EAAE,IAAI,CAAC;IACX,GAAG,EAAE,OAAO,CAAC;CACd;AAED;;GAEG;AACH,wBAAgB,SAAS,CAAC,IAAI,EAAE,MAAM,GAAG,eAAe,CAMvD;AAED;;GAEG;AACH,wBAAgB,oBAAoB,IAAI,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAI7D;AAED;;;;GAIG;AACH,wBAAgB,gBAAgB,CAAC,CAAC,EAAE,GAAG,EAAE,OAAO,EAAE,EAAE,EAAE,MAAM,CAAC,GAAG,CAAC,CAEhE"}

package/dist/esm/tracing/tracingClient.js

@@ -0,0 +1,37 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+/**
+ * Thin wrapper around `\@opentelemetry/api` for tracing.
+ * Uses direct imports since `\@opentelemetry/api` is an explicit dependency.
+ * @internal
+ */
+import { trace, context, propagation, SpanKind, defaultTextMapSetter } from "@opentelemetry/api";
+import { PACKAGE_NAME, SDK_VERSION } from "../constants.js";
+export { SpanStatusCode } from "@opentelemetry/api";
+/**
+ * Starts a new client span using the global OTel TracerProvider.
+ */
+export function startSpan(name) {
+    const tracer = trace.getTracer(PACKAGE_NAME, SDK_VERSION);
+    const ctx = context.active();
+    const span = tracer.startSpan(name, { kind: SpanKind.CLIENT }, ctx);
+    const spanCtx = trace.setSpan(ctx, span);
+    return { span, ctx: spanCtx };
+}
+/**
+ * Injects trace context headers (traceparent/tracestate) for the current active span.
+ */
+export function createRequestHeaders() {
+    const headers = {};
+    propagation.inject(context.active(), headers, defaultTextMapSetter);
+    return headers;
+}
+/**
+ * Runs a callback with the given span context as the active context,
+ * so that child operations (e.g. tracingFetch header injection) see this span
+ * as their parent.
+ */
+export function runInSpanContext(ctx, fn) {
+    return context.with(ctx, fn);
+}
+//# sourceMappingURL=tracingClient.js.map

package/dist/esm/tracing/tracingClient.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tracingClient.js","sourceRoot":"","sources":["../../../src/tracing/tracingClient.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAElC;;;;GAIG;AAEH,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,WAAW,EAAE,QAAQ,EAAE,oBAAoB,EAAE,MAAM,oBAAoB,CAAC;AAEjG,OAAO,EAAE,YAAY,EAAE,WAAW,EAAE,MAAM,iBAAiB,CAAC;AAG5D,OAAO,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AAOpD;;GAEG;AACH,MAAM,UAAU,SAAS,CAAC,IAAY;IACpC,MAAM,MAAM,GAAG,KAAK,CAAC,SAAS,CAAC,YAAY,EAAE,WAAW,CAAC,CAAC;IAC1D,MAAM,GAAG,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IAC7B,MAAM,IAAI,GAAG,MAAM,CAAC,SAAS,CAAC,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,CAAC,MAAM,EAAE,EAAE,GAAG,CAAC,CAAC;IACpE,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;IACzC,OAAO,EAAE,IAAI,EAAE,GAAG,EAAE,OAAO,EAAE,CAAC;AAChC,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,oBAAoB;IAClC,MAAM,OAAO,GAA2B,EAAE,CAAC;IAC3C,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,OAAO,EAAE,oBAAoB,CAAC,CAAC;IACpE,OAAO,OAAO,CAAC;AACjB,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,gBAAgB,CAAI,GAAY,EAAE,EAAW;IAC3D,OAAO,OAAO,CAAC,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;AAC/B,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\n/**\n * Thin wrapper around `\\@opentelemetry/api` for tracing.\n * Uses direct imports since `\\@opentelemetry/api` is an explicit dependency.\n * @internal\n */\n\nimport { trace, context, propagation, SpanKind, defaultTextMapSetter } from \"@opentelemetry/api\";\nimport type { Span, Context } from \"@opentelemetry/api\";\nimport { PACKAGE_NAME, SDK_VERSION } from \"../constants.js\";\n\nexport type { Span as OTelSpan } from \"@opentelemetry/api\";\nexport { SpanStatusCode } from \"@opentelemetry/api\";\n\nexport interface StartSpanResult {\n  span: Span;\n  ctx: Context;\n}\n\n/**\n * Starts a new client span using the global OTel TracerProvider.\n */\nexport function startSpan(name: string): StartSpanResult {\n  const tracer = trace.getTracer(PACKAGE_NAME, SDK_VERSION);\n  const ctx = context.active();\n  const span = tracer.startSpan(name, { kind: SpanKind.CLIENT }, ctx);\n  const spanCtx = trace.setSpan(ctx, span);\n  return { span, ctx: spanCtx };\n}\n\n/**\n * Injects trace context headers (traceparent/tracestate) for the current active span.\n */\nexport function createRequestHeaders(): Record<string, string> {\n  const headers: Record<string, string> = {};\n  propagation.inject(context.active(), headers, defaultTextMapSetter);\n  return headers;\n}\n\n/**\n * Runs a callback with the given span context as the active context,\n * so that child operations (e.g. tracingFetch header injection) see this span\n * as their parent.\n */\nexport function runInSpanContext<T>(ctx: Context, fn: () => T): T {\n  return context.with(ctx, fn);\n}\n"]}

package/dist/esm/tracing/tracingFetch.d.ts

@@ -0,0 +1,14 @@
+import type OpenAI from "openai";
+import type { ResolvedTracingConfig } from "./configuration.js";
+type FetchFn = NonNullable<NonNullable<ConstructorParameters<typeof OpenAI>[0]>["fetch"]>;
+/**
+ * Wraps a fetch function to inject OpenTelemetry trace context headers
+ * (traceparent, tracestate) into outgoing requests when GenAI tracing is enabled.
+ *
+ * @param innerFetch - The underlying fetch function to wrap. If undefined, uses globalThis.fetch.
+ * @param tracingConfig - The resolved tracing config for this client instance.
+ * @returns A fetch-compatible function that injects tracing headers.
+ */
+export declare function getTracingFetch(innerFetch?: FetchFn, tracingConfig?: ResolvedTracingConfig): FetchFn;
+export {};
+//# sourceMappingURL=tracingFetch.d.ts.map

package/dist/esm/tracing/tracingFetch.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tracingFetch.d.ts","sourceRoot":"","sources":["../../../src/tracing/tracingFetch.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,MAAM,MAAM,QAAQ,CAAC;AACjC,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAGhE,KAAK,OAAO,GAAG,WAAW,CAAC,WAAW,CAAC,qBAAqB,CAAC,OAAO,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;AAE1F;;;;;;;GAOG;AACH,wBAAgB,eAAe,CAC7B,UAAU,CAAC,EAAE,OAAO,EACpB,aAAa,CAAC,EAAE,qBAAqB,GACpC,OAAO,CAkBT"}

package/dist/esm/tracing/tracingFetch.js

@@ -0,0 +1,30 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+import { createRequestHeaders } from "./tracingClient.js";
+/**
+ * Wraps a fetch function to inject OpenTelemetry trace context headers
+ * (traceparent, tracestate) into outgoing requests when GenAI tracing is enabled.
+ *
+ * @param innerFetch - The underlying fetch function to wrap. If undefined, uses globalThis.fetch.
+ * @param tracingConfig - The resolved tracing config for this client instance.
+ * @returns A fetch-compatible function that injects tracing headers.
+ */
+export function getTracingFetch(innerFetch, tracingConfig) {
+    const baseFetch = innerFetch ?? globalThis.fetch;
+    return async function (resource, options) {
+        if (!tracingConfig?.enabled || !tracingConfig.traceContextPropagation) {
+            return baseFetch(resource, options);
+        }
+        const tracingHeaders = createRequestHeaders();
+        if (Object.keys(tracingHeaders).length === 0) {
+            return baseFetch(resource, options);
+        }
+        // Merge tracing headers into the request
+        const fetchRequest = new Request(resource, options);
+        for (const [key, value] of Object.entries(tracingHeaders)) {
+            fetchRequest.headers.set(key, value);
+        }
+        return baseFetch(fetchRequest);
+    };
+}
+//# sourceMappingURL=tracingFetch.js.map

package/dist/esm/tracing/tracingFetch.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tracingFetch.js","sourceRoot":"","sources":["../../../src/tracing/tracingFetch.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAIlC,OAAO,EAAE,oBAAoB,EAAE,MAAM,oBAAoB,CAAC;AAI1D;;;;;;;GAOG;AACH,MAAM,UAAU,eAAe,CAC7B,UAAoB,EACpB,aAAqC;IAErC,MAAM,SAAS,GAAG,UAAU,IAAK,UAAU,CAAC,KAAiB,CAAC;IAC9D,OAAO,KAAK,WAAW,QAAQ,EAAE,OAAO;QACtC,IAAI,CAAC,aAAa,EAAE,OAAO,IAAI,CAAC,aAAa,CAAC,uBAAuB,EAAE,CAAC;YACtE,OAAO,SAAS,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QACtC,CAAC;QACD,MAAM,cAAc,GAAG,oBAAoB,EAAE,CAAC;QAC9C,IAAI,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC7C,OAAO,SAAS,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QACtC,CAAC;QAED,yCAAyC;QACzC,MAAM,YAAY,GAAG,IAAI,OAAO,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QACpD,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,EAAE,CAAC;YAC1D,YAAY,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;QACvC,CAAC;QACD,OAAO,SAAS,CAAC,YAAY,CAAC,CAAC;IACjC,CAAC,CAAC;AACJ,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport type OpenAI from \"openai\";\nimport type { ResolvedTracingConfig } from \"./configuration.js\";\nimport { createRequestHeaders } from \"./tracingClient.js\";\n\ntype FetchFn = NonNullable<NonNullable<ConstructorParameters<typeof OpenAI>[0]>[\"fetch\"]>;\n\n/**\n * Wraps a fetch function to inject OpenTelemetry trace context headers\n * (traceparent, tracestate) into outgoing requests when GenAI tracing is enabled.\n *\n * @param innerFetch - The underlying fetch function to wrap. If undefined, uses globalThis.fetch.\n * @param tracingConfig - The resolved tracing config for this client instance.\n * @returns A fetch-compatible function that injects tracing headers.\n */\nexport function getTracingFetch(\n  innerFetch?: FetchFn,\n  tracingConfig?: ResolvedTracingConfig,\n): FetchFn {\n  const baseFetch = innerFetch ?? (globalThis.fetch as FetchFn);\n  return async function (resource, options): Promise<Response> {\n    if (!tracingConfig?.enabled || !tracingConfig.traceContextPropagation) {\n      return baseFetch(resource, options);\n    }\n    const tracingHeaders = createRequestHeaders();\n    if (Object.keys(tracingHeaders).length === 0) {\n      return baseFetch(resource, options);\n    }\n\n    // Merge tracing headers into the request\n    const fetchRequest = new Request(resource, options);\n    for (const [key, value] of Object.entries(tracingHeaders)) {\n      fetchRequest.headers.set(key, value);\n    }\n    return baseFetch(fetchRequest);\n  };\n}\n"]}

package/dist/react-native/aiProjectClient.d.ts

@@ -1,6 +1,7 @@
 import OpenAI from "openai";
 import type { AIProjectClientOptionalParams } from "./api/index.js";
 import type { AgentsOperations } from "./classic/agents/index.js";
+import type { ToolboxesOperations } from "./classic/toolboxes/index.js";
 import type { BetaOperations } from "./classic/beta/index.js";
 import type { ConnectionsOperations } from "./classic/connections/index.js";
 import type { DatasetsOperations } from "./classic/datasets/index.js";
@@ -25,6 +26,7 @@ export type { AIProjectClientOptionalParams } from "./api/aiProjectContext.js";
  * @property {DatasetsOperations} datasets - The operation groups for datasets
  * @property {ConnectionsOperations} connections - The operation groups for connections
  * @property {AgentsOperations} agents - The operation groups for agents
+ * @property {ToolboxesOperations} toolboxes - The operation groups for toolboxes
  * @property {BetaOperations} beta - The operation groups for beta include beta features:
  * - Memory Stores
  * - Evaluators
@@ -32,9 +34,10 @@ export type { AIProjectClientOptionalParams } from "./api/aiProjectContext.js";
  * - Insights
  * - Schedules
  * - Red Teams
- * - Toolboxes
  * - agents
  * - skills
+ * - routines
+ * - models
  * @property {TelemetryOperations} telemetry - The operation groups for telemetry
  * @property {getEndpointUrl} getEndpointUrl - gets the endpoint of the client
  * @property {getOpenAIClient} getOpenAIClient - gets the OpenAI client with optional OpenAI client options
@@ -45,7 +48,10 @@ export declare class AIProjectClient {
     private _endpoint;
     private _credential;
     private _options;
+    private _tracingConfig;
     constructor(endpoint: string, credential: TokenCredential, options?: AIProjectClientOptionalParams);
+    /** The operation groups for toolboxes */
+    readonly toolboxes: ToolboxesOperations;
     /** The operation groups for indexes */
     readonly indexes: IndexesOperations;
     /** The operation groups for deployments */
@@ -65,9 +71,10 @@ export declare class AIProjectClient {
      * - Insights
      * - Schedules
      * - Red Teams
-     * - Toolboxes
      * - agents
      * - skills
+     * - routines
+     * - models
      */
     readonly beta: BetaOperations;
     /** The operation groups for telemetry */

package/dist/react-native/aiProjectClient.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"aiProjectClient.d.ts","sourceRoot":"","sources":["../../src/aiProjectClient.ts"],"names":[],"mappings":"AAIA,OAAO,MAAM,MAAM,QAAQ,CAAC;AAE5B,OAAO,KAAK,EAAoB,6BAA6B,EAAE,MAAM,gBAAgB,CAAC;AAEtF,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,2BAA2B,CAAC;AAElE,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AAE9D,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,gCAAgC,CAAC;AAE5E,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,6BAA6B,CAAC;AAEtE,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,gCAAgC,CAAC;AAE5E,OAAO,KAAK,EAAE,yBAAyB,EAAE,MAAM,oCAAoC,CAAC;AAEpF,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,4BAA4B,CAAC;AAEpE,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,8BAA8B,CAAC;AAExE,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAIxD,OAAO,KAAK,EAAE,iCAAiC,EAAE,MAAM,2BAA2B,CAAC;AAGnF,YAAY,EAAE,6BAA6B,EAAE,MAAM,2BAA2B,CAAC;AAE/E;;;;;;;;;;;;;;;;;;;;;;;;;;;GA2BG;AACH,qBAAa,eAAe;IAC1B,OAAO,CAAC,qBAAqB,CAAmB;IAChD,OAAO,CAAC,iBAAiB,CAAmB;IAC5C,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,WAAW,CAAkB;IACrC,OAAO,CAAC,QAAQ,CAAgC;gBAG9C,QAAQ,EAAE,MAAM,EAChB,UAAU,EAAE,eAAe,EAC3B,OAAO,GAAE,6BAAkC;IA8B7C,uCAAuC;IACvC,SAAgB,OAAO,EAAE,iBAAiB,CAAC;IAC3C,2CAA2C;IAC3C,SAAgB,WAAW,EAAE,qBAAqB,CAAC;IACnD,wCAAwC;IACxC,SAAgB,QAAQ,EAAE,kBAAkB,CAAC;IAC7C,2CAA2C;IAC3C,SAAgB,WAAW,EAAE,qBAAqB,CAAC;IACnD,+CAA+C;IAC/C,SAAgB,eAAe,EAAE,yBAAyB,CAAC;IAC3D,sCAAsC;IACtC,SAAgB,MAAM,EAAE,gBAAgB,CAAC;IACzC;;;;;;;;;;OAUG;IACH,SAAgB,IAAI,EAAE,cAAc,CAAC;IACrC,yCAAyC;IACzC,SAAgB,SAAS,EAAE,mBAAmB,CAAC;IAC/C;;;OAGG;IAEI,eAAe,CAAC,kBAAkB,CAAC,EAAE,iCAAiC,GAAG,MAAM;IAiEtF;;;OAGG;IACH,IAAW,QAAQ,IAAI,MAAM,CAE5B;CACF"}
+{"version":3,"file":"aiProjectClient.d.ts","sourceRoot":"","sources":["../../src/aiProjectClient.ts"],"names":[],"mappings":"AAIA,OAAO,MAAM,MAAM,QAAQ,CAAC;AAE5B,OAAO,KAAK,EAAoB,6BAA6B,EAAE,MAAM,gBAAgB,CAAC;AAEtF,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,2BAA2B,CAAC;AAElE,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,8BAA8B,CAAC;AAExE,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AAE9D,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,gCAAgC,CAAC;AAE5E,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,6BAA6B,CAAC;AAEtE,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,gCAAgC,CAAC;AAE5E,OAAO,KAAK,EAAE,yBAAyB,EAAE,MAAM,oCAAoC,CAAC;AAEpF,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,4BAA4B,CAAC;AAEpE,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,8BAA8B,CAAC;AAExE,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAIxD,OAAO,KAAK,EAAE,iCAAiC,EAAE,MAAM,2BAA2B,CAAC;AAMnF,YAAY,EAAE,6BAA6B,EAAE,MAAM,2BAA2B,CAAC;AAE/E;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA6BG;AACH,qBAAa,eAAe;IAC1B,OAAO,CAAC,qBAAqB,CAAmB;IAChD,OAAO,CAAC,iBAAiB,CAAmB;IAC5C,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,WAAW,CAAkB;IACrC,OAAO,CAAC,QAAQ,CAAgC;IAChD,OAAO,CAAC,cAAc,CAAwB;gBAG5C,QAAQ,EAAE,MAAM,EAChB,UAAU,EAAE,eAAe,EAC3B,OAAO,GAAE,6BAAkC;IAkC7C,yCAAyC;IACzC,SAAgB,SAAS,EAAE,mBAAmB,CAAC;IAC/C,uCAAuC;IACvC,SAAgB,OAAO,EAAE,iBAAiB,CAAC;IAC3C,2CAA2C;IAC3C,SAAgB,WAAW,EAAE,qBAAqB,CAAC;IACnD,wCAAwC;IACxC,SAAgB,QAAQ,EAAE,kBAAkB,CAAC;IAC7C,2CAA2C;IAC3C,SAAgB,WAAW,EAAE,qBAAqB,CAAC;IACnD,+CAA+C;IAC/C,SAAgB,eAAe,EAAE,yBAAyB,CAAC;IAC3D,sCAAsC;IACtC,SAAgB,MAAM,EAAE,gBAAgB,CAAC;IACzC;;;;;;;;;;;OAWG;IACH,SAAgB,IAAI,EAAE,cAAc,CAAC;IACrC,yCAAyC;IACzC,SAAgB,SAAS,EAAE,mBAAmB,CAAC;IAC/C;;;OAGG;IAEI,eAAe,CAAC,kBAAkB,CAAC,EAAE,iCAAiC,GAAG,MAAM;IA+DtF;;;OAGG;IACH,IAAW,QAAQ,IAAI,MAAM,CAE5B;CACF"}

package/dist/react-native/aiProjectClient.js

@@ -5,6 +5,7 @@ import OpenAI from "openai";
 import { getBearerTokenProvider } from "@azure/identity";
 import { createAIProject } from "./api/index.js";
 import { _getAgentsOperations } from "./classic/agents/index.js";
+import { _getToolboxesOperations } from "./classic/toolboxes/index.js";
 import { _getBetaOperations } from "./classic/beta/index.js";
 import { _getConnectionsOperations } from "./classic/connections/index.js";
 import { _getDatasetsOperations } from "./classic/datasets/index.js";
@@ -16,6 +17,8 @@ import { overwriteOpenAIClient } from "./overwriteOpenAIClient.js";
 import { getCustomFetch } from "./getCustomFetch.js";
 import { getOpenAIDefaultHeaders } from "./util.js";
 import { KnownApiVersions } from "./models/models.js";
+import { getTracingFetch } from "./tracing/tracingFetch.js";
+import { resolveTracingConfig } from "./tracing/configuration.js";
 /**
  * The main client for the AIProjectClient service. It provides access to the various operations available in the service.
  * @class AIProjectClient
@@ -30,6 +33,7 @@ import { KnownApiVersions } from "./models/models.js";
  * @property {DatasetsOperations} datasets - The operation groups for datasets
  * @property {ConnectionsOperations} connections - The operation groups for connections
  * @property {AgentsOperations} agents - The operation groups for agents
+ * @property {ToolboxesOperations} toolboxes - The operation groups for toolboxes
  * @property {BetaOperations} beta - The operation groups for beta include beta features:
  * - Memory Stores
  * - Evaluators
@@ -37,9 +41,10 @@ import { KnownApiVersions } from "./models/models.js";
  * - Insights
  * - Schedules
  * - Red Teams
- * - Toolboxes
  * - agents
  * - skills
+ * - routines
+ * - models
  * @property {TelemetryOperations} telemetry - The operation groups for telemetry
  * @property {getEndpointUrl} getEndpointUrl - gets the endpoint of the client
  * @property {getOpenAIClient} getOpenAIClient - gets the OpenAI client with optional OpenAI client options
@@ -50,12 +55,16 @@ export class AIProjectClient {
     _endpoint;
     _credential;
     _options;
+    _tracingConfig;
     constructor(endpoint, credential, options = {}) {
         this._endpoint = endpoint;
         this._credential = credential;
         this._options = options;
+        this._tracingConfig = resolveTracingConfig(options.tracingOptions);
         const prefixFromOptions = options?.userAgentOptions?.userAgentPrefix;
-        const userAgentPrefix = prefixFromOptions ? `${prefixFromOptions}` : "";
+        const userAgentPrefix = prefixFromOptions
+            ? `${prefixFromOptions} azsdk-js-client`
+            : `azsdk-js-client`;
         this._cognitiveScopeClient = createAIProject(endpoint, this._credential, {
             ...options,
             userAgentOptions: { userAgentPrefix },
@@ -68,15 +77,18 @@ export class AIProjectClient {
             ...options,
             userAgentOptions: { userAgentPrefix },
         });
+        this.toolboxes = _getToolboxesOperations(this._cognitiveScopeClient);
         this.indexes = _getIndexesOperations(this._azureScopeClient);
         this.deployments = _getDeploymentsOperations(this._azureScopeClient);
         this.datasets = _getDatasetsOperations(this._azureScopeClient, this._options);
         this.connections = _getConnectionsOperations(this._azureScopeClient);
         this.evaluationRules = _getEvaluationRulesOperations(this._azureScopeClient);
-        this.agents = _getAgentsOperations(this._azureScopeClient);
+        this.agents = _getAgentsOperations(this._azureScopeClient, this._tracingConfig);
         this.beta = _getBetaOperations(this._cognitiveScopeClient);
         this.telemetry = _getTelemetryOperations(this.connections);
     }
+    /** The operation groups for toolboxes */
+    toolboxes;
     /** The operation groups for indexes */
     indexes;
     /** The operation groups for deployments */
@@ -96,9 +108,10 @@ export class AIProjectClient {
      * - Insights
      * - Schedules
      * - Red Teams
-     * - Toolboxes
      * - agents
      * - skills
+     * - routines
+     * - models
      */
     beta;
     /** The operation groups for telemetry */
@@ -115,6 +128,8 @@ export class AIProjectClient {
         if (this._options.additionalPolicies?.find((policy) => policy.policy.name === "recording policy")) {
             customFetch = getCustomFetch(this._azureScopeClient.pipeline, this._options.httpClient);
         }
+        // Wrap fetch with tracing to inject traceparent/tracestate headers
+        customFetch = getTracingFetch(customFetch, this._tracingConfig);
         let baseURL;
         if (opts?.baseURL) {
             baseURL = opts.baseURL;
@@ -131,10 +146,6 @@ export class AIProjectClient {
             baseURL = `${this._endpoint}/openai/v1`;
         }
         const defaultHeaders = getOpenAIDefaultHeaders(opts?.defaultHeaders, this._options?.userAgentOptions?.userAgentPrefix);
-        // When targeting an agent endpoint, add the foundry-features header if not already set
-        if (azureConfig?.agentName && !defaultHeaders.has("foundry-features")) {
-            defaultHeaders.set("foundry-features", "HostedAgents=V1Preview,AgentEndpoints=V1Preview");
-        }
         // When targeting an agent endpoint, add api-version to defaultQuery if not already present
         const defaultQuery = {
             ...(opts?.defaultQuery || {}),
@@ -154,7 +165,7 @@ export class AIProjectClient {
             fetch: customFetch,
         };
         const openaiClient = new OpenAI(openAIOptions);
-        return overwriteOpenAIClient(openaiClient);
+        return overwriteOpenAIClient(openaiClient, this._endpoint, this._tracingConfig);
     }
     /**
      * gets the endpoint of the client

package/dist/react-native/aiProjectClient.js.map

@@ -1 +1 @@
-{"version":3,"file":"aiProjectClient.js","sourceRoot":"","sources":["../../src/aiProjectClient.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAClC,iCAAiC;AAEjC,OAAO,MAAM,MAAM,QAAQ,CAAC;AAC5B,OAAO,EAAE,sBAAsB,EAAE,MAAM,iBAAiB,CAAC;AAEzD,OAAO,EAAE,eAAe,EAAE,MAAM,gBAAgB,CAAC;AAEjD,OAAO,EAAE,oBAAoB,EAAE,MAAM,2BAA2B,CAAC;AAEjE,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAE7D,OAAO,EAAE,yBAAyB,EAAE,MAAM,gCAAgC,CAAC;AAE3E,OAAO,EAAE,sBAAsB,EAAE,MAAM,6BAA6B,CAAC;AAErE,OAAO,EAAE,yBAAyB,EAAE,MAAM,gCAAgC,CAAC;AAE3E,OAAO,EAAE,6BAA6B,EAAE,MAAM,oCAAoC,CAAC;AAEnF,OAAO,EAAE,qBAAqB,EAAE,MAAM,4BAA4B,CAAC;AAEnE,OAAO,EAAE,uBAAuB,EAAE,MAAM,8BAA8B,CAAC;AAEvE,OAAO,EAAE,qBAAqB,EAAE,MAAM,4BAA4B,CAAC;AACnE,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AACrD,OAAO,EAAE,uBAAuB,EAAE,MAAM,WAAW,CAAC;AAEpD,OAAO,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AAItD;;;;;;;;;;;;;;;;;;;;;;;;;;;GA2BG;AACH,MAAM,OAAO,eAAe;IAClB,qBAAqB,CAAmB;IACxC,iBAAiB,CAAmB;IACpC,SAAS,CAAS;IAClB,WAAW,CAAkB;IAC7B,QAAQ,CAAgC;IAEhD,YACE,QAAgB,EAChB,UAA2B,EAC3B,UAAyC,EAAE;QAE3C,IAAI,CAAC,SAAS,GAAG,QAAQ,CAAC;QAC1B,IAAI,CAAC,WAAW,GAAG,UAAU,CAAC;QAC9B,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC;QACxB,MAAM,iBAAiB,GAAG,OAAO,EAAE,gBAAgB,EAAE,eAAe,CAAC;QACrE,MAAM,eAAe,GAAG,iBAAiB,CAAC,CAAC,CAAC,GAAG,iBAAiB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACxE,IAAI,CAAC,qBAAqB,GAAG,eAAe,CAAC,QAAQ,EAAE,IAAI,CAAC,WAAW,EAAE;YACvE,GAAG,OAAO;YACV,gBAAgB,EAAE,EAAE,eAAe,EAAE;YACrC,WAAW,EAAE;gBACX,GAAG,OAAO,CAAC,WAAW;gBACtB,MAAM,EAAE,CAAC,+BAA+B,CAAC;aAC1C;SACF,CAAC,CAAC;QACH,IAAI,CAAC,iBAAiB,GAAG,eAAe,CAAC,QAAQ,EAAE,UAAU,EAAE;YAC7D,GAAG,OAAO;YACV,gBAAgB,EAAE,EAAE,eAAe,EAAE;SACtC,CAAC,CAAC;QAEH,IAAI,CAAC,OAAO,GAAG,qBAAqB,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAC7D,IAAI,CAAC,WAAW,GAAG,yBAAyB,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QACrE,IAAI,CAAC,QAAQ,GAAG,sBAAsB,CAAC,IAAI,CAAC,iBAAiB,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC9E,IAAI,CAAC,WAAW,GAAG,yBAAyB,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QACrE,IAAI,CAAC,eAAe,GAAG,6BAA6B,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAC7E,IAAI,CAAC,MAAM,GAAG,oBAAoB,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAC3D,IAAI,CAAC,IAAI,GAAG,kBAAkB,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QAC3D,IAAI,CAAC,SAAS,GAAG,uBAAuB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IAC7D,CAAC;IAED,uCAAuC;IACvB,OAAO,CAAoB;IAC3C,2CAA2C;IAC3B,WAAW,CAAwB;IACnD,wCAAwC;IACxB,QAAQ,CAAqB;IAC7C,2CAA2C;IAC3B,WAAW,CAAwB;IACnD,+CAA+C;IAC/B,eAAe,CAA4B;IAC3D,sCAAsC;IACtB,MAAM,CAAmB;IACzC;;;;;;;;;;OAUG;IACa,IAAI,CAAiB;IACrC,yCAAyC;IACzB,SAAS,CAAsB;IAC/C;;;OAGG;IACH,8DAA8D;IACvD,eAAe,CAAC,kBAAsD;QAC3E,MAAM,KAAK,GAAG,+BAA+B,CAAC;QAC9C,MAAM,EAAE,WAAW,EAAE,GAAG,IAAI,EAAE,GAAG,kBAAkB,IAAI,EAAE,CAAC;QAC1D,IAAI,WAA0E,CAAC;QAE/E,IACE,IAAI,CAAC,QAAQ,CAAC,kBAAkB,EAAE,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,KAAK,kBAAkB,CAAC,EAC7F,CAAC;YACD,WAAW,GAAG,cAAc,CAAC,IAAI,CAAC,iBAAiB,CAAC,QAAQ,EAAE,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;QAC1F,CAAC;QAED,IAAI,OAAe,CAAC;QACpB,IAAI,IAAI,EAAE,OAAO,EAAE,CAAC;YAClB,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC;QACzB,CAAC;aAAM,IAAI,WAAW,EAAE,SAAS,EAAE,CAAC;YAClC,IAAI,CAAC,WAAW,CAAC,YAAY,EAAE,CAAC;gBAC9B,MAAM,IAAI,KAAK,CACb,wFAAwF;oBACtF,qFAAqF;oBACrF,0EAA0E,CAC7E,CAAC;YACJ,CAAC;YACD,OAAO,GAAG,GAAG,IAAI,CAAC,SAAS,WAAW,WAAW,CAAC,SAAS,4BAA4B,CAAC;QAC1F,CAAC;aAAM,CAAC;YACN,OAAO,GAAG,GAAG,IAAI,CAAC,SAAS,YAAY,CAAC;QAC1C,CAAC;QAED,MAAM,cAAc,GAAG,uBAAuB,CAC5C,IAAI,EAAE,cAAc,EACpB,IAAI,CAAC,QAAQ,EAAE,gBAAgB,EAAE,eAAe,CACjD,CAAC;QAEF,uFAAuF;QACvF,IAAI,WAAW,EAAE,SAAS,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,kBAAkB,CAAC,EAAE,CAAC;YACtE,cAAc,CAAC,GAAG,CAAC,kBAAkB,EAAE,iDAAiD,CAAC,CAAC;QAC5F,CAAC;QAED,2FAA2F;QAC3F,MAAM,YAAY,GAAG;YACnB,GAAG,CAAC,IAAI,EAAE,YAAY,IAAI,EAAE,CAAC;SAC9B,CAAC;QACF,IAAI,WAAW,EAAE,SAAS,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,EAAE,CAAC;YAC3D,YAAY,CAAC,aAAa,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,UAAU,IAAI,gBAAgB,CAAC,EAAE,CAAC;QAChF,CAAC;QAED,2GAA2G;QAC3G,MAAM,EACJ,cAAc,EAAE,eAAe,EAC/B,OAAO,EAAE,eAAe,EACxB,YAAY,EAAE,aAAa,EAC3B,GAAG,QAAQ,EACZ,GAAG,IAAI,IAAI,EAAE,CAAC;QACf,MAAM,aAAa,GAA4C;YAC7D,GAAG,QAAQ;YACX,MAAM,EAAE,sBAAsB,CAAC,IAAI,CAAC,WAAW,EAAE,KAAK,CAAC;YACvD,OAAO;YACP,uBAAuB,EAAE,IAAI;YAC7B,cAAc,EAAE,cAAc,CAAC,MAAM,CAAC,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC;YAC7D,YAAY,EAAE,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,SAAS;YAC7E,KAAK,EAAE,WAAW;SACnB,CAAC;QAEF,MAAM,YAAY,GAAG,IAAI,MAAM,CAAC,aAAa,CAAC,CAAC;QAC/C,OAAO,qBAAqB,CAAC,YAAY,CAAC,CAAC;IAC7C,CAAC;IACD;;;OAGG;IACH,IAAW,QAAQ;QACjB,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;CACF","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n/* eslint-disable tsdoc/syntax */\n\nimport OpenAI from \"openai\";\nimport { getBearerTokenProvider } from \"@azure/identity\";\nimport type { AIProjectContext, AIProjectClientOptionalParams } from \"./api/index.js\";\nimport { createAIProject } from \"./api/index.js\";\nimport type { AgentsOperations } from \"./classic/agents/index.js\";\nimport { _getAgentsOperations } from \"./classic/agents/index.js\";\nimport type { BetaOperations } from \"./classic/beta/index.js\";\nimport { _getBetaOperations } from \"./classic/beta/index.js\";\nimport type { ConnectionsOperations } from \"./classic/connections/index.js\";\nimport { _getConnectionsOperations } from \"./classic/connections/index.js\";\nimport type { DatasetsOperations } from \"./classic/datasets/index.js\";\nimport { _getDatasetsOperations } from \"./classic/datasets/index.js\";\nimport type { DeploymentsOperations } from \"./classic/deployments/index.js\";\nimport { _getDeploymentsOperations } from \"./classic/deployments/index.js\";\nimport type { EvaluationRulesOperations } from \"./classic/evaluationRules/index.js\";\nimport { _getEvaluationRulesOperations } from \"./classic/evaluationRules/index.js\";\nimport type { IndexesOperations } from \"./classic/indexes/index.js\";\nimport { _getIndexesOperations } from \"./classic/indexes/index.js\";\nimport type { TelemetryOperations } from \"./classic/telemetry/index.js\";\nimport { _getTelemetryOperations } from \"./classic/telemetry/index.js\";\nimport type { TokenCredential } from \"@azure/core-auth\";\nimport { overwriteOpenAIClient } from \"./overwriteOpenAIClient.js\";\nimport { getCustomFetch } from \"./getCustomFetch.js\";\nimport { getOpenAIDefaultHeaders } from \"./util.js\";\nimport type { OpenAIClientOptionsWithAzureAgent } from \"./azureAgent.interface.js\";\nimport { KnownApiVersions } from \"./models/models.js\";\n\nexport type { AIProjectClientOptionalParams } from \"./api/aiProjectContext.js\";\n\n/**\n * The main client for the AIProjectClient service. It provides access to the various operations available in the service.\n * @class AIProjectClient\n * @extends {AIProjectContext}\n * @description The main client for the AIProjectClient\n * @constructor\n * @param {string} endpoint - The endpoint to use\n * @param {TokenCredential} credential - The credential to use\n * @param {AIProjectClientOptionalParams} [options] - Optional parameters for the client.\n * @property {DeploymentsOperations} deployments - The operation groups for deployments\n * @property {IndexesOperations} indexes - The operation groups for indexes\n * @property {DatasetsOperations} datasets - The operation groups for datasets\n * @property {ConnectionsOperations} connections - The operation groups for connections\n * @property {AgentsOperations} agents - The operation groups for agents\n * @property {BetaOperations} beta - The operation groups for beta include beta features:\n * - Memory Stores\n * - Evaluators\n * - Evaluation Taxonomies\n * - Insights\n * - Schedules\n * - Red Teams\n * - Toolboxes\n * - agents\n * - skills\n * @property {TelemetryOperations} telemetry - The operation groups for telemetry\n * @property {getEndpointUrl} getEndpointUrl - gets the endpoint of the client\n * @property {getOpenAIClient} getOpenAIClient - gets the OpenAI client with optional OpenAI client options\n */\nexport class AIProjectClient {\n  private _cognitiveScopeClient: AIProjectContext;\n  private _azureScopeClient: AIProjectContext;\n  private _endpoint: string;\n  private _credential: TokenCredential;\n  private _options: AIProjectClientOptionalParams;\n\n  constructor(\n    endpoint: string,\n    credential: TokenCredential,\n    options: AIProjectClientOptionalParams = {},\n  ) {\n    this._endpoint = endpoint;\n    this._credential = credential;\n    this._options = options;\n    const prefixFromOptions = options?.userAgentOptions?.userAgentPrefix;\n    const userAgentPrefix = prefixFromOptions ? `${prefixFromOptions}` : \"\";\n    this._cognitiveScopeClient = createAIProject(endpoint, this._credential, {\n      ...options,\n      userAgentOptions: { userAgentPrefix },\n      credentials: {\n        ...options.credentials,\n        scopes: [\"https://ai.azure.com/.default\"],\n      },\n    });\n    this._azureScopeClient = createAIProject(endpoint, credential, {\n      ...options,\n      userAgentOptions: { userAgentPrefix },\n    });\n\n    this.indexes = _getIndexesOperations(this._azureScopeClient);\n    this.deployments = _getDeploymentsOperations(this._azureScopeClient);\n    this.datasets = _getDatasetsOperations(this._azureScopeClient, this._options);\n    this.connections = _getConnectionsOperations(this._azureScopeClient);\n    this.evaluationRules = _getEvaluationRulesOperations(this._azureScopeClient);\n    this.agents = _getAgentsOperations(this._azureScopeClient);\n    this.beta = _getBetaOperations(this._cognitiveScopeClient);\n    this.telemetry = _getTelemetryOperations(this.connections);\n  }\n\n  /** The operation groups for indexes */\n  public readonly indexes: IndexesOperations;\n  /** The operation groups for deployments */\n  public readonly deployments: DeploymentsOperations;\n  /** The operation groups for datasets */\n  public readonly datasets: DatasetsOperations;\n  /** The operation groups for connections */\n  public readonly connections: ConnectionsOperations;\n  /** The operation groups for evaluationRules */\n  public readonly evaluationRules: EvaluationRulesOperations;\n  /** The operation groups for agents */\n  public readonly agents: AgentsOperations;\n  /** The operation groups for beta include beta features:\n   * - Memory Stores\n   * - Evaluators\n   * - Evaluation Taxonomies\n   * - Insights\n   * - Schedules\n   * - Red Teams\n   * - Toolboxes\n   * - agents\n   * - skills\n   */\n  public readonly beta: BetaOperations;\n  /** The operation groups for telemetry */\n  public readonly telemetry: TelemetryOperations;\n  /**\n   * gets the OpenAI client\n   * @returns the OpenAI client\n   */\n  // eslint-disable-next-line @azure/azure-sdk/ts-naming-options\n  public getOpenAIClient(optsWithAzureAgent?: OpenAIClientOptionsWithAzureAgent): OpenAI {\n    const scope = \"https://ai.azure.com/.default\";\n    const { azureConfig, ...opts } = optsWithAzureAgent || {};\n    let customFetch: NonNullable<ConstructorParameters<typeof OpenAI>[0]>[\"fetch\"];\n\n    if (\n      this._options.additionalPolicies?.find((policy) => policy.policy.name === \"recording policy\")\n    ) {\n      customFetch = getCustomFetch(this._azureScopeClient.pipeline, this._options.httpClient);\n    }\n\n    let baseURL: string;\n    if (opts?.baseURL) {\n      baseURL = opts.baseURL;\n    } else if (azureConfig?.agentName) {\n      if (!azureConfig.allowPreview) {\n        throw new Error(\n          \"Calling `getOpenAIClient` with an `agentName` requires you to set `allowPreview: true`\" +\n            \"\\nwhen providing azureConfig. Note that preview features are under development and \" +\n            \"\\nsubject to change. They should not be used in production environments.\",\n        );\n      }\n      baseURL = `${this._endpoint}/agents/${azureConfig.agentName}/endpoint/protocols/openai`;\n    } else {\n      baseURL = `${this._endpoint}/openai/v1`;\n    }\n\n    const defaultHeaders = getOpenAIDefaultHeaders(\n      opts?.defaultHeaders,\n      this._options?.userAgentOptions?.userAgentPrefix,\n    );\n\n    // When targeting an agent endpoint, add the foundry-features header if not already set\n    if (azureConfig?.agentName && !defaultHeaders.has(\"foundry-features\")) {\n      defaultHeaders.set(\"foundry-features\", \"HostedAgents=V1Preview,AgentEndpoints=V1Preview\");\n    }\n\n    // When targeting an agent endpoint, add api-version to defaultQuery if not already present\n    const defaultQuery = {\n      ...(opts?.defaultQuery || {}),\n    };\n    if (azureConfig?.agentName && !defaultQuery[\"api-version\"]) {\n      defaultQuery[\"api-version\"] = this._options.apiVersion ?? KnownApiVersions.v1;\n    }\n\n    // Destructure opts to exclude defaultHeaders, baseURL, and defaultQuery, then override specific properties\n    const {\n      defaultHeaders: _ignoredHeaders,\n      baseURL: _ignoredBaseURL,\n      defaultQuery: _ignoredQuery,\n      ...restOpts\n    } = opts || {};\n    const openAIOptions: ConstructorParameters<typeof OpenAI>[0] = {\n      ...restOpts,\n      apiKey: getBearerTokenProvider(this._credential, scope),\n      baseURL,\n      dangerouslyAllowBrowser: true,\n      defaultHeaders: defaultHeaders.toJSON({ preserveCase: true }),\n      defaultQuery: Object.keys(defaultQuery).length > 0 ? defaultQuery : undefined,\n      fetch: customFetch,\n    };\n\n    const openaiClient = new OpenAI(openAIOptions);\n    return overwriteOpenAIClient(openaiClient);\n  }\n  /**\n   * gets the endpoint of the client\n   * @returns the endpoint of the client\n   */\n  public get endpoint(): string {\n    return this._endpoint;\n  }\n}\n"]}
+{"version":3,"file":"aiProjectClient.js","sourceRoot":"","sources":["../../src/aiProjectClient.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAClC,iCAAiC;AAEjC,OAAO,MAAM,MAAM,QAAQ,CAAC;AAC5B,OAAO,EAAE,sBAAsB,EAAE,MAAM,iBAAiB,CAAC;AAEzD,OAAO,EAAE,eAAe,EAAE,MAAM,gBAAgB,CAAC;AAEjD,OAAO,EAAE,oBAAoB,EAAE,MAAM,2BAA2B,CAAC;AAEjE,OAAO,EAAE,uBAAuB,EAAE,MAAM,8BAA8B,CAAC;AAEvE,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAE7D,OAAO,EAAE,yBAAyB,EAAE,MAAM,gCAAgC,CAAC;AAE3E,OAAO,EAAE,sBAAsB,EAAE,MAAM,6BAA6B,CAAC;AAErE,OAAO,EAAE,yBAAyB,EAAE,MAAM,gCAAgC,CAAC;AAE3E,OAAO,EAAE,6BAA6B,EAAE,MAAM,oCAAoC,CAAC;AAEnF,OAAO,EAAE,qBAAqB,EAAE,MAAM,4BAA4B,CAAC;AAEnE,OAAO,EAAE,uBAAuB,EAAE,MAAM,8BAA8B,CAAC;AAEvE,OAAO,EAAE,qBAAqB,EAAE,MAAM,4BAA4B,CAAC;AACnE,OAAO,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;AAC1D,OAAO,EAAE,uBAAuB,EAAE,MAAM,WAAW,CAAC;AAEpD,OAAO,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AACtD,OAAO,EAAE,eAAe,EAAE,MAAM,2BAA2B,CAAC;AAC5D,OAAO,EAAE,oBAAoB,EAAE,MAAM,4BAA4B,CAAC;AAKlE;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA6BG;AACH,MAAM,OAAO,eAAe;IAClB,qBAAqB,CAAmB;IACxC,iBAAiB,CAAmB;IACpC,SAAS,CAAS;IAClB,WAAW,CAAkB;IAC7B,QAAQ,CAAgC;IACxC,cAAc,CAAwB;IAE9C,YACE,QAAgB,EAChB,UAA2B,EAC3B,UAAyC,EAAE;QAE3C,IAAI,CAAC,SAAS,GAAG,QAAQ,CAAC;QAC1B,IAAI,CAAC,WAAW,GAAG,UAAU,CAAC;QAC9B,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC;QACxB,IAAI,CAAC,cAAc,GAAG,oBAAoB,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;QACnE,MAAM,iBAAiB,GAAG,OAAO,EAAE,gBAAgB,EAAE,eAAe,CAAC;QACrE,MAAM,eAAe,GAAG,iBAAiB;YACvC,CAAC,CAAC,GAAG,iBAAiB,kBAAkB;YACxC,CAAC,CAAC,iBAAiB,CAAC;QACtB,IAAI,CAAC,qBAAqB,GAAG,eAAe,CAAC,QAAQ,EAAE,IAAI,CAAC,WAAW,EAAE;YACvE,GAAG,OAAO;YACV,gBAAgB,EAAE,EAAE,eAAe,EAAE;YACrC,WAAW,EAAE;gBACX,GAAG,OAAO,CAAC,WAAW;gBACtB,MAAM,EAAE,CAAC,+BAA+B,CAAC;aAC1C;SACF,CAAC,CAAC;QACH,IAAI,CAAC,iBAAiB,GAAG,eAAe,CAAC,QAAQ,EAAE,UAAU,EAAE;YAC7D,GAAG,OAAO;YACV,gBAAgB,EAAE,EAAE,eAAe,EAAE;SACtC,CAAC,CAAC;QAEH,IAAI,CAAC,SAAS,GAAG,uBAAuB,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QACrE,IAAI,CAAC,OAAO,GAAG,qBAAqB,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAC7D,IAAI,CAAC,WAAW,GAAG,yBAAyB,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QACrE,IAAI,CAAC,QAAQ,GAAG,sBAAsB,CAAC,IAAI,CAAC,iBAAiB,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC9E,IAAI,CAAC,WAAW,GAAG,yBAAyB,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QACrE,IAAI,CAAC,eAAe,GAAG,6BAA6B,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAC7E,IAAI,CAAC,MAAM,GAAG,oBAAoB,CAAC,IAAI,CAAC,iBAAiB,EAAE,IAAI,CAAC,cAAc,CAAC,CAAC;QAChF,IAAI,CAAC,IAAI,GAAG,kBAAkB,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QAC3D,IAAI,CAAC,SAAS,GAAG,uBAAuB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IAC7D,CAAC;IAED,yCAAyC;IACzB,SAAS,CAAsB;IAC/C,uCAAuC;IACvB,OAAO,CAAoB;IAC3C,2CAA2C;IAC3B,WAAW,CAAwB;IACnD,wCAAwC;IACxB,QAAQ,CAAqB;IAC7C,2CAA2C;IAC3B,WAAW,CAAwB;IACnD,+CAA+C;IAC/B,eAAe,CAA4B;IAC3D,sCAAsC;IACtB,MAAM,CAAmB;IACzC;;;;;;;;;;;OAWG;IACa,IAAI,CAAiB;IACrC,yCAAyC;IACzB,SAAS,CAAsB;IAC/C;;;OAGG;IACH,8DAA8D;IACvD,eAAe,CAAC,kBAAsD;QAC3E,MAAM,KAAK,GAAG,+BAA+B,CAAC;QAC9C,MAAM,EAAE,WAAW,EAAE,GAAG,IAAI,EAAE,GAAG,kBAAkB,IAAI,EAAE,CAAC;QAC1D,IAAI,WAA0E,CAAC;QAE/E,IACE,IAAI,CAAC,QAAQ,CAAC,kBAAkB,EAAE,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,KAAK,kBAAkB,CAAC,EAC7F,CAAC;YACD,WAAW,GAAG,cAAc,CAAC,IAAI,CAAC,iBAAiB,CAAC,QAAQ,EAAE,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;QAC1F,CAAC;QAED,mEAAmE;QACnE,WAAW,GAAG,eAAe,CAAC,WAAW,EAAE,IAAI,CAAC,cAAc,CAAC,CAAC;QAEhE,IAAI,OAAe,CAAC;QACpB,IAAI,IAAI,EAAE,OAAO,EAAE,CAAC;YAClB,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC;QACzB,CAAC;aAAM,IAAI,WAAW,EAAE,SAAS,EAAE,CAAC;YAClC,IAAI,CAAC,WAAW,CAAC,YAAY,EAAE,CAAC;gBAC9B,MAAM,IAAI,KAAK,CACb,wFAAwF;oBACtF,qFAAqF;oBACrF,0EAA0E,CAC7E,CAAC;YACJ,CAAC;YACD,OAAO,GAAG,GAAG,IAAI,CAAC,SAAS,WAAW,WAAW,CAAC,SAAS,4BAA4B,CAAC;QAC1F,CAAC;aAAM,CAAC;YACN,OAAO,GAAG,GAAG,IAAI,CAAC,SAAS,YAAY,CAAC;QAC1C,CAAC;QAED,MAAM,cAAc,GAAG,uBAAuB,CAC5C,IAAI,EAAE,cAAc,EACpB,IAAI,CAAC,QAAQ,EAAE,gBAAgB,EAAE,eAAe,CACjD,CAAC;QAEF,2FAA2F;QAC3F,MAAM,YAAY,GAAG;YACnB,GAAG,CAAC,IAAI,EAAE,YAAY,IAAI,EAAE,CAAC;SAC9B,CAAC;QACF,IAAI,WAAW,EAAE,SAAS,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,EAAE,CAAC;YAC3D,YAAY,CAAC,aAAa,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,UAAU,IAAI,gBAAgB,CAAC,EAAE,CAAC;QAChF,CAAC;QAED,2GAA2G;QAC3G,MAAM,EACJ,cAAc,EAAE,eAAe,EAC/B,OAAO,EAAE,eAAe,EACxB,YAAY,EAAE,aAAa,EAC3B,GAAG,QAAQ,EACZ,GAAG,IAAI,IAAI,EAAE,CAAC;QACf,MAAM,aAAa,GAA4C;YAC7D,GAAG,QAAQ;YACX,MAAM,EAAE,sBAAsB,CAAC,IAAI,CAAC,WAAW,EAAE,KAAK,CAAC;YACvD,OAAO;YACP,uBAAuB,EAAE,IAAI;YAC7B,cAAc,EAAE,cAAc,CAAC,MAAM,CAAC,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC;YAC7D,YAAY,EAAE,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,SAAS;YAC7E,KAAK,EAAE,WAAW;SACnB,CAAC;QAEF,MAAM,YAAY,GAAG,IAAI,MAAM,CAAC,aAAa,CAAC,CAAC;QAC/C,OAAO,qBAAqB,CAAC,YAAY,EAAE,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,cAAc,CAAC,CAAC;IAClF,CAAC;IACD;;;OAGG;IACH,IAAW,QAAQ;QACjB,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;CACF","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n/* eslint-disable tsdoc/syntax */\n\nimport OpenAI from \"openai\";\nimport { getBearerTokenProvider } from \"@azure/identity\";\nimport type { AIProjectContext, AIProjectClientOptionalParams } from \"./api/index.js\";\nimport { createAIProject } from \"./api/index.js\";\nimport type { AgentsOperations } from \"./classic/agents/index.js\";\nimport { _getAgentsOperations } from \"./classic/agents/index.js\";\nimport type { ToolboxesOperations } from \"./classic/toolboxes/index.js\";\nimport { _getToolboxesOperations } from \"./classic/toolboxes/index.js\";\nimport type { BetaOperations } from \"./classic/beta/index.js\";\nimport { _getBetaOperations } from \"./classic/beta/index.js\";\nimport type { ConnectionsOperations } from \"./classic/connections/index.js\";\nimport { _getConnectionsOperations } from \"./classic/connections/index.js\";\nimport type { DatasetsOperations } from \"./classic/datasets/index.js\";\nimport { _getDatasetsOperations } from \"./classic/datasets/index.js\";\nimport type { DeploymentsOperations } from \"./classic/deployments/index.js\";\nimport { _getDeploymentsOperations } from \"./classic/deployments/index.js\";\nimport type { EvaluationRulesOperations } from \"./classic/evaluationRules/index.js\";\nimport { _getEvaluationRulesOperations } from \"./classic/evaluationRules/index.js\";\nimport type { IndexesOperations } from \"./classic/indexes/index.js\";\nimport { _getIndexesOperations } from \"./classic/indexes/index.js\";\nimport type { TelemetryOperations } from \"./classic/telemetry/index.js\";\nimport { _getTelemetryOperations } from \"./classic/telemetry/index.js\";\nimport type { TokenCredential } from \"@azure/core-auth\";\nimport { overwriteOpenAIClient } from \"./overwriteOpenAIClient.js\";\nimport { getCustomFetch } from \"#platform/getCustomFetch\";\nimport { getOpenAIDefaultHeaders } from \"./util.js\";\nimport type { OpenAIClientOptionsWithAzureAgent } from \"./azureAgent.interface.js\";\nimport { KnownApiVersions } from \"./models/models.js\";\nimport { getTracingFetch } from \"./tracing/tracingFetch.js\";\nimport { resolveTracingConfig } from \"./tracing/configuration.js\";\nimport type { ResolvedTracingConfig } from \"./tracing/configuration.js\";\n\nexport type { AIProjectClientOptionalParams } from \"./api/aiProjectContext.js\";\n\n/**\n * The main client for the AIProjectClient service. It provides access to the various operations available in the service.\n * @class AIProjectClient\n * @extends {AIProjectContext}\n * @description The main client for the AIProjectClient\n * @constructor\n * @param {string} endpoint - The endpoint to use\n * @param {TokenCredential} credential - The credential to use\n * @param {AIProjectClientOptionalParams} [options] - Optional parameters for the client.\n * @property {DeploymentsOperations} deployments - The operation groups for deployments\n * @property {IndexesOperations} indexes - The operation groups for indexes\n * @property {DatasetsOperations} datasets - The operation groups for datasets\n * @property {ConnectionsOperations} connections - The operation groups for connections\n * @property {AgentsOperations} agents - The operation groups for agents\n * @property {ToolboxesOperations} toolboxes - The operation groups for toolboxes\n * @property {BetaOperations} beta - The operation groups for beta include beta features:\n * - Memory Stores\n * - Evaluators\n * - Evaluation Taxonomies\n * - Insights\n * - Schedules\n * - Red Teams\n * - agents\n * - skills\n * - routines\n * - models\n * @property {TelemetryOperations} telemetry - The operation groups for telemetry\n * @property {getEndpointUrl} getEndpointUrl - gets the endpoint of the client\n * @property {getOpenAIClient} getOpenAIClient - gets the OpenAI client with optional OpenAI client options\n */\nexport class AIProjectClient {\n  private _cognitiveScopeClient: AIProjectContext;\n  private _azureScopeClient: AIProjectContext;\n  private _endpoint: string;\n  private _credential: TokenCredential;\n  private _options: AIProjectClientOptionalParams;\n  private _tracingConfig: ResolvedTracingConfig;\n\n  constructor(\n    endpoint: string,\n    credential: TokenCredential,\n    options: AIProjectClientOptionalParams = {},\n  ) {\n    this._endpoint = endpoint;\n    this._credential = credential;\n    this._options = options;\n    this._tracingConfig = resolveTracingConfig(options.tracingOptions);\n    const prefixFromOptions = options?.userAgentOptions?.userAgentPrefix;\n    const userAgentPrefix = prefixFromOptions\n      ? `${prefixFromOptions} azsdk-js-client`\n      : `azsdk-js-client`;\n    this._cognitiveScopeClient = createAIProject(endpoint, this._credential, {\n      ...options,\n      userAgentOptions: { userAgentPrefix },\n      credentials: {\n        ...options.credentials,\n        scopes: [\"https://ai.azure.com/.default\"],\n      },\n    });\n    this._azureScopeClient = createAIProject(endpoint, credential, {\n      ...options,\n      userAgentOptions: { userAgentPrefix },\n    });\n\n    this.toolboxes = _getToolboxesOperations(this._cognitiveScopeClient);\n    this.indexes = _getIndexesOperations(this._azureScopeClient);\n    this.deployments = _getDeploymentsOperations(this._azureScopeClient);\n    this.datasets = _getDatasetsOperations(this._azureScopeClient, this._options);\n    this.connections = _getConnectionsOperations(this._azureScopeClient);\n    this.evaluationRules = _getEvaluationRulesOperations(this._azureScopeClient);\n    this.agents = _getAgentsOperations(this._azureScopeClient, this._tracingConfig);\n    this.beta = _getBetaOperations(this._cognitiveScopeClient);\n    this.telemetry = _getTelemetryOperations(this.connections);\n  }\n\n  /** The operation groups for toolboxes */\n  public readonly toolboxes: ToolboxesOperations;\n  /** The operation groups for indexes */\n  public readonly indexes: IndexesOperations;\n  /** The operation groups for deployments */\n  public readonly deployments: DeploymentsOperations;\n  /** The operation groups for datasets */\n  public readonly datasets: DatasetsOperations;\n  /** The operation groups for connections */\n  public readonly connections: ConnectionsOperations;\n  /** The operation groups for evaluationRules */\n  public readonly evaluationRules: EvaluationRulesOperations;\n  /** The operation groups for agents */\n  public readonly agents: AgentsOperations;\n  /** The operation groups for beta include beta features:\n   * - Memory Stores\n   * - Evaluators\n   * - Evaluation Taxonomies\n   * - Insights\n   * - Schedules\n   * - Red Teams\n   * - agents\n   * - skills\n   * - routines\n   * - models\n   */\n  public readonly beta: BetaOperations;\n  /** The operation groups for telemetry */\n  public readonly telemetry: TelemetryOperations;\n  /**\n   * gets the OpenAI client\n   * @returns the OpenAI client\n   */\n  // eslint-disable-next-line @azure/azure-sdk/ts-naming-options\n  public getOpenAIClient(optsWithAzureAgent?: OpenAIClientOptionsWithAzureAgent): OpenAI {\n    const scope = \"https://ai.azure.com/.default\";\n    const { azureConfig, ...opts } = optsWithAzureAgent || {};\n    let customFetch: NonNullable<ConstructorParameters<typeof OpenAI>[0]>[\"fetch\"];\n\n    if (\n      this._options.additionalPolicies?.find((policy) => policy.policy.name === \"recording policy\")\n    ) {\n      customFetch = getCustomFetch(this._azureScopeClient.pipeline, this._options.httpClient);\n    }\n\n    // Wrap fetch with tracing to inject traceparent/tracestate headers\n    customFetch = getTracingFetch(customFetch, this._tracingConfig);\n\n    let baseURL: string;\n    if (opts?.baseURL) {\n      baseURL = opts.baseURL;\n    } else if (azureConfig?.agentName) {\n      if (!azureConfig.allowPreview) {\n        throw new Error(\n          \"Calling `getOpenAIClient` with an `agentName` requires you to set `allowPreview: true`\" +\n            \"\\nwhen providing azureConfig. Note that preview features are under development and \" +\n            \"\\nsubject to change. They should not be used in production environments.\",\n        );\n      }\n      baseURL = `${this._endpoint}/agents/${azureConfig.agentName}/endpoint/protocols/openai`;\n    } else {\n      baseURL = `${this._endpoint}/openai/v1`;\n    }\n\n    const defaultHeaders = getOpenAIDefaultHeaders(\n      opts?.defaultHeaders,\n      this._options?.userAgentOptions?.userAgentPrefix,\n    );\n\n    // When targeting an agent endpoint, add api-version to defaultQuery if not already present\n    const defaultQuery = {\n      ...(opts?.defaultQuery || {}),\n    };\n    if (azureConfig?.agentName && !defaultQuery[\"api-version\"]) {\n      defaultQuery[\"api-version\"] = this._options.apiVersion ?? KnownApiVersions.v1;\n    }\n\n    // Destructure opts to exclude defaultHeaders, baseURL, and defaultQuery, then override specific properties\n    const {\n      defaultHeaders: _ignoredHeaders,\n      baseURL: _ignoredBaseURL,\n      defaultQuery: _ignoredQuery,\n      ...restOpts\n    } = opts || {};\n    const openAIOptions: ConstructorParameters<typeof OpenAI>[0] = {\n      ...restOpts,\n      apiKey: getBearerTokenProvider(this._credential, scope),\n      baseURL,\n      dangerouslyAllowBrowser: true,\n      defaultHeaders: defaultHeaders.toJSON({ preserveCase: true }),\n      defaultQuery: Object.keys(defaultQuery).length > 0 ? defaultQuery : undefined,\n      fetch: customFetch,\n    };\n\n    const openaiClient = new OpenAI(openAIOptions);\n    return overwriteOpenAIClient(openaiClient, this._endpoint, this._tracingConfig);\n  }\n  /**\n   * gets the endpoint of the client\n   * @returns the endpoint of the client\n   */\n  public get endpoint(): string {\n    return this._endpoint;\n  }\n}\n"]}

package/dist/react-native/api/agents/index.d.ts

@@ -1,3 +1,3 @@
-export { listVersions, deleteVersion, getVersion, createAgentVersionFromManifest, createVersion, list, $delete, updateAgentFromManifest, createAgentFromManifest, update, create, get, } from "./operations.js";
-export type { AgentsListVersionsOptionalParams, AgentsDeleteVersionOptionalParams, AgentsGetVersionOptionalParams, AgentsCreateAgentVersionFromManifestOptionalParams, AgentsCreateVersionOptionalParams, AgentsListOptionalParams, AgentsDeleteOptionalParams, AgentsUpdateAgentFromManifestOptionalParams, AgentsCreateAgentFromManifestOptionalParams, AgentsUpdateOptionalParams, AgentsCreateOptionalParams, AgentsGetOptionalParams, } from "./options.js";
+export { deleteSessionFile, listSessionFiles, downloadSessionFile, uploadSessionFile, getSessionLogStream, listSessions, stopSession, deleteSession, getSession, createSession, disable, enable, downloadAgentCode, createVersionFromCode, updateAgentObject, listVersions, deleteVersion, getVersion, createAgentVersionFromManifest, createVersion, list, $delete, updateAgentFromManifest, createAgentFromManifest, update, create, get, } from "./operations.js";
+export type { AgentsDeleteSessionFileOptionalParams, AgentsListSessionFilesOptionalParams, AgentsDownloadSessionFileOptionalParams, AgentsUploadSessionFileOptionalParams, AgentsGetSessionLogStreamOptionalParams, AgentsListSessionsOptionalParams, AgentsStopSessionOptionalParams, AgentsDeleteSessionOptionalParams, AgentsGetSessionOptionalParams, AgentsCreateSessionOptionalParams, AgentsDisableOptionalParams, AgentsEnableOptionalParams, AgentsDownloadAgentCodeOptionalParams, AgentsCreateVersionFromCodeOptionalParams, AgentsUpdateAgentObjectOptionalParams, AgentsListVersionsOptionalParams, AgentsDeleteVersionOptionalParams, AgentsGetVersionOptionalParams, AgentsCreateAgentVersionFromManifestOptionalParams, AgentsCreateVersionOptionalParams, AgentsListOptionalParams, AgentsDeleteOptionalParams, AgentsUpdateAgentFromManifestOptionalParams, AgentsCreateAgentFromManifestOptionalParams, AgentsUpdateOptionalParams, AgentsCreateOptionalParams, AgentsGetOptionalParams, } from "./options.js";
 //# sourceMappingURL=index.d.ts.map

package/dist/react-native/api/agents/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/api/agents/index.ts"],"names":[],"mappings":"AAGA,OAAO,EACL,YAAY,EACZ,aAAa,EACb,UAAU,EACV,8BAA8B,EAC9B,aAAa,EACb,IAAI,EACJ,OAAO,EACP,uBAAuB,EACvB,uBAAuB,EACvB,MAAM,EACN,MAAM,EACN,GAAG,GACJ,MAAM,iBAAiB,CAAC;AACzB,YAAY,EACV,gCAAgC,EAChC,iCAAiC,EACjC,8BAA8B,EAC9B,kDAAkD,EAClD,iCAAiC,EACjC,wBAAwB,EACxB,0BAA0B,EAC1B,2CAA2C,EAC3C,2CAA2C,EAC3C,0BAA0B,EAC1B,0BAA0B,EAC1B,uBAAuB,GACxB,MAAM,cAAc,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/api/agents/index.ts"],"names":[],"mappings":"AAGA,OAAO,EACL,iBAAiB,EACjB,gBAAgB,EAChB,mBAAmB,EACnB,iBAAiB,EACjB,mBAAmB,EACnB,YAAY,EACZ,WAAW,EACX,aAAa,EACb,UAAU,EACV,aAAa,EACb,OAAO,EACP,MAAM,EACN,iBAAiB,EACjB,qBAAqB,EACrB,iBAAiB,EACjB,YAAY,EACZ,aAAa,EACb,UAAU,EACV,8BAA8B,EAC9B,aAAa,EACb,IAAI,EACJ,OAAO,EACP,uBAAuB,EACvB,uBAAuB,EACvB,MAAM,EACN,MAAM,EACN,GAAG,GACJ,MAAM,iBAAiB,CAAC;AACzB,YAAY,EACV,qCAAqC,EACrC,oCAAoC,EACpC,uCAAuC,EACvC,qCAAqC,EACrC,uCAAuC,EACvC,gCAAgC,EAChC,+BAA+B,EAC/B,iCAAiC,EACjC,8BAA8B,EAC9B,iCAAiC,EACjC,2BAA2B,EAC3B,0BAA0B,EAC1B,qCAAqC,EACrC,yCAAyC,EACzC,qCAAqC,EACrC,gCAAgC,EAChC,iCAAiC,EACjC,8BAA8B,EAC9B,kDAAkD,EAClD,iCAAiC,EACjC,wBAAwB,EACxB,0BAA0B,EAC1B,2CAA2C,EAC3C,2CAA2C,EAC3C,0BAA0B,EAC1B,0BAA0B,EAC1B,uBAAuB,GACxB,MAAM,cAAc,CAAC"}

package/dist/react-native/api/agents/index.js

@@ -1,4 +1,4 @@
 // Copyright (c) Microsoft Corporation.
 // Licensed under the MIT License.
-export { listVersions, deleteVersion, getVersion, createAgentVersionFromManifest, createVersion, list, $delete, updateAgentFromManifest, createAgentFromManifest, update, create, get, } from "./operations.js";
+export { deleteSessionFile, listSessionFiles, downloadSessionFile, uploadSessionFile, getSessionLogStream, listSessions, stopSession, deleteSession, getSession, createSession, disable, enable, downloadAgentCode, createVersionFromCode, updateAgentObject, listVersions, deleteVersion, getVersion, createAgentVersionFromManifest, createVersion, list, $delete, updateAgentFromManifest, createAgentFromManifest, update, create, get, } from "./operations.js";
 //# sourceMappingURL=index.js.map

package/dist/react-native/api/agents/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/api/agents/index.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAElC,OAAO,EACL,YAAY,EACZ,aAAa,EACb,UAAU,EACV,8BAA8B,EAC9B,aAAa,EACb,IAAI,EACJ,OAAO,EACP,uBAAuB,EACvB,uBAAuB,EACvB,MAAM,EACN,MAAM,EACN,GAAG,GACJ,MAAM,iBAAiB,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nexport {\n  listVersions,\n  deleteVersion,\n  getVersion,\n  createAgentVersionFromManifest,\n  createVersion,\n  list,\n  $delete,\n  updateAgentFromManifest,\n  createAgentFromManifest,\n  update,\n  create,\n  get,\n} from \"./operations.js\";\nexport type {\n  AgentsListVersionsOptionalParams,\n  AgentsDeleteVersionOptionalParams,\n  AgentsGetVersionOptionalParams,\n  AgentsCreateAgentVersionFromManifestOptionalParams,\n  AgentsCreateVersionOptionalParams,\n  AgentsListOptionalParams,\n  AgentsDeleteOptionalParams,\n  AgentsUpdateAgentFromManifestOptionalParams,\n  AgentsCreateAgentFromManifestOptionalParams,\n  AgentsUpdateOptionalParams,\n  AgentsCreateOptionalParams,\n  AgentsGetOptionalParams,\n} from \"./options.js\";\n"]}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/api/agents/index.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAElC,OAAO,EACL,iBAAiB,EACjB,gBAAgB,EAChB,mBAAmB,EACnB,iBAAiB,EACjB,mBAAmB,EACnB,YAAY,EACZ,WAAW,EACX,aAAa,EACb,UAAU,EACV,aAAa,EACb,OAAO,EACP,MAAM,EACN,iBAAiB,EACjB,qBAAqB,EACrB,iBAAiB,EACjB,YAAY,EACZ,aAAa,EACb,UAAU,EACV,8BAA8B,EAC9B,aAAa,EACb,IAAI,EACJ,OAAO,EACP,uBAAuB,EACvB,uBAAuB,EACvB,MAAM,EACN,MAAM,EACN,GAAG,GACJ,MAAM,iBAAiB,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nexport {\n  deleteSessionFile,\n  listSessionFiles,\n  downloadSessionFile,\n  uploadSessionFile,\n  getSessionLogStream,\n  listSessions,\n  stopSession,\n  deleteSession,\n  getSession,\n  createSession,\n  disable,\n  enable,\n  downloadAgentCode,\n  createVersionFromCode,\n  updateAgentObject,\n  listVersions,\n  deleteVersion,\n  getVersion,\n  createAgentVersionFromManifest,\n  createVersion,\n  list,\n  $delete,\n  updateAgentFromManifest,\n  createAgentFromManifest,\n  update,\n  create,\n  get,\n} from \"./operations.js\";\nexport type {\n  AgentsDeleteSessionFileOptionalParams,\n  AgentsListSessionFilesOptionalParams,\n  AgentsDownloadSessionFileOptionalParams,\n  AgentsUploadSessionFileOptionalParams,\n  AgentsGetSessionLogStreamOptionalParams,\n  AgentsListSessionsOptionalParams,\n  AgentsStopSessionOptionalParams,\n  AgentsDeleteSessionOptionalParams,\n  AgentsGetSessionOptionalParams,\n  AgentsCreateSessionOptionalParams,\n  AgentsDisableOptionalParams,\n  AgentsEnableOptionalParams,\n  AgentsDownloadAgentCodeOptionalParams,\n  AgentsCreateVersionFromCodeOptionalParams,\n  AgentsUpdateAgentObjectOptionalParams,\n  AgentsListVersionsOptionalParams,\n  AgentsDeleteVersionOptionalParams,\n  AgentsGetVersionOptionalParams,\n  AgentsCreateAgentVersionFromManifestOptionalParams,\n  AgentsCreateVersionOptionalParams,\n  AgentsListOptionalParams,\n  AgentsDeleteOptionalParams,\n  AgentsUpdateAgentFromManifestOptionalParams,\n  AgentsCreateAgentFromManifestOptionalParams,\n  AgentsUpdateOptionalParams,\n  AgentsCreateOptionalParams,\n  AgentsGetOptionalParams,\n} from \"./options.js\";\n"]}