@@ -2,7 +2,12 @@ import {
2
2
BROKEN_EDGES_WARNING ,
3
3
componentsToIgnoreUpdate ,
4
4
} from "@/constants/constants" ;
5
- import { track } from "@/customization/utils/analytics" ;
5
+ import { ENABLE_DATASTAX_LANGFLOW } from "@/customization/feature-flags" ;
6
+ import {
7
+ track ,
8
+ trackDataLoaded ,
9
+ trackFlowBuild ,
10
+ } from "@/customization/utils/analytics" ;
6
11
import { brokenEdgeMessage } from "@/utils/utils" ;
7
12
import {
8
13
EdgeChange ,
@@ -19,7 +24,7 @@ import {
19
24
MISSED_ERROR_ALERT ,
20
25
} from "../constants/alerts_constants" ;
21
26
import { BuildStatus } from "../constants/enums" ;
22
- import { VertexBuildTypeAPI } from "../types/api" ;
27
+ import { LogsLogType , VertexBuildTypeAPI } from "../types/api" ;
23
28
import { ChatInputType , ChatOutputType } from "../types/chat" ;
24
29
import {
25
30
AllNodeType ,
@@ -699,6 +704,28 @@ const useFlowStore = create<FlowStoreType>((set, get) => ({
699
704
...get ( ) . verticesBuild ! . verticesIds ,
700
705
...next_vertices_ids ,
701
706
] ;
707
+ if (
708
+ ENABLE_DATASTAX_LANGFLOW &&
709
+ vertexBuildData ?. id ?. includes ( "AstraDB" )
710
+ ) {
711
+ const search_results : LogsLogType [ ] = Object . values (
712
+ vertexBuildData ?. data ?. logs ?. search_results ,
713
+ ) ;
714
+ search_results . forEach ( ( log ) => {
715
+ if (
716
+ log . message . includes ( "Adding" ) &&
717
+ log . message . includes ( "documents" ) &&
718
+ log . message . includes ( "Vector Store" )
719
+ ) {
720
+ trackDataLoaded (
721
+ get ( ) . currentFlow ?. id ,
722
+ get ( ) . currentFlow ?. name ,
723
+ "AstraDB Vector Store" ,
724
+ vertexBuildData ?. id ,
725
+ ) ;
726
+ }
727
+ } ) ;
728
+ }
702
729
get ( ) . updateVerticesBuild ( {
703
730
verticesIds : newIds ,
704
731
verticesLayers : newLayers ,
@@ -747,6 +774,9 @@ const useFlowStore = create<FlowStoreType>((set, get) => ({
747
774
false ,
748
775
) ;
749
776
get ( ) . setIsBuilding ( false ) ;
777
+ trackFlowBuild ( get ( ) . currentFlow ?. name ?? "Unknown" , false , {
778
+ flowId : get ( ) . currentFlow ?. id ,
779
+ } ) ;
750
780
} ,
751
781
onBuildUpdate : handleBuildUpdate ,
752
782
onBuildError : ( title : string , list : string [ ] , elementList ) => {
@@ -767,6 +797,10 @@ const useFlowStore = create<FlowStoreType>((set, get) => ({
767
797
setErrorData ( { list, title } ) ;
768
798
get ( ) . setIsBuilding ( false ) ;
769
799
get ( ) . buildController . abort ( ) ;
800
+ trackFlowBuild ( get ( ) . currentFlow ?. name ?? "Unknown" , true , {
801
+ flowId : get ( ) . currentFlow ?. id ,
802
+ error : list ,
803
+ } ) ;
770
804
} ,
771
805
onBuildStart : ( elementList ) => {
772
806
const idList = elementList
0 commit comments