weave-typescript 0.43.0 → 0.45.0

package/dist/weaveapi/ingestion/v1/ingestion.pb.js

@@ -5,8 +5,8 @@
 //   protoc               unknown
 // source: weaveapi/ingestion/v1/ingestion.proto
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.ListRestrictedGroupsRequest = exports.DeleteDocumentResponse = exports.DeleteDocumentRequest = exports.UpdateDocumentTagsResponse = exports.UpdateDocumentTagsRequest = exports.UpdateDocumentSecurityResponse = exports.UpdateDocumentSecurityRequest = exports.GetDocumentSourceResponse = exports.GetDocumentSourceRequest = exports.GetDocumentDetailResponse = exports.GetDocumentDetailRequest = exports.ListDocumentsResponse = exports.ListDocumentsRequest = exports.ListIngestionHistoryResponse = exports.ListIngestionHistoryRequest = exports.GetIngestionRunRequest = exports.StreamPipelineRequest = exports.UploadFileResponse = exports.UploadFileRequest = exports.IngestionHistoryEntry = exports.IngestionRun = exports.Thread = exports.DocumentSourceBlock = exports.DocumentDetail = exports.DocumentActionPermissions = exports.DocumentSummary = exports.DocumentThreadCounts = exports.Document = exports.Entity = exports.FacetConfig = exports.FacetFieldDefinition = exports.TemporalValidity = exports.SourceLocation = exports.RestrictedGroupReplacement = exports.RestrictedGroup = exports.UserSensitivityClearance = exports.SensitivityTagDefinition = exports.DocumentTag = exports.RestrictedGroupReplacementTarget = exports.DocumentSourceBlockKind = exports.SortDirection = exports.DocumentSortField = exports.TemporalValidityKind = exports.ThreadActivityAction = exports.ThreadRetrievalStatus = exports.ThreadClassification = exports.IngestionRunStatus = exports.DocumentScope = exports.DocumentStatus = exports.protobufPackage = void 0;
-exports.PipelineEvent_PipelineFailed = exports.PipelineEvent_PipelineComplete = exports.PipelineEvent_DiffDecision = exports.PipelineEvent_ReconcileMatch = exports.PipelineEvent_EmbedComplete = exports.PipelineEvent_EmbedStarted = exports.PipelineEvent_ExtractThread = exports.PipelineEvent_ClassifyTag = exports.PipelineEvent_ParseComplete = exports.PipelineEvent_ParseStarted = exports.PipelineEvent = exports.RejectThreadResponse = exports.ApproveThreadResponse = exports.EditThreadResponse = exports.GetThreadResponse = exports.GetIngestionRunResponse = exports.StreamPipelineResponse = exports.BulkRejectThreadsResponse = exports.BulkRejectThreadsRequest = exports.BulkApproveThreadsResponse = exports.BulkApproveThreadsRequest = exports.RejectThreadRequest = exports.ApproveThreadRequest = exports.DeleteDocumentThreadResponse = exports.DeleteDocumentThreadRequest = exports.UpdateDocumentThreadResponse = exports.UpdateDocumentThreadRequest = exports.CreateDocumentThreadResponse = exports.CreateDocumentThreadRequest = exports.EditThreadRequest = exports.GetThreadRequest = exports.RetryIngestionResponse = exports.RetryIngestionRequest = exports.DeleteRestrictedGroupResponse = exports.DeleteRestrictedGroupRequest = exports.UpdateRestrictedGroupResponse = exports.UpdateRestrictedGroupRequest = exports.CreateRestrictedGroupResponse = exports.CreateRestrictedGroupRequest = exports.GetRestrictedGroupResponse = exports.GetRestrictedGroupRequest = exports.ListRestrictedGroupsResponse = void 0;
+exports.ListRestrictedGroupsResponse = exports.ListRestrictedGroupsRequest = exports.DeleteDocumentResponse = exports.DeleteDocumentRequest = exports.UpdateDocumentTagsResponse = exports.UpdateDocumentTagsRequest = exports.UpdateDocumentSecurityResponse = exports.UpdateDocumentSecurityRequest = exports.GetDocumentSourceResponse = exports.GetDocumentSourceRequest = exports.GetDocumentDetailResponse = exports.GetDocumentDetailRequest = exports.ListDocumentsResponse = exports.ListDocumentsRequest = exports.ListIngestionHistoryResponse = exports.ListIngestionHistoryRequest = exports.GetIngestionRunRequest = exports.StreamPipelineRequest = exports.UploadFileResponse = exports.UploadFileRequest = exports.IngestionHistoryEntry = exports.IngestionRun = exports.Thread = exports.DocumentSourceBlock = exports.DocumentDetail = exports.DocumentActionPermissions = exports.DocumentSummary = exports.DocumentThreadCounts = exports.Document = exports.ThreadFacet = exports.Entity = exports.FacetConfig = exports.FacetFieldDefinition = exports.TemporalValidity = exports.SourceLocation = exports.RestrictedGroupReplacement = exports.RestrictedGroup = exports.DocumentTag = exports.RestrictedGroupReplacementTarget = exports.DocumentSourceBlockKind = exports.SortDirection = exports.DocumentSortField = exports.TemporalValidityKind = exports.ThreadActivityAction = exports.ThreadRetrievalStatus = exports.ThreadClassification = exports.IngestionRunStatus = exports.DocumentScope = exports.DocumentStatus = exports.protobufPackage = void 0;
+exports.PipelineEvent_PipelineFailed = exports.PipelineEvent_PipelineComplete = exports.PipelineEvent_DiffDecision = exports.PipelineEvent_ReconcileMatch = exports.PipelineEvent_EmbedComplete = exports.PipelineEvent_EmbedStarted = exports.PipelineEvent_ExtractThread = exports.PipelineEvent_RefineComplete = exports.PipelineEvent_RefineThread = exports.PipelineEvent_RefineStarted = exports.PipelineEvent_ClassifyTag = exports.PipelineEvent_ParseComplete = exports.PipelineEvent_ParseStarted = exports.PipelineEvent = exports.RejectThreadResponse = exports.ApproveThreadResponse = exports.EditThreadResponse = exports.GetThreadResponse = exports.GetIngestionRunResponse = exports.StreamPipelineResponse = exports.BulkRejectThreadsResponse = exports.BulkRejectThreadsRequest = exports.BulkApproveThreadsResponse = exports.BulkApproveThreadsRequest = exports.RejectThreadRequest = exports.ApproveThreadRequest = exports.DeleteDocumentThreadResponse = exports.DeleteDocumentThreadRequest = exports.UpdateDocumentThreadResponse = exports.UpdateDocumentThreadRequest = exports.CreateDocumentThreadResponse = exports.CreateDocumentThreadRequest = exports.EditThreadRequest = exports.GetThreadRequest = exports.RetryIngestionResponse = exports.RetryIngestionRequest = exports.DeleteRestrictedGroupResponse = exports.DeleteRestrictedGroupRequest = exports.UpdateRestrictedGroupResponse = exports.UpdateRestrictedGroupRequest = exports.CreateRestrictedGroupResponse = exports.CreateRestrictedGroupRequest = exports.GetRestrictedGroupResponse = exports.GetRestrictedGroupRequest = void 0;
 exports.documentStatusFromJSON = documentStatusFromJSON;
 exports.documentStatusToJSON = documentStatusToJSON;
 exports.documentScopeFromJSON = documentScopeFromJSON;
@@ -47,6 +47,7 @@ var DocumentStatus;
     DocumentStatus[DocumentStatus["DOCUMENT_STATUS_COMMITTED"] = 8] = "DOCUMENT_STATUS_COMMITTED";
     DocumentStatus[DocumentStatus["DOCUMENT_STATUS_FAILED"] = 9] = "DOCUMENT_STATUS_FAILED";
     DocumentStatus[DocumentStatus["DOCUMENT_STATUS_STALE"] = 10] = "DOCUMENT_STATUS_STALE";
+    DocumentStatus[DocumentStatus["DOCUMENT_STATUS_REFINING"] = 11] = "DOCUMENT_STATUS_REFINING";
     DocumentStatus[DocumentStatus["UNRECOGNIZED"] = -1] = "UNRECOGNIZED";
 })(DocumentStatus || (exports.DocumentStatus = DocumentStatus = {}));
 function documentStatusFromJSON(object) {
@@ -84,6 +85,9 @@ function documentStatusFromJSON(object) {
         case 10:
         case "DOCUMENT_STATUS_STALE":
             return DocumentStatus.DOCUMENT_STATUS_STALE;
+        case 11:
+        case "DOCUMENT_STATUS_REFINING":
+            return DocumentStatus.DOCUMENT_STATUS_REFINING;
         case -1:
         case "UNRECOGNIZED":
         default:
@@ -114,6 +118,8 @@ function documentStatusToJSON(object) {
             return "DOCUMENT_STATUS_FAILED";
         case DocumentStatus.DOCUMENT_STATUS_STALE:
             return "DOCUMENT_STATUS_STALE";
+        case DocumentStatus.DOCUMENT_STATUS_REFINING:
+            return "DOCUMENT_STATUS_REFINING";
         case DocumentStatus.UNRECOGNIZED:
         default:
             return "UNRECOGNIZED";
@@ -175,6 +181,7 @@ var IngestionRunStatus;
     IngestionRunStatus[IngestionRunStatus["INGESTION_RUN_STATUS_REVIEW_PENDING"] = 8] = "INGESTION_RUN_STATUS_REVIEW_PENDING";
     IngestionRunStatus[IngestionRunStatus["INGESTION_RUN_STATUS_COMMITTED"] = 9] = "INGESTION_RUN_STATUS_COMMITTED";
     IngestionRunStatus[IngestionRunStatus["INGESTION_RUN_STATUS_FAILED"] = 10] = "INGESTION_RUN_STATUS_FAILED";
+    IngestionRunStatus[IngestionRunStatus["INGESTION_RUN_STATUS_REFINING"] = 11] = "INGESTION_RUN_STATUS_REFINING";
     IngestionRunStatus[IngestionRunStatus["UNRECOGNIZED"] = -1] = "UNRECOGNIZED";
 })(IngestionRunStatus || (exports.IngestionRunStatus = IngestionRunStatus = {}));
 function ingestionRunStatusFromJSON(object) {
@@ -212,6 +219,9 @@ function ingestionRunStatusFromJSON(object) {
         case 10:
         case "INGESTION_RUN_STATUS_FAILED":
             return IngestionRunStatus.INGESTION_RUN_STATUS_FAILED;
+        case 11:
+        case "INGESTION_RUN_STATUS_REFINING":
+            return IngestionRunStatus.INGESTION_RUN_STATUS_REFINING;
         case -1:
         case "UNRECOGNIZED":
         default:
@@ -242,6 +252,8 @@ function ingestionRunStatusToJSON(object) {
             return "INGESTION_RUN_STATUS_COMMITTED";
         case IngestionRunStatus.INGESTION_RUN_STATUS_FAILED:
             return "INGESTION_RUN_STATUS_FAILED";
+        case IngestionRunStatus.INGESTION_RUN_STATUS_REFINING:
+            return "INGESTION_RUN_STATUS_REFINING";
         case IngestionRunStatus.UNRECOGNIZED:
         default:
             return "UNRECOGNIZED";
@@ -690,305 +702,6 @@ exports.DocumentTag = {
         return message;
     },
 };
-function createBaseSensitivityTagDefinition() {
-    return {
-        organizationId: "",
-        slug: "",
-        label: "",
-        description: "",
-        extractionHint: "",
-        active: false,
-        createdAt: undefined,
-        updatedAt: undefined,
-    };
-}
-exports.SensitivityTagDefinition = {
-    encode(message, writer = new wire_1.BinaryWriter()) {
-        if (message.organizationId !== "") {
-            writer.uint32(10).string(message.organizationId);
-        }
-        if (message.slug !== "") {
-            writer.uint32(18).string(message.slug);
-        }
-        if (message.label !== "") {
-            writer.uint32(26).string(message.label);
-        }
-        if (message.description !== "") {
-            writer.uint32(34).string(message.description);
-        }
-        if (message.extractionHint !== "") {
-            writer.uint32(42).string(message.extractionHint);
-        }
-        if (message.active !== false) {
-            writer.uint32(48).bool(message.active);
-        }
-        if (message.createdAt !== undefined) {
-            timestamp_pb_1.Timestamp.encode(toTimestamp(message.createdAt), writer.uint32(58).fork()).join();
-        }
-        if (message.updatedAt !== undefined) {
-            timestamp_pb_1.Timestamp.encode(toTimestamp(message.updatedAt), writer.uint32(66).fork()).join();
-        }
-        return writer;
-    },
-    decode(input, length) {
-        const reader = input instanceof wire_1.BinaryReader ? input : new wire_1.BinaryReader(input);
-        const end = length === undefined ? reader.len : reader.pos + length;
-        const message = createBaseSensitivityTagDefinition();
-        while (reader.pos < end) {
-            const tag = reader.uint32();
-            switch (tag >>> 3) {
-                case 1: {
-                    if (tag !== 10) {
-                        break;
-                    }
-                    message.organizationId = reader.string();
-                    continue;
-                }
-                case 2: {
-                    if (tag !== 18) {
-                        break;
-                    }
-                    message.slug = reader.string();
-                    continue;
-                }
-                case 3: {
-                    if (tag !== 26) {
-                        break;
-                    }
-                    message.label = reader.string();
-                    continue;
-                }
-                case 4: {
-                    if (tag !== 34) {
-                        break;
-                    }
-                    message.description = reader.string();
-                    continue;
-                }
-                case 5: {
-                    if (tag !== 42) {
-                        break;
-                    }
-                    message.extractionHint = reader.string();
-                    continue;
-                }
-                case 6: {
-                    if (tag !== 48) {
-                        break;
-                    }
-                    message.active = reader.bool();
-                    continue;
-                }
-                case 7: {
-                    if (tag !== 58) {
-                        break;
-                    }
-                    message.createdAt = fromTimestamp(timestamp_pb_1.Timestamp.decode(reader, reader.uint32()));
-                    continue;
-                }
-                case 8: {
-                    if (tag !== 66) {
-                        break;
-                    }
-                    message.updatedAt = fromTimestamp(timestamp_pb_1.Timestamp.decode(reader, reader.uint32()));
-                    continue;
-                }
-            }
-            if ((tag & 7) === 4 || tag === 0) {
-                break;
-            }
-            reader.skip(tag & 7);
-        }
-        return message;
-    },
-    fromJSON(object) {
-        return {
-            organizationId: isSet(object.organizationId)
-                ? globalThis.String(object.organizationId)
-                : isSet(object.organization_id)
-                    ? globalThis.String(object.organization_id)
-                    : "",
-            slug: isSet(object.slug) ? globalThis.String(object.slug) : "",
-            label: isSet(object.label) ? globalThis.String(object.label) : "",
-            description: isSet(object.description) ? globalThis.String(object.description) : "",
-            extractionHint: isSet(object.extractionHint)
-                ? globalThis.String(object.extractionHint)
-                : isSet(object.extraction_hint)
-                    ? globalThis.String(object.extraction_hint)
-                    : "",
-            active: isSet(object.active) ? globalThis.Boolean(object.active) : false,
-            createdAt: isSet(object.createdAt)
-                ? fromJsonTimestamp(object.createdAt)
-                : isSet(object.created_at)
-                    ? fromJsonTimestamp(object.created_at)
-                    : undefined,
-            updatedAt: isSet(object.updatedAt)
-                ? fromJsonTimestamp(object.updatedAt)
-                : isSet(object.updated_at)
-                    ? fromJsonTimestamp(object.updated_at)
-                    : undefined,
-        };
-    },
-    toJSON(message) {
-        const obj = {};
-        if (message.organizationId !== "") {
-            obj.organizationId = message.organizationId;
-        }
-        if (message.slug !== "") {
-            obj.slug = message.slug;
-        }
-        if (message.label !== "") {
-            obj.label = message.label;
-        }
-        if (message.description !== "") {
-            obj.description = message.description;
-        }
-        if (message.extractionHint !== "") {
-            obj.extractionHint = message.extractionHint;
-        }
-        if (message.active !== false) {
-            obj.active = message.active;
-        }
-        if (message.createdAt !== undefined) {
-            obj.createdAt = message.createdAt.toISOString();
-        }
-        if (message.updatedAt !== undefined) {
-            obj.updatedAt = message.updatedAt.toISOString();
-        }
-        return obj;
-    },
-    create(base) {
-        return exports.SensitivityTagDefinition.fromPartial(base !== null && base !== void 0 ? base : {});
-    },
-    fromPartial(object) {
-        var _a, _b, _c, _d, _e, _f, _g, _h;
-        const message = createBaseSensitivityTagDefinition();
-        message.organizationId = (_a = object.organizationId) !== null && _a !== void 0 ? _a : "";
-        message.slug = (_b = object.slug) !== null && _b !== void 0 ? _b : "";
-        message.label = (_c = object.label) !== null && _c !== void 0 ? _c : "";
-        message.description = (_d = object.description) !== null && _d !== void 0 ? _d : "";
-        message.extractionHint = (_e = object.extractionHint) !== null && _e !== void 0 ? _e : "";
-        message.active = (_f = object.active) !== null && _f !== void 0 ? _f : false;
-        message.createdAt = (_g = object.createdAt) !== null && _g !== void 0 ? _g : undefined;
-        message.updatedAt = (_h = object.updatedAt) !== null && _h !== void 0 ? _h : undefined;
-        return message;
-    },
-};
-function createBaseUserSensitivityClearance() {
-    return { organizationId: "", userId: "", sensitivityTagSlug: "", createdAt: undefined };
-}
-exports.UserSensitivityClearance = {
-    encode(message, writer = new wire_1.BinaryWriter()) {
-        if (message.organizationId !== "") {
-            writer.uint32(10).string(message.organizationId);
-        }
-        if (message.userId !== "") {
-            writer.uint32(18).string(message.userId);
-        }
-        if (message.sensitivityTagSlug !== "") {
-            writer.uint32(26).string(message.sensitivityTagSlug);
-        }
-        if (message.createdAt !== undefined) {
-            timestamp_pb_1.Timestamp.encode(toTimestamp(message.createdAt), writer.uint32(34).fork()).join();
-        }
-        return writer;
-    },
-    decode(input, length) {
-        const reader = input instanceof wire_1.BinaryReader ? input : new wire_1.BinaryReader(input);
-        const end = length === undefined ? reader.len : reader.pos + length;
-        const message = createBaseUserSensitivityClearance();
-        while (reader.pos < end) {
-            const tag = reader.uint32();
-            switch (tag >>> 3) {
-                case 1: {
-                    if (tag !== 10) {
-                        break;
-                    }
-                    message.organizationId = reader.string();
-                    continue;
-                }
-                case 2: {
-                    if (tag !== 18) {
-                        break;
-                    }
-                    message.userId = reader.string();
-                    continue;
-                }
-                case 3: {
-                    if (tag !== 26) {
-                        break;
-                    }
-                    message.sensitivityTagSlug = reader.string();
-                    continue;
-                }
-                case 4: {
-                    if (tag !== 34) {
-                        break;
-                    }
-                    message.createdAt = fromTimestamp(timestamp_pb_1.Timestamp.decode(reader, reader.uint32()));
-                    continue;
-                }
-            }
-            if ((tag & 7) === 4 || tag === 0) {
-                break;
-            }
-            reader.skip(tag & 7);
-        }
-        return message;
-    },
-    fromJSON(object) {
-        return {
-            organizationId: isSet(object.organizationId)
-                ? globalThis.String(object.organizationId)
-                : isSet(object.organization_id)
-                    ? globalThis.String(object.organization_id)
-                    : "",
-            userId: isSet(object.userId)
-                ? globalThis.String(object.userId)
-                : isSet(object.user_id)
-                    ? globalThis.String(object.user_id)
-                    : "",
-            sensitivityTagSlug: isSet(object.sensitivityTagSlug)
-                ? globalThis.String(object.sensitivityTagSlug)
-                : isSet(object.sensitivity_tag_slug)
-                    ? globalThis.String(object.sensitivity_tag_slug)
-                    : "",
-            createdAt: isSet(object.createdAt)
-                ? fromJsonTimestamp(object.createdAt)
-                : isSet(object.created_at)
-                    ? fromJsonTimestamp(object.created_at)
-                    : undefined,
-        };
-    },
-    toJSON(message) {
-        const obj = {};
-        if (message.organizationId !== "") {
-            obj.organizationId = message.organizationId;
-        }
-        if (message.userId !== "") {
-            obj.userId = message.userId;
-        }
-        if (message.sensitivityTagSlug !== "") {
-            obj.sensitivityTagSlug = message.sensitivityTagSlug;
-        }
-        if (message.createdAt !== undefined) {
-            obj.createdAt = message.createdAt.toISOString();
-        }
-        return obj;
-    },
-    create(base) {
-        return exports.UserSensitivityClearance.fromPartial(base !== null && base !== void 0 ? base : {});
-    },
-    fromPartial(object) {
-        var _a, _b, _c, _d;
-        const message = createBaseUserSensitivityClearance();
-        message.organizationId = (_a = object.organizationId) !== null && _a !== void 0 ? _a : "";
-        message.userId = (_b = object.userId) !== null && _b !== void 0 ? _b : "";
-        message.sensitivityTagSlug = (_c = object.sensitivityTagSlug) !== null && _c !== void 0 ? _c : "";
-        message.createdAt = (_d = object.createdAt) !== null && _d !== void 0 ? _d : undefined;
-        return message;
-    },
-};
 function createBaseRestrictedGroup() {
     return {
         id: "",
@@ -1797,6 +1510,132 @@ exports.Entity = {
         return message;
     },
 };
+function createBaseThreadFacet() {
+    return { knowledgeType: "", score: 0, facetFields: undefined, rationale: "", isPrimary: false };
+}
+exports.ThreadFacet = {
+    encode(message, writer = new wire_1.BinaryWriter()) {
+        if (message.knowledgeType !== "") {
+            writer.uint32(10).string(message.knowledgeType);
+        }
+        if (message.score !== 0) {
+            writer.uint32(21).float(message.score);
+        }
+        if (message.facetFields !== undefined) {
+            struct_pb_1.Struct.encode(struct_pb_1.Struct.wrap(message.facetFields), writer.uint32(26).fork()).join();
+        }
+        if (message.rationale !== "") {
+            writer.uint32(34).string(message.rationale);
+        }
+        if (message.isPrimary !== false) {
+            writer.uint32(40).bool(message.isPrimary);
+        }
+        return writer;
+    },
+    decode(input, length) {
+        const reader = input instanceof wire_1.BinaryReader ? input : new wire_1.BinaryReader(input);
+        const end = length === undefined ? reader.len : reader.pos + length;
+        const message = createBaseThreadFacet();
+        while (reader.pos < end) {
+            const tag = reader.uint32();
+            switch (tag >>> 3) {
+                case 1: {
+                    if (tag !== 10) {
+                        break;
+                    }
+                    message.knowledgeType = reader.string();
+                    continue;
+                }
+                case 2: {
+                    if (tag !== 21) {
+                        break;
+                    }
+                    message.score = reader.float();
+                    continue;
+                }
+                case 3: {
+                    if (tag !== 26) {
+                        break;
+                    }
+                    message.facetFields = struct_pb_1.Struct.unwrap(struct_pb_1.Struct.decode(reader, reader.uint32()));
+                    continue;
+                }
+                case 4: {
+                    if (tag !== 34) {
+                        break;
+                    }
+                    message.rationale = reader.string();
+                    continue;
+                }
+                case 5: {
+                    if (tag !== 40) {
+                        break;
+                    }
+                    message.isPrimary = reader.bool();
+                    continue;
+                }
+            }
+            if ((tag & 7) === 4 || tag === 0) {
+                break;
+            }
+            reader.skip(tag & 7);
+        }
+        return message;
+    },
+    fromJSON(object) {
+        return {
+            knowledgeType: isSet(object.knowledgeType)
+                ? globalThis.String(object.knowledgeType)
+                : isSet(object.knowledge_type)
+                    ? globalThis.String(object.knowledge_type)
+                    : "",
+            score: isSet(object.score) ? globalThis.Number(object.score) : 0,
+            facetFields: isObject(object.facetFields)
+                ? object.facetFields
+                : isObject(object.facet_fields)
+                    ? object.facet_fields
+                    : undefined,
+            rationale: isSet(object.rationale) ? globalThis.String(object.rationale) : "",
+            isPrimary: isSet(object.isPrimary)
+                ? globalThis.Boolean(object.isPrimary)
+                : isSet(object.is_primary)
+                    ? globalThis.Boolean(object.is_primary)
+                    : false,
+        };
+    },
+    toJSON(message) {
+        const obj = {};
+        if (message.knowledgeType !== "") {
+            obj.knowledgeType = message.knowledgeType;
+        }
+        if (message.score !== 0) {
+            obj.score = message.score;
+        }
+        if (message.facetFields !== undefined) {
+            obj.facetFields = message.facetFields;
+        }
+        if (message.rationale !== "") {
+            obj.rationale = message.rationale;
+        }
+        if (message.isPrimary !== false) {
+            obj.isPrimary = message.isPrimary;
+        }
+        return obj;
+    },
+    create(base) {
+        return exports.ThreadFacet.fromPartial(base !== null && base !== void 0 ? base : {});
+    },
+    fromPartial(object) {
+        var _a, _b, _c, _d, _e;
+        const message = createBaseThreadFacet();
+        message.knowledgeType = (_a = object.knowledgeType) !== null && _a !== void 0 ? _a : "";
+        message.score = (_b = object.score) !== null && _b !== void 0 ? _b : 0;
+        message.facetFields = (_c = object.facetFields) !== null && _c !== void 0 ? _c : undefined;
+        message.rationale = (_d = object.rationale) !== null && _d !== void 0 ? _d : "";
+        message.isPrimary = (_e = object.isPrimary) !== null && _e !== void 0 ? _e : false;
+        return message;
+    },
+};
 function createBaseDocument() {
     return {
         id: "",
@@ -1813,7 +1652,6 @@ function createBaseDocument() {
         charCount: 0,
         tags: [],
         metadata: undefined,
-        sensitivityTags: [],
         restrictedGroups: [],
     };
 }
@@ -1861,9 +1699,6 @@ exports.Document = {
         if (message.metadata !== undefined) {
             struct_pb_1.Struct.encode(struct_pb_1.Struct.wrap(message.metadata), writer.uint32(122).fork()).join();
         }
-        for (const v of message.sensitivityTags) {
-            writer.uint32(130).string(v);
-        }
         for (const v of message.restrictedGroups) {
             exports.RestrictedGroup.encode(v, writer.uint32(138).fork()).join();
         }
@@ -1974,13 +1809,6 @@ exports.Document = {
                     message.metadata = struct_pb_1.Struct.unwrap(struct_pb_1.Struct.decode(reader, reader.uint32()));
                     continue;
                 }
-                case 16: {
-                    if (tag !== 130) {
-                        break;
-                    }
-                    message.sensitivityTags.push(reader.string());
-                    continue;
-                }
                 case 17: {
                     if (tag !== 138) {
                         break;
@@ -2046,11 +1874,6 @@ exports.Document = {
                 ? object.tags.map((e) => exports.DocumentTag.fromJSON(e))
                 : [],
             metadata: isObject(object.metadata) ? object.metadata : undefined,
-            sensitivityTags: globalThis.Array.isArray(object === null || object === void 0 ? void 0 : object.sensitivityTags)
-                ? object.sensitivityTags.map((e) => globalThis.String(e))
-                : globalThis.Array.isArray(object === null || object === void 0 ? void 0 : object.sensitivity_tags)
-                    ? object.sensitivity_tags.map((e) => globalThis.String(e))
-                    : [],
             restrictedGroups: globalThis.Array.isArray(object === null || object === void 0 ? void 0 : object.restrictedGroups)
                 ? object.restrictedGroups.map((e) => exports.RestrictedGroup.fromJSON(e))
                 : globalThis.Array.isArray(object === null || object === void 0 ? void 0 : object.restricted_groups)
@@ -2059,7 +1882,7 @@ exports.Document = {
         };
     },
     toJSON(message) {
-        var _a, _b, _c;
+        var _a, _b;
         const obj = {};
         if (message.id !== "") {
             obj.id = message.id;
@@ -2103,10 +1926,7 @@ exports.Document = {
         if (message.metadata !== undefined) {
             obj.metadata = message.metadata;
         }
-        if ((_b = message.sensitivityTags) === null || _b === void 0 ? void 0 : _b.length) {
-            obj.sensitivityTags = message.sensitivityTags;
-        }
-        if ((_c = message.restrictedGroups) === null || _c === void 0 ? void 0 : _c.length) {
+        if ((_b = message.restrictedGroups) === null || _b === void 0 ? void 0 : _b.length) {
             obj.restrictedGroups = message.restrictedGroups.map((e) => exports.RestrictedGroup.toJSON(e));
         }
         return obj;
@@ -2115,25 +1935,24 @@ exports.Document = {
         return exports.Document.fromPartial(base !== null && base !== void 0 ? base : {});
     },
     fromPartial(object) {
-        var _a, _b, _c;
-        var _d, _e, _f, _g, _h, _j, _k, _l, _m, _o, _p, _q, _r;
+        var _a, _b;
+        var _c, _d, _e, _f, _g, _h, _j, _k, _l, _m, _o, _p, _q;
         const message = createBaseDocument();
-        message.id = (_d = object.id) !== null && _d !== void 0 ? _d : "";
-        message.organizationId = (_e = object.organizationId) !== null && _e !== void 0 ? _e : "";
-        message.filename = (_f = object.filename) !== null && _f !== void 0 ? _f : "";
-        message.mimeType = (_g = object.mimeType) !== null && _g !== void 0 ? _g : "";
-        message.parserName = (_h = object.parserName) !== null && _h !== void 0 ? _h : "";
-        message.status = (_j = object.status) !== null && _j !== void 0 ? _j : 0;
-        message.uploadedByUserId = (_k = object.uploadedByUserId) !== null && _k !== void 0 ? _k : "";
-        message.uploadedAt = (_l = object.uploadedAt) !== null && _l !== void 0 ? _l : undefined;
-        message.updatedAt = (_m = object.updatedAt) !== null && _m !== void 0 ? _m : undefined;
-        message.scope = (_o = object.scope) !== null && _o !== void 0 ? _o : 0;
-        message.pageCount = (_p = object.pageCount) !== null && _p !== void 0 ? _p : 0;
-        message.charCount = (_q = object.charCount) !== null && _q !== void 0 ? _q : 0;
+        message.id = (_c = object.id) !== null && _c !== void 0 ? _c : "";
+        message.organizationId = (_d = object.organizationId) !== null && _d !== void 0 ? _d : "";
+        message.filename = (_e = object.filename) !== null && _e !== void 0 ? _e : "";
+        message.mimeType = (_f = object.mimeType) !== null && _f !== void 0 ? _f : "";
+        message.parserName = (_g = object.parserName) !== null && _g !== void 0 ? _g : "";
+        message.status = (_h = object.status) !== null && _h !== void 0 ? _h : 0;
+        message.uploadedByUserId = (_j = object.uploadedByUserId) !== null && _j !== void 0 ? _j : "";
+        message.uploadedAt = (_k = object.uploadedAt) !== null && _k !== void 0 ? _k : undefined;
+        message.updatedAt = (_l = object.updatedAt) !== null && _l !== void 0 ? _l : undefined;
+        message.scope = (_m = object.scope) !== null && _m !== void 0 ? _m : 0;
+        message.pageCount = (_o = object.pageCount) !== null && _o !== void 0 ? _o : 0;
+        message.charCount = (_p = object.charCount) !== null && _p !== void 0 ? _p : 0;
         message.tags = ((_a = object.tags) === null || _a === void 0 ? void 0 : _a.map((e) => exports.DocumentTag.fromPartial(e))) || [];
-        message.metadata = (_r = object.metadata) !== null && _r !== void 0 ? _r : undefined;
-        message.sensitivityTags = ((_b = object.sensitivityTags) === null || _b === void 0 ? void 0 : _b.map((e) => e)) || [];
-        message.restrictedGroups = ((_c = object.restrictedGroups) === null || _c === void 0 ? void 0 : _c.map((e) => exports.RestrictedGroup.fromPartial(e))) || [];
+        message.metadata = (_q = object.metadata) !== null && _q !== void 0 ? _q : undefined;
+        message.restrictedGroups = ((_b = object.restrictedGroups) === null || _b === void 0 ? void 0 : _b.map((e) => exports.RestrictedGroup.fromPartial(e))) || [];
         return message;
     },
 };
@@ -2246,7 +2065,6 @@ function createBaseDocumentSummary() {
         status: 0,
         scope: 0,
         tags: [],
-        sensitivityTags: [],
         uploadedByUserId: "",
         ownerDisplayName: "",
         uploadedAt: undefined,
@@ -2286,9 +2104,6 @@ exports.DocumentSummary = {
         for (const v of message.tags) {
             exports.DocumentTag.encode(v, writer.uint32(66).fork()).join();
         }
-        for (const v of message.sensitivityTags) {
-            writer.uint32(74).string(v);
-        }
         if (message.uploadedByUserId !== "") {
             writer.uint32(82).string(message.uploadedByUserId);
         }
@@ -2384,14 +2199,7 @@ exports.DocumentSummary = {
                     if (tag !== 66) {
                         break;
                     }
-                    message.tags.push(exports.DocumentTag.decode(reader, reader.uint32()));
-                    continue;
-                }
-                case 9: {
-                    if (tag !== 74) {
-                        break;
-                    }
-                    message.sensitivityTags.push(reader.string());
+                    message.tags.push(exports.DocumentTag.decode(reader, reader.uint32()));
                     continue;
                 }
                 case 10: {
@@ -2503,11 +2311,6 @@ exports.DocumentSummary = {
             tags: globalThis.Array.isArray(object === null || object === void 0 ? void 0 : object.tags)
                 ? object.tags.map((e) => exports.DocumentTag.fromJSON(e))
                 : [],
-            sensitivityTags: globalThis.Array.isArray(object === null || object === void 0 ? void 0 : object.sensitivityTags)
-                ? object.sensitivityTags.map((e) => globalThis.String(e))
-                : globalThis.Array.isArray(object === null || object === void 0 ? void 0 : object.sensitivity_tags)
-                    ? object.sensitivity_tags.map((e) => globalThis.String(e))
-                    : [],
             uploadedByUserId: isSet(object.uploadedByUserId)
                 ? globalThis.String(object.uploadedByUserId)
                 : isSet(object.uploaded_by_user_id)
@@ -2562,7 +2365,7 @@ exports.DocumentSummary = {
         };
     },
     toJSON(message) {
-        var _a, _b, _c;
+        var _a, _b;
         const obj = {};
         if (message.id !== "") {
             obj.id = message.id;
@@ -2588,9 +2391,6 @@ exports.DocumentSummary = {
         if ((_a = message.tags) === null || _a === void 0 ? void 0 : _a.length) {
             obj.tags = message.tags.map((e) => exports.DocumentTag.toJSON(e));
         }
-        if ((_b = message.sensitivityTags) === null || _b === void 0 ? void 0 : _b.length) {
-            obj.sensitivityTags = message.sensitivityTags;
-        }
         if (message.uploadedByUserId !== "") {
             obj.uploadedByUserId = message.uploadedByUserId;
         }
@@ -2621,7 +2421,7 @@ exports.DocumentSummary = {
         if (message.metadata !== undefined) {
             obj.metadata = message.metadata;
         }
-        if ((_c = message.restrictedGroups) === null || _c === void 0 ? void 0 : _c.length) {
+        if ((_b = message.restrictedGroups) === null || _b === void 0 ? void 0 : _b.length) {
             obj.restrictedGroups = message.restrictedGroups.map((e) => exports.RestrictedGroup.toJSON(e));
         }
         return obj;
@@ -2630,31 +2430,30 @@ exports.DocumentSummary = {
         return exports.DocumentSummary.fromPartial(base !== null && base !== void 0 ? base : {});
     },
     fromPartial(object) {
-        var _a, _b, _c;
-        var _d, _e, _f, _g, _h, _j, _k, _l, _m, _o, _p, _q, _r, _s, _t, _u;
+        var _a, _b;
+        var _c, _d, _e, _f, _g, _h, _j, _k, _l, _m, _o, _p, _q, _r, _s, _t;
         const message = createBaseDocumentSummary();
-        message.id = (_d = object.id) !== null && _d !== void 0 ? _d : "";
-        message.organizationId = (_e = object.organizationId) !== null && _e !== void 0 ? _e : "";
-        message.filename = (_f = object.filename) !== null && _f !== void 0 ? _f : "";
-        message.mimeType = (_g = object.mimeType) !== null && _g !== void 0 ? _g : "";
-        message.fileType = (_h = object.fileType) !== null && _h !== void 0 ? _h : "";
-        message.status = (_j = object.status) !== null && _j !== void 0 ? _j : 0;
-        message.scope = (_k = object.scope) !== null && _k !== void 0 ? _k : 0;
+        message.id = (_c = object.id) !== null && _c !== void 0 ? _c : "";
+        message.organizationId = (_d = object.organizationId) !== null && _d !== void 0 ? _d : "";
+        message.filename = (_e = object.filename) !== null && _e !== void 0 ? _e : "";
+        message.mimeType = (_f = object.mimeType) !== null && _f !== void 0 ? _f : "";
+        message.fileType = (_g = object.fileType) !== null && _g !== void 0 ? _g : "";
+        message.status = (_h = object.status) !== null && _h !== void 0 ? _h : 0;
+        message.scope = (_j = object.scope) !== null && _j !== void 0 ? _j : 0;
         message.tags = ((_a = object.tags) === null || _a === void 0 ? void 0 : _a.map((e) => exports.DocumentTag.fromPartial(e))) || [];
-        message.sensitivityTags = ((_b = object.sensitivityTags) === null || _b === void 0 ? void 0 : _b.map((e) => e)) || [];
-        message.uploadedByUserId = (_l = object.uploadedByUserId) !== null && _l !== void 0 ? _l : "";
-        message.ownerDisplayName = (_m = object.ownerDisplayName) !== null && _m !== void 0 ? _m : "";
-        message.uploadedAt = (_o = object.uploadedAt) !== null && _o !== void 0 ? _o : undefined;
-        message.updatedAt = (_p = object.updatedAt) !== null && _p !== void 0 ? _p : undefined;
-        message.pageCount = (_q = object.pageCount) !== null && _q !== void 0 ? _q : 0;
-        message.charCount = (_r = object.charCount) !== null && _r !== void 0 ? _r : 0;
-        message.latestIngestionRunId = (_s = object.latestIngestionRunId) !== null && _s !== void 0 ? _s : "";
-        message.latestIngestionStatus = (_t = object.latestIngestionStatus) !== null && _t !== void 0 ? _t : 0;
+        message.uploadedByUserId = (_k = object.uploadedByUserId) !== null && _k !== void 0 ? _k : "";
+        message.ownerDisplayName = (_l = object.ownerDisplayName) !== null && _l !== void 0 ? _l : "";
+        message.uploadedAt = (_m = object.uploadedAt) !== null && _m !== void 0 ? _m : undefined;
+        message.updatedAt = (_o = object.updatedAt) !== null && _o !== void 0 ? _o : undefined;
+        message.pageCount = (_p = object.pageCount) !== null && _p !== void 0 ? _p : 0;
+        message.charCount = (_q = object.charCount) !== null && _q !== void 0 ? _q : 0;
+        message.latestIngestionRunId = (_r = object.latestIngestionRunId) !== null && _r !== void 0 ? _r : "";
+        message.latestIngestionStatus = (_s = object.latestIngestionStatus) !== null && _s !== void 0 ? _s : 0;
         message.threadCounts = (object.threadCounts !== undefined && object.threadCounts !== null)
             ? exports.DocumentThreadCounts.fromPartial(object.threadCounts)
             : undefined;
-        message.metadata = (_u = object.metadata) !== null && _u !== void 0 ? _u : undefined;
-        message.restrictedGroups = ((_c = object.restrictedGroups) === null || _c === void 0 ? void 0 : _c.map((e) => exports.RestrictedGroup.fromPartial(e))) || [];
+        message.metadata = (_t = object.metadata) !== null && _t !== void 0 ? _t : undefined;
+        message.restrictedGroups = ((_b = object.restrictedGroups) === null || _b === void 0 ? void 0 : _b.map((e) => exports.RestrictedGroup.fromPartial(e))) || [];
         return message;
     },
 };
@@ -3113,6 +2912,7 @@ function createBaseThread() {
         edited: false,
         metadata: undefined,
         entities: [],
+        facets: [],
     };
 }
 exports.Thread = {
@@ -3186,6 +2986,9 @@ exports.Thread = {
         for (const v of message.entities) {
             exports.Entity.encode(v, writer.uint32(186).fork()).join();
         }
+        for (const v of message.facets) {
+            exports.ThreadFacet.encode(v, writer.uint32(194).fork()).join();
+        }
         return writer;
     },
     decode(input, length) {
@@ -3356,6 +3159,13 @@ exports.Thread = {
                     message.entities.push(exports.Entity.decode(reader, reader.uint32()));
                     continue;
                 }
+                case 24: {
+                    if (tag !== 194) {
+                        break;
+                    }
+                    message.facets.push(exports.ThreadFacet.decode(reader, reader.uint32()));
+                    continue;
+                }
             }
             if ((tag & 7) === 4 || tag === 0) {
                 break;
@@ -3455,10 +3265,13 @@ exports.Thread = {
             entities: globalThis.Array.isArray(object === null || object === void 0 ? void 0 : object.entities)
                 ? object.entities.map((e) => exports.Entity.fromJSON(e))
                 : [],
+            facets: globalThis.Array.isArray(object === null || object === void 0 ? void 0 : object.facets)
+                ? object.facets.map((e) => exports.ThreadFacet.fromJSON(e))
+                : [],
         };
     },
     toJSON(message) {
-        var _a, _b;
+        var _a, _b, _c;
         const obj = {};
         if (message.id !== "") {
             obj.id = message.id;
@@ -3529,42 +3342,46 @@ exports.Thread = {
         if ((_b = message.entities) === null || _b === void 0 ? void 0 : _b.length) {
             obj.entities = message.entities.map((e) => exports.Entity.toJSON(e));
         }
+        if ((_c = message.facets) === null || _c === void 0 ? void 0 : _c.length) {
+            obj.facets = message.facets.map((e) => exports.ThreadFacet.toJSON(e));
+        }
         return obj;
     },
     create(base) {
         return exports.Thread.fromPartial(base !== null && base !== void 0 ? base : {});
     },
     fromPartial(object) {
-        var _a, _b;
-        var _c, _d, _e, _f, _g, _h, _j, _k, _l, _m, _o, _p, _q, _r, _s, _t, _u, _v, _w;
+        var _a, _b, _c;
+        var _d, _e, _f, _g, _h, _j, _k, _l, _m, _o, _p, _q, _r, _s, _t, _u, _v, _w, _x;
         const message = createBaseThread();
-        message.id = (_c = object.id) !== null && _c !== void 0 ? _c : "";
-        message.organizationId = (_d = object.organizationId) !== null && _d !== void 0 ? _d : "";
-        message.documentId = (_e = object.documentId) !== null && _e !== void 0 ? _e : "";
-        message.documentName = (_f = object.documentName) !== null && _f !== void 0 ? _f : "";
-        message.ingestionRunId = (_g = object.ingestionRunId) !== null && _g !== void 0 ? _g : "";
-        message.text = (_h = object.text) !== null && _h !== void 0 ? _h : "";
-        message.textOriginal = (_j = object.textOriginal) !== null && _j !== void 0 ? _j : "";
-        message.knowledgeType = (_k = object.knowledgeType) !== null && _k !== void 0 ? _k : "";
-        message.facetFields = (_l = object.facetFields) !== null && _l !== void 0 ? _l : undefined;
-        message.confidence = (_m = object.confidence) !== null && _m !== void 0 ? _m : 0;
+        message.id = (_d = object.id) !== null && _d !== void 0 ? _d : "";
+        message.organizationId = (_e = object.organizationId) !== null && _e !== void 0 ? _e : "";
+        message.documentId = (_f = object.documentId) !== null && _f !== void 0 ? _f : "";
+        message.documentName = (_g = object.documentName) !== null && _g !== void 0 ? _g : "";
+        message.ingestionRunId = (_h = object.ingestionRunId) !== null && _h !== void 0 ? _h : "";
+        message.text = (_j = object.text) !== null && _j !== void 0 ? _j : "";
+        message.textOriginal = (_k = object.textOriginal) !== null && _k !== void 0 ? _k : "";
+        message.knowledgeType = (_l = object.knowledgeType) !== null && _l !== void 0 ? _l : "";
+        message.facetFields = (_m = object.facetFields) !== null && _m !== void 0 ? _m : undefined;
+        message.confidence = (_o = object.confidence) !== null && _o !== void 0 ? _o : 0;
         message.temporalValidity = (object.temporalValidity !== undefined && object.temporalValidity !== null)
             ? exports.TemporalValidity.fromPartial(object.temporalValidity)
             : undefined;
-        message.classification = (_o = object.classification) !== null && _o !== void 0 ? _o : 0;
-        message.retrievalStatus = (_p = object.retrievalStatus) !== null && _p !== void 0 ? _p : 0;
+        message.classification = (_p = object.classification) !== null && _p !== void 0 ? _p : 0;
+        message.retrievalStatus = (_q = object.retrievalStatus) !== null && _q !== void 0 ? _q : 0;
         message.sourceLocation = (object.sourceLocation !== undefined && object.sourceLocation !== null)
             ? exports.SourceLocation.fromPartial(object.sourceLocation)
             : undefined;
         message.supersedesThreadIds = ((_a = object.supersedesThreadIds) === null || _a === void 0 ? void 0 : _a.map((e) => e)) || [];
-        message.supersededByThreadId = (_q = object.supersededByThreadId) !== null && _q !== void 0 ? _q : "";
-        message.clusterId = (_r = object.clusterId) !== null && _r !== void 0 ? _r : "";
-        message.extractedAt = (_s = object.extractedAt) !== null && _s !== void 0 ? _s : undefined;
-        message.approvedAt = (_t = object.approvedAt) !== null && _t !== void 0 ? _t : undefined;
-        message.approvedByUserId = (_u = object.approvedByUserId) !== null && _u !== void 0 ? _u : "";
-        message.edited = (_v = object.edited) !== null && _v !== void 0 ? _v : false;
-        message.metadata = (_w = object.metadata) !== null && _w !== void 0 ? _w : undefined;
+        message.supersededByThreadId = (_r = object.supersededByThreadId) !== null && _r !== void 0 ? _r : "";
+        message.clusterId = (_s = object.clusterId) !== null && _s !== void 0 ? _s : "";
+        message.extractedAt = (_t = object.extractedAt) !== null && _t !== void 0 ? _t : undefined;
+        message.approvedAt = (_u = object.approvedAt) !== null && _u !== void 0 ? _u : undefined;
+        message.approvedByUserId = (_v = object.approvedByUserId) !== null && _v !== void 0 ? _v : "";
+        message.edited = (_w = object.edited) !== null && _w !== void 0 ? _w : false;
+        message.metadata = (_x = object.metadata) !== null && _x !== void 0 ? _x : undefined;
         message.entities = ((_b = object.entities) === null || _b === void 0 ? void 0 : _b.map((e) => exports.Entity.fromPartial(e))) || [];
+        message.facets = ((_c = object.facets) === null || _c === void 0 ? void 0 : _c.map((e) => exports.ThreadFacet.fromPartial(e))) || [];
         return message;
     },
 };
@@ -3585,6 +3402,8 @@ function createBaseIngestionRun() {
         completedAt: undefined,
         committedAt: undefined,
         threads: [],
+        stageModelSettings: undefined,
+        refineLoopCount: 0,
     };
 }
 exports.IngestionRun = {
@@ -3634,6 +3453,12 @@ exports.IngestionRun = {
         for (const v of message.threads) {
             exports.Thread.encode(v, writer.uint32(122).fork()).join();
         }
+        if (message.stageModelSettings !== undefined) {
+            struct_pb_1.Struct.encode(struct_pb_1.Struct.wrap(message.stageModelSettings), writer.uint32(130).fork()).join();
+        }
+        if (message.refineLoopCount !== 0) {
+            writer.uint32(136).int32(message.refineLoopCount);
+        }
         return writer;
     },
     decode(input, length) {
@@ -3748,6 +3573,20 @@ exports.IngestionRun = {
                     message.threads.push(exports.Thread.decode(reader, reader.uint32()));
                     continue;
                 }
+                case 16: {
+                    if (tag !== 130) {
+                        break;
+                    }
+                    message.stageModelSettings = struct_pb_1.Struct.unwrap(struct_pb_1.Struct.decode(reader, reader.uint32()));
+                    continue;
+                }
+                case 17: {
+                    if (tag !== 136) {
+                        break;
+                    }
+                    message.refineLoopCount = reader.int32();
+                    continue;
+                }
             }
             if ((tag & 7) === 4 || tag === 0) {
                 break;
@@ -3819,6 +3658,16 @@ exports.IngestionRun = {
             threads: globalThis.Array.isArray(object === null || object === void 0 ? void 0 : object.threads)
                 ? object.threads.map((e) => exports.Thread.fromJSON(e))
                 : [],
+            stageModelSettings: isObject(object.stageModelSettings)
+                ? object.stageModelSettings
+                : isObject(object.stage_model_settings)
+                    ? object.stage_model_settings
+                    : undefined,
+            refineLoopCount: isSet(object.refineLoopCount)
+                ? globalThis.Number(object.refineLoopCount)
+                : isSet(object.refine_loop_count)
+                    ? globalThis.Number(object.refine_loop_count)
+                    : 0,
         };
     },
     toJSON(message) {
@@ -3869,6 +3718,12 @@ exports.IngestionRun = {
         if ((_a = message.threads) === null || _a === void 0 ? void 0 : _a.length) {
             obj.threads = message.threads.map((e) => exports.Thread.toJSON(e));
         }
+        if (message.stageModelSettings !== undefined) {
+            obj.stageModelSettings = message.stageModelSettings;
+        }
+        if (message.refineLoopCount !== 0) {
+            obj.refineLoopCount = Math.round(message.refineLoopCount);
+        }
         return obj;
     },
     create(base) {
@@ -3876,7 +3731,7 @@ exports.IngestionRun = {
     },
     fromPartial(object) {
         var _a;
-        var _b, _c, _d, _e, _f, _g, _h, _j, _k, _l, _m, _o, _p, _q;
+        var _b, _c, _d, _e, _f, _g, _h, _j, _k, _l, _m, _o, _p, _q, _r, _s;
         const message = createBaseIngestionRun();
         message.id = (_b = object.id) !== null && _b !== void 0 ? _b : "";
         message.organizationId = (_c = object.organizationId) !== null && _c !== void 0 ? _c : "";
@@ -3893,6 +3748,8 @@ exports.IngestionRun = {
         message.completedAt = (_p = object.completedAt) !== null && _p !== void 0 ? _p : undefined;
         message.committedAt = (_q = object.committedAt) !== null && _q !== void 0 ? _q : undefined;
         message.threads = ((_a = object.threads) === null || _a === void 0 ? void 0 : _a.map((e) => exports.Thread.fromPartial(e))) || [];
+        message.stageModelSettings = (_r = object.stageModelSettings) !== null && _r !== void 0 ? _r : undefined;
+        message.refineLoopCount = (_s = object.refineLoopCount) !== null && _s !== void 0 ? _s : 0;
         return message;
     },
 };
@@ -8938,6 +8795,9 @@ function createBasePipelineEvent() {
         diffDecision: undefined,
         pipelineComplete: undefined,
         pipelineFailed: undefined,
+        refineStarted: undefined,
+        refineThread: undefined,
+        refineComplete: undefined,
     };
 }
 exports.PipelineEvent = {
@@ -8978,6 +8838,15 @@ exports.PipelineEvent = {
         if (message.pipelineFailed !== undefined) {
             exports.PipelineEvent_PipelineFailed.encode(message.pipelineFailed, writer.uint32(98).fork()).join();
         }
+        if (message.refineStarted !== undefined) {
+            exports.PipelineEvent_RefineStarted.encode(message.refineStarted, writer.uint32(106).fork()).join();
+        }
+        if (message.refineThread !== undefined) {
+            exports.PipelineEvent_RefineThread.encode(message.refineThread, writer.uint32(114).fork()).join();
+        }
+        if (message.refineComplete !== undefined) {
+            exports.PipelineEvent_RefineComplete.encode(message.refineComplete, writer.uint32(122).fork()).join();
+        }
         return writer;
     },
     decode(input, length) {
@@ -9071,6 +8940,27 @@ exports.PipelineEvent = {
                     message.pipelineFailed = exports.PipelineEvent_PipelineFailed.decode(reader, reader.uint32());
                     continue;
                 }
+                case 13: {
+                    if (tag !== 106) {
+                        break;
+                    }
+                    message.refineStarted = exports.PipelineEvent_RefineStarted.decode(reader, reader.uint32());
+                    continue;
+                }
+                case 14: {
+                    if (tag !== 114) {
+                        break;
+                    }
+                    message.refineThread = exports.PipelineEvent_RefineThread.decode(reader, reader.uint32());
+                    continue;
+                }
+                case 15: {
+                    if (tag !== 122) {
+                        break;
+                    }
+                    message.refineComplete = exports.PipelineEvent_RefineComplete.decode(reader, reader.uint32());
+                    continue;
+                }
             }
             if ((tag & 7) === 4 || tag === 0) {
                 break;
@@ -9141,6 +9031,21 @@ exports.PipelineEvent = {
                 : isSet(object.pipeline_failed)
                     ? exports.PipelineEvent_PipelineFailed.fromJSON(object.pipeline_failed)
                     : undefined,
+            refineStarted: isSet(object.refineStarted)
+                ? exports.PipelineEvent_RefineStarted.fromJSON(object.refineStarted)
+                : isSet(object.refine_started)
+                    ? exports.PipelineEvent_RefineStarted.fromJSON(object.refine_started)
+                    : undefined,
+            refineThread: isSet(object.refineThread)
+                ? exports.PipelineEvent_RefineThread.fromJSON(object.refineThread)
+                : isSet(object.refine_thread)
+                    ? exports.PipelineEvent_RefineThread.fromJSON(object.refine_thread)
+                    : undefined,
+            refineComplete: isSet(object.refineComplete)
+                ? exports.PipelineEvent_RefineComplete.fromJSON(object.refineComplete)
+                : isSet(object.refine_complete)
+                    ? exports.PipelineEvent_RefineComplete.fromJSON(object.refine_complete)
+                    : undefined,
         };
     },
     toJSON(message) {
@@ -9181,6 +9086,15 @@ exports.PipelineEvent = {
         if (message.pipelineFailed !== undefined) {
             obj.pipelineFailed = exports.PipelineEvent_PipelineFailed.toJSON(message.pipelineFailed);
         }
+        if (message.refineStarted !== undefined) {
+            obj.refineStarted = exports.PipelineEvent_RefineStarted.toJSON(message.refineStarted);
+        }
+        if (message.refineThread !== undefined) {
+            obj.refineThread = exports.PipelineEvent_RefineThread.toJSON(message.refineThread);
+        }
+        if (message.refineComplete !== undefined) {
+            obj.refineComplete = exports.PipelineEvent_RefineComplete.toJSON(message.refineComplete);
+        }
         return obj;
     },
     create(base) {
@@ -9221,6 +9135,15 @@ exports.PipelineEvent = {
         message.pipelineFailed = (object.pipelineFailed !== undefined && object.pipelineFailed !== null)
             ? exports.PipelineEvent_PipelineFailed.fromPartial(object.pipelineFailed)
             : undefined;
+        message.refineStarted = (object.refineStarted !== undefined && object.refineStarted !== null)
+            ? exports.PipelineEvent_RefineStarted.fromPartial(object.refineStarted)
+            : undefined;
+        message.refineThread = (object.refineThread !== undefined && object.refineThread !== null)
+            ? exports.PipelineEvent_RefineThread.fromPartial(object.refineThread)
+            : undefined;
+        message.refineComplete = (object.refineComplete !== undefined && object.refineComplete !== null)
+            ? exports.PipelineEvent_RefineComplete.fromPartial(object.refineComplete)
+            : undefined;
         return message;
     },
 };
@@ -9439,6 +9362,235 @@ exports.PipelineEvent_ClassifyTag = {
         return message;
     },
 };
+function createBasePipelineEvent_RefineStarted() {
+    return { blockCount: 0, loopCount: 0 };
+}
+exports.PipelineEvent_RefineStarted = {
+    encode(message, writer = new wire_1.BinaryWriter()) {
+        if (message.blockCount !== 0) {
+            writer.uint32(8).int32(message.blockCount);
+        }
+        if (message.loopCount !== 0) {
+            writer.uint32(16).int32(message.loopCount);
+        }
+        return writer;
+    },
+    decode(input, length) {
+        const reader = input instanceof wire_1.BinaryReader ? input : new wire_1.BinaryReader(input);
+        const end = length === undefined ? reader.len : reader.pos + length;
+        const message = createBasePipelineEvent_RefineStarted();
+        while (reader.pos < end) {
+            const tag = reader.uint32();
+            switch (tag >>> 3) {
+                case 1: {
+                    if (tag !== 8) {
+                        break;
+                    }
+                    message.blockCount = reader.int32();
+                    continue;
+                }
+                case 2: {
+                    if (tag !== 16) {
+                        break;
+                    }
+                    message.loopCount = reader.int32();
+                    continue;
+                }
+            }
+            if ((tag & 7) === 4 || tag === 0) {
+                break;
+            }
+            reader.skip(tag & 7);
+        }
+        return message;
+    },
+    fromJSON(object) {
+        return {
+            blockCount: isSet(object.blockCount)
+                ? globalThis.Number(object.blockCount)
+                : isSet(object.block_count)
+                    ? globalThis.Number(object.block_count)
+                    : 0,
+            loopCount: isSet(object.loopCount)
+                ? globalThis.Number(object.loopCount)
+                : isSet(object.loop_count)
+                    ? globalThis.Number(object.loop_count)
+                    : 0,
+        };
+    },
+    toJSON(message) {
+        const obj = {};
+        if (message.blockCount !== 0) {
+            obj.blockCount = Math.round(message.blockCount);
+        }
+        if (message.loopCount !== 0) {
+            obj.loopCount = Math.round(message.loopCount);
+        }
+        return obj;
+    },
+    create(base) {
+        return exports.PipelineEvent_RefineStarted.fromPartial(base !== null && base !== void 0 ? base : {});
+    },
+    fromPartial(object) {
+        var _a, _b;
+        const message = createBasePipelineEvent_RefineStarted();
+        message.blockCount = (_a = object.blockCount) !== null && _a !== void 0 ? _a : 0;
+        message.loopCount = (_b = object.loopCount) !== null && _b !== void 0 ? _b : 0;
+        return message;
+    },
+};
+function createBasePipelineEvent_RefineThread() {
+    return { text: "", sourceLocation: undefined, loopIndex: 0 };
+}
+exports.PipelineEvent_RefineThread = {
+    encode(message, writer = new wire_1.BinaryWriter()) {
+        if (message.text !== "") {
+            writer.uint32(10).string(message.text);
+        }
+        if (message.sourceLocation !== undefined) {
+            exports.SourceLocation.encode(message.sourceLocation, writer.uint32(18).fork()).join();
+        }
+        if (message.loopIndex !== 0) {
+            writer.uint32(24).int32(message.loopIndex);
+        }
+        return writer;
+    },
+    decode(input, length) {
+        const reader = input instanceof wire_1.BinaryReader ? input : new wire_1.BinaryReader(input);
+        const end = length === undefined ? reader.len : reader.pos + length;
+        const message = createBasePipelineEvent_RefineThread();
+        while (reader.pos < end) {
+            const tag = reader.uint32();
+            switch (tag >>> 3) {
+                case 1: {
+                    if (tag !== 10) {
+                        break;
+                    }
+                    message.text = reader.string();
+                    continue;
+                }
+                case 2: {
+                    if (tag !== 18) {
+                        break;
+                    }
+                    message.sourceLocation = exports.SourceLocation.decode(reader, reader.uint32());
+                    continue;
+                }
+                case 3: {
+                    if (tag !== 24) {
+                        break;
+                    }
+                    message.loopIndex = reader.int32();
+                    continue;
+                }
+            }
+            if ((tag & 7) === 4 || tag === 0) {
+                break;
+            }
+            reader.skip(tag & 7);
+        }
+        return message;
+    },
+    fromJSON(object) {
+        return {
+            text: isSet(object.text) ? globalThis.String(object.text) : "",
+            sourceLocation: isSet(object.sourceLocation)
+                ? exports.SourceLocation.fromJSON(object.sourceLocation)
+                : isSet(object.source_location)
+                    ? exports.SourceLocation.fromJSON(object.source_location)
+                    : undefined,
+            loopIndex: isSet(object.loopIndex)
+                ? globalThis.Number(object.loopIndex)
+                : isSet(object.loop_index)
+                    ? globalThis.Number(object.loop_index)
+                    : 0,
+        };
+    },
+    toJSON(message) {
+        const obj = {};
+        if (message.text !== "") {
+            obj.text = message.text;
+        }
+        if (message.sourceLocation !== undefined) {
+            obj.sourceLocation = exports.SourceLocation.toJSON(message.sourceLocation);
+        }
+        if (message.loopIndex !== 0) {
+            obj.loopIndex = Math.round(message.loopIndex);
+        }
+        return obj;
+    },
+    create(base) {
+        return exports.PipelineEvent_RefineThread.fromPartial(base !== null && base !== void 0 ? base : {});
+    },
+    fromPartial(object) {
+        var _a, _b;
+        const message = createBasePipelineEvent_RefineThread();
+        message.text = (_a = object.text) !== null && _a !== void 0 ? _a : "";
+        message.sourceLocation = (object.sourceLocation !== undefined && object.sourceLocation !== null)
+            ? exports.SourceLocation.fromPartial(object.sourceLocation)
+            : undefined;
+        message.loopIndex = (_b = object.loopIndex) !== null && _b !== void 0 ? _b : 0;
+        return message;
+    },
+};
+function createBasePipelineEvent_RefineComplete() {
+    return { threadCount: 0 };
+}
+exports.PipelineEvent_RefineComplete = {
+    encode(message, writer = new wire_1.BinaryWriter()) {
+        if (message.threadCount !== 0) {
+            writer.uint32(8).int32(message.threadCount);
+        }
+        return writer;
+    },
+    decode(input, length) {
+        const reader = input instanceof wire_1.BinaryReader ? input : new wire_1.BinaryReader(input);
+        const end = length === undefined ? reader.len : reader.pos + length;
+        const message = createBasePipelineEvent_RefineComplete();
+        while (reader.pos < end) {
+            const tag = reader.uint32();
+            switch (tag >>> 3) {
+                case 1: {
+                    if (tag !== 8) {
+                        break;
+                    }
+                    message.threadCount = reader.int32();
+                    continue;
+                }
+            }
+            if ((tag & 7) === 4 || tag === 0) {
+                break;
+            }
+            reader.skip(tag & 7);
+        }
+        return message;
+    },
+    fromJSON(object) {
+        return {
+            threadCount: isSet(object.threadCount)
+                ? globalThis.Number(object.threadCount)
+                : isSet(object.thread_count)
+                    ? globalThis.Number(object.thread_count)
+                    : 0,
+        };
+    },
+    toJSON(message) {
+        const obj = {};
+        if (message.threadCount !== 0) {
+            obj.threadCount = Math.round(message.threadCount);
+        }
+        return obj;
+    },
+    create(base) {
+        return exports.PipelineEvent_RefineComplete.fromPartial(base !== null && base !== void 0 ? base : {});
+    },
+    fromPartial(object) {
+        var _a;
+        const message = createBasePipelineEvent_RefineComplete();
+        message.threadCount = (_a = object.threadCount) !== null && _a !== void 0 ? _a : 0;
+        return message;
+    },
+};
 function createBasePipelineEvent_ExtractThread() {
     return { threadId: "", text: "", entities: [], sourceLocation: undefined };
 }