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

{ai/chronon → gen_thrift}/api/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.observability.ttypes
+import gen_thrift.common.ttypes
+import gen_thrift.observability.ttypes
 
 from thrift.transport import TTransport
 all_structs = []
@@ -121,6 +121,27 @@ class EngineType(object):
     }
 
 
+class ConfType(object):
+    GROUP_BY = 0
+    JOIN = 1
+    STAGING_QUERY = 2
+    MODEL = 3
+
+    _VALUES_TO_NAMES = {
+        0: "GROUP_BY",
+        1: "JOIN",
+        2: "STAGING_QUERY",
+        3: "MODEL",
+    }
+
+    _NAMES_TO_VALUES = {
+        "GROUP_BY": 0,
+        "JOIN": 1,
+        "STAGING_QUERY": 2,
+        "MODEL": 3,
+    }
+
+
 class DataKind(object):
     BOOLEAN = 0
     BYTE = 1
@@ -336,13 +357,13 @@ class Query(object):
                     iprot.skip(ftype)
             elif fid == 22:
                 if ftype == TType.STRUCT:
-                    self.partitionInterval = ai.chronon.api.common.ttypes.Window()
+                    self.partitionInterval = gen_thrift.common.ttypes.Window()
                     self.partitionInterval.read(iprot)
                 else:
                     iprot.skip(ftype)
             elif fid == 23:
                 if ftype == TType.STRUCT:
-                    self.partitionLag = ai.chronon.api.common.ttypes.Window()
+                    self.partitionLag = gen_thrift.common.ttypes.Window()
                     self.partitionLag.read(iprot)
                 else:
                     iprot.skip(ftype)
@@ -472,8 +493,6 @@ class StagingQuery(object):
 
      - setups: Spark SQL setup statements. Used typically to register UDFs.
 
-     - partitionColumn: Only needed for `max_date` template
-
      - engineType: By default, spark is the compute engine. You can specify an override (eg. bigquery, etc.)
 
      - tableDependencies
@@ -483,12 +502,11 @@ class StagingQuery(object):
     thrift_spec = None
 
 
-    def __init__(self, metaData = None, query = None, startPartition = None, setups = None, partitionColumn = None, engineType = None, tableDependencies = None, recomputeDays = None,):
+    def __init__(self, metaData = None, query = None, startPartition = None, setups = None, engineType = None, tableDependencies = None, recomputeDays = None,):
         self.metaData = metaData
         self.query = query
         self.startPartition = startPartition
         self.setups = setups
-        self.partitionColumn = partitionColumn
         self.engineType = engineType
         self.tableDependencies = tableDependencies
         self.recomputeDays = recomputeDays
@@ -529,21 +547,16 @@ class StagingQuery(object):
                 else:
                     iprot.skip(ftype)
             elif fid == 5:
-                if ftype == TType.STRING:
-                    self.partitionColumn = iprot.readString().decode('utf-8', errors='replace') if sys.version_info[0] == 2 else iprot.readString()
-                else:
-                    iprot.skip(ftype)
-            elif fid == 6:
                 if ftype == TType.I32:
                     self.engineType = iprot.readI32()
                 else:
                     iprot.skip(ftype)
-            elif fid == 7:
+            elif fid == 6:
                 if ftype == TType.LIST:
                     self.tableDependencies = []
                     (_etype39, _size36) = iprot.readListBegin()
                     for _i40 in range(_size36):
-                        _elem41 = ai.chronon.api.common.ttypes.TableDependency()
+                        _elem41 = gen_thrift.common.ttypes.TableDependency()
                         _elem41.read(iprot)
                         self.tableDependencies.append(_elem41)
                     iprot.readListEnd()
@@ -584,16 +597,12 @@ class StagingQuery(object):
                 oprot.writeString(iter42.encode('utf-8') if sys.version_info[0] == 2 else iter42)
             oprot.writeListEnd()
             oprot.writeFieldEnd()
-        if self.partitionColumn is not None:
-            oprot.writeFieldBegin('partitionColumn', TType.STRING, 5)
-            oprot.writeString(self.partitionColumn.encode('utf-8') if sys.version_info[0] == 2 else self.partitionColumn)
-            oprot.writeFieldEnd()
         if self.engineType is not None:
-            oprot.writeFieldBegin('engineType', TType.I32, 6)
+            oprot.writeFieldBegin('engineType', TType.I32, 5)
             oprot.writeI32(self.engineType)
             oprot.writeFieldEnd()
         if self.tableDependencies is not None:
-            oprot.writeFieldBegin('tableDependencies', TType.LIST, 7)
+            oprot.writeFieldBegin('tableDependencies', TType.LIST, 6)
             oprot.writeListBegin(TType.STRUCT, len(self.tableDependencies))
             for iter43 in self.tableDependencies:
                 iter43.write(oprot)
@@ -1137,7 +1146,7 @@ class Aggregation(object):
                     self.windows = []
                     (_etype54, _size51) = iprot.readListBegin()
                     for _i55 in range(_size51):
-                        _elem56 = ai.chronon.api.common.ttypes.Window()
+                        _elem56 = gen_thrift.common.ttypes.Window()
                         _elem56.read(iprot)
                         self.windows.append(_elem56)
                     iprot.readListEnd()
@@ -1264,7 +1273,7 @@ class AggregationPart(object):
                     iprot.skip(ftype)
             elif fid == 4:
                 if ftype == TType.STRUCT:
-                    self.window = ai.chronon.api.common.ttypes.Window()
+                    self.window = gen_thrift.common.ttypes.Window()
                     self.window.read(iprot)
                 else:
                     iprot.skip(ftype)
@@ -1342,6 +1351,7 @@ class MetaData(object):
      - tableProperties
      - tags
      - columnTags
+     - columnHashes
      - online
      - production
      - sourceFile
@@ -1356,7 +1366,7 @@ class MetaData(object):
     thrift_spec = None
 
 
-    def __init__(self, name = None, team = None, version = None, outputNamespace = None, additionalOutputPartitionColumns = None, tableProperties = None, tags = None, columnTags = None, online = None, production = None, sourceFile = None, customJson = None, consistencyCheck = None, samplePercent = None, consistencySamplePercent = None, driftSpec = None, executionInfo = None,):
+    def __init__(self, name = None, team = None, version = None, outputNamespace = None, additionalOutputPartitionColumns = None, tableProperties = None, tags = None, columnTags = None, columnHashes = None, online = None, production = None, sourceFile = None, customJson = None, consistencyCheck = None, samplePercent = None, consistencySamplePercent = None, driftSpec = None, executionInfo = None,):
         self.name = name
         self.team = team
         self.version = version
@@ -1365,6 +1375,7 @@ class MetaData(object):
         self.tableProperties = tableProperties
         self.tags = tags
         self.columnTags = columnTags
+        self.columnHashes = columnHashes
         self.online = online
         self.production = production
         self.sourceFile = sourceFile
@@ -1453,6 +1464,17 @@ class MetaData(object):
                     iprot.readMapEnd()
                 else:
                     iprot.skip(ftype)
+            elif fid == 22:
+                if ftype == TType.MAP:
+                    self.columnHashes = {}
+                    (_ktype111, _vtype112, _size110) = iprot.readMapBegin()
+                    for _i114 in range(_size110):
+                        _key115 = iprot.readString().decode('utf-8', errors='replace') if sys.version_info[0] == 2 else iprot.readString()
+                        _val116 = iprot.readString().decode('utf-8', errors='replace') if sys.version_info[0] == 2 else iprot.readString()
+                        self.columnHashes[_key115] = _val116
+                    iprot.readMapEnd()
+                else:
+                    iprot.skip(ftype)
             elif fid == 100:
                 if ftype == TType.BOOL:
                     self.online = iprot.readBool()
@@ -1490,13 +1512,13 @@ class MetaData(object):
                     iprot.skip(ftype)
             elif fid == 203:
                 if ftype == TType.STRUCT:
-                    self.driftSpec = ai.chronon.observability.ttypes.DriftSpec()
+                    self.driftSpec = gen_thrift.observability.ttypes.DriftSpec()
                     self.driftSpec.read(iprot)
                 else:
                     iprot.skip(ftype)
             elif fid == 204:
                 if ftype == TType.STRUCT:
-                    self.executionInfo = ai.chronon.api.common.ttypes.ExecutionInfo()
+                    self.executionInfo = gen_thrift.common.ttypes.ExecutionInfo()
                     self.executionInfo.read(iprot)
                 else:
                     iprot.skip(ftype)
@@ -1530,38 +1552,46 @@ class MetaData(object):
         if self.additionalOutputPartitionColumns is not None:
             oprot.writeFieldBegin('additionalOutputPartitionColumns', TType.LIST, 5)
             oprot.writeListBegin(TType.STRING, len(self.additionalOutputPartitionColumns))
-            for iter110 in self.additionalOutputPartitionColumns:
-                oprot.writeString(iter110.encode('utf-8') if sys.version_info[0] == 2 else iter110)
+            for iter117 in self.additionalOutputPartitionColumns:
+                oprot.writeString(iter117.encode('utf-8') if sys.version_info[0] == 2 else iter117)
             oprot.writeListEnd()
             oprot.writeFieldEnd()
         if self.tableProperties is not None:
             oprot.writeFieldBegin('tableProperties', TType.MAP, 6)
             oprot.writeMapBegin(TType.STRING, TType.STRING, len(self.tableProperties))
-            for kiter111, viter112 in self.tableProperties.items():
-                oprot.writeString(kiter111.encode('utf-8') if sys.version_info[0] == 2 else kiter111)
-                oprot.writeString(viter112.encode('utf-8') if sys.version_info[0] == 2 else viter112)
+            for kiter118, viter119 in self.tableProperties.items():
+                oprot.writeString(kiter118.encode('utf-8') if sys.version_info[0] == 2 else kiter118)
+                oprot.writeString(viter119.encode('utf-8') if sys.version_info[0] == 2 else viter119)
             oprot.writeMapEnd()
             oprot.writeFieldEnd()
         if self.tags is not None:
             oprot.writeFieldBegin('tags', TType.MAP, 20)
             oprot.writeMapBegin(TType.STRING, TType.STRING, len(self.tags))
-            for kiter113, viter114 in self.tags.items():
-                oprot.writeString(kiter113.encode('utf-8') if sys.version_info[0] == 2 else kiter113)
-                oprot.writeString(viter114.encode('utf-8') if sys.version_info[0] == 2 else viter114)
+            for kiter120, viter121 in self.tags.items():
+                oprot.writeString(kiter120.encode('utf-8') if sys.version_info[0] == 2 else kiter120)
+                oprot.writeString(viter121.encode('utf-8') if sys.version_info[0] == 2 else viter121)
             oprot.writeMapEnd()
             oprot.writeFieldEnd()
         if self.columnTags is not None:
             oprot.writeFieldBegin('columnTags', TType.MAP, 21)
             oprot.writeMapBegin(TType.STRING, TType.MAP, len(self.columnTags))
-            for kiter115, viter116 in self.columnTags.items():
-                oprot.writeString(kiter115.encode('utf-8') if sys.version_info[0] == 2 else kiter115)
-                oprot.writeMapBegin(TType.STRING, TType.STRING, len(viter116))
-                for kiter117, viter118 in viter116.items():
-                    oprot.writeString(kiter117.encode('utf-8') if sys.version_info[0] == 2 else kiter117)
-                    oprot.writeString(viter118.encode('utf-8') if sys.version_info[0] == 2 else viter118)
+            for kiter122, viter123 in self.columnTags.items():
+                oprot.writeString(kiter122.encode('utf-8') if sys.version_info[0] == 2 else kiter122)
+                oprot.writeMapBegin(TType.STRING, TType.STRING, len(viter123))
+                for kiter124, viter125 in viter123.items():
+                    oprot.writeString(kiter124.encode('utf-8') if sys.version_info[0] == 2 else kiter124)
+                    oprot.writeString(viter125.encode('utf-8') if sys.version_info[0] == 2 else viter125)
                 oprot.writeMapEnd()
             oprot.writeMapEnd()
             oprot.writeFieldEnd()
+        if self.columnHashes is not None:
+            oprot.writeFieldBegin('columnHashes', TType.MAP, 22)
+            oprot.writeMapBegin(TType.STRING, TType.STRING, len(self.columnHashes))
+            for kiter126, viter127 in self.columnHashes.items():
+                oprot.writeString(kiter126.encode('utf-8') if sys.version_info[0] == 2 else kiter126)
+                oprot.writeString(viter127.encode('utf-8') if sys.version_info[0] == 2 else viter127)
+            oprot.writeMapEnd()
+            oprot.writeFieldEnd()
         if self.online is not None:
             oprot.writeFieldBegin('online', TType.BOOL, 100)
             oprot.writeBool(self.online)
@@ -1616,6 +1646,98 @@ class MetaData(object):
         return not (self == other)
 
 
+class Conf(object):
+    """
+    Attributes:
+     - name
+     - hash
+     - contents
+     - confType
+
+    """
+    thrift_spec = None
+
+
+    def __init__(self, name = None, hash = None, contents = None, confType = None,):
+        self.name = name
+        self.hash = hash
+        self.contents = contents
+        self.confType = confType
+
+    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.STRING:
+                    self.contents = iprot.readString().decode('utf-8', errors='replace') if sys.version_info[0] == 2 else iprot.readString()
+                else:
+                    iprot.skip(ftype)
+            elif fid == 4:
+                if ftype == TType.I32:
+                    self.confType = 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('Conf')
+        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.contents is not None:
+            oprot.writeFieldBegin('contents', TType.STRING, 3)
+            oprot.writeString(self.contents.encode('utf-8') if sys.version_info[0] == 2 else self.contents)
+            oprot.writeFieldEnd()
+        if self.confType is not None:
+            oprot.writeFieldBegin('confType', TType.I32, 4)
+            oprot.writeI32(self.confType)
+            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 GroupBy(object):
     """
     Attributes:
@@ -1658,32 +1780,32 @@ class GroupBy(object):
             elif fid == 2:
                 if ftype == TType.LIST:
                     self.sources = []
-                    (_etype122, _size119) = iprot.readListBegin()
-                    for _i123 in range(_size119):
-                        _elem124 = Source()
-                        _elem124.read(iprot)
-                        self.sources.append(_elem124)
+                    (_etype131, _size128) = iprot.readListBegin()
+                    for _i132 in range(_size128):
+                        _elem133 = Source()
+                        _elem133.read(iprot)
+                        self.sources.append(_elem133)
                     iprot.readListEnd()
                 else:
                     iprot.skip(ftype)
             elif fid == 3:
                 if ftype == TType.LIST:
                     self.keyColumns = []
-                    (_etype128, _size125) = iprot.readListBegin()
-                    for _i129 in range(_size125):
-                        _elem130 = iprot.readString().decode('utf-8', errors='replace') if sys.version_info[0] == 2 else iprot.readString()
-                        self.keyColumns.append(_elem130)
+                    (_etype137, _size134) = iprot.readListBegin()
+                    for _i138 in range(_size134):
+                        _elem139 = iprot.readString().decode('utf-8', errors='replace') if sys.version_info[0] == 2 else iprot.readString()
+                        self.keyColumns.append(_elem139)
                     iprot.readListEnd()
                 else:
                     iprot.skip(ftype)
             elif fid == 4:
                 if ftype == TType.LIST:
                     self.aggregations = []
-                    (_etype134, _size131) = iprot.readListBegin()
-                    for _i135 in range(_size131):
-                        _elem136 = Aggregation()
-                        _elem136.read(iprot)
-                        self.aggregations.append(_elem136)
+                    (_etype143, _size140) = iprot.readListBegin()
+                    for _i144 in range(_size140):
+                        _elem145 = Aggregation()
+                        _elem145.read(iprot)
+                        self.aggregations.append(_elem145)
                     iprot.readListEnd()
                 else:
                     iprot.skip(ftype)
@@ -1700,11 +1822,11 @@ class GroupBy(object):
             elif fid == 7:
                 if ftype == TType.LIST:
                     self.derivations = []
-                    (_etype140, _size137) = iprot.readListBegin()
-                    for _i141 in range(_size137):
-                        _elem142 = Derivation()
-                        _elem142.read(iprot)
-                        self.derivations.append(_elem142)
+                    (_etype149, _size146) = iprot.readListBegin()
+                    for _i150 in range(_size146):
+                        _elem151 = Derivation()
+                        _elem151.read(iprot)
+                        self.derivations.append(_elem151)
                     iprot.readListEnd()
                 else:
                     iprot.skip(ftype)
@@ -1726,22 +1848,22 @@ class GroupBy(object):
         if self.sources is not None:
             oprot.writeFieldBegin('sources', TType.LIST, 2)
             oprot.writeListBegin(TType.STRUCT, len(self.sources))
-            for iter143 in self.sources:
-                iter143.write(oprot)
+            for iter152 in self.sources:
+                iter152.write(oprot)
             oprot.writeListEnd()
             oprot.writeFieldEnd()
         if self.keyColumns is not None:
             oprot.writeFieldBegin('keyColumns', TType.LIST, 3)
             oprot.writeListBegin(TType.STRING, len(self.keyColumns))
-            for iter144 in self.keyColumns:
-                oprot.writeString(iter144.encode('utf-8') if sys.version_info[0] == 2 else iter144)
+            for iter153 in self.keyColumns:
+                oprot.writeString(iter153.encode('utf-8') if sys.version_info[0] == 2 else iter153)
             oprot.writeListEnd()
             oprot.writeFieldEnd()
         if self.aggregations is not None:
             oprot.writeFieldBegin('aggregations', TType.LIST, 4)
             oprot.writeListBegin(TType.STRUCT, len(self.aggregations))
-            for iter145 in self.aggregations:
-                iter145.write(oprot)
+            for iter154 in self.aggregations:
+                iter154.write(oprot)
             oprot.writeListEnd()
             oprot.writeFieldEnd()
         if self.accuracy is not None:
@@ -1755,8 +1877,8 @@ class GroupBy(object):
         if self.derivations is not None:
             oprot.writeFieldBegin('derivations', TType.LIST, 7)
             oprot.writeListBegin(TType.STRUCT, len(self.derivations))
-            for iter146 in self.derivations:
-                iter146.write(oprot)
+            for iter155 in self.derivations:
+                iter155.write(oprot)
             oprot.writeListEnd()
             oprot.writeFieldEnd()
         oprot.writeFieldStop()
@@ -1813,11 +1935,11 @@ class JoinPart(object):
             elif fid == 2:
                 if ftype == TType.MAP:
                     self.keyMapping = {}
-                    (_ktype148, _vtype149, _size147) = iprot.readMapBegin()
-                    for _i151 in range(_size147):
-                        _key152 = iprot.readString().decode('utf-8', errors='replace') if sys.version_info[0] == 2 else iprot.readString()
-                        _val153 = iprot.readString().decode('utf-8', errors='replace') if sys.version_info[0] == 2 else iprot.readString()
-                        self.keyMapping[_key152] = _val153
+                    (_ktype157, _vtype158, _size156) = iprot.readMapBegin()
+                    for _i160 in range(_size156):
+                        _key161 = iprot.readString().decode('utf-8', errors='replace') if sys.version_info[0] == 2 else iprot.readString()
+                        _val162 = iprot.readString().decode('utf-8', errors='replace') if sys.version_info[0] == 2 else iprot.readString()
+                        self.keyMapping[_key161] = _val162
                     iprot.readMapEnd()
                 else:
                     iprot.skip(ftype)
@@ -1849,9 +1971,9 @@ class JoinPart(object):
         if self.keyMapping is not None:
             oprot.writeFieldBegin('keyMapping', TType.MAP, 2)
             oprot.writeMapBegin(TType.STRING, TType.STRING, len(self.keyMapping))
-            for kiter154, viter155 in self.keyMapping.items():
-                oprot.writeString(kiter154.encode('utf-8') if sys.version_info[0] == 2 else kiter154)
-                oprot.writeString(viter155.encode('utf-8') if sys.version_info[0] == 2 else viter155)
+            for kiter163, viter164 in self.keyMapping.items():
+                oprot.writeString(kiter163.encode('utf-8') if sys.version_info[0] == 2 else kiter163)
+                oprot.writeString(viter164.encode('utf-8') if sys.version_info[0] == 2 else viter164)
             oprot.writeMapEnd()
             oprot.writeFieldEnd()
         if self.prefix is not None:
@@ -1914,11 +2036,11 @@ class ExternalPart(object):
             elif fid == 2:
                 if ftype == TType.MAP:
                     self.keyMapping = {}
-                    (_ktype157, _vtype158, _size156) = iprot.readMapBegin()
-                    for _i160 in range(_size156):
-                        _key161 = iprot.readString().decode('utf-8', errors='replace') if sys.version_info[0] == 2 else iprot.readString()
-                        _val162 = iprot.readString().decode('utf-8', errors='replace') if sys.version_info[0] == 2 else iprot.readString()
-                        self.keyMapping[_key161] = _val162
+                    (_ktype166, _vtype167, _size165) = iprot.readMapBegin()
+                    for _i169 in range(_size165):
+                        _key170 = iprot.readString().decode('utf-8', errors='replace') if sys.version_info[0] == 2 else iprot.readString()
+                        _val171 = iprot.readString().decode('utf-8', errors='replace') if sys.version_info[0] == 2 else iprot.readString()
+                        self.keyMapping[_key170] = _val171
                     iprot.readMapEnd()
                 else:
                     iprot.skip(ftype)
@@ -1945,9 +2067,9 @@ class ExternalPart(object):
         if self.keyMapping is not None:
             oprot.writeFieldBegin('keyMapping', TType.MAP, 2)
             oprot.writeMapBegin(TType.STRING, TType.STRING, len(self.keyMapping))
-            for kiter163, viter164 in self.keyMapping.items():
-                oprot.writeString(kiter163.encode('utf-8') if sys.version_info[0] == 2 else kiter163)
-                oprot.writeString(viter164.encode('utf-8') if sys.version_info[0] == 2 else viter164)
+            for kiter172, viter173 in self.keyMapping.items():
+                oprot.writeString(kiter172.encode('utf-8') if sys.version_info[0] == 2 else kiter172)
+                oprot.writeString(viter173.encode('utf-8') if sys.version_info[0] == 2 else viter173)
             oprot.writeMapEnd()
             oprot.writeFieldEnd()
         if self.prefix is not None:
@@ -2109,38 +2231,38 @@ class Join(object):
             elif fid == 3:
                 if ftype == TType.LIST:
                     self.joinParts = []
-                    (_etype168, _size165) = iprot.readListBegin()
-                    for _i169 in range(_size165):
-                        _elem170 = JoinPart()
-                        _elem170.read(iprot)
-                        self.joinParts.append(_elem170)
+                    (_etype177, _size174) = iprot.readListBegin()
+                    for _i178 in range(_size174):
+                        _elem179 = JoinPart()
+                        _elem179.read(iprot)
+                        self.joinParts.append(_elem179)
                     iprot.readListEnd()
                 else:
                     iprot.skip(ftype)
             elif fid == 4:
                 if ftype == TType.MAP:
                     self.skewKeys = {}
-                    (_ktype172, _vtype173, _size171) = iprot.readMapBegin()
-                    for _i175 in range(_size171):
-                        _key176 = iprot.readString().decode('utf-8', errors='replace') if sys.version_info[0] == 2 else iprot.readString()
-                        _val177 = []
-                        (_etype181, _size178) = iprot.readListBegin()
-                        for _i182 in range(_size178):
-                            _elem183 = iprot.readString().decode('utf-8', errors='replace') if sys.version_info[0] == 2 else iprot.readString()
-                            _val177.append(_elem183)
+                    (_ktype181, _vtype182, _size180) = iprot.readMapBegin()
+                    for _i184 in range(_size180):
+                        _key185 = iprot.readString().decode('utf-8', errors='replace') if sys.version_info[0] == 2 else iprot.readString()
+                        _val186 = []
+                        (_etype190, _size187) = iprot.readListBegin()
+                        for _i191 in range(_size187):
+                            _elem192 = iprot.readString().decode('utf-8', errors='replace') if sys.version_info[0] == 2 else iprot.readString()
+                            _val186.append(_elem192)
                         iprot.readListEnd()
-                        self.skewKeys[_key176] = _val177
+                        self.skewKeys[_key185] = _val186
                     iprot.readMapEnd()
                 else:
                     iprot.skip(ftype)
             elif fid == 5:
                 if ftype == TType.LIST:
                     self.onlineExternalParts = []
-                    (_etype187, _size184) = iprot.readListBegin()
-                    for _i188 in range(_size184):
-                        _elem189 = ExternalPart()
-                        _elem189.read(iprot)
-                        self.onlineExternalParts.append(_elem189)
+                    (_etype196, _size193) = iprot.readListBegin()
+                    for _i197 in range(_size193):
+                        _elem198 = ExternalPart()
+                        _elem198.read(iprot)
+                        self.onlineExternalParts.append(_elem198)
                     iprot.readListEnd()
                 else:
                     iprot.skip(ftype)
@@ -2153,32 +2275,32 @@ class Join(object):
             elif fid == 7:
                 if ftype == TType.LIST:
                     self.bootstrapParts = []
-                    (_etype193, _size190) = iprot.readListBegin()
-                    for _i194 in range(_size190):
-                        _elem195 = BootstrapPart()
-                        _elem195.read(iprot)
-                        self.bootstrapParts.append(_elem195)
+                    (_etype202, _size199) = iprot.readListBegin()
+                    for _i203 in range(_size199):
+                        _elem204 = BootstrapPart()
+                        _elem204.read(iprot)
+                        self.bootstrapParts.append(_elem204)
                     iprot.readListEnd()
                 else:
                     iprot.skip(ftype)
             elif fid == 8:
                 if ftype == TType.LIST:
                     self.rowIds = []
-                    (_etype199, _size196) = iprot.readListBegin()
-                    for _i200 in range(_size196):
-                        _elem201 = iprot.readString().decode('utf-8', errors='replace') if sys.version_info[0] == 2 else iprot.readString()
-                        self.rowIds.append(_elem201)
+                    (_etype208, _size205) = iprot.readListBegin()
+                    for _i209 in range(_size205):
+                        _elem210 = iprot.readString().decode('utf-8', errors='replace') if sys.version_info[0] == 2 else iprot.readString()
+                        self.rowIds.append(_elem210)
                     iprot.readListEnd()
                 else:
                     iprot.skip(ftype)
             elif fid == 9:
                 if ftype == TType.LIST:
                     self.derivations = []
-                    (_etype205, _size202) = iprot.readListBegin()
-                    for _i206 in range(_size202):
-                        _elem207 = Derivation()
-                        _elem207.read(iprot)
-                        self.derivations.append(_elem207)
+                    (_etype214, _size211) = iprot.readListBegin()
+                    for _i215 in range(_size211):
+                        _elem216 = Derivation()
+                        _elem216.read(iprot)
+                        self.derivations.append(_elem216)
                     iprot.readListEnd()
                 else:
                     iprot.skip(ftype)
@@ -2209,26 +2331,26 @@ class Join(object):
         if self.joinParts is not None:
             oprot.writeFieldBegin('joinParts', TType.LIST, 3)
             oprot.writeListBegin(TType.STRUCT, len(self.joinParts))
-            for iter208 in self.joinParts:
-                iter208.write(oprot)
+            for iter217 in self.joinParts:
+                iter217.write(oprot)
             oprot.writeListEnd()
             oprot.writeFieldEnd()
         if self.skewKeys is not None:
             oprot.writeFieldBegin('skewKeys', TType.MAP, 4)
             oprot.writeMapBegin(TType.STRING, TType.LIST, len(self.skewKeys))
-            for kiter209, viter210 in self.skewKeys.items():
-                oprot.writeString(kiter209.encode('utf-8') if sys.version_info[0] == 2 else kiter209)
-                oprot.writeListBegin(TType.STRING, len(viter210))
-                for iter211 in viter210:
-                    oprot.writeString(iter211.encode('utf-8') if sys.version_info[0] == 2 else iter211)
+            for kiter218, viter219 in self.skewKeys.items():
+                oprot.writeString(kiter218.encode('utf-8') if sys.version_info[0] == 2 else kiter218)
+                oprot.writeListBegin(TType.STRING, len(viter219))
+                for iter220 in viter219:
+                    oprot.writeString(iter220.encode('utf-8') if sys.version_info[0] == 2 else iter220)
                 oprot.writeListEnd()
             oprot.writeMapEnd()
             oprot.writeFieldEnd()
         if self.onlineExternalParts is not None:
             oprot.writeFieldBegin('onlineExternalParts', TType.LIST, 5)
             oprot.writeListBegin(TType.STRUCT, len(self.onlineExternalParts))
-            for iter212 in self.onlineExternalParts:
-                iter212.write(oprot)
+            for iter221 in self.onlineExternalParts:
+                iter221.write(oprot)
             oprot.writeListEnd()
             oprot.writeFieldEnd()
         if self.labelParts is not None:
@@ -2238,22 +2360,22 @@ class Join(object):
         if self.bootstrapParts is not None:
             oprot.writeFieldBegin('bootstrapParts', TType.LIST, 7)
             oprot.writeListBegin(TType.STRUCT, len(self.bootstrapParts))
-            for iter213 in self.bootstrapParts:
-                iter213.write(oprot)
+            for iter222 in self.bootstrapParts:
+                iter222.write(oprot)
             oprot.writeListEnd()
             oprot.writeFieldEnd()
         if self.rowIds is not None:
             oprot.writeFieldBegin('rowIds', TType.LIST, 8)
             oprot.writeListBegin(TType.STRING, len(self.rowIds))
-            for iter214 in self.rowIds:
-                oprot.writeString(iter214.encode('utf-8') if sys.version_info[0] == 2 else iter214)
+            for iter223 in self.rowIds:
+                oprot.writeString(iter223.encode('utf-8') if sys.version_info[0] == 2 else iter223)
             oprot.writeListEnd()
             oprot.writeFieldEnd()
         if self.derivations is not None:
             oprot.writeFieldBegin('derivations', TType.LIST, 9)
             oprot.writeListBegin(TType.STRUCT, len(self.derivations))
-            for iter215 in self.derivations:
-                iter215.write(oprot)
+            for iter224 in self.derivations:
+                iter224.write(oprot)
             oprot.writeListEnd()
             oprot.writeFieldEnd()
         if self.useLongNames is not None:
@@ -2325,10 +2447,10 @@ class BootstrapPart(object):
             elif fid == 4:
                 if ftype == TType.LIST:
                     self.keyColumns = []
-                    (_etype219, _size216) = iprot.readListBegin()
-                    for _i220 in range(_size216):
-                        _elem221 = iprot.readString().decode('utf-8', errors='replace') if sys.version_info[0] == 2 else iprot.readString()
-                        self.keyColumns.append(_elem221)
+                    (_etype228, _size225) = iprot.readListBegin()
+                    for _i229 in range(_size225):
+                        _elem230 = iprot.readString().decode('utf-8', errors='replace') if sys.version_info[0] == 2 else iprot.readString()
+                        self.keyColumns.append(_elem230)
                     iprot.readListEnd()
                 else:
                     iprot.skip(ftype)
@@ -2358,8 +2480,8 @@ class BootstrapPart(object):
         if self.keyColumns is not None:
             oprot.writeFieldBegin('keyColumns', TType.LIST, 4)
             oprot.writeListBegin(TType.STRING, len(self.keyColumns))
-            for iter222 in self.keyColumns:
-                oprot.writeString(iter222.encode('utf-8') if sys.version_info[0] == 2 else iter222)
+            for iter231 in self.keyColumns:
+                oprot.writeString(iter231.encode('utf-8') if sys.version_info[0] == 2 else iter231)
             oprot.writeListEnd()
             oprot.writeFieldEnd()
         oprot.writeFieldStop()
@@ -2410,11 +2532,11 @@ class LabelParts(object):
             if fid == 1:
                 if ftype == TType.LIST:
                     self.labels = []
-                    (_etype226, _size223) = iprot.readListBegin()
-                    for _i227 in range(_size223):
-                        _elem228 = JoinPart()
-                        _elem228.read(iprot)
-                        self.labels.append(_elem228)
+                    (_etype235, _size232) = iprot.readListBegin()
+                    for _i236 in range(_size232):
+                        _elem237 = JoinPart()
+                        _elem237.read(iprot)
+                        self.labels.append(_elem237)
                     iprot.readListEnd()
                 else:
                     iprot.skip(ftype)
@@ -2448,8 +2570,8 @@ class LabelParts(object):
         if self.labels is not None:
             oprot.writeFieldBegin('labels', TType.LIST, 1)
             oprot.writeListBegin(TType.STRUCT, len(self.labels))
-            for iter229 in self.labels:
-                iter229.write(oprot)
+            for iter238 in self.labels:
+                iter238.write(oprot)
             oprot.writeListEnd()
             oprot.writeFieldEnd()
         if self.leftStartOffset is not None:
@@ -2701,11 +2823,11 @@ class TDataType(object):
             elif fid == 2:
                 if ftype == TType.LIST:
                     self.params = []
-                    (_etype233, _size230) = iprot.readListBegin()
-                    for _i234 in range(_size230):
-                        _elem235 = DataField()
-                        _elem235.read(iprot)
-                        self.params.append(_elem235)
+                    (_etype242, _size239) = iprot.readListBegin()
+                    for _i243 in range(_size239):
+                        _elem244 = DataField()
+                        _elem244.read(iprot)
+                        self.params.append(_elem244)
                     iprot.readListEnd()
                 else:
                     iprot.skip(ftype)
@@ -2732,8 +2854,8 @@ class TDataType(object):
         if self.params is not None:
             oprot.writeFieldBegin('params', TType.LIST, 2)
             oprot.writeListBegin(TType.STRUCT, len(self.params))
-            for iter236 in self.params:
-                iter236.write(oprot)
+            for iter245 in self.params:
+                iter245.write(oprot)
             oprot.writeListEnd()
             oprot.writeFieldEnd()
         if self.name is not None:
@@ -2794,10 +2916,10 @@ class DataSpec(object):
             elif fid == 2:
                 if ftype == TType.LIST:
                     self.partitionColumns = []
-                    (_etype240, _size237) = iprot.readListBegin()
-                    for _i241 in range(_size237):
-                        _elem242 = iprot.readString().decode('utf-8', errors='replace') if sys.version_info[0] == 2 else iprot.readString()
-                        self.partitionColumns.append(_elem242)
+                    (_etype249, _size246) = iprot.readListBegin()
+                    for _i250 in range(_size246):
+                        _elem251 = iprot.readString().decode('utf-8', errors='replace') if sys.version_info[0] == 2 else iprot.readString()
+                        self.partitionColumns.append(_elem251)
                     iprot.readListEnd()
                 else:
                     iprot.skip(ftype)
@@ -2809,11 +2931,11 @@ class DataSpec(object):
             elif fid == 4:
                 if ftype == TType.MAP:
                     self.props = {}
-                    (_ktype244, _vtype245, _size243) = iprot.readMapBegin()
-                    for _i247 in range(_size243):
-                        _key248 = iprot.readString().decode('utf-8', errors='replace') if sys.version_info[0] == 2 else iprot.readString()
-                        _val249 = iprot.readString().decode('utf-8', errors='replace') if sys.version_info[0] == 2 else iprot.readString()
-                        self.props[_key248] = _val249
+                    (_ktype253, _vtype254, _size252) = iprot.readMapBegin()
+                    for _i256 in range(_size252):
+                        _key257 = iprot.readString().decode('utf-8', errors='replace') if sys.version_info[0] == 2 else iprot.readString()
+                        _val258 = iprot.readString().decode('utf-8', errors='replace') if sys.version_info[0] == 2 else iprot.readString()
+                        self.props[_key257] = _val258
                     iprot.readMapEnd()
                 else:
                     iprot.skip(ftype)
@@ -2835,8 +2957,8 @@ class DataSpec(object):
         if self.partitionColumns is not None:
             oprot.writeFieldBegin('partitionColumns', TType.LIST, 2)
             oprot.writeListBegin(TType.STRING, len(self.partitionColumns))
-            for iter250 in self.partitionColumns:
-                oprot.writeString(iter250.encode('utf-8') if sys.version_info[0] == 2 else iter250)
+            for iter259 in self.partitionColumns:
+                oprot.writeString(iter259.encode('utf-8') if sys.version_info[0] == 2 else iter259)
             oprot.writeListEnd()
             oprot.writeFieldEnd()
         if self.retentionDays is not None:
@@ -2846,9 +2968,9 @@ class DataSpec(object):
         if self.props is not None:
             oprot.writeFieldBegin('props', TType.MAP, 4)
             oprot.writeMapBegin(TType.STRING, TType.STRING, len(self.props))
-            for kiter251, viter252 in self.props.items():
-                oprot.writeString(kiter251.encode('utf-8') if sys.version_info[0] == 2 else kiter251)
-                oprot.writeString(viter252.encode('utf-8') if sys.version_info[0] == 2 else viter252)
+            for kiter260, viter261 in self.props.items():
+                oprot.writeString(kiter260.encode('utf-8') if sys.version_info[0] == 2 else kiter260)
+                oprot.writeString(viter261.encode('utf-8') if sys.version_info[0] == 2 else viter261)
             oprot.writeMapEnd()
             oprot.writeFieldEnd()
         oprot.writeFieldStop()
@@ -2924,11 +3046,11 @@ class Model(object):
             elif fid == 5:
                 if ftype == TType.MAP:
                     self.modelParams = {}
-                    (_ktype254, _vtype255, _size253) = iprot.readMapBegin()
-                    for _i257 in range(_size253):
-                        _key258 = iprot.readString().decode('utf-8', errors='replace') if sys.version_info[0] == 2 else iprot.readString()
-                        _val259 = iprot.readString().decode('utf-8', errors='replace') if sys.version_info[0] == 2 else iprot.readString()
-                        self.modelParams[_key258] = _val259
+                    (_ktype263, _vtype264, _size262) = iprot.readMapBegin()
+                    for _i266 in range(_size262):
+                        _key267 = iprot.readString().decode('utf-8', errors='replace') if sys.version_info[0] == 2 else iprot.readString()
+                        _val268 = iprot.readString().decode('utf-8', errors='replace') if sys.version_info[0] == 2 else iprot.readString()
+                        self.modelParams[_key267] = _val268
                     iprot.readMapEnd()
                 else:
                     iprot.skip(ftype)
@@ -2962,9 +3084,9 @@ class Model(object):
         if self.modelParams is not None:
             oprot.writeFieldBegin('modelParams', TType.MAP, 5)
             oprot.writeMapBegin(TType.STRING, TType.STRING, len(self.modelParams))
-            for kiter260, viter261 in self.modelParams.items():
-                oprot.writeString(kiter260.encode('utf-8') if sys.version_info[0] == 2 else kiter260)
-                oprot.writeString(viter261.encode('utf-8') if sys.version_info[0] == 2 else viter261)
+            for kiter269, viter270 in self.modelParams.items():
+                oprot.writeString(kiter269.encode('utf-8') if sys.version_info[0] == 2 else kiter269)
+                oprot.writeString(viter270.encode('utf-8') if sys.version_info[0] == 2 else viter270)
             oprot.writeMapEnd()
             oprot.writeFieldEnd()
         oprot.writeFieldStop()
@@ -3043,29 +3165,29 @@ class Team(object):
             elif fid == 11:
                 if ftype == TType.MAP:
                     self.tableProperties = {}
-                    (_ktype263, _vtype264, _size262) = iprot.readMapBegin()
-                    for _i266 in range(_size262):
-                        _key267 = iprot.readString().decode('utf-8', errors='replace') if sys.version_info[0] == 2 else iprot.readString()
-                        _val268 = iprot.readString().decode('utf-8', errors='replace') if sys.version_info[0] == 2 else iprot.readString()
-                        self.tableProperties[_key267] = _val268
+                    (_ktype272, _vtype273, _size271) = iprot.readMapBegin()
+                    for _i275 in range(_size271):
+                        _key276 = iprot.readString().decode('utf-8', errors='replace') if sys.version_info[0] == 2 else iprot.readString()
+                        _val277 = iprot.readString().decode('utf-8', errors='replace') if sys.version_info[0] == 2 else iprot.readString()
+                        self.tableProperties[_key276] = _val277
                     iprot.readMapEnd()
                 else:
                     iprot.skip(ftype)
             elif fid == 20:
                 if ftype == TType.STRUCT:
-                    self.env = ai.chronon.api.common.ttypes.EnvironmentVariables()
+                    self.env = gen_thrift.common.ttypes.EnvironmentVariables()
                     self.env.read(iprot)
                 else:
                     iprot.skip(ftype)
             elif fid == 21:
                 if ftype == TType.STRUCT:
-                    self.conf = ai.chronon.api.common.ttypes.ConfigProperties()
+                    self.conf = gen_thrift.common.ttypes.ConfigProperties()
                     self.conf.read(iprot)
                 else:
                     iprot.skip(ftype)
             elif fid == 22:
                 if ftype == TType.STRUCT:
-                    self.clusterConf = ai.chronon.api.common.ttypes.ClusterConfigProperties()
+                    self.clusterConf = gen_thrift.common.ttypes.ClusterConfigProperties()
                     self.clusterConf.read(iprot)
                 else:
                     iprot.skip(ftype)
@@ -3099,9 +3221,9 @@ class Team(object):
         if self.tableProperties is not None:
             oprot.writeFieldBegin('tableProperties', TType.MAP, 11)
             oprot.writeMapBegin(TType.STRING, TType.STRING, len(self.tableProperties))
-            for kiter269, viter270 in self.tableProperties.items():
-                oprot.writeString(kiter269.encode('utf-8') if sys.version_info[0] == 2 else kiter269)
-                oprot.writeString(viter270.encode('utf-8') if sys.version_info[0] == 2 else viter270)
+            for kiter278, viter279 in self.tableProperties.items():
+                oprot.writeString(kiter278.encode('utf-8') if sys.version_info[0] == 2 else kiter278)
+                oprot.writeString(viter279.encode('utf-8') if sys.version_info[0] == 2 else viter279)
             oprot.writeMapEnd()
             oprot.writeFieldEnd()
         if self.env is not None:
@@ -3156,8 +3278,8 @@ Query.thrift_spec = (
     None,  # 19
     (20, TType.STRING, 'partitionColumn', 'UTF8', None, ),  # 20
     (21, TType.STRING, 'partitionFormat', 'UTF8', None, ),  # 21
-    (22, TType.STRUCT, 'partitionInterval', [ai.chronon.api.common.ttypes.Window, None], None, ),  # 22
-    (23, TType.STRUCT, 'partitionLag', [ai.chronon.api.common.ttypes.Window, None], None, ),  # 23
+    (22, TType.STRUCT, 'partitionInterval', [gen_thrift.common.ttypes.Window, None], None, ),  # 22
+    (23, TType.STRUCT, 'partitionLag', [gen_thrift.common.ttypes.Window, None], None, ),  # 23
     (24, TType.LIST, 'subPartitionsToWaitFor', (TType.STRING, 'UTF8', False), None, ),  # 24
 )
 all_structs.append(StagingQuery)
@@ -3167,9 +3289,9 @@ StagingQuery.thrift_spec = (
     (2, TType.STRING, 'query', 'UTF8', None, ),  # 2
     (3, TType.STRING, 'startPartition', 'UTF8', None, ),  # 3
     (4, TType.LIST, 'setups', (TType.STRING, 'UTF8', False), None, ),  # 4
-    (5, TType.STRING, 'partitionColumn', 'UTF8', None, ),  # 5
-    (6, TType.I32, 'engineType', None, None, ),  # 6
-    (7, TType.LIST, 'tableDependencies', (TType.STRUCT, [ai.chronon.api.common.ttypes.TableDependency, None], False), None, ),  # 7
+    (5, TType.I32, 'engineType', None, None, ),  # 5
+    (6, TType.LIST, 'tableDependencies', (TType.STRUCT, [gen_thrift.common.ttypes.TableDependency, None], False), None, ),  # 6
+    None,  # 7
     None,  # 8
     None,  # 9
     None,  # 10
@@ -3226,7 +3348,7 @@ Aggregation.thrift_spec = (
     (1, TType.STRING, 'inputColumn', 'UTF8', None, ),  # 1
     (2, TType.I32, 'operation', None, None, ),  # 2
     (3, TType.MAP, 'argMap', (TType.STRING, 'UTF8', TType.STRING, 'UTF8', False), None, ),  # 3
-    (4, TType.LIST, 'windows', (TType.STRUCT, [ai.chronon.api.common.ttypes.Window, None], False), None, ),  # 4
+    (4, TType.LIST, 'windows', (TType.STRUCT, [gen_thrift.common.ttypes.Window, None], False), None, ),  # 4
     (5, TType.LIST, 'buckets', (TType.STRING, 'UTF8', False), None, ),  # 5
 )
 all_structs.append(AggregationPart)
@@ -3235,7 +3357,7 @@ AggregationPart.thrift_spec = (
     (1, TType.STRING, 'inputColumn', 'UTF8', None, ),  # 1
     (2, TType.I32, 'operation', None, None, ),  # 2
     (3, TType.MAP, 'argMap', (TType.STRING, 'UTF8', TType.STRING, 'UTF8', False), None, ),  # 3
-    (4, TType.STRUCT, 'window', [ai.chronon.api.common.ttypes.Window, None], None, ),  # 4
+    (4, TType.STRUCT, 'window', [gen_thrift.common.ttypes.Window, None], None, ),  # 4
     (5, TType.STRING, 'bucket', 'UTF8', None, ),  # 5
 )
 all_structs.append(MetaData)
@@ -3262,7 +3384,7 @@ MetaData.thrift_spec = (
     None,  # 19
     (20, TType.MAP, 'tags', (TType.STRING, 'UTF8', TType.STRING, 'UTF8', False), None, ),  # 20
     (21, TType.MAP, 'columnTags', (TType.STRING, 'UTF8', TType.MAP, (TType.STRING, 'UTF8', TType.STRING, 'UTF8', False), False), None, ),  # 21
-    None,  # 22
+    (22, TType.MAP, 'columnHashes', (TType.STRING, 'UTF8', TType.STRING, 'UTF8', False), None, ),  # 22
     None,  # 23
     None,  # 24
     None,  # 25
@@ -3443,8 +3565,16 @@ MetaData.thrift_spec = (
     (200, TType.BOOL, 'consistencyCheck', None, None, ),  # 200
     (201, TType.DOUBLE, 'samplePercent', None, None, ),  # 201
     (202, TType.DOUBLE, 'consistencySamplePercent', None, None, ),  # 202
-    (203, TType.STRUCT, 'driftSpec', [ai.chronon.observability.ttypes.DriftSpec, None], None, ),  # 203
-    (204, TType.STRUCT, 'executionInfo', [ai.chronon.api.common.ttypes.ExecutionInfo, None], None, ),  # 204
+    (203, TType.STRUCT, 'driftSpec', [gen_thrift.observability.ttypes.DriftSpec, None], None, ),  # 203
+    (204, TType.STRUCT, 'executionInfo', [gen_thrift.common.ttypes.ExecutionInfo, None], None, ),  # 204
+)
+all_structs.append(Conf)
+Conf.thrift_spec = (
+    None,  # 0
+    (1, TType.STRING, 'name', 'UTF8', None, ),  # 1
+    (2, TType.STRING, 'hash', 'UTF8', None, ),  # 2
+    (3, TType.STRING, 'contents', 'UTF8', None, ),  # 3
+    (4, TType.I32, 'confType', None, None, ),  # 4
 )
 all_structs.append(GroupBy)
 GroupBy.thrift_spec = (
@@ -3616,9 +3746,9 @@ Team.thrift_spec = (
     None,  # 17
     None,  # 18
     None,  # 19
-    (20, TType.STRUCT, 'env', [ai.chronon.api.common.ttypes.EnvironmentVariables, None], None, ),  # 20
-    (21, TType.STRUCT, 'conf', [ai.chronon.api.common.ttypes.ConfigProperties, None], None, ),  # 21
-    (22, TType.STRUCT, 'clusterConf', [ai.chronon.api.common.ttypes.ClusterConfigProperties, None], None, ),  # 22
+    (20, TType.STRUCT, 'env', [gen_thrift.common.ttypes.EnvironmentVariables, None], None, ),  # 20
+    (21, TType.STRUCT, 'conf', [gen_thrift.common.ttypes.ConfigProperties, None], None, ),  # 21
+    (22, TType.STRUCT, 'clusterConf', [gen_thrift.common.ttypes.ClusterConfigProperties, None], None, ),  # 22
 )
 fix_spec(all_structs)
 del all_structs