weave-typescript 0.27.0 → 0.28.0

package/dist/weaveapi/retrieval/v1/service.pb.d.ts

@@ -0,0 +1,51 @@
+import { ExplainRetrievalRequest, ExplainRetrievalResponse, RetrieveRequest, RetrieveResponse } from "./retrieval.pb";
+export declare const protobufPackage = "weaveapi.retrieval.v1";
+export interface RetrievalService {
+    Retrieve(request: RetrieveRequest): Promise<RetrieveResponse>;
+    ExplainRetrieval(request: ExplainRetrievalRequest): Promise<ExplainRetrievalResponse>;
+}
+export declare const RetrievalServiceServiceName = "weaveapi.retrieval.v1.RetrievalService";
+export declare class RetrievalServiceClientImpl implements RetrievalService {
+    private readonly rpc;
+    private readonly service;
+    constructor(rpc: Rpc, opts?: {
+        service?: string;
+    });
+    Retrieve(request: RetrieveRequest): Promise<RetrieveResponse>;
+    ExplainRetrieval(request: ExplainRetrievalRequest): Promise<ExplainRetrievalResponse>;
+}
+export type RetrievalServiceDefinition = typeof RetrievalServiceDefinition;
+export declare const RetrievalServiceDefinition: {
+    readonly name: "RetrievalService";
+    readonly fullName: "weaveapi.retrieval.v1.RetrievalService";
+    readonly methods: {
+        readonly retrieve: {
+            readonly name: "Retrieve";
+            readonly requestType: typeof RetrieveRequest;
+            readonly requestStream: false;
+            readonly responseType: typeof RetrieveResponse;
+            readonly responseStream: false;
+            readonly options: {
+                readonly _unknownFields: {
+                    readonly 578365826: readonly [Uint8Array];
+                };
+            };
+        };
+        readonly explainRetrieval: {
+            readonly name: "ExplainRetrieval";
+            readonly requestType: typeof ExplainRetrievalRequest;
+            readonly requestStream: false;
+            readonly responseType: typeof ExplainRetrievalResponse;
+            readonly responseStream: false;
+            readonly options: {
+                readonly _unknownFields: {
+                    readonly 578365826: readonly [Uint8Array];
+                };
+            };
+        };
+    };
+};
+interface Rpc {
+    request(service: string, method: string, data: Uint8Array): Promise<Uint8Array>;
+}
+export {};

package/dist/weaveapi/retrieval/v1/service.pb.js

@@ -0,0 +1,164 @@
+"use strict";
+// Code generated by protoc-gen-ts_proto. DO NOT EDIT.
+// versions:
+//   protoc-gen-ts_proto  v2.11.6
+//   protoc               unknown
+// source: weaveapi/retrieval/v1/service.proto
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.RetrievalServiceDefinition = exports.RetrievalServiceClientImpl = exports.RetrievalServiceServiceName = exports.protobufPackage = void 0;
+/* eslint-disable */
+const wire_1 = require("@bufbuild/protobuf/wire");
+const retrieval_pb_1 = require("./retrieval.pb");
+exports.protobufPackage = "weaveapi.retrieval.v1";
+exports.RetrievalServiceServiceName = "weaveapi.retrieval.v1.RetrievalService";
+class RetrievalServiceClientImpl {
+    constructor(rpc, opts) {
+        this.service = (opts === null || opts === void 0 ? void 0 : opts.service) || exports.RetrievalServiceServiceName;
+        this.rpc = rpc;
+        this.Retrieve = this.Retrieve.bind(this);
+        this.ExplainRetrieval = this.ExplainRetrieval.bind(this);
+    }
+    Retrieve(request) {
+        const data = retrieval_pb_1.RetrieveRequest.encode(request).finish();
+        const promise = this.rpc.request(this.service, "Retrieve", data);
+        return promise.then((data) => retrieval_pb_1.RetrieveResponse.decode(new wire_1.BinaryReader(data)));
+    }
+    ExplainRetrieval(request) {
+        const data = retrieval_pb_1.ExplainRetrievalRequest.encode(request).finish();
+        const promise = this.rpc.request(this.service, "ExplainRetrieval", data);
+        return promise.then((data) => retrieval_pb_1.ExplainRetrievalResponse.decode(new wire_1.BinaryReader(data)));
+    }
+}
+exports.RetrievalServiceClientImpl = RetrievalServiceClientImpl;
+exports.RetrievalServiceDefinition = {
+    name: "RetrievalService",
+    fullName: "weaveapi.retrieval.v1.RetrievalService",
+    methods: {
+        retrieve: {
+            name: "Retrieve",
+            requestType: retrieval_pb_1.RetrieveRequest,
+            requestStream: false,
+            responseType: retrieval_pb_1.RetrieveResponse,
+            responseStream: false,
+            options: {
+                _unknownFields: {
+                    578365826: [
+                        new Uint8Array([
+                            46,
+                            58,
+                            1,
+                            42,
+                            34,
+                            41,
+                            47,
+                            118,
+                            49,
+                            47,
+                            111,
+                            114,
+                            103,
+                            47,
+                            123,
+                            111,
+                            114,
+                            103,
+                            97,
+                            110,
+                            105,
+                            122,
+                            97,
+                            116,
+                            105,
+                            111,
+                            110,
+                            95,
+                            105,
+                            100,
+                            125,
+                            47,
+                            114,
+                            101,
+                            116,
+                            114,
+                            105,
+                            101,
+                            118,
+                            97,
+                            108,
+                            47,
+                            113,
+                            117,
+                            101,
+                            114,
+                            121,
+                        ]),
+                    ],
+                },
+            },
+        },
+        explainRetrieval: {
+            name: "ExplainRetrieval",
+            requestType: retrieval_pb_1.ExplainRetrievalRequest,
+            requestStream: false,
+            responseType: retrieval_pb_1.ExplainRetrievalResponse,
+            responseStream: false,
+            options: {
+                _unknownFields: {
+                    578365826: [
+                        new Uint8Array([
+                            48,
+                            58,
+                            1,
+                            42,
+                            34,
+                            43,
+                            47,
+                            118,
+                            49,
+                            47,
+                            111,
+                            114,
+                            103,
+                            47,
+                            123,
+                            111,
+                            114,
+                            103,
+                            97,
+                            110,
+                            105,
+                            122,
+                            97,
+                            116,
+                            105,
+                            111,
+                            110,
+                            95,
+                            105,
+                            100,
+                            125,
+                            47,
+                            114,
+                            101,
+                            116,
+                            114,
+                            105,
+                            101,
+                            118,
+                            97,
+                            108,
+                            47,
+                            101,
+                            120,
+                            112,
+                            108,
+                            97,
+                            105,
+                            110,
+                        ]),
+                    ],
+                },
+            },
+        },
+    },
+};

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

@@ -0,0 +1,212 @@
+import { QueryArrayConfig, QueryArrayResult } from "pg";
+interface Client {
+    query: (config: QueryArrayConfig) => Promise<QueryArrayResult>;
+}
+export declare const createConsolidationRunQuery = "-- name: CreateConsolidationRun :one\nINSERT INTO weave.consolidation_runs (\n    id,\n    organization_id,\n    status\n) VALUES (\n    $1,\n    $2,\n    $3\n)\nRETURNING\n    id,\n    organization_id,\n    status,\n    thread_count,\n    cluster_count,\n    hierarchy_depth,\n    error,\n    started_at,\n    completed_at";
+export interface CreateConsolidationRunArgs {
+    id: string;
+    organizationId: string;
+    status: string;
+}
+export interface CreateConsolidationRunRow {
+    id: string;
+    organizationId: string;
+    status: string;
+    threadCount: number;
+    clusterCount: number;
+    hierarchyDepth: number;
+    error: string;
+    startedAt: Date;
+    completedAt: Date | null;
+}
+export declare function createConsolidationRun(client: Client, args: CreateConsolidationRunArgs): Promise<CreateConsolidationRunRow | null>;
+export declare const getConsolidationRunByOrganizationAndIDQuery = "-- name: GetConsolidationRunByOrganizationAndID :one\nSELECT\n    id,\n    organization_id,\n    status,\n    thread_count,\n    cluster_count,\n    hierarchy_depth,\n    error,\n    started_at,\n    completed_at\nFROM weave.consolidation_runs\nWHERE organization_id = $1\n  AND id = $2";
+export interface GetConsolidationRunByOrganizationAndIDArgs {
+    organizationId: string;
+    id: string;
+}
+export interface GetConsolidationRunByOrganizationAndIDRow {
+    id: string;
+    organizationId: string;
+    status: string;
+    threadCount: number;
+    clusterCount: number;
+    hierarchyDepth: number;
+    error: string;
+    startedAt: Date;
+    completedAt: Date | null;
+}
+export declare function getConsolidationRunByOrganizationAndID(client: Client, args: GetConsolidationRunByOrganizationAndIDArgs): Promise<GetConsolidationRunByOrganizationAndIDRow | null>;
+export declare const updateConsolidationRunStateQuery = "-- name: UpdateConsolidationRunState :one\nUPDATE weave.consolidation_runs\nSET\n    status = $1,\n    thread_count = $2,\n    cluster_count = $3,\n    hierarchy_depth = $4,\n    error = $5,\n    completed_at = $6\nWHERE id = $7\n  AND organization_id = $8\nRETURNING\n    id,\n    organization_id,\n    status,\n    thread_count,\n    cluster_count,\n    hierarchy_depth,\n    error,\n    started_at,\n    completed_at";
+export interface UpdateConsolidationRunStateArgs {
+    status: string;
+    threadCount: number;
+    clusterCount: number;
+    hierarchyDepth: number;
+    error: string;
+    completedAt: Date | null;
+    id: string;
+    organizationId: string;
+}
+export interface UpdateConsolidationRunStateRow {
+    id: string;
+    organizationId: string;
+    status: string;
+    threadCount: number;
+    clusterCount: number;
+    hierarchyDepth: number;
+    error: string;
+    startedAt: Date;
+    completedAt: Date | null;
+}
+export declare function updateConsolidationRunState(client: Client, args: UpdateConsolidationRunStateArgs): Promise<UpdateConsolidationRunStateRow | null>;
+export declare const listConsolidationRunsByOrganizationQuery = "-- name: ListConsolidationRunsByOrganization :many\nSELECT\n    id,\n    organization_id,\n    status,\n    thread_count,\n    cluster_count,\n    hierarchy_depth,\n    error,\n    started_at,\n    completed_at\nFROM weave.consolidation_runs\nWHERE organization_id = $1\nORDER BY started_at DESC\nLIMIT $3 OFFSET $2";
+export interface ListConsolidationRunsByOrganizationArgs {
+    organizationId: string;
+    pageOffset: string;
+    pageSize: string;
+}
+export interface ListConsolidationRunsByOrganizationRow {
+    id: string;
+    organizationId: string;
+    status: string;
+    threadCount: number;
+    clusterCount: number;
+    hierarchyDepth: number;
+    error: string;
+    startedAt: Date;
+    completedAt: Date | null;
+}
+export declare function listConsolidationRunsByOrganization(client: Client, args: ListConsolidationRunsByOrganizationArgs): Promise<ListConsolidationRunsByOrganizationRow[]>;
+export declare const upsertConceptClusterQuery = "-- name: UpsertConceptCluster :one\nINSERT INTO weave.concept_clusters (\n    id,\n    organization_id,\n    realm_kind,\n    realm_key,\n    name,\n    description,\n    knowledge_type,\n    level,\n    parent_id,\n    embedding,\n    centroid,\n    thread_count,\n    consolidation_run_id\n) VALUES (\n    $1,\n    $2,\n    $3,\n    $4,\n    $5,\n    $6,\n    $7,\n    $8,\n    $9,\n    $10::vector,\n    $11::vector,\n    $12,\n    $13\n)\nON CONFLICT (id) DO UPDATE\nSET\n    name = EXCLUDED.name,\n    description = EXCLUDED.description,\n    realm_kind = EXCLUDED.realm_kind,\n    realm_key = EXCLUDED.realm_key,\n    knowledge_type = EXCLUDED.knowledge_type,\n    level = EXCLUDED.level,\n    parent_id = EXCLUDED.parent_id,\n    embedding = EXCLUDED.embedding,\n    centroid = EXCLUDED.centroid,\n    thread_count = EXCLUDED.thread_count,\n    consolidation_run_id = EXCLUDED.consolidation_run_id,\n    updated_at = now()\nRETURNING\n    id,\n    organization_id,\n    realm_kind,\n    realm_key,\n    name,\n    description,\n    knowledge_type,\n    level,\n    parent_id,\n    thread_count,\n    consolidation_run_id,\n    created_at,\n    updated_at";
+export interface UpsertConceptClusterArgs {
+    id: string;
+    organizationId: string;
+    realmKind: string;
+    realmKey: string;
+    name: string;
+    description: string;
+    knowledgeType: string;
+    level: number;
+    parentId: string | null;
+    embedding: string;
+    centroid: string;
+    threadCount: number;
+    consolidationRunId: string;
+}
+export interface UpsertConceptClusterRow {
+    id: string;
+    organizationId: string;
+    realmKind: string;
+    realmKey: string;
+    name: string;
+    description: string;
+    knowledgeType: string;
+    level: number;
+    parentId: string | null;
+    threadCount: number;
+    consolidationRunId: string;
+    createdAt: Date;
+    updatedAt: Date;
+}
+export declare function upsertConceptCluster(client: Client, args: UpsertConceptClusterArgs): Promise<UpsertConceptClusterRow | null>;
+export declare const getConceptClusterByOrganizationRealmAndIDQuery = "-- name: GetConceptClusterByOrganizationRealmAndID :one\nSELECT\n    id,\n    organization_id,\n    realm_kind,\n    realm_key,\n    name,\n    description,\n    knowledge_type,\n    level,\n    parent_id,\n    thread_count,\n    consolidation_run_id,\n    created_at,\n    updated_at\nFROM weave.concept_clusters\nWHERE organization_id = $1\n  AND realm_kind = $2\n  AND realm_key = $3\n  AND id = $4";
+export interface GetConceptClusterByOrganizationRealmAndIDArgs {
+    organizationId: string;
+    realmKind: string;
+    realmKey: string;
+    id: string;
+}
+export interface GetConceptClusterByOrganizationRealmAndIDRow {
+    id: string;
+    organizationId: string;
+    realmKind: string;
+    realmKey: string;
+    name: string;
+    description: string;
+    knowledgeType: string;
+    level: number;
+    parentId: string | null;
+    threadCount: number;
+    consolidationRunId: string;
+    createdAt: Date;
+    updatedAt: Date;
+}
+export declare function getConceptClusterByOrganizationRealmAndID(client: Client, args: GetConceptClusterByOrganizationRealmAndIDArgs): Promise<GetConceptClusterByOrganizationRealmAndIDRow | null>;
+export declare const listConceptClustersByRealmQuery = "-- name: ListConceptClustersByRealm :many\nSELECT\n    id,\n    organization_id,\n    realm_kind,\n    realm_key,\n    name,\n    description,\n    knowledge_type,\n    level,\n    parent_id,\n    thread_count,\n    consolidation_run_id,\n    created_at,\n    updated_at\nFROM weave.concept_clusters\nWHERE organization_id = $1\n  AND realm_kind = $2\n  AND realm_key = $3\n  AND ($4 = '' OR knowledge_type = $4)\n  AND ($5 <= 0 OR level <= $5)\nORDER BY level ASC, updated_at DESC, id ASC";
+export interface ListConceptClustersByRealmArgs {
+    organizationId: string;
+    realmKind: string;
+    realmKey: string;
+    knowledgeType: string | null;
+    maxDepth: string | null;
+}
+export interface ListConceptClustersByRealmRow {
+    id: string;
+    organizationId: string;
+    realmKind: string;
+    realmKey: string;
+    name: string;
+    description: string;
+    knowledgeType: string;
+    level: number;
+    parentId: string | null;
+    threadCount: number;
+    consolidationRunId: string;
+    createdAt: Date;
+    updatedAt: Date;
+}
+export declare function listConceptClustersByRealm(client: Client, args: ListConceptClustersByRealmArgs): Promise<ListConceptClustersByRealmRow[]>;
+export declare const listConceptClustersByParentAndRealmQuery = "-- name: ListConceptClustersByParentAndRealm :many\nSELECT\n    id,\n    organization_id,\n    realm_kind,\n    realm_key,\n    name,\n    description,\n    knowledge_type,\n    level,\n    parent_id,\n    thread_count,\n    consolidation_run_id,\n    created_at,\n    updated_at\nFROM weave.concept_clusters\nWHERE organization_id = $1\n  AND realm_kind = $2\n  AND realm_key = $3\n  AND parent_id = $4\nORDER BY updated_at DESC, id ASC";
+export interface ListConceptClustersByParentAndRealmArgs {
+    organizationId: string;
+    realmKind: string;
+    realmKey: string;
+    parentId: string | null;
+}
+export interface ListConceptClustersByParentAndRealmRow {
+    id: string;
+    organizationId: string;
+    realmKind: string;
+    realmKey: string;
+    name: string;
+    description: string;
+    knowledgeType: string;
+    level: number;
+    parentId: string | null;
+    threadCount: number;
+    consolidationRunId: string;
+    createdAt: Date;
+    updatedAt: Date;
+}
+export declare function listConceptClustersByParentAndRealm(client: Client, args: ListConceptClustersByParentAndRealmArgs): Promise<ListConceptClustersByParentAndRealmRow[]>;
+export declare const listApprovedThreadsForConsolidationQuery = "-- name: ListApprovedThreadsForConsolidation :many\nSELECT\n    t.id,\n    t.organization_id,\n    t.document_id,\n    t.ingestion_run_id,\n    t.text,\n    t.knowledge_type,\n    t.embedding::text AS embedding_text,\n    t.approved_at,\n    CASE\n        WHEN d.scope = 'DOCUMENT_SCOPE_ORG'\n            THEN 'KNOWLEDGE_REALM_KIND_ORG'::text\n        WHEN d.scope = 'DOCUMENT_SCOPE_RESTRICTED'\n            THEN 'KNOWLEDGE_REALM_KIND_TAG'::text\n        WHEN d.scope = 'DOCUMENT_SCOPE_PRIVATE'\n            THEN 'KNOWLEDGE_REALM_KIND_USER'::text\n    END AS realm_kind,\n    CASE\n        WHEN d.scope = 'DOCUMENT_SCOPE_ORG'\n            THEN ''::text\n        WHEN d.scope = 'DOCUMENT_SCOPE_RESTRICTED'\n            THEN COALESCE(d.sensitivity_tags[1], ''::text)\n        WHEN d.scope = 'DOCUMENT_SCOPE_PRIVATE'\n            THEN d.uploaded_by_user_id::text\n    END AS realm_key\nFROM weave.threads t\nJOIN weave.documents d ON d.id = t.document_id\nWHERE t.organization_id = $1\n  AND t.retrieval_status = 'THREAD_RETRIEVAL_STATUS_APPROVED'\n  AND t.embedding IS NOT NULL\nORDER BY t.approved_at DESC, t.id ASC\nLIMIT $3 OFFSET $2";
+export interface ListApprovedThreadsForConsolidationArgs {
+    organizationId: string;
+    pageOffset: string;
+    pageSize: string;
+}
+export interface ListApprovedThreadsForConsolidationRow {
+    id: string;
+    organizationId: string;
+    documentId: string;
+    ingestionRunId: string;
+    text: string;
+    knowledgeType: string;
+    embeddingText: string;
+    approvedAt: Date | null;
+    realmKind: string | null;
+    realmKey: string | null;
+}
+export declare function listApprovedThreadsForConsolidation(client: Client, args: ListApprovedThreadsForConsolidationArgs): Promise<ListApprovedThreadsForConsolidationRow[]>;
+export declare const assignThreadsToClusterQuery = "-- name: AssignThreadsToCluster :execrows\nUPDATE weave.threads AS t\nSET cluster_id = cc.id\nFROM weave.concept_clusters cc,\n     weave.documents d\nWHERE t.organization_id = $1\n  AND t.id = ANY($2::uuid[])\n  AND cc.organization_id = t.organization_id\n  AND cc.id = $3\n  AND d.id = t.document_id\n  AND (\n      (cc.realm_kind = 'KNOWLEDGE_REALM_KIND_ORG'\n          AND cc.realm_key = ''\n          AND d.scope = 'DOCUMENT_SCOPE_ORG')\n      OR (cc.realm_kind = 'KNOWLEDGE_REALM_KIND_TAG'\n          AND d.scope = 'DOCUMENT_SCOPE_RESTRICTED'\n          AND cardinality(d.sensitivity_tags) = 1\n          AND cc.realm_key = d.sensitivity_tags[1])\n      OR (cc.realm_kind = 'KNOWLEDGE_REALM_KIND_USER'\n          AND d.scope = 'DOCUMENT_SCOPE_PRIVATE'\n          AND cc.realm_key = d.uploaded_by_user_id::text)\n  )";
+export interface AssignThreadsToClusterArgs {
+    organizationId: string;
+    threadIds: string[];
+    clusterId: string;
+}
+export declare const clearThreadClusterAssignmentsByOrganizationQuery = "-- name: ClearThreadClusterAssignmentsByOrganization :execrows\nUPDATE weave.threads\nSET cluster_id = NULL\nWHERE organization_id = $1\n  AND cluster_id IS NOT NULL";
+export interface ClearThreadClusterAssignmentsByOrganizationArgs {
+    organizationId: string;
+}
+export {};