@azure/ai-projects 1.0.0-beta.1

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

@@ -0,0 +1,101 @@
+import type { MessageDeltaChunk, RunStepDeltaChunk } from "../customization/streamingModels.js";
+import type { AgentThreadOutput, RunStepOutput, ThreadMessageOutput, ThreadRunOutput } from "../customization/outputModels.js";
+/**
+Each event in a server-sent events stream has an `event` and `data` property:
+  
+  ```
+  event: thread.created
+  data: {"id": "thread_123", "object": "thread", ...}
+  ```
+  
+  We emit events whenever a new object is created, transitions to a new state, or is being
+  streamed in parts (deltas). For example, we emit `thread.run.created` when a new run
+  is created, `thread.run.completed` when a run completes, and so on. When an Agent chooses
+  to create a message during a run, we emit a `thread.message.created event`, a
+  `thread.message.in_progress` event, many `thread.message.delta` events, and finally a
+  `thread.message.completed` event.
+  
+  We may add additional events over time, so we recommend handling unknown events gracefully
+  in your code.**/
+export interface AgentEventMessage {
+    /** The data of the event. The data can be of type AgentThreadOutput, ThreadRunOutput, RunStepOutput, ThreadMessageOutput, MessageDeltaChunk,RunStepDeltaChunk  */
+    data: AgentEventStreamDataOutput;
+    /** The type of the event. */
+    event: AgentStreamEventType | string;
+}
+/** Represents a stream event data in the agent. */
+export type AgentEventStreamDataOutput = AgentThreadOutput | ThreadRunOutput | RunStepOutput | ThreadMessageOutput | MessageDeltaChunk | RunStepDeltaChunk | string;
+/** Thread operation related streaming events */
+export declare enum ThreadStreamEvent {
+    /** Event sent when a new thread is created. The data of this event is of type AgentThread */
+    ThreadCreated = "thread.created"
+}
+/** Run operation related streaming events */
+export declare enum RunStreamEvent {
+    /** Event sent when a new run is created. The data of this event is of type ThreadRun */
+    ThreadRunCreated = "thread.run.created",
+    /** Event sent when a run moves to `queued` status. The data of this event is of type ThreadRun */
+    ThreadRunQueued = "thread.run.queued",
+    /** Event sent when a run moves to `in_progress` status. The data of this event is of type ThreadRun */
+    ThreadRunInProgress = "thread.run.in_progress",
+    /** Event sent when a run moves to `requires_action` status. The data of this event is of type ThreadRun */
+    ThreadRunRequiresAction = "thread.run.requires_action",
+    /** Event sent when a run is completed. The data of this event is of type ThreadRun */
+    ThreadRunCompleted = "thread.run.completed",
+    /** Event sent when a run fails. The data of this event is of type ThreadRun */
+    ThreadRunFailed = "thread.run.failed",
+    /** Event sent when a run moves to `cancelling` status. The data of this event is of type ThreadRun */
+    ThreadRunCancelling = "thread.run.cancelling",
+    /** Event sent when a run is cancelled. The data of this event is of type ThreadRun */
+    ThreadRunCancelled = "thread.run.cancelled",
+    /** Event sent when a run is expired. The data of this event is of type ThreadRun */
+    ThreadRunExpired = "thread.run.expired"
+}
+/** Run step operation related streaming events */
+export declare enum RunStepStreamEvent {
+    /** Event sent when a new thread run step is created. The data of this event is of type RunStep */
+    ThreadRunStepCreated = "thread.run.step.created",
+    /** Event sent when a run step moves to `in_progress` status. The data of this event is of type RunStep */
+    ThreadRunStepInProgress = "thread.run.step.in_progress",
+    /** Event sent when a run step is being streamed. The data of this event is of type RunStepDeltaChunk */
+    ThreadRunStepDelta = "thread.run.step.delta",
+    /** Event sent when a run step is completed. The data of this event is of type RunStep */
+    ThreadRunStepCompleted = "thread.run.step.completed",
+    /** Event sent when a run step fails. The data of this event is of type RunStep */
+    ThreadRunStepFailed = "thread.run.step.failed",
+    /** Event sent when a run step is cancelled. The data of this event is of type RunStep */
+    ThreadRunStepCancelled = "thread.run.step.cancelled",
+    /** Event sent when a run step is expired. The data of this event is of type RunStep */
+    ThreadRunStepExpired = "thread.run.step.expired"
+}
+/** Message operation related streaming events */
+export declare enum MessageStreamEvent {
+    /** Event sent when a new message is created. The data of this event is of type ThreadMessage */
+    ThreadMessageCreated = "thread.message.created",
+    /** Event sent when a message moves to `in_progress` status. The data of this event is of type ThreadMessage */
+    ThreadMessageInProgress = "thread.message.in_progress",
+    /** Event sent when a message is being streamed. The data of this event is of type MessageDeltaChunk */
+    ThreadMessageDelta = "thread.message.delta",
+    /** Event sent when a message is completed. The data of this event is of type ThreadMessage */
+    ThreadMessageCompleted = "thread.message.completed",
+    /** Event sent before a message is completed. The data of this event is of type ThreadMessage */
+    ThreadMessageIncomplete = "thread.message.incomplete"
+}
+/** Terminal event indicating a server side error while streaming. */
+export declare enum ErrorEvent {
+    /** Event sent when an error occurs, such as an internal server error or a timeout. */
+    Error = "error"
+}
+/** Terminal event indicating the successful end of a stream. */
+export declare enum DoneEvent {
+    /** Event sent when the stream is done. */
+    Done = "done"
+}
+/**
+  Represents the type of an agent stream event.
+ */
+export type AgentStreamEventType = ThreadStreamEvent | RunStreamEvent | RunStepStreamEvent | MessageStreamEvent | ErrorEvent | DoneEvent;
+/** Represents a stream of agent event message. */
+export interface AgentEventMessageStream extends AsyncDisposable, AsyncIterable<AgentEventMessage> {
+}
+//# sourceMappingURL=streamingModels.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"streamingModels.d.ts","sourceRoot":"","sources":["../../../src/agents/streamingModels.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,MAAM,qCAAqC,CAAC;AAChG,OAAO,KAAK,EACV,iBAAiB,EACjB,aAAa,EACb,mBAAmB,EACnB,eAAe,EAChB,MAAM,kCAAkC,CAAC;AAE1C;;;;;;;;;;;;;;;;kBAgBkB;AAClB,MAAM,WAAW,iBAAiB;IAChC,kKAAkK;IAClK,IAAI,EAAE,0BAA0B,CAAC;IACjC,6BAA6B;IAC7B,KAAK,EAAE,oBAAoB,GAAG,MAAM,CAAC;CACtC;AAED,mDAAmD;AACnD,MAAM,MAAM,0BAA0B,GAClC,iBAAiB,GACjB,eAAe,GACf,aAAa,GACb,mBAAmB,GACnB,iBAAiB,GACjB,iBAAiB,GACjB,MAAM,CAAC;AAEX,gDAAgD;AAChD,oBAAY,iBAAiB;IAC3B,6FAA6F;IAC7F,aAAa,mBAAmB;CACjC;AAED,6CAA6C;AAC7C,oBAAY,cAAc;IACxB,wFAAwF;IACxF,gBAAgB,uBAAuB;IAEvC,kGAAkG;IAClG,eAAe,sBAAsB;IAErC,uGAAuG;IACvG,mBAAmB,2BAA2B;IAE9C,2GAA2G;IAC3G,uBAAuB,+BAA+B;IAEtD,sFAAsF;IACtF,kBAAkB,yBAAyB;IAE3C,+EAA+E;IAC/E,eAAe,sBAAsB;IAErC,sGAAsG;IACtG,mBAAmB,0BAA0B;IAE7C,sFAAsF;IACtF,kBAAkB,yBAAyB;IAE3C,oFAAoF;IACpF,gBAAgB,uBAAuB;CACxC;AAED,kDAAkD;AAClD,oBAAY,kBAAkB;IAC5B,kGAAkG;IAClG,oBAAoB,4BAA4B;IAEhD,0GAA0G;IAC1G,uBAAuB,gCAAgC;IAEvD,wGAAwG;IACxG,kBAAkB,0BAA0B;IAE5C,yFAAyF;IACzF,sBAAsB,8BAA8B;IAEpD,kFAAkF;IAClF,mBAAmB,2BAA2B;IAE9C,yFAAyF;IACzF,sBAAsB,8BAA8B;IAEpD,uFAAuF;IACvF,oBAAoB,4BAA4B;CACjD;AAED,iDAAiD;AACjD,oBAAY,kBAAkB;IAC5B,gGAAgG;IAChG,oBAAoB,2BAA2B;IAE/C,+GAA+G;IAC/G,uBAAuB,+BAA+B;IAEtD,uGAAuG;IACvG,kBAAkB,yBAAyB;IAE3C,8FAA8F;IAC9F,sBAAsB,6BAA6B;IAEnD,gGAAgG;IAChG,uBAAuB,8BAA8B;CACtD;AAED,qEAAqE;AACrE,oBAAY,UAAU;IACpB,sFAAsF;IACtF,KAAK,UAAU;CAChB;AAED,gEAAgE;AAChE,oBAAY,SAAS;IACnB,0CAA0C;IAC1C,IAAI,SAAS;CACd;AAED;;GAEG;AACH,MAAM,MAAM,oBAAoB,GAC5B,iBAAiB,GACjB,cAAc,GACd,kBAAkB,GAClB,kBAAkB,GAClB,UAAU,GACV,SAAS,CAAC;AAEd,kDAAkD;AAClD,MAAM,WAAW,uBACf,SAAQ,eAAe,EACrB,aAAa,CAAC,iBAAiB,CAAC;CAAG"}

package/dist/commonjs/agents/streamingModels.js

@@ -0,0 +1,78 @@
+"use strict";
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DoneEvent = exports.ErrorEvent = exports.MessageStreamEvent = exports.RunStepStreamEvent = exports.RunStreamEvent = exports.ThreadStreamEvent = void 0;
+/** Thread operation related streaming events */
+var ThreadStreamEvent;
+(function (ThreadStreamEvent) {
+    /** Event sent when a new thread is created. The data of this event is of type AgentThread */
+    ThreadStreamEvent["ThreadCreated"] = "thread.created";
+})(ThreadStreamEvent || (exports.ThreadStreamEvent = ThreadStreamEvent = {}));
+/** Run operation related streaming events */
+var RunStreamEvent;
+(function (RunStreamEvent) {
+    /** Event sent when a new run is created. The data of this event is of type ThreadRun */
+    RunStreamEvent["ThreadRunCreated"] = "thread.run.created";
+    /** Event sent when a run moves to `queued` status. The data of this event is of type ThreadRun */
+    RunStreamEvent["ThreadRunQueued"] = "thread.run.queued";
+    /** Event sent when a run moves to `in_progress` status. The data of this event is of type ThreadRun */
+    RunStreamEvent["ThreadRunInProgress"] = "thread.run.in_progress";
+    /** Event sent when a run moves to `requires_action` status. The data of this event is of type ThreadRun */
+    RunStreamEvent["ThreadRunRequiresAction"] = "thread.run.requires_action";
+    /** Event sent when a run is completed. The data of this event is of type ThreadRun */
+    RunStreamEvent["ThreadRunCompleted"] = "thread.run.completed";
+    /** Event sent when a run fails. The data of this event is of type ThreadRun */
+    RunStreamEvent["ThreadRunFailed"] = "thread.run.failed";
+    /** Event sent when a run moves to `cancelling` status. The data of this event is of type ThreadRun */
+    RunStreamEvent["ThreadRunCancelling"] = "thread.run.cancelling";
+    /** Event sent when a run is cancelled. The data of this event is of type ThreadRun */
+    RunStreamEvent["ThreadRunCancelled"] = "thread.run.cancelled";
+    /** Event sent when a run is expired. The data of this event is of type ThreadRun */
+    RunStreamEvent["ThreadRunExpired"] = "thread.run.expired";
+})(RunStreamEvent || (exports.RunStreamEvent = RunStreamEvent = {}));
+/** Run step operation related streaming events */
+var RunStepStreamEvent;
+(function (RunStepStreamEvent) {
+    /** Event sent when a new thread run step is created. The data of this event is of type RunStep */
+    RunStepStreamEvent["ThreadRunStepCreated"] = "thread.run.step.created";
+    /** Event sent when a run step moves to `in_progress` status. The data of this event is of type RunStep */
+    RunStepStreamEvent["ThreadRunStepInProgress"] = "thread.run.step.in_progress";
+    /** Event sent when a run step is being streamed. The data of this event is of type RunStepDeltaChunk */
+    RunStepStreamEvent["ThreadRunStepDelta"] = "thread.run.step.delta";
+    /** Event sent when a run step is completed. The data of this event is of type RunStep */
+    RunStepStreamEvent["ThreadRunStepCompleted"] = "thread.run.step.completed";
+    /** Event sent when a run step fails. The data of this event is of type RunStep */
+    RunStepStreamEvent["ThreadRunStepFailed"] = "thread.run.step.failed";
+    /** Event sent when a run step is cancelled. The data of this event is of type RunStep */
+    RunStepStreamEvent["ThreadRunStepCancelled"] = "thread.run.step.cancelled";
+    /** Event sent when a run step is expired. The data of this event is of type RunStep */
+    RunStepStreamEvent["ThreadRunStepExpired"] = "thread.run.step.expired";
+})(RunStepStreamEvent || (exports.RunStepStreamEvent = RunStepStreamEvent = {}));
+/** Message operation related streaming events */
+var MessageStreamEvent;
+(function (MessageStreamEvent) {
+    /** Event sent when a new message is created. The data of this event is of type ThreadMessage */
+    MessageStreamEvent["ThreadMessageCreated"] = "thread.message.created";
+    /** Event sent when a message moves to `in_progress` status. The data of this event is of type ThreadMessage */
+    MessageStreamEvent["ThreadMessageInProgress"] = "thread.message.in_progress";
+    /** Event sent when a message is being streamed. The data of this event is of type MessageDeltaChunk */
+    MessageStreamEvent["ThreadMessageDelta"] = "thread.message.delta";
+    /** Event sent when a message is completed. The data of this event is of type ThreadMessage */
+    MessageStreamEvent["ThreadMessageCompleted"] = "thread.message.completed";
+    /** Event sent before a message is completed. The data of this event is of type ThreadMessage */
+    MessageStreamEvent["ThreadMessageIncomplete"] = "thread.message.incomplete";
+})(MessageStreamEvent || (exports.MessageStreamEvent = MessageStreamEvent = {}));
+/** Terminal event indicating a server side error while streaming. */
+var ErrorEvent;
+(function (ErrorEvent) {
+    /** Event sent when an error occurs, such as an internal server error or a timeout. */
+    ErrorEvent["Error"] = "error";
+})(ErrorEvent || (exports.ErrorEvent = ErrorEvent = {}));
+/** Terminal event indicating the successful end of a stream. */
+var DoneEvent;
+(function (DoneEvent) {
+    /** Event sent when the stream is done. */
+    DoneEvent["Done"] = "done";
+})(DoneEvent || (exports.DoneEvent = DoneEvent = {}));
+//# sourceMappingURL=streamingModels.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"streamingModels.js","sourceRoot":"","sources":["../../../src/agents/streamingModels.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;;AA4ClC,gDAAgD;AAChD,IAAY,iBAGX;AAHD,WAAY,iBAAiB;IAC3B,6FAA6F;IAC7F,qDAAgC,CAAA;AAClC,CAAC,EAHW,iBAAiB,iCAAjB,iBAAiB,QAG5B;AAED,6CAA6C;AAC7C,IAAY,cA2BX;AA3BD,WAAY,cAAc;IACxB,wFAAwF;IACxF,yDAAuC,CAAA;IAEvC,kGAAkG;IAClG,uDAAqC,CAAA;IAErC,uGAAuG;IACvG,gEAA8C,CAAA;IAE9C,2GAA2G;IAC3G,wEAAsD,CAAA;IAEtD,sFAAsF;IACtF,6DAA2C,CAAA;IAE3C,+EAA+E;IAC/E,uDAAqC,CAAA;IAErC,sGAAsG;IACtG,+DAA6C,CAAA;IAE7C,sFAAsF;IACtF,6DAA2C,CAAA;IAE3C,oFAAoF;IACpF,yDAAuC,CAAA;AACzC,CAAC,EA3BW,cAAc,8BAAd,cAAc,QA2BzB;AAED,kDAAkD;AAClD,IAAY,kBAqBX;AArBD,WAAY,kBAAkB;IAC5B,kGAAkG;IAClG,sEAAgD,CAAA;IAEhD,0GAA0G;IAC1G,6EAAuD,CAAA;IAEvD,wGAAwG;IACxG,kEAA4C,CAAA;IAE5C,yFAAyF;IACzF,0EAAoD,CAAA;IAEpD,kFAAkF;IAClF,oEAA8C,CAAA;IAE9C,yFAAyF;IACzF,0EAAoD,CAAA;IAEpD,uFAAuF;IACvF,sEAAgD,CAAA;AAClD,CAAC,EArBW,kBAAkB,kCAAlB,kBAAkB,QAqB7B;AAED,iDAAiD;AACjD,IAAY,kBAeX;AAfD,WAAY,kBAAkB;IAC5B,gGAAgG;IAChG,qEAA+C,CAAA;IAE/C,+GAA+G;IAC/G,4EAAsD,CAAA;IAEtD,uGAAuG;IACvG,iEAA2C,CAAA;IAE3C,8FAA8F;IAC9F,yEAAmD,CAAA;IAEnD,gGAAgG;IAChG,2EAAqD,CAAA;AACvD,CAAC,EAfW,kBAAkB,kCAAlB,kBAAkB,QAe7B;AAED,qEAAqE;AACrE,IAAY,UAGX;AAHD,WAAY,UAAU;IACpB,sFAAsF;IACtF,6BAAe,CAAA;AACjB,CAAC,EAHW,UAAU,0BAAV,UAAU,QAGrB;AAED,gEAAgE;AAChE,IAAY,SAGX;AAHD,WAAY,SAAS;IACnB,0CAA0C;IAC1C,0BAAa,CAAA;AACf,CAAC,EAHW,SAAS,yBAAT,SAAS,QAGpB","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport type { MessageDeltaChunk, RunStepDeltaChunk } from \"../customization/streamingModels.js\";\nimport type {\n  AgentThreadOutput,\n  RunStepOutput,\n  ThreadMessageOutput,\n  ThreadRunOutput,\n} from \"../customization/outputModels.js\";\n\n/**\nEach event in a server-sent events stream has an `event` and `data` property:\n  \n  ```\n  event: thread.created\n  data: {\"id\": \"thread_123\", \"object\": \"thread\", ...}\n  ```\n  \n  We emit events whenever a new object is created, transitions to a new state, or is being\n  streamed in parts (deltas). For example, we emit `thread.run.created` when a new run\n  is created, `thread.run.completed` when a run completes, and so on. When an Agent chooses\n  to create a message during a run, we emit a `thread.message.created event`, a\n  `thread.message.in_progress` event, many `thread.message.delta` events, and finally a\n  `thread.message.completed` event.\n  \n  We may add additional events over time, so we recommend handling unknown events gracefully\n  in your code.**/\nexport interface AgentEventMessage {\n  /** The data of the event. The data can be of type AgentThreadOutput, ThreadRunOutput, RunStepOutput, ThreadMessageOutput, MessageDeltaChunk,RunStepDeltaChunk  */\n  data: AgentEventStreamDataOutput;\n  /** The type of the event. */\n  event: AgentStreamEventType | string;\n}\n\n/** Represents a stream event data in the agent. */\nexport type AgentEventStreamDataOutput =\n  | AgentThreadOutput\n  | ThreadRunOutput\n  | RunStepOutput\n  | ThreadMessageOutput\n  | MessageDeltaChunk\n  | RunStepDeltaChunk\n  | string;\n\n/** Thread operation related streaming events */\nexport enum ThreadStreamEvent {\n  /** Event sent when a new thread is created. The data of this event is of type AgentThread */\n  ThreadCreated = \"thread.created\",\n}\n\n/** Run operation related streaming events */\nexport enum RunStreamEvent {\n  /** Event sent when a new run is created. The data of this event is of type ThreadRun */\n  ThreadRunCreated = \"thread.run.created\",\n\n  /** Event sent when a run moves to `queued` status. The data of this event is of type ThreadRun */\n  ThreadRunQueued = \"thread.run.queued\",\n\n  /** Event sent when a run moves to `in_progress` status. The data of this event is of type ThreadRun */\n  ThreadRunInProgress = \"thread.run.in_progress\",\n\n  /** Event sent when a run moves to `requires_action` status. The data of this event is of type ThreadRun */\n  ThreadRunRequiresAction = \"thread.run.requires_action\",\n\n  /** Event sent when a run is completed. The data of this event is of type ThreadRun */\n  ThreadRunCompleted = \"thread.run.completed\",\n\n  /** Event sent when a run fails. The data of this event is of type ThreadRun */\n  ThreadRunFailed = \"thread.run.failed\",\n\n  /** Event sent when a run moves to `cancelling` status. The data of this event is of type ThreadRun */\n  ThreadRunCancelling = \"thread.run.cancelling\",\n\n  /** Event sent when a run is cancelled. The data of this event is of type ThreadRun */\n  ThreadRunCancelled = \"thread.run.cancelled\",\n\n  /** Event sent when a run is expired. The data of this event is of type ThreadRun */\n  ThreadRunExpired = \"thread.run.expired\",\n}\n\n/** Run step operation related streaming events */\nexport enum RunStepStreamEvent {\n  /** Event sent when a new thread run step is created. The data of this event is of type RunStep */\n  ThreadRunStepCreated = \"thread.run.step.created\",\n\n  /** Event sent when a run step moves to `in_progress` status. The data of this event is of type RunStep */\n  ThreadRunStepInProgress = \"thread.run.step.in_progress\",\n\n  /** Event sent when a run step is being streamed. The data of this event is of type RunStepDeltaChunk */\n  ThreadRunStepDelta = \"thread.run.step.delta\",\n\n  /** Event sent when a run step is completed. The data of this event is of type RunStep */\n  ThreadRunStepCompleted = \"thread.run.step.completed\",\n\n  /** Event sent when a run step fails. The data of this event is of type RunStep */\n  ThreadRunStepFailed = \"thread.run.step.failed\",\n\n  /** Event sent when a run step is cancelled. The data of this event is of type RunStep */\n  ThreadRunStepCancelled = \"thread.run.step.cancelled\",\n\n  /** Event sent when a run step is expired. The data of this event is of type RunStep */\n  ThreadRunStepExpired = \"thread.run.step.expired\",\n}\n\n/** Message operation related streaming events */\nexport enum MessageStreamEvent {\n  /** Event sent when a new message is created. The data of this event is of type ThreadMessage */\n  ThreadMessageCreated = \"thread.message.created\",\n\n  /** Event sent when a message moves to `in_progress` status. The data of this event is of type ThreadMessage */\n  ThreadMessageInProgress = \"thread.message.in_progress\",\n\n  /** Event sent when a message is being streamed. The data of this event is of type MessageDeltaChunk */\n  ThreadMessageDelta = \"thread.message.delta\",\n\n  /** Event sent when a message is completed. The data of this event is of type ThreadMessage */\n  ThreadMessageCompleted = \"thread.message.completed\",\n\n  /** Event sent before a message is completed. The data of this event is of type ThreadMessage */\n  ThreadMessageIncomplete = \"thread.message.incomplete\",\n}\n\n/** Terminal event indicating a server side error while streaming. */\nexport enum ErrorEvent {\n  /** Event sent when an error occurs, such as an internal server error or a timeout. */\n  Error = \"error\",\n}\n\n/** Terminal event indicating the successful end of a stream. */\nexport enum DoneEvent {\n  /** Event sent when the stream is done. */\n  Done = \"done\",\n}\n\n/**\n  Represents the type of an agent stream event.\n */\nexport type AgentStreamEventType =\n  | ThreadStreamEvent\n  | RunStreamEvent\n  | RunStepStreamEvent\n  | MessageStreamEvent\n  | ErrorEvent\n  | DoneEvent;\n\n/** Represents a stream of agent event message. */\nexport interface AgentEventMessageStream\n  extends AsyncDisposable,\n    AsyncIterable<AgentEventMessage> {}\n"]}

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

@@ -0,0 +1,12 @@
+import type { Client } from "@azure-rest/core-client";
+import type { AgentThreadOutput, ThreadDeletionStatusOutput } from "../customization/outputModels.js";
+import type { CreateAgentThreadOptionalParams, DeleteAgentThreadOptionalParams, GetAgentThreadOptionalParams, UpdateAgentThreadOptionalParams } from "./customModels.js";
+/** Creates a new thread. Threads contain messages and can be run by agents. */
+export declare function createThread(context: Client, options?: CreateAgentThreadOptionalParams): Promise<AgentThreadOutput>;
+/** Gets information about an existing thread. */
+export declare function getThread(context: Client, threadId: string, options?: GetAgentThreadOptionalParams): Promise<AgentThreadOutput>;
+/** Modifies an existing thread. */
+export declare function updateThread(context: Client, threadId: string, options?: UpdateAgentThreadOptionalParams): Promise<AgentThreadOutput>;
+/** Deletes an existing thread. */
+export declare function deleteThread(context: Client, threadId: string, options?: DeleteAgentThreadOptionalParams): Promise<ThreadDeletionStatusOutput>;
+//# sourceMappingURL=threads.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"threads.d.ts","sourceRoot":"","sources":["../../../src/agents/threads.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,yBAAyB,CAAC;AAKtD,OAAO,KAAK,EACV,iBAAiB,EACjB,0BAA0B,EAC3B,MAAM,kCAAkC,CAAC;AAU1C,OAAO,KAAK,EACV,+BAA+B,EAC/B,+BAA+B,EAC/B,4BAA4B,EAC5B,+BAA+B,EAChC,MAAM,mBAAmB,CAAC;AAM3B,+EAA+E;AAC/E,wBAAsB,YAAY,CAChC,OAAO,EAAE,MAAM,EACf,OAAO,GAAE,+BAAoC,GAC5C,OAAO,CAAC,iBAAiB,CAAC,CAwB5B;AAED,iDAAiD;AACjD,wBAAsB,SAAS,CAC7B,OAAO,EAAE,MAAM,EACf,QAAQ,EAAE,MAAM,EAChB,OAAO,GAAE,4BAAiC,GACzC,OAAO,CAAC,iBAAiB,CAAC,CAwB5B;AAED,mCAAmC;AACnC,wBAAsB,YAAY,CAChC,OAAO,EAAE,MAAM,EACf,QAAQ,EAAE,MAAM,EAChB,OAAO,GAAE,+BAAoC,GAC5C,OAAO,CAAC,iBAAiB,CAAC,CA2B5B;AAED,kCAAkC;AAClC,wBAAsB,YAAY,CAChC,OAAO,EAAE,MAAM,EACf,QAAQ,EAAE,MAAM,EAChB,OAAO,GAAE,+BAAoC,GAC5C,OAAO,CAAC,0BAA0B,CAAC,CAwBrC"}

package/dist/commonjs/agents/threads.js

@@ -0,0 +1,92 @@
+"use strict";
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createThread = createThread;
+exports.getThread = getThread;
+exports.updateThread = updateThread;
+exports.deleteThread = deleteThread;
+const tslib_1 = require("tslib");
+const core_client_1 = require("@azure-rest/core-client");
+const ConverterToWire = tslib_1.__importStar(require("../customization/convertModelsToWrite.js"));
+const ConverterFromWire = tslib_1.__importStar(require("../customization/convertOutputModelsFromWire.js"));
+const tracing_js_1 = require("../tracing.js");
+const threadsTrace_js_1 = require("./threadsTrace.js");
+const inputValidations_js_1 = require("./inputValidations.js");
+const traceUtility_js_1 = require("./traceUtility.js");
+const convertOutputModelsFromWire_js_1 = require("../customization/convertOutputModelsFromWire.js");
+const openAIError_js_1 = require("./openAIError.js");
+const expectedStatuses = ["200"];
+/** Creates a new thread. Threads contain messages and can be run by agents. */
+async function createThread(context, options = {}) {
+    const createThreadOptions = Object.assign(Object.assign({}, (0, core_client_1.operationOptionsToRequestParameters)(options)), { body: Object.assign({}, ConverterToWire.convertAgentThreadCreationOptions(options)) });
+    validateCreateThreadParameters(createThreadOptions);
+    const response = await tracing_js_1.TracingUtility.withSpan("CreateThread", createThreadOptions, async (updatedOptions) => {
+        const result = await context.path("/threads").post(updatedOptions);
+        if (!expectedStatuses.includes(result.status)) {
+            throw (0, openAIError_js_1.createOpenAIError)(result);
+        }
+        return result.body;
+    }, threadsTrace_js_1.traceStartCreateThread, threadsTrace_js_1.traceEndCreateThread);
+    return ConverterFromWire.convertAgentThreadOutput(response);
+}
+/** Gets information about an existing thread. */
+async function getThread(context, threadId, options = {}) {
+    const getThreadOptions = Object.assign({}, (0, core_client_1.operationOptionsToRequestParameters)(options));
+    (0, inputValidations_js_1.validateThreadId)(threadId);
+    const response = await tracing_js_1.TracingUtility.withSpan("GetThread", getThreadOptions, async (updatedOptions) => {
+        const result = await context.path("/threads/{threadId}", threadId).get(updatedOptions);
+        if (!expectedStatuses.includes(result.status)) {
+            throw (0, openAIError_js_1.createOpenAIError)(result);
+        }
+        return result.body;
+    }, (span, updatedOptions) => (0, traceUtility_js_1.traceStartAgentGeneric)(span, Object.assign(Object.assign({}, updatedOptions), { tracingAttributeOptions: { threadId: threadId } })));
+    return ConverterFromWire.convertAgentThreadOutput(response);
+}
+/** Modifies an existing thread. */
+async function updateThread(context, threadId, options = {}) {
+    const updateThreadOptions = Object.assign(Object.assign({}, (0, core_client_1.operationOptionsToRequestParameters)(options)), { body: Object.assign({}, ConverterToWire.convertAgentThreadUpdateOptions(options)) });
+    validateUpdateThreadParameters(threadId, updateThreadOptions);
+    const response = await tracing_js_1.TracingUtility.withSpan("UpdateThread", updateThreadOptions, async (updatedOptions) => {
+        const result = await context.path("/threads/{threadId}", threadId).post(updatedOptions);
+        if (!expectedStatuses.includes(result.status)) {
+            throw (0, openAIError_js_1.createOpenAIError)(result);
+        }
+        return result.body;
+    }, (span, updatedOptions) => (0, traceUtility_js_1.traceStartAgentGeneric)(span, Object.assign(Object.assign({}, updatedOptions), { tracingAttributeOptions: { threadId: threadId } })));
+    return ConverterFromWire.convertAgentThreadOutput(response);
+}
+/** Deletes an existing thread. */
+async function deleteThread(context, threadId, options = {}) {
+    const deleteThreadOptions = Object.assign({}, (0, core_client_1.operationOptionsToRequestParameters)(options));
+    (0, inputValidations_js_1.validateThreadId)(threadId);
+    const response = await tracing_js_1.TracingUtility.withSpan("DeleteThread", deleteThreadOptions, async (updatedOptions) => {
+        const result = await context.path("/threads/{threadId}", threadId).delete(updatedOptions);
+        if (!expectedStatuses.includes(result.status)) {
+            throw (0, openAIError_js_1.createOpenAIError)(result);
+        }
+        return result.body;
+    }, (span, updatedOptions) => (0, traceUtility_js_1.traceStartAgentGeneric)(span, Object.assign(Object.assign({}, updatedOptions), { tracingAttributeOptions: { threadId: threadId } })));
+    return (0, convertOutputModelsFromWire_js_1.convertThreadDeletionStatusOutput)(response);
+}
+function validateCreateThreadParameters(options) {
+    if (options === null || options === void 0 ? void 0 : options.body.messages) {
+        options.body.messages.forEach((message) => (0, inputValidations_js_1.validateMessages)(message.role));
+    }
+    if (options === null || options === void 0 ? void 0 : options.body.tool_resources) {
+        (0, inputValidations_js_1.validateToolResources)(options.body.tool_resources);
+    }
+    if (options === null || options === void 0 ? void 0 : options.body.metadata) {
+        (0, inputValidations_js_1.validateMetadata)(options.body.metadata);
+    }
+}
+function validateUpdateThreadParameters(threadId, options) {
+    (0, inputValidations_js_1.validateThreadId)(threadId);
+    if (options === null || options === void 0 ? void 0 : options.body.tool_resources) {
+        (0, inputValidations_js_1.validateToolResources)(options.body.tool_resources);
+    }
+    if (options === null || options === void 0 ? void 0 : options.body.metadata) {
+        (0, inputValidations_js_1.validateMetadata)(options.body.metadata);
+    }
+}
+//# sourceMappingURL=threads.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"threads.js","sourceRoot":"","sources":["../../../src/agents/threads.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;AAgClC,oCA2BC;AAGD,8BA4BC;AAGD,oCA+BC;AAGD,oCA4BC;;AAxJD,yDAA8E;AAE9E,kGAA4E;AAC5E,2GAAqF;AAKrF,8CAA+C;AAC/C,uDAAiF;AACjF,+DAK+B;AAC/B,uDAA2D;AAO3D,oGAAoG;AACpG,qDAAqD;AAErD,MAAM,gBAAgB,GAAG,CAAC,KAAK,CAAC,CAAC;AAEjC,+EAA+E;AACxE,KAAK,UAAU,YAAY,CAChC,OAAe,EACf,UAA2C,EAAE;IAE7C,MAAM,mBAAmB,mCACpB,IAAA,iDAAmC,EAAC,OAAO,CAAC,KAC/C,IAAI,oBACC,eAAe,CAAC,iCAAiC,CAAC,OAAO,CAAC,IAEhE,CAAC;IAEF,8BAA8B,CAAC,mBAAmB,CAAC,CAAC;IACpD,MAAM,QAAQ,GAAG,MAAM,2BAAc,CAAC,QAAQ,CAC5C,cAAc,EACd,mBAAmB,EACnB,KAAK,EAAE,cAAc,EAAE,EAAE;QACvB,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACnE,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;YAC9C,MAAM,IAAA,kCAAiB,EAAC,MAAM,CAAC,CAAC;QAClC,CAAC;QACD,OAAO,MAAM,CAAC,IAAI,CAAC;IACrB,CAAC,EACD,wCAAsB,EACtB,sCAAoB,CACrB,CAAC;IAEF,OAAO,iBAAiB,CAAC,wBAAwB,CAAC,QAAQ,CAAC,CAAC;AAC9D,CAAC;AAED,iDAAiD;AAC1C,KAAK,UAAU,SAAS,CAC7B,OAAe,EACf,QAAgB,EAChB,UAAwC,EAAE;IAE1C,MAAM,gBAAgB,qBACjB,IAAA,iDAAmC,EAAC,OAAO,CAAC,CAChD,CAAC;IAEF,IAAA,sCAAgB,EAAC,QAAQ,CAAC,CAAC;IAC3B,MAAM,QAAQ,GAAG,MAAM,2BAAc,CAAC,QAAQ,CAC5C,WAAW,EACX,gBAAgB,EAChB,KAAK,EAAE,cAAc,EAAE,EAAE;QACvB,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,qBAAqB,EAAE,QAAQ,CAAC,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;QACvF,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;YAC9C,MAAM,IAAA,kCAAiB,EAAC,MAAM,CAAC,CAAC;QAClC,CAAC;QACD,OAAO,MAAM,CAAC,IAAI,CAAC;IACrB,CAAC,EACD,CAAC,IAAI,EAAE,cAAc,EAAE,EAAE,CACvB,IAAA,wCAAsB,EAAC,IAAI,kCACtB,cAAc,KACjB,uBAAuB,EAAE,EAAE,QAAQ,EAAE,QAAQ,EAAE,IAC/C,CACL,CAAC;IAEF,OAAO,iBAAiB,CAAC,wBAAwB,CAAC,QAAQ,CAAC,CAAC;AAC9D,CAAC;AAED,mCAAmC;AAC5B,KAAK,UAAU,YAAY,CAChC,OAAe,EACf,QAAgB,EAChB,UAA2C,EAAE;IAE7C,MAAM,mBAAmB,mCACpB,IAAA,iDAAmC,EAAC,OAAO,CAAC,KAC/C,IAAI,oBACC,eAAe,CAAC,+BAA+B,CAAC,OAAO,CAAC,IAE9D,CAAC;IAEF,8BAA8B,CAAC,QAAQ,EAAE,mBAAmB,CAAC,CAAC;IAC9D,MAAM,QAAQ,GAAG,MAAM,2BAAc,CAAC,QAAQ,CAC5C,cAAc,EACd,mBAAmB,EACnB,KAAK,EAAE,cAAc,EAAE,EAAE;QACvB,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,qBAAqB,EAAE,QAAQ,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACxF,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;YAC9C,MAAM,IAAA,kCAAiB,EAAC,MAAM,CAAC,CAAC;QAClC,CAAC;QACD,OAAO,MAAM,CAAC,IAAI,CAAC;IACrB,CAAC,EACD,CAAC,IAAI,EAAE,cAAc,EAAE,EAAE,CACvB,IAAA,wCAAsB,EAAC,IAAI,kCACtB,cAAc,KACjB,uBAAuB,EAAE,EAAE,QAAQ,EAAE,QAAQ,EAAE,IAC/C,CACL,CAAC;IAEF,OAAO,iBAAiB,CAAC,wBAAwB,CAAC,QAAQ,CAAC,CAAC;AAC9D,CAAC;AAED,kCAAkC;AAC3B,KAAK,UAAU,YAAY,CAChC,OAAe,EACf,QAAgB,EAChB,UAA2C,EAAE;IAE7C,MAAM,mBAAmB,qBACpB,IAAA,iDAAmC,EAAC,OAAO,CAAC,CAChD,CAAC;IAEF,IAAA,sCAAgB,EAAC,QAAQ,CAAC,CAAC;IAC3B,MAAM,QAAQ,GAAG,MAAM,2BAAc,CAAC,QAAQ,CAC5C,cAAc,EACd,mBAAmB,EACnB,KAAK,EAAE,cAAc,EAAE,EAAE;QACvB,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,qBAAqB,EAAE,QAAQ,CAAC,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;QAC1F,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;YAC9C,MAAM,IAAA,kCAAiB,EAAC,MAAM,CAAC,CAAC;QAClC,CAAC;QACD,OAAO,MAAM,CAAC,IAAI,CAAC;IACrB,CAAC,EACD,CAAC,IAAI,EAAE,cAAc,EAAE,EAAE,CACvB,IAAA,wCAAsB,EAAC,IAAI,kCACtB,cAAc,KACjB,uBAAuB,EAAE,EAAE,QAAQ,EAAE,QAAQ,EAAE,IAC/C,CACL,CAAC;IAEF,OAAO,IAAA,kEAAiC,EAAC,QAAQ,CAAC,CAAC;AACrD,CAAC;AAED,SAAS,8BAA8B,CACrC,OAAoD;IAEpD,IAAI,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,IAAI,CAAC,QAAQ,EAAE,CAAC;QAC3B,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,IAAA,sCAAgB,EAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC;IAC7E,CAAC;IACD,IAAI,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,IAAI,CAAC,cAAc,EAAE,CAAC;QACjC,IAAA,2CAAqB,EAAC,OAAO,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;IACrD,CAAC;IACD,IAAI,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,IAAI,CAAC,QAAQ,EAAE,CAAC;QAC3B,IAAA,sCAAgB,EAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC1C,CAAC;AACH,CAAC;AAED,SAAS,8BAA8B,CACrC,QAAgB,EAChB,OAAoD;IAEpD,IAAA,sCAAgB,EAAC,QAAQ,CAAC,CAAC;IAC3B,IAAI,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,IAAI,CAAC,cAAc,EAAE,CAAC;QACjC,IAAA,2CAAqB,EAAC,OAAO,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;IACrD,CAAC;IACD,IAAI,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,IAAI,CAAC,QAAQ,EAAE,CAAC;QAC3B,IAAA,sCAAgB,EAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC1C,CAAC;AACH,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport type { Client } from \"@azure-rest/core-client\";\nimport { operationOptionsToRequestParameters } from \"@azure-rest/core-client\";\nimport type * as GeneratedParameters from \"../generated/src/parameters.js\";\nimport * as ConverterToWire from \"../customization/convertModelsToWrite.js\";\nimport * as ConverterFromWire from \"../customization/convertOutputModelsFromWire.js\";\nimport type {\n  AgentThreadOutput,\n  ThreadDeletionStatusOutput,\n} from \"../customization/outputModels.js\";\nimport { TracingUtility } from \"../tracing.js\";\nimport { traceEndCreateThread, traceStartCreateThread } from \"./threadsTrace.js\";\nimport {\n  validateMessages,\n  validateMetadata,\n  validateThreadId,\n  validateToolResources,\n} from \"./inputValidations.js\";\nimport { traceStartAgentGeneric } from \"./traceUtility.js\";\nimport type {\n  CreateAgentThreadOptionalParams,\n  DeleteAgentThreadOptionalParams,\n  GetAgentThreadOptionalParams,\n  UpdateAgentThreadOptionalParams,\n} from \"./customModels.js\";\nimport { convertThreadDeletionStatusOutput } from \"../customization/convertOutputModelsFromWire.js\";\nimport { createOpenAIError } from \"./openAIError.js\";\n\nconst expectedStatuses = [\"200\"];\n\n/** Creates a new thread. Threads contain messages and can be run by agents. */\nexport async function createThread(\n  context: Client,\n  options: CreateAgentThreadOptionalParams = {},\n): Promise<AgentThreadOutput> {\n  const createThreadOptions: GeneratedParameters.CreateThreadParameters = {\n    ...operationOptionsToRequestParameters(options),\n    body: {\n      ...ConverterToWire.convertAgentThreadCreationOptions(options),\n    },\n  };\n\n  validateCreateThreadParameters(createThreadOptions);\n  const response = await TracingUtility.withSpan(\n    \"CreateThread\",\n    createThreadOptions,\n    async (updatedOptions) => {\n      const result = await context.path(\"/threads\").post(updatedOptions);\n      if (!expectedStatuses.includes(result.status)) {\n        throw createOpenAIError(result);\n      }\n      return result.body;\n    },\n    traceStartCreateThread,\n    traceEndCreateThread,\n  );\n\n  return ConverterFromWire.convertAgentThreadOutput(response);\n}\n\n/** Gets information about an existing thread. */\nexport async function getThread(\n  context: Client,\n  threadId: string,\n  options: GetAgentThreadOptionalParams = {},\n): Promise<AgentThreadOutput> {\n  const getThreadOptions: GeneratedParameters.GetThreadParameters = {\n    ...operationOptionsToRequestParameters(options),\n  };\n\n  validateThreadId(threadId);\n  const response = await TracingUtility.withSpan(\n    \"GetThread\",\n    getThreadOptions,\n    async (updatedOptions) => {\n      const result = await context.path(\"/threads/{threadId}\", threadId).get(updatedOptions);\n      if (!expectedStatuses.includes(result.status)) {\n        throw createOpenAIError(result);\n      }\n      return result.body;\n    },\n    (span, updatedOptions) =>\n      traceStartAgentGeneric(span, {\n        ...updatedOptions,\n        tracingAttributeOptions: { threadId: threadId },\n      }),\n  );\n\n  return ConverterFromWire.convertAgentThreadOutput(response);\n}\n\n/** Modifies an existing thread. */\nexport async function updateThread(\n  context: Client,\n  threadId: string,\n  options: UpdateAgentThreadOptionalParams = {},\n): Promise<AgentThreadOutput> {\n  const updateThreadOptions: GeneratedParameters.UpdateThreadParameters = {\n    ...operationOptionsToRequestParameters(options),\n    body: {\n      ...ConverterToWire.convertAgentThreadUpdateOptions(options),\n    },\n  };\n\n  validateUpdateThreadParameters(threadId, updateThreadOptions);\n  const response = await TracingUtility.withSpan(\n    \"UpdateThread\",\n    updateThreadOptions,\n    async (updatedOptions) => {\n      const result = await context.path(\"/threads/{threadId}\", threadId).post(updatedOptions);\n      if (!expectedStatuses.includes(result.status)) {\n        throw createOpenAIError(result);\n      }\n      return result.body;\n    },\n    (span, updatedOptions) =>\n      traceStartAgentGeneric(span, {\n        ...updatedOptions,\n        tracingAttributeOptions: { threadId: threadId },\n      }),\n  );\n\n  return ConverterFromWire.convertAgentThreadOutput(response);\n}\n\n/** Deletes an existing thread. */\nexport async function deleteThread(\n  context: Client,\n  threadId: string,\n  options: DeleteAgentThreadOptionalParams = {},\n): Promise<ThreadDeletionStatusOutput> {\n  const deleteThreadOptions: GeneratedParameters.DeleteAgentParameters = {\n    ...operationOptionsToRequestParameters(options),\n  };\n\n  validateThreadId(threadId);\n  const response = await TracingUtility.withSpan(\n    \"DeleteThread\",\n    deleteThreadOptions,\n    async (updatedOptions) => {\n      const result = await context.path(\"/threads/{threadId}\", threadId).delete(updatedOptions);\n      if (!expectedStatuses.includes(result.status)) {\n        throw createOpenAIError(result);\n      }\n      return result.body;\n    },\n    (span, updatedOptions) =>\n      traceStartAgentGeneric(span, {\n        ...updatedOptions,\n        tracingAttributeOptions: { threadId: threadId },\n      }),\n  );\n\n  return convertThreadDeletionStatusOutput(response);\n}\n\nfunction validateCreateThreadParameters(\n  options?: GeneratedParameters.CreateThreadParameters,\n): void {\n  if (options?.body.messages) {\n    options.body.messages.forEach((message) => validateMessages(message.role));\n  }\n  if (options?.body.tool_resources) {\n    validateToolResources(options.body.tool_resources);\n  }\n  if (options?.body.metadata) {\n    validateMetadata(options.body.metadata);\n  }\n}\n\nfunction validateUpdateThreadParameters(\n  threadId: string,\n  options?: GeneratedParameters.UpdateThreadParameters,\n): void {\n  validateThreadId(threadId);\n  if (options?.body.tool_resources) {\n    validateToolResources(options.body.tool_resources);\n  }\n  if (options?.body.metadata) {\n    validateMetadata(options.body.metadata);\n  }\n}\n"]}

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

@@ -0,0 +1,6 @@
+import type { AgentThreadOutput } from "../generated/src/outputModels.js";
+import type { Span } from "../tracing.js";
+import type { CreateThreadParameters } from "../generated/src/parameters.js";
+export declare function traceStartCreateThread(span: Span, options: CreateThreadParameters): void;
+export declare function traceEndCreateThread(span: Span, _options: CreateThreadParameters, result: Promise<AgentThreadOutput>): Promise<void>;
+//# sourceMappingURL=threadsTrace.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"threadsTrace.d.ts","sourceRoot":"","sources":["../../../src/agents/threadsTrace.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,kCAAkC,CAAC;AAC1E,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,eAAe,CAAC;AAE1C,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,gCAAgC,CAAC;AAG7E,wBAAgB,sBAAsB,CAAC,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE,sBAAsB,GAAG,IAAI,CAOxF;AAED,wBAAsB,oBAAoB,CACxC,IAAI,EAAE,IAAI,EACV,QAAQ,EAAE,sBAAsB,EAChC,MAAM,EAAE,OAAO,CAAC,iBAAiB,CAAC,GACjC,OAAO,CAAC,IAAI,CAAC,CAGf"}

package/dist/commonjs/agents/threadsTrace.js

@@ -0,0 +1,23 @@
+"use strict";
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.traceStartCreateThread = traceStartCreateThread;
+exports.traceEndCreateThread = traceEndCreateThread;
+const tracing_js_1 = require("../tracing.js");
+const traceUtility_js_1 = require("./traceUtility.js");
+function traceStartCreateThread(span, options) {
+    tracing_js_1.TracingUtility.setSpanAttributes(span, tracing_js_1.TracingOperationName.CREATE_THREAD, (0, traceUtility_js_1.UpdateWithAgentAttributes)({}));
+    setSpanEvents(span, options);
+}
+async function traceEndCreateThread(span, _options, result) {
+    const resolvedResult = await result;
+    tracing_js_1.TracingUtility.updateSpanAttributes(span, { threadId: resolvedResult.id });
+}
+function setSpanEvents(span, options) {
+    var _a;
+    (_a = options.body.messages) === null || _a === void 0 ? void 0 : _a.forEach((message) => {
+        (0, traceUtility_js_1.addMessageEvent)(span, message);
+    });
+}
+//# sourceMappingURL=threadsTrace.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"threadsTrace.js","sourceRoot":"","sources":["../../../src/agents/threadsTrace.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;AAQlC,wDAOC;AAED,oDAOC;AApBD,8CAAqE;AAErE,uDAA+E;AAE/E,SAAgB,sBAAsB,CAAC,IAAU,EAAE,OAA+B;IAChF,2BAAc,CAAC,iBAAiB,CAC9B,IAAI,EACJ,iCAAoB,CAAC,aAAa,EAClC,IAAA,2CAAyB,EAAC,EAAE,CAAC,CAC9B,CAAC;IACF,aAAa,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;AAC/B,CAAC;AAEM,KAAK,UAAU,oBAAoB,CACxC,IAAU,EACV,QAAgC,EAChC,MAAkC;IAElC,MAAM,cAAc,GAAG,MAAM,MAAM,CAAC;IACpC,2BAAc,CAAC,oBAAoB,CAAC,IAAI,EAAE,EAAE,QAAQ,EAAE,cAAc,CAAC,EAAE,EAAE,CAAC,CAAC;AAC7E,CAAC;AAED,SAAS,aAAa,CAAC,IAAU,EAAE,OAA+B;;IAChE,MAAA,OAAO,CAAC,IAAI,CAAC,QAAQ,0CAAE,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QACzC,IAAA,iCAAe,EAAC,IAAI,EAAE,OAAO,CAAC,CAAC;IACjC,CAAC,CAAC,CAAC;AACL,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport type { AgentThreadOutput } from \"../generated/src/outputModels.js\";\nimport type { Span } from \"../tracing.js\";\nimport { TracingUtility, TracingOperationName } from \"../tracing.js\";\nimport type { CreateThreadParameters } from \"../generated/src/parameters.js\";\nimport { addMessageEvent, UpdateWithAgentAttributes } from \"./traceUtility.js\";\n\nexport function traceStartCreateThread(span: Span, options: CreateThreadParameters): void {\n  TracingUtility.setSpanAttributes(\n    span,\n    TracingOperationName.CREATE_THREAD,\n    UpdateWithAgentAttributes({}),\n  );\n  setSpanEvents(span, options);\n}\n\nexport async function traceEndCreateThread(\n  span: Span,\n  _options: CreateThreadParameters,\n  result: Promise<AgentThreadOutput>,\n): Promise<void> {\n  const resolvedResult = await result;\n  TracingUtility.updateSpanAttributes(span, { threadId: resolvedResult.id });\n}\n\nfunction setSpanEvents(span: Span, options: CreateThreadParameters): void {\n  options.body.messages?.forEach((message) => {\n    addMessageEvent(span, message);\n  });\n}\n"]}

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

@@ -0,0 +1,36 @@
+import type { AgentsApiResponseFormatOption, ThreadMessage, ThreadMessageOptions, ToolOutput } from "../generated/src/models.js";
+import type { RunStepCompletionUsageOutput } from "../generated/src/outputModels.js";
+import type { OptionsWithTracing, Span, TracingAttributeOptions } from "../tracing.js";
+export declare function traceStartAgentGeneric<Options extends OptionsWithTracing>(span: Span, options: Options): void;
+export declare function traceEndAgentGeneric<Options extends OptionsWithTracing>(span: Span, _options: Options): void;
+export declare function UpdateWithAgentAttributes(attributeOptions: Omit<TracingAttributeOptions, "operationName">): Omit<TracingAttributeOptions, "operationName">;
+/**
+ * Adds a message event to the span.
+ * @param span - The span to add the event to.
+ * @param messageAttributes - The attributes of the message event.
+ */
+export declare function addMessageEvent(span: Span, messageAttributes: ThreadMessageOptions | ThreadMessage, usage?: RunStepCompletionUsageOutput): void;
+/**
+ * Adds an instruction event to the span.
+ * @param span - The span to add the event to.
+ * @param instructionAttributes - The attributes of the instruction event.
+ */
+export declare function addInstructionsEvent(span: Span, instructionAttributes: {
+    instructions?: string | null;
+    additional_instructions?: string | null;
+    threadId?: string;
+    agentId?: string;
+}): void;
+/**
+ * Formats the agent API response.
+ * @param responseFormat - The response format option.
+ * @returns The formatted response as a string, or null/undefined.
+ */
+export declare function formatAgentApiResponse(responseFormat: AgentsApiResponseFormatOption | null | undefined): string | undefined;
+/**
+ * Adds a tool messages event to the span
+ * @param span - The span to add the event to.
+ * @param tool_outputs - List of tool oupts
+ */
+export declare function addToolMessagesEvent(span: Span, tool_outputs: Array<ToolOutput>): void;
+//# sourceMappingURL=traceUtility.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"traceUtility.d.ts","sourceRoot":"","sources":["../../../src/agents/traceUtility.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAEV,6BAA6B,EAE7B,aAAa,EACb,oBAAoB,EACpB,UAAU,EACX,MAAM,4BAA4B,CAAC;AACpC,OAAO,KAAK,EAAE,4BAA4B,EAAE,MAAM,kCAAkC,CAAC;AACrF,OAAO,KAAK,EAAE,kBAAkB,EAAE,IAAI,EAAE,uBAAuB,EAAE,MAAM,eAAe,CAAC;AAIvF,wBAAgB,sBAAsB,CAAC,OAAO,SAAS,kBAAkB,EACvE,IAAI,EAAE,IAAI,EACV,OAAO,EAAE,OAAO,GACf,IAAI,CAON;AACD,wBAAgB,oBAAoB,CAAC,OAAO,SAAS,kBAAkB,EACrE,IAAI,EAAE,IAAI,EACV,QAAQ,EAAE,OAAO,GAChB,IAAI,CAGN;AAED,wBAAgB,yBAAyB,CACvC,gBAAgB,EAAE,IAAI,CAAC,uBAAuB,EAAE,eAAe,CAAC,GAC/D,IAAI,CAAC,uBAAuB,EAAE,eAAe,CAAC,CAGhD;AAED;;;;GAIG;AACH,wBAAgB,eAAe,CAC7B,IAAI,EAAE,IAAI,EACV,iBAAiB,EAAE,oBAAoB,GAAG,aAAa,EACvD,KAAK,CAAC,EAAE,4BAA4B,GACnC,IAAI,CAsCN;AAED;;;;GAIG;AACH,wBAAgB,oBAAoB,CAClC,IAAI,EAAE,IAAI,EACV,qBAAqB,EAAE;IACrB,YAAY,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC7B,uBAAuB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACxC,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB,GACA,IAAI,CAeN;AAED;;;;GAIG;AACH,wBAAgB,sBAAsB,CACpC,cAAc,EAAE,6BAA6B,GAAG,IAAI,GAAG,SAAS,GAC/D,MAAM,GAAG,SAAS,CAYpB;AAED;;;;GAIG;AACH,wBAAgB,oBAAoB,CAAC,IAAI,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,CAAC,UAAU,CAAC,GAAG,IAAI,CAQtF"}

package/dist/commonjs/agents/traceUtility.js

@@ -0,0 +1,136 @@
+"use strict";
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.traceStartAgentGeneric = traceStartAgentGeneric;
+exports.traceEndAgentGeneric = traceEndAgentGeneric;
+exports.UpdateWithAgentAttributes = UpdateWithAgentAttributes;
+exports.addMessageEvent = addMessageEvent;
+exports.addInstructionsEvent = addInstructionsEvent;
+exports.formatAgentApiResponse = formatAgentApiResponse;
+exports.addToolMessagesEvent = addToolMessagesEvent;
+const tracing_js_1 = require("../tracing.js");
+const telemetry_js_1 = require("../telemetry/telemetry.js");
+function traceStartAgentGeneric(span, options) {
+    var _a;
+    const attributeOptions = options.tracingAttributeOptions || {};
+    tracing_js_1.TracingUtility.setSpanAttributes(span, ((_a = options.tracingAttributeOptions) === null || _a === void 0 ? void 0 : _a.operationName) || "Agent_Operation", UpdateWithAgentAttributes(attributeOptions));
+}
+function traceEndAgentGeneric(span, _options) {
+    const attributeOptions = {};
+    tracing_js_1.TracingUtility.updateSpanAttributes(span, UpdateWithAgentAttributes(attributeOptions));
+}
+function UpdateWithAgentAttributes(attributeOptions) {
+    attributeOptions.genAiSystem = tracing_js_1.TracingAttributes.AZ_AI_AGENT_SYSTEM;
+    return attributeOptions;
+}
+/**
+ * Adds a message event to the span.
+ * @param span - The span to add the event to.
+ * @param messageAttributes - The attributes of the message event.
+ */
+function addMessageEvent(span, messageAttributes, usage) {
+    var _a;
+    const eventBody = {};
+    const telemetryOptions = (0, telemetry_js_1.getTelemetryOptions)();
+    if (telemetryOptions.enableContentRecording) {
+        eventBody.content = getMessageContent(messageAttributes.content);
+    }
+    eventBody.role = messageAttributes.role;
+    if (messageAttributes.attachments) {
+        eventBody.attachments = messageAttributes.attachments.map((attachment) => {
+            return {
+                id: attachment.file_id,
+                tools: attachment.tools.map((tool) => tool.type),
+            };
+        });
+    }
+    const threadId = messageAttributes.thread_id;
+    const agentId = (_a = messageAttributes.assistant_id) !== null && _a !== void 0 ? _a : undefined;
+    const threadRunId = messageAttributes.run_id;
+    const messageStatus = messageAttributes.status;
+    const messageId = messageAttributes.id;
+    const incompleteDetails = messageAttributes.incomplete_details;
+    if (incompleteDetails) {
+        eventBody.incomplete_details = incompleteDetails;
+    }
+    const usagePromptTokens = usage === null || usage === void 0 ? void 0 : usage.prompt_tokens;
+    const usageCompletionTokens = usage === null || usage === void 0 ? void 0 : usage.completion_tokens;
+    const attributes = {
+        eventContent: JSON.stringify(eventBody),
+        threadId,
+        agentId,
+        threadRunId,
+        messageStatus,
+        messageId,
+        usagePromptTokens,
+        usageCompletionTokens,
+        genAiSystem: tracing_js_1.TracingAttributes.AZ_AI_AGENT_SYSTEM,
+    };
+    tracing_js_1.TracingUtility.addSpanEvent(span, `gen_ai.${messageAttributes.role}.message`, attributes);
+}
+/**
+ * Adds an instruction event to the span.
+ * @param span - The span to add the event to.
+ * @param instructionAttributes - The attributes of the instruction event.
+ */
+function addInstructionsEvent(span, instructionAttributes) {
+    const eventBody = {};
+    if (instructionAttributes.instructions || instructionAttributes.additional_instructions) {
+        eventBody.content =
+            instructionAttributes.instructions && instructionAttributes.additional_instructions
+                ? `${instructionAttributes.instructions} ${instructionAttributes.additional_instructions}`
+                : instructionAttributes.instructions || instructionAttributes.additional_instructions;
+    }
+    const attributes = {
+        eventContent: JSON.stringify(eventBody),
+        threadId: instructionAttributes.threadId,
+        agentId: instructionAttributes.agentId,
+        genAiSystem: tracing_js_1.TracingAttributes.AZ_AI_AGENT_SYSTEM,
+    };
+    tracing_js_1.TracingUtility.addSpanEvent(span, "gen_ai.system.message", attributes);
+}
+/**
+ * Formats the agent API response.
+ * @param responseFormat - The response format option.
+ * @returns The formatted response as a string, or null/undefined.
+ */
+function formatAgentApiResponse(responseFormat) {
+    var _a;
+    if (typeof responseFormat === "string" ||
+        responseFormat === undefined ||
+        responseFormat === null) {
+        return responseFormat !== null && responseFormat !== void 0 ? responseFormat : undefined;
+    }
+    if (responseFormat.type) {
+        return (_a = responseFormat.type) !== null && _a !== void 0 ? _a : undefined;
+    }
+    return undefined;
+}
+/**
+ * Adds a tool messages event to the span
+ * @param span - The span to add the event to.
+ * @param tool_outputs - List of tool oupts
+ */
+function addToolMessagesEvent(span, tool_outputs) {
+    tool_outputs.forEach((tool_output) => {
+        const eventBody = { content: tool_output.output, id: tool_output.tool_call_id };
+        tracing_js_1.TracingUtility.addSpanEvent(span, "gen_ai.tool.message", {
+            eventContent: JSON.stringify(eventBody),
+            genAiSystem: tracing_js_1.TracingAttributes.AZ_AI_AGENT_SYSTEM,
+        });
+    });
+}
+function getMessageContent(messageContent) {
+    if (!Array.isArray(messageContent)) {
+        return messageContent;
+    }
+    const contentBody = {};
+    messageContent.forEach((content) => {
+        const typedContent = content.type;
+        const { value, annotations } = content[typedContent];
+        contentBody[typedContent] = { value, annotations };
+    });
+    return contentBody;
+}
+//# sourceMappingURL=traceUtility.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"traceUtility.js","sourceRoot":"","sources":["../../../src/agents/traceUtility.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;AAelC,wDAUC;AACD,oDAMC;AAED,8DAKC;AAOD,0CA0CC;AAOD,oDAuBC;AAOD,wDAcC;AAOD,oDAQC;AA9ID,8CAAkE;AAClE,4DAAgE;AAEhE,SAAgB,sBAAsB,CACpC,IAAU,EACV,OAAgB;;IAEhB,MAAM,gBAAgB,GAAG,OAAO,CAAC,uBAAuB,IAAI,EAAE,CAAC;IAC/D,2BAAc,CAAC,iBAAiB,CAC9B,IAAI,EACJ,CAAA,MAAA,OAAO,CAAC,uBAAuB,0CAAE,aAAa,KAAI,iBAAiB,EACnE,yBAAyB,CAAC,gBAAgB,CAAC,CAC5C,CAAC;AACJ,CAAC;AACD,SAAgB,oBAAoB,CAClC,IAAU,EACV,QAAiB;IAEjB,MAAM,gBAAgB,GAAG,EAAE,CAAC;IAC5B,2BAAc,CAAC,oBAAoB,CAAC,IAAI,EAAE,yBAAyB,CAAC,gBAAgB,CAAC,CAAC,CAAC;AACzF,CAAC;AAED,SAAgB,yBAAyB,CACvC,gBAAgE;IAEhE,gBAAgB,CAAC,WAAW,GAAG,8BAAiB,CAAC,kBAAkB,CAAC;IACpE,OAAO,gBAAgB,CAAC;AAC1B,CAAC;AAED;;;;GAIG;AACH,SAAgB,eAAe,CAC7B,IAAU,EACV,iBAAuD,EACvD,KAAoC;;IAEpC,MAAM,SAAS,GAA4B,EAAE,CAAC;IAC9C,MAAM,gBAAgB,GAAG,IAAA,kCAAmB,GAAE,CAAC;IAC/C,IAAI,gBAAgB,CAAC,sBAAsB,EAAE,CAAC;QAC5C,SAAS,CAAC,OAAO,GAAG,iBAAiB,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;IACnE,CAAC;IACD,SAAS,CAAC,IAAI,GAAG,iBAAiB,CAAC,IAAI,CAAC;IACxC,IAAI,iBAAiB,CAAC,WAAW,EAAE,CAAC;QAClC,SAAS,CAAC,WAAW,GAAG,iBAAiB,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE;YACvE,OAAO;gBACL,EAAE,EAAE,UAAU,CAAC,OAAO;gBACtB,KAAK,EAAE,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;aACjD,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC;IACD,MAAM,QAAQ,GAAI,iBAAmC,CAAC,SAAS,CAAC;IAChE,MAAM,OAAO,GAAG,MAAC,iBAAmC,CAAC,YAAY,mCAAI,SAAS,CAAC;IAC/E,MAAM,WAAW,GAAI,iBAAmC,CAAC,MAAM,CAAC;IAChE,MAAM,aAAa,GAAI,iBAAmC,CAAC,MAAM,CAAC;IAClE,MAAM,SAAS,GAAI,iBAAmC,CAAC,EAAE,CAAC;IAC1D,MAAM,iBAAiB,GAAI,iBAAmC,CAAC,kBAAkB,CAAC;IAClF,IAAI,iBAAiB,EAAE,CAAC;QACtB,SAAS,CAAC,kBAAkB,GAAG,iBAAiB,CAAC;IACnD,CAAC;IACD,MAAM,iBAAiB,GAAG,KAAK,aAAL,KAAK,uBAAL,KAAK,CAAE,aAAa,CAAC;IAC/C,MAAM,qBAAqB,GAAG,KAAK,aAAL,KAAK,uBAAL,KAAK,CAAE,iBAAiB,CAAC;IACvD,MAAM,UAAU,GAAG;QACjB,YAAY,EAAE,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC;QACvC,QAAQ;QACR,OAAO;QACP,WAAW;QACX,aAAa;QACb,SAAS;QACT,iBAAiB;QACjB,qBAAqB;QACrB,WAAW,EAAE,8BAAiB,CAAC,kBAAkB;KAClD,CAAC;IACF,2BAAc,CAAC,YAAY,CAAC,IAAI,EAAE,UAAU,iBAAiB,CAAC,IAAI,UAAU,EAAE,UAAU,CAAC,CAAC;AAC5F,CAAC;AAED;;;;GAIG;AACH,SAAgB,oBAAoB,CAClC,IAAU,EACV,qBAKC;IAED,MAAM,SAAS,GAA4B,EAAE,CAAC;IAC9C,IAAI,qBAAqB,CAAC,YAAY,IAAI,qBAAqB,CAAC,uBAAuB,EAAE,CAAC;QACxF,SAAS,CAAC,OAAO;YACf,qBAAqB,CAAC,YAAY,IAAI,qBAAqB,CAAC,uBAAuB;gBACjF,CAAC,CAAC,GAAG,qBAAqB,CAAC,YAAY,IAAI,qBAAqB,CAAC,uBAAuB,EAAE;gBAC1F,CAAC,CAAC,qBAAqB,CAAC,YAAY,IAAI,qBAAqB,CAAC,uBAAuB,CAAC;IAC5F,CAAC;IACD,MAAM,UAAU,GAAG;QACjB,YAAY,EAAE,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC;QACvC,QAAQ,EAAE,qBAAqB,CAAC,QAAQ;QACxC,OAAO,EAAE,qBAAqB,CAAC,OAAO;QACtC,WAAW,EAAE,8BAAiB,CAAC,kBAAkB;KAClD,CAAC;IACF,2BAAc,CAAC,YAAY,CAAC,IAAI,EAAE,uBAAuB,EAAE,UAAU,CAAC,CAAC;AACzE,CAAC;AAED;;;;GAIG;AACH,SAAgB,sBAAsB,CACpC,cAAgE;;IAEhE,IACE,OAAO,cAAc,KAAK,QAAQ;QAClC,cAAc,KAAK,SAAS;QAC5B,cAAc,KAAK,IAAI,EACvB,CAAC;QACD,OAAO,cAAc,aAAd,cAAc,cAAd,cAAc,GAAI,SAAS,CAAC;IACrC,CAAC;IACD,IAAK,cAA0C,CAAC,IAAI,EAAE,CAAC;QACrD,OAAO,MAAC,cAA0C,CAAC,IAAI,mCAAI,SAAS,CAAC;IACvE,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAED;;;;GAIG;AACH,SAAgB,oBAAoB,CAAC,IAAU,EAAE,YAA+B;IAC9E,YAAY,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,EAAE;QACnC,MAAM,SAAS,GAAG,EAAE,OAAO,EAAE,WAAW,CAAC,MAAM,EAAE,EAAE,EAAE,WAAW,CAAC,YAAY,EAAE,CAAC;QAChF,2BAAc,CAAC,YAAY,CAAC,IAAI,EAAE,qBAAqB,EAAE;YACvD,YAAY,EAAE,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC;YACvC,WAAW,EAAE,8BAAiB,CAAC,kBAAkB;SAClD,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,iBAAiB,CAAC,cAAyC;IAElE,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,cAAc,CAAC,EAAE,CAAC;QACnC,OAAO,cAAc,CAAC;IACxB,CAAC;IACD,MAAM,WAAW,GAA2B,EAAE,CAAC;IAC/C,cAAc,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QACjC,MAAM,YAAY,GAAG,OAAO,CAAC,IAAI,CAAC;QAClC,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,GAAI,OAAkC,CAAC,YAAY,CAAC,CAAC;QACjF,WAAW,CAAC,YAAY,CAAC,GAAG,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC;IACrD,CAAC,CAAC,CAAC;IACH,OAAO,WAAW,CAAC;AACrB,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport type {\n  AgentsApiResponseFormat,\n  AgentsApiResponseFormatOption,\n  MessageContent,\n  ThreadMessage,\n  ThreadMessageOptions,\n  ToolOutput,\n} from \"../generated/src/models.js\";\nimport type { RunStepCompletionUsageOutput } from \"../generated/src/outputModels.js\";\nimport type { OptionsWithTracing, Span, TracingAttributeOptions } from \"../tracing.js\";\nimport { TracingAttributes, TracingUtility } from \"../tracing.js\";\nimport { getTelemetryOptions } from \"../telemetry/telemetry.js\";\n\nexport function traceStartAgentGeneric<Options extends OptionsWithTracing>(\n  span: Span,\n  options: Options,\n): void {\n  const attributeOptions = options.tracingAttributeOptions || {};\n  TracingUtility.setSpanAttributes(\n    span,\n    options.tracingAttributeOptions?.operationName || \"Agent_Operation\",\n    UpdateWithAgentAttributes(attributeOptions),\n  );\n}\nexport function traceEndAgentGeneric<Options extends OptionsWithTracing>(\n  span: Span,\n  _options: Options,\n): void {\n  const attributeOptions = {};\n  TracingUtility.updateSpanAttributes(span, UpdateWithAgentAttributes(attributeOptions));\n}\n\nexport function UpdateWithAgentAttributes(\n  attributeOptions: Omit<TracingAttributeOptions, \"operationName\">,\n): Omit<TracingAttributeOptions, \"operationName\"> {\n  attributeOptions.genAiSystem = TracingAttributes.AZ_AI_AGENT_SYSTEM;\n  return attributeOptions;\n}\n\n/**\n * Adds a message event to the span.\n * @param span - The span to add the event to.\n * @param messageAttributes - The attributes of the message event.\n */\nexport function addMessageEvent(\n  span: Span,\n  messageAttributes: ThreadMessageOptions | ThreadMessage,\n  usage?: RunStepCompletionUsageOutput,\n): void {\n  const eventBody: Record<string, unknown> = {};\n  const telemetryOptions = getTelemetryOptions();\n  if (telemetryOptions.enableContentRecording) {\n    eventBody.content = getMessageContent(messageAttributes.content);\n  }\n  eventBody.role = messageAttributes.role;\n  if (messageAttributes.attachments) {\n    eventBody.attachments = messageAttributes.attachments.map((attachment) => {\n      return {\n        id: attachment.file_id,\n        tools: attachment.tools.map((tool) => tool.type),\n      };\n    });\n  }\n  const threadId = (messageAttributes as ThreadMessage).thread_id;\n  const agentId = (messageAttributes as ThreadMessage).assistant_id ?? undefined;\n  const threadRunId = (messageAttributes as ThreadMessage).run_id;\n  const messageStatus = (messageAttributes as ThreadMessage).status;\n  const messageId = (messageAttributes as ThreadMessage).id;\n  const incompleteDetails = (messageAttributes as ThreadMessage).incomplete_details;\n  if (incompleteDetails) {\n    eventBody.incomplete_details = incompleteDetails;\n  }\n  const usagePromptTokens = usage?.prompt_tokens;\n  const usageCompletionTokens = usage?.completion_tokens;\n  const attributes = {\n    eventContent: JSON.stringify(eventBody),\n    threadId,\n    agentId,\n    threadRunId,\n    messageStatus,\n    messageId,\n    usagePromptTokens,\n    usageCompletionTokens,\n    genAiSystem: TracingAttributes.AZ_AI_AGENT_SYSTEM,\n  };\n  TracingUtility.addSpanEvent(span, `gen_ai.${messageAttributes.role}.message`, attributes);\n}\n\n/**\n * Adds an instruction event to the span.\n * @param span - The span to add the event to.\n * @param instructionAttributes - The attributes of the instruction event.\n */\nexport function addInstructionsEvent(\n  span: Span,\n  instructionAttributes: {\n    instructions?: string | null;\n    additional_instructions?: string | null;\n    threadId?: string;\n    agentId?: string;\n  },\n): void {\n  const eventBody: Record<string, unknown> = {};\n  if (instructionAttributes.instructions || instructionAttributes.additional_instructions) {\n    eventBody.content =\n      instructionAttributes.instructions && instructionAttributes.additional_instructions\n        ? `${instructionAttributes.instructions} ${instructionAttributes.additional_instructions}`\n        : instructionAttributes.instructions || instructionAttributes.additional_instructions;\n  }\n  const attributes = {\n    eventContent: JSON.stringify(eventBody),\n    threadId: instructionAttributes.threadId,\n    agentId: instructionAttributes.agentId,\n    genAiSystem: TracingAttributes.AZ_AI_AGENT_SYSTEM,\n  };\n  TracingUtility.addSpanEvent(span, \"gen_ai.system.message\", attributes);\n}\n\n/**\n * Formats the agent API response.\n * @param responseFormat - The response format option.\n * @returns The formatted response as a string, or null/undefined.\n */\nexport function formatAgentApiResponse(\n  responseFormat: AgentsApiResponseFormatOption | null | undefined,\n): string | undefined {\n  if (\n    typeof responseFormat === \"string\" ||\n    responseFormat === undefined ||\n    responseFormat === null\n  ) {\n    return responseFormat ?? undefined;\n  }\n  if ((responseFormat as AgentsApiResponseFormat).type) {\n    return (responseFormat as AgentsApiResponseFormat).type ?? undefined;\n  }\n  return undefined;\n}\n\n/**\n * Adds a tool messages event to the span\n * @param span - The span to add the event to.\n * @param tool_outputs - List of tool oupts\n */\nexport function addToolMessagesEvent(span: Span, tool_outputs: Array<ToolOutput>): void {\n  tool_outputs.forEach((tool_output) => {\n    const eventBody = { content: tool_output.output, id: tool_output.tool_call_id };\n    TracingUtility.addSpanEvent(span, \"gen_ai.tool.message\", {\n      eventContent: JSON.stringify(eventBody),\n      genAiSystem: TracingAttributes.AZ_AI_AGENT_SYSTEM,\n    });\n  });\n}\n\nfunction getMessageContent(messageContent: string | MessageContent[]): string | {} {\n  type MessageContentExtended = MessageContent & { [key: string]: any };\n  if (!Array.isArray(messageContent)) {\n    return messageContent;\n  }\n  const contentBody: { [key: string]: any } = {};\n  messageContent.forEach((content) => {\n    const typedContent = content.type;\n    const { value, annotations } = (content as MessageContentExtended)[typedContent];\n    contentBody[typedContent] = { value, annotations };\n  });\n  return contentBody;\n}\n"]}