agent-relay 1.3.1 → 1.3.3

package/dist/cloud/db/schema.d.ts

@@ -423,6 +423,7 @@ export declare const githubInstallations: import("drizzle-orm/pg-core").PgTableW
             identity: undefined;
             generated: undefined;
         }, {}, {
+            size: undefined;
             baseBuilder: import("drizzle-orm/pg-core").PgColumnBuilder<{
                 name: "events";
                 dataType: "string";
@@ -431,7 +432,6 @@ export declare const githubInstallations: import("drizzle-orm/pg-core").PgTableW
                 enumValues: [string, ...string[]];
                 driverParam: string;
             }, {}, {}, import("drizzle-orm").ColumnBuilderExtraConfig>;
-            size: undefined;
         }>;
         suspended: import("drizzle-orm/pg-core").PgColumn<{
             name: "suspended";
@@ -617,6 +617,7 @@ export declare const credentials: import("drizzle-orm/pg-core").PgTableWithColum
             identity: undefined;
             generated: undefined;
         }, {}, {
+            size: undefined;
             baseBuilder: import("drizzle-orm/pg-core").PgColumnBuilder<{
                 name: "scopes";
                 dataType: "string";
@@ -625,7 +626,6 @@ export declare const credentials: import("drizzle-orm/pg-core").PgTableWithColum
                 enumValues: [string, ...string[]];
                 driverParam: string;
             }, {}, {}, import("drizzle-orm").ColumnBuilderExtraConfig>;
-            size: undefined;
         }>;
         providerAccountId: import("drizzle-orm/pg-core").PgColumn<{
             name: "provider_account_id";
@@ -993,6 +993,8 @@ export declare const workspacesRelations: import("drizzle-orm").Relations<"works
     user: import("drizzle-orm").One<"users", true>;
     members: import("drizzle-orm").Many<"workspace_members">;
     repositories: import("drizzle-orm").Many<"repositories">;
+    linkedDaemons: import("drizzle-orm").Many<"linked_daemons">;
+    messages: import("drizzle-orm").Many<"agent_messages">;
 }>;
 export declare const workspaceMembers: import("drizzle-orm/pg-core").PgTableWithColumns<{
     name: "workspace_members";
@@ -1654,6 +1656,23 @@ export declare const linkedDaemons: import("drizzle-orm/pg-core").PgTableWithCol
             identity: undefined;
             generated: undefined;
         }, {}, {}>;
+        workspaceId: import("drizzle-orm/pg-core").PgColumn<{
+            name: "workspace_id";
+            tableName: "linked_daemons";
+            dataType: "string";
+            columnType: "PgUUID";
+            data: string;
+            driverParam: string;
+            notNull: false;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: undefined;
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {}>;
         name: import("drizzle-orm/pg-core").PgColumn<{
             name: "name";
             tableName: "linked_daemons";
@@ -1837,6 +1856,8 @@ export declare const linkedDaemons: import("drizzle-orm/pg-core").PgTableWithCol
 }>;
 export declare const linkedDaemonsRelations: import("drizzle-orm").Relations<"linked_daemons", {
     user: import("drizzle-orm").One<"users", true>;
+    workspace: import("drizzle-orm").One<"workspaces", false>;
+    messages: import("drizzle-orm").Many<"agent_messages">;
 }>;
 export declare const subscriptions: import("drizzle-orm/pg-core").PgTableWithColumns<{
     name: "subscriptions";
@@ -4054,6 +4075,7 @@ export declare const issueAssignments: import("drizzle-orm/pg-core").PgTableWith
             identity: undefined;
             generated: undefined;
         }, {}, {
+            size: undefined;
             baseBuilder: import("drizzle-orm/pg-core").PgColumnBuilder<{
                 name: "labels";
                 dataType: "string";
@@ -4062,7 +4084,6 @@ export declare const issueAssignments: import("drizzle-orm/pg-core").PgTableWith
                 enumValues: [string, ...string[]];
                 driverParam: string;
             }, {}, {}, import("drizzle-orm").ColumnBuilderExtraConfig>;
-            size: undefined;
         }>;
         priority: import("drizzle-orm/pg-core").PgColumn<{
             name: "priority";
@@ -4484,4 +4505,377 @@ export type IssueAssignment = typeof issueAssignments.$inferSelect;
 export type NewIssueAssignment = typeof issueAssignments.$inferInsert;
 export type CommentMention = typeof commentMentions.$inferSelect;
 export type NewCommentMention = typeof commentMentions.$inferInsert;
+/**
+ * Message payload metadata (mirrors SendMeta from protocol)
+ */
+export interface MessagePayloadMeta {
+    requires_ack?: boolean;
+    ttl_ms?: number;
+    importance?: number;
+    replyTo?: string;
+}
+/**
+ * Agent messages table - stores all relay messages for search and history.
+ *
+ * Retention policy:
+ * - Free tier: 30 days (enforced via expires_at)
+ * - Pro tier: 90 days
+ * - Enterprise: Unlimited (no expires_at set)
+ *
+ * Messages are synced from daemon SQLite to cloud PostgreSQL.
+ */
+export declare const agentMessages: import("drizzle-orm/pg-core").PgTableWithColumns<{
+    name: "agent_messages";
+    schema: undefined;
+    columns: {
+        id: import("drizzle-orm/pg-core").PgColumn<{
+            name: "id";
+            tableName: "agent_messages";
+            dataType: "string";
+            columnType: "PgUUID";
+            data: string;
+            driverParam: string;
+            notNull: true;
+            hasDefault: true;
+            isPrimaryKey: true;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: undefined;
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {}>;
+        workspaceId: import("drizzle-orm/pg-core").PgColumn<{
+            name: "workspace_id";
+            tableName: "agent_messages";
+            dataType: "string";
+            columnType: "PgUUID";
+            data: string;
+            driverParam: string;
+            notNull: true;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: undefined;
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {}>;
+        daemonId: import("drizzle-orm/pg-core").PgColumn<{
+            name: "daemon_id";
+            tableName: "agent_messages";
+            dataType: "string";
+            columnType: "PgUUID";
+            data: string;
+            driverParam: string;
+            notNull: false;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: undefined;
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {}>;
+        originalId: import("drizzle-orm/pg-core").PgColumn<{
+            name: "original_id";
+            tableName: "agent_messages";
+            dataType: "string";
+            columnType: "PgVarchar";
+            data: string;
+            driverParam: string;
+            notNull: true;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: [string, ...string[]];
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {
+            length: 255;
+        }>;
+        fromAgent: import("drizzle-orm/pg-core").PgColumn<{
+            name: "from_agent";
+            tableName: "agent_messages";
+            dataType: "string";
+            columnType: "PgVarchar";
+            data: string;
+            driverParam: string;
+            notNull: true;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: [string, ...string[]];
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {
+            length: 255;
+        }>;
+        toAgent: import("drizzle-orm/pg-core").PgColumn<{
+            name: "to_agent";
+            tableName: "agent_messages";
+            dataType: "string";
+            columnType: "PgVarchar";
+            data: string;
+            driverParam: string;
+            notNull: true;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: [string, ...string[]];
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {
+            length: 255;
+        }>;
+        body: import("drizzle-orm/pg-core").PgColumn<{
+            name: "body";
+            tableName: "agent_messages";
+            dataType: "string";
+            columnType: "PgText";
+            data: string;
+            driverParam: string;
+            notNull: true;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: [string, ...string[]];
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {}>;
+        kind: import("drizzle-orm/pg-core").PgColumn<{
+            name: "kind";
+            tableName: "agent_messages";
+            dataType: "string";
+            columnType: "PgVarchar";
+            data: string;
+            driverParam: string;
+            notNull: true;
+            hasDefault: true;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: [string, ...string[]];
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {
+            length: 50;
+        }>;
+        topic: import("drizzle-orm/pg-core").PgColumn<{
+            name: "topic";
+            tableName: "agent_messages";
+            dataType: "string";
+            columnType: "PgVarchar";
+            data: string;
+            driverParam: string;
+            notNull: false;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: [string, ...string[]];
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {
+            length: 255;
+        }>;
+        thread: import("drizzle-orm/pg-core").PgColumn<{
+            name: "thread";
+            tableName: "agent_messages";
+            dataType: "string";
+            columnType: "PgVarchar";
+            data: string;
+            driverParam: string;
+            notNull: false;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: [string, ...string[]];
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {
+            length: 255;
+        }>;
+        channel: import("drizzle-orm/pg-core").PgColumn<{
+            name: "channel";
+            tableName: "agent_messages";
+            dataType: "string";
+            columnType: "PgVarchar";
+            data: string;
+            driverParam: string;
+            notNull: false;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: [string, ...string[]];
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {
+            length: 255;
+        }>;
+        isBroadcast: import("drizzle-orm/pg-core").PgColumn<{
+            name: "is_broadcast";
+            tableName: "agent_messages";
+            dataType: "boolean";
+            columnType: "PgBoolean";
+            data: boolean;
+            driverParam: boolean;
+            notNull: true;
+            hasDefault: true;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: undefined;
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {}>;
+        isUrgent: import("drizzle-orm/pg-core").PgColumn<{
+            name: "is_urgent";
+            tableName: "agent_messages";
+            dataType: "boolean";
+            columnType: "PgBoolean";
+            data: boolean;
+            driverParam: boolean;
+            notNull: true;
+            hasDefault: true;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: undefined;
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {}>;
+        data: import("drizzle-orm/pg-core").PgColumn<{
+            name: "data";
+            tableName: "agent_messages";
+            dataType: "json";
+            columnType: "PgJsonb";
+            data: Record<string, unknown>;
+            driverParam: unknown;
+            notNull: false;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: undefined;
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {
+            $type: Record<string, unknown>;
+        }>;
+        payloadMeta: import("drizzle-orm/pg-core").PgColumn<{
+            name: "payload_meta";
+            tableName: "agent_messages";
+            dataType: "json";
+            columnType: "PgJsonb";
+            data: MessagePayloadMeta;
+            driverParam: unknown;
+            notNull: false;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: undefined;
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {
+            $type: MessagePayloadMeta;
+        }>;
+        messageTs: import("drizzle-orm/pg-core").PgColumn<{
+            name: "message_ts";
+            tableName: "agent_messages";
+            dataType: "date";
+            columnType: "PgTimestamp";
+            data: Date;
+            driverParam: string;
+            notNull: true;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: undefined;
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {}>;
+        createdAt: import("drizzle-orm/pg-core").PgColumn<{
+            name: "created_at";
+            tableName: "agent_messages";
+            dataType: "date";
+            columnType: "PgTimestamp";
+            data: Date;
+            driverParam: string;
+            notNull: true;
+            hasDefault: true;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: undefined;
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {}>;
+        expiresAt: import("drizzle-orm/pg-core").PgColumn<{
+            name: "expires_at";
+            tableName: "agent_messages";
+            dataType: "date";
+            columnType: "PgTimestamp";
+            data: Date;
+            driverParam: string;
+            notNull: false;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: undefined;
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {}>;
+        indexedAt: import("drizzle-orm/pg-core").PgColumn<{
+            name: "indexed_at";
+            tableName: "agent_messages";
+            dataType: "date";
+            columnType: "PgTimestamp";
+            data: Date;
+            driverParam: string;
+            notNull: false;
+            hasDefault: false;
+            isPrimaryKey: false;
+            isAutoincrement: false;
+            hasRuntimeDefault: false;
+            enumValues: undefined;
+            baseColumn: never;
+            identity: undefined;
+            generated: undefined;
+        }, {}, {}>;
+    };
+    dialect: "pg";
+}>;
+export declare const agentMessagesRelations: import("drizzle-orm").Relations<"agent_messages", {
+    workspace: import("drizzle-orm").One<"workspaces", true>;
+    daemon: import("drizzle-orm").One<"linked_daemons", false>;
+}>;
+export type AgentMessage = typeof agentMessages.$inferSelect;
+export type NewAgentMessage = typeof agentMessages.$inferInsert;
 //# sourceMappingURL=schema.d.ts.map

package/dist/cloud/db/schema.js

@@ -118,6 +118,8 @@ export const workspacesRelations = relations(workspaces, ({ one, many }) => ({
     }),
     members: many(workspaceMembers),
     repositories: many(repositories),
+    linkedDaemons: many(linkedDaemons),
+    messages: many(agentMessages),
 }));
 // ============================================================================
 // Workspace Members
@@ -229,6 +231,7 @@ export const repositoriesRelations = relations(repositories, ({ one }) => ({
 export const linkedDaemons = pgTable('linked_daemons', {
     id: uuid('id').primaryKey().defaultRandom(),
     userId: uuid('user_id').notNull().references(() => users.id, { onDelete: 'cascade' }),
+    workspaceId: uuid('workspace_id').references(() => workspaces.id, { onDelete: 'set null' }),
     name: varchar('name', { length: 255 }).notNull(),
     machineId: varchar('machine_id', { length: 255 }).notNull(),
     apiKeyHash: varchar('api_key_hash', { length: 255 }).notNull(),
@@ -242,14 +245,20 @@ export const linkedDaemons = pgTable('linked_daemons', {
 }, (table) => ({
     userMachineIdx: unique('linked_daemons_user_machine_unique').on(table.userId, table.machineId),
     userIdIdx: index('idx_linked_daemons_user_id').on(table.userId),
+    workspaceIdIdx: index('idx_linked_daemons_workspace_id').on(table.workspaceId),
     apiKeyHashIdx: index('idx_linked_daemons_api_key_hash').on(table.apiKeyHash),
     statusIdx: index('idx_linked_daemons_status').on(table.status),
 }));
-export const linkedDaemonsRelations = relations(linkedDaemons, ({ one }) => ({
+export const linkedDaemonsRelations = relations(linkedDaemons, ({ one, many }) => ({
     user: one(users, {
         fields: [linkedDaemons.userId],
         references: [users.id],
     }),
+    workspace: one(workspaces, {
+        fields: [linkedDaemons.workspaceId],
+        references: [workspaces.id],
+    }),
+    messages: many(agentMessages),
 }));
 // ============================================================================
 // Subscriptions (billing)
@@ -545,4 +554,69 @@ export const commentMentionsRelations = relations(commentMentions, ({ one }) =>
         references: [repositories.id],
     }),
 }));
+/**
+ * Agent messages table - stores all relay messages for search and history.
+ *
+ * Retention policy:
+ * - Free tier: 30 days (enforced via expires_at)
+ * - Pro tier: 90 days
+ * - Enterprise: Unlimited (no expires_at set)
+ *
+ * Messages are synced from daemon SQLite to cloud PostgreSQL.
+ */
+export const agentMessages = pgTable('agent_messages', {
+    id: uuid('id').primaryKey().defaultRandom(),
+    // Scoping
+    workspaceId: uuid('workspace_id').notNull().references(() => workspaces.id, { onDelete: 'cascade' }),
+    daemonId: uuid('daemon_id').references(() => linkedDaemons.id, { onDelete: 'set null' }),
+    // Original message ID from daemon (for deduplication)
+    originalId: varchar('original_id', { length: 255 }).notNull(),
+    // Core message fields (aligned with StoredMessage)
+    fromAgent: varchar('from_agent', { length: 255 }).notNull(),
+    toAgent: varchar('to_agent', { length: 255 }).notNull(), // '*' for broadcast
+    body: text('body').notNull(),
+    // Message classification
+    kind: varchar('kind', { length: 50 }).notNull().default('message'), // message, action, state, thinking
+    topic: varchar('topic', { length: 255 }),
+    thread: varchar('thread', { length: 255 }), // Thread ID for grouping
+    channel: varchar('channel', { length: 255 }), // Channel name if channel message
+    // Flags
+    isBroadcast: boolean('is_broadcast').notNull().default(false),
+    isUrgent: boolean('is_urgent').notNull().default(false),
+    // Optional structured data
+    data: jsonb('data').$type(),
+    payloadMeta: jsonb('payload_meta').$type(),
+    // Timestamps
+    messageTs: timestamp('message_ts').notNull(), // Original message timestamp from daemon
+    createdAt: timestamp('created_at').defaultNow().notNull(), // When synced to cloud
+    expiresAt: timestamp('expires_at'), // Retention policy - null = never expires
+    // Search/indexing tracking
+    indexedAt: timestamp('indexed_at'), // When sent to Algolia (null = not indexed)
+}, (table) => ({
+    // Primary lookups
+    workspaceIdIdx: index('idx_agent_messages_workspace_id').on(table.workspaceId),
+    daemonIdIdx: index('idx_agent_messages_daemon_id').on(table.daemonId),
+    // Deduplication - prevent duplicate sync
+    workspaceOriginalIdx: unique('agent_messages_workspace_original_unique').on(table.workspaceId, table.originalId),
+    // Query patterns
+    fromAgentIdx: index('idx_agent_messages_from_agent').on(table.fromAgent),
+    toAgentIdx: index('idx_agent_messages_to_agent').on(table.toAgent),
+    threadIdx: index('idx_agent_messages_thread').on(table.thread),
+    channelIdx: index('idx_agent_messages_channel').on(table.channel),
+    messageTsIdx: index('idx_agent_messages_message_ts').on(table.messageTs),
+    // Retention cleanup
+    expiresAtIdx: index('idx_agent_messages_expires_at').on(table.expiresAt),
+    // Search indexing queue
+    indexedAtIdx: index('idx_agent_messages_indexed_at').on(table.indexedAt),
+}));
+export const agentMessagesRelations = relations(agentMessages, ({ one }) => ({
+    workspace: one(workspaces, {
+        fields: [agentMessages.workspaceId],
+        references: [workspaces.id],
+    }),
+    daemon: one(linkedDaemons, {
+        fields: [agentMessages.daemonId],
+        references: [linkedDaemons.id],
+    }),
+}));
 //# sourceMappingURL=schema.js.map

package/dist/cloud/provisioner/index.d.ts

@@ -70,6 +70,10 @@ export declare class WorkspaceProvisioner {
      * Restart a workspace
      */
     restart(workspaceId: string): Promise<void>;
+    /**
+     * Update environment variables for a workspace instance.
+     */
+    setWorkspaceEnvVars(workspace: Workspace, envVars: Record<string, string>): Promise<void>;
     /**
      * Stop a workspace
      */
@@ -130,6 +134,7 @@ export declare class WorkspaceProvisioner {
         readonly UPDATED: "updated";
         readonly UPDATED_PENDING_RESTART: "updated_pending_restart";
         readonly SKIPPED_ACTIVE_AGENTS: "skipped_active_agents";
+        readonly SKIPPED_VERIFICATION_FAILED: "skipped_verification_failed";
         readonly SKIPPED_NOT_RUNNING: "skipped_not_running";
         readonly NOT_SUPPORTED: "not_supported";
         readonly ERROR: "error";
@@ -160,6 +165,7 @@ export declare class WorkspaceProvisioner {
             status: string;
         }>;
         error?: string;
+        reason?: string;
     }>;
     /**
      * Gracefully update all workspaces to a new image
@@ -183,6 +189,7 @@ export declare class WorkspaceProvisioner {
             updated: number;
             pendingRestart: number;
             skippedActiveAgents: number;
+            skippedVerificationFailed: number;
             skippedNotRunning: number;
             errors: number;
         };
@@ -192,6 +199,7 @@ export declare class WorkspaceProvisioner {
             machineState?: string;
             agentCount?: number;
             error?: string;
+            reason?: string;
         }>;
     }>;
 }