@mastra/ai-sdk 0.2.3 → 0.2.5-alpha.0

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/CHANGELOG.md CHANGED
@@ -1,5 +1,23 @@
1
1
  # @mastra/ai-sdk
2
2
 
3
+ ## 0.2.5-alpha.0
4
+
5
+ ### Patch Changes
6
+
7
+ - Fix usage tracking with agent network ([#9413](https://github.com/mastra-ai/mastra/pull/9413))
8
+
9
+ - Updated dependencies [[`f57a81e`](https://github.com/mastra-ai/mastra/commit/f57a81e6ce644e45bf1c9618778cc54c50a84ad4), [`fc79af3`](https://github.com/mastra-ai/mastra/commit/fc79af3915d1c456729cbd753673b0c0564340d8), [`60bd45d`](https://github.com/mastra-ai/mastra/commit/60bd45de021f0dfbe6583928f6da5169cb5585ba), [`a30093d`](https://github.com/mastra-ai/mastra/commit/a30093de98c1836dcd5dfddf09649010712b8c95), [`e40d4d0`](https://github.com/mastra-ai/mastra/commit/e40d4d0a0971b4505e7c9de73c656066c7565653), [`ff16f9b`](https://github.com/mastra-ai/mastra/commit/ff16f9b9dbc701b26b6c4e9872f759f3880f9327), [`35e6cf7`](https://github.com/mastra-ai/mastra/commit/35e6cf722fef16ea0301eb9cf5a32fe9ccb12d22), [`30a2e36`](https://github.com/mastra-ai/mastra/commit/30a2e369485e0e59c4faa1d83c5635c2260b304c)]:
10
+ - @mastra/core@0.23.2-alpha.1
11
+
12
+ ## 0.2.4
13
+
14
+ ### Patch Changes
15
+
16
+ - Fix peerdependencies ([`eb7c1c8`](https://github.com/mastra-ai/mastra/commit/eb7c1c8c592d8fb16dfd250e337d9cdc73c8d5de))
17
+
18
+ - Updated dependencies []:
19
+ - @mastra/core@0.23.1
20
+
3
21
  ## 0.2.3
4
22
 
5
23
  ### Patch Changes
package/dist/index.cjs CHANGED
@@ -702,19 +702,22 @@ function transformWorkflow(payload, bufferedWorkflows, isNested) {
702
702
  function transformNetwork(payload, bufferedNetworks, isNested) {
703
703
  switch (payload.type) {
704
704
  case "routing-agent-start": {
705
- if (!bufferedNetworks.has(payload.payload.runId)) {
706
- bufferedNetworks.set(payload.payload.runId, {
705
+ if (!bufferedNetworks.has(payload.runId)) {
706
+ bufferedNetworks.set(payload.runId, {
707
707
  name: payload.payload.agentId,
708
- steps: []
708
+ steps: [],
709
+ usage: null,
710
+ output: null
709
711
  });
710
712
  }
711
713
  return {
712
714
  type: isNested ? "data-tool-network" : "data-network",
713
- id: payload.payload.runId,
715
+ id: payload.runId,
714
716
  data: {
715
- name: bufferedNetworks.get(payload.payload.runId).name,
717
+ name: bufferedNetworks.get(payload.runId).name,
716
718
  status: "running",
717
- steps: bufferedNetworks.get(payload.payload.runId).steps,
719
+ usage: null,
720
+ steps: bufferedNetworks.get(payload.runId).steps,
718
721
  output: null
719
722
  }
720
723
  };
@@ -737,62 +740,56 @@ function transformNetwork(payload, bufferedNetworks, isNested) {
737
740
  };
738
741
  }
739
742
  case "agent-execution-start": {
740
- const current = bufferedNetworks.get(payload.payload.runId) || { name: "", steps: [] };
743
+ const current = bufferedNetworks.get(payload.runId) || { name: "", steps: [], usage: null, output: null };
741
744
  current.steps.push({
742
745
  name: payload.payload.agentId,
743
746
  status: "running",
744
747
  input: payload.payload.args || null,
745
748
  output: null
746
749
  });
747
- bufferedNetworks.set(payload.payload.runId, current);
750
+ bufferedNetworks.set(payload.runId, current);
748
751
  return {
749
752
  type: isNested ? "data-tool-network" : "data-network",
750
- id: payload.payload.runId,
753
+ id: payload.runId,
751
754
  data: {
752
- name: current.name,
753
- status: "running",
754
- steps: current.steps,
755
- output: null
755
+ ...current,
756
+ status: "running"
756
757
  }
757
758
  };
758
759
  }
759
760
  case "workflow-execution-start": {
760
- const current = bufferedNetworks.get(payload.payload.runId) || { name: "", steps: [] };
761
+ const current = bufferedNetworks.get(payload.runId) || { name: "", steps: [], usage: null, output: null };
761
762
  current.steps.push({
762
763
  name: payload.payload.name,
763
764
  status: "running",
764
765
  input: payload.payload.args || null,
765
766
  output: null
766
767
  });
767
- bufferedNetworks.set(payload.payload.runId, current);
768
+ bufferedNetworks.set(payload.runId, current);
768
769
  return {
769
770
  type: isNested ? "data-tool-network" : "data-network",
770
- id: payload.payload.runId,
771
+ id: payload.runId,
771
772
  data: {
772
- name: current.name,
773
- status: "running",
774
- steps: current.steps,
775
- output: null
773
+ ...current,
774
+ status: "running"
776
775
  }
777
776
  };
778
777
  }
779
778
  case "tool-execution-start": {
780
- const current = bufferedNetworks.get(payload.payload.runId) || { name: "", steps: [] };
779
+ const current = bufferedNetworks.get(payload.runId) || { name: "", steps: [], usage: null, output: null };
781
780
  current.steps.push({
782
781
  name: payload.payload.args?.toolName,
783
782
  status: "running",
784
783
  input: payload.payload.args?.args || null,
785
784
  output: null
786
785
  });
787
- bufferedNetworks.set(payload.payload.runId, current);
786
+ bufferedNetworks.set(payload.runId, current);
788
787
  return {
789
788
  type: isNested ? "data-tool-network" : "data-network",
790
- id: payload.payload.runId,
789
+ id: payload.runId,
791
790
  data: {
792
- name: current.name,
793
- status: "running",
794
- steps: current.steps,
795
- output: null
791
+ ...current,
792
+ status: "running"
796
793
  }
797
794
  };
798
795
  }
@@ -809,10 +806,10 @@ function transformNetwork(payload, bufferedNetworks, isNested) {
809
806
  type: isNested ? "data-tool-network" : "data-network",
810
807
  id: payload.runId,
811
808
  data: {
812
- name: current.name,
809
+ ...current,
810
+ usage: payload.payload?.usage ?? current.usage,
813
811
  status: "running",
814
- steps: current.steps,
815
- output: payload.payload.result ?? null
812
+ output: payload.payload.result ?? current.output
816
813
  }
817
814
  };
818
815
  }
@@ -829,10 +826,9 @@ function transformNetwork(payload, bufferedNetworks, isNested) {
829
826
  type: isNested ? "data-tool-network" : "data-network",
830
827
  id: payload.runId,
831
828
  data: {
832
- name: current.name,
829
+ ...current,
833
830
  status: "running",
834
- steps: current.steps,
835
- output: payload.payload.result ?? null
831
+ output: payload.payload.result ?? current.output
836
832
  }
837
833
  };
838
834
  }
@@ -849,38 +845,37 @@ function transformNetwork(payload, bufferedNetworks, isNested) {
849
845
  type: isNested ? "data-tool-network" : "data-network",
850
846
  id: payload.runId,
851
847
  data: {
852
- name: current.name,
848
+ ...current,
849
+ usage: payload.payload?.usage ?? current.usage,
853
850
  status: "running",
854
- steps: current.steps,
855
- output: payload.payload.result ?? null
851
+ output: payload.payload.result ?? current.output
856
852
  }
857
853
  };
858
854
  }
859
855
  case "routing-agent-end": {
860
- const current = bufferedNetworks.get(payload.payload.runId);
856
+ const current = bufferedNetworks.get(payload.runId);
861
857
  if (!current) return null;
862
858
  return {
863
859
  type: isNested ? "data-tool-network" : "data-network",
864
- id: payload.payload.runId,
860
+ id: payload.runId,
865
861
  data: {
866
- name: current.name,
862
+ ...current,
867
863
  status: "finished",
868
- steps: current.steps,
869
- output: payload.payload?.result ?? null
864
+ usage: payload.payload?.usage ?? current.usage,
865
+ output: payload.payload?.result ?? current.output
870
866
  }
871
867
  };
872
868
  }
873
869
  case "network-execution-event-step-finish": {
874
- const current = bufferedNetworks.get(payload.payload.runId);
870
+ const current = bufferedNetworks.get(payload.runId);
875
871
  if (!current) return null;
876
872
  return {
877
873
  type: isNested ? "data-tool-network" : "data-network",
878
- id: payload.payload.runId,
874
+ id: payload.runId,
879
875
  data: {
880
- name: current.name,
876
+ ...current,
881
877
  status: "finished",
882
- steps: current.steps,
883
- output: payload.payload?.result ?? null
878
+ output: payload.payload?.result ?? current.output
884
879
  }
885
880
  };
886
881
  }
@@ -891,10 +886,10 @@ function transformNetwork(payload, bufferedNetworks, isNested) {
891
886
  type: isNested ? "data-tool-network" : "data-network",
892
887
  id: payload.runId,
893
888
  data: {
894
- name: current.name,
889
+ ...current,
890
+ usage: payload.payload?.usage ?? current.usage,
895
891
  status: "finished",
896
- steps: current.steps,
897
- output: payload.payload?.result ?? null
892
+ output: payload.payload?.result ?? current.output
898
893
  }
899
894
  };
900
895
  }