fix: add missing tools for automation generation and status generation

Author: ochmanski

apps/api/src/workflows/schedules/generate-schedule/generate-schedule.ts

@@ -4,8 +4,14 @@ import type { Db } from "../../../db/db";
 import { createOrganizationScheduleTool } from "../../tools/create-organization-schedule-tool";
 import { getDiscordIntegrationTool } from "../../tools/get-discord-integration-tool";
 import { getGithubIntegrationTool } from "../../tools/get-github-integration-tool";
+import { getGitlabIntegrationTool } from "../../tools/get-gitlab-integration-tool";
+import { getLinearIntegrationTool } from "../../tools/get-linear-integration-tool";
 import { getSlackIntegrationTool } from "../../tools/get-slack-integration-tool";
 import { listOrganizationDiscordChannelsTool } from "../../tools/list-organization-discord-channels-tool";
+import { listOrganizationGithubRepositoriesTool } from "../../tools/list-organization-github-repositories-tool";
+import { listOrganizationGitlabProjectsTool } from "../../tools/list-organization-gitlab-projects-tool";
+import { listOrganizationLinearProjectsTool } from "../../tools/list-organization-linear-projects-tool";
+import { listOrganizationLinearTeamsTool } from "../../tools/list-organization-linear-teams-tool";
 import { listOrganizationMembersTool } from "../../tools/list-organization-members-tool";
 import { listOrganizationSlackChannelsTool } from "../../tools/list-organization-slack-channels-tool";
 import { listOrganizationTeamsTool } from "../../tools/list-organization-teams-tool";
@@ -50,11 +56,17 @@ Request: "${naturalLanguageRequest}"`,
     tools: {
       listOrganizationTeams: listOrganizationTeamsTool(db),
       listOrganizationMembers: listOrganizationMembersTool(db),
+      listOrganizationGithubRepositories: listOrganizationGithubRepositoriesTool(db),
       listOrganizationSlackChannels: listOrganizationSlackChannelsTool(db),
       listOrganizationDiscordChannels: listOrganizationDiscordChannelsTool(db),
+      listOrganizationGitlabProjects: listOrganizationGitlabProjectsTool(db),
+      listOrganizationLinearTeams: listOrganizationLinearTeamsTool(db),
+      listOrganizationLinearProjects: listOrganizationLinearProjectsTool(db),
       getSlackIntegration: getSlackIntegrationTool(db),
       getDiscordIntegration: getDiscordIntegrationTool(db),
       getGithubIntegration: getGithubIntegrationTool(db),
+      getGitlabIntegration: getGitlabIntegrationTool(db),
+      getLinearIntegration: getLinearIntegrationTool(db),
       createOrganizationSchedule: createOrganizationScheduleTool(db),
     },
   });

apps/api/src/workflows/schedules/generate-schedule/system-prompt.ts

@@ -31,11 +31,13 @@ GENERATE TARGETS (for generateUpdates): array generateFor[] where each item is o
 UsingActivityFrom is an array of discriminated objects (not strings). Each item is exactly one of:
 - { type: "anyIntegration", value: "anyIntegration" }
 - { type: "anyGithub", value: "anyGithub" }
+- { type: "anyGitlab", value: "anyGitlab" }
 - { type: "anySlack", value: "anySlack" }
 - { type: "anyDiscord", value: "anyDiscord" }
 - { type: "anyLinear", value: "anyLinear" }
 - { type: "slackChannel", value: slackChannelId }
 - { type: "githubRepository", value: githubRepositoryId }
+- { type: "gitlabProject", value: gitlabProjectId }
 - { type: "discordChannel", value: discordChannelId }
 - { type: "linearTeam", value: linearTeamId }
 - { type: "linearProject", value: linearProjectId }
@@ -45,11 +47,13 @@ SUMMARY TARGETS (for sendSummaries): array summaryFor[] where each item is one o
 - { type: "member", value: memberId }
 - { type: "team", value: teamId }
 - { type: "anyGithub", value: "anyGithub" }
+- { type: "anyGitlab", value: "anyGitlab" }
 - { type: "anySlack", value: "anySlack" }
 - { type: "anyDiscord", value: "anyDiscord" }
 - { type: "anyLinear", value: "anyLinear" }
 - { type: "slackChannel", value: slackChannelId }
 - { type: "githubRepository", value: githubRepositoryId }
+- { type: "gitlabProject", value: gitlabProjectId }
 - { type: "discordChannel", value: discordChannelId }
 - { type: "linearTeam", value: linearTeamId }
 - { type: "linearProject", value: linearProjectId }
@@ -61,12 +65,14 @@ RULES:
     - For generation: add { type: "organization", value: organizationSlug, usingActivityFrom: [...] }
   * If a specific team is mentioned (e.g., "Engineering team"), resolve to { type: "team", value: teamId }
   * If a specific person is mentioned, resolve to { type: "member", value: memberId }
-- Never invent IDs. Use list tools to resolve: members, teams, Slack/Discord channels. Use integration tools for github/slack/discord integration IDs.
+- Never invent IDs. Use list tools to resolve: members, teams, Slack/Discord channels. Use integration tools for github/slack/discord/gitlab/linear integration IDs.
 - Validate timezone and time format; if missing, default timezone to UTC and timeOfDay to 09:00.
 - If recurrence not specified, default to daily.
 - If weekly/monthly, ensure the correct day value exists; if not provided, choose the nearest valid future day based on current day.
 - For send routes (slackChannel/discordChannel), resolve channels by exact name match; prefer public channels when multiple; if ambiguity remains, pick the first and proceed.
 - If you cannot resolve a specific GitHub repository, prefer { type: "anyGithub", value: "anyGithub" }.
+- If you cannot resolve a specific GitLab project, prefer { type: "anyGitlab", value: "anyGitlab" }.
+- If you cannot resolve a specific Linear team or project, prefer { type: "anyLinear", value: "anyLinear" }.
 - After constructing a typed config, call create-organization-schedule (this will also create the initial run if active).
 - Return the created schedule id and the initial run id from the tool response.
 `;

apps/api/src/workflows/status-updates/generate-status-update/generate-status-update.ts

@@ -12,12 +12,15 @@ import { getDiscordServerTool } from "../../tools/get-discord-server-tool";
 import { getDiscordUserTool } from "../../tools/get-discord-user-tool";
 import { getExistingStatusUpdateItemsTool } from "../../tools/get-existing-status-update-items-tool";
 import { getGithubEventDetailTool } from "../../tools/get-github-event-detail-tool";
+import { getGithubIntegrationTool } from "../../tools/get-github-integration-tool";
 import { getGithubRepositoryTool } from "../../tools/get-github-repository-tool";
 import { getGithubUserTool } from "../../tools/get-github-user-tool";
 import { getGitlabEventDetailTool } from "../../tools/get-gitlab-event-detail-tool";
+import { getGitlabIntegrationTool } from "../../tools/get-gitlab-integration-tool";
 import { getGitlabProjectTool } from "../../tools/get-gitlab-project-tool";
 import { getGitlabUserTool } from "../../tools/get-gitlab-user-tool";
 import { getLinearEventDetailTool } from "../../tools/get-linear-event-detail-tool";
+import { getLinearIntegrationTool } from "../../tools/get-linear-integration-tool";
 import { getLinearIssueTool } from "../../tools/get-linear-issue-tool";
 import { getLinearProjectTool } from "../../tools/get-linear-project-tool";
 import { getLinearTeamTool } from "../../tools/get-linear-team-tool";
@@ -127,11 +130,13 @@ Available integrations: ${JSON.stringify(availableIntegrations)}.`,
       getGitHubEventDetail: getGithubEventDetailTool(db),
       getGitHubUser: getGithubUserTool(db),
       getGitHubRepository: getGithubRepositoryTool(db),
+      getGithubIntegration: getGithubIntegrationTool(db),
 
       getMemberGitlabEvents: getMemberGitlabEventsTool(db),
       getGitlabEventDetail: getGitlabEventDetailTool(db),
       getGitlabUser: getGitlabUserTool(db),
       getGitlabProject: getGitlabProjectTool(db),
+      getGitlabIntegration: getGitlabIntegrationTool(db),
 
       getMemberSlackEvents: getMemberSlackEventsTool(db),
       getSlackEventDetail: getSlackEventDetailTool(db),
@@ -152,6 +157,7 @@ Available integrations: ${JSON.stringify(availableIntegrations)}.`,
       getLinearTeam: getLinearTeamTool(db),
       getLinearProject: getLinearProjectTool(db),
       getLinearIssue: getLinearIssueTool(db),
+      getLinearIntegration: getLinearIntegrationTool(db),
     },
   });
 

apps/api/src/workflows/tools/get-linear-integration-tool.ts

@@ -0,0 +1,29 @@
+import { tool } from "ai";
+import { eq } from "drizzle-orm";
+import { z } from "zod";
+import * as schema from "../../db";
+import type { Db } from "../../db/db";
+
+export function getLinearIntegrationTool(db: Db) {
+  return tool({
+    description: `Get Linear integration details including team info for constructing links or defaults.`,
+    parameters: z.object({
+      organizationId: z.string().describe("The organization ID to get the Linear integration for"),
+    }),
+    execute: async (params) => {
+      const integration = await db
+        .select({
+          id: schema.linearIntegration.id,
+          teamId: schema.linearIntegration.teamId,
+          teamName: schema.linearIntegration.teamName,
+          teamKey: schema.linearIntegration.teamKey,
+          createdAt: schema.linearIntegration.createdAt,
+          updatedAt: schema.linearIntegration.updatedAt,
+        })
+        .from(schema.linearIntegration)
+        .where(eq(schema.linearIntegration.organizationId, params.organizationId))
+        .limit(1);
+      return integration[0] || null;
+    },
+  });
+}

apps/api/src/workflows/tools/list-organization-github-repositories-tool.ts

@@ -0,0 +1,35 @@
+import { tool } from "ai";
+import { eq } from "drizzle-orm";
+import { z } from "zod";
+import * as schema from "../../db";
+import type { Db } from "../../db/db";
+
+export function listOrganizationGithubRepositoriesTool(db: Db) {
+  return tool({
+    description: `List GitHub repositories for an organization's GitHub integration. Use to map names to repository IDs.`,
+    parameters: z.object({
+      organizationId: z.string().describe("The organization ID that owns the GitHub integration"),
+    }),
+    execute: async (params) => {
+      const repositories = await db
+        .select({
+          id: schema.githubRepository.id,
+          repoId: schema.githubRepository.repoId,
+          name: schema.githubRepository.name,
+          owner: schema.githubRepository.owner,
+          fullName: schema.githubRepository.fullName,
+          private: schema.githubRepository.private,
+          htmlUrl: schema.githubRepository.htmlUrl,
+          createdAt: schema.githubRepository.createdAt,
+        })
+        .from(schema.githubRepository)
+        .innerJoin(
+          schema.githubIntegration,
+          eq(schema.githubRepository.integrationId, schema.githubIntegration.id),
+        )
+        .where(eq(schema.githubIntegration.organizationId, params.organizationId));
+
+      return repositories;
+    },
+  });
+}

apps/api/src/workflows/tools/list-organization-gitlab-projects-tool.ts

@@ -0,0 +1,34 @@
+import { tool } from "ai";
+import { eq } from "drizzle-orm";
+import { z } from "zod";
+import * as schema from "../../db";
+import type { Db } from "../../db/db";
+
+export function listOrganizationGitlabProjectsTool(db: Db) {
+  return tool({
+    description: `List GitLab projects for an organization's GitLab integration. Use to map names to project IDs.`,
+    parameters: z.object({
+      organizationId: z.string().describe("The organization ID that owns the GitLab integration"),
+    }),
+    execute: async (params) => {
+      const projects = await db
+        .select({
+          id: schema.gitlabProject.id,
+          projectId: schema.gitlabProject.projectId,
+          name: schema.gitlabProject.name,
+          pathWithNamespace: schema.gitlabProject.pathWithNamespace,
+          visibility: schema.gitlabProject.visibility,
+          webUrl: schema.gitlabProject.webUrl,
+          createdAt: schema.gitlabProject.createdAt,
+        })
+        .from(schema.gitlabProject)
+        .innerJoin(
+          schema.gitlabIntegration,
+          eq(schema.gitlabProject.integrationId, schema.gitlabIntegration.id),
+        )
+        .where(eq(schema.gitlabIntegration.organizationId, params.organizationId));
+
+      return projects;
+    },
+  });
+}

apps/api/src/workflows/tools/list-organization-linear-projects-tool.ts

@@ -0,0 +1,33 @@
+import { tool } from "ai";
+import { eq } from "drizzle-orm";
+import { z } from "zod";
+import * as schema from "../../db";
+import type { Db } from "../../db/db";
+
+export function listOrganizationLinearProjectsTool(db: Db) {
+  return tool({
+    description: `List Linear projects for an organization's Linear integration. Use to map project names to IDs.`,
+    parameters: z.object({
+      organizationId: z.string().describe("The organization ID that owns the Linear integration"),
+    }),
+    execute: async (params) => {
+      const projects = await db
+        .select({
+          id: schema.linearProject.id,
+          projectId: schema.linearProject.projectId,
+          teamId: schema.linearProject.teamId,
+          name: schema.linearProject.name,
+          key: schema.linearProject.key,
+          createdAt: schema.linearProject.createdAt,
+        })
+        .from(schema.linearProject)
+        .innerJoin(
+          schema.linearIntegration,
+          eq(schema.linearProject.integrationId, schema.linearIntegration.id),
+        )
+        .where(eq(schema.linearIntegration.organizationId, params.organizationId));
+
+      return projects;
+    },
+  });
+}

apps/api/src/workflows/tools/list-organization-linear-teams-tool.ts

@@ -0,0 +1,34 @@
+import { tool } from "ai";
+import { eq } from "drizzle-orm";
+import { z } from "zod";
+import * as schema from "../../db";
+import type { Db } from "../../db/db";
+
+export function listOrganizationLinearTeamsTool(db: Db) {
+  return tool({
+    description: `List Linear teams for an organization's Linear integration. Use to map team names to IDs.`,
+    parameters: z.object({
+      organizationId: z.string().describe("The organization ID that owns the Linear integration"),
+    }),
+    execute: async (params) => {
+      const teams = await db
+        .select({
+          id: schema.linearTeam.id,
+          teamId: schema.linearTeam.teamId,
+          name: schema.linearTeam.name,
+          key: schema.linearTeam.key,
+          timezone: schema.linearTeam.timezone,
+          private: schema.linearTeam.private,
+          createdAt: schema.linearTeam.createdAt,
+        })
+        .from(schema.linearTeam)
+        .innerJoin(
+          schema.linearIntegration,
+          eq(schema.linearTeam.integrationId, schema.linearIntegration.id),
+        )
+        .where(eq(schema.linearIntegration.organizationId, params.organizationId));
+
+      return teams;
+    },
+  });
+}