@mastra/openai 0.1.0-alpha.1

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
package/dist/index.cjs ADDED
@@ -0,0 +1,1076 @@
1
+ 'use strict';
2
+
3
+ var crypto = require('crypto');
4
+ var web = require('stream/web');
5
+ var agent = require('@mastra/core/agent');
6
+ var requestContext = require('@mastra/core/request-context');
7
+ var stream = require('@mastra/core/stream');
8
+ var agents = require('@openai/agents');
9
+ var messageList = require('@mastra/core/agent/message-list');
10
+ var observability = require('@mastra/core/observability');
11
+
12
+ // src/index.ts
13
+ function createNoopModel({ modelId, provider }) {
14
+ return {
15
+ modelId,
16
+ provider,
17
+ specificationVersion: "v3",
18
+ supportedUrls: {},
19
+ doGenerate: async () => createNoopStreamResult(),
20
+ doStream: async () => createNoopStreamResult()
21
+ };
22
+ }
23
+ function createNoopStreamResult() {
24
+ return {
25
+ stream: new web.ReadableStream({
26
+ start: (controller) => controller.close()
27
+ })
28
+ };
29
+ }
30
+ function createCompletedMastraStream({
31
+ runId,
32
+ prompt,
33
+ text,
34
+ responseId,
35
+ modelId,
36
+ usage,
37
+ providerMetadata,
38
+ costContext
39
+ }) {
40
+ return new web.ReadableStream({
41
+ start(controller) {
42
+ const textId = crypto.randomUUID();
43
+ enqueueStartChunks(controller, {
44
+ runId,
45
+ prompt,
46
+ textId,
47
+ responseId,
48
+ modelId,
49
+ providerMetadata
50
+ });
51
+ if (text) {
52
+ enqueueTextDelta(controller, runId, textId, text);
53
+ }
54
+ enqueueFinishChunks(controller, {
55
+ runId,
56
+ prompt,
57
+ textId,
58
+ text,
59
+ responseId,
60
+ modelId,
61
+ usage,
62
+ providerMetadata,
63
+ costContext
64
+ });
65
+ controller.close();
66
+ }
67
+ });
68
+ }
69
+ function createMastraOutput({
70
+ messages,
71
+ runId,
72
+ modelId,
73
+ provider,
74
+ stream: stream$1,
75
+ options
76
+ }) {
77
+ const messageList$1 = new messageList.MessageList();
78
+ messageList$1.add(messages, "input");
79
+ messageList$1.add([{ role: "assistant", content: "" }], "response");
80
+ return new stream.MastraModelOutput({
81
+ model: {
82
+ modelId,
83
+ provider,
84
+ version: "v3"
85
+ },
86
+ stream: stream$1,
87
+ messageList: messageList$1,
88
+ messageId: crypto.randomUUID(),
89
+ options: {
90
+ ...options,
91
+ runId
92
+ }
93
+ });
94
+ }
95
+ function toFullOutput({
96
+ messages,
97
+ runId,
98
+ provider,
99
+ result,
100
+ options
101
+ }) {
102
+ const text = result.content.map((part) => part.text).join("");
103
+ const stream = createCompletedMastraStream({
104
+ runId,
105
+ prompt: promptToText(messages),
106
+ text,
107
+ responseId: result.response.id,
108
+ modelId: result.response.modelId,
109
+ usage: toLanguageModelUsage(result.usage),
110
+ providerMetadata: result.providerMetadata,
111
+ costContext: result.costContext
112
+ });
113
+ return createMastraOutput({
114
+ messages,
115
+ runId,
116
+ modelId: result.response.modelId,
117
+ provider,
118
+ stream,
119
+ options
120
+ }).getFullOutput();
121
+ }
122
+ function createSDKAgentTelemetry({
123
+ agentId,
124
+ agentName,
125
+ provider,
126
+ modelId,
127
+ messages,
128
+ prompt,
129
+ runId,
130
+ streaming,
131
+ method,
132
+ requestContext,
133
+ instructions,
134
+ maxSteps,
135
+ tracingOptions,
136
+ tracingContext,
137
+ onFinish,
138
+ onStepFinish,
139
+ mastra
140
+ }) {
141
+ const agentSpan = observability.getOrCreateSpan({
142
+ type: observability.SpanType.AGENT_RUN,
143
+ name: `agent run: '${agentId}'`,
144
+ entityType: observability.EntityType.AGENT,
145
+ entityId: agentId,
146
+ entityName: agentName,
147
+ input: messages,
148
+ attributes: {
149
+ prompt,
150
+ instructions,
151
+ maxSteps
152
+ },
153
+ metadata: {
154
+ runId,
155
+ sdkAgent: true,
156
+ sdkProvider: provider,
157
+ sdkMethod: method
158
+ },
159
+ tracingOptions,
160
+ tracingContext,
161
+ requestContext,
162
+ mastra
163
+ });
164
+ const modelSpan = agentSpan?.createChildSpan({
165
+ type: observability.SpanType.MODEL_GENERATION,
166
+ name: `llm: '${modelId}'`,
167
+ input: {
168
+ messages
169
+ },
170
+ attributes: {
171
+ model: modelId,
172
+ provider,
173
+ streaming
174
+ },
175
+ metadata: {
176
+ runId,
177
+ sdkAgent: true,
178
+ sdkProvider: provider,
179
+ sdkMethod: method
180
+ },
181
+ requestContext
182
+ });
183
+ const modelSpanTracker = getModelSpanTracker(modelSpan);
184
+ const toolSpans = /* @__PURE__ */ new Map();
185
+ let ended = false;
186
+ const startToolCall = ({ toolCallId, toolName, input }) => {
187
+ if (toolSpans.has(toolCallId)) {
188
+ return;
189
+ }
190
+ const parentSpan = agentSpan ?? modelSpan;
191
+ if (!parentSpan) {
192
+ return;
193
+ }
194
+ const mcp = parseMcpToolName(toolName);
195
+ const span = mcp ? parentSpan.createChildSpan({
196
+ type: observability.SpanType.MCP_TOOL_CALL,
197
+ name: `mcp_tool: '${toolName}' on '${mcp.serverName}'`,
198
+ input,
199
+ entityType: observability.EntityType.TOOL,
200
+ entityId: toolName,
201
+ entityName: toolName,
202
+ attributes: {
203
+ mcpServer: mcp.serverName
204
+ },
205
+ metadata: {
206
+ runId,
207
+ sdkAgent: true,
208
+ sdkProvider: provider,
209
+ sdkMethod: method,
210
+ toolCallId
211
+ },
212
+ requestContext
213
+ }) : parentSpan.createChildSpan({
214
+ type: observability.SpanType.TOOL_CALL,
215
+ name: `tool: '${toolName}'`,
216
+ input,
217
+ entityType: observability.EntityType.TOOL,
218
+ entityId: toolName,
219
+ entityName: toolName,
220
+ attributes: {
221
+ toolType: "tool"
222
+ },
223
+ metadata: {
224
+ runId,
225
+ sdkAgent: true,
226
+ sdkProvider: provider,
227
+ sdkMethod: method,
228
+ toolCallId
229
+ },
230
+ requestContext
231
+ });
232
+ toolSpans.set(toolCallId, span);
233
+ };
234
+ const endToolCall = ({ toolCallId, output, isError }) => {
235
+ const span = toolSpans.get(toolCallId);
236
+ if (!span) {
237
+ return;
238
+ }
239
+ toolSpans.delete(toolCallId);
240
+ if (isError) {
241
+ span.error({
242
+ error: output instanceof Error ? output : new Error(typeof output === "string" ? output : "SDK tool call failed"),
243
+ attributes: { success: false }
244
+ });
245
+ return;
246
+ }
247
+ span.end({
248
+ output,
249
+ attributes: { success: true }
250
+ });
251
+ };
252
+ const closeOpenToolSpans = (success, error) => {
253
+ for (const [toolCallId, span] of toolSpans) {
254
+ toolSpans.delete(toolCallId);
255
+ if (success) {
256
+ span.end({ attributes: { success: true } });
257
+ continue;
258
+ }
259
+ const normalized = error instanceof Error ? error : new Error(String(error ?? "SDK agent run failed"));
260
+ span.error({ error: normalized, attributes: { success: false } });
261
+ }
262
+ };
263
+ const endModel = ({
264
+ text,
265
+ usage,
266
+ providerMetadata,
267
+ finishReason = "stop",
268
+ responseId,
269
+ responseModel,
270
+ costContext
271
+ }) => {
272
+ if (modelSpanTracker) {
273
+ modelSpanTracker.endGeneration({
274
+ output: {
275
+ text
276
+ },
277
+ attributes: {
278
+ finishReason,
279
+ responseId,
280
+ responseModel,
281
+ costContext
282
+ },
283
+ usage,
284
+ providerMetadata
285
+ });
286
+ return;
287
+ }
288
+ modelSpan?.end({
289
+ output: {
290
+ text
291
+ },
292
+ attributes: {
293
+ finishReason,
294
+ responseId,
295
+ responseModel,
296
+ usage: usage ? toUsageStats(usage) : void 0,
297
+ costContext
298
+ }
299
+ });
300
+ };
301
+ const end = (result) => {
302
+ if (ended) {
303
+ return;
304
+ }
305
+ ended = true;
306
+ closeOpenToolSpans(true);
307
+ endModel(result);
308
+ agentSpan?.end({
309
+ output: {
310
+ text: result.text
311
+ }
312
+ });
313
+ };
314
+ const fail = (error) => {
315
+ if (ended) {
316
+ return;
317
+ }
318
+ ended = true;
319
+ const normalized = error instanceof Error ? error : new Error(String(error));
320
+ closeOpenToolSpans(false, normalized);
321
+ if (modelSpanTracker) {
322
+ modelSpanTracker.reportGenerationError({ error: normalized });
323
+ } else {
324
+ modelSpan?.error({ error: normalized });
325
+ }
326
+ agentSpan?.error({ error: normalized });
327
+ };
328
+ return {
329
+ execute: (fn) => observability.executeWithContext({ span: modelSpan ?? agentSpan, fn }),
330
+ endGenerate(result) {
331
+ end({
332
+ text: result.content.map((part) => part.text).join(""),
333
+ usage: toLanguageModelUsage(result.usage),
334
+ providerMetadata: result.providerMetadata,
335
+ finishReason: result.finishReason.unified,
336
+ responseId: result.response.id,
337
+ responseModel: result.response.modelId,
338
+ costContext: result.costContext
339
+ });
340
+ },
341
+ fail,
342
+ startToolCall,
343
+ endToolCall,
344
+ wrapStream(stream) {
345
+ const trackedStream = modelSpanTracker?.wrapStream(stream) ?? stream;
346
+ return wrapStreamForAgentSpan(trackedStream, {
347
+ end,
348
+ fail
349
+ });
350
+ },
351
+ outputOptions() {
352
+ return {
353
+ onFinish,
354
+ onStepFinish,
355
+ requestContext,
356
+ tracingContext: agentSpan ? { currentSpan: agentSpan } : tracingContext
357
+ };
358
+ }
359
+ };
360
+ }
361
+ function parseMcpToolName(toolName) {
362
+ const match = /^mcp__([^_].*?)__(.+)$/.exec(toolName);
363
+ if (!match?.[1] || !match[2]) {
364
+ return void 0;
365
+ }
366
+ return {
367
+ serverName: match[1],
368
+ toolName: match[2]
369
+ };
370
+ }
371
+ function getModelSpanTracker(modelSpan) {
372
+ if (!modelSpan || !("createTracker" in modelSpan)) {
373
+ return void 0;
374
+ }
375
+ return modelSpan.createTracker();
376
+ }
377
+ function wrapStreamForAgentSpan(stream, telemetry) {
378
+ let text = "";
379
+ return stream.pipeThrough(
380
+ new web.TransformStream({
381
+ transform(chunk, controller) {
382
+ if (chunk.type === "text-delta") {
383
+ text += chunk.payload.text;
384
+ }
385
+ if (chunk.type === "finish") {
386
+ telemetry.end({
387
+ text,
388
+ usage: chunk.payload.output.usage,
389
+ providerMetadata: chunk.payload.providerMetadata,
390
+ finishReason: chunk.payload.stepResult.reason,
391
+ responseId: chunk.payload.response?.id,
392
+ responseModel: chunk.payload.response?.modelId,
393
+ costContext: getCostContext(chunk.payload.metadata?.costContext)
394
+ });
395
+ }
396
+ if (chunk.type === "error") {
397
+ telemetry.fail(chunk.payload.error);
398
+ }
399
+ controller.enqueue(chunk);
400
+ },
401
+ flush() {
402
+ telemetry.end({ text });
403
+ }
404
+ })
405
+ );
406
+ }
407
+ function toUsageStats(usage) {
408
+ return {
409
+ inputTokens: usage.inputTokens,
410
+ outputTokens: usage.outputTokens,
411
+ inputDetails: {
412
+ cacheRead: usage.cachedInputTokens,
413
+ cacheWrite: usage.cacheCreationInputTokens
414
+ },
415
+ outputDetails: {
416
+ text: usage.outputTokens,
417
+ reasoning: usage.reasoningTokens
418
+ }
419
+ };
420
+ }
421
+ function getCostContext(value) {
422
+ if (!value || typeof value !== "object") {
423
+ return void 0;
424
+ }
425
+ return value;
426
+ }
427
+ function enqueueStartChunks(controller, {
428
+ runId,
429
+ prompt,
430
+ textId,
431
+ responseId,
432
+ modelId,
433
+ providerMetadata
434
+ }) {
435
+ controller.enqueue({
436
+ type: "start",
437
+ runId,
438
+ from: stream.ChunkFrom.AGENT,
439
+ payload: {}
440
+ });
441
+ controller.enqueue({
442
+ type: "step-start",
443
+ runId,
444
+ from: stream.ChunkFrom.AGENT,
445
+ payload: {
446
+ request: { body: prompt }
447
+ }
448
+ });
449
+ controller.enqueue({
450
+ type: "response-metadata",
451
+ runId,
452
+ from: stream.ChunkFrom.AGENT,
453
+ payload: {
454
+ ...responseId ? { id: responseId } : {},
455
+ modelId,
456
+ timestamp: (/* @__PURE__ */ new Date()).toISOString()
457
+ }
458
+ });
459
+ controller.enqueue({
460
+ type: "text-start",
461
+ runId,
462
+ from: stream.ChunkFrom.AGENT,
463
+ payload: {
464
+ id: textId,
465
+ providerMetadata
466
+ }
467
+ });
468
+ }
469
+ function enqueueTextDelta(controller, runId, textId, text) {
470
+ controller.enqueue({
471
+ type: "text-delta",
472
+ runId,
473
+ from: stream.ChunkFrom.AGENT,
474
+ payload: {
475
+ id: textId,
476
+ text
477
+ }
478
+ });
479
+ }
480
+ function enqueueFinishChunks(controller, {
481
+ runId,
482
+ prompt,
483
+ textId,
484
+ text,
485
+ responseId,
486
+ modelId,
487
+ usage,
488
+ providerMetadata,
489
+ costContext
490
+ }) {
491
+ const timestamp = /* @__PURE__ */ new Date();
492
+ const response = {
493
+ ...responseId ? { id: responseId } : {},
494
+ modelId,
495
+ timestamp
496
+ };
497
+ const metadata = {
498
+ providerMetadata,
499
+ costContext,
500
+ request: { body: prompt },
501
+ modelId,
502
+ timestamp
503
+ };
504
+ controller.enqueue({
505
+ type: "text-end",
506
+ runId,
507
+ from: stream.ChunkFrom.AGENT,
508
+ payload: {
509
+ id: textId,
510
+ providerMetadata
511
+ }
512
+ });
513
+ controller.enqueue({
514
+ type: "step-finish",
515
+ runId,
516
+ from: stream.ChunkFrom.AGENT,
517
+ payload: {
518
+ ...responseId ? { id: responseId } : {},
519
+ providerMetadata,
520
+ totalUsage: usage,
521
+ response,
522
+ stepResult: {
523
+ reason: "stop",
524
+ warnings: []
525
+ },
526
+ output: {
527
+ text,
528
+ usage,
529
+ steps: []
530
+ },
531
+ metadata
532
+ }
533
+ });
534
+ controller.enqueue({
535
+ type: "finish",
536
+ runId,
537
+ from: stream.ChunkFrom.AGENT,
538
+ payload: {
539
+ stepResult: {
540
+ reason: "stop",
541
+ warnings: []
542
+ },
543
+ output: {
544
+ usage,
545
+ steps: []
546
+ },
547
+ metadata,
548
+ providerMetadata,
549
+ messages: {
550
+ all: [],
551
+ user: [],
552
+ nonUser: []
553
+ },
554
+ response
555
+ }
556
+ });
557
+ }
558
+ function toLanguageModelUsage(usage) {
559
+ const inputTokens = usage.inputTokens.total ?? 0;
560
+ const outputTokens = usage.outputTokens.total ?? 0;
561
+ return {
562
+ inputTokens,
563
+ outputTokens,
564
+ totalTokens: inputTokens + outputTokens,
565
+ cachedInputTokens: usage.inputTokens.cacheRead,
566
+ cacheCreationInputTokens: usage.inputTokens.cacheWrite,
567
+ reasoningTokens: usage.outputTokens.reasoning,
568
+ raw: usage
569
+ };
570
+ }
571
+ function createProviderMetadata(provider, metadata) {
572
+ return {
573
+ [provider]: toJsonRecord(metadata)
574
+ };
575
+ }
576
+ function toJsonRecord(record) {
577
+ return Object.fromEntries(
578
+ Object.entries(record).filter((entry) => entry[1] !== void 0).map(([key, value]) => [key, toJsonValue(value)])
579
+ );
580
+ }
581
+ function toJsonValue(value) {
582
+ if (value === null || typeof value === "string" || typeof value === "number" || typeof value === "boolean") {
583
+ return value;
584
+ }
585
+ if (Array.isArray(value)) {
586
+ return value.filter((item) => item !== void 0).map(toJsonValue);
587
+ }
588
+ if (value instanceof Date) {
589
+ return value.toISOString();
590
+ }
591
+ if (typeof value === "object") {
592
+ return toJsonRecord(value);
593
+ }
594
+ return String(value);
595
+ }
596
+ function promptToText(prompt) {
597
+ if (typeof prompt === "string") {
598
+ return prompt;
599
+ }
600
+ if (Array.isArray(prompt)) {
601
+ return prompt.map(promptToText).filter(Boolean).join("\n");
602
+ }
603
+ if (!prompt || typeof prompt !== "object") {
604
+ return "";
605
+ }
606
+ const record = prompt;
607
+ if (typeof record.text === "string") {
608
+ return record.text;
609
+ }
610
+ if (typeof record.content === "string") {
611
+ return record.content;
612
+ }
613
+ if (record.content) {
614
+ return promptToText(record.content);
615
+ }
616
+ return "";
617
+ }
618
+ function sumDefined(...values) {
619
+ const defined = values.filter((value) => typeof value === "number");
620
+ if (defined.length === 0) {
621
+ return void 0;
622
+ }
623
+ return defined.reduce((sum, value) => sum + value, 0);
624
+ }
625
+
626
+ // src/index.ts
627
+ var PROVIDER = "@openai/agents";
628
+ var MODEL_ID = "openai-agents-sdk";
629
+ var OpenAISDKAgent = class extends agent.Agent {
630
+ options;
631
+ #mastra;
632
+ #createdAgent;
633
+ constructor(options) {
634
+ super({
635
+ id: options.id,
636
+ name: options.name ?? options.id,
637
+ description: options.description,
638
+ instructions: "",
639
+ model: createNoopModel({
640
+ modelId: getModelId(options),
641
+ provider: PROVIDER
642
+ })
643
+ });
644
+ this.options = options;
645
+ }
646
+ __registerMastra(mastra) {
647
+ super.__registerMastra(mastra);
648
+ this.#mastra = mastra;
649
+ }
650
+ supportsMemory() {
651
+ return false;
652
+ }
653
+ async generate(messages, options) {
654
+ const prompt = promptToText(messages);
655
+ const runId = options?.runId ?? crypto.randomUUID();
656
+ const sdkAgent = this.resolveOpenAIAgent();
657
+ const modelId = getModelId(this.options, sdkAgent);
658
+ const requestContext$1 = options?.requestContext ?? new requestContext.RequestContext();
659
+ const instructions = options?.instructions ? promptToText(options.instructions) : void 0;
660
+ const telemetry = createSDKAgentTelemetry({
661
+ agentId: this.id,
662
+ agentName: this.name,
663
+ provider: PROVIDER,
664
+ modelId,
665
+ messages,
666
+ prompt,
667
+ runId,
668
+ streaming: false,
669
+ method: "generate",
670
+ requestContext: requestContext$1,
671
+ instructions,
672
+ maxSteps: options?.maxSteps,
673
+ tracingOptions: options?.tracingOptions,
674
+ tracingContext: options?.tracingContext,
675
+ onFinish: options?.onFinish,
676
+ onStepFinish: options?.onStepFinish,
677
+ mastra: this.#mastra
678
+ });
679
+ let result;
680
+ try {
681
+ result = await telemetry.execute(() => runOpenAIGenerate(prompt, sdkAgent, runId, telemetry, options));
682
+ telemetry.endGenerate(result);
683
+ } catch (error) {
684
+ telemetry.fail(error);
685
+ throw error;
686
+ }
687
+ return toFullOutput({
688
+ messages,
689
+ runId,
690
+ provider: PROVIDER,
691
+ result,
692
+ options: telemetry.outputOptions()
693
+ });
694
+ }
695
+ async stream(messages, options) {
696
+ const prompt = promptToText(messages);
697
+ const runId = options?.runId ?? crypto.randomUUID();
698
+ const sdkAgent = this.resolveOpenAIAgent();
699
+ const modelId = getModelId(this.options, sdkAgent);
700
+ const requestContext$1 = options?.requestContext ?? new requestContext.RequestContext();
701
+ const instructions = options?.instructions ? promptToText(options.instructions) : void 0;
702
+ const telemetry = createSDKAgentTelemetry({
703
+ agentId: this.id,
704
+ agentName: this.name,
705
+ provider: PROVIDER,
706
+ modelId,
707
+ messages,
708
+ prompt,
709
+ runId,
710
+ streaming: true,
711
+ method: "stream",
712
+ requestContext: requestContext$1,
713
+ instructions,
714
+ maxSteps: options?.maxSteps,
715
+ tracingOptions: options?.tracingOptions,
716
+ tracingContext: options?.tracingContext,
717
+ onFinish: options?.onFinish,
718
+ onStepFinish: options?.onStepFinish,
719
+ mastra: this.#mastra
720
+ });
721
+ return createMastraOutput({
722
+ messages,
723
+ runId,
724
+ modelId,
725
+ provider: PROVIDER,
726
+ stream: telemetry.wrapStream(runOpenAIAsMastraStream(prompt, sdkAgent, runId, modelId, telemetry, options)),
727
+ options: telemetry.outputOptions()
728
+ });
729
+ }
730
+ resolveOpenAIAgent() {
731
+ this.#createdAgent ??= this.options.agent ?? new agents.Agent(toOpenAIAgentOptions(this.options));
732
+ return this.#createdAgent;
733
+ }
734
+ };
735
+ async function runOpenAIGenerate(prompt, agent, runId, telemetry, options) {
736
+ const result = await agents.run(agent, prompt, {
737
+ stream: false,
738
+ maxTurns: options?.maxSteps,
739
+ signal: getAbortSignal(options)
740
+ });
741
+ recordOpenAIToolTelemetry(result.newItems, telemetry);
742
+ const text = getTextFromFinalOutput(result.finalOutput);
743
+ const responseId = result.lastResponseId;
744
+ const modelId = getModelId(void 0, result.lastAgent ?? agent, result.rawResponses.at(-1));
745
+ const usage = createOpenAIUsageTotals(result.state.usage);
746
+ const providerMetadata = getOpenAIProviderMetadata({
747
+ modelId,
748
+ responseId,
749
+ lastResponseId: result.lastResponseId,
750
+ rawResponseCount: result.rawResponses.length,
751
+ itemCount: result.newItems.length,
752
+ usage
753
+ });
754
+ return {
755
+ content: [{ type: "text", text }],
756
+ finishReason: { unified: "stop", raw: "stop" },
757
+ usage: toV3Usage(usage),
758
+ response: {
759
+ id: responseId,
760
+ modelId,
761
+ timestamp: /* @__PURE__ */ new Date()
762
+ },
763
+ providerMetadata
764
+ };
765
+ }
766
+ function runOpenAIAsMastraStream(prompt, agent, runId, requestedModelId, telemetry, options) {
767
+ return new web.ReadableStream({
768
+ start: async (controller) => {
769
+ const textId = crypto.randomUUID();
770
+ let text = "";
771
+ let responseId;
772
+ let modelId = requestedModelId;
773
+ try {
774
+ const result = await agents.run(agent, prompt, {
775
+ stream: true,
776
+ maxTurns: options?.maxSteps,
777
+ signal: getAbortSignal(options)
778
+ });
779
+ enqueueStartChunks(controller, {
780
+ runId,
781
+ prompt,
782
+ textId,
783
+ responseId,
784
+ modelId
785
+ });
786
+ for await (const event of result) {
787
+ const delta = getTextDelta(event);
788
+ if (delta) {
789
+ text += delta;
790
+ enqueueTextDelta(controller, runId, textId, delta);
791
+ }
792
+ recordOpenAIStreamToolTelemetry(event, telemetry);
793
+ }
794
+ await result.completed;
795
+ responseId = result.lastResponseId ?? responseId;
796
+ modelId = getModelId(void 0, result.lastAgent ?? agent, result.rawResponses.at(-1));
797
+ if (!text) {
798
+ text = getTextFromFinalOutput(result.finalOutput);
799
+ if (text) {
800
+ enqueueTextDelta(controller, runId, textId, text);
801
+ }
802
+ }
803
+ const usage = createOpenAIUsageTotals(result.state.usage);
804
+ const providerMetadata = getOpenAIProviderMetadata({
805
+ modelId,
806
+ responseId,
807
+ lastResponseId: result.lastResponseId,
808
+ rawResponseCount: result.rawResponses.length,
809
+ itemCount: result.newItems.length,
810
+ usage
811
+ });
812
+ enqueueFinishChunks(controller, {
813
+ runId,
814
+ prompt,
815
+ textId,
816
+ text,
817
+ responseId,
818
+ modelId,
819
+ usage: toLanguageModelUsage(toV3Usage(usage)),
820
+ providerMetadata
821
+ });
822
+ controller.close();
823
+ } catch (error) {
824
+ controller.enqueue({
825
+ type: "error",
826
+ runId,
827
+ from: stream.ChunkFrom.AGENT,
828
+ payload: { error }
829
+ });
830
+ controller.close();
831
+ }
832
+ }
833
+ });
834
+ }
835
+ function toOpenAIAgentOptions(options) {
836
+ return {
837
+ name: options.name ?? options.id,
838
+ ...options.sdkOptions
839
+ };
840
+ }
841
+ function getAbortSignal(options) {
842
+ return options?.abortSignal ?? options?.signal;
843
+ }
844
+ function getModelId(options, agent, rawResponse) {
845
+ return getModelNameFromUnknown(rawResponse) ?? getModelNameFromUnknown(agent?.model) ?? getModelNameFromUnknown(options?.sdkOptions?.model) ?? MODEL_ID;
846
+ }
847
+ function getModelNameFromUnknown(value) {
848
+ if (typeof value === "string") {
849
+ return value;
850
+ }
851
+ const record = toRecord(value);
852
+ return getString(record, "model") ?? getString(record, "modelId") ?? getString(record, "modelName") ?? getString(toRecord(record?.providerData), "model");
853
+ }
854
+ function getTextFromFinalOutput(output) {
855
+ if (typeof output === "string") {
856
+ return output;
857
+ }
858
+ if (output === void 0 || output === null) {
859
+ return "";
860
+ }
861
+ return JSON.stringify(output);
862
+ }
863
+ function createOpenAIUsageTotals(usage) {
864
+ const record = toRecord(usage);
865
+ if (!record) {
866
+ return {};
867
+ }
868
+ const inputDetails = getDetailRecords(record.inputTokensDetails);
869
+ const outputDetails = getDetailRecords(record.outputTokensDetails);
870
+ const requestUsageEntries = Array.isArray(record.requestUsageEntries) ? record.requestUsageEntries : void 0;
871
+ return {
872
+ inputTokens: getNumber(record.inputTokens),
873
+ outputTokens: getNumber(record.outputTokens),
874
+ cacheReadInputTokens: sumDetails(inputDetails, "cachedTokens", "cached_tokens", "cacheReadInputTokens"),
875
+ cacheCreationInputTokens: sumDetails(inputDetails, "cacheCreationInputTokens", "cache_creation_input_tokens"),
876
+ reasoningTokens: sumDetails(outputDetails, "reasoningTokens", "reasoning_tokens"),
877
+ requests: getNumber(record.requests),
878
+ requestUsageEntries
879
+ };
880
+ }
881
+ function getDetailRecords(value) {
882
+ if (Array.isArray(value)) {
883
+ return value.filter(isRecord);
884
+ }
885
+ const record = toRecord(value);
886
+ return record ? [record] : [];
887
+ }
888
+ function sumDetails(records, ...keys) {
889
+ let total = 0;
890
+ let found = false;
891
+ for (const record of records) {
892
+ for (const key of keys) {
893
+ const value = getNumber(record[key]);
894
+ if (value !== void 0) {
895
+ total += value;
896
+ found = true;
897
+ }
898
+ }
899
+ }
900
+ return found ? total : void 0;
901
+ }
902
+ function toV3Usage(usage) {
903
+ const totalInputTokens = usage.inputTokens;
904
+ const cacheRead = usage.cacheReadInputTokens;
905
+ const cacheWrite = usage.cacheCreationInputTokens;
906
+ const noCache = totalInputTokens === void 0 ? void 0 : Math.max(totalInputTokens - (sumDefined(cacheRead, cacheWrite) ?? 0), 0);
907
+ const outputTokens = usage.outputTokens;
908
+ const reasoningTokens = usage.reasoningTokens;
909
+ return {
910
+ inputTokens: {
911
+ total: totalInputTokens,
912
+ noCache,
913
+ cacheRead,
914
+ cacheWrite
915
+ },
916
+ outputTokens: {
917
+ total: outputTokens,
918
+ text: outputTokens,
919
+ reasoning: reasoningTokens
920
+ }
921
+ };
922
+ }
923
+ function getOpenAIProviderMetadata({
924
+ modelId,
925
+ responseId,
926
+ lastResponseId,
927
+ rawResponseCount,
928
+ itemCount,
929
+ usage
930
+ }) {
931
+ return createProviderMetadata("openai", {
932
+ model: modelId,
933
+ responseId,
934
+ lastResponseId,
935
+ rawResponseCount,
936
+ itemCount,
937
+ usage
938
+ });
939
+ }
940
+ function recordOpenAIToolTelemetry(items, telemetry) {
941
+ for (const item of items) {
942
+ if (item.type === "tool_call_item") {
943
+ const toolCall = getOpenAIToolCall(item.rawItem);
944
+ if (toolCall) {
945
+ telemetry.startToolCall(toolCall);
946
+ }
947
+ continue;
948
+ }
949
+ if (item.type === "tool_call_output_item") {
950
+ const toolOutput = getOpenAIToolOutput(item.rawItem, item.output);
951
+ if (toolOutput) {
952
+ telemetry.endToolCall(toolOutput);
953
+ }
954
+ }
955
+ }
956
+ }
957
+ function recordOpenAIStreamToolTelemetry(event, telemetry) {
958
+ if (event.type !== "run_item_stream_event") {
959
+ return;
960
+ }
961
+ if (event.name === "tool_called") {
962
+ const toolCall = getOpenAIToolCall(event.item.rawItem);
963
+ if (toolCall) {
964
+ telemetry.startToolCall(toolCall);
965
+ }
966
+ return;
967
+ }
968
+ if (event.name === "tool_output") {
969
+ const toolOutput = getOpenAIToolOutput(event.item.rawItem, getObjectValue(event.item, "output"));
970
+ if (toolOutput) {
971
+ telemetry.endToolCall(toolOutput);
972
+ }
973
+ }
974
+ }
975
+ function getOpenAIToolCall(rawItem) {
976
+ const record = toRecord(rawItem);
977
+ if (!record) {
978
+ return void 0;
979
+ }
980
+ if (record.type === "function_call") {
981
+ const toolCallId = getString(record, "callId") ?? getString(record, "id");
982
+ const toolName = getNamespacedToolName(record);
983
+ if (!toolCallId || !toolName) {
984
+ return void 0;
985
+ }
986
+ return {
987
+ toolCallId,
988
+ toolName,
989
+ input: parseJsonString(record.arguments)
990
+ };
991
+ }
992
+ if (record.type === "hosted_tool_call") {
993
+ const toolCallId = getString(record, "id") ?? getString(record, "name");
994
+ const toolName = getString(record, "name");
995
+ if (!toolCallId || !toolName) {
996
+ return void 0;
997
+ }
998
+ return {
999
+ toolCallId,
1000
+ toolName,
1001
+ input: parseJsonString(record.arguments)
1002
+ };
1003
+ }
1004
+ if (record.type === "shell_call" || record.type === "apply_patch_call") {
1005
+ const toolCallId = getString(record, "callId");
1006
+ if (!toolCallId) {
1007
+ return void 0;
1008
+ }
1009
+ return {
1010
+ toolCallId,
1011
+ toolName: String(record.type),
1012
+ input: record.action ?? record.operation
1013
+ };
1014
+ }
1015
+ return void 0;
1016
+ }
1017
+ function getOpenAIToolOutput(rawItem, fallbackOutput) {
1018
+ const record = toRecord(rawItem);
1019
+ if (!record) {
1020
+ return void 0;
1021
+ }
1022
+ const toolCallId = getString(record, "callId") ?? getString(record, "id");
1023
+ if (!toolCallId) {
1024
+ return void 0;
1025
+ }
1026
+ return {
1027
+ toolCallId,
1028
+ output: record.output ?? fallbackOutput,
1029
+ isError: record.status === "failed" || record.status === "incomplete"
1030
+ };
1031
+ }
1032
+ function getNamespacedToolName(record) {
1033
+ const name = getString(record, "name");
1034
+ const namespace = getString(record, "namespace");
1035
+ if (!name) {
1036
+ return void 0;
1037
+ }
1038
+ return namespace ? `${namespace}.${name}` : name;
1039
+ }
1040
+ function getTextDelta(event) {
1041
+ if (event.type !== "raw_model_stream_event") {
1042
+ return "";
1043
+ }
1044
+ const data = toRecord(event.data);
1045
+ return data?.type === "output_text_delta" && typeof data.delta === "string" ? data.delta : "";
1046
+ }
1047
+ function parseJsonString(value) {
1048
+ if (typeof value !== "string") {
1049
+ return value;
1050
+ }
1051
+ try {
1052
+ return JSON.parse(value);
1053
+ } catch {
1054
+ return value;
1055
+ }
1056
+ }
1057
+ function getNumber(value) {
1058
+ return typeof value === "number" ? value : void 0;
1059
+ }
1060
+ function getString(record, key) {
1061
+ const value = record?.[key];
1062
+ return typeof value === "string" ? value : void 0;
1063
+ }
1064
+ function getObjectValue(value, key) {
1065
+ return toRecord(value)?.[key];
1066
+ }
1067
+ function toRecord(value) {
1068
+ return isRecord(value) ? value : void 0;
1069
+ }
1070
+ function isRecord(value) {
1071
+ return value !== null && typeof value === "object";
1072
+ }
1073
+
1074
+ exports.OpenAISDKAgent = OpenAISDKAgent;
1075
+ //# sourceMappingURL=index.cjs.map
1076
+ //# sourceMappingURL=index.cjs.map