Tool factories

2025-12-15 23:04:20 -08:00
parent 0fced1dfa6
commit 26336d0ac2
21 changed files with 14461 additions and 15875 deletions
@@ -1,37 +1,40 @@
 import { type } from "arktype";
-import { contextualize, tool } from "./shared.ts";
+import type { Context } from "../main.ts";
+import { execute, tool } from "./shared.ts";

 export const PullRequestInfo = type({
  pull_number: type.number.describe("The pull request number to fetch"),
 });

-export const PullRequestInfoTool = tool({
-  name: "get_pull_request",
-  description:
-    "Retrieve PR information (metadata only). PR branch is already checked out during setup.",
-  parameters: PullRequestInfo,
-  execute: contextualize(async ({ pull_number }, ctx) => {
-    const pr = await ctx.octokit.rest.pulls.get({
-      owner: ctx.owner,
-      repo: ctx.name,
-      pull_number,
-    });
+export function PullRequestInfoTool(ctx: Context) {
+  return tool({
+    name: "get_pull_request",
+    description:
+      "Retrieve PR information (metadata only). PR branch is already checked out during setup.",
+    parameters: PullRequestInfo,
+    execute: execute(ctx, async ({ pull_number }) => {
+      const pr = await ctx.octokit.rest.pulls.get({
+        owner: ctx.owner,
+        repo: ctx.name,
+        pull_number,
+      });

-    const data = pr.data;
+      const data = pr.data;

-    // detect fork PRs - head repo differs from base repo
-    const isFork = data.head.repo.full_name !== data.base.repo.full_name;
+      // detect fork PRs - head repo differs from base repo
+      const isFork = data.head.repo.full_name !== data.base.repo.full_name;

-    return {
-      number: data.number,
-      url: data.html_url,
-      title: data.title,
-      state: data.state,
-      draft: data.draft,
-      merged: data.merged,
-      base: data.base.ref,
-      head: data.head.ref,
-      isFork,
-    };
-  }),
-});
+      return {
+        number: data.number,
+        url: data.html_url,
+        title: data.title,
+        state: data.state,
+        draft: data.draft,
+        merged: data.merged,
+        base: data.base.ref,
+        head: data.head.ref,
+        isFork,
+      };
+    }),
+  });
+}