weave-typescript 0.38.0 → 0.40.0

package/dist/weavesql/weavedb/retrieval_sql.d.ts

@@ -28,7 +28,7 @@ export interface SearchApprovedThreadsSemanticRow {
     conceptCluster: string;
 }
 export declare function searchApprovedThreadsSemantic(client: Client, args: SearchApprovedThreadsSemanticArgs): Promise<SearchApprovedThreadsSemanticRow[]>;
-export declare const searchApprovedThreadsSemanticByUserPermissionsQuery = "-- name: SearchApprovedThreadsSemanticByUserPermissions :many\nSELECT\n    t.id AS thread_id,\n    t.text,\n    t.knowledge_type,\n    t.facet_fields,\n    similarity.score,\n    t.classification,\n    t.document_id,\n    d.filename AS document_name,\n    t.source_location,\n    t.approved_at,\n    t.temporal_validity,\n    COALESCE(entity_data.entities, '{}'::text[]) AS entities,\n    COALESCE(rs.reinforcement_count, 0) AS reinforcement_count,\n    COALESCE(cc.name, '') AS concept_cluster\nFROM weave.threads t\nJOIN weave.documents d ON d.id = t.document_id\nCROSS JOIN LATERAL (\n    SELECT CAST(\n        1.0::double precision\n        - (t.embedding <=> $1::vector)::double precision\n        AS double precision\n    ) AS score\n) similarity\nLEFT JOIN weave.thread_reinforcement_scores rs ON rs.thread_id = t.id\nLEFT JOIN weave.concept_clusters cc ON cc.id = t.cluster_id\nLEFT JOIN LATERAL (\n    SELECT array_agg(DISTINCT e.canonical ORDER BY e.canonical) AS entities\n    FROM weave.thread_entities te\n    JOIN weave.entities e ON e.id = te.entity_id\n    WHERE te.thread_id = t.id\n) entity_data ON TRUE\nWHERE t.organization_id = $2\n  AND t.retrieval_status = 'THREAD_RETRIEVAL_STATUS_APPROVED'\n  AND t.embedding IS NOT NULL\n  AND similarity.score >= $3::double precision\n  AND (\n      d.scope = 'DOCUMENT_SCOPE_ORG'\n      OR (d.scope = 'DOCUMENT_SCOPE_PRIVATE'\n          AND d.uploaded_by_user_id = $4)\n      OR (d.scope = 'DOCUMENT_SCOPE_RESTRICTED'\n          AND EXISTS (\n              SELECT 1\n              FROM weave.user_sensitivity_clearances c\n              WHERE c.organization_id = $2\n                AND c.user_id = $4\n                AND cardinality(d.sensitivity_tags) = 1\n                AND c.sensitivity_tag_slug = d.sensitivity_tags[1]\n          ))\n  )\nORDER BY score DESC, t.approved_at DESC\nLIMIT $5";
+export declare const searchApprovedThreadsSemanticByUserPermissionsQuery = "-- name: SearchApprovedThreadsSemanticByUserPermissions :many\nSELECT\n    t.id AS thread_id,\n    t.text,\n    t.knowledge_type,\n    t.facet_fields,\n    similarity.score,\n    t.classification,\n    t.document_id,\n    d.filename AS document_name,\n    t.source_location,\n    t.approved_at,\n    t.temporal_validity,\n    COALESCE(entity_data.entities, '{}'::text[]) AS entities,\n    COALESCE(rs.reinforcement_count, 0) AS reinforcement_count,\n    COALESCE(cc.name, '') AS concept_cluster\nFROM weave.threads t\nJOIN weave.documents d ON d.id = t.document_id\nCROSS JOIN LATERAL (\n    SELECT CAST(\n        1.0::double precision\n        - (t.embedding <=> $1::vector)::double precision\n        AS double precision\n    ) AS score\n) similarity\nLEFT JOIN weave.thread_reinforcement_scores rs ON rs.thread_id = t.id\nLEFT JOIN weave.concept_clusters cc ON cc.id = t.cluster_id\nLEFT JOIN LATERAL (\n    SELECT array_agg(DISTINCT e.canonical ORDER BY e.canonical) AS entities\n    FROM weave.thread_entities te\n    JOIN weave.entities e ON e.id = te.entity_id\n    WHERE te.thread_id = t.id\n) entity_data ON TRUE\nWHERE t.organization_id = $2\n  AND t.retrieval_status = 'THREAD_RETRIEVAL_STATUS_APPROVED'\n  AND t.embedding IS NOT NULL\n  AND similarity.score >= $3::double precision\n  AND (\n      d.scope = 'DOCUMENT_SCOPE_ORG'\n      OR (d.scope = 'DOCUMENT_SCOPE_PRIVATE'\n          AND d.uploaded_by_user_id = $4)\n      OR (d.scope = 'DOCUMENT_SCOPE_RESTRICTED'\n          AND (\n          EXISTS (\n              SELECT 1\n              FROM weave.document_restricted_groups drg\n              JOIN weave.restricted_group_user_memberships rgum\n                ON rgum.organization_id = drg.organization_id\n               AND rgum.restricted_group_id = drg.restricted_group_id\n               AND rgum.user_id = $4\n              WHERE drg.organization_id = d.organization_id\n                AND drg.document_id = d.id\n          )\n          OR EXISTS (\n              SELECT 1\n              FROM weave.document_restricted_groups drg\n              JOIN weave.restricted_group_team_memberships rgtm\n                ON rgtm.organization_id = drg.organization_id\n               AND rgtm.restricted_group_id = drg.restricted_group_id\n              JOIN weave.org_team_memberships otm\n                ON otm.organization_id = rgtm.organization_id\n               AND otm.org_team_id = rgtm.org_team_id\n               AND otm.user_id = $4\n              WHERE drg.organization_id = d.organization_id\n                AND drg.document_id = d.id\n          )))\n  )\nORDER BY score DESC, t.approved_at DESC\nLIMIT $5";
 export interface SearchApprovedThreadsSemanticByUserPermissionsArgs {
     queryEmbedding: string;
     organizationId: string;
@@ -53,7 +53,7 @@ export interface SearchApprovedThreadsSemanticByUserPermissionsRow {
     conceptCluster: string;
 }
 export declare function searchApprovedThreadsSemanticByUserPermissions(client: Client, args: SearchApprovedThreadsSemanticByUserPermissionsArgs): Promise<SearchApprovedThreadsSemanticByUserPermissionsRow[]>;
-export declare const searchAccessibleDocumentsSemanticQuery = "-- name: SearchAccessibleDocumentsSemantic :many\nWITH tag_data AS (\n    SELECT\n        dt.document_id,\n        array_agg(dt.tag ORDER BY dt.tag ASC) AS tags,\n        array_agg(dt.confidence ORDER BY dt.tag ASC) AS tag_confidences\n    FROM weave.document_tags dt\n    JOIN weave.documents d ON d.id = dt.document_id\n    WHERE d.organization_id = $4\n    GROUP BY dt.document_id\n),\nmatches AS (\n    SELECT\n        t.id AS thread_id,\n        t.text,\n        t.classification,\n        t.document_id,\n        d.filename AS document_name,\n        d.mime_type,\n        d.scope,\n        d.sensitivity_tags,\n        d.uploaded_at,\n        COALESCE(tag_data.tags, '{}'::text[]) AS tags,\n        COALESCE(tag_data.tag_confidences, '{}'::real[]) AS tag_confidences,\n        t.source_location,\n        t.approved_at,\n        CAST(\n            1.0::double precision\n            - (t.embedding <=> $5::vector)::double precision\n            AS double precision\n        ) AS score\n    FROM weave.threads t\n    JOIN weave.documents d ON d.id = t.document_id\n    LEFT JOIN tag_data ON tag_data.document_id = d.id\n    WHERE t.organization_id = $4\n      AND t.retrieval_status = 'THREAD_RETRIEVAL_STATUS_APPROVED'\n      AND t.embedding IS NOT NULL\n      AND (\n          d.scope = 'DOCUMENT_SCOPE_ORG'\n          OR (d.scope = 'DOCUMENT_SCOPE_PRIVATE'\n              AND d.uploaded_by_user_id = $6)\n          OR (d.scope = 'DOCUMENT_SCOPE_RESTRICTED'\n              AND EXISTS (\n                  SELECT 1\n                  FROM weave.user_sensitivity_clearances c\n                  WHERE c.organization_id = $4\n                    AND c.user_id = $6\n                    AND cardinality(d.sensitivity_tags) = 1\n                    AND c.sensitivity_tag_slug = d.sensitivity_tags[1]\n              ))\n      )\n      AND (cardinality($7::text[]) = 0 OR tag_data.tags && $7::text[] OR d.sensitivity_tags && $7::text[])\n      AND (cardinality($8::text[]) = 0 OR d.sensitivity_tags && $8::text[])\n      AND (cardinality($9::text[]) = 0 OR d.scope = ANY($9::text[]))\n      AND ($10::timestamptz IS NULL OR d.uploaded_at >= $10::timestamptz)\n      AND ($11::timestamptz IS NULL OR d.uploaded_at <= $11::timestamptz)\n)\nSELECT\n    document_id,\n    document_name,\n    mime_type,\n    scope,\n    sensitivity_tags,\n    tags,\n    tag_confidences,\n    uploaded_at,\n    MAX(score)::double precision AS score,\n    COUNT(*)::int AS matched_thread_count,\n    (array_agg(thread_id ORDER BY score DESC, approved_at DESC))[1] AS best_thread_id,\n    (array_agg(text ORDER BY score DESC, approved_at DESC))[1] AS best_text,\n    (array_agg(classification ORDER BY score DESC, approved_at DESC))[1] AS best_classification,\n    (array_agg(source_location ORDER BY score DESC, approved_at DESC))[1] AS best_source_location\nFROM matches\nWHERE score >= $1::double precision\nGROUP BY\n    document_id,\n    document_name,\n    mime_type,\n    scope,\n    sensitivity_tags,\n    tags,\n    tag_confidences,\n    uploaded_at\nORDER BY score DESC, uploaded_at DESC\nLIMIT $3 OFFSET $2";
+export declare const searchAccessibleDocumentsSemanticQuery = "-- name: SearchAccessibleDocumentsSemantic :many\nWITH tag_data AS (\n    SELECT\n        dt.document_id,\n        array_agg(dt.tag ORDER BY dt.tag ASC) AS tags,\n        array_agg(dt.confidence ORDER BY dt.tag ASC) AS tag_confidences\n    FROM weave.document_tags dt\n    JOIN weave.documents d ON d.id = dt.document_id\n    WHERE d.organization_id = $4\n    GROUP BY dt.document_id\n),\nmatches AS (\n    SELECT\n        t.id AS thread_id,\n        t.text,\n        t.classification,\n        t.document_id,\n        d.filename AS document_name,\n        d.mime_type,\n        d.scope,\n        d.uploaded_at,\n        COALESCE(tag_data.tags, '{}'::text[]) AS tags,\n        COALESCE(tag_data.tag_confidences, '{}'::real[]) AS tag_confidences,\n        t.source_location,\n        t.approved_at,\n        CAST(\n            1.0::double precision\n            - (t.embedding <=> $5::vector)::double precision\n            AS double precision\n        ) AS score\n    FROM weave.threads t\n    JOIN weave.documents d ON d.id = t.document_id\n    LEFT JOIN tag_data ON tag_data.document_id = d.id\n    WHERE t.organization_id = $4\n      AND t.retrieval_status = 'THREAD_RETRIEVAL_STATUS_APPROVED'\n      AND t.embedding IS NOT NULL\n      AND (\n          d.scope = 'DOCUMENT_SCOPE_ORG'\n          OR (d.scope = 'DOCUMENT_SCOPE_PRIVATE'\n              AND d.uploaded_by_user_id = $6)\n          OR (d.scope = 'DOCUMENT_SCOPE_RESTRICTED'\n              AND (\n              EXISTS (\n                  SELECT 1\n                  FROM weave.document_restricted_groups drg\n                  JOIN weave.restricted_group_user_memberships rgum\n                    ON rgum.organization_id = drg.organization_id\n                   AND rgum.restricted_group_id = drg.restricted_group_id\n                   AND rgum.user_id = $6\n                  WHERE drg.organization_id = d.organization_id\n                    AND drg.document_id = d.id\n              )\n              OR EXISTS (\n                  SELECT 1\n                  FROM weave.document_restricted_groups drg\n                  JOIN weave.restricted_group_team_memberships rgtm\n                    ON rgtm.organization_id = drg.organization_id\n                   AND rgtm.restricted_group_id = drg.restricted_group_id\n                  JOIN weave.org_team_memberships otm\n                    ON otm.organization_id = rgtm.organization_id\n                   AND otm.org_team_id = rgtm.org_team_id\n                   AND otm.user_id = $6\n                  WHERE drg.organization_id = d.organization_id\n                    AND drg.document_id = d.id\n              )))\n      )\n      AND (cardinality($7::text[]) = 0 OR tag_data.tags && $7::text[])\n      AND (\n          cardinality($8::uuid[]) = 0\n          OR EXISTS (\n              SELECT 1\n              FROM weave.document_restricted_groups drg\n              WHERE drg.organization_id = d.organization_id\n                AND drg.document_id = d.id\n                AND drg.restricted_group_id = ANY($8::uuid[])\n          )\n      )\n      AND (cardinality($9::text[]) = 0 OR d.scope = ANY($9::text[]))\n      AND ($10::timestamptz IS NULL OR d.uploaded_at >= $10::timestamptz)\n      AND ($11::timestamptz IS NULL OR d.uploaded_at <= $11::timestamptz)\n)\nSELECT\n    document_id,\n    document_name,\n    mime_type,\n    scope,\n    tags,\n    tag_confidences,\n    uploaded_at,\n    MAX(score)::double precision AS score,\n    COUNT(*)::int AS matched_thread_count,\n    (array_agg(thread_id ORDER BY score DESC, approved_at DESC))[1] AS best_thread_id,\n    (array_agg(text ORDER BY score DESC, approved_at DESC))[1] AS best_text,\n    (array_agg(classification ORDER BY score DESC, approved_at DESC))[1] AS best_classification,\n    (array_agg(source_location ORDER BY score DESC, approved_at DESC))[1] AS best_source_location\nFROM matches\nWHERE score >= $1::double precision\nGROUP BY\n    document_id,\n    document_name,\n    mime_type,\n    scope,\n    tags,\n    tag_confidences,\n    uploaded_at\nORDER BY score DESC, uploaded_at DESC\nLIMIT $3 OFFSET $2";
 export interface SearchAccessibleDocumentsSemanticArgs {
     minSimilarity: number;
     pageOffset: string;
@@ -62,7 +62,7 @@ export interface SearchAccessibleDocumentsSemanticArgs {
     queryEmbedding: string;
     actingUserId: string;
     tags: string[];
-    restrictedTags: string[];
+    restrictedGroupIds: string[];
     scopes: string[];
     uploadedAfter: Date | null;
     uploadedBefore: Date | null;
@@ -72,7 +72,6 @@ export interface SearchAccessibleDocumentsSemanticRow {
     documentName: string;
     mimeType: string;
     scope: string;
-    sensitivityTags: string[];
     tags: string;
     tagConfidences: string;
     uploadedAt: Date;
@@ -109,7 +108,7 @@ export interface SearchApprovedThreadsSemanticByKnowledgeTypesRow {
     conceptCluster: string;
 }
 export declare function searchApprovedThreadsSemanticByKnowledgeTypes(client: Client, args: SearchApprovedThreadsSemanticByKnowledgeTypesArgs): Promise<SearchApprovedThreadsSemanticByKnowledgeTypesRow[]>;
-export declare const searchApprovedThreadsSemanticByKnowledgeTypesAndUserPermissionsQuery = "-- name: SearchApprovedThreadsSemanticByKnowledgeTypesAndUserPermissions :many\nSELECT\n    t.id AS thread_id,\n    t.text,\n    t.knowledge_type,\n    t.facet_fields,\n    similarity.score,\n    t.classification,\n    t.document_id,\n    d.filename AS document_name,\n    t.source_location,\n    t.approved_at,\n    t.temporal_validity,\n    COALESCE(entity_data.entities, '{}'::text[]) AS entities,\n    COALESCE(rs.reinforcement_count, 0) AS reinforcement_count,\n    COALESCE(cc.name, '') AS concept_cluster\nFROM weave.threads t\nJOIN weave.documents d ON d.id = t.document_id\nCROSS JOIN LATERAL (\n    SELECT CAST(\n        1.0::double precision\n        - (t.embedding <=> $1::vector)::double precision\n        AS double precision\n    ) AS score\n) similarity\nLEFT JOIN weave.thread_reinforcement_scores rs ON rs.thread_id = t.id\nLEFT JOIN weave.concept_clusters cc ON cc.id = t.cluster_id\nLEFT JOIN LATERAL (\n    SELECT array_agg(DISTINCT e.canonical ORDER BY e.canonical) AS entities\n    FROM weave.thread_entities te\n    JOIN weave.entities e ON e.id = te.entity_id\n    WHERE te.thread_id = t.id\n) entity_data ON TRUE\nWHERE t.organization_id = $2\n  AND t.retrieval_status = 'THREAD_RETRIEVAL_STATUS_APPROVED'\n  AND t.embedding IS NOT NULL\n  AND t.knowledge_type = ANY($3::text[])\n  AND similarity.score >= $4::double precision\n  AND (\n      d.scope = 'DOCUMENT_SCOPE_ORG'\n      OR (d.scope = 'DOCUMENT_SCOPE_PRIVATE'\n          AND d.uploaded_by_user_id = $5)\n      OR (d.scope = 'DOCUMENT_SCOPE_RESTRICTED'\n          AND EXISTS (\n              SELECT 1\n              FROM weave.user_sensitivity_clearances c\n              WHERE c.organization_id = $2\n                AND c.user_id = $5\n                AND cardinality(d.sensitivity_tags) = 1\n                AND c.sensitivity_tag_slug = d.sensitivity_tags[1]\n          ))\n  )\nORDER BY score DESC, t.approved_at DESC\nLIMIT $6";
+export declare const searchApprovedThreadsSemanticByKnowledgeTypesAndUserPermissionsQuery = "-- name: SearchApprovedThreadsSemanticByKnowledgeTypesAndUserPermissions :many\nSELECT\n    t.id AS thread_id,\n    t.text,\n    t.knowledge_type,\n    t.facet_fields,\n    similarity.score,\n    t.classification,\n    t.document_id,\n    d.filename AS document_name,\n    t.source_location,\n    t.approved_at,\n    t.temporal_validity,\n    COALESCE(entity_data.entities, '{}'::text[]) AS entities,\n    COALESCE(rs.reinforcement_count, 0) AS reinforcement_count,\n    COALESCE(cc.name, '') AS concept_cluster\nFROM weave.threads t\nJOIN weave.documents d ON d.id = t.document_id\nCROSS JOIN LATERAL (\n    SELECT CAST(\n        1.0::double precision\n        - (t.embedding <=> $1::vector)::double precision\n        AS double precision\n    ) AS score\n) similarity\nLEFT JOIN weave.thread_reinforcement_scores rs ON rs.thread_id = t.id\nLEFT JOIN weave.concept_clusters cc ON cc.id = t.cluster_id\nLEFT JOIN LATERAL (\n    SELECT array_agg(DISTINCT e.canonical ORDER BY e.canonical) AS entities\n    FROM weave.thread_entities te\n    JOIN weave.entities e ON e.id = te.entity_id\n    WHERE te.thread_id = t.id\n) entity_data ON TRUE\nWHERE t.organization_id = $2\n  AND t.retrieval_status = 'THREAD_RETRIEVAL_STATUS_APPROVED'\n  AND t.embedding IS NOT NULL\n  AND t.knowledge_type = ANY($3::text[])\n  AND similarity.score >= $4::double precision\n  AND (\n      d.scope = 'DOCUMENT_SCOPE_ORG'\n      OR (d.scope = 'DOCUMENT_SCOPE_PRIVATE'\n          AND d.uploaded_by_user_id = $5)\n      OR (d.scope = 'DOCUMENT_SCOPE_RESTRICTED'\n          AND (\n          EXISTS (\n              SELECT 1\n              FROM weave.document_restricted_groups drg\n              JOIN weave.restricted_group_user_memberships rgum\n                ON rgum.organization_id = drg.organization_id\n               AND rgum.restricted_group_id = drg.restricted_group_id\n               AND rgum.user_id = $5\n              WHERE drg.organization_id = d.organization_id\n                AND drg.document_id = d.id\n          )\n          OR EXISTS (\n              SELECT 1\n              FROM weave.document_restricted_groups drg\n              JOIN weave.restricted_group_team_memberships rgtm\n                ON rgtm.organization_id = drg.organization_id\n               AND rgtm.restricted_group_id = drg.restricted_group_id\n              JOIN weave.org_team_memberships otm\n                ON otm.organization_id = rgtm.organization_id\n               AND otm.org_team_id = rgtm.org_team_id\n               AND otm.user_id = $5\n              WHERE drg.organization_id = d.organization_id\n                AND drg.document_id = d.id\n          )))\n  )\nORDER BY score DESC, t.approved_at DESC\nLIMIT $6";
 export interface SearchApprovedThreadsSemanticByKnowledgeTypesAndUserPermissionsArgs {
     queryEmbedding: string;
     organizationId: string;
@@ -158,7 +157,7 @@ export interface SearchApprovedThreadsByEntitiesRow {
     conceptCluster: string;
 }
 export declare function searchApprovedThreadsByEntities(client: Client, args: SearchApprovedThreadsByEntitiesArgs): Promise<SearchApprovedThreadsByEntitiesRow[]>;
-export declare const searchApprovedThreadsByEntitiesAndUserPermissionsQuery = "-- name: SearchApprovedThreadsByEntitiesAndUserPermissions :many\nSELECT DISTINCT\n    t.id AS thread_id,\n    t.text,\n    t.knowledge_type,\n    t.facet_fields,\n    te.confidence::double precision AS score,\n    t.classification,\n    t.document_id,\n    d.filename AS document_name,\n    t.source_location,\n    t.approved_at,\n    t.temporal_validity,\n    COALESCE(entity_data.entities, '{}'::text[]) AS entities,\n    COALESCE(rs.reinforcement_count, 0) AS reinforcement_count,\n    COALESCE(cc.name, '') AS concept_cluster\nFROM weave.threads t\nJOIN weave.documents d ON d.id = t.document_id\nJOIN weave.thread_entities te ON te.thread_id = t.id\nJOIN weave.entities e ON e.id = te.entity_id\nLEFT JOIN weave.thread_reinforcement_scores rs ON rs.thread_id = t.id\nLEFT JOIN weave.concept_clusters cc ON cc.id = t.cluster_id\nLEFT JOIN LATERAL (\n    SELECT array_agg(DISTINCT e2.canonical ORDER BY e2.canonical) AS entities\n    FROM weave.thread_entities te2\n    JOIN weave.entities e2 ON e2.id = te2.entity_id\n    WHERE te2.thread_id = t.id\n) entity_data ON TRUE\nWHERE t.organization_id = $1\n  AND t.retrieval_status = 'THREAD_RETRIEVAL_STATUS_APPROVED'\n  AND e.canonical = ANY($2::text[])\n  AND (\n      d.scope = 'DOCUMENT_SCOPE_ORG'\n      OR (d.scope = 'DOCUMENT_SCOPE_PRIVATE'\n          AND d.uploaded_by_user_id = $3)\n      OR (d.scope = 'DOCUMENT_SCOPE_RESTRICTED'\n          AND EXISTS (\n              SELECT 1\n              FROM weave.user_sensitivity_clearances c\n              WHERE c.organization_id = $1\n                AND c.user_id = $3\n                AND cardinality(d.sensitivity_tags) = 1\n                AND c.sensitivity_tag_slug = d.sensitivity_tags[1]\n          ))\n  )\nORDER BY score DESC, t.approved_at DESC\nLIMIT $4";
+export declare const searchApprovedThreadsByEntitiesAndUserPermissionsQuery = "-- name: SearchApprovedThreadsByEntitiesAndUserPermissions :many\nSELECT DISTINCT\n    t.id AS thread_id,\n    t.text,\n    t.knowledge_type,\n    t.facet_fields,\n    te.confidence::double precision AS score,\n    t.classification,\n    t.document_id,\n    d.filename AS document_name,\n    t.source_location,\n    t.approved_at,\n    t.temporal_validity,\n    COALESCE(entity_data.entities, '{}'::text[]) AS entities,\n    COALESCE(rs.reinforcement_count, 0) AS reinforcement_count,\n    COALESCE(cc.name, '') AS concept_cluster\nFROM weave.threads t\nJOIN weave.documents d ON d.id = t.document_id\nJOIN weave.thread_entities te ON te.thread_id = t.id\nJOIN weave.entities e ON e.id = te.entity_id\nLEFT JOIN weave.thread_reinforcement_scores rs ON rs.thread_id = t.id\nLEFT JOIN weave.concept_clusters cc ON cc.id = t.cluster_id\nLEFT JOIN LATERAL (\n    SELECT array_agg(DISTINCT e2.canonical ORDER BY e2.canonical) AS entities\n    FROM weave.thread_entities te2\n    JOIN weave.entities e2 ON e2.id = te2.entity_id\n    WHERE te2.thread_id = t.id\n) entity_data ON TRUE\nWHERE t.organization_id = $1\n  AND t.retrieval_status = 'THREAD_RETRIEVAL_STATUS_APPROVED'\n  AND e.canonical = ANY($2::text[])\n  AND (\n      d.scope = 'DOCUMENT_SCOPE_ORG'\n      OR (d.scope = 'DOCUMENT_SCOPE_PRIVATE'\n          AND d.uploaded_by_user_id = $3)\n      OR (d.scope = 'DOCUMENT_SCOPE_RESTRICTED'\n          AND (\n          EXISTS (\n              SELECT 1\n              FROM weave.document_restricted_groups drg\n              JOIN weave.restricted_group_user_memberships rgum\n                ON rgum.organization_id = drg.organization_id\n               AND rgum.restricted_group_id = drg.restricted_group_id\n               AND rgum.user_id = $3\n              WHERE drg.organization_id = d.organization_id\n                AND drg.document_id = d.id\n          )\n          OR EXISTS (\n              SELECT 1\n              FROM weave.document_restricted_groups drg\n              JOIN weave.restricted_group_team_memberships rgtm\n                ON rgtm.organization_id = drg.organization_id\n               AND rgtm.restricted_group_id = drg.restricted_group_id\n              JOIN weave.org_team_memberships otm\n                ON otm.organization_id = rgtm.organization_id\n               AND otm.org_team_id = rgtm.org_team_id\n               AND otm.user_id = $3\n              WHERE drg.organization_id = d.organization_id\n                AND drg.document_id = d.id\n          )))\n  )\nORDER BY score DESC, t.approved_at DESC\nLIMIT $4";
 export interface SearchApprovedThreadsByEntitiesAndUserPermissionsArgs {
     organizationId: string;
     entityCanonicals: string[];
@@ -234,7 +233,7 @@ export interface SearchConceptClusterThreadsBySimilarityRow {
     conceptCluster: string;
 }
 export declare function searchConceptClusterThreadsBySimilarity(client: Client, args: SearchConceptClusterThreadsBySimilarityArgs): Promise<SearchConceptClusterThreadsBySimilarityRow[]>;
-export declare const searchConceptClusterThreadsBySimilarityAndUserPermissionsQuery = "-- name: SearchConceptClusterThreadsBySimilarityAndUserPermissions :many\nSELECT\n    t.id AS thread_id,\n    t.text,\n    t.knowledge_type,\n    t.facet_fields,\n    similarity.score,\n    t.classification,\n    t.document_id,\n    d.filename AS document_name,\n    t.source_location,\n    t.approved_at,\n    t.temporal_validity,\n    COALESCE(entity_data.entities, '{}'::text[]) AS entities,\n    COALESCE(rs.reinforcement_count, 0) AS reinforcement_count,\n    cc.name AS concept_cluster\nFROM weave.concept_clusters cc\nCROSS JOIN LATERAL (\n    SELECT CAST(\n        1.0::double precision\n        - (cc.embedding <=> $1::vector)::double precision\n        AS double precision\n    ) AS score\n) similarity\nJOIN weave.threads t ON t.cluster_id = cc.id\nJOIN weave.documents d ON d.id = t.document_id\nLEFT JOIN weave.thread_reinforcement_scores rs ON rs.thread_id = t.id\nLEFT JOIN LATERAL (\n    SELECT array_agg(DISTINCT e.canonical ORDER BY e.canonical) AS entities\n    FROM weave.thread_entities te\n    JOIN weave.entities e ON e.id = te.entity_id\n    WHERE te.thread_id = t.id\n) entity_data ON TRUE\nWHERE cc.organization_id = $2\n  AND cc.realm_kind = $3\n  AND cc.realm_key = $4\n  AND cc.level = 0\n  AND cc.embedding IS NOT NULL\n  AND t.retrieval_status = 'THREAD_RETRIEVAL_STATUS_APPROVED'\n  AND similarity.score >= $5::double precision\n  AND (\n      d.scope = 'DOCUMENT_SCOPE_ORG'\n      OR (d.scope = 'DOCUMENT_SCOPE_PRIVATE'\n          AND d.uploaded_by_user_id = $6)\n      OR (d.scope = 'DOCUMENT_SCOPE_RESTRICTED'\n          AND EXISTS (\n              SELECT 1\n              FROM weave.user_sensitivity_clearances c\n              WHERE c.organization_id = $2\n                AND c.user_id = $6\n                AND cardinality(d.sensitivity_tags) = 1\n                AND c.sensitivity_tag_slug = d.sensitivity_tags[1]\n          ))\n  )\nORDER BY score DESC, t.approved_at DESC\nLIMIT $7";
+export declare const searchConceptClusterThreadsBySimilarityAndUserPermissionsQuery = "-- name: SearchConceptClusterThreadsBySimilarityAndUserPermissions :many\nSELECT\n    t.id AS thread_id,\n    t.text,\n    t.knowledge_type,\n    t.facet_fields,\n    similarity.score,\n    t.classification,\n    t.document_id,\n    d.filename AS document_name,\n    t.source_location,\n    t.approved_at,\n    t.temporal_validity,\n    COALESCE(entity_data.entities, '{}'::text[]) AS entities,\n    COALESCE(rs.reinforcement_count, 0) AS reinforcement_count,\n    cc.name AS concept_cluster\nFROM weave.concept_clusters cc\nCROSS JOIN LATERAL (\n    SELECT CAST(\n        1.0::double precision\n        - (cc.embedding <=> $1::vector)::double precision\n        AS double precision\n    ) AS score\n) similarity\nJOIN weave.threads t ON t.cluster_id = cc.id\nJOIN weave.documents d ON d.id = t.document_id\nLEFT JOIN weave.thread_reinforcement_scores rs ON rs.thread_id = t.id\nLEFT JOIN LATERAL (\n    SELECT array_agg(DISTINCT e.canonical ORDER BY e.canonical) AS entities\n    FROM weave.thread_entities te\n    JOIN weave.entities e ON e.id = te.entity_id\n    WHERE te.thread_id = t.id\n) entity_data ON TRUE\nWHERE cc.organization_id = $2\n  AND cc.realm_kind = $3\n  AND cc.realm_key = $4\n  AND cc.level = 0\n  AND cc.embedding IS NOT NULL\n  AND t.retrieval_status = 'THREAD_RETRIEVAL_STATUS_APPROVED'\n  AND similarity.score >= $5::double precision\n  AND (\n      d.scope = 'DOCUMENT_SCOPE_ORG'\n      OR (d.scope = 'DOCUMENT_SCOPE_PRIVATE'\n          AND d.uploaded_by_user_id = $6)\n      OR (d.scope = 'DOCUMENT_SCOPE_RESTRICTED'\n          AND (\n          EXISTS (\n              SELECT 1\n              FROM weave.document_restricted_groups drg\n              JOIN weave.restricted_group_user_memberships rgum\n                ON rgum.organization_id = drg.organization_id\n               AND rgum.restricted_group_id = drg.restricted_group_id\n               AND rgum.user_id = $6\n              WHERE drg.organization_id = d.organization_id\n                AND drg.document_id = d.id\n          )\n          OR EXISTS (\n              SELECT 1\n              FROM weave.document_restricted_groups drg\n              JOIN weave.restricted_group_team_memberships rgtm\n                ON rgtm.organization_id = drg.organization_id\n               AND rgtm.restricted_group_id = drg.restricted_group_id\n              JOIN weave.org_team_memberships otm\n                ON otm.organization_id = rgtm.organization_id\n               AND otm.org_team_id = rgtm.org_team_id\n               AND otm.user_id = $6\n              WHERE drg.organization_id = d.organization_id\n                AND drg.document_id = d.id\n          )))\n  )\nORDER BY score DESC, t.approved_at DESC\nLIMIT $7";
 export interface SearchConceptClusterThreadsBySimilarityAndUserPermissionsArgs {
     queryEmbedding: string;
     organizationId: string;

package/dist/weavesql/weavedb/retrieval_sql.js

@@ -131,14 +131,30 @@ WHERE t.organization_id = $2
       OR (d.scope = 'DOCUMENT_SCOPE_PRIVATE'
           AND d.uploaded_by_user_id = $4)
       OR (d.scope = 'DOCUMENT_SCOPE_RESTRICTED'
-          AND EXISTS (
+          AND (
+          EXISTS (
               SELECT 1
-              FROM weave.user_sensitivity_clearances c
-              WHERE c.organization_id = $2
-                AND c.user_id = $4
-                AND cardinality(d.sensitivity_tags) = 1
-                AND c.sensitivity_tag_slug = d.sensitivity_tags[1]
-          ))
+              FROM weave.document_restricted_groups drg
+              JOIN weave.restricted_group_user_memberships rgum
+                ON rgum.organization_id = drg.organization_id
+               AND rgum.restricted_group_id = drg.restricted_group_id
+               AND rgum.user_id = $4
+              WHERE drg.organization_id = d.organization_id
+                AND drg.document_id = d.id
+          )
+          OR EXISTS (
+              SELECT 1
+              FROM weave.document_restricted_groups drg
+              JOIN weave.restricted_group_team_memberships rgtm
+                ON rgtm.organization_id = drg.organization_id
+               AND rgtm.restricted_group_id = drg.restricted_group_id
+              JOIN weave.org_team_memberships otm
+                ON otm.organization_id = rgtm.organization_id
+               AND otm.org_team_id = rgtm.org_team_id
+               AND otm.user_id = $4
+              WHERE drg.organization_id = d.organization_id
+                AND drg.document_id = d.id
+          )))
   )
 ORDER BY score DESC, t.approved_at DESC
 LIMIT $5`;
@@ -187,7 +203,6 @@ matches AS (
         d.filename AS document_name,
         d.mime_type,
         d.scope,
-        d.sensitivity_tags,
         d.uploaded_at,
         COALESCE(tag_data.tags, '{}'::text[]) AS tags,
         COALESCE(tag_data.tag_confidences, '{}'::real[]) AS tag_confidences,
@@ -209,17 +224,42 @@ matches AS (
           OR (d.scope = 'DOCUMENT_SCOPE_PRIVATE'
               AND d.uploaded_by_user_id = $6)
           OR (d.scope = 'DOCUMENT_SCOPE_RESTRICTED'
-              AND EXISTS (
+              AND (
+              EXISTS (
+                  SELECT 1
+                  FROM weave.document_restricted_groups drg
+                  JOIN weave.restricted_group_user_memberships rgum
+                    ON rgum.organization_id = drg.organization_id
+                   AND rgum.restricted_group_id = drg.restricted_group_id
+                   AND rgum.user_id = $6
+                  WHERE drg.organization_id = d.organization_id
+                    AND drg.document_id = d.id
+              )
+              OR EXISTS (
                   SELECT 1
-                  FROM weave.user_sensitivity_clearances c
-                  WHERE c.organization_id = $4
-                    AND c.user_id = $6
-                    AND cardinality(d.sensitivity_tags) = 1
-                    AND c.sensitivity_tag_slug = d.sensitivity_tags[1]
-              ))
+                  FROM weave.document_restricted_groups drg
+                  JOIN weave.restricted_group_team_memberships rgtm
+                    ON rgtm.organization_id = drg.organization_id
+                   AND rgtm.restricted_group_id = drg.restricted_group_id
+                  JOIN weave.org_team_memberships otm
+                    ON otm.organization_id = rgtm.organization_id
+                   AND otm.org_team_id = rgtm.org_team_id
+                   AND otm.user_id = $6
+                  WHERE drg.organization_id = d.organization_id
+                    AND drg.document_id = d.id
+              )))
+      )
+      AND (cardinality($7::text[]) = 0 OR tag_data.tags && $7::text[])
+      AND (
+          cardinality($8::uuid[]) = 0
+          OR EXISTS (
+              SELECT 1
+              FROM weave.document_restricted_groups drg
+              WHERE drg.organization_id = d.organization_id
+                AND drg.document_id = d.id
+                AND drg.restricted_group_id = ANY($8::uuid[])
+          )
       )
-      AND (cardinality($7::text[]) = 0 OR tag_data.tags && $7::text[] OR d.sensitivity_tags && $7::text[])
-      AND (cardinality($8::text[]) = 0 OR d.sensitivity_tags && $8::text[])
       AND (cardinality($9::text[]) = 0 OR d.scope = ANY($9::text[]))
       AND ($10::timestamptz IS NULL OR d.uploaded_at >= $10::timestamptz)
       AND ($11::timestamptz IS NULL OR d.uploaded_at <= $11::timestamptz)
@@ -229,7 +269,6 @@ SELECT
     document_name,
     mime_type,
     scope,
-    sensitivity_tags,
     tags,
     tag_confidences,
     uploaded_at,
@@ -246,7 +285,6 @@ GROUP BY
     document_name,
     mime_type,
     scope,
-    sensitivity_tags,
     tags,
     tag_confidences,
     uploaded_at
@@ -255,7 +293,7 @@ LIMIT $3 OFFSET $2`;
 async function searchAccessibleDocumentsSemantic(client, args) {
     const result = await client.query({
         text: exports.searchAccessibleDocumentsSemanticQuery,
-        values: [args.minSimilarity, args.pageOffset, args.resultLimit, args.organizationId, args.queryEmbedding, args.actingUserId, args.tags, args.restrictedTags, args.scopes, args.uploadedAfter, args.uploadedBefore],
+        values: [args.minSimilarity, args.pageOffset, args.resultLimit, args.organizationId, args.queryEmbedding, args.actingUserId, args.tags, args.restrictedGroupIds, args.scopes, args.uploadedAfter, args.uploadedBefore],
         rowMode: "array"
     });
     return result.rows.map(row => {
@@ -264,16 +302,15 @@ async function searchAccessibleDocumentsSemantic(client, args) {
             documentName: row[1],
             mimeType: row[2],
             scope: row[3],
-            sensitivityTags: row[4],
-            tags: row[5],
-            tagConfidences: row[6],
-            uploadedAt: row[7],
-            score: row[8],
-            matchedThreadCount: row[9],
-            bestThreadId: row[10],
-            bestText: row[11],
-            bestClassification: row[12],
-            bestSourceLocation: row[13]
+            tags: row[4],
+            tagConfidences: row[5],
+            uploadedAt: row[6],
+            score: row[7],
+            matchedThreadCount: row[8],
+            bestThreadId: row[9],
+            bestText: row[10],
+            bestClassification: row[11],
+            bestSourceLocation: row[12]
         };
     });
 }
@@ -385,14 +422,30 @@ WHERE t.organization_id = $2
       OR (d.scope = 'DOCUMENT_SCOPE_PRIVATE'
           AND d.uploaded_by_user_id = $5)
       OR (d.scope = 'DOCUMENT_SCOPE_RESTRICTED'
-          AND EXISTS (
+          AND (
+          EXISTS (
+              SELECT 1
+              FROM weave.document_restricted_groups drg
+              JOIN weave.restricted_group_user_memberships rgum
+                ON rgum.organization_id = drg.organization_id
+               AND rgum.restricted_group_id = drg.restricted_group_id
+               AND rgum.user_id = $5
+              WHERE drg.organization_id = d.organization_id
+                AND drg.document_id = d.id
+          )
+          OR EXISTS (
               SELECT 1
-              FROM weave.user_sensitivity_clearances c
-              WHERE c.organization_id = $2
-                AND c.user_id = $5
-                AND cardinality(d.sensitivity_tags) = 1
-                AND c.sensitivity_tag_slug = d.sensitivity_tags[1]
-          ))
+              FROM weave.document_restricted_groups drg
+              JOIN weave.restricted_group_team_memberships rgtm
+                ON rgtm.organization_id = drg.organization_id
+               AND rgtm.restricted_group_id = drg.restricted_group_id
+              JOIN weave.org_team_memberships otm
+                ON otm.organization_id = rgtm.organization_id
+               AND otm.org_team_id = rgtm.org_team_id
+               AND otm.user_id = $5
+              WHERE drg.organization_id = d.organization_id
+                AND drg.document_id = d.id
+          )))
   )
 ORDER BY score DESC, t.approved_at DESC
 LIMIT $6`;
@@ -515,14 +568,30 @@ WHERE t.organization_id = $1
       OR (d.scope = 'DOCUMENT_SCOPE_PRIVATE'
           AND d.uploaded_by_user_id = $3)
       OR (d.scope = 'DOCUMENT_SCOPE_RESTRICTED'
-          AND EXISTS (
+          AND (
+          EXISTS (
               SELECT 1
-              FROM weave.user_sensitivity_clearances c
-              WHERE c.organization_id = $1
-                AND c.user_id = $3
-                AND cardinality(d.sensitivity_tags) = 1
-                AND c.sensitivity_tag_slug = d.sensitivity_tags[1]
-          ))
+              FROM weave.document_restricted_groups drg
+              JOIN weave.restricted_group_user_memberships rgum
+                ON rgum.organization_id = drg.organization_id
+               AND rgum.restricted_group_id = drg.restricted_group_id
+               AND rgum.user_id = $3
+              WHERE drg.organization_id = d.organization_id
+                AND drg.document_id = d.id
+          )
+          OR EXISTS (
+              SELECT 1
+              FROM weave.document_restricted_groups drg
+              JOIN weave.restricted_group_team_memberships rgtm
+                ON rgtm.organization_id = drg.organization_id
+               AND rgtm.restricted_group_id = drg.restricted_group_id
+              JOIN weave.org_team_memberships otm
+                ON otm.organization_id = rgtm.organization_id
+               AND otm.org_team_id = rgtm.org_team_id
+               AND otm.user_id = $3
+              WHERE drg.organization_id = d.organization_id
+                AND drg.document_id = d.id
+          )))
   )
 ORDER BY score DESC, t.approved_at DESC
 LIMIT $4`;
@@ -720,14 +789,30 @@ WHERE cc.organization_id = $2
       OR (d.scope = 'DOCUMENT_SCOPE_PRIVATE'
           AND d.uploaded_by_user_id = $6)
       OR (d.scope = 'DOCUMENT_SCOPE_RESTRICTED'
-          AND EXISTS (
+          AND (
+          EXISTS (
+              SELECT 1
+              FROM weave.document_restricted_groups drg
+              JOIN weave.restricted_group_user_memberships rgum
+                ON rgum.organization_id = drg.organization_id
+               AND rgum.restricted_group_id = drg.restricted_group_id
+               AND rgum.user_id = $6
+              WHERE drg.organization_id = d.organization_id
+                AND drg.document_id = d.id
+          )
+          OR EXISTS (
               SELECT 1
-              FROM weave.user_sensitivity_clearances c
-              WHERE c.organization_id = $2
-                AND c.user_id = $6
-                AND cardinality(d.sensitivity_tags) = 1
-                AND c.sensitivity_tag_slug = d.sensitivity_tags[1]
-          ))
+              FROM weave.document_restricted_groups drg
+              JOIN weave.restricted_group_team_memberships rgtm
+                ON rgtm.organization_id = drg.organization_id
+               AND rgtm.restricted_group_id = drg.restricted_group_id
+              JOIN weave.org_team_memberships otm
+                ON otm.organization_id = rgtm.organization_id
+               AND otm.org_team_id = rgtm.org_team_id
+               AND otm.user_id = $6
+              WHERE drg.organization_id = d.organization_id
+                AND drg.document_id = d.id
+          )))
   )
 ORDER BY score DESC, t.approved_at DESC
 LIMIT $7`;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "weave-typescript",
-  "version": "0.38.0",
+  "version": "0.40.0",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
   "files": [
@@ -31,7 +31,7 @@
   "devDependencies": {
     "@types/node": "^25.2.0",
     "@types/pg": "^8.15.5",
-    "@typescript/native-preview": "7.0.0-dev.20260503.1"
+    "@typescript/native-preview": "7.0.0-dev.20260510.1"
   },
   "scripts": {
     "test": "node tools/sqlcgen.test.js",