Single-roundtrip liveness check used between batches.
( connectorId: string, knowledgeBaseId: string )
| 148 | |
| 149 | /** Single-roundtrip liveness check used between batches. */ |
| 150 | async function checkSyncLiveness( |
| 151 | connectorId: string, |
| 152 | knowledgeBaseId: string |
| 153 | ): Promise<{ connectorDeleted: boolean; knowledgeBaseDeleted: boolean }> { |
| 154 | const rows = await db |
| 155 | .select({ |
| 156 | connectorArchivedAt: knowledgeConnector.archivedAt, |
| 157 | connectorDeletedAt: knowledgeConnector.deletedAt, |
| 158 | kbDeletedAt: knowledgeBase.deletedAt, |
| 159 | }) |
| 160 | .from(knowledgeConnector) |
| 161 | .innerJoin(knowledgeBase, eq(knowledgeBase.id, knowledgeConnector.knowledgeBaseId)) |
| 162 | .where(and(eq(knowledgeConnector.id, connectorId), eq(knowledgeBase.id, knowledgeBaseId))) |
| 163 | .limit(1) |
| 164 | |
| 165 | if (rows.length === 0) { |
| 166 | return { connectorDeleted: true, knowledgeBaseDeleted: true } |
| 167 | } |
| 168 | const row = rows[0] |
| 169 | return { |
| 170 | connectorDeleted: row.connectorArchivedAt !== null || row.connectorDeletedAt !== null, |
| 171 | knowledgeBaseDeleted: row.kbDeletedAt !== null, |
| 172 | } |
| 173 | } |
| 174 | |
| 175 | async function isKnowledgeBaseActiveInTx( |
| 176 | tx: KnowledgeBaseLockingTx, |