awx-zipline-ai 0.2.1__py3-none-any.whl → 0.3.1__py3-none-any.whl

{ai/chronon → gen_thrift}/planner/ttypes.py

@@ -12,8 +12,8 @@ from thrift.TRecursive import fix_spec
 from uuid import UUID
 
 import sys
-import ai.chronon.api.common.ttypes
-import ai.chronon.api.ttypes
+import gen_thrift.common.ttypes
+import gen_thrift.api.ttypes
 
 from thrift.transport import TTransport
 all_structs = []
@@ -62,7 +62,7 @@ class SourceWithFilterNode(object):
                 break
             if fid == 2:
                 if ftype == TType.STRUCT:
-                    self.source = ai.chronon.api.ttypes.Source()
+                    self.source = gen_thrift.api.ttypes.Source()
                     self.source.read(iprot)
                 else:
                     iprot.skip(ftype)
@@ -149,7 +149,7 @@ class JoinBootstrapNode(object):
                 break
             if fid == 2:
                 if ftype == TType.STRUCT:
-                    self.join = ai.chronon.api.ttypes.Join()
+                    self.join = gen_thrift.api.ttypes.Join()
                     self.join.read(iprot)
                 else:
                     iprot.skip(ftype)
@@ -190,13 +190,15 @@ class JoinMergeNode(object):
     """
     Attributes:
      - join
+     - productionJoin
 
     """
     thrift_spec = None
 
 
-    def __init__(self, join = None,):
+    def __init__(self, join = None, productionJoin = None,):
         self.join = join
+        self.productionJoin = productionJoin
 
     def read(self, iprot):
         if iprot._fast_decode is not None and isinstance(iprot.trans, TTransport.CReadableTransport) and self.thrift_spec is not None:
@@ -209,10 +211,16 @@ class JoinMergeNode(object):
                 break
             if fid == 2:
                 if ftype == TType.STRUCT:
-                    self.join = ai.chronon.api.ttypes.Join()
+                    self.join = gen_thrift.api.ttypes.Join()
                     self.join.read(iprot)
                 else:
                     iprot.skip(ftype)
+            elif fid == 3:
+                if ftype == TType.STRUCT:
+                    self.productionJoin = gen_thrift.api.ttypes.Join()
+                    self.productionJoin.read(iprot)
+                else:
+                    iprot.skip(ftype)
             else:
                 iprot.skip(ftype)
             iprot.readFieldEnd()
@@ -228,6 +236,10 @@ class JoinMergeNode(object):
             oprot.writeFieldBegin('join', TType.STRUCT, 2)
             self.join.write(oprot)
             oprot.writeFieldEnd()
+        if self.productionJoin is not None:
+            oprot.writeFieldBegin('productionJoin', TType.STRUCT, 3)
+            self.productionJoin.write(oprot)
+            oprot.writeFieldEnd()
         oprot.writeFieldStop()
         oprot.writeStructEnd()
 
@@ -269,7 +281,7 @@ class JoinDerivationNode(object):
                 break
             if fid == 2:
                 if ftype == TType.STRUCT:
-                    self.join = ai.chronon.api.ttypes.Join()
+                    self.join = gen_thrift.api.ttypes.Join()
                     self.join.read(iprot)
                 else:
                     iprot.skip(ftype)
@@ -313,16 +325,18 @@ class JoinPartNode(object):
      - leftDataModel
      - joinPart
      - skewKeys
+     - productionJoinPart
 
     """
     thrift_spec = None
 
 
-    def __init__(self, leftSourceTable = None, leftDataModel = None, joinPart = None, skewKeys = None,):
+    def __init__(self, leftSourceTable = None, leftDataModel = None, joinPart = None, skewKeys = None, productionJoinPart = None,):
         self.leftSourceTable = leftSourceTable
         self.leftDataModel = leftDataModel
         self.joinPart = joinPart
         self.skewKeys = skewKeys
+        self.productionJoinPart = productionJoinPart
 
     def read(self, iprot):
         if iprot._fast_decode is not None and isinstance(iprot.trans, TTransport.CReadableTransport) and self.thrift_spec is not None:
@@ -345,7 +359,7 @@ class JoinPartNode(object):
                     iprot.skip(ftype)
             elif fid == 4:
                 if ftype == TType.STRUCT:
-                    self.joinPart = ai.chronon.api.ttypes.JoinPart()
+                    self.joinPart = gen_thrift.api.ttypes.JoinPart()
                     self.joinPart.read(iprot)
                 else:
                     iprot.skip(ftype)
@@ -365,6 +379,12 @@ class JoinPartNode(object):
                     iprot.readMapEnd()
                 else:
                     iprot.skip(ftype)
+            elif fid == 6:
+                if ftype == TType.STRUCT:
+                    self.productionJoinPart = gen_thrift.api.ttypes.JoinPart()
+                    self.productionJoinPart.read(iprot)
+                else:
+                    iprot.skip(ftype)
             else:
                 iprot.skip(ftype)
             iprot.readFieldEnd()
@@ -399,6 +419,10 @@ class JoinPartNode(object):
                 oprot.writeListEnd()
             oprot.writeMapEnd()
             oprot.writeFieldEnd()
+        if self.productionJoinPart is not None:
+            oprot.writeFieldBegin('productionJoinPart', TType.STRUCT, 6)
+            self.productionJoinPart.write(oprot)
+            oprot.writeFieldEnd()
         oprot.writeFieldStop()
         oprot.writeStructEnd()
 
@@ -440,7 +464,7 @@ class LabelJoinNode(object):
                 break
             if fid == 2:
                 if ftype == TType.STRUCT:
-                    self.join = ai.chronon.api.ttypes.Join()
+                    self.join = gen_thrift.api.ttypes.Join()
                     self.join.read(iprot)
                 else:
                     iprot.skip(ftype)
@@ -500,7 +524,7 @@ class MonolithJoinNode(object):
                 break
             if fid == 1:
                 if ftype == TType.STRUCT:
-                    self.join = ai.chronon.api.ttypes.Join()
+                    self.join = gen_thrift.api.ttypes.Join()
                     self.join.read(iprot)
                 else:
                     iprot.skip(ftype)
@@ -560,7 +584,7 @@ class StagingQueryNode(object):
                 break
             if fid == 2:
                 if ftype == TType.STRUCT:
-                    self.stagingQuery = ai.chronon.api.ttypes.StagingQuery()
+                    self.stagingQuery = gen_thrift.api.ttypes.StagingQuery()
                     self.stagingQuery.read(iprot)
                 else:
                     iprot.skip(ftype)
@@ -620,7 +644,7 @@ class GroupByBackfillNode(object):
                 break
             if fid == 2:
                 if ftype == TType.STRUCT:
-                    self.groupBy = ai.chronon.api.ttypes.GroupBy()
+                    self.groupBy = gen_thrift.api.ttypes.GroupBy()
                     self.groupBy.read(iprot)
                 else:
                     iprot.skip(ftype)
@@ -680,7 +704,7 @@ class GroupByUploadNode(object):
                 break
             if fid == 2:
                 if ftype == TType.STRUCT:
-                    self.groupBy = ai.chronon.api.ttypes.GroupBy()
+                    self.groupBy = gen_thrift.api.ttypes.GroupBy()
                     self.groupBy.read(iprot)
                 else:
                     iprot.skip(ftype)
@@ -740,7 +764,7 @@ class GroupByUploadToKVNode(object):
                 break
             if fid == 2:
                 if ftype == TType.STRUCT:
-                    self.groupBy = ai.chronon.api.ttypes.GroupBy()
+                    self.groupBy = gen_thrift.api.ttypes.GroupBy()
                     self.groupBy.read(iprot)
                 else:
                     iprot.skip(ftype)
@@ -800,7 +824,7 @@ class GroupByStreamingNode(object):
                 break
             if fid == 2:
                 if ftype == TType.STRUCT:
-                    self.groupBy = ai.chronon.api.ttypes.GroupBy()
+                    self.groupBy = gen_thrift.api.ttypes.GroupBy()
                     self.groupBy.read(iprot)
                 else:
                     iprot.skip(ftype)
@@ -860,7 +884,7 @@ class JoinMetadataUpload(object):
                 break
             if fid == 2:
                 if ftype == TType.STRUCT:
-                    self.join = ai.chronon.api.ttypes.Join()
+                    self.join = gen_thrift.api.ttypes.Join()
                     self.join.read(iprot)
                 else:
                     iprot.skip(ftype)
@@ -901,15 +925,19 @@ class ExternalSourceSensorNode(object):
     """
     Attributes:
      - metaData
-     - sourceName
+     - sourceTableDependency
+     - retryCount
+     - retryIntervalMin
 
     """
     thrift_spec = None
 
 
-    def __init__(self, metaData = None, sourceName = None,):
+    def __init__(self, metaData = None, sourceTableDependency = None, retryCount = None, retryIntervalMin = None,):
         self.metaData = metaData
-        self.sourceName = sourceName
+        self.sourceTableDependency = sourceTableDependency
+        self.retryCount = retryCount
+        self.retryIntervalMin = retryIntervalMin
 
     def read(self, iprot):
         if iprot._fast_decode is not None and isinstance(iprot.trans, TTransport.CReadableTransport) and self.thrift_spec is not None:
@@ -922,13 +950,24 @@ class ExternalSourceSensorNode(object):
                 break
             if fid == 1:
                 if ftype == TType.STRUCT:
-                    self.metaData = ai.chronon.api.ttypes.MetaData()
+                    self.metaData = gen_thrift.api.ttypes.MetaData()
                     self.metaData.read(iprot)
                 else:
                     iprot.skip(ftype)
             elif fid == 2:
-                if ftype == TType.STRING:
-                    self.sourceName = iprot.readString().decode('utf-8', errors='replace') if sys.version_info[0] == 2 else iprot.readString()
+                if ftype == TType.STRUCT:
+                    self.sourceTableDependency = gen_thrift.common.ttypes.TableDependency()
+                    self.sourceTableDependency.read(iprot)
+                else:
+                    iprot.skip(ftype)
+            elif fid == 3:
+                if ftype == TType.I64:
+                    self.retryCount = iprot.readI64()
+                else:
+                    iprot.skip(ftype)
+            elif fid == 4:
+                if ftype == TType.I64:
+                    self.retryIntervalMin = iprot.readI64()
                 else:
                     iprot.skip(ftype)
             else:
@@ -946,9 +985,17 @@ class ExternalSourceSensorNode(object):
             oprot.writeFieldBegin('metaData', TType.STRUCT, 1)
             self.metaData.write(oprot)
             oprot.writeFieldEnd()
-        if self.sourceName is not None:
-            oprot.writeFieldBegin('sourceName', TType.STRING, 2)
-            oprot.writeString(self.sourceName.encode('utf-8') if sys.version_info[0] == 2 else self.sourceName)
+        if self.sourceTableDependency is not None:
+            oprot.writeFieldBegin('sourceTableDependency', TType.STRUCT, 2)
+            self.sourceTableDependency.write(oprot)
+            oprot.writeFieldEnd()
+        if self.retryCount is not None:
+            oprot.writeFieldBegin('retryCount', TType.I64, 3)
+            oprot.writeI64(self.retryCount)
+            oprot.writeFieldEnd()
+        if self.retryIntervalMin is not None:
+            oprot.writeFieldBegin('retryIntervalMin', TType.I64, 4)
+            oprot.writeI64(self.retryIntervalMin)
             oprot.writeFieldEnd()
         oprot.writeFieldStop()
         oprot.writeStructEnd()
@@ -1103,7 +1150,7 @@ class NodeContent(object):
                     iprot.skip(ftype)
             elif fid == 200:
                 if ftype == TType.STRUCT:
-                    self.stagingQueryBackfill = ai.chronon.api.ttypes.StagingQuery()
+                    self.stagingQueryBackfill = gen_thrift.api.ttypes.StagingQuery()
                     self.stagingQueryBackfill.read(iprot)
                 else:
                     iprot.skip(ftype)
@@ -1225,7 +1272,7 @@ class Node(object):
                 break
             if fid == 1:
                 if ftype == TType.STRUCT:
-                    self.metaData = ai.chronon.api.ttypes.MetaData()
+                    self.metaData = gen_thrift.api.ttypes.MetaData()
                     self.metaData.read(iprot)
                 else:
                     iprot.skip(ftype)
@@ -1377,30 +1424,244 @@ class ConfPlan(object):
 
     def __ne__(self, other):
         return not (self == other)
+
+
+class NodeMeta(object):
+    """
+    Attributes:
+     - name
+     - hash
+     - deps
+     - continuous
+     - stepSize
+     - outputTable
+     - retries
+
+    """
+    thrift_spec = None
+
+
+    def __init__(self, name = None, hash = None, deps = None, continuous = None, stepSize = None, outputTable = None, retries = None,):
+        self.name = name
+        self.hash = hash
+        self.deps = deps
+        self.continuous = continuous
+        self.stepSize = stepSize
+        self.outputTable = outputTable
+        self.retries = retries
+
+    def read(self, iprot):
+        if iprot._fast_decode is not None and isinstance(iprot.trans, TTransport.CReadableTransport) and self.thrift_spec is not None:
+            iprot._fast_decode(self, iprot, [self.__class__, self.thrift_spec])
+            return
+        iprot.readStructBegin()
+        while True:
+            (fname, ftype, fid) = iprot.readFieldBegin()
+            if ftype == TType.STOP:
+                break
+            if fid == 1:
+                if ftype == TType.STRING:
+                    self.name = iprot.readString().decode('utf-8', errors='replace') if sys.version_info[0] == 2 else iprot.readString()
+                else:
+                    iprot.skip(ftype)
+            elif fid == 2:
+                if ftype == TType.STRING:
+                    self.hash = iprot.readString().decode('utf-8', errors='replace') if sys.version_info[0] == 2 else iprot.readString()
+                else:
+                    iprot.skip(ftype)
+            elif fid == 3:
+                if ftype == TType.LIST:
+                    self.deps = []
+                    (_etype51, _size48) = iprot.readListBegin()
+                    for _i52 in range(_size48):
+                        _elem53 = gen_thrift.common.ttypes.TableDependency()
+                        _elem53.read(iprot)
+                        self.deps.append(_elem53)
+                    iprot.readListEnd()
+                else:
+                    iprot.skip(ftype)
+            elif fid == 4:
+                if ftype == TType.BOOL:
+                    self.continuous = iprot.readBool()
+                else:
+                    iprot.skip(ftype)
+            elif fid == 5:
+                if ftype == TType.I32:
+                    self.stepSize = iprot.readI32()
+                else:
+                    iprot.skip(ftype)
+            elif fid == 6:
+                if ftype == TType.STRING:
+                    self.outputTable = iprot.readString().decode('utf-8', errors='replace') if sys.version_info[0] == 2 else iprot.readString()
+                else:
+                    iprot.skip(ftype)
+            elif fid == 7:
+                if ftype == TType.I32:
+                    self.retries = iprot.readI32()
+                else:
+                    iprot.skip(ftype)
+            else:
+                iprot.skip(ftype)
+            iprot.readFieldEnd()
+        iprot.readStructEnd()
+
+    def write(self, oprot):
+        self.validate()
+        if oprot._fast_encode is not None and self.thrift_spec is not None:
+            oprot.trans.write(oprot._fast_encode(self, [self.__class__, self.thrift_spec]))
+            return
+        oprot.writeStructBegin('NodeMeta')
+        if self.name is not None:
+            oprot.writeFieldBegin('name', TType.STRING, 1)
+            oprot.writeString(self.name.encode('utf-8') if sys.version_info[0] == 2 else self.name)
+            oprot.writeFieldEnd()
+        if self.hash is not None:
+            oprot.writeFieldBegin('hash', TType.STRING, 2)
+            oprot.writeString(self.hash.encode('utf-8') if sys.version_info[0] == 2 else self.hash)
+            oprot.writeFieldEnd()
+        if self.deps is not None:
+            oprot.writeFieldBegin('deps', TType.LIST, 3)
+            oprot.writeListBegin(TType.STRUCT, len(self.deps))
+            for iter54 in self.deps:
+                iter54.write(oprot)
+            oprot.writeListEnd()
+            oprot.writeFieldEnd()
+        if self.continuous is not None:
+            oprot.writeFieldBegin('continuous', TType.BOOL, 4)
+            oprot.writeBool(self.continuous)
+            oprot.writeFieldEnd()
+        if self.stepSize is not None:
+            oprot.writeFieldBegin('stepSize', TType.I32, 5)
+            oprot.writeI32(self.stepSize)
+            oprot.writeFieldEnd()
+        if self.outputTable is not None:
+            oprot.writeFieldBegin('outputTable', TType.STRING, 6)
+            oprot.writeString(self.outputTable.encode('utf-8') if sys.version_info[0] == 2 else self.outputTable)
+            oprot.writeFieldEnd()
+        if self.retries is not None:
+            oprot.writeFieldBegin('retries', TType.I32, 7)
+            oprot.writeI32(self.retries)
+            oprot.writeFieldEnd()
+        oprot.writeFieldStop()
+        oprot.writeStructEnd()
+
+    def validate(self):
+        return
+
+    def __repr__(self):
+        L = ['%s=%r' % (key, value)
+             for key, value in self.__dict__.items()]
+        return '%s(%s)' % (self.__class__.__name__, ', '.join(L))
+
+    def __eq__(self, other):
+        return isinstance(other, self.__class__) and self.__dict__ == other.__dict__
+
+    def __ne__(self, other):
+        return not (self == other)
+
+
+class WorkflowPlan(object):
+    """
+    Attributes:
+     - nodes
+     - terminalNode
+
+    """
+    thrift_spec = None
+
+
+    def __init__(self, nodes = None, terminalNode = None,):
+        self.nodes = nodes
+        self.terminalNode = terminalNode
+
+    def read(self, iprot):
+        if iprot._fast_decode is not None and isinstance(iprot.trans, TTransport.CReadableTransport) and self.thrift_spec is not None:
+            iprot._fast_decode(self, iprot, [self.__class__, self.thrift_spec])
+            return
+        iprot.readStructBegin()
+        while True:
+            (fname, ftype, fid) = iprot.readFieldBegin()
+            if ftype == TType.STOP:
+                break
+            if fid == 1:
+                if ftype == TType.LIST:
+                    self.nodes = []
+                    (_etype58, _size55) = iprot.readListBegin()
+                    for _i59 in range(_size55):
+                        _elem60 = NodeMeta()
+                        _elem60.read(iprot)
+                        self.nodes.append(_elem60)
+                    iprot.readListEnd()
+                else:
+                    iprot.skip(ftype)
+            elif fid == 2:
+                if ftype == TType.STRING:
+                    self.terminalNode = iprot.readString().decode('utf-8', errors='replace') if sys.version_info[0] == 2 else iprot.readString()
+                else:
+                    iprot.skip(ftype)
+            else:
+                iprot.skip(ftype)
+            iprot.readFieldEnd()
+        iprot.readStructEnd()
+
+    def write(self, oprot):
+        self.validate()
+        if oprot._fast_encode is not None and self.thrift_spec is not None:
+            oprot.trans.write(oprot._fast_encode(self, [self.__class__, self.thrift_spec]))
+            return
+        oprot.writeStructBegin('WorkflowPlan')
+        if self.nodes is not None:
+            oprot.writeFieldBegin('nodes', TType.LIST, 1)
+            oprot.writeListBegin(TType.STRUCT, len(self.nodes))
+            for iter61 in self.nodes:
+                iter61.write(oprot)
+            oprot.writeListEnd()
+            oprot.writeFieldEnd()
+        if self.terminalNode is not None:
+            oprot.writeFieldBegin('terminalNode', TType.STRING, 2)
+            oprot.writeString(self.terminalNode.encode('utf-8') if sys.version_info[0] == 2 else self.terminalNode)
+            oprot.writeFieldEnd()
+        oprot.writeFieldStop()
+        oprot.writeStructEnd()
+
+    def validate(self):
+        return
+
+    def __repr__(self):
+        L = ['%s=%r' % (key, value)
+             for key, value in self.__dict__.items()]
+        return '%s(%s)' % (self.__class__.__name__, ', '.join(L))
+
+    def __eq__(self, other):
+        return isinstance(other, self.__class__) and self.__dict__ == other.__dict__
+
+    def __ne__(self, other):
+        return not (self == other)
 all_structs.append(SourceWithFilterNode)
 SourceWithFilterNode.thrift_spec = (
     None,  # 0
     None,  # 1
-    (2, TType.STRUCT, 'source', [ai.chronon.api.ttypes.Source, None], None, ),  # 2
+    (2, TType.STRUCT, 'source', [gen_thrift.api.ttypes.Source, None], None, ),  # 2
     (3, TType.MAP, 'excludeKeys', (TType.STRING, 'UTF8', TType.LIST, (TType.STRING, 'UTF8', False), False), None, ),  # 3
 )
 all_structs.append(JoinBootstrapNode)
 JoinBootstrapNode.thrift_spec = (
     None,  # 0
     None,  # 1
-    (2, TType.STRUCT, 'join', [ai.chronon.api.ttypes.Join, None], None, ),  # 2
+    (2, TType.STRUCT, 'join', [gen_thrift.api.ttypes.Join, None], None, ),  # 2
 )
 all_structs.append(JoinMergeNode)
 JoinMergeNode.thrift_spec = (
     None,  # 0
     None,  # 1
-    (2, TType.STRUCT, 'join', [ai.chronon.api.ttypes.Join, None], None, ),  # 2
+    (2, TType.STRUCT, 'join', [gen_thrift.api.ttypes.Join, None], None, ),  # 2
+    (3, TType.STRUCT, 'productionJoin', [gen_thrift.api.ttypes.Join, None], None, ),  # 3
 )
 all_structs.append(JoinDerivationNode)
 JoinDerivationNode.thrift_spec = (
     None,  # 0
     None,  # 1
-    (2, TType.STRUCT, 'join', [ai.chronon.api.ttypes.Join, None], None, ),  # 2
+    (2, TType.STRUCT, 'join', [gen_thrift.api.ttypes.Join, None], None, ),  # 2
 )
 all_structs.append(JoinPartNode)
 JoinPartNode.thrift_spec = (
@@ -1408,61 +1669,64 @@ JoinPartNode.thrift_spec = (
     None,  # 1
     (2, TType.STRING, 'leftSourceTable', 'UTF8', None, ),  # 2
     (3, TType.I32, 'leftDataModel', None, None, ),  # 3
-    (4, TType.STRUCT, 'joinPart', [ai.chronon.api.ttypes.JoinPart, None], None, ),  # 4
+    (4, TType.STRUCT, 'joinPart', [gen_thrift.api.ttypes.JoinPart, None], None, ),  # 4
     (5, TType.MAP, 'skewKeys', (TType.STRING, 'UTF8', TType.LIST, (TType.STRING, 'UTF8', False), False), None, ),  # 5
+    (6, TType.STRUCT, 'productionJoinPart', [gen_thrift.api.ttypes.JoinPart, None], None, ),  # 6
 )
 all_structs.append(LabelJoinNode)
 LabelJoinNode.thrift_spec = (
     None,  # 0
     None,  # 1
-    (2, TType.STRUCT, 'join', [ai.chronon.api.ttypes.Join, None], None, ),  # 2
+    (2, TType.STRUCT, 'join', [gen_thrift.api.ttypes.Join, None], None, ),  # 2
 )
 all_structs.append(MonolithJoinNode)
 MonolithJoinNode.thrift_spec = (
     None,  # 0
-    (1, TType.STRUCT, 'join', [ai.chronon.api.ttypes.Join, None], None, ),  # 1
+    (1, TType.STRUCT, 'join', [gen_thrift.api.ttypes.Join, None], None, ),  # 1
 )
 all_structs.append(StagingQueryNode)
 StagingQueryNode.thrift_spec = (
     None,  # 0
     None,  # 1
-    (2, TType.STRUCT, 'stagingQuery', [ai.chronon.api.ttypes.StagingQuery, None], None, ),  # 2
+    (2, TType.STRUCT, 'stagingQuery', [gen_thrift.api.ttypes.StagingQuery, None], None, ),  # 2
 )
 all_structs.append(GroupByBackfillNode)
 GroupByBackfillNode.thrift_spec = (
     None,  # 0
     None,  # 1
-    (2, TType.STRUCT, 'groupBy', [ai.chronon.api.ttypes.GroupBy, None], None, ),  # 2
+    (2, TType.STRUCT, 'groupBy', [gen_thrift.api.ttypes.GroupBy, None], None, ),  # 2
 )
 all_structs.append(GroupByUploadNode)
 GroupByUploadNode.thrift_spec = (
     None,  # 0
     None,  # 1
-    (2, TType.STRUCT, 'groupBy', [ai.chronon.api.ttypes.GroupBy, None], None, ),  # 2
+    (2, TType.STRUCT, 'groupBy', [gen_thrift.api.ttypes.GroupBy, None], None, ),  # 2
 )
 all_structs.append(GroupByUploadToKVNode)
 GroupByUploadToKVNode.thrift_spec = (
     None,  # 0
     None,  # 1
-    (2, TType.STRUCT, 'groupBy', [ai.chronon.api.ttypes.GroupBy, None], None, ),  # 2
+    (2, TType.STRUCT, 'groupBy', [gen_thrift.api.ttypes.GroupBy, None], None, ),  # 2
 )
 all_structs.append(GroupByStreamingNode)
 GroupByStreamingNode.thrift_spec = (
     None,  # 0
     None,  # 1
-    (2, TType.STRUCT, 'groupBy', [ai.chronon.api.ttypes.GroupBy, None], None, ),  # 2
+    (2, TType.STRUCT, 'groupBy', [gen_thrift.api.ttypes.GroupBy, None], None, ),  # 2
 )
 all_structs.append(JoinMetadataUpload)
 JoinMetadataUpload.thrift_spec = (
     None,  # 0
     None,  # 1
-    (2, TType.STRUCT, 'join', [ai.chronon.api.ttypes.Join, None], None, ),  # 2
+    (2, TType.STRUCT, 'join', [gen_thrift.api.ttypes.Join, None], None, ),  # 2
 )
 all_structs.append(ExternalSourceSensorNode)
 ExternalSourceSensorNode.thrift_spec = (
     None,  # 0
-    (1, TType.STRUCT, 'metaData', [ai.chronon.api.ttypes.MetaData, None], None, ),  # 1
-    (2, TType.STRING, 'sourceName', 'UTF8', None, ),  # 2
+    (1, TType.STRUCT, 'metaData', [gen_thrift.api.ttypes.MetaData, None], None, ),  # 1
+    (2, TType.STRUCT, 'sourceTableDependency', [gen_thrift.common.ttypes.TableDependency, None], None, ),  # 2
+    (3, TType.I64, 'retryCount', None, None, ),  # 3
+    (4, TType.I64, 'retryIntervalMin', None, None, ),  # 4
 )
 all_structs.append(NodeContent)
 NodeContent.thrift_spec = (
@@ -1666,12 +1930,12 @@ NodeContent.thrift_spec = (
     None,  # 197
     None,  # 198
     None,  # 199
-    (200, TType.STRUCT, 'stagingQueryBackfill', [ai.chronon.api.ttypes.StagingQuery, None], None, ),  # 200
+    (200, TType.STRUCT, 'stagingQueryBackfill', [gen_thrift.api.ttypes.StagingQuery, None], None, ),  # 200
 )
 all_structs.append(Node)
 Node.thrift_spec = (
     None,  # 0
-    (1, TType.STRUCT, 'metaData', [ai.chronon.api.ttypes.MetaData, None], None, ),  # 1
+    (1, TType.STRUCT, 'metaData', [gen_thrift.api.ttypes.MetaData, None], None, ),  # 1
     (2, TType.STRUCT, 'content', [NodeContent, None], None, ),  # 2
     (3, TType.STRING, 'semanticHash', 'UTF8', None, ),  # 3
     (4, TType.BOOL, 'isLongRunning', None, None, ),  # 4
@@ -1682,5 +1946,22 @@ ConfPlan.thrift_spec = (
     (1, TType.LIST, 'nodes', (TType.STRUCT, [Node, None], False), None, ),  # 1
     (2, TType.MAP, 'terminalNodeNames', (TType.I32, None, TType.STRING, 'UTF8', False), None, ),  # 2
 )
+all_structs.append(NodeMeta)
+NodeMeta.thrift_spec = (
+    None,  # 0
+    (1, TType.STRING, 'name', 'UTF8', None, ),  # 1
+    (2, TType.STRING, 'hash', 'UTF8', None, ),  # 2
+    (3, TType.LIST, 'deps', (TType.STRUCT, [gen_thrift.common.ttypes.TableDependency, None], False), None, ),  # 3
+    (4, TType.BOOL, 'continuous', None, None, ),  # 4
+    (5, TType.I32, 'stepSize', None, None, ),  # 5
+    (6, TType.STRING, 'outputTable', 'UTF8', None, ),  # 6
+    (7, TType.I32, 'retries', None, None, ),  # 7
+)
+all_structs.append(WorkflowPlan)
+WorkflowPlan.thrift_spec = (
+    None,  # 0
+    (1, TType.LIST, 'nodes', (TType.STRUCT, [NodeMeta, None], False), None, ),  # 1
+    (2, TType.STRING, 'terminalNode', 'UTF8', None, ),  # 2
+)
 fix_spec(all_structs)
 del all_structs