@azure/ai-projects 2.2.0 → 2.3.0

package/dist/commonjs/tracing/responseTracing.js

@@ -0,0 +1,267 @@
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var responseTracing_exports = {};
+__export(responseTracing_exports, {
+  traceConversationCreate: () => traceConversationCreate,
+  traceNonStreamingResponse: () => traceNonStreamingResponse,
+  traceStreamingResponse: () => traceStreamingResponse
+});
+module.exports = __toCommonJS(responseTracing_exports);
+var import_tracingClient = require("./tracingClient.js");
+var import_attributes = require("./attributes.js");
+var import_constants = require("./constants.js");
+var import_metrics = require("./metrics.js");
+async function traceNonStreamingResponse(responsesCreate, body, options, spanName, operationName, endpoint, agentName, contentRecording = false) {
+  const startTime = performance.now();
+  const { serverAddress, serverPort } = (0, import_attributes.parseEndpoint)(endpoint);
+  let errorType;
+  let responseModel;
+  const { span, ctx } = (0, import_tracingClient.startSpan)(spanName);
+  try {
+    (0, import_attributes.setCommonSpanAttributes)(
+      span,
+      operationName,
+      serverAddress,
+      serverPort,
+      body,
+      agentName,
+      contentRecording
+    );
+    const response = await (0, import_tracingClient.runInSpanContext)(
+      ctx,
+      () => responsesCreate(body, options)
+    );
+    (0, import_attributes.setResponseSpanAttributes)(span, response, contentRecording);
+    addWorkflowActionEvents(span, response, contentRecording);
+    responseModel = typeof response.model === "string" ? response.model : void 0;
+    const durationSeconds = (performance.now() - startTime) / 1e3;
+    (0, import_metrics.recordOperationDuration)(durationSeconds, {
+      operationName,
+      serverAddress,
+      serverPort,
+      responseModel
+    });
+    if (response.usage) {
+      (0, import_metrics.recordTokenUsage)(response.usage.input_tokens, response.usage.output_tokens, {
+        operationName,
+        serverAddress,
+        serverPort,
+        responseModel
+      });
+    }
+    span.setStatus({ code: 1 });
+    span.end();
+    return response;
+  } catch (error) {
+    (0, import_attributes.setErrorAttributes)(span, error);
+    errorType = error instanceof Error ? error.name || error.constructor?.name || "Error" : "Error";
+    const durationSeconds = (performance.now() - startTime) / 1e3;
+    (0, import_metrics.recordOperationDuration)(durationSeconds, {
+      operationName,
+      serverAddress,
+      serverPort,
+      errorType
+    });
+    span.setStatus({ code: 2, message: errorType });
+    span.end();
+    throw error;
+  }
+}
+async function traceStreamingResponse(responsesCreate, body, options, spanName, operationName, endpoint, agentName, contentRecording = false) {
+  const startTime = performance.now();
+  const { serverAddress, serverPort } = (0, import_attributes.parseEndpoint)(endpoint);
+  const { span, ctx } = (0, import_tracingClient.startSpan)(spanName);
+  (0, import_attributes.setCommonSpanAttributes)(
+    span,
+    operationName,
+    serverAddress,
+    serverPort,
+    body,
+    agentName,
+    contentRecording
+  );
+  try {
+    const stream = await (0, import_tracingClient.runInSpanContext)(
+      ctx,
+      () => responsesCreate(body, options)
+    );
+    return wrapStream(
+      stream,
+      span,
+      {
+        startTime,
+        operationName,
+        serverAddress,
+        serverPort
+      },
+      contentRecording
+    );
+  } catch (error) {
+    (0, import_attributes.setErrorAttributes)(span, error);
+    span.setStatus({ code: 2, message: error instanceof Error ? error.message : "Error" });
+    span.end();
+    const errorType = error instanceof Error ? error.name || error.constructor?.name || "Error" : "Error";
+    const durationSeconds = (performance.now() - startTime) / 1e3;
+    (0, import_metrics.recordOperationDuration)(durationSeconds, {
+      operationName,
+      serverAddress,
+      serverPort,
+      errorType
+    });
+    throw error;
+  }
+}
+function wrapStream(innerStream, span, metricsCtx, contentRecording = false) {
+  const iterator = innerStream[Symbol.asyncIterator]();
+  let completedResponse;
+  function recordStreamMetrics(errorType) {
+    const durationSeconds = (performance.now() - metricsCtx.startTime) / 1e3;
+    const responseModel = completedResponse && typeof completedResponse.model === "string" ? completedResponse.model : void 0;
+    (0, import_metrics.recordOperationDuration)(durationSeconds, {
+      operationName: metricsCtx.operationName,
+      serverAddress: metricsCtx.serverAddress,
+      serverPort: metricsCtx.serverPort,
+      responseModel,
+      errorType
+    });
+    if (completedResponse?.usage) {
+      (0, import_metrics.recordTokenUsage)(
+        completedResponse.usage.input_tokens,
+        completedResponse.usage.output_tokens,
+        {
+          operationName: metricsCtx.operationName,
+          serverAddress: metricsCtx.serverAddress,
+          serverPort: metricsCtx.serverPort,
+          responseModel
+        }
+      );
+    }
+  }
+  const wrappedIterator = {
+    async next() {
+      try {
+        const result = await iterator.next();
+        if (result.done) {
+          span.setStatus({ code: 1 });
+          span.end();
+          recordStreamMetrics();
+          return result;
+        }
+        const event = result.value;
+        if (event?.type === "response.output_item.done") {
+          const item = event.item;
+          if (item?.type === "workflow_action") {
+            addSingleWorkflowActionEvent(span, item, contentRecording);
+          }
+        }
+        if (event?.type === "response.completed" && event.response) {
+          completedResponse = event.response;
+          (0, import_attributes.setResponseSpanAttributes)(span, completedResponse, contentRecording);
+        }
+        return result;
+      } catch (error) {
+        (0, import_attributes.setErrorAttributes)(span, error);
+        span.setStatus({ code: 2, message: error instanceof Error ? error.message : "Error" });
+        span.end();
+        const errType = error instanceof Error ? error.name || error.constructor?.name || "Error" : "Error";
+        recordStreamMetrics(errType);
+        throw error;
+      }
+    },
+    async return(value) {
+      span.setStatus({ code: 1 });
+      span.end();
+      recordStreamMetrics();
+      if (iterator.return) {
+        return iterator.return(value);
+      }
+      return { done: true, value: void 0 };
+    },
+    async throw(error) {
+      (0, import_attributes.setErrorAttributes)(span, error);
+      span.setStatus({
+        code: 2,
+        message: error instanceof Error ? error.message : "Error"
+      });
+      span.end();
+      const errType = error instanceof Error ? error.name || error.constructor?.name || "Error" : "Error";
+      recordStreamMetrics(errType);
+      if (iterator.throw) {
+        return iterator.throw(error);
+      }
+      throw error;
+    }
+  };
+  const wrappedStream = Object.create(
+    Object.getPrototypeOf(innerStream),
+    Object.getOwnPropertyDescriptors(innerStream)
+  );
+  wrappedStream[Symbol.asyncIterator] = () => wrappedIterator;
+  return wrappedStream;
+}
+function addSingleWorkflowActionEvent(span, item, contentRecording = false) {
+  const workflowDetails = {};
+  if (item.status) workflowDetails.status = item.status;
+  if (contentRecording) {
+    if (item.action_id) workflowDetails.action_id = item.action_id;
+    if (item.previous_action_id) workflowDetails.previous_action_id = item.previous_action_id;
+  }
+  const contentArray = [
+    {
+      role: "workflow",
+      parts: [{ type: "workflow_action", content: workflowDetails }]
+    }
+  ];
+  span.addEvent(import_constants.GEN_AI_WORKFLOW_ACTION_EVENT, {
+    [import_constants.GEN_AI_PROVIDER_NAME]: import_constants.AGENTS_PROVIDER,
+    [import_constants.GEN_AI_EVENT_CONTENT]: JSON.stringify(contentArray)
+  });
+}
+function addWorkflowActionEvents(span, response, contentRecording = false) {
+  const output = response.output;
+  if (!Array.isArray(output)) return;
+  for (const item of output) {
+    if (item && typeof item === "object" && item.type === "workflow_action") {
+      addSingleWorkflowActionEvent(span, item, contentRecording);
+    }
+  }
+}
+async function traceConversationCreate(conversationsCreate, args, endpoint) {
+  const { span, ctx } = (0, import_tracingClient.startSpan)(import_constants.OperationName.CREATE_CONVERSATION);
+  try {
+    (0, import_attributes.setCommonAttributes)(span, import_constants.OperationName.CREATE_CONVERSATION, endpoint);
+    const result = await (0, import_tracingClient.runInSpanContext)(
+      ctx,
+      () => conversationsCreate(...args)
+    );
+    const conversation = result;
+    if (typeof conversation.id === "string") {
+      span.setAttribute(import_constants.GEN_AI_CONVERSATION_ID, conversation.id);
+    }
+    return result;
+  } finally {
+    span.end();
+  }
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  traceConversationCreate,
+  traceNonStreamingResponse,
+  traceStreamingResponse
+});
+//# sourceMappingURL=responseTracing.js.map

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

@@ -0,0 +1,7 @@
+{
+  "version": 3,
+  "sources": ["../../../src/tracing/responseTracing.ts"],
+  "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 \u2014 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"],
+  "mappings": ";;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAKA,2BAA4C;AAC5C,wBAMO;AACP,uBAOO;AACP,qBAA0D;AAS1D,eAAsB,0BACpB,iBACA,MACA,SACA,UACA,eACA,UACA,WACA,mBAA4B,OACV;AAClB,QAAM,YAAY,YAAY,IAAI;AAClC,QAAM,EAAE,eAAe,WAAW,QAAI,iCAAc,QAAQ;AAC5D,MAAI;AACJ,MAAI;AAEJ,QAAM,EAAE,MAAM,IAAI,QAAI,gCAAU,QAAQ;AACxC,MAAI;AACF;AAAA,MACE;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF;AACA,UAAM,WAAY,UAAM;AAAA,MAAiB;AAAA,MAAK,MAC5C,gBAAgB,MAAM,OAAO;AAAA,IAC/B;AACA,qDAA0B,MAAM,UAAU,gBAAgB;AAC1D,4BAAwB,MAAM,UAAU,gBAAgB;AACxD,oBAAgB,OAAO,SAAS,UAAU,WAAW,SAAS,QAAQ;AAGtE,UAAM,mBAAmB,YAAY,IAAI,IAAI,aAAa;AAC1D,gDAAwB,iBAAiB;AAAA,MACvC;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF,CAAC;AACD,QAAI,SAAS,OAAO;AAClB,2CAAiB,SAAS,MAAM,cAAc,SAAS,MAAM,eAAe;AAAA,QAC1E;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,MACF,CAAC;AAAA,IACH;AAEA,SAAK,UAAU,EAAE,MAAM,EAAE,CAAC;AAC1B,SAAK,IAAI;AACT,WAAO;AAAA,EACT,SAAS,OAAO;AACd,8CAAmB,MAAM,KAAK;AAC9B,gBAAY,iBAAiB,QAAQ,MAAM,QAAQ,MAAM,aAAa,QAAQ,UAAU;AACxF,UAAM,mBAAmB,YAAY,IAAI,IAAI,aAAa;AAC1D,gDAAwB,iBAAiB;AAAA,MACvC;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF,CAAC;AACD,SAAK,UAAU,EAAE,MAAM,GAAG,SAAS,UAAU,CAAC;AAC9C,SAAK,IAAI;AACT,UAAM;AAAA,EACR;AACF;AAEA,eAAsB,uBACpB,iBACA,MACA,SACA,UACA,eACA,UACA,WACA,mBAA4B,OACV;AAClB,QAAM,YAAY,YAAY,IAAI;AAClC,QAAM,EAAE,eAAe,WAAW,QAAI,iCAAc,QAAQ;AAC5D,QAAM,EAAE,MAAM,IAAI,QAAI,gCAAU,QAAQ;AACxC;AAAA,IACE;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AAEA,MAAI;AACF,UAAM,SAAS,UAAM;AAAA,MACnB;AAAA,MACA,MAAM,gBAAgB,MAAM,OAAO;AAAA,IACrC;AACA,WAAO;AAAA,MACL;AAAA,MACA;AAAA,MACA;AAAA,QACE;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,MACF;AAAA,MACA;AAAA,IACF;AAAA,EACF,SAAS,OAAO;AACd,8CAAmB,MAAM,KAAK;AAC9B,SAAK,UAAU,EAAE,MAAM,GAAG,SAAS,iBAAiB,QAAQ,MAAM,UAAU,QAAQ,CAAC;AACrF,SAAK,IAAI;AACT,UAAM,YACJ,iBAAiB,QAAQ,MAAM,QAAQ,MAAM,aAAa,QAAQ,UAAU;AAC9E,UAAM,mBAAmB,YAAY,IAAI,IAAI,aAAa;AAC1D,gDAAwB,iBAAiB;AAAA,MACvC;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF,CAAC;AACD,UAAM;AAAA,EACR;AACF;AAEA,SAAS,WACP,aACA,MACA,YACA,mBAA4B,OACJ;AACxB,QAAM,WAAW,YAAY,OAAO,aAAa,EAAE;AACnD,MAAI;AAEJ,WAAS,oBAAoB,WAA0B;AACrD,UAAM,mBAAmB,YAAY,IAAI,IAAI,WAAW,aAAa;AACrE,UAAM,gBACJ,qBAAqB,OAAO,kBAAkB,UAAU,WACpD,kBAAkB,QAClB;AACN,gDAAwB,iBAAiB;AAAA,MACvC,eAAe,WAAW;AAAA,MAC1B,eAAe,WAAW;AAAA,MAC1B,YAAY,WAAW;AAAA,MACvB;AAAA,MACA;AAAA,IACF,CAAC;AACD,QAAI,mBAAmB,OAAO;AAC5B;AAAA,QACE,kBAAkB,MAAM;AAAA,QACxB,kBAAkB,MAAM;AAAA,QACxB;AAAA,UACE,eAAe,WAAW;AAAA,UAC1B,eAAe,WAAW;AAAA,UAC1B,YAAY,WAAW;AAAA,UACvB;AAAA,QACF;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAEA,QAAM,kBAA0C;AAAA,IAC9C,MAAM,OAAO;AACX,UAAI;AACF,cAAM,SAAS,MAAM,SAAS,KAAK;AACnC,YAAI,OAAO,MAAM;AACf,eAAK,UAAU,EAAE,MAAM,EAAE,CAAC;AAC1B,eAAK,IAAI;AACT,8BAAoB;AACpB,iBAAO;AAAA,QACT;AAEA,cAAM,QAAQ,OAAO;AACrB,YAAI,OAAO,SAAS,6BAA6B;AAC/C,gBAAM,OAAQ,MAA6C;AAC3D,cAAI,MAAM,SAAS,mBAAmB;AACpC,yCAA6B,MAAM,MAAM,gBAAgB;AAAA,UAC3D;AAAA,QACF;AACA,YAAI,OAAO,SAAS,wBAAwB,MAAM,UAAU;AAC1D,8BAAoB,MAAM;AAC1B,2DAA0B,MAAM,mBAAmB,gBAAgB;AAAA,QACrE;AACA,eAAO;AAAA,MACT,SAAS,OAAO;AACd,kDAAmB,MAAM,KAAK;AAC9B,aAAK,UAAU,EAAE,MAAM,GAAG,SAAS,iBAAiB,QAAQ,MAAM,UAAU,QAAQ,CAAC;AACrF,aAAK,IAAI;AACT,cAAM,UACJ,iBAAiB,QAAQ,MAAM,QAAQ,MAAM,aAAa,QAAQ,UAAU;AAC9E,4BAAoB,OAAO;AAC3B,cAAM;AAAA,MACR;AAAA,IACF;AAAA,IACA,MAAM,OAAO,OAAiB;AAC5B,WAAK,UAAU,EAAE,MAAM,EAAE,CAAC;AAC1B,WAAK,IAAI;AACT,0BAAoB;AACpB,UAAI,SAAS,QAAQ;AACnB,eAAO,SAAS,OAAO,KAAK;AAAA,MAC9B;AACA,aAAO,EAAE,MAAM,MAAM,OAAO,OAAU;AAAA,IACxC;AAAA,IACA,MAAM,MAAM,OAAiB;AAC3B,gDAAmB,MAAM,KAAK;AAC9B,WAAK,UAAU;AAAA,QACb,MAAM;AAAA,QACN,SAAS,iBAAiB,QAAS,MAAgB,UAAU;AAAA,MAC/D,CAAC;AACD,WAAK,IAAI;AACT,YAAM,UACJ,iBAAiB,QAAQ,MAAM,QAAQ,MAAM,aAAa,QAAQ,UAAU;AAC9E,0BAAoB,OAAO;AAC3B,UAAI,SAAS,OAAO;AAClB,eAAO,SAAS,MAAM,KAAK;AAAA,MAC7B;AACA,YAAM;AAAA,IACR;AAAA,EACF;AAGA,QAAM,gBAAgB,OAAO;AAAA,IAC3B,OAAO,eAAe,WAAW;AAAA,IACjC,OAAO,0BAA0B,WAAW;AAAA,EAC9C;AACA,EAAC,cAA8D,OAAO,aAAa,IAAI,MACrF;AAEF,SAAO;AACT;AAKA,SAAS,6BACP,MACA,MACA,mBAA4B,OACtB;AACN,QAAM,kBAA2C,CAAC;AAClD,MAAI,KAAK,OAAQ,iBAAgB,SAAS,KAAK;AAC/C,MAAI,kBAAkB;AACpB,QAAI,KAAK,UAAW,iBAAgB,YAAY,KAAK;AACrD,QAAI,KAAK,mBAAoB,iBAAgB,qBAAqB,KAAK;AAAA,EACzE;AAEA,QAAM,eAAe;AAAA,IACnB;AAAA,MACE,MAAM;AAAA,MACN,OAAO,CAAC,EAAE,MAAM,mBAAmB,SAAS,gBAAgB,CAAC;AAAA,IAC/D;AAAA,EACF;AAEA,OAAK,SAAS,+CAA8B;AAAA,IAC1C,CAAC,qCAAoB,GAAG;AAAA,IACxB,CAAC,qCAAoB,GAAG,KAAK,UAAU,YAAY;AAAA,EACrD,CAAC;AACH;AAKA,SAAS,wBACP,MACA,UACA,mBAA4B,OACtB;AACN,QAAM,SAAU,SAAgD;AAChE,MAAI,CAAC,MAAM,QAAQ,MAAM,EAAG;AAC5B,aAAW,QAAQ,QAAQ;AACzB,QACE,QACA,OAAO,SAAS,YACf,KAAiC,SAAS,mBAC3C;AACA,mCAA6B,MAAM,MAAiC,gBAAgB;AAAA,IACtF;AAAA,EACF;AACF;AAEA,eAAsB,wBACpB,qBACA,MACA,UACkB;AAClB,QAAM,EAAE,MAAM,IAAI,QAAI,gCAAU,+BAAc,mBAAmB;AACjE,MAAI;AACF,+CAAoB,MAAM,+BAAc,qBAAqB,QAAQ;AACrE,UAAM,SAAS,UAAM;AAAA,MAAiB;AAAA,MAAK,MACxC,oBAA8D,GAAG,IAAI;AAAA,IACxE;AACA,UAAM,eAAe;AACrB,QAAI,OAAO,aAAa,OAAO,UAAU;AACvC,WAAK,aAAa,yCAAwB,aAAa,EAAE;AAAA,IAC3D;AACA,WAAO;AAAA,EACT,UAAE;AACA,SAAK,IAAI;AAAA,EACX;AACF;",
+  "names": []
+}

package/dist/commonjs/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/commonjs/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/commonjs/tracing/tracingClient.js

@@ -0,0 +1,51 @@
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var tracingClient_exports = {};
+__export(tracingClient_exports, {
+  SpanStatusCode: () => import_api2.SpanStatusCode,
+  createRequestHeaders: () => createRequestHeaders,
+  runInSpanContext: () => runInSpanContext,
+  startSpan: () => startSpan
+});
+module.exports = __toCommonJS(tracingClient_exports);
+var import_api = require("@opentelemetry/api");
+var import_constants = require("../constants.js");
+var import_api2 = require("@opentelemetry/api");
+function startSpan(name) {
+  const tracer = import_api.trace.getTracer(import_constants.PACKAGE_NAME, import_constants.SDK_VERSION);
+  const ctx = import_api.context.active();
+  const span = tracer.startSpan(name, { kind: import_api.SpanKind.CLIENT }, ctx);
+  const spanCtx = import_api.trace.setSpan(ctx, span);
+  return { span, ctx: spanCtx };
+}
+function createRequestHeaders() {
+  const headers = {};
+  import_api.propagation.inject(import_api.context.active(), headers, import_api.defaultTextMapSetter);
+  return headers;
+}
+function runInSpanContext(ctx, fn) {
+  return import_api.context.with(ctx, fn);
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  SpanStatusCode,
+  createRequestHeaders,
+  runInSpanContext,
+  startSpan
+});
+//# sourceMappingURL=tracingClient.js.map

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

@@ -0,0 +1,7 @@
+{
+  "version": 3,
+  "sources": ["../../../src/tracing/tracingClient.ts"],
+  "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"],
+  "mappings": ";;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AASA,iBAA4E;AAE5E,uBAA0C;AAG1C,IAAAA,cAA+B;AAUxB,SAAS,UAAU,MAA+B;AACvD,QAAM,SAAS,iBAAM,UAAU,+BAAc,4BAAW;AACxD,QAAM,MAAM,mBAAQ,OAAO;AAC3B,QAAM,OAAO,OAAO,UAAU,MAAM,EAAE,MAAM,oBAAS,OAAO,GAAG,GAAG;AAClE,QAAM,UAAU,iBAAM,QAAQ,KAAK,IAAI;AACvC,SAAO,EAAE,MAAM,KAAK,QAAQ;AAC9B;AAKO,SAAS,uBAA+C;AAC7D,QAAM,UAAkC,CAAC;AACzC,yBAAY,OAAO,mBAAQ,OAAO,GAAG,SAAS,+BAAoB;AAClE,SAAO;AACT;AAOO,SAAS,iBAAoB,KAAc,IAAgB;AAChE,SAAO,mBAAQ,KAAK,KAAK,EAAE;AAC7B;",
+  "names": ["import_api"]
+}

package/dist/commonjs/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/commonjs/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/commonjs/tracing/tracingFetch.js

@@ -0,0 +1,45 @@
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var tracingFetch_exports = {};
+__export(tracingFetch_exports, {
+  getTracingFetch: () => getTracingFetch
+});
+module.exports = __toCommonJS(tracingFetch_exports);
+var import_tracingClient = require("./tracingClient.js");
+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 = (0, import_tracingClient.createRequestHeaders)();
+    if (Object.keys(tracingHeaders).length === 0) {
+      return baseFetch(resource, options);
+    }
+    const fetchRequest = new Request(resource, options);
+    for (const [key, value] of Object.entries(tracingHeaders)) {
+      fetchRequest.headers.set(key, value);
+    }
+    return baseFetch(fetchRequest);
+  };
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  getTracingFetch
+});
+//# sourceMappingURL=tracingFetch.js.map

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

@@ -0,0 +1,7 @@
+{
+  "version": 3,
+  "sources": ["../../../src/tracing/tracingFetch.ts"],
+  "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"],
+  "mappings": ";;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAKA,2BAAqC;AAY9B,SAAS,gBACd,YACA,eACS;AACT,QAAM,YAAY,cAAe,WAAW;AAC5C,SAAO,eAAgB,UAAU,SAA4B;AAC3D,QAAI,CAAC,eAAe,WAAW,CAAC,cAAc,yBAAyB;AACrE,aAAO,UAAU,UAAU,OAAO;AAAA,IACpC;AACA,UAAM,qBAAiB,2CAAqB;AAC5C,QAAI,OAAO,KAAK,cAAc,EAAE,WAAW,GAAG;AAC5C,aAAO,UAAU,UAAU,OAAO;AAAA,IACpC;AAGA,UAAM,eAAe,IAAI,QAAQ,UAAU,OAAO;AAClD,eAAW,CAAC,KAAK,KAAK,KAAK,OAAO,QAAQ,cAAc,GAAG;AACzD,mBAAa,QAAQ,IAAI,KAAK,KAAK;AAAA,IACrC;AACA,WAAO,UAAU,YAAY;AAAA,EAC/B;AACF;",
+  "names": []
+}

package/dist/commonjs/tsdoc-metadata.json

@@ -5,7 +5,7 @@
   "toolPackages": [
     {
       "packageName": "@microsoft/api-extractor",
-      "packageVersion": "7.58.7"
+      "packageVersion": "7.58.9"
     }
   ]
 }

package/dist/esm/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,7 +34,6 @@ export type { AIProjectClientOptionalParams } from "./api/aiProjectContext.js";
  * - Insights
  * - Schedules
  * - Red Teams
- * - Toolboxes
  * - agents
  * - skills
  * - routines
@@ -47,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 */
@@ -67,7 +71,6 @@ export declare class AIProjectClient {
      * - Insights
      * - Schedules
      * - Red Teams
-     * - Toolboxes
      * - agents
      * - skills
      * - routines

package/dist/esm/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;;;;;;;;;;;;;;;;;;;;;;;;;;;;;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;gBAG9C,QAAQ,EAAE,MAAM,EAChB,UAAU,EAAE,eAAe,EAC3B,OAAO,GAAE,6BAAkC;IAgC7C,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;;;;;;;;;;;;OAYG;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/esm/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,7 +41,6 @@ import { KnownApiVersions } from "./models/models.js";
  * - Insights
  * - Schedules
  * - Red Teams
- * - Toolboxes
  * - agents
  * - skills
  * - routines
@@ -52,10 +55,12 @@ 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} azsdk-js-client`
@@ -72,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 */
@@ -100,7 +108,6 @@ export class AIProjectClient {
      * - Insights
      * - Schedules
      * - Red Teams
-     * - Toolboxes
      * - agents
      * - skills
      * - routines
@@ -121,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;
@@ -137,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 || {}),
@@ -160,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