weave-typescript 0.51.2 → 0.52.0

package/dist/weavesql/weavedb/chat_sql.js

@@ -1,49 +1,48 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.listChatArtifactVersionsQuery = exports.getChatArtifactVersionQuery = exports.insertChatArtifactVersionWithDocumentReferencesQuery = exports.insertChatArtifactVersionQuery = exports.getChatArtifactQuery = exports.listChatArtifactsForSessionQuery = exports.bumpChatArtifactVersionQuery = exports.createChatArtifactQuery = exports.clearOrganizationChatSettingsDefaultIfMatchesQuery = exports.getOrganizationChatSettingsQuery = exports.upsertOrganizationChatSettingsQuery = exports.getOrganizationApprovedModelQuery = exports.listOrganizationApprovedModelsQuery = exports.deleteOrganizationApprovedModelQuery = exports.createOrganizationApprovedModelQuery = exports.completeChatMessageQuery = exports.updateChatMessageToolCallStateQuery = exports.listChatMessagesForSessionQuery = exports.getChatMessageQuery = exports.insertChatMessageQuery = exports.listProjectsForChatSessionQuery = exports.detachProjectFromChatSessionQuery = exports.attachProjectToChatSessionQuery = exports.softDeleteChatSessionQuery = exports.touchChatSessionLastMessageQuery = exports.updateChatSessionTitleQuery = exports.updateChatSessionModelQuery = exports.listChatSessionsForUserByProjectQuery = exports.listChatSessionsForUserQuery = exports.getChatSessionQuery = exports.createChatSessionQuery = void 0;
+exports.getCurrentChatArtifactVersionQuery = exports.getChatArtifactVersionQuery = exports.createChatArtifactVersionQuery = exports.listChatArtifactsQuery = exports.getChatArtifactQuery = exports.bumpChatArtifactVersionQuery = exports.createChatArtifactQuery = exports.listChatMessagesQuery = exports.createChatMessageQuery = exports.updateChatSessionTitleQuery = exports.listChatSessionsByUserQuery = exports.getChatSessionQuery = exports.createChatSessionQuery = void 0;
 exports.createChatSession = createChatSession;
 exports.getChatSession = getChatSession;
-exports.listChatSessionsForUser = listChatSessionsForUser;
-exports.listChatSessionsForUserByProject = listChatSessionsForUserByProject;
-exports.updateChatSessionModel = updateChatSessionModel;
+exports.listChatSessionsByUser = listChatSessionsByUser;
 exports.updateChatSessionTitle = updateChatSessionTitle;
-exports.touchChatSessionLastMessage = touchChatSessionLastMessage;
-exports.softDeleteChatSession = softDeleteChatSession;
-exports.attachProjectToChatSession = attachProjectToChatSession;
-exports.detachProjectFromChatSession = detachProjectFromChatSession;
-exports.listProjectsForChatSession = listProjectsForChatSession;
-exports.insertChatMessage = insertChatMessage;
-exports.getChatMessage = getChatMessage;
-exports.listChatMessagesForSession = listChatMessagesForSession;
-exports.updateChatMessageToolCallState = updateChatMessageToolCallState;
-exports.completeChatMessage = completeChatMessage;
-exports.createOrganizationApprovedModel = createOrganizationApprovedModel;
-exports.deleteOrganizationApprovedModel = deleteOrganizationApprovedModel;
-exports.listOrganizationApprovedModels = listOrganizationApprovedModels;
-exports.getOrganizationApprovedModel = getOrganizationApprovedModel;
-exports.upsertOrganizationChatSettings = upsertOrganizationChatSettings;
-exports.getOrganizationChatSettings = getOrganizationChatSettings;
-exports.clearOrganizationChatSettingsDefaultIfMatches = clearOrganizationChatSettingsDefaultIfMatches;
+exports.createChatMessage = createChatMessage;
+exports.listChatMessages = listChatMessages;
 exports.createChatArtifact = createChatArtifact;
 exports.bumpChatArtifactVersion = bumpChatArtifactVersion;
-exports.listChatArtifactsForSession = listChatArtifactsForSession;
 exports.getChatArtifact = getChatArtifact;
-exports.insertChatArtifactVersion = insertChatArtifactVersion;
-exports.insertChatArtifactVersionWithDocumentReferences = insertChatArtifactVersionWithDocumentReferences;
+exports.listChatArtifacts = listChatArtifacts;
+exports.createChatArtifactVersion = createChatArtifactVersion;
 exports.getChatArtifactVersion = getChatArtifactVersion;
-exports.listChatArtifactVersions = listChatArtifactVersions;
+exports.getCurrentChatArtifactVersion = getCurrentChatArtifactVersion;
 exports.createChatSessionQuery = `-- name: CreateChatSession :one
-INSERT INTO weave.chat_sessions (
-    id, organization_id, created_by_user_id, title,
-    model_id, provider_configuration_id
-) VALUES (
-    $1, $2, $3, $4, $5, $6
+WITH resolved_agent AS (
+    SELECT COALESCE($4::uuid, o.default_chat_agent_id) AS root_agent_id
+    FROM weave.organization o
+    WHERE o.id = $1
+),
+valid_agent AS (
+    SELECT r.root_agent_id
+    FROM resolved_agent r
+             JOIN weave.agent a
+                  ON a.organization_id = $1
+                      AND a.id = r.root_agent_id
 )
-RETURNING id, organization_id, created_by_user_id, title, model_id, provider_configuration_id, created_at, updated_at, last_message_at, deleted_at`;
+INSERT INTO weave.chat_session (
+    organization_id,
+    created_by_user_id,
+    root_agent_id,
+    title
+) SELECT
+    $1,
+    $2,
+    valid_agent.root_agent_id,
+    COALESCE($3::text, '')
+FROM valid_agent
+RETURNING id, organization_id, created_by_user_id, root_agent_id, title, last_message_at, created_at, updated_at`;
 async function createChatSession(client, args) {
     const result = await client.query({
         text: exports.createChatSessionQuery,
-        values: [args.id, args.organizationId, args.createdByUserId, args.title, args.modelId, args.providerConfigurationId],
+        values: [args.organizationId, args.createdByUserId, args.title, args.rootAgentId],
         rowMode: "array"
     });
     if (result.rows.length !== 1) {
@@ -54,22 +53,23 @@ async function createChatSession(client, args) {
         id: row[0],
         organizationId: row[1],
         createdByUserId: row[2],
-        title: row[3],
-        modelId: row[4],
-        providerConfigurationId: row[5],
+        rootAgentId: row[3],
+        title: row[4],
+        lastMessageAt: row[5],
         createdAt: row[6],
-        updatedAt: row[7],
-        lastMessageAt: row[8],
-        deletedAt: row[9]
+        updatedAt: row[7]
     };
 }
 exports.getChatSessionQuery = `-- name: GetChatSession :one
-SELECT id, organization_id, created_by_user_id, title, model_id, provider_configuration_id, created_at, updated_at, last_message_at, deleted_at FROM weave.chat_sessions
-WHERE id = $1 AND deleted_at IS NULL`;
+SELECT id, organization_id, created_by_user_id, root_agent_id, title, last_message_at, created_at, updated_at
+FROM weave.chat_session
+WHERE organization_id = $1
+  AND id = $2
+  AND created_by_user_id = $3`;
 async function getChatSession(client, args) {
     const result = await client.query({
         text: exports.getChatSessionQuery,
-        values: [args.id],
+        values: [args.organizationId, args.sessionId, args.createdByUserId],
         rowMode: "array"
     });
     if (result.rows.length !== 1) {
@@ -80,56 +80,25 @@ async function getChatSession(client, args) {
         id: row[0],
         organizationId: row[1],
         createdByUserId: row[2],
-        title: row[3],
-        modelId: row[4],
-        providerConfigurationId: row[5],
+        rootAgentId: row[3],
+        title: row[4],
+        lastMessageAt: row[5],
         createdAt: row[6],
-        updatedAt: row[7],
-        lastMessageAt: row[8],
-        deletedAt: row[9]
+        updatedAt: row[7]
     };
 }
-exports.listChatSessionsForUserQuery = `-- name: ListChatSessionsForUser :many
-SELECT id, organization_id, created_by_user_id, title, model_id, provider_configuration_id, created_at, updated_at, last_message_at, deleted_at FROM weave.chat_sessions
+exports.listChatSessionsByUserQuery = `-- name: ListChatSessionsByUser :many
+SELECT id, organization_id, created_by_user_id, root_agent_id, title, last_message_at, created_at, updated_at
+FROM weave.chat_session
 WHERE organization_id = $1
   AND created_by_user_id = $2
-  AND deleted_at IS NULL
-ORDER BY last_message_at DESC
-LIMIT $3 OFFSET $4`;
-async function listChatSessionsForUser(client, args) {
-    const result = await client.query({
-        text: exports.listChatSessionsForUserQuery,
-        values: [args.organizationId, args.createdByUserId, args.limit, args.offset],
-        rowMode: "array"
-    });
-    return result.rows.map(row => {
-        return {
-            id: row[0],
-            organizationId: row[1],
-            createdByUserId: row[2],
-            title: row[3],
-            modelId: row[4],
-            providerConfigurationId: row[5],
-            createdAt: row[6],
-            updatedAt: row[7],
-            lastMessageAt: row[8],
-            deletedAt: row[9]
-        };
-    });
-}
-exports.listChatSessionsForUserByProjectQuery = `-- name: ListChatSessionsForUserByProject :many
-SELECT s.id, s.organization_id, s.created_by_user_id, s.title, s.model_id, s.provider_configuration_id, s.created_at, s.updated_at, s.last_message_at, s.deleted_at FROM weave.chat_sessions s
-JOIN weave.chat_session_projects p ON p.session_id = s.id
-WHERE s.organization_id = $1
-  AND s.created_by_user_id = $2
-  AND p.project_id = $3
-  AND s.deleted_at IS NULL
-ORDER BY s.last_message_at DESC
-LIMIT $4 OFFSET $5`;
-async function listChatSessionsForUserByProject(client, args) {
+ORDER BY last_message_at DESC, created_at DESC
+LIMIT $4::integer
+OFFSET $3::integer`;
+async function listChatSessionsByUser(client, args) {
     const result = await client.query({
-        text: exports.listChatSessionsForUserByProjectQuery,
-        values: [args.organizationId, args.createdByUserId, args.projectId, args.limit, args.offset],
+        text: exports.listChatSessionsByUserQuery,
+        values: [args.organizationId, args.createdByUserId, args.offsetCount, args.limitCount],
         rowMode: "array"
     });
     return result.rows.map(row => {
@@ -137,55 +106,25 @@ async function listChatSessionsForUserByProject(client, args) {
             id: row[0],
             organizationId: row[1],
             createdByUserId: row[2],
-            title: row[3],
-            modelId: row[4],
-            providerConfigurationId: row[5],
+            rootAgentId: row[3],
+            title: row[4],
+            lastMessageAt: row[5],
             createdAt: row[6],
-            updatedAt: row[7],
-            lastMessageAt: row[8],
-            deletedAt: row[9]
+            updatedAt: row[7]
         };
     });
 }
-exports.updateChatSessionModelQuery = `-- name: UpdateChatSessionModel :one
-UPDATE weave.chat_sessions
-SET model_id = $2,
-    provider_configuration_id = $3,
-    updated_at = now()
-WHERE id = $1
-RETURNING id, organization_id, created_by_user_id, title, model_id, provider_configuration_id, created_at, updated_at, last_message_at, deleted_at`;
-async function updateChatSessionModel(client, args) {
-    const result = await client.query({
-        text: exports.updateChatSessionModelQuery,
-        values: [args.id, args.modelId, args.providerConfigurationId],
-        rowMode: "array"
-    });
-    if (result.rows.length !== 1) {
-        return null;
-    }
-    const row = result.rows[0];
-    return {
-        id: row[0],
-        organizationId: row[1],
-        createdByUserId: row[2],
-        title: row[3],
-        modelId: row[4],
-        providerConfigurationId: row[5],
-        createdAt: row[6],
-        updatedAt: row[7],
-        lastMessageAt: row[8],
-        deletedAt: row[9]
-    };
-}
 exports.updateChatSessionTitleQuery = `-- name: UpdateChatSessionTitle :one
-UPDATE weave.chat_sessions
-SET title = $2, updated_at = now()
-WHERE id = $1
-RETURNING id, organization_id, created_by_user_id, title, model_id, provider_configuration_id, created_at, updated_at, last_message_at, deleted_at`;
+UPDATE weave.chat_session
+SET title = $1
+WHERE organization_id = $2
+  AND id = $3
+  AND created_by_user_id = $4
+RETURNING id, organization_id, created_by_user_id, root_agent_id, title, last_message_at, created_at, updated_at`;
 async function updateChatSessionTitle(client, args) {
     const result = await client.query({
         text: exports.updateChatSessionTitleQuery,
-        values: [args.id, args.title],
+        values: [args.title, args.organizationId, args.sessionId, args.createdByUserId],
         rowMode: "array"
     });
     if (result.rows.length !== 1) {
@@ -196,254 +135,51 @@ async function updateChatSessionTitle(client, args) {
         id: row[0],
         organizationId: row[1],
         createdByUserId: row[2],
-        title: row[3],
-        modelId: row[4],
-        providerConfigurationId: row[5],
+        rootAgentId: row[3],
+        title: row[4],
+        lastMessageAt: row[5],
         createdAt: row[6],
-        updatedAt: row[7],
-        lastMessageAt: row[8],
-        deletedAt: row[9]
-    };
-}
-exports.touchChatSessionLastMessageQuery = `-- name: TouchChatSessionLastMessage :exec
-UPDATE weave.chat_sessions
-SET last_message_at = $2, updated_at = $2
-WHERE id = $1`;
-async function touchChatSessionLastMessage(client, args) {
-    await client.query({
-        text: exports.touchChatSessionLastMessageQuery,
-        values: [args.id, args.lastMessageAt],
-        rowMode: "array"
-    });
-}
-exports.softDeleteChatSessionQuery = `-- name: SoftDeleteChatSession :exec
-UPDATE weave.chat_sessions
-SET deleted_at = now(), updated_at = now()
-WHERE id = $1`;
-async function softDeleteChatSession(client, args) {
-    await client.query({
-        text: exports.softDeleteChatSessionQuery,
-        values: [args.id],
-        rowMode: "array"
-    });
-}
-exports.attachProjectToChatSessionQuery = `-- name: AttachProjectToChatSession :exec
-INSERT INTO weave.chat_session_projects (
-    session_id, project_id, attached_by_user_id
-) VALUES ($1, $2, $3)
-ON CONFLICT DO NOTHING`;
-async function attachProjectToChatSession(client, args) {
-    await client.query({
-        text: exports.attachProjectToChatSessionQuery,
-        values: [args.sessionId, args.projectId, args.attachedByUserId],
-        rowMode: "array"
-    });
-}
-exports.detachProjectFromChatSessionQuery = `-- name: DetachProjectFromChatSession :exec
-DELETE FROM weave.chat_session_projects
-WHERE session_id = $1 AND project_id = $2`;
-async function detachProjectFromChatSession(client, args) {
-    await client.query({
-        text: exports.detachProjectFromChatSessionQuery,
-        values: [args.sessionId, args.projectId],
-        rowMode: "array"
-    });
-}
-exports.listProjectsForChatSessionQuery = `-- name: ListProjectsForChatSession :many
-SELECT project_id FROM weave.chat_session_projects
-WHERE session_id = $1`;
-async function listProjectsForChatSession(client, args) {
-    const result = await client.query({
-        text: exports.listProjectsForChatSessionQuery,
-        values: [args.sessionId],
-        rowMode: "array"
-    });
-    return result.rows.map(row => {
-        return {
-            projectId: row[0]
-        };
-    });
-}
-exports.insertChatMessageQuery = `-- name: InsertChatMessage :one
-INSERT INTO weave.chat_messages (
-    id, session_id, role, content,
-    reasoning_content,
-    parent_message_id, tool_call_id, tool_call_state,
-    model_id, ephemeral_attachment_count, ephemeral_attachment_mime_types,
-    token_usage, stream_state
-) VALUES (
-    $1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13
-)
-RETURNING id, session_id, role, content, parent_message_id, tool_call_id, tool_call_state, model_id, ephemeral_attachment_count, ephemeral_attachment_mime_types, token_usage, stream_state, created_at, completed_at, reasoning_content`;
-async function insertChatMessage(client, args) {
-    const result = await client.query({
-        text: exports.insertChatMessageQuery,
-        values: [args.id, args.sessionId, args.role, args.content, args.reasoningContent, args.parentMessageId, args.toolCallId, args.toolCallState, args.modelId, args.ephemeralAttachmentCount, args.ephemeralAttachmentMimeTypes, args.tokenUsage, args.streamState],
-        rowMode: "array"
-    });
-    if (result.rows.length !== 1) {
-        return null;
-    }
-    const row = result.rows[0];
-    return {
-        id: row[0],
-        sessionId: row[1],
-        role: row[2],
-        content: row[3],
-        parentMessageId: row[4],
-        toolCallId: row[5],
-        toolCallState: row[6],
-        modelId: row[7],
-        ephemeralAttachmentCount: row[8],
-        ephemeralAttachmentMimeTypes: row[9],
-        tokenUsage: row[10],
-        streamState: row[11],
-        createdAt: row[12],
-        completedAt: row[13],
-        reasoningContent: row[14]
-    };
-}
-exports.getChatMessageQuery = `-- name: GetChatMessage :one
-SELECT id, session_id, role, content, parent_message_id, tool_call_id, tool_call_state, model_id, ephemeral_attachment_count, ephemeral_attachment_mime_types, token_usage, stream_state, created_at, completed_at, reasoning_content FROM weave.chat_messages WHERE id = $1`;
-async function getChatMessage(client, args) {
-    const result = await client.query({
-        text: exports.getChatMessageQuery,
-        values: [args.id],
-        rowMode: "array"
-    });
-    if (result.rows.length !== 1) {
-        return null;
-    }
-    const row = result.rows[0];
-    return {
-        id: row[0],
-        sessionId: row[1],
-        role: row[2],
-        content: row[3],
-        parentMessageId: row[4],
-        toolCallId: row[5],
-        toolCallState: row[6],
-        modelId: row[7],
-        ephemeralAttachmentCount: row[8],
-        ephemeralAttachmentMimeTypes: row[9],
-        tokenUsage: row[10],
-        streamState: row[11],
-        createdAt: row[12],
-        completedAt: row[13],
-        reasoningContent: row[14]
-    };
-}
-exports.listChatMessagesForSessionQuery = `-- name: ListChatMessagesForSession :many
-SELECT id, session_id, role, content, parent_message_id, tool_call_id, tool_call_state, model_id, ephemeral_attachment_count, ephemeral_attachment_mime_types, token_usage, stream_state, created_at, completed_at, reasoning_content FROM weave.chat_messages
-WHERE session_id = $1
-ORDER BY created_at`;
-async function listChatMessagesForSession(client, args) {
-    const result = await client.query({
-        text: exports.listChatMessagesForSessionQuery,
-        values: [args.sessionId],
-        rowMode: "array"
-    });
-    return result.rows.map(row => {
-        return {
-            id: row[0],
-            sessionId: row[1],
-            role: row[2],
-            content: row[3],
-            parentMessageId: row[4],
-            toolCallId: row[5],
-            toolCallState: row[6],
-            modelId: row[7],
-            ephemeralAttachmentCount: row[8],
-            ephemeralAttachmentMimeTypes: row[9],
-            tokenUsage: row[10],
-            streamState: row[11],
-            createdAt: row[12],
-            completedAt: row[13],
-            reasoningContent: row[14]
-        };
-    });
-}
-exports.updateChatMessageToolCallStateQuery = `-- name: UpdateChatMessageToolCallState :one
-UPDATE weave.chat_messages
-SET tool_call_state = $2
-WHERE id = $1
-RETURNING id, session_id, role, content, parent_message_id, tool_call_id, tool_call_state, model_id, ephemeral_attachment_count, ephemeral_attachment_mime_types, token_usage, stream_state, created_at, completed_at, reasoning_content`;
-async function updateChatMessageToolCallState(client, args) {
-    const result = await client.query({
-        text: exports.updateChatMessageToolCallStateQuery,
-        values: [args.id, args.toolCallState],
-        rowMode: "array"
-    });
-    if (result.rows.length !== 1) {
-        return null;
-    }
-    const row = result.rows[0];
-    return {
-        id: row[0],
-        sessionId: row[1],
-        role: row[2],
-        content: row[3],
-        parentMessageId: row[4],
-        toolCallId: row[5],
-        toolCallState: row[6],
-        modelId: row[7],
-        ephemeralAttachmentCount: row[8],
-        ephemeralAttachmentMimeTypes: row[9],
-        tokenUsage: row[10],
-        streamState: row[11],
-        createdAt: row[12],
-        completedAt: row[13],
-        reasoningContent: row[14]
-    };
-}
-exports.completeChatMessageQuery = `-- name: CompleteChatMessage :one
-UPDATE weave.chat_messages
-SET stream_state = $2,
-    content = $3,
-    reasoning_content = $4,
-    token_usage = $5,
-    completed_at = now()
-WHERE id = $1
-RETURNING id, session_id, role, content, parent_message_id, tool_call_id, tool_call_state, model_id, ephemeral_attachment_count, ephemeral_attachment_mime_types, token_usage, stream_state, created_at, completed_at, reasoning_content`;
-async function completeChatMessage(client, args) {
-    const result = await client.query({
-        text: exports.completeChatMessageQuery,
-        values: [args.id, args.streamState, args.content, args.reasoningContent, args.tokenUsage],
-        rowMode: "array"
-    });
-    if (result.rows.length !== 1) {
-        return null;
-    }
-    const row = result.rows[0];
-    return {
-        id: row[0],
-        sessionId: row[1],
-        role: row[2],
-        content: row[3],
-        parentMessageId: row[4],
-        toolCallId: row[5],
-        toolCallState: row[6],
-        modelId: row[7],
-        ephemeralAttachmentCount: row[8],
-        ephemeralAttachmentMimeTypes: row[9],
-        tokenUsage: row[10],
-        streamState: row[11],
-        createdAt: row[12],
-        completedAt: row[13],
-        reasoningContent: row[14]
+        updatedAt: row[7]
     };
 }
-exports.createOrganizationApprovedModelQuery = `-- name: CreateOrganizationApprovedModel :one
-INSERT INTO weave.organization_approved_models (
-    id, organization_id, provider_configuration_id, model_id, approved_by_user_id
-) VALUES (
-    $1, $2, $3, $4, $5
+exports.createChatMessageQuery = `-- name: CreateChatMessage :one
+WITH created_message AS (
+    INSERT INTO weave.chat_message (
+        organization_id,
+        session_id,
+        agent_run_id,
+        role,
+        parts
+    )
+    SELECT
+        $1,
+        $2,
+        $3::uuid,
+        $4,
+        COALESCE($5::jsonb, '[]'::jsonb)
+    WHERE EXISTS (
+        SELECT 1
+        FROM weave.chat_session s
+        WHERE s.organization_id = $1
+          AND s.id = $2
+          AND s.created_by_user_id = $6
+    )
+    RETURNING id, organization_id, session_id, agent_run_id, role, parts, created_at
+),
+touched_session AS (
+    UPDATE weave.chat_session
+    SET last_message_at = created_message.created_at
+    FROM created_message
+    WHERE weave.chat_session.organization_id = created_message.organization_id
+      AND weave.chat_session.id = created_message.session_id
+    RETURNING weave.chat_session.id
 )
-RETURNING id, organization_id, provider_configuration_id, model_id, approved_by_user_id, approved_at`;
-async function createOrganizationApprovedModel(client, args) {
+SELECT id, organization_id, session_id, agent_run_id, role, parts, created_at
+FROM created_message`;
+async function createChatMessage(client, args) {
     const result = await client.query({
-        text: exports.createOrganizationApprovedModelQuery,
-        values: [args.id, args.organizationId, args.providerConfigurationId, args.modelId, args.approvedByUserId],
+        text: exports.createChatMessageQuery,
+        values: [args.organizationId, args.sessionId, args.agentRunId, args.role, args.parts, args.createdByUserId],
         rowMode: "array"
     });
     if (result.rows.length !== 1) {
@@ -453,52 +189,70 @@ async function createOrganizationApprovedModel(client, args) {
     return {
         id: row[0],
         organizationId: row[1],
-        providerConfigurationId: row[2],
-        modelId: row[3],
-        approvedByUserId: row[4],
-        approvedAt: row[5]
+        sessionId: row[2],
+        agentRunId: row[3],
+        role: row[4],
+        parts: row[5],
+        createdAt: row[6]
     };
 }
-exports.deleteOrganizationApprovedModelQuery = `-- name: DeleteOrganizationApprovedModel :exec
-DELETE FROM weave.organization_approved_models
-WHERE id = $1 AND organization_id = $2`;
-async function deleteOrganizationApprovedModel(client, args) {
-    await client.query({
-        text: exports.deleteOrganizationApprovedModelQuery,
-        values: [args.id, args.organizationId],
-        rowMode: "array"
-    });
-}
-exports.listOrganizationApprovedModelsQuery = `-- name: ListOrganizationApprovedModels :many
-SELECT id, organization_id, provider_configuration_id, model_id, approved_by_user_id, approved_at FROM weave.organization_approved_models
-WHERE organization_id = $1
-ORDER BY approved_at DESC`;
-async function listOrganizationApprovedModels(client, args) {
+exports.listChatMessagesQuery = `-- name: ListChatMessages :many
+SELECT m.id, m.organization_id, m.session_id, m.agent_run_id, m.role, m.parts, m.created_at
+FROM weave.chat_message m
+         JOIN weave.chat_session s
+              ON s.organization_id = m.organization_id
+                  AND s.id = m.session_id
+WHERE m.organization_id = $1
+  AND m.session_id = $2
+  AND s.created_by_user_id = $3
+ORDER BY m.created_at ASC, m.id ASC
+LIMIT $5::integer
+OFFSET $4::integer`;
+async function listChatMessages(client, args) {
     const result = await client.query({
-        text: exports.listOrganizationApprovedModelsQuery,
-        values: [args.organizationId],
+        text: exports.listChatMessagesQuery,
+        values: [args.organizationId, args.sessionId, args.createdByUserId, args.offsetCount, args.limitCount],
         rowMode: "array"
     });
     return result.rows.map(row => {
         return {
             id: row[0],
             organizationId: row[1],
-            providerConfigurationId: row[2],
-            modelId: row[3],
-            approvedByUserId: row[4],
-            approvedAt: row[5]
+            sessionId: row[2],
+            agentRunId: row[3],
+            role: row[4],
+            parts: row[5],
+            createdAt: row[6]
         };
     });
 }
-exports.getOrganizationApprovedModelQuery = `-- name: GetOrganizationApprovedModel :one
-SELECT id, organization_id, provider_configuration_id, model_id, approved_by_user_id, approved_at FROM weave.organization_approved_models
-WHERE organization_id = $1
-  AND provider_configuration_id = $2
-  AND model_id = $3`;
-async function getOrganizationApprovedModel(client, args) {
+exports.createChatArtifactQuery = `-- name: CreateChatArtifact :one
+WITH visible_session AS (
+    SELECT s.id
+    FROM weave.chat_session s
+    WHERE s.organization_id = $1
+      AND s.id = $5
+      AND s.created_by_user_id = $6
+)
+INSERT INTO weave.chat_artifact (
+    organization_id,
+    session_id,
+    title,
+    kind,
+    language
+)
+SELECT
+    $1,
+    visible_session.id,
+    $2,
+    $3,
+    $4::text
+FROM visible_session
+RETURNING id, organization_id, session_id, title, kind, language, current_version, created_at, updated_at`;
+async function createChatArtifact(client, args) {
     const result = await client.query({
-        text: exports.getOrganizationApprovedModelQuery,
-        values: [args.organizationId, args.providerConfigurationId, args.modelId],
+        text: exports.createChatArtifactQuery,
+        values: [args.organizationId, args.title, args.kind, args.language, args.sessionId, args.createdByUserId],
         rowMode: "array"
     });
     if (result.rows.length !== 1) {
@@ -508,82 +262,34 @@ async function getOrganizationApprovedModel(client, args) {
     return {
         id: row[0],
         organizationId: row[1],
-        providerConfigurationId: row[2],
-        modelId: row[3],
-        approvedByUserId: row[4],
-        approvedAt: row[5]
-    };
-}
-exports.upsertOrganizationChatSettingsQuery = `-- name: UpsertOrganizationChatSettings :one
-INSERT INTO weave.organization_chat_settings (
-    organization_id, default_chat_model_id, default_provider_configuration_id
-) VALUES ($1, $2, $3)
-ON CONFLICT (organization_id) DO UPDATE
-    SET default_chat_model_id = EXCLUDED.default_chat_model_id,
-        default_provider_configuration_id = EXCLUDED.default_provider_configuration_id,
-        updated_at = now()
-RETURNING organization_id, default_chat_model_id, default_provider_configuration_id, updated_at`;
-async function upsertOrganizationChatSettings(client, args) {
-    const result = await client.query({
-        text: exports.upsertOrganizationChatSettingsQuery,
-        values: [args.organizationId, args.defaultChatModelId, args.defaultProviderConfigurationId],
-        rowMode: "array"
-    });
-    if (result.rows.length !== 1) {
-        return null;
-    }
-    const row = result.rows[0];
-    return {
-        organizationId: row[0],
-        defaultChatModelId: row[1],
-        defaultProviderConfigurationId: row[2],
-        updatedAt: row[3]
-    };
-}
-exports.getOrganizationChatSettingsQuery = `-- name: GetOrganizationChatSettings :one
-SELECT organization_id, default_chat_model_id, default_provider_configuration_id, updated_at FROM weave.organization_chat_settings
-WHERE organization_id = $1`;
-async function getOrganizationChatSettings(client, args) {
-    const result = await client.query({
-        text: exports.getOrganizationChatSettingsQuery,
-        values: [args.organizationId],
-        rowMode: "array"
-    });
-    if (result.rows.length !== 1) {
-        return null;
-    }
-    const row = result.rows[0];
-    return {
-        organizationId: row[0],
-        defaultChatModelId: row[1],
-        defaultProviderConfigurationId: row[2],
-        updatedAt: row[3]
+        sessionId: row[2],
+        title: row[3],
+        kind: row[4],
+        language: row[5],
+        currentVersion: row[6],
+        createdAt: row[7],
+        updatedAt: row[8]
     };
 }
-exports.clearOrganizationChatSettingsDefaultIfMatchesQuery = `-- name: ClearOrganizationChatSettingsDefaultIfMatches :exec
-UPDATE weave.organization_chat_settings
-SET default_chat_model_id = NULL,
-    default_provider_configuration_id = NULL,
-    updated_at = now()
-WHERE organization_id = $1
-  AND default_chat_model_id = $2
-  AND default_provider_configuration_id = $3`;
-async function clearOrganizationChatSettingsDefaultIfMatches(client, args) {
-    await client.query({
-        text: exports.clearOrganizationChatSettingsDefaultIfMatchesQuery,
-        values: [args.organizationId, args.defaultChatModelId, args.defaultProviderConfigurationId],
-        rowMode: "array"
-    });
-}
-exports.createChatArtifactQuery = `-- name: CreateChatArtifact :one
-INSERT INTO weave.chat_artifacts (
-    id, session_id, title, kind, language, current_version
-) VALUES ($1, $2, $3, $4, $5, 1)
-RETURNING id, session_id, title, kind, language, current_version, created_at, updated_at`;
-async function createChatArtifact(client, args) {
+exports.bumpChatArtifactVersionQuery = `-- name: BumpChatArtifactVersion :one
+WITH visible_session AS (
+    SELECT s.id
+    FROM weave.chat_session s
+    WHERE s.organization_id = $1
+      AND s.id = $3
+      AND s.created_by_user_id = $4
+)
+UPDATE weave.chat_artifact
+SET current_version = current_version + 1
+FROM visible_session
+WHERE chat_artifact.organization_id = $1
+  AND chat_artifact.session_id = visible_session.id
+  AND chat_artifact.id = $2
+RETURNING chat_artifact.id, chat_artifact.organization_id, chat_artifact.session_id, chat_artifact.title, chat_artifact.kind, chat_artifact.language, chat_artifact.current_version, chat_artifact.created_at, chat_artifact.updated_at`;
+async function bumpChatArtifactVersion(client, args) {
     const result = await client.query({
-        text: exports.createChatArtifactQuery,
-        values: [args.id, args.sessionId, args.title, args.kind, args.language],
+        text: exports.bumpChatArtifactVersionQuery,
+        values: [args.organizationId, args.artifactId, args.sessionId, args.createdByUserId],
         rowMode: "array"
     });
     if (result.rows.length !== 1) {
@@ -592,25 +298,30 @@ async function createChatArtifact(client, args) {
     const row = result.rows[0];
     return {
         id: row[0],
-        sessionId: row[1],
-        title: row[2],
-        kind: row[3],
-        language: row[4],
-        currentVersion: row[5],
-        createdAt: row[6],
-        updatedAt: row[7]
+        organizationId: row[1],
+        sessionId: row[2],
+        title: row[3],
+        kind: row[4],
+        language: row[5],
+        currentVersion: row[6],
+        createdAt: row[7],
+        updatedAt: row[8]
     };
 }
-exports.bumpChatArtifactVersionQuery = `-- name: BumpChatArtifactVersion :one
-UPDATE weave.chat_artifacts
-SET current_version = current_version + 1,
-    updated_at = now()
-WHERE id = $1
-RETURNING id, session_id, title, kind, language, current_version, created_at, updated_at`;
-async function bumpChatArtifactVersion(client, args) {
+exports.getChatArtifactQuery = `-- name: GetChatArtifact :one
+SELECT a.id, a.organization_id, a.session_id, a.title, a.kind, a.language, a.current_version, a.created_at, a.updated_at
+FROM weave.chat_artifact a
+         JOIN weave.chat_session s
+              ON s.organization_id = a.organization_id
+                  AND s.id = a.session_id
+WHERE a.organization_id = $1
+  AND a.session_id = $2
+  AND a.id = $3
+  AND s.created_by_user_id = $4`;
+async function getChatArtifact(client, args) {
     const result = await client.query({
-        text: exports.bumpChatArtifactVersionQuery,
-        values: [args.id],
+        text: exports.getChatArtifactQuery,
+        values: [args.organizationId, args.sessionId, args.artifactId, args.createdByUserId],
         rowMode: "array"
     });
     if (result.rows.length !== 1) {
@@ -619,70 +330,70 @@ async function bumpChatArtifactVersion(client, args) {
     const row = result.rows[0];
     return {
         id: row[0],
-        sessionId: row[1],
-        title: row[2],
-        kind: row[3],
-        language: row[4],
-        currentVersion: row[5],
-        createdAt: row[6],
-        updatedAt: row[7]
+        organizationId: row[1],
+        sessionId: row[2],
+        title: row[3],
+        kind: row[4],
+        language: row[5],
+        currentVersion: row[6],
+        createdAt: row[7],
+        updatedAt: row[8]
     };
 }
-exports.listChatArtifactsForSessionQuery = `-- name: ListChatArtifactsForSession :many
-SELECT id, session_id, title, kind, language, current_version, created_at, updated_at FROM weave.chat_artifacts
-WHERE session_id = $1
-ORDER BY created_at`;
-async function listChatArtifactsForSession(client, args) {
+exports.listChatArtifactsQuery = `-- name: ListChatArtifacts :many
+SELECT a.id, a.organization_id, a.session_id, a.title, a.kind, a.language, a.current_version, a.created_at, a.updated_at
+FROM weave.chat_artifact a
+         JOIN weave.chat_session s
+              ON s.organization_id = a.organization_id
+                  AND s.id = a.session_id
+WHERE a.organization_id = $1
+  AND a.session_id = $2
+  AND s.created_by_user_id = $3
+ORDER BY a.created_at ASC, a.id ASC`;
+async function listChatArtifacts(client, args) {
     const result = await client.query({
-        text: exports.listChatArtifactsForSessionQuery,
-        values: [args.sessionId],
+        text: exports.listChatArtifactsQuery,
+        values: [args.organizationId, args.sessionId, args.createdByUserId],
         rowMode: "array"
     });
     return result.rows.map(row => {
         return {
             id: row[0],
-            sessionId: row[1],
-            title: row[2],
-            kind: row[3],
-            language: row[4],
-            currentVersion: row[5],
-            createdAt: row[6],
-            updatedAt: row[7]
+            organizationId: row[1],
+            sessionId: row[2],
+            title: row[3],
+            kind: row[4],
+            language: row[5],
+            currentVersion: row[6],
+            createdAt: row[7],
+            updatedAt: row[8]
         };
     });
 }
-exports.getChatArtifactQuery = `-- name: GetChatArtifact :one
-SELECT id, session_id, title, kind, language, current_version, created_at, updated_at FROM weave.chat_artifacts WHERE id = $1`;
-async function getChatArtifact(client, args) {
-    const result = await client.query({
-        text: exports.getChatArtifactQuery,
-        values: [args.id],
-        rowMode: "array"
-    });
-    if (result.rows.length !== 1) {
-        return null;
-    }
-    const row = result.rows[0];
-    return {
-        id: row[0],
-        sessionId: row[1],
-        title: row[2],
-        kind: row[3],
-        language: row[4],
-        currentVersion: row[5],
-        createdAt: row[6],
-        updatedAt: row[7]
-    };
-}
-exports.insertChatArtifactVersionQuery = `-- name: InsertChatArtifactVersion :one
-INSERT INTO weave.chat_artifact_versions (
-    id, artifact_id, version, content, created_by_message_id
-) VALUES ($1, $2, $3, $4, $5)
-RETURNING id, artifact_id, version, content, created_by_message_id, created_at, document_references`;
-async function insertChatArtifactVersion(client, args) {
+exports.createChatArtifactVersionQuery = `-- name: CreateChatArtifactVersion :one
+INSERT INTO weave.chat_artifact_version (
+    organization_id,
+    artifact_id,
+    version,
+    content,
+    created_by_agent_run_id,
+    tool_call_id,
+    document_references
+)
+VALUES (
+    $1,
+    $2,
+    $3,
+    $4,
+    $5::uuid,
+    COALESCE($6::text, ''),
+    COALESCE($7::jsonb, '[]'::jsonb)
+)
+RETURNING id, organization_id, artifact_id, version, content, created_by_agent_run_id, tool_call_id, document_references, created_at`;
+async function createChatArtifactVersion(client, args) {
     const result = await client.query({
-        text: exports.insertChatArtifactVersionQuery,
-        values: [args.id, args.artifactId, args.version, args.content, args.createdByMessageId],
+        text: exports.createChatArtifactVersionQuery,
+        values: [args.organizationId, args.artifactId, args.version, args.content, args.createdByAgentRunId, args.toolCallId, args.documentReferences],
         rowMode: "array"
     });
     if (result.rows.length !== 1) {
@@ -691,23 +402,34 @@ async function insertChatArtifactVersion(client, args) {
     const row = result.rows[0];
     return {
         id: row[0],
-        artifactId: row[1],
-        version: row[2],
-        content: row[3],
-        createdByMessageId: row[4],
-        createdAt: row[5],
-        documentReferences: row[6]
+        organizationId: row[1],
+        artifactId: row[2],
+        version: row[3],
+        content: row[4],
+        createdByAgentRunId: row[5],
+        toolCallId: row[6],
+        documentReferences: row[7],
+        createdAt: row[8]
     };
 }
-exports.insertChatArtifactVersionWithDocumentReferencesQuery = `-- name: InsertChatArtifactVersionWithDocumentReferences :one
-INSERT INTO weave.chat_artifact_versions (
-    id, artifact_id, version, content, created_by_message_id, document_references
-) VALUES ($1, $2, $3, $4, $5, $6)
-RETURNING id, artifact_id, version, content, created_by_message_id, created_at, document_references`;
-async function insertChatArtifactVersionWithDocumentReferences(client, args) {
+exports.getChatArtifactVersionQuery = `-- name: GetChatArtifactVersion :one
+SELECT v.id, v.organization_id, v.artifact_id, v.version, v.content, v.created_by_agent_run_id, v.tool_call_id, v.document_references, v.created_at
+FROM weave.chat_artifact_version v
+         JOIN weave.chat_artifact a
+              ON a.organization_id = v.organization_id
+                  AND a.id = v.artifact_id
+         JOIN weave.chat_session s
+              ON s.organization_id = a.organization_id
+                  AND s.id = a.session_id
+WHERE v.organization_id = $1
+  AND a.session_id = $2
+  AND v.artifact_id = $3
+  AND v.version = $4
+  AND s.created_by_user_id = $5`;
+async function getChatArtifactVersion(client, args) {
     const result = await client.query({
-        text: exports.insertChatArtifactVersionWithDocumentReferencesQuery,
-        values: [args.id, args.artifactId, args.version, args.content, args.createdByMessageId, args.documentReferences],
+        text: exports.getChatArtifactVersionQuery,
+        values: [args.organizationId, args.sessionId, args.artifactId, args.version, args.createdByUserId],
         rowMode: "array"
     });
     if (result.rows.length !== 1) {
@@ -716,21 +438,34 @@ async function insertChatArtifactVersionWithDocumentReferences(client, args) {
     const row = result.rows[0];
     return {
         id: row[0],
-        artifactId: row[1],
-        version: row[2],
-        content: row[3],
-        createdByMessageId: row[4],
-        createdAt: row[5],
-        documentReferences: row[6]
+        organizationId: row[1],
+        artifactId: row[2],
+        version: row[3],
+        content: row[4],
+        createdByAgentRunId: row[5],
+        toolCallId: row[6],
+        documentReferences: row[7],
+        createdAt: row[8]
     };
 }
-exports.getChatArtifactVersionQuery = `-- name: GetChatArtifactVersion :one
-SELECT id, artifact_id, version, content, created_by_message_id, created_at, document_references FROM weave.chat_artifact_versions
-WHERE artifact_id = $1 AND version = $2`;
-async function getChatArtifactVersion(client, args) {
+exports.getCurrentChatArtifactVersionQuery = `-- name: GetCurrentChatArtifactVersion :one
+SELECT v.id, v.organization_id, v.artifact_id, v.version, v.content, v.created_by_agent_run_id, v.tool_call_id, v.document_references, v.created_at
+FROM weave.chat_artifact_version v
+         JOIN weave.chat_artifact a
+              ON a.organization_id = v.organization_id
+                  AND a.id = v.artifact_id
+         JOIN weave.chat_session s
+              ON s.organization_id = a.organization_id
+                  AND s.id = a.session_id
+WHERE v.organization_id = $1
+  AND a.session_id = $2
+  AND v.artifact_id = $3
+  AND v.version = a.current_version
+  AND s.created_by_user_id = $4`;
+async function getCurrentChatArtifactVersion(client, args) {
     const result = await client.query({
-        text: exports.getChatArtifactVersionQuery,
-        values: [args.artifactId, args.version],
+        text: exports.getCurrentChatArtifactVersionQuery,
+        values: [args.organizationId, args.sessionId, args.artifactId, args.createdByUserId],
         rowMode: "array"
     });
     if (result.rows.length !== 1) {
@@ -739,33 +474,13 @@ async function getChatArtifactVersion(client, args) {
     const row = result.rows[0];
     return {
         id: row[0],
-        artifactId: row[1],
-        version: row[2],
-        content: row[3],
-        createdByMessageId: row[4],
-        createdAt: row[5],
-        documentReferences: row[6]
+        organizationId: row[1],
+        artifactId: row[2],
+        version: row[3],
+        content: row[4],
+        createdByAgentRunId: row[5],
+        toolCallId: row[6],
+        documentReferences: row[7],
+        createdAt: row[8]
     };
 }
-exports.listChatArtifactVersionsQuery = `-- name: ListChatArtifactVersions :many
-SELECT id, artifact_id, version, content, created_by_message_id, created_at, document_references FROM weave.chat_artifact_versions
-WHERE artifact_id = $1
-ORDER BY version`;
-async function listChatArtifactVersions(client, args) {
-    const result = await client.query({
-        text: exports.listChatArtifactVersionsQuery,
-        values: [args.artifactId],
-        rowMode: "array"
-    });
-    return result.rows.map(row => {
-        return {
-            id: row[0],
-            artifactId: row[1],
-            version: row[2],
-            content: row[3],
-            createdByMessageId: row[4],
-            createdAt: row[5],
-            documentReferences: row[6]
-        };
-    });
-}